xref: /freebsd-src/sys/contrib/openzfs/module/zfs/zcp_synctask.c (revision c7046f76c2c027b00c0e6ba57cfd28f1a78f5e23)
1eda14cbcSMatt Macy /*
2eda14cbcSMatt Macy  * CDDL HEADER START
3eda14cbcSMatt Macy  *
4eda14cbcSMatt Macy  * This file and its contents are supplied under the terms of the
5eda14cbcSMatt Macy  * Common Development and Distribution License ("CDDL"), version 1.0.
6eda14cbcSMatt Macy  * You may only use this file in accordance with the terms of version
7eda14cbcSMatt Macy  * 1.0 of the CDDL.
8eda14cbcSMatt Macy  *
9eda14cbcSMatt Macy  * A full copy of the text of the CDDL should have accompanied this
10eda14cbcSMatt Macy  * source.  A copy of the CDDL is also available via the Internet at
11eda14cbcSMatt Macy  * http://www.illumos.org/license/CDDL.
12eda14cbcSMatt Macy  *
13eda14cbcSMatt Macy  * CDDL HEADER END
14eda14cbcSMatt Macy  */
15eda14cbcSMatt Macy 
16eda14cbcSMatt Macy /*
17eda14cbcSMatt Macy  * Copyright (c) 2016, 2017 by Delphix. All rights reserved.
18eda14cbcSMatt Macy  * Copyright (c) 2019, 2020 by Christian Schwarz. All rights reserved.
19eda14cbcSMatt Macy  * Copyright 2020 Joyent, Inc.
20eda14cbcSMatt Macy  */
21eda14cbcSMatt Macy 
22eda14cbcSMatt Macy #include <sys/lua/lua.h>
23eda14cbcSMatt Macy #include <sys/lua/lauxlib.h>
24eda14cbcSMatt Macy 
25eda14cbcSMatt Macy #include <sys/zcp.h>
26eda14cbcSMatt Macy #include <sys/zcp_set.h>
27eda14cbcSMatt Macy #include <sys/dsl_dir.h>
28eda14cbcSMatt Macy #include <sys/dsl_pool.h>
29eda14cbcSMatt Macy #include <sys/dsl_prop.h>
30eda14cbcSMatt Macy #include <sys/dsl_synctask.h>
31eda14cbcSMatt Macy #include <sys/dsl_dataset.h>
32eda14cbcSMatt Macy #include <sys/dsl_bookmark.h>
33eda14cbcSMatt Macy #include <sys/dsl_destroy.h>
34eda14cbcSMatt Macy #include <sys/dmu_objset.h>
35eda14cbcSMatt Macy #include <sys/zfs_znode.h>
36eda14cbcSMatt Macy #include <sys/zfeature.h>
37eda14cbcSMatt Macy #include <sys/metaslab.h>
38eda14cbcSMatt Macy 
39eda14cbcSMatt Macy #define	DST_AVG_BLKSHIFT 14
40eda14cbcSMatt Macy 
41eda14cbcSMatt Macy typedef struct zcp_inherit_prop_arg {
42eda14cbcSMatt Macy 	lua_State		*zipa_state;
43eda14cbcSMatt Macy 	const char		*zipa_prop;
44eda14cbcSMatt Macy 	dsl_props_set_arg_t	zipa_dpsa;
45eda14cbcSMatt Macy } zcp_inherit_prop_arg_t;
46eda14cbcSMatt Macy 
47eda14cbcSMatt Macy typedef int (zcp_synctask_func_t)(lua_State *, boolean_t, nvlist_t *);
48eda14cbcSMatt Macy typedef struct zcp_synctask_info {
49eda14cbcSMatt Macy 	const char *name;
50eda14cbcSMatt Macy 	zcp_synctask_func_t *func;
51eda14cbcSMatt Macy 	const zcp_arg_t pargs[4];
52eda14cbcSMatt Macy 	const zcp_arg_t kwargs[2];
53eda14cbcSMatt Macy 	zfs_space_check_t space_check;
54eda14cbcSMatt Macy 	int blocks_modified;
55eda14cbcSMatt Macy } zcp_synctask_info_t;
56eda14cbcSMatt Macy 
573f9d360cSMartin Matuska static void
zcp_synctask_cleanup(void * arg)583f9d360cSMartin Matuska zcp_synctask_cleanup(void *arg)
593f9d360cSMartin Matuska {
603f9d360cSMartin Matuska 	fnvlist_free(arg);
613f9d360cSMartin Matuska }
623f9d360cSMartin Matuska 
63eda14cbcSMatt Macy /*
64eda14cbcSMatt Macy  * Generic synctask interface for channel program syncfuncs.
65eda14cbcSMatt Macy  *
66eda14cbcSMatt Macy  * To perform some action in syncing context, we'd generally call
67eda14cbcSMatt Macy  * dsl_sync_task(), but since the Lua script is already running inside a
68eda14cbcSMatt Macy  * synctask we need to leave out some actions (such as acquiring the config
69eda14cbcSMatt Macy  * rwlock and performing space checks).
70eda14cbcSMatt Macy  *
71eda14cbcSMatt Macy  * If 'sync' is false, executes a dry run and returns the error code.
72eda14cbcSMatt Macy  *
73eda14cbcSMatt Macy  * If we are not running in syncing context and we are not doing a dry run
74eda14cbcSMatt Macy  * (meaning we are running a zfs.sync function in open-context) then we
75eda14cbcSMatt Macy  * return a Lua error.
76eda14cbcSMatt Macy  *
77eda14cbcSMatt Macy  * This function also handles common fatal error cases for channel program
78eda14cbcSMatt Macy  * library functions. If a fatal error occurs, err_dsname will be the dataset
79eda14cbcSMatt Macy  * name reported in error messages, if supplied.
80eda14cbcSMatt Macy  */
81eda14cbcSMatt Macy static int
zcp_sync_task(lua_State * state,dsl_checkfunc_t * checkfunc,dsl_syncfunc_t * syncfunc,void * arg,boolean_t sync,const char * err_dsname)82eda14cbcSMatt Macy zcp_sync_task(lua_State *state, dsl_checkfunc_t *checkfunc,
83eda14cbcSMatt Macy     dsl_syncfunc_t *syncfunc, void *arg, boolean_t sync, const char *err_dsname)
84eda14cbcSMatt Macy {
85eda14cbcSMatt Macy 	int err;
86eda14cbcSMatt Macy 	zcp_run_info_t *ri = zcp_run_info(state);
87eda14cbcSMatt Macy 
88eda14cbcSMatt Macy 	err = checkfunc(arg, ri->zri_tx);
89eda14cbcSMatt Macy 	if (!sync)
90eda14cbcSMatt Macy 		return (err);
91eda14cbcSMatt Macy 
92eda14cbcSMatt Macy 	if (!ri->zri_sync) {
93eda14cbcSMatt Macy 		return (luaL_error(state, "running functions from the zfs.sync "
94eda14cbcSMatt Macy 		    "submodule requires passing sync=TRUE to "
95eda14cbcSMatt Macy 		    "lzc_channel_program() (i.e. do not specify the \"-n\" "
96eda14cbcSMatt Macy 		    "command line argument)"));
97eda14cbcSMatt Macy 	}
98eda14cbcSMatt Macy 
99eda14cbcSMatt Macy 	if (err == 0) {
100eda14cbcSMatt Macy 		syncfunc(arg, ri->zri_tx);
101eda14cbcSMatt Macy 	} else if (err == EIO) {
102eda14cbcSMatt Macy 		if (err_dsname != NULL) {
103eda14cbcSMatt Macy 			return (luaL_error(state,
104eda14cbcSMatt Macy 			    "I/O error while accessing dataset '%s'",
105eda14cbcSMatt Macy 			    err_dsname));
106eda14cbcSMatt Macy 		} else {
107eda14cbcSMatt Macy 			return (luaL_error(state,
108eda14cbcSMatt Macy 			    "I/O error while accessing dataset."));
109eda14cbcSMatt Macy 		}
110eda14cbcSMatt Macy 	}
111eda14cbcSMatt Macy 
112eda14cbcSMatt Macy 	return (err);
113eda14cbcSMatt Macy }
114eda14cbcSMatt Macy 
115eda14cbcSMatt Macy 
116eda14cbcSMatt Macy static int zcp_synctask_destroy(lua_State *, boolean_t, nvlist_t *);
117e92ffd9bSMartin Matuska static const zcp_synctask_info_t zcp_synctask_destroy_info = {
118eda14cbcSMatt Macy 	.name = "destroy",
119eda14cbcSMatt Macy 	.func = zcp_synctask_destroy,
120eda14cbcSMatt Macy 	.pargs = {
121eda14cbcSMatt Macy 	    {.za_name = "filesystem | snapshot", .za_lua_type = LUA_TSTRING },
122eda14cbcSMatt Macy 	    {NULL, 0}
123eda14cbcSMatt Macy 	},
124eda14cbcSMatt Macy 	.kwargs = {
125eda14cbcSMatt Macy 	    {.za_name = "defer", .za_lua_type = LUA_TBOOLEAN },
126eda14cbcSMatt Macy 	    {NULL, 0}
127eda14cbcSMatt Macy 	},
128eda14cbcSMatt Macy 	.space_check = ZFS_SPACE_CHECK_DESTROY,
129eda14cbcSMatt Macy 	.blocks_modified = 0
130eda14cbcSMatt Macy };
131eda14cbcSMatt Macy 
132eda14cbcSMatt Macy static int
zcp_synctask_destroy(lua_State * state,boolean_t sync,nvlist_t * err_details)133eda14cbcSMatt Macy zcp_synctask_destroy(lua_State *state, boolean_t sync, nvlist_t *err_details)
134eda14cbcSMatt Macy {
135e92ffd9bSMartin Matuska 	(void) err_details;
136eda14cbcSMatt Macy 	int err;
137eda14cbcSMatt Macy 	const char *dsname = lua_tostring(state, 1);
138eda14cbcSMatt Macy 
139eda14cbcSMatt Macy 	boolean_t issnap = (strchr(dsname, '@') != NULL);
140eda14cbcSMatt Macy 
141eda14cbcSMatt Macy 	if (!issnap && !lua_isnil(state, 2)) {
142eda14cbcSMatt Macy 		return (luaL_error(state,
143eda14cbcSMatt Macy 		    "'deferred' kwarg only supported for snapshots: %s",
144eda14cbcSMatt Macy 		    dsname));
145eda14cbcSMatt Macy 	}
146eda14cbcSMatt Macy 
147eda14cbcSMatt Macy 	if (issnap) {
148eda14cbcSMatt Macy 		dsl_destroy_snapshot_arg_t ddsa = { 0 };
149eda14cbcSMatt Macy 		ddsa.ddsa_name = dsname;
150eda14cbcSMatt Macy 		if (!lua_isnil(state, 2)) {
151eda14cbcSMatt Macy 			ddsa.ddsa_defer = lua_toboolean(state, 2);
152eda14cbcSMatt Macy 		} else {
153eda14cbcSMatt Macy 			ddsa.ddsa_defer = B_FALSE;
154eda14cbcSMatt Macy 		}
155eda14cbcSMatt Macy 
156eda14cbcSMatt Macy 		err = zcp_sync_task(state, dsl_destroy_snapshot_check,
157eda14cbcSMatt Macy 		    dsl_destroy_snapshot_sync, &ddsa, sync, dsname);
158eda14cbcSMatt Macy 	} else {
159eda14cbcSMatt Macy 		dsl_destroy_head_arg_t ddha = { 0 };
160eda14cbcSMatt Macy 		ddha.ddha_name = dsname;
161eda14cbcSMatt Macy 
162eda14cbcSMatt Macy 		err = zcp_sync_task(state, dsl_destroy_head_check,
163eda14cbcSMatt Macy 		    dsl_destroy_head_sync, &ddha, sync, dsname);
164eda14cbcSMatt Macy 	}
165eda14cbcSMatt Macy 
166eda14cbcSMatt Macy 	return (err);
167eda14cbcSMatt Macy }
168eda14cbcSMatt Macy 
169eda14cbcSMatt Macy static int zcp_synctask_promote(lua_State *, boolean_t, nvlist_t *);
170e92ffd9bSMartin Matuska static const zcp_synctask_info_t zcp_synctask_promote_info = {
171eda14cbcSMatt Macy 	.name = "promote",
172eda14cbcSMatt Macy 	.func = zcp_synctask_promote,
173eda14cbcSMatt Macy 	.pargs = {
174eda14cbcSMatt Macy 	    {.za_name = "clone", .za_lua_type = LUA_TSTRING },
175eda14cbcSMatt Macy 	    {NULL, 0}
176eda14cbcSMatt Macy 	},
177eda14cbcSMatt Macy 	.kwargs = {
178eda14cbcSMatt Macy 	    {NULL, 0}
179eda14cbcSMatt Macy 	},
180eda14cbcSMatt Macy 	.space_check = ZFS_SPACE_CHECK_RESERVED,
181eda14cbcSMatt Macy 	.blocks_modified = 3
182eda14cbcSMatt Macy };
183eda14cbcSMatt Macy 
184eda14cbcSMatt Macy static int
zcp_synctask_promote(lua_State * state,boolean_t sync,nvlist_t * err_details)185eda14cbcSMatt Macy zcp_synctask_promote(lua_State *state, boolean_t sync, nvlist_t *err_details)
186eda14cbcSMatt Macy {
187eda14cbcSMatt Macy 	int err;
188eda14cbcSMatt Macy 	dsl_dataset_promote_arg_t ddpa = { 0 };
189eda14cbcSMatt Macy 	const char *dsname = lua_tostring(state, 1);
190eda14cbcSMatt Macy 	zcp_run_info_t *ri = zcp_run_info(state);
191eda14cbcSMatt Macy 
192eda14cbcSMatt Macy 	ddpa.ddpa_clonename = dsname;
193eda14cbcSMatt Macy 	ddpa.err_ds = err_details;
194eda14cbcSMatt Macy 	ddpa.cr = ri->zri_cred;
195eda14cbcSMatt Macy 	ddpa.proc = ri->zri_proc;
196eda14cbcSMatt Macy 
197eda14cbcSMatt Macy 	/*
198eda14cbcSMatt Macy 	 * If there was a snapshot name conflict, then err_ds will be filled
199eda14cbcSMatt Macy 	 * with a list of conflicting snapshot names.
200eda14cbcSMatt Macy 	 */
201eda14cbcSMatt Macy 	err = zcp_sync_task(state, dsl_dataset_promote_check,
202eda14cbcSMatt Macy 	    dsl_dataset_promote_sync, &ddpa, sync, dsname);
203eda14cbcSMatt Macy 
204eda14cbcSMatt Macy 	return (err);
205eda14cbcSMatt Macy }
206eda14cbcSMatt Macy 
207eda14cbcSMatt Macy static int zcp_synctask_rollback(lua_State *, boolean_t, nvlist_t *err_details);
208e92ffd9bSMartin Matuska static const zcp_synctask_info_t zcp_synctask_rollback_info = {
209eda14cbcSMatt Macy 	.name = "rollback",
210eda14cbcSMatt Macy 	.func = zcp_synctask_rollback,
211eda14cbcSMatt Macy 	.space_check = ZFS_SPACE_CHECK_RESERVED,
212eda14cbcSMatt Macy 	.blocks_modified = 1,
213eda14cbcSMatt Macy 	.pargs = {
214eda14cbcSMatt Macy 	    {.za_name = "filesystem", .za_lua_type = LUA_TSTRING },
215eda14cbcSMatt Macy 	    {0, 0}
216eda14cbcSMatt Macy 	},
217eda14cbcSMatt Macy 	.kwargs = {
218eda14cbcSMatt Macy 	    {0, 0}
219eda14cbcSMatt Macy 	}
220eda14cbcSMatt Macy };
221eda14cbcSMatt Macy 
222eda14cbcSMatt Macy static int
zcp_synctask_rollback(lua_State * state,boolean_t sync,nvlist_t * err_details)223eda14cbcSMatt Macy zcp_synctask_rollback(lua_State *state, boolean_t sync, nvlist_t *err_details)
224eda14cbcSMatt Macy {
225eda14cbcSMatt Macy 	int err;
226eda14cbcSMatt Macy 	const char *dsname = lua_tostring(state, 1);
227eda14cbcSMatt Macy 	dsl_dataset_rollback_arg_t ddra = { 0 };
228eda14cbcSMatt Macy 
229eda14cbcSMatt Macy 	ddra.ddra_fsname = dsname;
230eda14cbcSMatt Macy 	ddra.ddra_result = err_details;
231eda14cbcSMatt Macy 
232eda14cbcSMatt Macy 	err = zcp_sync_task(state, dsl_dataset_rollback_check,
233eda14cbcSMatt Macy 	    dsl_dataset_rollback_sync, &ddra, sync, dsname);
234eda14cbcSMatt Macy 
235eda14cbcSMatt Macy 	return (err);
236eda14cbcSMatt Macy }
237eda14cbcSMatt Macy 
238eda14cbcSMatt Macy static int zcp_synctask_snapshot(lua_State *, boolean_t, nvlist_t *);
239e92ffd9bSMartin Matuska static const zcp_synctask_info_t zcp_synctask_snapshot_info = {
240eda14cbcSMatt Macy 	.name = "snapshot",
241eda14cbcSMatt Macy 	.func = zcp_synctask_snapshot,
242eda14cbcSMatt Macy 	.pargs = {
243eda14cbcSMatt Macy 	    {.za_name = "filesystem@snapname | volume@snapname",
244eda14cbcSMatt Macy 	    .za_lua_type = LUA_TSTRING },
245eda14cbcSMatt Macy 	    {NULL, 0}
246eda14cbcSMatt Macy 	},
247eda14cbcSMatt Macy 	.kwargs = {
248eda14cbcSMatt Macy 	    {NULL, 0}
249eda14cbcSMatt Macy 	},
250eda14cbcSMatt Macy 	.space_check = ZFS_SPACE_CHECK_NORMAL,
251eda14cbcSMatt Macy 	.blocks_modified = 3
252eda14cbcSMatt Macy };
253eda14cbcSMatt Macy 
254eda14cbcSMatt Macy static int
zcp_synctask_snapshot(lua_State * state,boolean_t sync,nvlist_t * err_details)255eda14cbcSMatt Macy zcp_synctask_snapshot(lua_State *state, boolean_t sync, nvlist_t *err_details)
256eda14cbcSMatt Macy {
257e92ffd9bSMartin Matuska 	(void) err_details;
258eda14cbcSMatt Macy 	int err;
259eda14cbcSMatt Macy 	dsl_dataset_snapshot_arg_t ddsa = { 0 };
260eda14cbcSMatt Macy 	const char *dsname = lua_tostring(state, 1);
261eda14cbcSMatt Macy 	zcp_run_info_t *ri = zcp_run_info(state);
262eda14cbcSMatt Macy 
263eda14cbcSMatt Macy 	/*
264eda14cbcSMatt Macy 	 * On old pools, the ZIL must not be active when a snapshot is created,
265eda14cbcSMatt Macy 	 * but we can't suspend the ZIL because we're already in syncing
266eda14cbcSMatt Macy 	 * context.
267eda14cbcSMatt Macy 	 */
268eda14cbcSMatt Macy 	if (spa_version(ri->zri_pool->dp_spa) < SPA_VERSION_FAST_SNAP) {
269eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
270eda14cbcSMatt Macy 	}
271eda14cbcSMatt Macy 
272eda14cbcSMatt Macy 	/*
273eda14cbcSMatt Macy 	 * We only allow for a single snapshot rather than a list, so the
274eda14cbcSMatt Macy 	 * error list output is unnecessary.
275eda14cbcSMatt Macy 	 */
276eda14cbcSMatt Macy 	ddsa.ddsa_errors = NULL;
277eda14cbcSMatt Macy 	ddsa.ddsa_props = NULL;
278eda14cbcSMatt Macy 	ddsa.ddsa_cr = ri->zri_cred;
279eda14cbcSMatt Macy 	ddsa.ddsa_proc = ri->zri_proc;
280eda14cbcSMatt Macy 	ddsa.ddsa_snaps = fnvlist_alloc();
281eda14cbcSMatt Macy 	fnvlist_add_boolean(ddsa.ddsa_snaps, dsname);
282eda14cbcSMatt Macy 
283eda14cbcSMatt Macy 	zcp_cleanup_handler_t *zch = zcp_register_cleanup(state,
2843f9d360cSMartin Matuska 	    zcp_synctask_cleanup, ddsa.ddsa_snaps);
285eda14cbcSMatt Macy 
286eda14cbcSMatt Macy 	err = zcp_sync_task(state, dsl_dataset_snapshot_check,
287eda14cbcSMatt Macy 	    dsl_dataset_snapshot_sync, &ddsa, sync, dsname);
288eda14cbcSMatt Macy 
289eda14cbcSMatt Macy 	if (err == 0) {
290eda14cbcSMatt Macy 		/*
291eda14cbcSMatt Macy 		 * We may need to create a new device minor node for this
292eda14cbcSMatt Macy 		 * dataset (if it is a zvol and the "snapdev" property is set).
293eda14cbcSMatt Macy 		 * Save it in the nvlist so that it can be processed in open
294eda14cbcSMatt Macy 		 * context.
295eda14cbcSMatt Macy 		 */
296eda14cbcSMatt Macy 		fnvlist_add_boolean(ri->zri_new_zvols, dsname);
297eda14cbcSMatt Macy 	}
298eda14cbcSMatt Macy 
299eda14cbcSMatt Macy 	zcp_deregister_cleanup(state, zch);
300eda14cbcSMatt Macy 	fnvlist_free(ddsa.ddsa_snaps);
301eda14cbcSMatt Macy 
302eda14cbcSMatt Macy 	return (err);
303eda14cbcSMatt Macy }
304eda14cbcSMatt Macy 
305*c7046f76SMartin Matuska static int zcp_synctask_rename_snapshot(lua_State *, boolean_t, nvlist_t *);
306*c7046f76SMartin Matuska static const zcp_synctask_info_t zcp_synctask_rename_snapshot_info = {
307*c7046f76SMartin Matuska 	.name = "rename_snapshot",
308*c7046f76SMartin Matuska 	.func = zcp_synctask_rename_snapshot,
309*c7046f76SMartin Matuska 	.pargs = {
310*c7046f76SMartin Matuska 	    {.za_name = "filesystem | volume", .za_lua_type = LUA_TSTRING },
311*c7046f76SMartin Matuska 	    {.za_name = "oldsnapname", .za_lua_type = LUA_TSTRING },
312*c7046f76SMartin Matuska 	    {.za_name = "newsnapname", .za_lua_type = LUA_TSTRING },
313*c7046f76SMartin Matuska 	    {NULL, 0}
314*c7046f76SMartin Matuska 	},
315*c7046f76SMartin Matuska 	.space_check = ZFS_SPACE_CHECK_RESERVED,
316*c7046f76SMartin Matuska 	.blocks_modified = 1
317*c7046f76SMartin Matuska };
318*c7046f76SMartin Matuska 
319*c7046f76SMartin Matuska static int
zcp_synctask_rename_snapshot(lua_State * state,boolean_t sync,nvlist_t * err_details)320*c7046f76SMartin Matuska zcp_synctask_rename_snapshot(lua_State *state, boolean_t sync,
321*c7046f76SMartin Matuska     nvlist_t *err_details)
322*c7046f76SMartin Matuska {
323*c7046f76SMartin Matuska 	(void) err_details;
324*c7046f76SMartin Matuska 	int err;
325*c7046f76SMartin Matuska 	const char *fsname = lua_tostring(state, 1);
326*c7046f76SMartin Matuska 	const char *oldsnapname = lua_tostring(state, 2);
327*c7046f76SMartin Matuska 	const char *newsnapname = lua_tostring(state, 3);
328*c7046f76SMartin Matuska 
329*c7046f76SMartin Matuska 	struct dsl_dataset_rename_snapshot_arg ddrsa = { 0 };
330*c7046f76SMartin Matuska 	ddrsa.ddrsa_fsname = fsname;
331*c7046f76SMartin Matuska 	ddrsa.ddrsa_oldsnapname = oldsnapname;
332*c7046f76SMartin Matuska 	ddrsa.ddrsa_newsnapname = newsnapname;
333*c7046f76SMartin Matuska 	ddrsa.ddrsa_recursive = B_FALSE;
334*c7046f76SMartin Matuska 
335*c7046f76SMartin Matuska 	err = zcp_sync_task(state, dsl_dataset_rename_snapshot_check,
336*c7046f76SMartin Matuska 	    dsl_dataset_rename_snapshot_sync, &ddrsa, sync, NULL);
337*c7046f76SMartin Matuska 
338*c7046f76SMartin Matuska 	return (err);
339*c7046f76SMartin Matuska }
340*c7046f76SMartin Matuska 
341eda14cbcSMatt Macy static int zcp_synctask_inherit_prop(lua_State *, boolean_t,
342eda14cbcSMatt Macy     nvlist_t *err_details);
343e92ffd9bSMartin Matuska static const zcp_synctask_info_t zcp_synctask_inherit_prop_info = {
344eda14cbcSMatt Macy 	.name = "inherit",
345eda14cbcSMatt Macy 	.func = zcp_synctask_inherit_prop,
346eda14cbcSMatt Macy 	.space_check = ZFS_SPACE_CHECK_RESERVED,
347eda14cbcSMatt Macy 	.blocks_modified = 2, /* 2 * numprops */
348eda14cbcSMatt Macy 	.pargs = {
349eda14cbcSMatt Macy 		{ .za_name = "dataset", .za_lua_type = LUA_TSTRING },
350eda14cbcSMatt Macy 		{ .za_name = "property", .za_lua_type = LUA_TSTRING },
351eda14cbcSMatt Macy 		{ NULL, 0 }
352eda14cbcSMatt Macy 	},
353eda14cbcSMatt Macy 	.kwargs = {
354eda14cbcSMatt Macy 		{ NULL, 0 }
355eda14cbcSMatt Macy 	},
356eda14cbcSMatt Macy };
357eda14cbcSMatt Macy 
358eda14cbcSMatt Macy static int
zcp_synctask_inherit_prop_check(void * arg,dmu_tx_t * tx)359eda14cbcSMatt Macy zcp_synctask_inherit_prop_check(void *arg, dmu_tx_t *tx)
360eda14cbcSMatt Macy {
361eda14cbcSMatt Macy 	zcp_inherit_prop_arg_t *args = arg;
362eda14cbcSMatt Macy 	zfs_prop_t prop = zfs_name_to_prop(args->zipa_prop);
363eda14cbcSMatt Macy 
3641f1e2261SMartin Matuska 	if (prop == ZPROP_USERPROP) {
365eda14cbcSMatt Macy 		if (zfs_prop_user(args->zipa_prop))
366eda14cbcSMatt Macy 			return (0);
367eda14cbcSMatt Macy 
368eda14cbcSMatt Macy 		return (EINVAL);
369eda14cbcSMatt Macy 	}
370eda14cbcSMatt Macy 
371eda14cbcSMatt Macy 	if (zfs_prop_readonly(prop))
372eda14cbcSMatt Macy 		return (EINVAL);
373eda14cbcSMatt Macy 
374eda14cbcSMatt Macy 	if (!zfs_prop_inheritable(prop))
375eda14cbcSMatt Macy 		return (EINVAL);
376eda14cbcSMatt Macy 
377eda14cbcSMatt Macy 	return (dsl_props_set_check(&args->zipa_dpsa, tx));
378eda14cbcSMatt Macy }
379eda14cbcSMatt Macy 
380eda14cbcSMatt Macy static void
zcp_synctask_inherit_prop_sync(void * arg,dmu_tx_t * tx)381eda14cbcSMatt Macy zcp_synctask_inherit_prop_sync(void *arg, dmu_tx_t *tx)
382eda14cbcSMatt Macy {
383eda14cbcSMatt Macy 	zcp_inherit_prop_arg_t *args = arg;
384eda14cbcSMatt Macy 	dsl_props_set_arg_t *dpsa = &args->zipa_dpsa;
385eda14cbcSMatt Macy 
386eda14cbcSMatt Macy 	dsl_props_set_sync(dpsa, tx);
387eda14cbcSMatt Macy }
388eda14cbcSMatt Macy 
389eda14cbcSMatt Macy static int
zcp_synctask_inherit_prop(lua_State * state,boolean_t sync,nvlist_t * err_details)390eda14cbcSMatt Macy zcp_synctask_inherit_prop(lua_State *state, boolean_t sync,
391eda14cbcSMatt Macy     nvlist_t *err_details)
392eda14cbcSMatt Macy {
393e92ffd9bSMartin Matuska 	(void) err_details;
394eda14cbcSMatt Macy 	int err;
395eda14cbcSMatt Macy 	zcp_inherit_prop_arg_t zipa = { 0 };
396eda14cbcSMatt Macy 	dsl_props_set_arg_t *dpsa = &zipa.zipa_dpsa;
397eda14cbcSMatt Macy 
398eda14cbcSMatt Macy 	const char *dsname = lua_tostring(state, 1);
399eda14cbcSMatt Macy 	const char *prop = lua_tostring(state, 2);
400eda14cbcSMatt Macy 
401eda14cbcSMatt Macy 	zipa.zipa_state = state;
402eda14cbcSMatt Macy 	zipa.zipa_prop = prop;
403eda14cbcSMatt Macy 	dpsa->dpsa_dsname = dsname;
404eda14cbcSMatt Macy 	dpsa->dpsa_source = ZPROP_SRC_INHERITED;
405eda14cbcSMatt Macy 	dpsa->dpsa_props = fnvlist_alloc();
406eda14cbcSMatt Macy 	fnvlist_add_boolean(dpsa->dpsa_props, prop);
407eda14cbcSMatt Macy 
408eda14cbcSMatt Macy 	zcp_cleanup_handler_t *zch = zcp_register_cleanup(state,
4093f9d360cSMartin Matuska 	    zcp_synctask_cleanup, dpsa->dpsa_props);
410eda14cbcSMatt Macy 
411eda14cbcSMatt Macy 	err = zcp_sync_task(state, zcp_synctask_inherit_prop_check,
412eda14cbcSMatt Macy 	    zcp_synctask_inherit_prop_sync, &zipa, sync, dsname);
413eda14cbcSMatt Macy 
414eda14cbcSMatt Macy 	zcp_deregister_cleanup(state, zch);
415eda14cbcSMatt Macy 	fnvlist_free(dpsa->dpsa_props);
416eda14cbcSMatt Macy 
417eda14cbcSMatt Macy 	return (err);
418eda14cbcSMatt Macy }
419eda14cbcSMatt Macy 
420eda14cbcSMatt Macy static int zcp_synctask_bookmark(lua_State *, boolean_t, nvlist_t *);
421e92ffd9bSMartin Matuska static const zcp_synctask_info_t zcp_synctask_bookmark_info = {
422eda14cbcSMatt Macy 	.name = "bookmark",
423eda14cbcSMatt Macy 	.func = zcp_synctask_bookmark,
424eda14cbcSMatt Macy 	.pargs = {
425eda14cbcSMatt Macy 	    {.za_name = "snapshot | bookmark", .za_lua_type = LUA_TSTRING },
426eda14cbcSMatt Macy 	    {.za_name = "bookmark", .za_lua_type = LUA_TSTRING },
427eda14cbcSMatt Macy 	    {NULL, 0}
428eda14cbcSMatt Macy 	},
429eda14cbcSMatt Macy 	.kwargs = {
430eda14cbcSMatt Macy 	    {NULL, 0}
431eda14cbcSMatt Macy 	},
432eda14cbcSMatt Macy 	.space_check = ZFS_SPACE_CHECK_NORMAL,
433eda14cbcSMatt Macy 	.blocks_modified = 1,
434eda14cbcSMatt Macy };
435eda14cbcSMatt Macy 
436eda14cbcSMatt Macy static int
zcp_synctask_bookmark(lua_State * state,boolean_t sync,nvlist_t * err_details)437eda14cbcSMatt Macy zcp_synctask_bookmark(lua_State *state, boolean_t sync, nvlist_t *err_details)
438eda14cbcSMatt Macy {
439e92ffd9bSMartin Matuska 	(void) err_details;
440eda14cbcSMatt Macy 	int err;
441eda14cbcSMatt Macy 	const char *source = lua_tostring(state, 1);
442eda14cbcSMatt Macy 	const char *new = lua_tostring(state, 2);
443eda14cbcSMatt Macy 
444eda14cbcSMatt Macy 	nvlist_t *bmarks = fnvlist_alloc();
445eda14cbcSMatt Macy 	fnvlist_add_string(bmarks, new, source);
446eda14cbcSMatt Macy 
447eda14cbcSMatt Macy 	zcp_cleanup_handler_t *zch = zcp_register_cleanup(state,
4483f9d360cSMartin Matuska 	    zcp_synctask_cleanup, bmarks);
449eda14cbcSMatt Macy 
450eda14cbcSMatt Macy 	dsl_bookmark_create_arg_t dbca = {
451eda14cbcSMatt Macy 		.dbca_bmarks = bmarks,
452eda14cbcSMatt Macy 		.dbca_errors = NULL,
453eda14cbcSMatt Macy 	};
454eda14cbcSMatt Macy 	err = zcp_sync_task(state, dsl_bookmark_create_check,
455eda14cbcSMatt Macy 	    dsl_bookmark_create_sync, &dbca, sync, source);
456eda14cbcSMatt Macy 
457eda14cbcSMatt Macy 	zcp_deregister_cleanup(state, zch);
458eda14cbcSMatt Macy 	fnvlist_free(bmarks);
459eda14cbcSMatt Macy 
460eda14cbcSMatt Macy 	return (err);
461eda14cbcSMatt Macy }
462eda14cbcSMatt Macy 
463eda14cbcSMatt Macy static int zcp_synctask_set_prop(lua_State *, boolean_t, nvlist_t *err_details);
464e92ffd9bSMartin Matuska static const zcp_synctask_info_t zcp_synctask_set_prop_info = {
465eda14cbcSMatt Macy 	.name = "set_prop",
466eda14cbcSMatt Macy 	.func = zcp_synctask_set_prop,
467eda14cbcSMatt Macy 	.space_check = ZFS_SPACE_CHECK_RESERVED,
468eda14cbcSMatt Macy 	.blocks_modified = 2,
469eda14cbcSMatt Macy 	.pargs = {
470eda14cbcSMatt Macy 		{ .za_name = "dataset", .za_lua_type = LUA_TSTRING },
471eda14cbcSMatt Macy 		{ .za_name = "property", .za_lua_type =  LUA_TSTRING },
472eda14cbcSMatt Macy 		{ .za_name = "value", .za_lua_type =  LUA_TSTRING },
473eda14cbcSMatt Macy 		{ NULL, 0 }
474eda14cbcSMatt Macy 	},
475eda14cbcSMatt Macy 	.kwargs = {
476eda14cbcSMatt Macy 		{ NULL, 0 }
477eda14cbcSMatt Macy 	}
478eda14cbcSMatt Macy };
479eda14cbcSMatt Macy 
480eda14cbcSMatt Macy static int
zcp_synctask_set_prop(lua_State * state,boolean_t sync,nvlist_t * err_details)481eda14cbcSMatt Macy zcp_synctask_set_prop(lua_State *state, boolean_t sync, nvlist_t *err_details)
482eda14cbcSMatt Macy {
483e92ffd9bSMartin Matuska 	(void) err_details;
484eda14cbcSMatt Macy 	int err;
485eda14cbcSMatt Macy 	zcp_set_prop_arg_t args = { 0 };
486eda14cbcSMatt Macy 
487eda14cbcSMatt Macy 	const char *dsname = lua_tostring(state, 1);
488eda14cbcSMatt Macy 	const char *prop = lua_tostring(state, 2);
489eda14cbcSMatt Macy 	const char *val = lua_tostring(state, 3);
490eda14cbcSMatt Macy 
491eda14cbcSMatt Macy 	args.state = state;
492eda14cbcSMatt Macy 	args.dsname = dsname;
493eda14cbcSMatt Macy 	args.prop = prop;
494eda14cbcSMatt Macy 	args.val = val;
495eda14cbcSMatt Macy 
496eda14cbcSMatt Macy 	err = zcp_sync_task(state, zcp_set_prop_check, zcp_set_prop_sync,
497eda14cbcSMatt Macy 	    &args, sync, dsname);
498eda14cbcSMatt Macy 
499eda14cbcSMatt Macy 	return (err);
500eda14cbcSMatt Macy }
501eda14cbcSMatt Macy 
502eda14cbcSMatt Macy static int
zcp_synctask_wrapper(lua_State * state)503eda14cbcSMatt Macy zcp_synctask_wrapper(lua_State *state)
504eda14cbcSMatt Macy {
505eda14cbcSMatt Macy 	int err;
506eda14cbcSMatt Macy 	zcp_cleanup_handler_t *zch;
507eda14cbcSMatt Macy 	int num_ret = 1;
508eda14cbcSMatt Macy 	nvlist_t *err_details = fnvlist_alloc();
509eda14cbcSMatt Macy 
510eda14cbcSMatt Macy 	/*
511eda14cbcSMatt Macy 	 * Make sure err_details is properly freed, even if a fatal error is
512eda14cbcSMatt Macy 	 * thrown during the synctask.
513eda14cbcSMatt Macy 	 */
5143f9d360cSMartin Matuska 	zch = zcp_register_cleanup(state, zcp_synctask_cleanup, err_details);
515eda14cbcSMatt Macy 
516eda14cbcSMatt Macy 	zcp_synctask_info_t *info = lua_touserdata(state, lua_upvalueindex(1));
517eda14cbcSMatt Macy 	boolean_t sync = lua_toboolean(state, lua_upvalueindex(2));
518eda14cbcSMatt Macy 
519eda14cbcSMatt Macy 	zcp_run_info_t *ri = zcp_run_info(state);
520eda14cbcSMatt Macy 	dsl_pool_t *dp = ri->zri_pool;
521eda14cbcSMatt Macy 
522eda14cbcSMatt Macy 	/* MOS space is triple-dittoed, so we multiply by 3. */
523eda14cbcSMatt Macy 	uint64_t funcspace =
524eda14cbcSMatt Macy 	    ((uint64_t)info->blocks_modified << DST_AVG_BLKSHIFT) * 3;
525eda14cbcSMatt Macy 
526eda14cbcSMatt Macy 	zcp_parse_args(state, info->name, info->pargs, info->kwargs);
527eda14cbcSMatt Macy 
528eda14cbcSMatt Macy 	err = 0;
529eda14cbcSMatt Macy 	if (info->space_check != ZFS_SPACE_CHECK_NONE) {
530eda14cbcSMatt Macy 		uint64_t quota = dsl_pool_unreserved_space(dp,
531eda14cbcSMatt Macy 		    info->space_check);
532eda14cbcSMatt Macy 		uint64_t used = dsl_dir_phys(dp->dp_root_dir)->dd_used_bytes +
533eda14cbcSMatt Macy 		    ri->zri_space_used;
534eda14cbcSMatt Macy 
535eda14cbcSMatt Macy 		if (used + funcspace > quota) {
536eda14cbcSMatt Macy 			err = SET_ERROR(ENOSPC);
537eda14cbcSMatt Macy 		}
538eda14cbcSMatt Macy 	}
539eda14cbcSMatt Macy 
540eda14cbcSMatt Macy 	if (err == 0) {
541eda14cbcSMatt Macy 		err = info->func(state, sync, err_details);
542eda14cbcSMatt Macy 	}
543eda14cbcSMatt Macy 
544eda14cbcSMatt Macy 	if (err == 0) {
545eda14cbcSMatt Macy 		ri->zri_space_used += funcspace;
546eda14cbcSMatt Macy 	}
547eda14cbcSMatt Macy 
548eda14cbcSMatt Macy 	lua_pushnumber(state, (lua_Number)err);
549eda14cbcSMatt Macy 	if (fnvlist_num_pairs(err_details) > 0) {
550eda14cbcSMatt Macy 		(void) zcp_nvlist_to_lua(state, err_details, NULL, 0);
551eda14cbcSMatt Macy 		num_ret++;
552eda14cbcSMatt Macy 	}
553eda14cbcSMatt Macy 
554eda14cbcSMatt Macy 	zcp_deregister_cleanup(state, zch);
555eda14cbcSMatt Macy 	fnvlist_free(err_details);
556eda14cbcSMatt Macy 
557eda14cbcSMatt Macy 	return (num_ret);
558eda14cbcSMatt Macy }
559eda14cbcSMatt Macy 
560eda14cbcSMatt Macy int
zcp_load_synctask_lib(lua_State * state,boolean_t sync)561eda14cbcSMatt Macy zcp_load_synctask_lib(lua_State *state, boolean_t sync)
562eda14cbcSMatt Macy {
563e92ffd9bSMartin Matuska 	const zcp_synctask_info_t *zcp_synctask_funcs[] = {
564eda14cbcSMatt Macy 		&zcp_synctask_destroy_info,
565eda14cbcSMatt Macy 		&zcp_synctask_promote_info,
566eda14cbcSMatt Macy 		&zcp_synctask_rollback_info,
567eda14cbcSMatt Macy 		&zcp_synctask_snapshot_info,
568*c7046f76SMartin Matuska 		&zcp_synctask_rename_snapshot_info,
569eda14cbcSMatt Macy 		&zcp_synctask_inherit_prop_info,
570eda14cbcSMatt Macy 		&zcp_synctask_bookmark_info,
571eda14cbcSMatt Macy 		&zcp_synctask_set_prop_info,
572eda14cbcSMatt Macy 		NULL
573eda14cbcSMatt Macy 	};
574eda14cbcSMatt Macy 
575eda14cbcSMatt Macy 	lua_newtable(state);
576eda14cbcSMatt Macy 
577e92ffd9bSMartin Matuska 	for (int i = 0; zcp_synctask_funcs[i] != NULL; i++) {
578e92ffd9bSMartin Matuska 		const zcp_synctask_info_t *info = zcp_synctask_funcs[i];
579e92ffd9bSMartin Matuska 		lua_pushlightuserdata(state, (void *)(uintptr_t)info);
580eda14cbcSMatt Macy 		lua_pushboolean(state, sync);
581eda14cbcSMatt Macy 		lua_pushcclosure(state, &zcp_synctask_wrapper, 2);
582eda14cbcSMatt Macy 		lua_setfield(state, -2, info->name);
583eda14cbcSMatt Macy 	}
584eda14cbcSMatt Macy 
585eda14cbcSMatt Macy 	return (1);
586eda14cbcSMatt Macy }
587