1*b2d2f356SSara Hartse /*
2*b2d2f356SSara Hartse * CDDL HEADER START
3*b2d2f356SSara Hartse *
4*b2d2f356SSara Hartse * This file and its contents are supplied under the terms of the
5*b2d2f356SSara Hartse * Common Development and Distribution License ("CDDL"), version 1.0.
6*b2d2f356SSara Hartse * You may only use this file in accordance with the terms of version
7*b2d2f356SSara Hartse * 1.0 of the CDDL.
8*b2d2f356SSara Hartse *
9*b2d2f356SSara Hartse * A full copy of the text of the CDDL should have accompanied this
10*b2d2f356SSara Hartse * source. A copy of the CDDL is also available via the Internet at
11*b2d2f356SSara Hartse * http://www.illumos.org/license/CDDL.
12*b2d2f356SSara Hartse *
13*b2d2f356SSara Hartse * CDDL HEADER END
14*b2d2f356SSara Hartse */
15*b2d2f356SSara Hartse
16*b2d2f356SSara Hartse /*
17*b2d2f356SSara Hartse * Copyright (c) 2016 by Delphix. All rights reserved.
18*b2d2f356SSara Hartse * Copyrigh 2020 Joyent, Inc.
19*b2d2f356SSara Hartse */
20*b2d2f356SSara Hartse
21*b2d2f356SSara Hartse #include "lua.h"
22*b2d2f356SSara Hartse #include "lualib.h"
23*b2d2f356SSara Hartse #include "lauxlib.h"
24*b2d2f356SSara Hartse
25*b2d2f356SSara Hartse #include <sys/dsl_prop.h>
26*b2d2f356SSara Hartse #include <sys/dsl_dir.h>
27*b2d2f356SSara Hartse #include <sys/dsl_synctask.h>
28*b2d2f356SSara Hartse #include <sys/dsl_dataset.h>
29*b2d2f356SSara Hartse #include <sys/zcp.h>
30*b2d2f356SSara Hartse #include <sys/zcp_set.h>
31*b2d2f356SSara Hartse #include <sys/zcp_iter.h>
32*b2d2f356SSara Hartse #include <sys/zcp_global.h>
33*b2d2f356SSara Hartse #include <sys/zvol.h>
34*b2d2f356SSara Hartse
35*b2d2f356SSara Hartse #include <zfs_prop.h>
36*b2d2f356SSara Hartse
37*b2d2f356SSara Hartse static void
zcp_set_user_prop(lua_State * state,dsl_pool_t * dp,const char * dsname,const char * prop_name,const char * prop_val,dmu_tx_t * tx)38*b2d2f356SSara Hartse zcp_set_user_prop(lua_State *state, dsl_pool_t *dp, const char *dsname,
39*b2d2f356SSara Hartse const char *prop_name, const char *prop_val, dmu_tx_t *tx)
40*b2d2f356SSara Hartse {
41*b2d2f356SSara Hartse dsl_dataset_t *ds = zcp_dataset_hold(state, dp, dsname, FTAG);
42*b2d2f356SSara Hartse if (ds == NULL)
43*b2d2f356SSara Hartse return; /* not reached; zcp_dataset_hold() longjmp'd */
44*b2d2f356SSara Hartse
45*b2d2f356SSara Hartse nvlist_t *nvl = fnvlist_alloc();
46*b2d2f356SSara Hartse fnvlist_add_string(nvl, prop_name, prop_val);
47*b2d2f356SSara Hartse
48*b2d2f356SSara Hartse dsl_props_set_sync_impl(ds, ZPROP_SRC_LOCAL, nvl, tx);
49*b2d2f356SSara Hartse
50*b2d2f356SSara Hartse fnvlist_free(nvl);
51*b2d2f356SSara Hartse dsl_dataset_rele(ds, FTAG);
52*b2d2f356SSara Hartse }
53*b2d2f356SSara Hartse
54*b2d2f356SSara Hartse int
zcp_set_prop_check(void * arg,dmu_tx_t * tx)55*b2d2f356SSara Hartse zcp_set_prop_check(void *arg, dmu_tx_t *tx)
56*b2d2f356SSara Hartse {
57*b2d2f356SSara Hartse zcp_set_prop_arg_t *args = arg;
58*b2d2f356SSara Hartse const char *prop_name = args->prop;
59*b2d2f356SSara Hartse dsl_props_set_arg_t dpsa = {
60*b2d2f356SSara Hartse .dpsa_dsname = args->dsname,
61*b2d2f356SSara Hartse .dpsa_source = ZPROP_SRC_LOCAL,
62*b2d2f356SSara Hartse };
63*b2d2f356SSara Hartse nvlist_t *nvl = NULL;
64*b2d2f356SSara Hartse int ret = 0;
65*b2d2f356SSara Hartse
66*b2d2f356SSara Hartse /*
67*b2d2f356SSara Hartse * Only user properties are currently supported. When non-user
68*b2d2f356SSara Hartse * properties are supported, we will want to use
69*b2d2f356SSara Hartse * zfs_valid_proplist() to verify the properties.
70*b2d2f356SSara Hartse */
71*b2d2f356SSara Hartse if (!zfs_prop_user(prop_name)) {
72*b2d2f356SSara Hartse return (EINVAL);
73*b2d2f356SSara Hartse }
74*b2d2f356SSara Hartse
75*b2d2f356SSara Hartse nvl = fnvlist_alloc();
76*b2d2f356SSara Hartse fnvlist_add_string(nvl, args->prop, args->val);
77*b2d2f356SSara Hartse dpsa.dpsa_props = nvl;
78*b2d2f356SSara Hartse
79*b2d2f356SSara Hartse ret = dsl_props_set_check(&dpsa, tx);
80*b2d2f356SSara Hartse nvlist_free(nvl);
81*b2d2f356SSara Hartse
82*b2d2f356SSara Hartse return (ret);
83*b2d2f356SSara Hartse }
84*b2d2f356SSara Hartse
85*b2d2f356SSara Hartse void
zcp_set_prop_sync(void * arg,dmu_tx_t * tx)86*b2d2f356SSara Hartse zcp_set_prop_sync(void *arg, dmu_tx_t *tx)
87*b2d2f356SSara Hartse {
88*b2d2f356SSara Hartse zcp_set_prop_arg_t *args = arg;
89*b2d2f356SSara Hartse zcp_run_info_t *ri = zcp_run_info(args->state);
90*b2d2f356SSara Hartse dsl_pool_t *dp = ri->zri_pool;
91*b2d2f356SSara Hartse
92*b2d2f356SSara Hartse const char *dsname = args->dsname;
93*b2d2f356SSara Hartse const char *prop_name = args->prop;
94*b2d2f356SSara Hartse const char *prop_val = args->val;
95*b2d2f356SSara Hartse
96*b2d2f356SSara Hartse if (zfs_prop_user(prop_name)) {
97*b2d2f356SSara Hartse zcp_set_user_prop(args->state, dp, dsname, prop_name,
98*b2d2f356SSara Hartse prop_val, tx);
99*b2d2f356SSara Hartse }
100*b2d2f356SSara Hartse }
101