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 2379396SMatthew.Ahrens@Sun.COM assert(zfs_prop_user(propname) || prop != ZPROP_INVAL); 2389396SMatthew.Ahrens@Sun.COM 2395367Sahrens if (!zfs_prop_user(propname) && zfs_prop_readonly(prop)) 2405367Sahrens continue; 2415367Sahrens 2425367Sahrens verify(nvpair_value_nvlist(elem, &propnv) == 0); 24310195SSanjeev.Bagewadi@Sun.COM if (prop == ZFS_PROP_QUOTA || prop == ZFS_PROP_RESERVATION || 24410195SSanjeev.Bagewadi@Sun.COM prop == ZFS_PROP_REFQUOTA || 24510195SSanjeev.Bagewadi@Sun.COM prop == ZFS_PROP_REFRESERVATION) { 2465367Sahrens /* these guys are modifyable, but have no source */ 2475367Sahrens uint64_t value; 2485367Sahrens verify(nvlist_lookup_uint64(propnv, 2495367Sahrens ZPROP_VALUE, &value) == 0); 2507265Sahrens if (zhp->zfs_type == ZFS_TYPE_SNAPSHOT) 2517265Sahrens continue; 2525367Sahrens } else { 2535367Sahrens char *source; 2545367Sahrens if (nvlist_lookup_string(propnv, 2555367Sahrens ZPROP_SOURCE, &source) != 0) 2565367Sahrens continue; 2575367Sahrens if (strcmp(source, zhp->zfs_name) != 0) 2585367Sahrens continue; 2595367Sahrens } 2605367Sahrens 2615367Sahrens if (zfs_prop_user(propname) || 2625367Sahrens zfs_prop_get_type(prop) == PROP_TYPE_STRING) { 2635367Sahrens char *value; 2645367Sahrens verify(nvlist_lookup_string(propnv, 2655367Sahrens ZPROP_VALUE, &value) == 0); 2665367Sahrens VERIFY(0 == nvlist_add_string(nv, propname, value)); 2675367Sahrens } else { 2685367Sahrens uint64_t value; 2695367Sahrens verify(nvlist_lookup_uint64(propnv, 2705367Sahrens ZPROP_VALUE, &value) == 0); 2715367Sahrens VERIFY(0 == nvlist_add_uint64(nv, propname, value)); 2725367Sahrens } 2735367Sahrens } 2745367Sahrens } 2755367Sahrens 27610200SLori.Alt@Sun.COM /* 27710200SLori.Alt@Sun.COM * recursively generate nvlists describing datasets. See comment 27810200SLori.Alt@Sun.COM * for the data structure send_data_t above for description of contents 27910200SLori.Alt@Sun.COM * of the nvlist. 28010200SLori.Alt@Sun.COM */ 2815367Sahrens static int 2825367Sahrens send_iterate_fs(zfs_handle_t *zhp, void *arg) 2835367Sahrens { 2845367Sahrens send_data_t *sd = arg; 2855367Sahrens nvlist_t *nvfs, *nv; 2865367Sahrens int rv; 2875367Sahrens uint64_t parent_fromsnap_guid_save = sd->parent_fromsnap_guid; 2885367Sahrens uint64_t guid = zhp->zfs_dmustats.dds_guid; 2895367Sahrens char guidstring[64]; 2905367Sahrens 2915367Sahrens VERIFY(0 == nvlist_alloc(&nvfs, NV_UNIQUE_NAME, 0)); 2925367Sahrens VERIFY(0 == nvlist_add_string(nvfs, "name", zhp->zfs_name)); 2935367Sahrens VERIFY(0 == nvlist_add_uint64(nvfs, "parentfromsnap", 2945367Sahrens sd->parent_fromsnap_guid)); 2955367Sahrens 2965367Sahrens if (zhp->zfs_dmustats.dds_origin[0]) { 2975367Sahrens zfs_handle_t *origin = zfs_open(zhp->zfs_hdl, 2985367Sahrens zhp->zfs_dmustats.dds_origin, ZFS_TYPE_SNAPSHOT); 2995367Sahrens if (origin == NULL) 3005367Sahrens return (-1); 3015367Sahrens VERIFY(0 == nvlist_add_uint64(nvfs, "origin", 3025367Sahrens origin->zfs_dmustats.dds_guid)); 3035367Sahrens } 3045367Sahrens 3055367Sahrens /* iterate over props */ 3065367Sahrens VERIFY(0 == nvlist_alloc(&nv, NV_UNIQUE_NAME, 0)); 3075367Sahrens send_iterate_prop(zhp, nv); 3085367Sahrens VERIFY(0 == nvlist_add_nvlist(nvfs, "props", nv)); 3095367Sahrens nvlist_free(nv); 3105367Sahrens 3115367Sahrens /* iterate over snaps, and set sd->parent_fromsnap_guid */ 3125367Sahrens sd->parent_fromsnap_guid = 0; 3135367Sahrens VERIFY(0 == nvlist_alloc(&sd->parent_snaps, NV_UNIQUE_NAME, 0)); 3147265Sahrens VERIFY(0 == nvlist_alloc(&sd->snapprops, NV_UNIQUE_NAME, 0)); 3155367Sahrens (void) zfs_iter_snapshots(zhp, send_iterate_snap, sd); 3165367Sahrens VERIFY(0 == nvlist_add_nvlist(nvfs, "snaps", sd->parent_snaps)); 3177265Sahrens VERIFY(0 == nvlist_add_nvlist(nvfs, "snapprops", sd->snapprops)); 3185367Sahrens nvlist_free(sd->parent_snaps); 3197265Sahrens nvlist_free(sd->snapprops); 3205367Sahrens 3215367Sahrens /* add this fs to nvlist */ 3225367Sahrens (void) snprintf(guidstring, sizeof (guidstring), 3235367Sahrens "0x%llx", (longlong_t)guid); 3245367Sahrens VERIFY(0 == nvlist_add_nvlist(sd->fss, guidstring, nvfs)); 3255367Sahrens nvlist_free(nvfs); 3265367Sahrens 3275367Sahrens /* iterate over children */ 3285367Sahrens rv = zfs_iter_filesystems(zhp, send_iterate_fs, sd); 3295367Sahrens 3305367Sahrens sd->parent_fromsnap_guid = parent_fromsnap_guid_save; 3315367Sahrens 3325367Sahrens zfs_close(zhp); 3335367Sahrens return (rv); 3345367Sahrens } 3355367Sahrens 3365367Sahrens static int 3375367Sahrens gather_nvlist(libzfs_handle_t *hdl, const char *fsname, const char *fromsnap, 3385367Sahrens const char *tosnap, nvlist_t **nvlp, avl_tree_t **avlp) 3395367Sahrens { 3405367Sahrens zfs_handle_t *zhp; 3415367Sahrens send_data_t sd = { 0 }; 3425367Sahrens int error; 3435367Sahrens 3445367Sahrens zhp = zfs_open(hdl, fsname, ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME); 3455367Sahrens if (zhp == NULL) 3465367Sahrens return (EZFS_BADTYPE); 3475367Sahrens 3485367Sahrens VERIFY(0 == nvlist_alloc(&sd.fss, NV_UNIQUE_NAME, 0)); 3495367Sahrens sd.fromsnap = fromsnap; 3505367Sahrens sd.tosnap = tosnap; 3515896Sahl 3525896Sahl if ((error = send_iterate_fs(zhp, &sd)) != 0) { 3535896Sahl nvlist_free(sd.fss); 3545896Sahl if (avlp != NULL) 3555896Sahl *avlp = NULL; 3565896Sahl *nvlp = NULL; 3575896Sahl return (error); 3585896Sahl } 3595896Sahl 3605896Sahl if (avlp != NULL && (*avlp = fsavl_create(sd.fss)) == NULL) { 3615896Sahl nvlist_free(sd.fss); 3625896Sahl *nvlp = NULL; 3635896Sahl return (EZFS_NOMEM); 3645896Sahl } 3655367Sahrens 3665367Sahrens *nvlp = sd.fss; 3675896Sahl return (0); 3685367Sahrens } 3695367Sahrens 3705367Sahrens /* 3715367Sahrens * Routines for dealing with the sorted snapshot functionality 3725367Sahrens */ 3735367Sahrens typedef struct zfs_node { 3745367Sahrens zfs_handle_t *zn_handle; 3755367Sahrens avl_node_t zn_avlnode; 3765367Sahrens } zfs_node_t; 3775367Sahrens 3785367Sahrens static int 3795367Sahrens zfs_sort_snaps(zfs_handle_t *zhp, void *data) 3805367Sahrens { 3815367Sahrens avl_tree_t *avl = data; 3825367Sahrens zfs_node_t *node = zfs_alloc(zhp->zfs_hdl, sizeof (zfs_node_t)); 3835367Sahrens 3845367Sahrens node->zn_handle = zhp; 3855367Sahrens avl_add(avl, node); 3865367Sahrens return (0); 3875367Sahrens } 3885367Sahrens 3895367Sahrens /* ARGSUSED */ 3905367Sahrens static int 3915367Sahrens zfs_snapshot_compare(const void *larg, const void *rarg) 3925367Sahrens { 3935367Sahrens zfs_handle_t *l = ((zfs_node_t *)larg)->zn_handle; 3945367Sahrens zfs_handle_t *r = ((zfs_node_t *)rarg)->zn_handle; 3955367Sahrens uint64_t lcreate, rcreate; 3965367Sahrens 3975367Sahrens /* 3985367Sahrens * Sort them according to creation time. We use the hidden 3995367Sahrens * CREATETXG property to get an absolute ordering of snapshots. 4005367Sahrens */ 4015367Sahrens lcreate = zfs_prop_get_int(l, ZFS_PROP_CREATETXG); 4025367Sahrens rcreate = zfs_prop_get_int(r, ZFS_PROP_CREATETXG); 4035367Sahrens 4045367Sahrens if (lcreate < rcreate) 4055367Sahrens return (-1); 4065367Sahrens else if (lcreate > rcreate) 4075367Sahrens return (+1); 4085367Sahrens else 4095367Sahrens return (0); 4105367Sahrens } 4115367Sahrens 41210342Schris.kirby@sun.com int 4135367Sahrens zfs_iter_snapshots_sorted(zfs_handle_t *zhp, zfs_iter_f callback, void *data) 4145367Sahrens { 4155367Sahrens int ret = 0; 4165367Sahrens zfs_node_t *node; 4175367Sahrens avl_tree_t avl; 4185367Sahrens void *cookie = NULL; 4195367Sahrens 4205367Sahrens avl_create(&avl, zfs_snapshot_compare, 4215367Sahrens sizeof (zfs_node_t), offsetof(zfs_node_t, zn_avlnode)); 4225367Sahrens 4235367Sahrens ret = zfs_iter_snapshots(zhp, zfs_sort_snaps, &avl); 4245367Sahrens 4255367Sahrens for (node = avl_first(&avl); node != NULL; node = AVL_NEXT(&avl, node)) 4265367Sahrens ret |= callback(node->zn_handle, data); 4275367Sahrens 4285367Sahrens while ((node = avl_destroy_nodes(&avl, &cookie)) != NULL) 4295367Sahrens free(node); 4305367Sahrens 4315367Sahrens avl_destroy(&avl); 4325367Sahrens 4335367Sahrens return (ret); 4345367Sahrens } 4355367Sahrens 4365367Sahrens /* 4375367Sahrens * Routines specific to "zfs send" 4385367Sahrens */ 4395367Sahrens typedef struct send_dump_data { 4405367Sahrens /* these are all just the short snapname (the part after the @) */ 4415367Sahrens const char *fromsnap; 4425367Sahrens const char *tosnap; 4435367Sahrens char lastsnap[ZFS_MAXNAMELEN]; 4445367Sahrens boolean_t seenfrom, seento, replicate, doall, fromorigin; 4455367Sahrens boolean_t verbose; 4465367Sahrens int outfd; 4475367Sahrens boolean_t err; 4485367Sahrens nvlist_t *fss; 4495367Sahrens avl_tree_t *fsavl; 4505367Sahrens } send_dump_data_t; 4515367Sahrens 4525367Sahrens /* 4535367Sahrens * Dumps a backup of the given snapshot (incremental from fromsnap if it's not 4545367Sahrens * NULL) to the file descriptor specified by outfd. 4555367Sahrens */ 4565367Sahrens static int 4575367Sahrens dump_ioctl(zfs_handle_t *zhp, const char *fromsnap, boolean_t fromorigin, 4585367Sahrens int outfd) 4595367Sahrens { 4605367Sahrens zfs_cmd_t zc = { 0 }; 4615367Sahrens libzfs_handle_t *hdl = zhp->zfs_hdl; 4625367Sahrens 4635367Sahrens assert(zhp->zfs_type == ZFS_TYPE_SNAPSHOT); 4645367Sahrens assert(fromsnap == NULL || fromsnap[0] == '\0' || !fromorigin); 4655367Sahrens 4665367Sahrens (void) strlcpy(zc.zc_name, zhp->zfs_name, sizeof (zc.zc_name)); 4675367Sahrens if (fromsnap) 4685765Sek110237 (void) strlcpy(zc.zc_value, fromsnap, sizeof (zc.zc_value)); 4695367Sahrens zc.zc_cookie = outfd; 4705367Sahrens zc.zc_obj = fromorigin; 4715367Sahrens 4725367Sahrens if (ioctl(zhp->zfs_hdl->libzfs_fd, ZFS_IOC_SEND, &zc) != 0) { 4735367Sahrens char errbuf[1024]; 4745367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN, 4755367Sahrens "warning: cannot send '%s'"), zhp->zfs_name); 4765367Sahrens 4775367Sahrens switch (errno) { 4785367Sahrens 4795367Sahrens case EXDEV: 4805367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 4815367Sahrens "not an earlier snapshot from the same fs")); 4825367Sahrens return (zfs_error(hdl, EZFS_CROSSTARGET, errbuf)); 4835367Sahrens 4845367Sahrens case ENOENT: 4855367Sahrens if (zfs_dataset_exists(hdl, zc.zc_name, 4865367Sahrens ZFS_TYPE_SNAPSHOT)) { 4875367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 4885367Sahrens "incremental source (@%s) does not exist"), 4895367Sahrens zc.zc_value); 4905367Sahrens } 4915367Sahrens return (zfs_error(hdl, EZFS_NOENT, errbuf)); 4925367Sahrens 4935367Sahrens case EDQUOT: 4945367Sahrens case EFBIG: 4955367Sahrens case EIO: 4965367Sahrens case ENOLINK: 4975367Sahrens case ENOSPC: 4985367Sahrens case ENOSTR: 4995367Sahrens case ENXIO: 5005367Sahrens case EPIPE: 5015367Sahrens case ERANGE: 5025367Sahrens case EFAULT: 5035367Sahrens case EROFS: 5045367Sahrens zfs_error_aux(hdl, strerror(errno)); 5055367Sahrens return (zfs_error(hdl, EZFS_BADBACKUP, errbuf)); 5065367Sahrens 5075367Sahrens default: 5085367Sahrens return (zfs_standard_error(hdl, errno, errbuf)); 5095367Sahrens } 5105367Sahrens } 5115367Sahrens 5125367Sahrens return (0); 5135367Sahrens } 5145367Sahrens 5155367Sahrens static int 5165367Sahrens dump_snapshot(zfs_handle_t *zhp, void *arg) 5175367Sahrens { 5185367Sahrens send_dump_data_t *sdd = arg; 5195367Sahrens const char *thissnap; 5205367Sahrens int err; 5215367Sahrens 5225367Sahrens thissnap = strchr(zhp->zfs_name, '@') + 1; 5235367Sahrens 5245367Sahrens if (sdd->fromsnap && !sdd->seenfrom && 5255367Sahrens strcmp(sdd->fromsnap, thissnap) == 0) { 5265367Sahrens sdd->seenfrom = B_TRUE; 5275367Sahrens (void) strcpy(sdd->lastsnap, thissnap); 5285367Sahrens zfs_close(zhp); 5295367Sahrens return (0); 5305367Sahrens } 5315367Sahrens 5325367Sahrens if (sdd->seento || !sdd->seenfrom) { 5335367Sahrens zfs_close(zhp); 5345367Sahrens return (0); 5355367Sahrens } 5365367Sahrens 5375367Sahrens /* send it */ 5385367Sahrens if (sdd->verbose) { 5395367Sahrens (void) fprintf(stderr, "sending from @%s to %s\n", 5405367Sahrens sdd->lastsnap, zhp->zfs_name); 5415367Sahrens } 5425367Sahrens 5435367Sahrens err = dump_ioctl(zhp, sdd->lastsnap, 5445367Sahrens sdd->lastsnap[0] == '\0' && (sdd->fromorigin || sdd->replicate), 5455367Sahrens sdd->outfd); 5465367Sahrens 5475367Sahrens if (!sdd->seento && strcmp(sdd->tosnap, thissnap) == 0) 5485367Sahrens sdd->seento = B_TRUE; 5495367Sahrens 5505367Sahrens (void) strcpy(sdd->lastsnap, thissnap); 5515367Sahrens zfs_close(zhp); 5525367Sahrens return (err); 5535367Sahrens } 5545367Sahrens 5555367Sahrens static int 5565367Sahrens dump_filesystem(zfs_handle_t *zhp, void *arg) 5575367Sahrens { 5585367Sahrens int rv = 0; 5595367Sahrens send_dump_data_t *sdd = arg; 5605367Sahrens boolean_t missingfrom = B_FALSE; 5615367Sahrens zfs_cmd_t zc = { 0 }; 5625367Sahrens 5635367Sahrens (void) snprintf(zc.zc_name, sizeof (zc.zc_name), "%s@%s", 5645367Sahrens zhp->zfs_name, sdd->tosnap); 5655367Sahrens if (ioctl(zhp->zfs_hdl->libzfs_fd, ZFS_IOC_OBJSET_STATS, &zc) != 0) { 5665367Sahrens (void) fprintf(stderr, "WARNING: " 5675367Sahrens "could not send %s@%s: does not exist\n", 5685367Sahrens zhp->zfs_name, sdd->tosnap); 5695367Sahrens sdd->err = B_TRUE; 5705367Sahrens return (0); 5715367Sahrens } 5725367Sahrens 5735367Sahrens if (sdd->replicate && sdd->fromsnap) { 5745367Sahrens /* 5755367Sahrens * If this fs does not have fromsnap, and we're doing 5765367Sahrens * recursive, we need to send a full stream from the 5775367Sahrens * beginning (or an incremental from the origin if this 5785367Sahrens * is a clone). If we're doing non-recursive, then let 5795367Sahrens * them get the error. 5805367Sahrens */ 5815367Sahrens (void) snprintf(zc.zc_name, sizeof (zc.zc_name), "%s@%s", 5825367Sahrens zhp->zfs_name, sdd->fromsnap); 5835367Sahrens if (ioctl(zhp->zfs_hdl->libzfs_fd, 5845367Sahrens ZFS_IOC_OBJSET_STATS, &zc) != 0) { 5855367Sahrens missingfrom = B_TRUE; 5865367Sahrens } 5875367Sahrens } 5885367Sahrens 5895367Sahrens if (sdd->doall) { 5905367Sahrens sdd->seenfrom = sdd->seento = sdd->lastsnap[0] = 0; 5915367Sahrens if (sdd->fromsnap == NULL || missingfrom) 5925367Sahrens sdd->seenfrom = B_TRUE; 5935367Sahrens 5945367Sahrens rv = zfs_iter_snapshots_sorted(zhp, dump_snapshot, arg); 5955367Sahrens if (!sdd->seenfrom) { 5965367Sahrens (void) fprintf(stderr, 5975367Sahrens "WARNING: could not send %s@%s:\n" 5985367Sahrens "incremental source (%s@%s) does not exist\n", 5995367Sahrens zhp->zfs_name, sdd->tosnap, 6005367Sahrens zhp->zfs_name, sdd->fromsnap); 6015367Sahrens sdd->err = B_TRUE; 6025367Sahrens } else if (!sdd->seento) { 6039396SMatthew.Ahrens@Sun.COM if (sdd->fromsnap) { 6049396SMatthew.Ahrens@Sun.COM (void) fprintf(stderr, 6059396SMatthew.Ahrens@Sun.COM "WARNING: could not send %s@%s:\n" 6069396SMatthew.Ahrens@Sun.COM "incremental source (%s@%s) " 6079396SMatthew.Ahrens@Sun.COM "is not earlier than it\n", 6089396SMatthew.Ahrens@Sun.COM zhp->zfs_name, sdd->tosnap, 6099396SMatthew.Ahrens@Sun.COM zhp->zfs_name, sdd->fromsnap); 6109396SMatthew.Ahrens@Sun.COM } else { 6119396SMatthew.Ahrens@Sun.COM (void) fprintf(stderr, "WARNING: " 6129396SMatthew.Ahrens@Sun.COM "could not send %s@%s: does not exist\n", 6139396SMatthew.Ahrens@Sun.COM zhp->zfs_name, sdd->tosnap); 6149396SMatthew.Ahrens@Sun.COM } 6155367Sahrens sdd->err = B_TRUE; 6165367Sahrens } 6175367Sahrens } else { 6185367Sahrens zfs_handle_t *snapzhp; 6195367Sahrens char snapname[ZFS_MAXNAMELEN]; 6205367Sahrens 6215367Sahrens (void) snprintf(snapname, sizeof (snapname), "%s@%s", 6225367Sahrens zfs_get_name(zhp), sdd->tosnap); 6235367Sahrens snapzhp = zfs_open(zhp->zfs_hdl, snapname, ZFS_TYPE_SNAPSHOT); 6246476Sahl if (snapzhp == NULL) { 6256476Sahl rv = -1; 6266476Sahl } else { 6276476Sahl rv = dump_ioctl(snapzhp, 6286476Sahl missingfrom ? NULL : sdd->fromsnap, 6296476Sahl sdd->fromorigin || missingfrom, 6306476Sahl sdd->outfd); 6316476Sahl sdd->seento = B_TRUE; 6326476Sahl zfs_close(snapzhp); 6336476Sahl } 6345367Sahrens } 6355367Sahrens 6365367Sahrens return (rv); 6375367Sahrens } 6385367Sahrens 6395367Sahrens static int 6405367Sahrens dump_filesystems(zfs_handle_t *rzhp, void *arg) 6415367Sahrens { 6425367Sahrens send_dump_data_t *sdd = arg; 6435367Sahrens nvpair_t *fspair; 6445367Sahrens boolean_t needagain, progress; 6455367Sahrens 6465367Sahrens if (!sdd->replicate) 6475367Sahrens return (dump_filesystem(rzhp, sdd)); 6485367Sahrens 6495367Sahrens again: 6505367Sahrens needagain = progress = B_FALSE; 6515367Sahrens for (fspair = nvlist_next_nvpair(sdd->fss, NULL); fspair; 6525367Sahrens fspair = nvlist_next_nvpair(sdd->fss, fspair)) { 6535367Sahrens nvlist_t *fslist; 6545367Sahrens char *fsname; 6555367Sahrens zfs_handle_t *zhp; 6565367Sahrens int err; 6575367Sahrens uint64_t origin_guid = 0; 6585367Sahrens nvlist_t *origin_nv; 6595367Sahrens 6605367Sahrens VERIFY(nvpair_value_nvlist(fspair, &fslist) == 0); 6615367Sahrens if (nvlist_lookup_boolean(fslist, "sent") == 0) 6625367Sahrens continue; 6635367Sahrens 6645367Sahrens VERIFY(nvlist_lookup_string(fslist, "name", &fsname) == 0); 6655367Sahrens (void) nvlist_lookup_uint64(fslist, "origin", &origin_guid); 6665367Sahrens 6675367Sahrens origin_nv = fsavl_find(sdd->fsavl, origin_guid, NULL); 6685367Sahrens if (origin_nv && 6695367Sahrens nvlist_lookup_boolean(origin_nv, "sent") == ENOENT) { 6705367Sahrens /* 6715367Sahrens * origin has not been sent yet; 6725367Sahrens * skip this clone. 6735367Sahrens */ 6745367Sahrens needagain = B_TRUE; 6755367Sahrens continue; 6765367Sahrens } 6775367Sahrens 6785367Sahrens zhp = zfs_open(rzhp->zfs_hdl, fsname, ZFS_TYPE_DATASET); 6796476Sahl if (zhp == NULL) 6806476Sahl return (-1); 6815367Sahrens err = dump_filesystem(zhp, sdd); 6825367Sahrens VERIFY(nvlist_add_boolean(fslist, "sent") == 0); 6835367Sahrens progress = B_TRUE; 6845367Sahrens zfs_close(zhp); 6855367Sahrens if (err) 6865367Sahrens return (err); 6875367Sahrens } 6885367Sahrens if (needagain) { 6895367Sahrens assert(progress); 6905367Sahrens goto again; 6915367Sahrens } 6925367Sahrens return (0); 6935367Sahrens } 6945367Sahrens 6955367Sahrens /* 69610200SLori.Alt@Sun.COM * Generate a send stream for the dataset identified by the argument zhp. 69710200SLori.Alt@Sun.COM * 69810200SLori.Alt@Sun.COM * The content of the send stream is the snapshot identified by 69910200SLori.Alt@Sun.COM * 'tosnap'. Incremental streams are requested in two ways: 70010200SLori.Alt@Sun.COM * - from the snapshot identified by "fromsnap" (if non-null) or 70110200SLori.Alt@Sun.COM * - from the origin of the dataset identified by zhp, which must 70210200SLori.Alt@Sun.COM * be a clone. In this case, "fromsnap" is null and "fromorigin" 70310200SLori.Alt@Sun.COM * is TRUE. 70410200SLori.Alt@Sun.COM * 70510200SLori.Alt@Sun.COM * The send stream is recursive (i.e. dumps a hierarchy of snapshots) and 70610200SLori.Alt@Sun.COM * uses a special header (with a version field of DMU_BACKUP_HEADER_VERSION) 70710200SLori.Alt@Sun.COM * if "replicate" is set. If "doall" is set, dump all the intermediate 70810200SLori.Alt@Sun.COM * snapshots. The DMU_BACKUP_HEADER_VERSION header is used in the "doall" 70910200SLori.Alt@Sun.COM * case too. 7105367Sahrens */ 7115367Sahrens int 7125367Sahrens zfs_send(zfs_handle_t *zhp, const char *fromsnap, const char *tosnap, 7135367Sahrens boolean_t replicate, boolean_t doall, boolean_t fromorigin, 7145367Sahrens boolean_t verbose, int outfd) 7155367Sahrens { 7165367Sahrens char errbuf[1024]; 7175367Sahrens send_dump_data_t sdd = { 0 }; 7185367Sahrens int err; 7195367Sahrens nvlist_t *fss = NULL; 7205367Sahrens avl_tree_t *fsavl = NULL; 72110342Schris.kirby@sun.com char holdtag[128]; 72210342Schris.kirby@sun.com static uint64_t holdseq; 723*10735Schris.kirby@sun.com int spa_version; 724*10735Schris.kirby@sun.com boolean_t holdsnaps = B_FALSE; 7255367Sahrens 7265367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN, 7275367Sahrens "cannot send '%s'"), zhp->zfs_name); 7285367Sahrens 7295367Sahrens if (fromsnap && fromsnap[0] == '\0') { 7305367Sahrens zfs_error_aux(zhp->zfs_hdl, dgettext(TEXT_DOMAIN, 7315367Sahrens "zero-length incremental source")); 7325367Sahrens return (zfs_error(zhp->zfs_hdl, EZFS_NOENT, errbuf)); 7335367Sahrens } 7345367Sahrens 735*10735Schris.kirby@sun.com if (zfs_spa_version(zhp, &spa_version) == 0 && 736*10735Schris.kirby@sun.com spa_version >= SPA_VERSION_USERREFS) 737*10735Schris.kirby@sun.com holdsnaps = B_TRUE; 738*10735Schris.kirby@sun.com 7395367Sahrens if (replicate || doall) { 7405367Sahrens dmu_replay_record_t drr = { 0 }; 7415367Sahrens char *packbuf = NULL; 7425367Sahrens size_t buflen = 0; 7435367Sahrens zio_cksum_t zc = { 0 }; 7445367Sahrens 7455367Sahrens assert(fromsnap || doall); 7465367Sahrens 747*10735Schris.kirby@sun.com if (holdsnaps) { 748*10735Schris.kirby@sun.com (void) snprintf(holdtag, sizeof (holdtag), 749*10735Schris.kirby@sun.com ".send-%d-%llu", getpid(), (u_longlong_t)holdseq); 750*10735Schris.kirby@sun.com ++holdseq; 751*10735Schris.kirby@sun.com err = zfs_hold_range(zhp, fromsnap, tosnap, 752*10735Schris.kirby@sun.com holdtag, B_TRUE); 753*10735Schris.kirby@sun.com if (err) 754*10735Schris.kirby@sun.com return (err); 755*10735Schris.kirby@sun.com } 7565367Sahrens if (replicate) { 7575367Sahrens nvlist_t *hdrnv; 7585367Sahrens 7595367Sahrens VERIFY(0 == nvlist_alloc(&hdrnv, NV_UNIQUE_NAME, 0)); 7605367Sahrens if (fromsnap) { 7615367Sahrens VERIFY(0 == nvlist_add_string(hdrnv, 7625367Sahrens "fromsnap", fromsnap)); 7635367Sahrens } 7645367Sahrens VERIFY(0 == nvlist_add_string(hdrnv, "tosnap", tosnap)); 7655367Sahrens 7665367Sahrens err = gather_nvlist(zhp->zfs_hdl, zhp->zfs_name, 7675367Sahrens fromsnap, tosnap, &fss, &fsavl); 76810342Schris.kirby@sun.com if (err) { 769*10735Schris.kirby@sun.com if (holdsnaps) { 770*10735Schris.kirby@sun.com (void) zfs_release_range(zhp, fromsnap, 771*10735Schris.kirby@sun.com tosnap, holdtag); 772*10735Schris.kirby@sun.com } 7735367Sahrens return (err); 77410342Schris.kirby@sun.com } 7755367Sahrens VERIFY(0 == nvlist_add_nvlist(hdrnv, "fss", fss)); 7765367Sahrens err = nvlist_pack(hdrnv, &packbuf, &buflen, 7775367Sahrens NV_ENCODE_XDR, 0); 7785367Sahrens nvlist_free(hdrnv); 7795367Sahrens if (err) { 7805367Sahrens fsavl_destroy(fsavl); 7815367Sahrens nvlist_free(fss); 782*10735Schris.kirby@sun.com if (holdsnaps) { 783*10735Schris.kirby@sun.com (void) zfs_release_range(zhp, fromsnap, 784*10735Schris.kirby@sun.com tosnap, holdtag); 785*10735Schris.kirby@sun.com } 7865367Sahrens return (zfs_standard_error(zhp->zfs_hdl, 7875367Sahrens err, errbuf)); 7885367Sahrens } 7895367Sahrens } 7905367Sahrens 7915367Sahrens /* write first begin record */ 7925367Sahrens drr.drr_type = DRR_BEGIN; 7935367Sahrens drr.drr_u.drr_begin.drr_magic = DMU_BACKUP_MAGIC; 7945367Sahrens drr.drr_u.drr_begin.drr_version = DMU_BACKUP_HEADER_VERSION; 7955367Sahrens (void) snprintf(drr.drr_u.drr_begin.drr_toname, 7965367Sahrens sizeof (drr.drr_u.drr_begin.drr_toname), 7975367Sahrens "%s@%s", zhp->zfs_name, tosnap); 7985367Sahrens drr.drr_payloadlen = buflen; 7995367Sahrens fletcher_4_incremental_native(&drr, sizeof (drr), &zc); 8005367Sahrens err = write(outfd, &drr, sizeof (drr)); 8015367Sahrens 8025367Sahrens /* write header nvlist */ 8035367Sahrens if (err != -1) { 8045367Sahrens fletcher_4_incremental_native(packbuf, buflen, &zc); 8055367Sahrens err = write(outfd, packbuf, buflen); 8065367Sahrens } 8075367Sahrens free(packbuf); 8085367Sahrens if (err == -1) { 8095367Sahrens fsavl_destroy(fsavl); 8105367Sahrens nvlist_free(fss); 811*10735Schris.kirby@sun.com if (holdsnaps) { 812*10735Schris.kirby@sun.com (void) zfs_release_range(zhp, fromsnap, tosnap, 813*10735Schris.kirby@sun.com holdtag); 814*10735Schris.kirby@sun.com } 8155367Sahrens return (zfs_standard_error(zhp->zfs_hdl, 8165367Sahrens errno, errbuf)); 8175367Sahrens } 8185367Sahrens 8195367Sahrens /* write end record */ 8205367Sahrens if (err != -1) { 8215367Sahrens bzero(&drr, sizeof (drr)); 8225367Sahrens drr.drr_type = DRR_END; 8235367Sahrens drr.drr_u.drr_end.drr_checksum = zc; 8245367Sahrens err = write(outfd, &drr, sizeof (drr)); 8255367Sahrens if (err == -1) { 8265367Sahrens fsavl_destroy(fsavl); 8275367Sahrens nvlist_free(fss); 828*10735Schris.kirby@sun.com if (holdsnaps) { 829*10735Schris.kirby@sun.com (void) zfs_release_range(zhp, fromsnap, 830*10735Schris.kirby@sun.com tosnap, holdtag); 831*10735Schris.kirby@sun.com } 8325367Sahrens return (zfs_standard_error(zhp->zfs_hdl, 8335367Sahrens errno, errbuf)); 8345367Sahrens } 8355367Sahrens } 8365367Sahrens } 8375367Sahrens 8385367Sahrens /* dump each stream */ 8395367Sahrens sdd.fromsnap = fromsnap; 8405367Sahrens sdd.tosnap = tosnap; 8415367Sahrens sdd.outfd = outfd; 8425367Sahrens sdd.replicate = replicate; 8435367Sahrens sdd.doall = doall; 8445367Sahrens sdd.fromorigin = fromorigin; 8455367Sahrens sdd.fss = fss; 8465367Sahrens sdd.fsavl = fsavl; 8475367Sahrens sdd.verbose = verbose; 8485367Sahrens err = dump_filesystems(zhp, &sdd); 8495367Sahrens fsavl_destroy(fsavl); 8505367Sahrens nvlist_free(fss); 8515367Sahrens 8525367Sahrens if (replicate || doall) { 8535367Sahrens /* 8545367Sahrens * write final end record. NB: want to do this even if 8555367Sahrens * there was some error, because it might not be totally 8565367Sahrens * failed. 8575367Sahrens */ 8585367Sahrens dmu_replay_record_t drr = { 0 }; 8595367Sahrens drr.drr_type = DRR_END; 860*10735Schris.kirby@sun.com if (holdsnaps) { 861*10735Schris.kirby@sun.com (void) zfs_release_range(zhp, fromsnap, tosnap, 862*10735Schris.kirby@sun.com holdtag); 863*10735Schris.kirby@sun.com } 8645367Sahrens if (write(outfd, &drr, sizeof (drr)) == -1) { 8655367Sahrens return (zfs_standard_error(zhp->zfs_hdl, 8665367Sahrens errno, errbuf)); 8675367Sahrens } 8685367Sahrens } 8695367Sahrens 8705367Sahrens return (err || sdd.err); 8715367Sahrens } 8725367Sahrens 8735367Sahrens /* 8745367Sahrens * Routines specific to "zfs recv" 8755367Sahrens */ 8765367Sahrens 8775367Sahrens static int 8785367Sahrens recv_read(libzfs_handle_t *hdl, int fd, void *buf, int ilen, 8795367Sahrens boolean_t byteswap, zio_cksum_t *zc) 8805367Sahrens { 8815367Sahrens char *cp = buf; 8825367Sahrens int rv; 8835367Sahrens int len = ilen; 8845367Sahrens 8855367Sahrens do { 8865367Sahrens rv = read(fd, cp, len); 8875367Sahrens cp += rv; 8885367Sahrens len -= rv; 8895367Sahrens } while (rv > 0); 8905367Sahrens 8915367Sahrens if (rv < 0 || len != 0) { 8925367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 8935367Sahrens "failed to read from stream")); 8945367Sahrens return (zfs_error(hdl, EZFS_BADSTREAM, dgettext(TEXT_DOMAIN, 8955367Sahrens "cannot receive"))); 8965367Sahrens } 8975367Sahrens 8985367Sahrens if (zc) { 8995367Sahrens if (byteswap) 9005367Sahrens fletcher_4_incremental_byteswap(buf, ilen, zc); 9015367Sahrens else 9025367Sahrens fletcher_4_incremental_native(buf, ilen, zc); 9035367Sahrens } 9045367Sahrens return (0); 9055367Sahrens } 9065367Sahrens 9075367Sahrens static int 9085367Sahrens recv_read_nvlist(libzfs_handle_t *hdl, int fd, int len, nvlist_t **nvp, 9095367Sahrens boolean_t byteswap, zio_cksum_t *zc) 9105367Sahrens { 9115367Sahrens char *buf; 9125367Sahrens int err; 9135367Sahrens 9145367Sahrens buf = zfs_alloc(hdl, len); 9155367Sahrens if (buf == NULL) 9165367Sahrens return (ENOMEM); 9175367Sahrens 9185367Sahrens err = recv_read(hdl, fd, buf, len, byteswap, zc); 9195367Sahrens if (err != 0) { 9205367Sahrens free(buf); 9215367Sahrens return (err); 9225367Sahrens } 9235367Sahrens 9245367Sahrens err = nvlist_unpack(buf, len, nvp, 0); 9255367Sahrens free(buf); 9265367Sahrens if (err != 0) { 9275367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid " 9285367Sahrens "stream (malformed nvlist)")); 9295367Sahrens return (EINVAL); 9305367Sahrens } 9315367Sahrens return (0); 9325367Sahrens } 9335367Sahrens 9345367Sahrens static int 9355367Sahrens recv_rename(libzfs_handle_t *hdl, const char *name, const char *tryname, 9365367Sahrens int baselen, char *newname, recvflags_t flags) 9375367Sahrens { 9385367Sahrens static int seq; 9395367Sahrens zfs_cmd_t zc = { 0 }; 9405367Sahrens int err; 9416047Sahrens prop_changelist_t *clp; 9426047Sahrens zfs_handle_t *zhp; 9435367Sahrens 9446047Sahrens zhp = zfs_open(hdl, name, ZFS_TYPE_DATASET); 9456047Sahrens if (zhp == NULL) 9466047Sahrens return (-1); 9477366STim.Haley@Sun.COM clp = changelist_gather(zhp, ZFS_PROP_NAME, 0, 9487366STim.Haley@Sun.COM flags.force ? MS_FORCE : 0); 9496047Sahrens zfs_close(zhp); 9506047Sahrens if (clp == NULL) 9516047Sahrens return (-1); 9526047Sahrens err = changelist_prefix(clp); 9536047Sahrens if (err) 9546047Sahrens return (err); 9555367Sahrens 95610242Schris.kirby@sun.com zc.zc_objset_type = DMU_OST_ZFS; 95710242Schris.kirby@sun.com (void) strlcpy(zc.zc_name, name, sizeof (zc.zc_name)); 95810242Schris.kirby@sun.com 9595367Sahrens if (tryname) { 9605367Sahrens (void) strcpy(newname, tryname); 9615367Sahrens 9625367Sahrens (void) strlcpy(zc.zc_value, tryname, sizeof (zc.zc_value)); 9635367Sahrens 9645367Sahrens if (flags.verbose) { 9655367Sahrens (void) printf("attempting rename %s to %s\n", 9665367Sahrens zc.zc_name, zc.zc_value); 9675367Sahrens } 9685367Sahrens err = ioctl(hdl->libzfs_fd, ZFS_IOC_RENAME, &zc); 9696047Sahrens if (err == 0) 9705367Sahrens changelist_rename(clp, name, tryname); 9715367Sahrens } else { 9725367Sahrens err = ENOENT; 9735367Sahrens } 9745367Sahrens 9755367Sahrens if (err != 0 && strncmp(name+baselen, "recv-", 5) != 0) { 9765367Sahrens seq++; 9775367Sahrens 9785367Sahrens (void) strncpy(newname, name, baselen); 9795367Sahrens (void) snprintf(newname+baselen, ZFS_MAXNAMELEN-baselen, 9805367Sahrens "recv-%u-%u", getpid(), seq); 9815367Sahrens (void) strlcpy(zc.zc_value, newname, sizeof (zc.zc_value)); 9825367Sahrens 9835367Sahrens if (flags.verbose) { 9845367Sahrens (void) printf("failed - trying rename %s to %s\n", 9855367Sahrens zc.zc_name, zc.zc_value); 9865367Sahrens } 9875367Sahrens err = ioctl(hdl->libzfs_fd, ZFS_IOC_RENAME, &zc); 9886047Sahrens if (err == 0) 9895367Sahrens changelist_rename(clp, name, newname); 9905367Sahrens if (err && flags.verbose) { 9915367Sahrens (void) printf("failed (%u) - " 9925367Sahrens "will try again on next pass\n", errno); 9935367Sahrens } 9945367Sahrens err = EAGAIN; 9955367Sahrens } else if (flags.verbose) { 9965367Sahrens if (err == 0) 9975367Sahrens (void) printf("success\n"); 9985367Sahrens else 9995367Sahrens (void) printf("failed (%u)\n", errno); 10005367Sahrens } 10015367Sahrens 10026047Sahrens (void) changelist_postfix(clp); 10036047Sahrens changelist_free(clp); 10045367Sahrens 10055367Sahrens return (err); 10065367Sahrens } 10075367Sahrens 10085367Sahrens static int 10095367Sahrens recv_destroy(libzfs_handle_t *hdl, const char *name, int baselen, 10105367Sahrens char *newname, recvflags_t flags) 10115367Sahrens { 10125367Sahrens zfs_cmd_t zc = { 0 }; 10136047Sahrens int err = 0; 10146047Sahrens prop_changelist_t *clp; 10156047Sahrens zfs_handle_t *zhp; 101610242Schris.kirby@sun.com boolean_t defer = B_FALSE; 101710242Schris.kirby@sun.com int spa_version; 10186047Sahrens 10196047Sahrens zhp = zfs_open(hdl, name, ZFS_TYPE_DATASET); 10206047Sahrens if (zhp == NULL) 10216047Sahrens return (-1); 10227366STim.Haley@Sun.COM clp = changelist_gather(zhp, ZFS_PROP_NAME, 0, 10237366STim.Haley@Sun.COM flags.force ? MS_FORCE : 0); 102410242Schris.kirby@sun.com if (zfs_get_type(zhp) == ZFS_TYPE_SNAPSHOT && 102510242Schris.kirby@sun.com zfs_spa_version(zhp, &spa_version) == 0 && 102610242Schris.kirby@sun.com spa_version >= SPA_VERSION_USERREFS) 102710242Schris.kirby@sun.com defer = B_TRUE; 10286047Sahrens zfs_close(zhp); 10296047Sahrens if (clp == NULL) 10306047Sahrens return (-1); 10316047Sahrens err = changelist_prefix(clp); 10326047Sahrens if (err) 10336047Sahrens return (err); 10345367Sahrens 10355367Sahrens zc.zc_objset_type = DMU_OST_ZFS; 103610242Schris.kirby@sun.com zc.zc_defer_destroy = defer; 10375367Sahrens (void) strlcpy(zc.zc_name, name, sizeof (zc.zc_name)); 10385367Sahrens 10395367Sahrens if (flags.verbose) 10405367Sahrens (void) printf("attempting destroy %s\n", zc.zc_name); 10415367Sahrens err = ioctl(hdl->libzfs_fd, ZFS_IOC_DESTROY, &zc); 10426047Sahrens if (err == 0) { 10436047Sahrens if (flags.verbose) 10446047Sahrens (void) printf("success\n"); 10456047Sahrens changelist_remove(clp, zc.zc_name); 10465367Sahrens } 10475367Sahrens 10486047Sahrens (void) changelist_postfix(clp); 10496047Sahrens changelist_free(clp); 10506047Sahrens 105110242Schris.kirby@sun.com /* 105210242Schris.kirby@sun.com * Deferred destroy should always succeed. Since we can't tell 105310242Schris.kirby@sun.com * if it destroyed the dataset or just marked it for deferred 105410242Schris.kirby@sun.com * destroy, always do the rename just in case. 105510242Schris.kirby@sun.com */ 105610242Schris.kirby@sun.com if (err != 0 || defer) 10576047Sahrens err = recv_rename(hdl, name, NULL, baselen, newname, flags); 10585367Sahrens 10595367Sahrens return (err); 10605367Sahrens } 10615367Sahrens 10625367Sahrens typedef struct guid_to_name_data { 10635367Sahrens uint64_t guid; 10645367Sahrens char *name; 10655367Sahrens } guid_to_name_data_t; 10665367Sahrens 10675367Sahrens static int 10685367Sahrens guid_to_name_cb(zfs_handle_t *zhp, void *arg) 10695367Sahrens { 10705367Sahrens guid_to_name_data_t *gtnd = arg; 10715367Sahrens int err; 10725367Sahrens 10735367Sahrens if (zhp->zfs_dmustats.dds_guid == gtnd->guid) { 10745367Sahrens (void) strcpy(gtnd->name, zhp->zfs_name); 10755367Sahrens return (EEXIST); 10765367Sahrens } 10775367Sahrens err = zfs_iter_children(zhp, guid_to_name_cb, gtnd); 10785367Sahrens zfs_close(zhp); 10795367Sahrens return (err); 10805367Sahrens } 10815367Sahrens 10825367Sahrens static int 10835367Sahrens guid_to_name(libzfs_handle_t *hdl, const char *parent, uint64_t guid, 10845367Sahrens char *name) 10855367Sahrens { 10865367Sahrens /* exhaustive search all local snapshots */ 10875367Sahrens guid_to_name_data_t gtnd; 10885367Sahrens int err = 0; 10895367Sahrens zfs_handle_t *zhp; 10905367Sahrens char *cp; 10915367Sahrens 10925367Sahrens gtnd.guid = guid; 10935367Sahrens gtnd.name = name; 10945367Sahrens 10955367Sahrens if (strchr(parent, '@') == NULL) { 10965367Sahrens zhp = make_dataset_handle(hdl, parent); 10975367Sahrens if (zhp != NULL) { 10985367Sahrens err = zfs_iter_children(zhp, guid_to_name_cb, >nd); 10995367Sahrens zfs_close(zhp); 11005367Sahrens if (err == EEXIST) 11015367Sahrens return (0); 11025367Sahrens } 11035367Sahrens } 11045367Sahrens 11055367Sahrens cp = strchr(parent, '/'); 11065367Sahrens if (cp) 11075367Sahrens *cp = '\0'; 11085367Sahrens zhp = make_dataset_handle(hdl, parent); 11095367Sahrens if (cp) 11105367Sahrens *cp = '/'; 11115367Sahrens 11125367Sahrens if (zhp) { 11135367Sahrens err = zfs_iter_children(zhp, guid_to_name_cb, >nd); 11145367Sahrens zfs_close(zhp); 11155367Sahrens } 11165367Sahrens 11175367Sahrens return (err == EEXIST ? 0 : ENOENT); 11185367Sahrens 11195367Sahrens } 11205367Sahrens 11215367Sahrens /* 11225367Sahrens * Return true if dataset guid1 is created before guid2. 11235367Sahrens */ 11246476Sahl static int 11255367Sahrens created_before(libzfs_handle_t *hdl, avl_tree_t *avl, 11265367Sahrens uint64_t guid1, uint64_t guid2) 11275367Sahrens { 11285367Sahrens nvlist_t *nvfs; 11295367Sahrens char *fsname, *snapname; 11305367Sahrens char buf[ZFS_MAXNAMELEN]; 11316476Sahl int rv; 11325367Sahrens zfs_node_t zn1, zn2; 11335367Sahrens 11345367Sahrens if (guid2 == 0) 11356476Sahl return (0); 11365367Sahrens if (guid1 == 0) 11376476Sahl return (1); 11385367Sahrens 11395367Sahrens nvfs = fsavl_find(avl, guid1, &snapname); 11405367Sahrens VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname)); 11415367Sahrens (void) snprintf(buf, sizeof (buf), "%s@%s", fsname, snapname); 11425367Sahrens zn1.zn_handle = zfs_open(hdl, buf, ZFS_TYPE_SNAPSHOT); 11436476Sahl if (zn1.zn_handle == NULL) 11446476Sahl return (-1); 11455367Sahrens 11465367Sahrens nvfs = fsavl_find(avl, guid2, &snapname); 11475367Sahrens VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname)); 11485367Sahrens (void) snprintf(buf, sizeof (buf), "%s@%s", fsname, snapname); 11495367Sahrens zn2.zn_handle = zfs_open(hdl, buf, ZFS_TYPE_SNAPSHOT); 11506476Sahl if (zn2.zn_handle == NULL) { 11516476Sahl zfs_close(zn2.zn_handle); 11526476Sahl return (-1); 11536476Sahl } 11545367Sahrens 11555367Sahrens rv = (zfs_snapshot_compare(&zn1, &zn2) == -1); 11565367Sahrens 11575367Sahrens zfs_close(zn1.zn_handle); 11585367Sahrens zfs_close(zn2.zn_handle); 11595367Sahrens 11605367Sahrens return (rv); 11615367Sahrens } 11625367Sahrens 11635367Sahrens static int 11645367Sahrens recv_incremental_replication(libzfs_handle_t *hdl, const char *tofs, 11655367Sahrens recvflags_t flags, nvlist_t *stream_nv, avl_tree_t *stream_avl) 11665367Sahrens { 11675367Sahrens nvlist_t *local_nv; 11685367Sahrens avl_tree_t *local_avl; 11695367Sahrens nvpair_t *fselem, *nextfselem; 11705367Sahrens char *tosnap, *fromsnap; 11715367Sahrens char newname[ZFS_MAXNAMELEN]; 11725367Sahrens int error; 11735367Sahrens boolean_t needagain, progress; 117410299SPrabahar.Jeyaram@Sun.COM char *s1, *s2; 11755367Sahrens 11765367Sahrens VERIFY(0 == nvlist_lookup_string(stream_nv, "fromsnap", &fromsnap)); 11775367Sahrens VERIFY(0 == nvlist_lookup_string(stream_nv, "tosnap", &tosnap)); 11785367Sahrens 11795367Sahrens if (flags.dryrun) 11805367Sahrens return (0); 11815367Sahrens 11825367Sahrens again: 11835367Sahrens needagain = progress = B_FALSE; 11845367Sahrens 11855367Sahrens if ((error = gather_nvlist(hdl, tofs, fromsnap, NULL, 11865367Sahrens &local_nv, &local_avl)) != 0) 11875367Sahrens return (error); 11885367Sahrens 11895367Sahrens /* 11905367Sahrens * Process deletes and renames 11915367Sahrens */ 11925367Sahrens for (fselem = nvlist_next_nvpair(local_nv, NULL); 11935367Sahrens fselem; fselem = nextfselem) { 11945367Sahrens nvlist_t *nvfs, *snaps; 11955367Sahrens nvlist_t *stream_nvfs = NULL; 11965367Sahrens nvpair_t *snapelem, *nextsnapelem; 11975367Sahrens uint64_t fromguid = 0; 11985367Sahrens uint64_t originguid = 0; 11995367Sahrens uint64_t stream_originguid = 0; 12005367Sahrens uint64_t parent_fromsnap_guid, stream_parent_fromsnap_guid; 12015367Sahrens char *fsname, *stream_fsname; 12025367Sahrens 12035367Sahrens nextfselem = nvlist_next_nvpair(local_nv, fselem); 12045367Sahrens 12055367Sahrens VERIFY(0 == nvpair_value_nvlist(fselem, &nvfs)); 12065367Sahrens VERIFY(0 == nvlist_lookup_nvlist(nvfs, "snaps", &snaps)); 12075367Sahrens VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname)); 12085367Sahrens VERIFY(0 == nvlist_lookup_uint64(nvfs, "parentfromsnap", 12095367Sahrens &parent_fromsnap_guid)); 12105367Sahrens (void) nvlist_lookup_uint64(nvfs, "origin", &originguid); 12115367Sahrens 12125367Sahrens /* 12135367Sahrens * First find the stream's fs, so we can check for 12145367Sahrens * a different origin (due to "zfs promote") 12155367Sahrens */ 12165367Sahrens for (snapelem = nvlist_next_nvpair(snaps, NULL); 12175367Sahrens snapelem; snapelem = nvlist_next_nvpair(snaps, snapelem)) { 12185367Sahrens uint64_t thisguid; 12195367Sahrens 12205367Sahrens VERIFY(0 == nvpair_value_uint64(snapelem, &thisguid)); 12215367Sahrens stream_nvfs = fsavl_find(stream_avl, thisguid, NULL); 12225367Sahrens 12235367Sahrens if (stream_nvfs != NULL) 12245367Sahrens break; 12255367Sahrens } 12265367Sahrens 12275367Sahrens /* check for promote */ 12285367Sahrens (void) nvlist_lookup_uint64(stream_nvfs, "origin", 12295367Sahrens &stream_originguid); 12305367Sahrens if (stream_nvfs && originguid != stream_originguid) { 12316476Sahl switch (created_before(hdl, local_avl, 12326476Sahl stream_originguid, originguid)) { 12336476Sahl case 1: { 12345367Sahrens /* promote it! */ 12355367Sahrens zfs_cmd_t zc = { 0 }; 12365367Sahrens nvlist_t *origin_nvfs; 12375367Sahrens char *origin_fsname; 12385367Sahrens 12395367Sahrens if (flags.verbose) 12405367Sahrens (void) printf("promoting %s\n", fsname); 12415367Sahrens 12425367Sahrens origin_nvfs = fsavl_find(local_avl, originguid, 12435367Sahrens NULL); 12445367Sahrens VERIFY(0 == nvlist_lookup_string(origin_nvfs, 12455367Sahrens "name", &origin_fsname)); 12465367Sahrens (void) strlcpy(zc.zc_value, origin_fsname, 12475367Sahrens sizeof (zc.zc_value)); 12485367Sahrens (void) strlcpy(zc.zc_name, fsname, 12495367Sahrens sizeof (zc.zc_name)); 12505367Sahrens error = zfs_ioctl(hdl, ZFS_IOC_PROMOTE, &zc); 12515367Sahrens if (error == 0) 12525367Sahrens progress = B_TRUE; 12536476Sahl break; 12546476Sahl } 12556476Sahl default: 12566476Sahl break; 12576476Sahl case -1: 12586476Sahl fsavl_destroy(local_avl); 12596476Sahl nvlist_free(local_nv); 12606476Sahl return (-1); 12615367Sahrens } 12625367Sahrens /* 12635367Sahrens * We had/have the wrong origin, therefore our 12645367Sahrens * list of snapshots is wrong. Need to handle 12655367Sahrens * them on the next pass. 12665367Sahrens */ 12675367Sahrens needagain = B_TRUE; 12685367Sahrens continue; 12695367Sahrens } 12705367Sahrens 12715367Sahrens for (snapelem = nvlist_next_nvpair(snaps, NULL); 12725367Sahrens snapelem; snapelem = nextsnapelem) { 12735367Sahrens uint64_t thisguid; 12745367Sahrens char *stream_snapname; 12757265Sahrens nvlist_t *found, *props; 12765367Sahrens 12775367Sahrens nextsnapelem = nvlist_next_nvpair(snaps, snapelem); 12785367Sahrens 12795367Sahrens VERIFY(0 == nvpair_value_uint64(snapelem, &thisguid)); 12805367Sahrens found = fsavl_find(stream_avl, thisguid, 12815367Sahrens &stream_snapname); 12825367Sahrens 12835367Sahrens /* check for delete */ 12845367Sahrens if (found == NULL) { 12855367Sahrens char name[ZFS_MAXNAMELEN]; 12865367Sahrens 12875367Sahrens if (!flags.force) 12885367Sahrens continue; 12895367Sahrens 12905367Sahrens (void) snprintf(name, sizeof (name), "%s@%s", 12915367Sahrens fsname, nvpair_name(snapelem)); 12925367Sahrens 12935367Sahrens error = recv_destroy(hdl, name, 12945367Sahrens strlen(fsname)+1, newname, flags); 12955367Sahrens if (error) 12965367Sahrens needagain = B_TRUE; 12975367Sahrens else 12985367Sahrens progress = B_TRUE; 12995367Sahrens continue; 13005367Sahrens } 13015367Sahrens 13025367Sahrens stream_nvfs = found; 13035367Sahrens 13047265Sahrens if (0 == nvlist_lookup_nvlist(stream_nvfs, "snapprops", 13057265Sahrens &props) && 0 == nvlist_lookup_nvlist(props, 13067265Sahrens stream_snapname, &props)) { 13077265Sahrens zfs_cmd_t zc = { 0 }; 13087265Sahrens 13097265Sahrens zc.zc_cookie = B_TRUE; /* clear current props */ 13107273Sahrens (void) snprintf(zc.zc_name, sizeof (zc.zc_name), 13117265Sahrens "%s@%s", fsname, nvpair_name(snapelem)); 13127265Sahrens if (zcmd_write_src_nvlist(hdl, &zc, 13137265Sahrens props) == 0) { 13147265Sahrens (void) zfs_ioctl(hdl, 13157265Sahrens ZFS_IOC_SET_PROP, &zc); 13167265Sahrens zcmd_free_nvlists(&zc); 13177265Sahrens } 13187265Sahrens } 13197265Sahrens 13205367Sahrens /* check for different snapname */ 13215367Sahrens if (strcmp(nvpair_name(snapelem), 13225367Sahrens stream_snapname) != 0) { 13235367Sahrens char name[ZFS_MAXNAMELEN]; 13245367Sahrens char tryname[ZFS_MAXNAMELEN]; 13255367Sahrens 13265367Sahrens (void) snprintf(name, sizeof (name), "%s@%s", 13275367Sahrens fsname, nvpair_name(snapelem)); 13285367Sahrens (void) snprintf(tryname, sizeof (name), "%s@%s", 13295367Sahrens fsname, stream_snapname); 13305367Sahrens 13315367Sahrens error = recv_rename(hdl, name, tryname, 13325367Sahrens strlen(fsname)+1, newname, flags); 13335367Sahrens if (error) 13345367Sahrens needagain = B_TRUE; 13355367Sahrens else 13365367Sahrens progress = B_TRUE; 13375367Sahrens } 13385367Sahrens 13395367Sahrens if (strcmp(stream_snapname, fromsnap) == 0) 13405367Sahrens fromguid = thisguid; 13415367Sahrens } 13425367Sahrens 13435367Sahrens /* check for delete */ 13445367Sahrens if (stream_nvfs == NULL) { 13455367Sahrens if (!flags.force) 13465367Sahrens continue; 13475367Sahrens 13485367Sahrens error = recv_destroy(hdl, fsname, strlen(tofs)+1, 13495367Sahrens newname, flags); 13505367Sahrens if (error) 13515367Sahrens needagain = B_TRUE; 13525367Sahrens else 13535367Sahrens progress = B_TRUE; 13545367Sahrens continue; 13555367Sahrens } 13565367Sahrens 13575367Sahrens if (fromguid == 0 && flags.verbose) { 13585367Sahrens (void) printf("local fs %s does not have fromsnap " 13595367Sahrens "(%s in stream); must have been deleted locally; " 13605367Sahrens "ignoring\n", fsname, fromsnap); 13615367Sahrens continue; 13625367Sahrens } 13635367Sahrens 13645367Sahrens VERIFY(0 == nvlist_lookup_string(stream_nvfs, 13655367Sahrens "name", &stream_fsname)); 13665367Sahrens VERIFY(0 == nvlist_lookup_uint64(stream_nvfs, 13675367Sahrens "parentfromsnap", &stream_parent_fromsnap_guid)); 13685367Sahrens 136910302SPrabahar.Jeyaram@Sun.COM s1 = strrchr(fsname, '/'); 137010302SPrabahar.Jeyaram@Sun.COM s2 = strrchr(stream_fsname, '/'); 137110302SPrabahar.Jeyaram@Sun.COM 13725367Sahrens /* check for rename */ 13735367Sahrens if ((stream_parent_fromsnap_guid != 0 && 13745367Sahrens stream_parent_fromsnap_guid != parent_fromsnap_guid) || 137510302SPrabahar.Jeyaram@Sun.COM ((s1 != NULL) && (s2 != NULL) && strcmp(s1, s2) != 0)) { 13765367Sahrens nvlist_t *parent; 13775367Sahrens char tryname[ZFS_MAXNAMELEN]; 13785367Sahrens 13795367Sahrens parent = fsavl_find(local_avl, 13805367Sahrens stream_parent_fromsnap_guid, NULL); 13815367Sahrens /* 13825367Sahrens * NB: parent might not be found if we used the 13835367Sahrens * tosnap for stream_parent_fromsnap_guid, 13845367Sahrens * because the parent is a newly-created fs; 13855367Sahrens * we'll be able to rename it after we recv the 13865367Sahrens * new fs. 13875367Sahrens */ 13885367Sahrens if (parent != NULL) { 13895367Sahrens char *pname; 13905367Sahrens 13915367Sahrens VERIFY(0 == nvlist_lookup_string(parent, "name", 13925367Sahrens &pname)); 13935367Sahrens (void) snprintf(tryname, sizeof (tryname), 13945367Sahrens "%s%s", pname, strrchr(stream_fsname, '/')); 13955367Sahrens } else { 13965367Sahrens tryname[0] = '\0'; 13975367Sahrens if (flags.verbose) { 13985367Sahrens (void) printf("local fs %s new parent " 13995367Sahrens "not found\n", fsname); 14005367Sahrens } 14015367Sahrens } 14025367Sahrens 14035367Sahrens error = recv_rename(hdl, fsname, tryname, 14045367Sahrens strlen(tofs)+1, newname, flags); 14055367Sahrens if (error) 14065367Sahrens needagain = B_TRUE; 14075367Sahrens else 14085367Sahrens progress = B_TRUE; 14095367Sahrens } 14105367Sahrens } 14115367Sahrens 14125367Sahrens fsavl_destroy(local_avl); 14135367Sahrens nvlist_free(local_nv); 14145367Sahrens 14155367Sahrens if (needagain && progress) { 14165367Sahrens /* do another pass to fix up temporary names */ 14175367Sahrens if (flags.verbose) 14185367Sahrens (void) printf("another pass:\n"); 14195367Sahrens goto again; 14205367Sahrens } 14215367Sahrens 14225367Sahrens return (needagain); 14235367Sahrens } 14245367Sahrens 14255367Sahrens static int 14265367Sahrens zfs_receive_package(libzfs_handle_t *hdl, int fd, const char *destname, 14277366STim.Haley@Sun.COM recvflags_t flags, dmu_replay_record_t *drr, zio_cksum_t *zc, 14287366STim.Haley@Sun.COM char **top_zfs) 14295367Sahrens { 14305367Sahrens nvlist_t *stream_nv = NULL; 14315367Sahrens avl_tree_t *stream_avl = NULL; 14325367Sahrens char *fromsnap = NULL; 14335367Sahrens char tofs[ZFS_MAXNAMELEN]; 14345367Sahrens char errbuf[1024]; 14355367Sahrens dmu_replay_record_t drre; 14365367Sahrens int error; 14375367Sahrens boolean_t anyerr = B_FALSE; 14386479Sahrens boolean_t softerr = B_FALSE; 14395367Sahrens 14405367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN, 14415367Sahrens "cannot receive")); 14425367Sahrens 14435367Sahrens if (strchr(destname, '@')) { 14445367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 14455367Sahrens "can not specify snapshot name for multi-snapshot stream")); 14465367Sahrens return (zfs_error(hdl, EZFS_BADSTREAM, errbuf)); 14475367Sahrens } 14485367Sahrens 14495367Sahrens assert(drr->drr_type == DRR_BEGIN); 14505367Sahrens assert(drr->drr_u.drr_begin.drr_magic == DMU_BACKUP_MAGIC); 14515367Sahrens assert(drr->drr_u.drr_begin.drr_version == DMU_BACKUP_HEADER_VERSION); 14525367Sahrens 14535367Sahrens /* 14545367Sahrens * Read in the nvlist from the stream. 14555367Sahrens */ 14565367Sahrens if (drr->drr_payloadlen != 0) { 14575367Sahrens if (!flags.isprefix) { 14585367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 14595367Sahrens "must use -d to receive replication " 14605367Sahrens "(send -R) stream")); 14615367Sahrens return (zfs_error(hdl, EZFS_BADSTREAM, errbuf)); 14625367Sahrens } 14635367Sahrens 14645367Sahrens error = recv_read_nvlist(hdl, fd, drr->drr_payloadlen, 14655367Sahrens &stream_nv, flags.byteswap, zc); 14665367Sahrens if (error) { 14675367Sahrens error = zfs_error(hdl, EZFS_BADSTREAM, errbuf); 14685367Sahrens goto out; 14695367Sahrens } 14705367Sahrens } 14715367Sahrens 14725367Sahrens /* 14735367Sahrens * Read in the end record and verify checksum. 14745367Sahrens */ 14755367Sahrens if (0 != (error = recv_read(hdl, fd, &drre, sizeof (drre), 14765367Sahrens flags.byteswap, NULL))) 14775367Sahrens goto out; 14785367Sahrens if (flags.byteswap) { 14795367Sahrens drre.drr_type = BSWAP_32(drre.drr_type); 14805367Sahrens drre.drr_u.drr_end.drr_checksum.zc_word[0] = 14815367Sahrens BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[0]); 14825367Sahrens drre.drr_u.drr_end.drr_checksum.zc_word[1] = 14835367Sahrens BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[1]); 14845367Sahrens drre.drr_u.drr_end.drr_checksum.zc_word[2] = 14855367Sahrens BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[2]); 14865367Sahrens drre.drr_u.drr_end.drr_checksum.zc_word[3] = 14875367Sahrens BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[3]); 14885367Sahrens } 14895367Sahrens if (drre.drr_type != DRR_END) { 14905367Sahrens error = zfs_error(hdl, EZFS_BADSTREAM, errbuf); 14915367Sahrens goto out; 14925367Sahrens } 14935367Sahrens if (!ZIO_CHECKSUM_EQUAL(drre.drr_u.drr_end.drr_checksum, *zc)) { 14945367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 14955367Sahrens "incorrect header checksum")); 14965367Sahrens error = zfs_error(hdl, EZFS_BADSTREAM, errbuf); 14975367Sahrens goto out; 14985367Sahrens } 14995367Sahrens 15005367Sahrens (void) nvlist_lookup_string(stream_nv, "fromsnap", &fromsnap); 15015367Sahrens 15025367Sahrens if (drr->drr_payloadlen != 0) { 15035367Sahrens nvlist_t *stream_fss; 15045367Sahrens 15055367Sahrens VERIFY(0 == nvlist_lookup_nvlist(stream_nv, "fss", 15065367Sahrens &stream_fss)); 15075896Sahl if ((stream_avl = fsavl_create(stream_fss)) == NULL) { 15085896Sahl zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 15095896Sahl "couldn't allocate avl tree")); 15105896Sahl error = zfs_error(hdl, EZFS_NOMEM, errbuf); 15115896Sahl goto out; 15125896Sahl } 15135367Sahrens 15145367Sahrens if (fromsnap != NULL) { 15155367Sahrens (void) strlcpy(tofs, destname, ZFS_MAXNAMELEN); 15165367Sahrens if (flags.isprefix) { 15175367Sahrens int i = strcspn(drr->drr_u.drr_begin.drr_toname, 15185367Sahrens "/@"); 15195367Sahrens /* zfs_receive_one() will create_parents() */ 15205367Sahrens (void) strlcat(tofs, 15215367Sahrens &drr->drr_u.drr_begin.drr_toname[i], 15225367Sahrens ZFS_MAXNAMELEN); 15235367Sahrens *strchr(tofs, '@') = '\0'; 15245367Sahrens } 15256479Sahrens softerr = recv_incremental_replication(hdl, tofs, 15265367Sahrens flags, stream_nv, stream_avl); 15275367Sahrens } 15285367Sahrens } 15295367Sahrens 15305367Sahrens 15315367Sahrens /* Finally, receive each contained stream */ 15325367Sahrens do { 15335367Sahrens /* 15345367Sahrens * we should figure out if it has a recoverable 15355367Sahrens * error, in which case do a recv_skip() and drive on. 15365367Sahrens * Note, if we fail due to already having this guid, 15375367Sahrens * zfs_receive_one() will take care of it (ie, 15385367Sahrens * recv_skip() and return 0). 15395367Sahrens */ 15407366STim.Haley@Sun.COM error = zfs_receive_impl(hdl, destname, flags, fd, 15417366STim.Haley@Sun.COM stream_avl, top_zfs); 15425367Sahrens if (error == ENODATA) { 15435367Sahrens error = 0; 15445367Sahrens break; 15455367Sahrens } 15465367Sahrens anyerr |= error; 15475367Sahrens } while (error == 0); 15485367Sahrens 15495367Sahrens if (drr->drr_payloadlen != 0 && fromsnap != NULL) { 15505367Sahrens /* 15515367Sahrens * Now that we have the fs's they sent us, try the 15525367Sahrens * renames again. 15535367Sahrens */ 15546479Sahrens softerr = recv_incremental_replication(hdl, tofs, flags, 15555367Sahrens stream_nv, stream_avl); 15565367Sahrens } 15575367Sahrens 15585367Sahrens out: 15595367Sahrens fsavl_destroy(stream_avl); 15605367Sahrens if (stream_nv) 15615367Sahrens nvlist_free(stream_nv); 15626479Sahrens if (softerr) 15636479Sahrens error = -2; 15645367Sahrens if (anyerr) 15655367Sahrens error = -1; 15665367Sahrens return (error); 15675367Sahrens } 15685367Sahrens 15695367Sahrens static int 15705367Sahrens recv_skip(libzfs_handle_t *hdl, int fd, boolean_t byteswap) 15715367Sahrens { 15725367Sahrens dmu_replay_record_t *drr; 15735367Sahrens void *buf = malloc(1<<20); 15745367Sahrens 15755367Sahrens /* XXX would be great to use lseek if possible... */ 15765367Sahrens drr = buf; 15775367Sahrens 15785367Sahrens while (recv_read(hdl, fd, drr, sizeof (dmu_replay_record_t), 15795367Sahrens byteswap, NULL) == 0) { 15805367Sahrens if (byteswap) 15815367Sahrens drr->drr_type = BSWAP_32(drr->drr_type); 15825367Sahrens 15835367Sahrens switch (drr->drr_type) { 15845367Sahrens case DRR_BEGIN: 15855367Sahrens /* NB: not to be used on v2 stream packages */ 15865367Sahrens assert(drr->drr_payloadlen == 0); 15875367Sahrens break; 15885367Sahrens 15895367Sahrens case DRR_END: 15905367Sahrens free(buf); 15915367Sahrens return (0); 15925367Sahrens 15935367Sahrens case DRR_OBJECT: 15945367Sahrens if (byteswap) { 15955367Sahrens drr->drr_u.drr_object.drr_bonuslen = 15965367Sahrens BSWAP_32(drr->drr_u.drr_object. 15975367Sahrens drr_bonuslen); 15985367Sahrens } 15995367Sahrens (void) recv_read(hdl, fd, buf, 16005367Sahrens P2ROUNDUP(drr->drr_u.drr_object.drr_bonuslen, 8), 16015367Sahrens B_FALSE, NULL); 16025367Sahrens break; 16035367Sahrens 16045367Sahrens case DRR_WRITE: 16055367Sahrens if (byteswap) { 16065367Sahrens drr->drr_u.drr_write.drr_length = 16077366STim.Haley@Sun.COM BSWAP_64(drr->drr_u.drr_write.drr_length); 16085367Sahrens } 16095367Sahrens (void) recv_read(hdl, fd, buf, 16105367Sahrens drr->drr_u.drr_write.drr_length, B_FALSE, NULL); 16115367Sahrens break; 16125367Sahrens 16135367Sahrens case DRR_FREEOBJECTS: 16145367Sahrens case DRR_FREE: 16155367Sahrens break; 16165367Sahrens 16175367Sahrens default: 16185367Sahrens assert(!"invalid record type"); 16195367Sahrens } 16205367Sahrens } 16215367Sahrens 16225367Sahrens free(buf); 16235367Sahrens return (-1); 16245367Sahrens } 16255367Sahrens 16265367Sahrens /* 16275367Sahrens * Restores a backup of tosnap from the file descriptor specified by infd. 16285367Sahrens */ 16295367Sahrens static int 16305367Sahrens zfs_receive_one(libzfs_handle_t *hdl, int infd, const char *tosnap, 16315367Sahrens recvflags_t flags, dmu_replay_record_t *drr, 16327366STim.Haley@Sun.COM dmu_replay_record_t *drr_noswap, avl_tree_t *stream_avl, 16337366STim.Haley@Sun.COM char **top_zfs) 16345367Sahrens { 16355367Sahrens zfs_cmd_t zc = { 0 }; 16365367Sahrens time_t begin_time; 16375367Sahrens int ioctl_err, ioctl_errno, err, choplen; 16385367Sahrens char *cp; 16395367Sahrens struct drr_begin *drrb = &drr->drr_u.drr_begin; 16405367Sahrens char errbuf[1024]; 16415367Sahrens char chopprefix[ZFS_MAXNAMELEN]; 16425367Sahrens boolean_t newfs = B_FALSE; 16435367Sahrens boolean_t stream_wantsnewfs; 16445367Sahrens uint64_t parent_snapguid = 0; 16455367Sahrens prop_changelist_t *clp = NULL; 16467265Sahrens nvlist_t *snapprops_nvlist = NULL; 16475367Sahrens 16485367Sahrens begin_time = time(NULL); 16495367Sahrens 16505367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN, 16515367Sahrens "cannot receive")); 16525367Sahrens 16535367Sahrens if (stream_avl != NULL) { 16547265Sahrens char *snapname; 16557265Sahrens nvlist_t *fs = fsavl_find(stream_avl, drrb->drr_toguid, 16567265Sahrens &snapname); 16575367Sahrens nvlist_t *props; 16585896Sahl int ret; 16595367Sahrens 16605367Sahrens (void) nvlist_lookup_uint64(fs, "parentfromsnap", 16615367Sahrens &parent_snapguid); 16625367Sahrens err = nvlist_lookup_nvlist(fs, "props", &props); 16635367Sahrens if (err) 16645367Sahrens VERIFY(0 == nvlist_alloc(&props, NV_UNIQUE_NAME, 0)); 16655896Sahl 16665367Sahrens if (flags.canmountoff) { 16675367Sahrens VERIFY(0 == nvlist_add_uint64(props, 16685367Sahrens zfs_prop_to_name(ZFS_PROP_CANMOUNT), 0)); 16695367Sahrens } 16705896Sahl ret = zcmd_write_src_nvlist(hdl, &zc, props); 16715367Sahrens if (err) 16725367Sahrens nvlist_free(props); 16735896Sahl 16747265Sahrens if (0 == nvlist_lookup_nvlist(fs, "snapprops", &props)) { 16757265Sahrens VERIFY(0 == nvlist_lookup_nvlist(props, 16767265Sahrens snapname, &snapprops_nvlist)); 16777265Sahrens } 16787265Sahrens 16795896Sahl if (ret != 0) 16805896Sahl return (-1); 16815367Sahrens } 16825367Sahrens 16835367Sahrens /* 16845367Sahrens * Determine how much of the snapshot name stored in the stream 16855367Sahrens * we are going to tack on to the name they specified on the 16865367Sahrens * command line, and how much we are going to chop off. 16875367Sahrens * 16885367Sahrens * If they specified a snapshot, chop the entire name stored in 16895367Sahrens * the stream. 16905367Sahrens */ 16915367Sahrens (void) strcpy(chopprefix, drrb->drr_toname); 16925367Sahrens if (flags.isprefix) { 16935367Sahrens /* 16945367Sahrens * They specified a fs with -d, we want to tack on 16955367Sahrens * everything but the pool name stored in the stream 16965367Sahrens */ 16975367Sahrens if (strchr(tosnap, '@')) { 16985367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid " 16995367Sahrens "argument - snapshot not allowed with -d")); 17005367Sahrens return (zfs_error(hdl, EZFS_INVALIDNAME, errbuf)); 17015367Sahrens } 17025367Sahrens cp = strchr(chopprefix, '/'); 17035367Sahrens if (cp == NULL) 17045367Sahrens cp = strchr(chopprefix, '@'); 17055367Sahrens *cp = '\0'; 17065367Sahrens } else if (strchr(tosnap, '@') == NULL) { 17075367Sahrens /* 17085367Sahrens * If they specified a filesystem without -d, we want to 17095367Sahrens * tack on everything after the fs specified in the 17105367Sahrens * first name from the stream. 17115367Sahrens */ 17125367Sahrens cp = strchr(chopprefix, '@'); 17135367Sahrens *cp = '\0'; 17145367Sahrens } 17155367Sahrens choplen = strlen(chopprefix); 17165367Sahrens 17175367Sahrens /* 17185367Sahrens * Determine name of destination snapshot, store in zc_value. 17195367Sahrens */ 17205367Sahrens (void) strcpy(zc.zc_value, tosnap); 17215367Sahrens (void) strncat(zc.zc_value, drrb->drr_toname+choplen, 17225367Sahrens sizeof (zc.zc_value)); 17235896Sahl if (!zfs_name_valid(zc.zc_value, ZFS_TYPE_SNAPSHOT)) { 17245896Sahl zcmd_free_nvlists(&zc); 17255367Sahrens return (zfs_error(hdl, EZFS_INVALIDNAME, errbuf)); 17265896Sahl } 17275367Sahrens 17285367Sahrens /* 17295367Sahrens * Determine the name of the origin snapshot, store in zc_string. 17305367Sahrens */ 17315367Sahrens if (drrb->drr_flags & DRR_FLAG_CLONE) { 17325367Sahrens if (guid_to_name(hdl, tosnap, 17335367Sahrens drrb->drr_fromguid, zc.zc_string) != 0) { 17345896Sahl zcmd_free_nvlists(&zc); 17355367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 17365367Sahrens "local origin for clone %s does not exist"), 17375367Sahrens zc.zc_value); 17385367Sahrens return (zfs_error(hdl, EZFS_NOENT, errbuf)); 17395367Sahrens } 17405367Sahrens if (flags.verbose) 17415367Sahrens (void) printf("found clone origin %s\n", zc.zc_string); 17425367Sahrens } 17435367Sahrens 17445367Sahrens stream_wantsnewfs = (drrb->drr_fromguid == NULL || 17455367Sahrens (drrb->drr_flags & DRR_FLAG_CLONE)); 17465367Sahrens 17475367Sahrens if (stream_wantsnewfs) { 17485367Sahrens /* 17495367Sahrens * if the parent fs does not exist, look for it based on 17505367Sahrens * the parent snap GUID 17515367Sahrens */ 17525367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN, 17535367Sahrens "cannot receive new filesystem stream")); 17545367Sahrens 17555367Sahrens (void) strcpy(zc.zc_name, zc.zc_value); 17565367Sahrens cp = strrchr(zc.zc_name, '/'); 17575367Sahrens if (cp) 17585367Sahrens *cp = '\0'; 17595367Sahrens if (cp && 17605367Sahrens !zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) { 17615367Sahrens char suffix[ZFS_MAXNAMELEN]; 17625367Sahrens (void) strcpy(suffix, strrchr(zc.zc_value, '/')); 17635367Sahrens if (guid_to_name(hdl, tosnap, parent_snapguid, 17645367Sahrens zc.zc_value) == 0) { 17655367Sahrens *strchr(zc.zc_value, '@') = '\0'; 17665367Sahrens (void) strcat(zc.zc_value, suffix); 17675367Sahrens } 17685367Sahrens } 17695367Sahrens } else { 17705367Sahrens /* 17715367Sahrens * if the fs does not exist, look for it based on the 17725367Sahrens * fromsnap GUID 17735367Sahrens */ 17745367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN, 17755367Sahrens "cannot receive incremental stream")); 17765367Sahrens 17775367Sahrens (void) strcpy(zc.zc_name, zc.zc_value); 17785367Sahrens *strchr(zc.zc_name, '@') = '\0'; 17795367Sahrens 17805367Sahrens if (!zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) { 17815367Sahrens char snap[ZFS_MAXNAMELEN]; 17825367Sahrens (void) strcpy(snap, strchr(zc.zc_value, '@')); 17835367Sahrens if (guid_to_name(hdl, tosnap, drrb->drr_fromguid, 17845367Sahrens zc.zc_value) == 0) { 17855367Sahrens *strchr(zc.zc_value, '@') = '\0'; 17865367Sahrens (void) strcat(zc.zc_value, snap); 17875367Sahrens } 17885367Sahrens } 17895367Sahrens } 17905367Sahrens 17915367Sahrens (void) strcpy(zc.zc_name, zc.zc_value); 17925367Sahrens *strchr(zc.zc_name, '@') = '\0'; 17935367Sahrens 17945367Sahrens if (zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) { 17955367Sahrens zfs_handle_t *zhp; 17965367Sahrens /* 17975367Sahrens * Destination fs exists. Therefore this should either 17985367Sahrens * be an incremental, or the stream specifies a new fs 17995367Sahrens * (full stream or clone) and they want us to blow it 18005367Sahrens * away (and have therefore specified -F and removed any 18015367Sahrens * snapshots). 18025367Sahrens */ 18035367Sahrens 18045367Sahrens if (stream_wantsnewfs) { 18055367Sahrens if (!flags.force) { 18065896Sahl zcmd_free_nvlists(&zc); 18075367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 18085367Sahrens "destination '%s' exists\n" 18095367Sahrens "must specify -F to overwrite it"), 18105367Sahrens zc.zc_name); 18115367Sahrens return (zfs_error(hdl, EZFS_EXISTS, errbuf)); 18125367Sahrens } 18135367Sahrens if (ioctl(hdl->libzfs_fd, ZFS_IOC_SNAPSHOT_LIST_NEXT, 18145367Sahrens &zc) == 0) { 18155896Sahl zcmd_free_nvlists(&zc); 18165367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 18175367Sahrens "destination has snapshots (eg. %s)\n" 18185367Sahrens "must destroy them to overwrite it"), 18195367Sahrens zc.zc_name); 18205367Sahrens return (zfs_error(hdl, EZFS_EXISTS, errbuf)); 18215367Sahrens } 18225367Sahrens } 18235367Sahrens 18245896Sahl if ((zhp = zfs_open(hdl, zc.zc_name, 18255896Sahl ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME)) == NULL) { 18265896Sahl zcmd_free_nvlists(&zc); 18275367Sahrens return (-1); 18285896Sahl } 18295896Sahl 18305367Sahrens if (stream_wantsnewfs && 18315367Sahrens zhp->zfs_dmustats.dds_origin[0]) { 18325896Sahl zcmd_free_nvlists(&zc); 18335367Sahrens zfs_close(zhp); 18345367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 18355367Sahrens "destination '%s' is a clone\n" 18365367Sahrens "must destroy it to overwrite it"), 18375367Sahrens zc.zc_name); 18385367Sahrens return (zfs_error(hdl, EZFS_EXISTS, errbuf)); 18395367Sahrens } 18405367Sahrens 18415367Sahrens if (!flags.dryrun && zhp->zfs_type == ZFS_TYPE_FILESYSTEM && 18425367Sahrens stream_wantsnewfs) { 18435367Sahrens /* We can't do online recv in this case */ 18447366STim.Haley@Sun.COM clp = changelist_gather(zhp, ZFS_PROP_NAME, 0, 0); 18455896Sahl if (clp == NULL) { 184610200SLori.Alt@Sun.COM zfs_close(zhp); 18475896Sahl zcmd_free_nvlists(&zc); 18485367Sahrens return (-1); 18495896Sahl } 18505367Sahrens if (changelist_prefix(clp) != 0) { 18515367Sahrens changelist_free(clp); 185210200SLori.Alt@Sun.COM zfs_close(zhp); 18535896Sahl zcmd_free_nvlists(&zc); 18545367Sahrens return (-1); 18555367Sahrens } 18565367Sahrens } 18575367Sahrens zfs_close(zhp); 18585367Sahrens } else { 18595367Sahrens /* 18605896Sahl * Destination filesystem does not exist. Therefore we better 18615896Sahl * be creating a new filesystem (either from a full backup, or 18625896Sahl * a clone). It would therefore be invalid if the user 18635896Sahl * specified only the pool name (i.e. if the destination name 18645896Sahl * contained no slash character). 18655367Sahrens */ 18665896Sahl if (!stream_wantsnewfs || 18675896Sahl (cp = strrchr(zc.zc_name, '/')) == NULL) { 18685896Sahl zcmd_free_nvlists(&zc); 18695367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 18705367Sahrens "destination '%s' does not exist"), zc.zc_name); 18715367Sahrens return (zfs_error(hdl, EZFS_NOENT, errbuf)); 18725367Sahrens } 18735367Sahrens 18745896Sahl /* 18755896Sahl * Trim off the final dataset component so we perform the 18765896Sahl * recvbackup ioctl to the filesystems's parent. 18775896Sahl */ 18785896Sahl *cp = '\0'; 18795367Sahrens 18805896Sahl if (flags.isprefix && !flags.dryrun && 18815896Sahl create_parents(hdl, zc.zc_value, strlen(tosnap)) != 0) { 18825896Sahl zcmd_free_nvlists(&zc); 18835896Sahl return (zfs_error(hdl, EZFS_BADRESTORE, errbuf)); 18845367Sahrens } 18855367Sahrens 18865367Sahrens newfs = B_TRUE; 18875367Sahrens } 18885367Sahrens 18895367Sahrens zc.zc_begin_record = drr_noswap->drr_u.drr_begin; 18905367Sahrens zc.zc_cookie = infd; 18915367Sahrens zc.zc_guid = flags.force; 18925367Sahrens if (flags.verbose) { 18935367Sahrens (void) printf("%s %s stream of %s into %s\n", 18945367Sahrens flags.dryrun ? "would receive" : "receiving", 18955367Sahrens drrb->drr_fromguid ? "incremental" : "full", 18965367Sahrens drrb->drr_toname, zc.zc_value); 18975367Sahrens (void) fflush(stdout); 18985367Sahrens } 18995367Sahrens 19005896Sahl if (flags.dryrun) { 19015896Sahl zcmd_free_nvlists(&zc); 19025367Sahrens return (recv_skip(hdl, infd, flags.byteswap)); 19035896Sahl } 19045367Sahrens 19055367Sahrens err = ioctl_err = zfs_ioctl(hdl, ZFS_IOC_RECV, &zc); 19065367Sahrens ioctl_errno = errno; 19077265Sahrens zcmd_free_nvlists(&zc); 19087265Sahrens 19097265Sahrens if (err == 0 && snapprops_nvlist) { 19107265Sahrens zfs_cmd_t zc2 = { 0 }; 19117265Sahrens 19127273Sahrens (void) strcpy(zc2.zc_name, zc.zc_value); 19137265Sahrens if (zcmd_write_src_nvlist(hdl, &zc2, snapprops_nvlist) == 0) { 19147265Sahrens (void) zfs_ioctl(hdl, ZFS_IOC_SET_PROP, &zc2); 19157265Sahrens zcmd_free_nvlists(&zc2); 19167265Sahrens } 19177265Sahrens } 19187265Sahrens 19195367Sahrens if (err && (ioctl_errno == ENOENT || ioctl_errno == ENODEV)) { 19205367Sahrens /* 19215367Sahrens * It may be that this snapshot already exists, 19225367Sahrens * in which case we want to consume & ignore it 19235367Sahrens * rather than failing. 19245367Sahrens */ 19255367Sahrens avl_tree_t *local_avl; 19265367Sahrens nvlist_t *local_nv, *fs; 19275367Sahrens char *cp = strchr(zc.zc_value, '@'); 19285367Sahrens 19295367Sahrens /* 19305367Sahrens * XXX Do this faster by just iterating over snaps in 19315367Sahrens * this fs. Also if zc_value does not exist, we will 19325367Sahrens * get a strange "does not exist" error message. 19335367Sahrens */ 19345367Sahrens *cp = '\0'; 19355367Sahrens if (gather_nvlist(hdl, zc.zc_value, NULL, NULL, 19365367Sahrens &local_nv, &local_avl) == 0) { 19375367Sahrens *cp = '@'; 19385367Sahrens fs = fsavl_find(local_avl, drrb->drr_toguid, NULL); 19395367Sahrens fsavl_destroy(local_avl); 19405367Sahrens nvlist_free(local_nv); 19415367Sahrens 19425367Sahrens if (fs != NULL) { 19435367Sahrens if (flags.verbose) { 19445367Sahrens (void) printf("snap %s already exists; " 19455367Sahrens "ignoring\n", zc.zc_value); 19465367Sahrens } 19475367Sahrens ioctl_err = recv_skip(hdl, infd, 19485367Sahrens flags.byteswap); 19495367Sahrens } 19505367Sahrens } 19515367Sahrens *cp = '@'; 19525367Sahrens } 19535367Sahrens 19545896Sahl 19555367Sahrens if (ioctl_err != 0) { 19565367Sahrens switch (ioctl_errno) { 19575367Sahrens case ENODEV: 19585367Sahrens cp = strchr(zc.zc_value, '@'); 19595367Sahrens *cp = '\0'; 19605367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 19615367Sahrens "most recent snapshot of %s does not\n" 19625367Sahrens "match incremental source"), zc.zc_value); 19635367Sahrens (void) zfs_error(hdl, EZFS_BADRESTORE, errbuf); 19645367Sahrens *cp = '@'; 19655367Sahrens break; 19665367Sahrens case ETXTBSY: 19675367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 19685367Sahrens "destination %s has been modified\n" 19695367Sahrens "since most recent snapshot"), zc.zc_name); 19705367Sahrens (void) zfs_error(hdl, EZFS_BADRESTORE, errbuf); 19715367Sahrens break; 19725367Sahrens case EEXIST: 19735367Sahrens cp = strchr(zc.zc_value, '@'); 19745367Sahrens if (newfs) { 19755367Sahrens /* it's the containing fs that exists */ 19765367Sahrens *cp = '\0'; 19775367Sahrens } 19785367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 19795367Sahrens "destination already exists")); 19805367Sahrens (void) zfs_error_fmt(hdl, EZFS_EXISTS, 19815367Sahrens dgettext(TEXT_DOMAIN, "cannot restore to %s"), 19825367Sahrens zc.zc_value); 19835367Sahrens *cp = '@'; 19845367Sahrens break; 19855367Sahrens case EINVAL: 19865367Sahrens (void) zfs_error(hdl, EZFS_BADSTREAM, errbuf); 19875367Sahrens break; 19885367Sahrens case ECKSUM: 19895367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 19905367Sahrens "invalid stream (checksum mismatch)")); 19915367Sahrens (void) zfs_error(hdl, EZFS_BADSTREAM, errbuf); 19925367Sahrens break; 19935367Sahrens default: 19945367Sahrens (void) zfs_standard_error(hdl, ioctl_errno, errbuf); 19955367Sahrens } 19965367Sahrens } 19975367Sahrens 19985367Sahrens /* 199910588SEric.Taylor@Sun.COM * Mount the target filesystem (if created). Also mount any 200010588SEric.Taylor@Sun.COM * children of the target filesystem if we did a replication 200110588SEric.Taylor@Sun.COM * receive (indicated by stream_avl being non-NULL). 20025367Sahrens */ 20035367Sahrens cp = strchr(zc.zc_value, '@'); 20045367Sahrens if (cp && (ioctl_err == 0 || !newfs)) { 20055367Sahrens zfs_handle_t *h; 20065367Sahrens 20075367Sahrens *cp = '\0'; 20085367Sahrens h = zfs_open(hdl, zc.zc_value, 20095367Sahrens ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME); 20106476Sahl if (h != NULL) { 20115367Sahrens if (h->zfs_type == ZFS_TYPE_VOLUME) { 20127366STim.Haley@Sun.COM *cp = '@'; 201310200SLori.Alt@Sun.COM } else if (newfs || stream_avl) { 20147366STim.Haley@Sun.COM /* 20157366STim.Haley@Sun.COM * Track the first/top of hierarchy fs, 20167366STim.Haley@Sun.COM * for mounting and sharing later. 20177366STim.Haley@Sun.COM */ 20187366STim.Haley@Sun.COM if (top_zfs && *top_zfs == NULL) 20197366STim.Haley@Sun.COM *top_zfs = zfs_strdup(hdl, zc.zc_value); 20205367Sahrens } 20216476Sahl zfs_close(h); 20225367Sahrens } 20237366STim.Haley@Sun.COM *cp = '@'; 20245367Sahrens } 20255367Sahrens 20265367Sahrens if (clp) { 20275367Sahrens err |= changelist_postfix(clp); 20285367Sahrens changelist_free(clp); 20295367Sahrens } 20305367Sahrens 20315367Sahrens if (err || ioctl_err) 20325367Sahrens return (-1); 20335367Sahrens 20345367Sahrens if (flags.verbose) { 20355367Sahrens char buf1[64]; 20365367Sahrens char buf2[64]; 20375367Sahrens uint64_t bytes = zc.zc_cookie; 20385367Sahrens time_t delta = time(NULL) - begin_time; 20395367Sahrens if (delta == 0) 20405367Sahrens delta = 1; 20415367Sahrens zfs_nicenum(bytes, buf1, sizeof (buf1)); 20425367Sahrens zfs_nicenum(bytes/delta, buf2, sizeof (buf1)); 20435367Sahrens 20445367Sahrens (void) printf("received %sB stream in %lu seconds (%sB/sec)\n", 20455367Sahrens buf1, delta, buf2); 20465367Sahrens } 20475367Sahrens 20485367Sahrens return (0); 20495367Sahrens } 20505367Sahrens 20517366STim.Haley@Sun.COM static int 20527366STim.Haley@Sun.COM zfs_receive_impl(libzfs_handle_t *hdl, const char *tosnap, recvflags_t flags, 20537366STim.Haley@Sun.COM int infd, avl_tree_t *stream_avl, char **top_zfs) 20545367Sahrens { 20555367Sahrens int err; 20565367Sahrens dmu_replay_record_t drr, drr_noswap; 20575367Sahrens struct drr_begin *drrb = &drr.drr_u.drr_begin; 20585367Sahrens char errbuf[1024]; 20595367Sahrens zio_cksum_t zcksum = { 0 }; 20605367Sahrens 20615367Sahrens (void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN, 20625367Sahrens "cannot receive")); 20635367Sahrens 20645367Sahrens if (flags.isprefix && 20655367Sahrens !zfs_dataset_exists(hdl, tosnap, ZFS_TYPE_DATASET)) { 20665367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "specified fs " 20675367Sahrens "(%s) does not exist"), tosnap); 20685367Sahrens return (zfs_error(hdl, EZFS_NOENT, errbuf)); 20695367Sahrens } 20705367Sahrens 20715367Sahrens /* read in the BEGIN record */ 20725367Sahrens if (0 != (err = recv_read(hdl, infd, &drr, sizeof (drr), B_FALSE, 20735367Sahrens &zcksum))) 20745367Sahrens return (err); 20755367Sahrens 20765367Sahrens if (drr.drr_type == DRR_END || drr.drr_type == BSWAP_32(DRR_END)) { 20775367Sahrens /* It's the double end record at the end of a package */ 20785367Sahrens return (ENODATA); 20795367Sahrens } 20805367Sahrens 20815367Sahrens /* the kernel needs the non-byteswapped begin record */ 20825367Sahrens drr_noswap = drr; 20835367Sahrens 20845446Sahrens flags.byteswap = B_FALSE; 20855367Sahrens if (drrb->drr_magic == BSWAP_64(DMU_BACKUP_MAGIC)) { 20865367Sahrens /* 20875367Sahrens * We computed the checksum in the wrong byteorder in 20885367Sahrens * recv_read() above; do it again correctly. 20895367Sahrens */ 20905367Sahrens bzero(&zcksum, sizeof (zio_cksum_t)); 20915367Sahrens fletcher_4_incremental_byteswap(&drr, sizeof (drr), &zcksum); 20925367Sahrens flags.byteswap = B_TRUE; 20935367Sahrens 20945367Sahrens drr.drr_type = BSWAP_32(drr.drr_type); 20955367Sahrens drr.drr_payloadlen = BSWAP_32(drr.drr_payloadlen); 20965367Sahrens drrb->drr_magic = BSWAP_64(drrb->drr_magic); 20975367Sahrens drrb->drr_version = BSWAP_64(drrb->drr_version); 20985367Sahrens drrb->drr_creation_time = BSWAP_64(drrb->drr_creation_time); 20995367Sahrens drrb->drr_type = BSWAP_32(drrb->drr_type); 21005367Sahrens drrb->drr_flags = BSWAP_32(drrb->drr_flags); 21015367Sahrens drrb->drr_toguid = BSWAP_64(drrb->drr_toguid); 21025367Sahrens drrb->drr_fromguid = BSWAP_64(drrb->drr_fromguid); 21035367Sahrens } 21045367Sahrens 21055367Sahrens if (drrb->drr_magic != DMU_BACKUP_MAGIC || drr.drr_type != DRR_BEGIN) { 21065367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid " 21075367Sahrens "stream (bad magic number)")); 21085367Sahrens return (zfs_error(hdl, EZFS_BADSTREAM, errbuf)); 21095367Sahrens } 21105367Sahrens 21115367Sahrens if (strchr(drrb->drr_toname, '@') == NULL) { 21125367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid " 21135367Sahrens "stream (bad snapshot name)")); 21145367Sahrens return (zfs_error(hdl, EZFS_BADSTREAM, errbuf)); 21155367Sahrens } 21165367Sahrens 21175367Sahrens if (drrb->drr_version == DMU_BACKUP_STREAM_VERSION) { 21185367Sahrens return (zfs_receive_one(hdl, infd, tosnap, flags, 21197366STim.Haley@Sun.COM &drr, &drr_noswap, stream_avl, top_zfs)); 21205367Sahrens } else if (drrb->drr_version == DMU_BACKUP_HEADER_VERSION) { 21215367Sahrens return (zfs_receive_package(hdl, infd, tosnap, flags, 21227366STim.Haley@Sun.COM &drr, &zcksum, top_zfs)); 21235367Sahrens } else { 21245367Sahrens zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, 21255367Sahrens "stream is unsupported version %llu"), 21265367Sahrens drrb->drr_version); 21275367Sahrens return (zfs_error(hdl, EZFS_BADSTREAM, errbuf)); 21285367Sahrens } 21295367Sahrens } 21307366STim.Haley@Sun.COM 21317366STim.Haley@Sun.COM /* 21327366STim.Haley@Sun.COM * Restores a backup of tosnap from the file descriptor specified by infd. 21337366STim.Haley@Sun.COM * Return 0 on total success, -2 if some things couldn't be 21347366STim.Haley@Sun.COM * destroyed/renamed/promoted, -1 if some things couldn't be received. 21357366STim.Haley@Sun.COM * (-1 will override -2). 21367366STim.Haley@Sun.COM */ 21377366STim.Haley@Sun.COM int 21387366STim.Haley@Sun.COM zfs_receive(libzfs_handle_t *hdl, const char *tosnap, recvflags_t flags, 21397366STim.Haley@Sun.COM int infd, avl_tree_t *stream_avl) 21407366STim.Haley@Sun.COM { 21417366STim.Haley@Sun.COM char *top_zfs = NULL; 21427366STim.Haley@Sun.COM int err; 21437366STim.Haley@Sun.COM 21447366STim.Haley@Sun.COM err = zfs_receive_impl(hdl, tosnap, flags, infd, stream_avl, &top_zfs); 21457366STim.Haley@Sun.COM 21468584SLori.Alt@Sun.COM if (err == 0 && !flags.nomount && top_zfs) { 21477366STim.Haley@Sun.COM zfs_handle_t *zhp; 21487366STim.Haley@Sun.COM prop_changelist_t *clp; 21497366STim.Haley@Sun.COM 21507366STim.Haley@Sun.COM zhp = zfs_open(hdl, top_zfs, ZFS_TYPE_FILESYSTEM); 21517366STim.Haley@Sun.COM if (zhp != NULL) { 21527366STim.Haley@Sun.COM clp = changelist_gather(zhp, ZFS_PROP_MOUNTPOINT, 21537366STim.Haley@Sun.COM CL_GATHER_MOUNT_ALWAYS, 0); 21547366STim.Haley@Sun.COM zfs_close(zhp); 21557366STim.Haley@Sun.COM if (clp != NULL) { 21567366STim.Haley@Sun.COM /* mount and share received datasets */ 21577366STim.Haley@Sun.COM err = changelist_postfix(clp); 21587366STim.Haley@Sun.COM changelist_free(clp); 21597366STim.Haley@Sun.COM } 21607366STim.Haley@Sun.COM } 21617366STim.Haley@Sun.COM if (zhp == NULL || clp == NULL || err) 21627366STim.Haley@Sun.COM err = -1; 21637366STim.Haley@Sun.COM } 21647366STim.Haley@Sun.COM if (top_zfs) 21657366STim.Haley@Sun.COM free(top_zfs); 21667366STim.Haley@Sun.COM 21677366STim.Haley@Sun.COM return (err); 21687366STim.Haley@Sun.COM } 2169