xref: /freebsd-11-stable/cddl/contrib/opensolaris/lib/libzfs_core/common/libzfs_core.c (revision 9679213c21e94271a92fed651c3d0a12e06e3e4d)
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 = -1;
98 static pthread_mutex_t g_lock = PTHREAD_MUTEX_INITIALIZER;
99 static int g_refcount;
100 
101 int
libzfs_core_init(void)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
libzfs_core_fini(void)119 libzfs_core_fini(void)
120 {
121 	(void) pthread_mutex_lock(&g_lock);
122 	ASSERT3S(g_refcount, >, 0);
123 
124 	if (g_refcount > 0)
125 		g_refcount--;
126 
127 	if (g_refcount == 0 && g_fd != -1) {
128 		(void) close(g_fd);
129 		g_fd = -1;
130 	}
131 	(void) pthread_mutex_unlock(&g_lock);
132 }
133 
134 static int
lzc_ioctl(zfs_ioc_t ioc,const char * name,nvlist_t * source,nvlist_t ** resultp)135 lzc_ioctl(zfs_ioc_t ioc, const char *name,
136     nvlist_t *source, nvlist_t **resultp)
137 {
138 	zfs_cmd_t zc = { 0 };
139 	int error = 0;
140 	char *packed;
141 #ifdef __FreeBSD__
142 	nvlist_t *oldsource;
143 #endif
144 	size_t size;
145 
146 	ASSERT3S(g_refcount, >, 0);
147 	VERIFY3S(g_fd, !=, -1);
148 
149 	(void) strlcpy(zc.zc_name, name, sizeof (zc.zc_name));
150 
151 #ifdef __FreeBSD__
152 	if (zfs_ioctl_version == ZFS_IOCVER_UNDEF)
153 		zfs_ioctl_version = get_zfs_ioctl_version();
154 
155 	if (zfs_ioctl_version < ZFS_IOCVER_LZC) {
156 		oldsource = source;
157 		error = lzc_compat_pre(&zc, &ioc, &source);
158 		if (error)
159 			return (error);
160 	}
161 #endif
162 
163 	packed = fnvlist_pack(source, &size);
164 	zc.zc_nvlist_src = (uint64_t)(uintptr_t)packed;
165 	zc.zc_nvlist_src_size = size;
166 
167 	if (resultp != NULL) {
168 		*resultp = NULL;
169 		if (ioc == ZFS_IOC_CHANNEL_PROGRAM) {
170 			zc.zc_nvlist_dst_size = fnvlist_lookup_uint64(source,
171 			    ZCP_ARG_MEMLIMIT);
172 		} else {
173 			zc.zc_nvlist_dst_size = MAX(size * 2, 128 * 1024);
174 		}
175 		zc.zc_nvlist_dst = (uint64_t)(uintptr_t)
176 		    malloc(zc.zc_nvlist_dst_size);
177 #ifdef illumos
178 		if (zc.zc_nvlist_dst == NULL) {
179 #else
180 		if (zc.zc_nvlist_dst == 0) {
181 #endif
182 			error = ENOMEM;
183 			goto out;
184 		}
185 	}
186 
187 	while (ioctl(g_fd, ioc, &zc) != 0) {
188 		/*
189 		 * If ioctl exited with ENOMEM, we retry the ioctl after
190 		 * increasing the size of the destination nvlist.
191 		 *
192 		 * Channel programs that exit with ENOMEM ran over the
193 		 * lua memory sandbox; they should not be retried.
194 		 */
195 		if (errno == ENOMEM && resultp != NULL &&
196 		    ioc != ZFS_IOC_CHANNEL_PROGRAM) {
197 			free((void *)(uintptr_t)zc.zc_nvlist_dst);
198 			zc.zc_nvlist_dst_size *= 2;
199 			zc.zc_nvlist_dst = (uint64_t)(uintptr_t)
200 			    malloc(zc.zc_nvlist_dst_size);
201 #ifdef illumos
202 			if (zc.zc_nvlist_dst == NULL) {
203 #else
204 			if (zc.zc_nvlist_dst == 0) {
205 #endif
206 				error = ENOMEM;
207 				goto out;
208 			}
209 		} else {
210 			error = errno;
211 			break;
212 		}
213 	}
214 
215 #ifdef __FreeBSD__
216 	if (zfs_ioctl_version < ZFS_IOCVER_LZC)
217 		lzc_compat_post(&zc, ioc);
218 #endif
219 	if (zc.zc_nvlist_dst_filled) {
220 		*resultp = fnvlist_unpack((void *)(uintptr_t)zc.zc_nvlist_dst,
221 		    zc.zc_nvlist_dst_size);
222 	}
223 #ifdef __FreeBSD__
224 	if (zfs_ioctl_version < ZFS_IOCVER_LZC)
225 		lzc_compat_outnvl(&zc, ioc, resultp);
226 #endif
227 out:
228 #ifdef __FreeBSD__
229 	if (zfs_ioctl_version < ZFS_IOCVER_LZC) {
230 		if (source != oldsource)
231 			nvlist_free(source);
232 		source = oldsource;
233 	}
234 #endif
235 	fnvlist_pack_free(packed, size);
236 	free((void *)(uintptr_t)zc.zc_nvlist_dst);
237 	return (error);
238 }
239 
240 int
241 lzc_create(const char *fsname, enum lzc_dataset_type type, nvlist_t *props)
242 {
243 	int error;
244 	nvlist_t *args = fnvlist_alloc();
245 	fnvlist_add_int32(args, "type", (dmu_objset_type_t)type);
246 	if (props != NULL)
247 		fnvlist_add_nvlist(args, "props", props);
248 	error = lzc_ioctl(ZFS_IOC_CREATE, fsname, args, NULL);
249 	nvlist_free(args);
250 	return (error);
251 }
252 
253 int
254 lzc_clone(const char *fsname, const char *origin,
255     nvlist_t *props)
256 {
257 	int error;
258 	nvlist_t *args = fnvlist_alloc();
259 	fnvlist_add_string(args, "origin", origin);
260 	if (props != NULL)
261 		fnvlist_add_nvlist(args, "props", props);
262 	error = lzc_ioctl(ZFS_IOC_CLONE, fsname, args, NULL);
263 	nvlist_free(args);
264 	return (error);
265 }
266 
267 int
268 lzc_promote(const char *fsname, char *snapnamebuf, int snapnamelen)
269 {
270 	/*
271 	 * The promote ioctl is still legacy, so we need to construct our
272 	 * own zfs_cmd_t rather than using lzc_ioctl().
273 	 */
274 	zfs_cmd_t zc = { 0 };
275 
276 	ASSERT3S(g_refcount, >, 0);
277 	VERIFY3S(g_fd, !=, -1);
278 
279 	(void) strlcpy(zc.zc_name, fsname, sizeof (zc.zc_name));
280 	if (ioctl(g_fd, ZFS_IOC_PROMOTE, &zc) != 0) {
281 		int error = errno;
282 		if (error == EEXIST && snapnamebuf != NULL)
283 			(void) strlcpy(snapnamebuf, zc.zc_string, snapnamelen);
284 		return (error);
285 	}
286 	return (0);
287 }
288 
289 int
290 lzc_remap(const char *fsname)
291 {
292 	int error;
293 	nvlist_t *args = fnvlist_alloc();
294 	error = lzc_ioctl(ZFS_IOC_REMAP, fsname, args, NULL);
295 	nvlist_free(args);
296 	return (error);
297 }
298 
299 int
300 lzc_rename(const char *source, const char *target)
301 {
302 	zfs_cmd_t zc = { 0 };
303 	int error;
304 
305 	ASSERT3S(g_refcount, >, 0);
306 	VERIFY3S(g_fd, !=, -1);
307 
308 	(void) strlcpy(zc.zc_name, source, sizeof (zc.zc_name));
309 	(void) strlcpy(zc.zc_value, target, sizeof (zc.zc_value));
310 	error = ioctl(g_fd, ZFS_IOC_RENAME, &zc);
311 	if (error != 0)
312 		error = errno;
313 	return (error);
314 }
315 
316 int
317 lzc_destroy(const char *fsname)
318 {
319 	int error;
320 
321 	nvlist_t *args = fnvlist_alloc();
322 	error = lzc_ioctl(ZFS_IOC_DESTROY, fsname, args, NULL);
323 	nvlist_free(args);
324 	return (error);
325 }
326 
327 /*
328  * Creates snapshots.
329  *
330  * The keys in the snaps nvlist are the snapshots to be created.
331  * They must all be in the same pool.
332  *
333  * The props nvlist is properties to set.  Currently only user properties
334  * are supported.  { user:prop_name -> string value }
335  *
336  * The returned results nvlist will have an entry for each snapshot that failed.
337  * The value will be the (int32) error code.
338  *
339  * The return value will be 0 if all snapshots were created, otherwise it will
340  * be the errno of a (unspecified) snapshot that failed.
341  */
342 int
343 lzc_snapshot(nvlist_t *snaps, nvlist_t *props, nvlist_t **errlist)
344 {
345 	nvpair_t *elem;
346 	nvlist_t *args;
347 	int error;
348 	char pool[ZFS_MAX_DATASET_NAME_LEN];
349 
350 	*errlist = NULL;
351 
352 	/* determine the pool name */
353 	elem = nvlist_next_nvpair(snaps, NULL);
354 	if (elem == NULL)
355 		return (0);
356 	(void) strlcpy(pool, nvpair_name(elem), sizeof (pool));
357 	pool[strcspn(pool, "/@")] = '\0';
358 
359 	args = fnvlist_alloc();
360 	fnvlist_add_nvlist(args, "snaps", snaps);
361 	if (props != NULL)
362 		fnvlist_add_nvlist(args, "props", props);
363 
364 	error = lzc_ioctl(ZFS_IOC_SNAPSHOT, pool, args, errlist);
365 	nvlist_free(args);
366 
367 	return (error);
368 }
369 
370 /*
371  * Destroys snapshots.
372  *
373  * The keys in the snaps nvlist are the snapshots to be destroyed.
374  * They must all be in the same pool.
375  *
376  * Snapshots that do not exist will be silently ignored.
377  *
378  * If 'defer' is not set, and a snapshot has user holds or clones, the
379  * destroy operation will fail and none of the snapshots will be
380  * destroyed.
381  *
382  * If 'defer' is set, and a snapshot has user holds or clones, it will be
383  * marked for deferred destruction, and will be destroyed when the last hold
384  * or clone is removed/destroyed.
385  *
386  * The return value will be 0 if all snapshots were destroyed (or marked for
387  * later destruction if 'defer' is set) or didn't exist to begin with.
388  *
389  * Otherwise the return value will be the errno of a (unspecified) snapshot
390  * that failed, no snapshots will be destroyed, and the errlist will have an
391  * entry for each snapshot that failed.  The value in the errlist will be
392  * the (int32) error code.
393  */
394 int
395 lzc_destroy_snaps(nvlist_t *snaps, boolean_t defer, nvlist_t **errlist)
396 {
397 	nvpair_t *elem;
398 	nvlist_t *args;
399 	int error;
400 	char pool[ZFS_MAX_DATASET_NAME_LEN];
401 
402 	/* determine the pool name */
403 	elem = nvlist_next_nvpair(snaps, NULL);
404 	if (elem == NULL)
405 		return (0);
406 	(void) strlcpy(pool, nvpair_name(elem), sizeof (pool));
407 	pool[strcspn(pool, "/@")] = '\0';
408 
409 	args = fnvlist_alloc();
410 	fnvlist_add_nvlist(args, "snaps", snaps);
411 	if (defer)
412 		fnvlist_add_boolean(args, "defer");
413 
414 	error = lzc_ioctl(ZFS_IOC_DESTROY_SNAPS, pool, args, errlist);
415 	nvlist_free(args);
416 
417 	return (error);
418 }
419 
420 int
421 lzc_snaprange_space(const char *firstsnap, const char *lastsnap,
422     uint64_t *usedp)
423 {
424 	nvlist_t *args;
425 	nvlist_t *result;
426 	int err;
427 	char fs[ZFS_MAX_DATASET_NAME_LEN];
428 	char *atp;
429 
430 	/* determine the fs name */
431 	(void) strlcpy(fs, firstsnap, sizeof (fs));
432 	atp = strchr(fs, '@');
433 	if (atp == NULL)
434 		return (EINVAL);
435 	*atp = '\0';
436 
437 	args = fnvlist_alloc();
438 	fnvlist_add_string(args, "firstsnap", firstsnap);
439 
440 	err = lzc_ioctl(ZFS_IOC_SPACE_SNAPS, lastsnap, args, &result);
441 	nvlist_free(args);
442 	if (err == 0)
443 		*usedp = fnvlist_lookup_uint64(result, "used");
444 	fnvlist_free(result);
445 
446 	return (err);
447 }
448 
449 boolean_t
450 lzc_exists(const char *dataset)
451 {
452 	/*
453 	 * The objset_stats ioctl is still legacy, so we need to construct our
454 	 * own zfs_cmd_t rather than using lzc_ioctl().
455 	 */
456 	zfs_cmd_t zc = { 0 };
457 
458 	ASSERT3S(g_refcount, >, 0);
459 	VERIFY3S(g_fd, !=, -1);
460 
461 	(void) strlcpy(zc.zc_name, dataset, sizeof (zc.zc_name));
462 	return (ioctl(g_fd, ZFS_IOC_OBJSET_STATS, &zc) == 0);
463 }
464 
465 /*
466  * Create "user holds" on snapshots.  If there is a hold on a snapshot,
467  * the snapshot can not be destroyed.  (However, it can be marked for deletion
468  * by lzc_destroy_snaps(defer=B_TRUE).)
469  *
470  * The keys in the nvlist are snapshot names.
471  * The snapshots must all be in the same pool.
472  * The value is the name of the hold (string type).
473  *
474  * If cleanup_fd is not -1, it must be the result of open("/dev/zfs", O_EXCL).
475  * In this case, when the cleanup_fd is closed (including on process
476  * termination), the holds will be released.  If the system is shut down
477  * uncleanly, the holds will be released when the pool is next opened
478  * or imported.
479  *
480  * Holds for snapshots which don't exist will be skipped and have an entry
481  * added to errlist, but will not cause an overall failure.
482  *
483  * The return value will be 0 if all holds, for snapshots that existed,
484  * were succesfully created.
485  *
486  * Otherwise the return value will be the errno of a (unspecified) hold that
487  * failed and no holds will be created.
488  *
489  * In all cases the errlist will have an entry for each hold that failed
490  * (name = snapshot), with its value being the error code (int32).
491  */
492 int
493 lzc_hold(nvlist_t *holds, int cleanup_fd, nvlist_t **errlist)
494 {
495 	char pool[ZFS_MAX_DATASET_NAME_LEN];
496 	nvlist_t *args;
497 	nvpair_t *elem;
498 	int error;
499 
500 	/* determine the pool name */
501 	elem = nvlist_next_nvpair(holds, NULL);
502 	if (elem == NULL)
503 		return (0);
504 	(void) strlcpy(pool, nvpair_name(elem), sizeof (pool));
505 	pool[strcspn(pool, "/@")] = '\0';
506 
507 	args = fnvlist_alloc();
508 	fnvlist_add_nvlist(args, "holds", holds);
509 	if (cleanup_fd != -1)
510 		fnvlist_add_int32(args, "cleanup_fd", cleanup_fd);
511 
512 	error = lzc_ioctl(ZFS_IOC_HOLD, pool, args, errlist);
513 	nvlist_free(args);
514 	return (error);
515 }
516 
517 /*
518  * Release "user holds" on snapshots.  If the snapshot has been marked for
519  * deferred destroy (by lzc_destroy_snaps(defer=B_TRUE)), it does not have
520  * any clones, and all the user holds are removed, then the snapshot will be
521  * destroyed.
522  *
523  * The keys in the nvlist are snapshot names.
524  * The snapshots must all be in the same pool.
525  * The value is a nvlist whose keys are the holds to remove.
526  *
527  * Holds which failed to release because they didn't exist will have an entry
528  * added to errlist, but will not cause an overall failure.
529  *
530  * The return value will be 0 if the nvl holds was empty or all holds that
531  * existed, were successfully removed.
532  *
533  * Otherwise the return value will be the errno of a (unspecified) hold that
534  * failed to release and no holds will be released.
535  *
536  * In all cases the errlist will have an entry for each hold that failed to
537  * to release.
538  */
539 int
540 lzc_release(nvlist_t *holds, nvlist_t **errlist)
541 {
542 	char pool[ZFS_MAX_DATASET_NAME_LEN];
543 	nvpair_t *elem;
544 
545 	/* determine the pool name */
546 	elem = nvlist_next_nvpair(holds, NULL);
547 	if (elem == NULL)
548 		return (0);
549 	(void) strlcpy(pool, nvpair_name(elem), sizeof (pool));
550 	pool[strcspn(pool, "/@")] = '\0';
551 
552 	return (lzc_ioctl(ZFS_IOC_RELEASE, pool, holds, errlist));
553 }
554 
555 /*
556  * Retrieve list of user holds on the specified snapshot.
557  *
558  * On success, *holdsp will be set to a nvlist which the caller must free.
559  * The keys are the names of the holds, and the value is the creation time
560  * of the hold (uint64) in seconds since the epoch.
561  */
562 int
563 lzc_get_holds(const char *snapname, nvlist_t **holdsp)
564 {
565 	int error;
566 	nvlist_t *innvl = fnvlist_alloc();
567 	error = lzc_ioctl(ZFS_IOC_GET_HOLDS, snapname, innvl, holdsp);
568 	fnvlist_free(innvl);
569 	return (error);
570 }
571 
572 /*
573  * Generate a zfs send stream for the specified snapshot and write it to
574  * the specified file descriptor.
575  *
576  * "snapname" is the full name of the snapshot to send (e.g. "pool/fs@snap")
577  *
578  * If "from" is NULL, a full (non-incremental) stream will be sent.
579  * If "from" is non-NULL, it must be the full name of a snapshot or
580  * bookmark to send an incremental from (e.g. "pool/fs@earlier_snap" or
581  * "pool/fs#earlier_bmark").  If non-NULL, the specified snapshot or
582  * bookmark must represent an earlier point in the history of "snapname").
583  * It can be an earlier snapshot in the same filesystem or zvol as "snapname",
584  * or it can be the origin of "snapname"'s filesystem, or an earlier
585  * snapshot in the origin, etc.
586  *
587  * "fd" is the file descriptor to write the send stream to.
588  *
589  * If "flags" contains LZC_SEND_FLAG_LARGE_BLOCK, the stream is permitted
590  * to contain DRR_WRITE records with drr_length > 128K, and DRR_OBJECT
591  * records with drr_blksz > 128K.
592  *
593  * If "flags" contains LZC_SEND_FLAG_EMBED_DATA, the stream is permitted
594  * to contain DRR_WRITE_EMBEDDED records with drr_etype==BP_EMBEDDED_TYPE_DATA,
595  * which the receiving system must support (as indicated by support
596  * for the "embedded_data" feature).
597  */
598 int
599 lzc_send(const char *snapname, const char *from, int fd,
600     enum lzc_send_flags flags)
601 {
602 	return (lzc_send_resume(snapname, from, fd, flags, 0, 0));
603 }
604 
605 int
606 lzc_send_resume(const char *snapname, const char *from, int fd,
607     enum lzc_send_flags flags, uint64_t resumeobj, uint64_t resumeoff)
608 {
609 	nvlist_t *args;
610 	int err;
611 
612 	args = fnvlist_alloc();
613 	fnvlist_add_int32(args, "fd", fd);
614 	if (from != NULL)
615 		fnvlist_add_string(args, "fromsnap", from);
616 	if (flags & LZC_SEND_FLAG_LARGE_BLOCK)
617 		fnvlist_add_boolean(args, "largeblockok");
618 	if (flags & LZC_SEND_FLAG_EMBED_DATA)
619 		fnvlist_add_boolean(args, "embedok");
620 	if (flags & LZC_SEND_FLAG_COMPRESS)
621 		fnvlist_add_boolean(args, "compressok");
622 	if (resumeobj != 0 || resumeoff != 0) {
623 		fnvlist_add_uint64(args, "resume_object", resumeobj);
624 		fnvlist_add_uint64(args, "resume_offset", resumeoff);
625 	}
626 	err = lzc_ioctl(ZFS_IOC_SEND_NEW, snapname, args, NULL);
627 	nvlist_free(args);
628 	return (err);
629 }
630 
631 /*
632  * "from" can be NULL, a snapshot, or a bookmark.
633  *
634  * If from is NULL, a full (non-incremental) stream will be estimated.  This
635  * is calculated very efficiently.
636  *
637  * If from is a snapshot, lzc_send_space uses the deadlists attached to
638  * each snapshot to efficiently estimate the stream size.
639  *
640  * If from is a bookmark, the indirect blocks in the destination snapshot
641  * are traversed, looking for blocks with a birth time since the creation TXG of
642  * the snapshot this bookmark was created from.  This will result in
643  * significantly more I/O and be less efficient than a send space estimation on
644  * an equivalent snapshot.
645  */
646 int
647 lzc_send_space(const char *snapname, const char *from,
648     enum lzc_send_flags flags, uint64_t *spacep)
649 {
650 	nvlist_t *args;
651 	nvlist_t *result;
652 	int err;
653 
654 	args = fnvlist_alloc();
655 	if (from != NULL)
656 		fnvlist_add_string(args, "from", from);
657 	if (flags & LZC_SEND_FLAG_LARGE_BLOCK)
658 		fnvlist_add_boolean(args, "largeblockok");
659 	if (flags & LZC_SEND_FLAG_EMBED_DATA)
660 		fnvlist_add_boolean(args, "embedok");
661 	if (flags & LZC_SEND_FLAG_COMPRESS)
662 		fnvlist_add_boolean(args, "compressok");
663 	err = lzc_ioctl(ZFS_IOC_SEND_SPACE, snapname, args, &result);
664 	nvlist_free(args);
665 	if (err == 0)
666 		*spacep = fnvlist_lookup_uint64(result, "space");
667 	nvlist_free(result);
668 	return (err);
669 }
670 
671 static int
672 recv_read(int fd, void *buf, int ilen)
673 {
674 	char *cp = buf;
675 	int rv;
676 	int len = ilen;
677 
678 	do {
679 		rv = read(fd, cp, len);
680 		cp += rv;
681 		len -= rv;
682 	} while (rv > 0);
683 
684 	if (rv < 0 || len != 0)
685 		return (EIO);
686 
687 	return (0);
688 }
689 
690 static int
691 recv_impl(const char *snapname, nvlist_t *props, const char *origin,
692     boolean_t force, boolean_t resumable, int fd,
693     const dmu_replay_record_t *begin_record)
694 {
695 	/*
696 	 * The receive ioctl is still legacy, so we need to construct our own
697 	 * zfs_cmd_t rather than using zfsc_ioctl().
698 	 */
699 	zfs_cmd_t zc = { 0 };
700 	char *atp;
701 	char *packed = NULL;
702 	size_t size;
703 	int error;
704 
705 	ASSERT3S(g_refcount, >, 0);
706 	VERIFY3S(g_fd, !=, -1);
707 
708 	/* zc_name is name of containing filesystem */
709 	(void) strlcpy(zc.zc_name, snapname, sizeof (zc.zc_name));
710 	atp = strchr(zc.zc_name, '@');
711 	if (atp == NULL)
712 		return (EINVAL);
713 	*atp = '\0';
714 
715 	/* if the fs does not exist, try its parent. */
716 	if (!lzc_exists(zc.zc_name)) {
717 		char *slashp = strrchr(zc.zc_name, '/');
718 		if (slashp == NULL)
719 			return (ENOENT);
720 		*slashp = '\0';
721 
722 	}
723 
724 	/* zc_value is full name of the snapshot to create */
725 	(void) strlcpy(zc.zc_value, snapname, sizeof (zc.zc_value));
726 
727 	if (props != NULL) {
728 		/* zc_nvlist_src is props to set */
729 		packed = fnvlist_pack(props, &size);
730 		zc.zc_nvlist_src = (uint64_t)(uintptr_t)packed;
731 		zc.zc_nvlist_src_size = size;
732 	}
733 
734 	/* zc_string is name of clone origin (if DRR_FLAG_CLONE) */
735 	if (origin != NULL)
736 		(void) strlcpy(zc.zc_string, origin, sizeof (zc.zc_string));
737 
738 	/* zc_begin_record is non-byteswapped BEGIN record */
739 	if (begin_record == NULL) {
740 		error = recv_read(fd, &zc.zc_begin_record,
741 		    sizeof (zc.zc_begin_record));
742 		if (error != 0)
743 			goto out;
744 	} else {
745 		zc.zc_begin_record = *begin_record;
746 	}
747 
748 	/* zc_cookie is fd to read from */
749 	zc.zc_cookie = fd;
750 
751 	/* zc guid is force flag */
752 	zc.zc_guid = force;
753 
754 	zc.zc_resumable = resumable;
755 
756 	/* zc_cleanup_fd is unused */
757 	zc.zc_cleanup_fd = -1;
758 
759 	error = ioctl(g_fd, ZFS_IOC_RECV, &zc);
760 	if (error != 0)
761 		error = errno;
762 
763 out:
764 	if (packed != NULL)
765 		fnvlist_pack_free(packed, size);
766 	free((void*)(uintptr_t)zc.zc_nvlist_dst);
767 	return (error);
768 }
769 
770 /*
771  * The simplest receive case: receive from the specified fd, creating the
772  * specified snapshot.  Apply the specified properties as "received" properties
773  * (which can be overridden by locally-set properties).  If the stream is a
774  * clone, its origin snapshot must be specified by 'origin'.  The 'force'
775  * flag will cause the target filesystem to be rolled back or destroyed if
776  * necessary to receive.
777  *
778  * Return 0 on success or an errno on failure.
779  *
780  * Note: this interface does not work on dedup'd streams
781  * (those with DMU_BACKUP_FEATURE_DEDUP).
782  */
783 int
784 lzc_receive(const char *snapname, nvlist_t *props, const char *origin,
785     boolean_t force, int fd)
786 {
787 	return (recv_impl(snapname, props, origin, force, B_FALSE, fd, NULL));
788 }
789 
790 /*
791  * Like lzc_receive, but if the receive fails due to premature stream
792  * termination, the intermediate state will be preserved on disk.  In this
793  * case, ECKSUM will be returned.  The receive may subsequently be resumed
794  * with a resuming send stream generated by lzc_send_resume().
795  */
796 int
797 lzc_receive_resumable(const char *snapname, nvlist_t *props, const char *origin,
798     boolean_t force, int fd)
799 {
800 	return (recv_impl(snapname, props, origin, force, B_TRUE, fd, NULL));
801 }
802 
803 /*
804  * Like lzc_receive, but allows the caller to read the begin record and then to
805  * pass it in.  That could be useful if the caller wants to derive, for example,
806  * the snapname or the origin parameters based on the information contained in
807  * the begin record.
808  * The begin record must be in its original form as read from the stream,
809  * in other words, it should not be byteswapped.
810  *
811  * The 'resumable' parameter allows to obtain the same behavior as with
812  * lzc_receive_resumable.
813  */
814 int
815 lzc_receive_with_header(const char *snapname, nvlist_t *props,
816     const char *origin, boolean_t force, boolean_t resumable, int fd,
817     const dmu_replay_record_t *begin_record)
818 {
819 	if (begin_record == NULL)
820 		return (EINVAL);
821 	return (recv_impl(snapname, props, origin, force, resumable, fd,
822 	    begin_record));
823 }
824 
825 /*
826  * Roll back this filesystem or volume to its most recent snapshot.
827  * If snapnamebuf is not NULL, it will be filled in with the name
828  * of the most recent snapshot.
829  * Note that the latest snapshot may change if a new one is concurrently
830  * created or the current one is destroyed.  lzc_rollback_to can be used
831  * to roll back to a specific latest snapshot.
832  *
833  * Return 0 on success or an errno on failure.
834  */
835 int
836 lzc_rollback(const char *fsname, char *snapnamebuf, int snapnamelen)
837 {
838 	nvlist_t *args;
839 	nvlist_t *result;
840 	int err;
841 
842 	args = fnvlist_alloc();
843 	err = lzc_ioctl(ZFS_IOC_ROLLBACK, fsname, args, &result);
844 	nvlist_free(args);
845 	if (err == 0 && snapnamebuf != NULL) {
846 		const char *snapname = fnvlist_lookup_string(result, "target");
847 		(void) strlcpy(snapnamebuf, snapname, snapnamelen);
848 	}
849 	nvlist_free(result);
850 
851 	return (err);
852 }
853 
854 /*
855  * Roll back this filesystem or volume to the specified snapshot,
856  * if possible.
857  *
858  * Return 0 on success or an errno on failure.
859  */
860 int
861 lzc_rollback_to(const char *fsname, const char *snapname)
862 {
863 	nvlist_t *args;
864 	nvlist_t *result;
865 	int err;
866 
867 	args = fnvlist_alloc();
868 	fnvlist_add_string(args, "target", snapname);
869 	err = lzc_ioctl(ZFS_IOC_ROLLBACK, fsname, args, &result);
870 	nvlist_free(args);
871 	nvlist_free(result);
872 	return (err);
873 }
874 
875 /*
876  * Creates bookmarks.
877  *
878  * The bookmarks nvlist maps from name of the bookmark (e.g. "pool/fs#bmark") to
879  * the name of the snapshot (e.g. "pool/fs@snap").  All the bookmarks and
880  * snapshots must be in the same pool.
881  *
882  * The returned results nvlist will have an entry for each bookmark that failed.
883  * The value will be the (int32) error code.
884  *
885  * The return value will be 0 if all bookmarks were created, otherwise it will
886  * be the errno of a (undetermined) bookmarks that failed.
887  */
888 int
889 lzc_bookmark(nvlist_t *bookmarks, nvlist_t **errlist)
890 {
891 	nvpair_t *elem;
892 	int error;
893 	char pool[ZFS_MAX_DATASET_NAME_LEN];
894 
895 	/* determine the pool name */
896 	elem = nvlist_next_nvpair(bookmarks, NULL);
897 	if (elem == NULL)
898 		return (0);
899 	(void) strlcpy(pool, nvpair_name(elem), sizeof (pool));
900 	pool[strcspn(pool, "/#")] = '\0';
901 
902 	error = lzc_ioctl(ZFS_IOC_BOOKMARK, pool, bookmarks, errlist);
903 
904 	return (error);
905 }
906 
907 /*
908  * Retrieve bookmarks.
909  *
910  * Retrieve the list of bookmarks for the given file system. The props
911  * parameter is an nvlist of property names (with no values) that will be
912  * returned for each bookmark.
913  *
914  * The following are valid properties on bookmarks, all of which are numbers
915  * (represented as uint64 in the nvlist)
916  *
917  * "guid" - globally unique identifier of the snapshot it refers to
918  * "createtxg" - txg when the snapshot it refers to was created
919  * "creation" - timestamp when the snapshot it refers to was created
920  *
921  * The format of the returned nvlist as follows:
922  * <short name of bookmark> -> {
923  *     <name of property> -> {
924  *         "value" -> uint64
925  *     }
926  *  }
927  */
928 int
929 lzc_get_bookmarks(const char *fsname, nvlist_t *props, nvlist_t **bmarks)
930 {
931 	return (lzc_ioctl(ZFS_IOC_GET_BOOKMARKS, fsname, props, bmarks));
932 }
933 
934 /*
935  * Destroys bookmarks.
936  *
937  * The keys in the bmarks nvlist are the bookmarks to be destroyed.
938  * They must all be in the same pool.  Bookmarks are specified as
939  * <fs>#<bmark>.
940  *
941  * Bookmarks that do not exist will be silently ignored.
942  *
943  * The return value will be 0 if all bookmarks that existed were destroyed.
944  *
945  * Otherwise the return value will be the errno of a (undetermined) bookmark
946  * that failed, no bookmarks will be destroyed, and the errlist will have an
947  * entry for each bookmarks that failed.  The value in the errlist will be
948  * the (int32) error code.
949  */
950 int
951 lzc_destroy_bookmarks(nvlist_t *bmarks, nvlist_t **errlist)
952 {
953 	nvpair_t *elem;
954 	int error;
955 	char pool[ZFS_MAX_DATASET_NAME_LEN];
956 
957 	/* determine the pool name */
958 	elem = nvlist_next_nvpair(bmarks, NULL);
959 	if (elem == NULL)
960 		return (0);
961 	(void) strlcpy(pool, nvpair_name(elem), sizeof (pool));
962 	pool[strcspn(pool, "/#")] = '\0';
963 
964 	error = lzc_ioctl(ZFS_IOC_DESTROY_BOOKMARKS, pool, bmarks, errlist);
965 
966 	return (error);
967 }
968 
969 static int
970 lzc_channel_program_impl(const char *pool, const char *program, boolean_t sync,
971     uint64_t instrlimit, uint64_t memlimit, nvlist_t *argnvl, nvlist_t **outnvl)
972 {
973 	int error;
974 	nvlist_t *args;
975 
976 	args = fnvlist_alloc();
977 	fnvlist_add_string(args, ZCP_ARG_PROGRAM, program);
978 	fnvlist_add_nvlist(args, ZCP_ARG_ARGLIST, argnvl);
979 	fnvlist_add_boolean_value(args, ZCP_ARG_SYNC, sync);
980 	fnvlist_add_uint64(args, ZCP_ARG_INSTRLIMIT, instrlimit);
981 	fnvlist_add_uint64(args, ZCP_ARG_MEMLIMIT, memlimit);
982 	error = lzc_ioctl(ZFS_IOC_CHANNEL_PROGRAM, pool, args, outnvl);
983 	fnvlist_free(args);
984 
985 	return (error);
986 }
987 
988 /*
989  * Executes a channel program.
990  *
991  * If this function returns 0 the channel program was successfully loaded and
992  * ran without failing. Note that individual commands the channel program ran
993  * may have failed and the channel program is responsible for reporting such
994  * errors through outnvl if they are important.
995  *
996  * This method may also return:
997  *
998  * EINVAL   The program contains syntax errors, or an invalid memory or time
999  *          limit was given. No part of the channel program was executed.
1000  *          If caused by syntax errors, 'outnvl' contains information about the
1001  *          errors.
1002  *
1003  * EDOM     The program was executed, but encountered a runtime error, such as
1004  *          calling a function with incorrect arguments, invoking the error()
1005  *          function directly, failing an assert() command, etc. Some portion
1006  *          of the channel program may have executed and committed changes.
1007  *          Information about the failure can be found in 'outnvl'.
1008  *
1009  * ENOMEM   The program fully executed, but the output buffer was not large
1010  *          enough to store the returned value. No output is returned through
1011  *          'outnvl'.
1012  *
1013  * ENOSPC   The program was terminated because it exceeded its memory usage
1014  *          limit. Some portion of the channel program may have executed and
1015  *          committed changes to disk. No output is returned through 'outnvl'.
1016  *
1017  * ETIMEDOUT The program was terminated because it exceeded its Lua instruction
1018  *           limit. Some portion of the channel program may have executed and
1019  *           committed changes to disk. No output is returned through 'outnvl'.
1020  */
1021 int
1022 lzc_channel_program(const char *pool, const char *program, uint64_t instrlimit,
1023     uint64_t memlimit, nvlist_t *argnvl, nvlist_t **outnvl)
1024 {
1025 	return (lzc_channel_program_impl(pool, program, B_TRUE, instrlimit,
1026 	    memlimit, argnvl, outnvl));
1027 }
1028 
1029 /*
1030  * Creates a checkpoint for the specified pool.
1031  *
1032  * If this function returns 0 the pool was successfully checkpointed.
1033  *
1034  * This method may also return:
1035  *
1036  * ZFS_ERR_CHECKPOINT_EXISTS
1037  *	The pool already has a checkpoint. A pools can only have one
1038  *	checkpoint at most, at any given time.
1039  *
1040  * ZFS_ERR_DISCARDING_CHECKPOINT
1041  * 	ZFS is in the middle of discarding a checkpoint for this pool.
1042  * 	The pool can be checkpointed again once the discard is done.
1043  *
1044  * ZFS_DEVRM_IN_PROGRESS
1045  * 	A vdev is currently being removed. The pool cannot be
1046  * 	checkpointed until the device removal is done.
1047  *
1048  * ZFS_VDEV_TOO_BIG
1049  * 	One or more top-level vdevs exceed the maximum vdev size
1050  * 	supported for this feature.
1051  */
1052 int
1053 lzc_pool_checkpoint(const char *pool)
1054 {
1055 	int error;
1056 
1057 	nvlist_t *result = NULL;
1058 	nvlist_t *args = fnvlist_alloc();
1059 
1060 	error = lzc_ioctl(ZFS_IOC_POOL_CHECKPOINT, pool, args, &result);
1061 
1062 	fnvlist_free(args);
1063 	fnvlist_free(result);
1064 
1065 	return (error);
1066 }
1067 
1068 /*
1069  * Discard the checkpoint from the specified pool.
1070  *
1071  * If this function returns 0 the checkpoint was successfully discarded.
1072  *
1073  * This method may also return:
1074  *
1075  * ZFS_ERR_NO_CHECKPOINT
1076  * 	The pool does not have a checkpoint.
1077  *
1078  * ZFS_ERR_DISCARDING_CHECKPOINT
1079  * 	ZFS is already in the middle of discarding the checkpoint.
1080  */
1081 int
1082 lzc_pool_checkpoint_discard(const char *pool)
1083 {
1084 	int error;
1085 
1086 	nvlist_t *result = NULL;
1087 	nvlist_t *args = fnvlist_alloc();
1088 
1089 	error = lzc_ioctl(ZFS_IOC_POOL_DISCARD_CHECKPOINT, pool, args, &result);
1090 
1091 	fnvlist_free(args);
1092 	fnvlist_free(result);
1093 
1094 	return (error);
1095 }
1096 
1097 /*
1098  * Executes a read-only channel program.
1099  *
1100  * A read-only channel program works programmatically the same way as a
1101  * normal channel program executed with lzc_channel_program(). The only
1102  * difference is it runs exclusively in open-context and therefore can
1103  * return faster. The downside to that, is that the program cannot change
1104  * on-disk state by calling functions from the zfs.sync submodule.
1105  *
1106  * The return values of this function (and their meaning) are exactly the
1107  * same as the ones described in lzc_channel_program().
1108  */
1109 int
1110 lzc_channel_program_nosync(const char *pool, const char *program,
1111     uint64_t timeout, uint64_t memlimit, nvlist_t *argnvl, nvlist_t **outnvl)
1112 {
1113 	return (lzc_channel_program_impl(pool, program, B_FALSE, timeout,
1114 	    memlimit, argnvl, outnvl));
1115 }
1116 
1117 /*
1118  * Changes initializing state.
1119  *
1120  * vdevs should be a list of (<key>, guid) where guid is a uint64 vdev GUID.
1121  * The key is ignored.
1122  *
1123  * If there are errors related to vdev arguments, per-vdev errors are returned
1124  * in an nvlist with the key "vdevs". Each error is a (guid, errno) pair where
1125  * guid is stringified with PRIu64, and errno is one of the following as
1126  * an int64_t:
1127  *	- ENODEV if the device was not found
1128  *	- EINVAL if the devices is not a leaf or is not concrete (e.g. missing)
1129  *	- EROFS if the device is not writeable
1130  *	- EBUSY start requested but the device is already being initialized
1131  *	- ESRCH cancel/suspend requested but device is not being initialized
1132  *
1133  * If the errlist is empty, then return value will be:
1134  *	- EINVAL if one or more arguments was invalid
1135  *	- Other spa_open failures
1136  *	- 0 if the operation succeeded
1137  */
1138 int
1139 lzc_initialize(const char *poolname, pool_initialize_func_t cmd_type,
1140     nvlist_t *vdevs, nvlist_t **errlist)
1141 {
1142 	int error;
1143 	nvlist_t *args = fnvlist_alloc();
1144 	fnvlist_add_uint64(args, ZPOOL_INITIALIZE_COMMAND, (uint64_t)cmd_type);
1145 	fnvlist_add_nvlist(args, ZPOOL_INITIALIZE_VDEVS, vdevs);
1146 
1147 	error = lzc_ioctl(ZFS_IOC_POOL_INITIALIZE, poolname, args, errlist);
1148 
1149 	fnvlist_free(args);
1150 
1151 	return (error);
1152 }
1153