]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - cddl/contrib/opensolaris/lib/libzfs_core/common/libzfs_core.c
Import tzdata 2018c
[FreeBSD/FreeBSD.git] / cddl / contrib / opensolaris / lib / libzfs_core / common / libzfs_core.c
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21
22 /*
23  * Copyright (c) 2012, 2017 by Delphix. All rights reserved.
24  * Copyright (c) 2013 Steven Hartland. All rights reserved.
25  * Copyright (c) 2014 Integros [integros.com]
26  * Copyright 2017 RackTop Systems.
27  */
28
29 /*
30  * LibZFS_Core (lzc) is intended to replace most functionality in libzfs.
31  * It has the following characteristics:
32  *
33  *  - Thread Safe.  libzfs_core is accessible concurrently from multiple
34  *  threads.  This is accomplished primarily by avoiding global data
35  *  (e.g. caching).  Since it's thread-safe, there is no reason for a
36  *  process to have multiple libzfs "instances".  Therefore, we store
37  *  our few pieces of data (e.g. the file descriptor) in global
38  *  variables.  The fd is reference-counted so that the libzfs_core
39  *  library can be "initialized" multiple times (e.g. by different
40  *  consumers within the same process).
41  *
42  *  - Committed Interface.  The libzfs_core interface will be committed,
43  *  therefore consumers can compile against it and be confident that
44  *  their code will continue to work on future releases of this code.
45  *  Currently, the interface is Evolving (not Committed), but we intend
46  *  to commit to it once it is more complete and we determine that it
47  *  meets the needs of all consumers.
48  *
49  *  - Programatic Error Handling.  libzfs_core communicates errors with
50  *  defined error numbers, and doesn't print anything to stdout/stderr.
51  *
52  *  - Thin Layer.  libzfs_core is a thin layer, marshaling arguments
53  *  to/from the kernel ioctls.  There is generally a 1:1 correspondence
54  *  between libzfs_core functions and ioctls to /dev/zfs.
55  *
56  *  - Clear Atomicity.  Because libzfs_core functions are generally 1:1
57  *  with kernel ioctls, and kernel ioctls are general atomic, each
58  *  libzfs_core function is atomic.  For example, creating multiple
59  *  snapshots with a single call to lzc_snapshot() is atomic -- it
60  *  can't fail with only some of the requested snapshots created, even
61  *  in the event of power loss or system crash.
62  *
63  *  - Continued libzfs Support.  Some higher-level operations (e.g.
64  *  support for "zfs send -R") are too complicated to fit the scope of
65  *  libzfs_core.  This functionality will continue to live in libzfs.
66  *  Where appropriate, libzfs will use the underlying atomic operations
67  *  of libzfs_core.  For example, libzfs may implement "zfs send -R |
68  *  zfs receive" by using individual "send one snapshot", rename,
69  *  destroy, and "receive one snapshot" operations in libzfs_core.
70  *  /sbin/zfs and /zbin/zpool will link with both libzfs and
71  *  libzfs_core.  Other consumers should aim to use only libzfs_core,
72  *  since that will be the supported, stable interface going forwards.
73  */
74
75 #define _IN_LIBZFS_CORE_
76
77 #include <libzfs_core.h>
78 #include <ctype.h>
79 #include <unistd.h>
80 #include <stdlib.h>
81 #include <string.h>
82 #include <errno.h>
83 #include <fcntl.h>
84 #include <pthread.h>
85 #include <sys/nvpair.h>
86 #include <sys/param.h>
87 #include <sys/types.h>
88 #include <sys/stat.h>
89 #include <sys/zfs_ioctl.h>
90 #include "libzfs_core_compat.h"
91 #include "libzfs_compat.h"
92
93 #ifdef __FreeBSD__
94 extern int zfs_ioctl_version;
95 #endif
96
97 static int g_fd;
98 static pthread_mutex_t g_lock = PTHREAD_MUTEX_INITIALIZER;
99 static int g_refcount;
100
101 int
102 libzfs_core_init(void)
103 {
104         (void) pthread_mutex_lock(&g_lock);
105         if (g_refcount == 0) {
106                 g_fd = open("/dev/zfs", O_RDWR);
107                 if (g_fd < 0) {
108                         (void) pthread_mutex_unlock(&g_lock);
109                         return (errno);
110                 }
111         }
112         g_refcount++;
113         (void) pthread_mutex_unlock(&g_lock);
114
115         return (0);
116 }
117
118 void
119 libzfs_core_fini(void)
120 {
121         (void) pthread_mutex_lock(&g_lock);
122         ASSERT3S(g_refcount, >, 0);
123         g_refcount--;
124         if (g_refcount == 0)
125                 (void) close(g_fd);
126         (void) pthread_mutex_unlock(&g_lock);
127 }
128
129 static int
130 lzc_ioctl(zfs_ioc_t ioc, const char *name,
131     nvlist_t *source, nvlist_t **resultp)
132 {
133         zfs_cmd_t zc = { 0 };
134         int error = 0;
135         char *packed;
136 #ifdef __FreeBSD__
137         nvlist_t *oldsource;
138 #endif
139         size_t size;
140
141         ASSERT3S(g_refcount, >, 0);
142
143         (void) strlcpy(zc.zc_name, name, sizeof (zc.zc_name));
144
145 #ifdef __FreeBSD__
146         if (zfs_ioctl_version == ZFS_IOCVER_UNDEF)
147                 zfs_ioctl_version = get_zfs_ioctl_version();
148
149         if (zfs_ioctl_version < ZFS_IOCVER_LZC) {
150                 oldsource = source;
151                 error = lzc_compat_pre(&zc, &ioc, &source);
152                 if (error)
153                         return (error);
154         }
155 #endif
156
157         packed = fnvlist_pack(source, &size);
158         zc.zc_nvlist_src = (uint64_t)(uintptr_t)packed;
159         zc.zc_nvlist_src_size = size;
160
161         if (resultp != NULL) {
162                 *resultp = NULL;
163                 if (ioc == ZFS_IOC_CHANNEL_PROGRAM) {
164                         zc.zc_nvlist_dst_size = fnvlist_lookup_uint64(source,
165                             ZCP_ARG_MEMLIMIT);
166                 } else {
167                         zc.zc_nvlist_dst_size = MAX(size * 2, 128 * 1024);
168                 }
169                 zc.zc_nvlist_dst = (uint64_t)(uintptr_t)
170                     malloc(zc.zc_nvlist_dst_size);
171 #ifdef illumos
172                 if (zc.zc_nvlist_dst == NULL) {
173 #else
174                 if (zc.zc_nvlist_dst == 0) {
175 #endif
176                         error = ENOMEM;
177                         goto out;
178                 }
179         }
180
181         while (ioctl(g_fd, ioc, &zc) != 0) {
182                 /*
183                  * If ioctl exited with ENOMEM, we retry the ioctl after
184                  * increasing the size of the destination nvlist.
185                  *
186                  * Channel programs that exit with ENOMEM ran over the
187                  * lua memory sandbox; they should not be retried.
188                  */
189                 if (errno == ENOMEM && resultp != NULL &&
190                     ioc != ZFS_IOC_CHANNEL_PROGRAM) {
191                         free((void *)(uintptr_t)zc.zc_nvlist_dst);
192                         zc.zc_nvlist_dst_size *= 2;
193                         zc.zc_nvlist_dst = (uint64_t)(uintptr_t)
194                             malloc(zc.zc_nvlist_dst_size);
195 #ifdef illumos
196                         if (zc.zc_nvlist_dst == NULL) {
197 #else
198                         if (zc.zc_nvlist_dst == 0) {
199 #endif
200                                 error = ENOMEM;
201                                 goto out;
202                         }
203                 } else {
204                         error = errno;
205                         break;
206                 }
207         }
208
209 #ifdef __FreeBSD__
210         if (zfs_ioctl_version < ZFS_IOCVER_LZC)
211                 lzc_compat_post(&zc, ioc);
212 #endif
213         if (zc.zc_nvlist_dst_filled) {
214                 *resultp = fnvlist_unpack((void *)(uintptr_t)zc.zc_nvlist_dst,
215                     zc.zc_nvlist_dst_size);
216         }
217 #ifdef __FreeBSD__
218         if (zfs_ioctl_version < ZFS_IOCVER_LZC)
219                 lzc_compat_outnvl(&zc, ioc, resultp);
220 #endif
221 out:
222 #ifdef __FreeBSD__
223         if (zfs_ioctl_version < ZFS_IOCVER_LZC) {
224                 if (source != oldsource)
225                         nvlist_free(source);
226                 source = oldsource;
227         }
228 #endif
229         fnvlist_pack_free(packed, size);
230         free((void *)(uintptr_t)zc.zc_nvlist_dst);
231         return (error);
232 }
233
234 int
235 lzc_create(const char *fsname, enum lzc_dataset_type type, nvlist_t *props)
236 {
237         int error;
238         nvlist_t *args = fnvlist_alloc();
239         fnvlist_add_int32(args, "type", (dmu_objset_type_t)type);
240         if (props != NULL)
241                 fnvlist_add_nvlist(args, "props", props);
242         error = lzc_ioctl(ZFS_IOC_CREATE, fsname, args, NULL);
243         nvlist_free(args);
244         return (error);
245 }
246
247 int
248 lzc_clone(const char *fsname, const char *origin,
249     nvlist_t *props)
250 {
251         int error;
252         nvlist_t *args = fnvlist_alloc();
253         fnvlist_add_string(args, "origin", origin);
254         if (props != NULL)
255                 fnvlist_add_nvlist(args, "props", props);
256         error = lzc_ioctl(ZFS_IOC_CLONE, fsname, args, NULL);
257         nvlist_free(args);
258         return (error);
259 }
260
261 int
262 lzc_promote(const char *fsname, char *snapnamebuf, int snapnamelen)
263 {
264         /*
265          * The promote ioctl is still legacy, so we need to construct our
266          * own zfs_cmd_t rather than using lzc_ioctl().
267          */
268         zfs_cmd_t zc = { 0 };
269
270         ASSERT3S(g_refcount, >, 0);
271         VERIFY3S(g_fd, !=, -1);
272
273         (void) strlcpy(zc.zc_name, fsname, sizeof (zc.zc_name));
274         if (ioctl(g_fd, ZFS_IOC_PROMOTE, &zc) != 0) {
275                 int error = errno;
276                 if (error == EEXIST && snapnamebuf != NULL)
277                         (void) strlcpy(snapnamebuf, zc.zc_string, snapnamelen);
278                 return (error);
279         }
280         return (0);
281 }
282
283 /*
284  * Creates snapshots.
285  *
286  * The keys in the snaps nvlist are the snapshots to be created.
287  * They must all be in the same pool.
288  *
289  * The props nvlist is properties to set.  Currently only user properties
290  * are supported.  { user:prop_name -> string value }
291  *
292  * The returned results nvlist will have an entry for each snapshot that failed.
293  * The value will be the (int32) error code.
294  *
295  * The return value will be 0 if all snapshots were created, otherwise it will
296  * be the errno of a (unspecified) snapshot that failed.
297  */
298 int
299 lzc_snapshot(nvlist_t *snaps, nvlist_t *props, nvlist_t **errlist)
300 {
301         nvpair_t *elem;
302         nvlist_t *args;
303         int error;
304         char pool[ZFS_MAX_DATASET_NAME_LEN];
305
306         *errlist = NULL;
307
308         /* determine the pool name */
309         elem = nvlist_next_nvpair(snaps, NULL);
310         if (elem == NULL)
311                 return (0);
312         (void) strlcpy(pool, nvpair_name(elem), sizeof (pool));
313         pool[strcspn(pool, "/@")] = '\0';
314
315         args = fnvlist_alloc();
316         fnvlist_add_nvlist(args, "snaps", snaps);
317         if (props != NULL)
318                 fnvlist_add_nvlist(args, "props", props);
319
320         error = lzc_ioctl(ZFS_IOC_SNAPSHOT, pool, args, errlist);
321         nvlist_free(args);
322
323         return (error);
324 }
325
326 /*
327  * Destroys snapshots.
328  *
329  * The keys in the snaps nvlist are the snapshots to be destroyed.
330  * They must all be in the same pool.
331  *
332  * Snapshots that do not exist will be silently ignored.
333  *
334  * If 'defer' is not set, and a snapshot has user holds or clones, the
335  * destroy operation will fail and none of the snapshots will be
336  * destroyed.
337  *
338  * If 'defer' is set, and a snapshot has user holds or clones, it will be
339  * marked for deferred destruction, and will be destroyed when the last hold
340  * or clone is removed/destroyed.
341  *
342  * The return value will be 0 if all snapshots were destroyed (or marked for
343  * later destruction if 'defer' is set) or didn't exist to begin with.
344  *
345  * Otherwise the return value will be the errno of a (unspecified) snapshot
346  * that failed, no snapshots will be destroyed, and the errlist will have an
347  * entry for each snapshot that failed.  The value in the errlist will be
348  * the (int32) error code.
349  */
350 int
351 lzc_destroy_snaps(nvlist_t *snaps, boolean_t defer, nvlist_t **errlist)
352 {
353         nvpair_t *elem;
354         nvlist_t *args;
355         int error;
356         char pool[ZFS_MAX_DATASET_NAME_LEN];
357
358         /* determine the pool name */
359         elem = nvlist_next_nvpair(snaps, NULL);
360         if (elem == NULL)
361                 return (0);
362         (void) strlcpy(pool, nvpair_name(elem), sizeof (pool));
363         pool[strcspn(pool, "/@")] = '\0';
364
365         args = fnvlist_alloc();
366         fnvlist_add_nvlist(args, "snaps", snaps);
367         if (defer)
368                 fnvlist_add_boolean(args, "defer");
369
370         error = lzc_ioctl(ZFS_IOC_DESTROY_SNAPS, pool, args, errlist);
371         nvlist_free(args);
372
373         return (error);
374 }
375
376 int
377 lzc_snaprange_space(const char *firstsnap, const char *lastsnap,
378     uint64_t *usedp)
379 {
380         nvlist_t *args;
381         nvlist_t *result;
382         int err;
383         char fs[ZFS_MAX_DATASET_NAME_LEN];
384         char *atp;
385
386         /* determine the fs name */
387         (void) strlcpy(fs, firstsnap, sizeof (fs));
388         atp = strchr(fs, '@');
389         if (atp == NULL)
390                 return (EINVAL);
391         *atp = '\0';
392
393         args = fnvlist_alloc();
394         fnvlist_add_string(args, "firstsnap", firstsnap);
395
396         err = lzc_ioctl(ZFS_IOC_SPACE_SNAPS, lastsnap, args, &result);
397         nvlist_free(args);
398         if (err == 0)
399                 *usedp = fnvlist_lookup_uint64(result, "used");
400         fnvlist_free(result);
401
402         return (err);
403 }
404
405 boolean_t
406 lzc_exists(const char *dataset)
407 {
408         /*
409          * The objset_stats ioctl is still legacy, so we need to construct our
410          * own zfs_cmd_t rather than using lzc_ioctl().
411          */
412         zfs_cmd_t zc = { 0 };
413
414         (void) strlcpy(zc.zc_name, dataset, sizeof (zc.zc_name));
415         return (ioctl(g_fd, ZFS_IOC_OBJSET_STATS, &zc) == 0);
416 }
417
418 /*
419  * Create "user holds" on snapshots.  If there is a hold on a snapshot,
420  * the snapshot can not be destroyed.  (However, it can be marked for deletion
421  * by lzc_destroy_snaps(defer=B_TRUE).)
422  *
423  * The keys in the nvlist are snapshot names.
424  * The snapshots must all be in the same pool.
425  * The value is the name of the hold (string type).
426  *
427  * If cleanup_fd is not -1, it must be the result of open("/dev/zfs", O_EXCL).
428  * In this case, when the cleanup_fd is closed (including on process
429  * termination), the holds will be released.  If the system is shut down
430  * uncleanly, the holds will be released when the pool is next opened
431  * or imported.
432  *
433  * Holds for snapshots which don't exist will be skipped and have an entry
434  * added to errlist, but will not cause an overall failure.
435  *
436  * The return value will be 0 if all holds, for snapshots that existed,
437  * were succesfully created.
438  *
439  * Otherwise the return value will be the errno of a (unspecified) hold that
440  * failed and no holds will be created.
441  *
442  * In all cases the errlist will have an entry for each hold that failed
443  * (name = snapshot), with its value being the error code (int32).
444  */
445 int
446 lzc_hold(nvlist_t *holds, int cleanup_fd, nvlist_t **errlist)
447 {
448         char pool[ZFS_MAX_DATASET_NAME_LEN];
449         nvlist_t *args;
450         nvpair_t *elem;
451         int error;
452
453         /* determine the pool name */
454         elem = nvlist_next_nvpair(holds, NULL);
455         if (elem == NULL)
456                 return (0);
457         (void) strlcpy(pool, nvpair_name(elem), sizeof (pool));
458         pool[strcspn(pool, "/@")] = '\0';
459
460         args = fnvlist_alloc();
461         fnvlist_add_nvlist(args, "holds", holds);
462         if (cleanup_fd != -1)
463                 fnvlist_add_int32(args, "cleanup_fd", cleanup_fd);
464
465         error = lzc_ioctl(ZFS_IOC_HOLD, pool, args, errlist);
466         nvlist_free(args);
467         return (error);
468 }
469
470 /*
471  * Release "user holds" on snapshots.  If the snapshot has been marked for
472  * deferred destroy (by lzc_destroy_snaps(defer=B_TRUE)), it does not have
473  * any clones, and all the user holds are removed, then the snapshot will be
474  * destroyed.
475  *
476  * The keys in the nvlist are snapshot names.
477  * The snapshots must all be in the same pool.
478  * The value is a nvlist whose keys are the holds to remove.
479  *
480  * Holds which failed to release because they didn't exist will have an entry
481  * added to errlist, but will not cause an overall failure.
482  *
483  * The return value will be 0 if the nvl holds was empty or all holds that
484  * existed, were successfully removed.
485  *
486  * Otherwise the return value will be the errno of a (unspecified) hold that
487  * failed to release and no holds will be released.
488  *
489  * In all cases the errlist will have an entry for each hold that failed to
490  * to release.
491  */
492 int
493 lzc_release(nvlist_t *holds, nvlist_t **errlist)
494 {
495         char pool[ZFS_MAX_DATASET_NAME_LEN];
496         nvpair_t *elem;
497
498         /* determine the pool name */
499         elem = nvlist_next_nvpair(holds, NULL);
500         if (elem == NULL)
501                 return (0);
502         (void) strlcpy(pool, nvpair_name(elem), sizeof (pool));
503         pool[strcspn(pool, "/@")] = '\0';
504
505         return (lzc_ioctl(ZFS_IOC_RELEASE, pool, holds, errlist));
506 }
507
508 /*
509  * Retrieve list of user holds on the specified snapshot.
510  *
511  * On success, *holdsp will be set to a nvlist which the caller must free.
512  * The keys are the names of the holds, and the value is the creation time
513  * of the hold (uint64) in seconds since the epoch.
514  */
515 int
516 lzc_get_holds(const char *snapname, nvlist_t **holdsp)
517 {
518         int error;
519         nvlist_t *innvl = fnvlist_alloc();
520         error = lzc_ioctl(ZFS_IOC_GET_HOLDS, snapname, innvl, holdsp);
521         fnvlist_free(innvl);
522         return (error);
523 }
524
525 /*
526  * Generate a zfs send stream for the specified snapshot and write it to
527  * the specified file descriptor.
528  *
529  * "snapname" is the full name of the snapshot to send (e.g. "pool/fs@snap")
530  *
531  * If "from" is NULL, a full (non-incremental) stream will be sent.
532  * If "from" is non-NULL, it must be the full name of a snapshot or
533  * bookmark to send an incremental from (e.g. "pool/fs@earlier_snap" or
534  * "pool/fs#earlier_bmark").  If non-NULL, the specified snapshot or
535  * bookmark must represent an earlier point in the history of "snapname").
536  * It can be an earlier snapshot in the same filesystem or zvol as "snapname",
537  * or it can be the origin of "snapname"'s filesystem, or an earlier
538  * snapshot in the origin, etc.
539  *
540  * "fd" is the file descriptor to write the send stream to.
541  *
542  * If "flags" contains LZC_SEND_FLAG_LARGE_BLOCK, the stream is permitted
543  * to contain DRR_WRITE records with drr_length > 128K, and DRR_OBJECT
544  * records with drr_blksz > 128K.
545  *
546  * If "flags" contains LZC_SEND_FLAG_EMBED_DATA, the stream is permitted
547  * to contain DRR_WRITE_EMBEDDED records with drr_etype==BP_EMBEDDED_TYPE_DATA,
548  * which the receiving system must support (as indicated by support
549  * for the "embedded_data" feature).
550  */
551 int
552 lzc_send(const char *snapname, const char *from, int fd,
553     enum lzc_send_flags flags)
554 {
555         return (lzc_send_resume(snapname, from, fd, flags, 0, 0));
556 }
557
558 int
559 lzc_send_resume(const char *snapname, const char *from, int fd,
560     enum lzc_send_flags flags, uint64_t resumeobj, uint64_t resumeoff)
561 {
562         nvlist_t *args;
563         int err;
564
565         args = fnvlist_alloc();
566         fnvlist_add_int32(args, "fd", fd);
567         if (from != NULL)
568                 fnvlist_add_string(args, "fromsnap", from);
569         if (flags & LZC_SEND_FLAG_LARGE_BLOCK)
570                 fnvlist_add_boolean(args, "largeblockok");
571         if (flags & LZC_SEND_FLAG_EMBED_DATA)
572                 fnvlist_add_boolean(args, "embedok");
573         if (flags & LZC_SEND_FLAG_COMPRESS)
574                 fnvlist_add_boolean(args, "compressok");
575         if (resumeobj != 0 || resumeoff != 0) {
576                 fnvlist_add_uint64(args, "resume_object", resumeobj);
577                 fnvlist_add_uint64(args, "resume_offset", resumeoff);
578         }
579         err = lzc_ioctl(ZFS_IOC_SEND_NEW, snapname, args, NULL);
580         nvlist_free(args);
581         return (err);
582 }
583
584 /*
585  * "from" can be NULL, a snapshot, or a bookmark.
586  *
587  * If from is NULL, a full (non-incremental) stream will be estimated.  This
588  * is calculated very efficiently.
589  *
590  * If from is a snapshot, lzc_send_space uses the deadlists attached to
591  * each snapshot to efficiently estimate the stream size.
592  *
593  * If from is a bookmark, the indirect blocks in the destination snapshot
594  * are traversed, looking for blocks with a birth time since the creation TXG of
595  * the snapshot this bookmark was created from.  This will result in
596  * significantly more I/O and be less efficient than a send space estimation on
597  * an equivalent snapshot.
598  */
599 int
600 lzc_send_space(const char *snapname, const char *from,
601     enum lzc_send_flags flags, uint64_t *spacep)
602 {
603         nvlist_t *args;
604         nvlist_t *result;
605         int err;
606
607         args = fnvlist_alloc();
608         if (from != NULL)
609                 fnvlist_add_string(args, "from", from);
610         if (flags & LZC_SEND_FLAG_LARGE_BLOCK)
611                 fnvlist_add_boolean(args, "largeblockok");
612         if (flags & LZC_SEND_FLAG_EMBED_DATA)
613                 fnvlist_add_boolean(args, "embedok");
614         if (flags & LZC_SEND_FLAG_COMPRESS)
615                 fnvlist_add_boolean(args, "compressok");
616         err = lzc_ioctl(ZFS_IOC_SEND_SPACE, snapname, args, &result);
617         nvlist_free(args);
618         if (err == 0)
619                 *spacep = fnvlist_lookup_uint64(result, "space");
620         nvlist_free(result);
621         return (err);
622 }
623
624 static int
625 recv_read(int fd, void *buf, int ilen)
626 {
627         char *cp = buf;
628         int rv;
629         int len = ilen;
630
631         do {
632                 rv = read(fd, cp, len);
633                 cp += rv;
634                 len -= rv;
635         } while (rv > 0);
636
637         if (rv < 0 || len != 0)
638                 return (EIO);
639
640         return (0);
641 }
642
643 static int
644 recv_impl(const char *snapname, nvlist_t *props, const char *origin,
645     boolean_t force, boolean_t resumable, int fd,
646     const dmu_replay_record_t *begin_record)
647 {
648         /*
649          * The receive ioctl is still legacy, so we need to construct our own
650          * zfs_cmd_t rather than using zfsc_ioctl().
651          */
652         zfs_cmd_t zc = { 0 };
653         char *atp;
654         char *packed = NULL;
655         size_t size;
656         int error;
657
658         ASSERT3S(g_refcount, >, 0);
659
660         /* zc_name is name of containing filesystem */
661         (void) strlcpy(zc.zc_name, snapname, sizeof (zc.zc_name));
662         atp = strchr(zc.zc_name, '@');
663         if (atp == NULL)
664                 return (EINVAL);
665         *atp = '\0';
666
667         /* if the fs does not exist, try its parent. */
668         if (!lzc_exists(zc.zc_name)) {
669                 char *slashp = strrchr(zc.zc_name, '/');
670                 if (slashp == NULL)
671                         return (ENOENT);
672                 *slashp = '\0';
673
674         }
675
676         /* zc_value is full name of the snapshot to create */
677         (void) strlcpy(zc.zc_value, snapname, sizeof (zc.zc_value));
678
679         if (props != NULL) {
680                 /* zc_nvlist_src is props to set */
681                 packed = fnvlist_pack(props, &size);
682                 zc.zc_nvlist_src = (uint64_t)(uintptr_t)packed;
683                 zc.zc_nvlist_src_size = size;
684         }
685
686         /* zc_string is name of clone origin (if DRR_FLAG_CLONE) */
687         if (origin != NULL)
688                 (void) strlcpy(zc.zc_string, origin, sizeof (zc.zc_string));
689
690         /* zc_begin_record is non-byteswapped BEGIN record */
691         if (begin_record == NULL) {
692                 error = recv_read(fd, &zc.zc_begin_record,
693                     sizeof (zc.zc_begin_record));
694                 if (error != 0)
695                         goto out;
696         } else {
697                 zc.zc_begin_record = *begin_record;
698         }
699
700         /* zc_cookie is fd to read from */
701         zc.zc_cookie = fd;
702
703         /* zc guid is force flag */
704         zc.zc_guid = force;
705
706         zc.zc_resumable = resumable;
707
708         /* zc_cleanup_fd is unused */
709         zc.zc_cleanup_fd = -1;
710
711         error = ioctl(g_fd, ZFS_IOC_RECV, &zc);
712         if (error != 0)
713                 error = errno;
714
715 out:
716         if (packed != NULL)
717                 fnvlist_pack_free(packed, size);
718         free((void*)(uintptr_t)zc.zc_nvlist_dst);
719         return (error);
720 }
721
722 /*
723  * The simplest receive case: receive from the specified fd, creating the
724  * specified snapshot.  Apply the specified properties as "received" properties
725  * (which can be overridden by locally-set properties).  If the stream is a
726  * clone, its origin snapshot must be specified by 'origin'.  The 'force'
727  * flag will cause the target filesystem to be rolled back or destroyed if
728  * necessary to receive.
729  *
730  * Return 0 on success or an errno on failure.
731  *
732  * Note: this interface does not work on dedup'd streams
733  * (those with DMU_BACKUP_FEATURE_DEDUP).
734  */
735 int
736 lzc_receive(const char *snapname, nvlist_t *props, const char *origin,
737     boolean_t force, int fd)
738 {
739         return (recv_impl(snapname, props, origin, force, B_FALSE, fd, NULL));
740 }
741
742 /*
743  * Like lzc_receive, but if the receive fails due to premature stream
744  * termination, the intermediate state will be preserved on disk.  In this
745  * case, ECKSUM will be returned.  The receive may subsequently be resumed
746  * with a resuming send stream generated by lzc_send_resume().
747  */
748 int
749 lzc_receive_resumable(const char *snapname, nvlist_t *props, const char *origin,
750     boolean_t force, int fd)
751 {
752         return (recv_impl(snapname, props, origin, force, B_TRUE, fd, NULL));
753 }
754
755 /*
756  * Like lzc_receive, but allows the caller to read the begin record and then to
757  * pass it in.  That could be useful if the caller wants to derive, for example,
758  * the snapname or the origin parameters based on the information contained in
759  * the begin record.
760  * The begin record must be in its original form as read from the stream,
761  * in other words, it should not be byteswapped.
762  *
763  * The 'resumable' parameter allows to obtain the same behavior as with
764  * lzc_receive_resumable.
765  */
766 int
767 lzc_receive_with_header(const char *snapname, nvlist_t *props,
768     const char *origin, boolean_t force, boolean_t resumable, int fd,
769     const dmu_replay_record_t *begin_record)
770 {
771         if (begin_record == NULL)
772                 return (EINVAL);
773         return (recv_impl(snapname, props, origin, force, resumable, fd,
774             begin_record));
775 }
776
777 /*
778  * Roll back this filesystem or volume to its most recent snapshot.
779  * If snapnamebuf is not NULL, it will be filled in with the name
780  * of the most recent snapshot.
781  * Note that the latest snapshot may change if a new one is concurrently
782  * created or the current one is destroyed.  lzc_rollback_to can be used
783  * to roll back to a specific latest snapshot.
784  *
785  * Return 0 on success or an errno on failure.
786  */
787 int
788 lzc_rollback(const char *fsname, char *snapnamebuf, int snapnamelen)
789 {
790         nvlist_t *args;
791         nvlist_t *result;
792         int err;
793
794         args = fnvlist_alloc();
795         err = lzc_ioctl(ZFS_IOC_ROLLBACK, fsname, args, &result);
796         nvlist_free(args);
797         if (err == 0 && snapnamebuf != NULL) {
798                 const char *snapname = fnvlist_lookup_string(result, "target");
799                 (void) strlcpy(snapnamebuf, snapname, snapnamelen);
800         }
801         nvlist_free(result);
802
803         return (err);
804 }
805
806 /*
807  * Roll back this filesystem or volume to the specified snapshot,
808  * if possible.
809  *
810  * Return 0 on success or an errno on failure.
811  */
812 int
813 lzc_rollback_to(const char *fsname, const char *snapname)
814 {
815         nvlist_t *args;
816         nvlist_t *result;
817         int err;
818
819         args = fnvlist_alloc();
820         fnvlist_add_string(args, "target", snapname);
821         err = lzc_ioctl(ZFS_IOC_ROLLBACK, fsname, args, &result);
822         nvlist_free(args);
823         nvlist_free(result);
824         return (err);
825 }
826
827 /*
828  * Creates bookmarks.
829  *
830  * The bookmarks nvlist maps from name of the bookmark (e.g. "pool/fs#bmark") to
831  * the name of the snapshot (e.g. "pool/fs@snap").  All the bookmarks and
832  * snapshots must be in the same pool.
833  *
834  * The returned results nvlist will have an entry for each bookmark that failed.
835  * The value will be the (int32) error code.
836  *
837  * The return value will be 0 if all bookmarks were created, otherwise it will
838  * be the errno of a (undetermined) bookmarks that failed.
839  */
840 int
841 lzc_bookmark(nvlist_t *bookmarks, nvlist_t **errlist)
842 {
843         nvpair_t *elem;
844         int error;
845         char pool[ZFS_MAX_DATASET_NAME_LEN];
846
847         /* determine the pool name */
848         elem = nvlist_next_nvpair(bookmarks, NULL);
849         if (elem == NULL)
850                 return (0);
851         (void) strlcpy(pool, nvpair_name(elem), sizeof (pool));
852         pool[strcspn(pool, "/#")] = '\0';
853
854         error = lzc_ioctl(ZFS_IOC_BOOKMARK, pool, bookmarks, errlist);
855
856         return (error);
857 }
858
859 /*
860  * Retrieve bookmarks.
861  *
862  * Retrieve the list of bookmarks for the given file system. The props
863  * parameter is an nvlist of property names (with no values) that will be
864  * returned for each bookmark.
865  *
866  * The following are valid properties on bookmarks, all of which are numbers
867  * (represented as uint64 in the nvlist)
868  *
869  * "guid" - globally unique identifier of the snapshot it refers to
870  * "createtxg" - txg when the snapshot it refers to was created
871  * "creation" - timestamp when the snapshot it refers to was created
872  *
873  * The format of the returned nvlist as follows:
874  * <short name of bookmark> -> {
875  *     <name of property> -> {
876  *         "value" -> uint64
877  *     }
878  *  }
879  */
880 int
881 lzc_get_bookmarks(const char *fsname, nvlist_t *props, nvlist_t **bmarks)
882 {
883         return (lzc_ioctl(ZFS_IOC_GET_BOOKMARKS, fsname, props, bmarks));
884 }
885
886 /*
887  * Destroys bookmarks.
888  *
889  * The keys in the bmarks nvlist are the bookmarks to be destroyed.
890  * They must all be in the same pool.  Bookmarks are specified as
891  * <fs>#<bmark>.
892  *
893  * Bookmarks that do not exist will be silently ignored.
894  *
895  * The return value will be 0 if all bookmarks that existed were destroyed.
896  *
897  * Otherwise the return value will be the errno of a (undetermined) bookmark
898  * that failed, no bookmarks will be destroyed, and the errlist will have an
899  * entry for each bookmarks that failed.  The value in the errlist will be
900  * the (int32) error code.
901  */
902 int
903 lzc_destroy_bookmarks(nvlist_t *bmarks, nvlist_t **errlist)
904 {
905         nvpair_t *elem;
906         int error;
907         char pool[ZFS_MAX_DATASET_NAME_LEN];
908
909         /* determine the pool name */
910         elem = nvlist_next_nvpair(bmarks, NULL);
911         if (elem == NULL)
912                 return (0);
913         (void) strlcpy(pool, nvpair_name(elem), sizeof (pool));
914         pool[strcspn(pool, "/#")] = '\0';
915
916         error = lzc_ioctl(ZFS_IOC_DESTROY_BOOKMARKS, pool, bmarks, errlist);
917
918         return (error);
919 }
920
921 static int
922 lzc_channel_program_impl(const char *pool, const char *program, boolean_t sync,
923     uint64_t instrlimit, uint64_t memlimit, nvlist_t *argnvl, nvlist_t **outnvl)
924 {
925         int error;
926         nvlist_t *args;
927
928         args = fnvlist_alloc();
929         fnvlist_add_string(args, ZCP_ARG_PROGRAM, program);
930         fnvlist_add_nvlist(args, ZCP_ARG_ARGLIST, argnvl);
931         fnvlist_add_boolean_value(args, ZCP_ARG_SYNC, sync);
932         fnvlist_add_uint64(args, ZCP_ARG_INSTRLIMIT, instrlimit);
933         fnvlist_add_uint64(args, ZCP_ARG_MEMLIMIT, memlimit);
934         error = lzc_ioctl(ZFS_IOC_CHANNEL_PROGRAM, pool, args, outnvl);
935         fnvlist_free(args);
936
937         return (error);
938 }
939
940 /*
941  * Executes a channel program.
942  *
943  * If this function returns 0 the channel program was successfully loaded and
944  * ran without failing. Note that individual commands the channel program ran
945  * may have failed and the channel program is responsible for reporting such
946  * errors through outnvl if they are important.
947  *
948  * This method may also return:
949  *
950  * EINVAL   The program contains syntax errors, or an invalid memory or time
951  *          limit was given. No part of the channel program was executed.
952  *          If caused by syntax errors, 'outnvl' contains information about the
953  *          errors.
954  *
955  * EDOM     The program was executed, but encountered a runtime error, such as
956  *          calling a function with incorrect arguments, invoking the error()
957  *          function directly, failing an assert() command, etc. Some portion
958  *          of the channel program may have executed and committed changes.
959  *          Information about the failure can be found in 'outnvl'.
960  *
961  * ENOMEM   The program fully executed, but the output buffer was not large
962  *          enough to store the returned value. No output is returned through
963  *          'outnvl'.
964  *
965  * ENOSPC   The program was terminated because it exceeded its memory usage
966  *          limit. Some portion of the channel program may have executed and
967  *          committed changes to disk. No output is returned through 'outnvl'.
968  *
969  * ETIMEDOUT The program was terminated because it exceeded its Lua instruction
970  *           limit. Some portion of the channel program may have executed and
971  *           committed changes to disk. No output is returned through 'outnvl'.
972  */
973 int
974 lzc_channel_program(const char *pool, const char *program, uint64_t instrlimit,
975     uint64_t memlimit, nvlist_t *argnvl, nvlist_t **outnvl)
976 {
977         return (lzc_channel_program_impl(pool, program, B_TRUE, instrlimit,
978             memlimit, argnvl, outnvl));
979 }
980
981 /*
982  * Executes a read-only channel program.
983  *
984  * A read-only channel program works programmatically the same way as a
985  * normal channel program executed with lzc_channel_program(). The only
986  * difference is it runs exclusively in open-context and therefore can
987  * return faster. The downside to that, is that the program cannot change
988  * on-disk state by calling functions from the zfs.sync submodule.
989  *
990  * The return values of this function (and their meaning) are exactly the
991  * same as the ones described in lzc_channel_program().
992  */
993 int
994 lzc_channel_program_nosync(const char *pool, const char *program,
995     uint64_t timeout, uint64_t memlimit, nvlist_t *argnvl, nvlist_t **outnvl)
996 {
997         return (lzc_channel_program_impl(pool, program, B_FALSE, timeout,
998             memlimit, argnvl, outnvl));
999 }