xref: /freebsd/sys/contrib/openzfs/module/zfs/zcp_get.c (revision a90b9d0159070121c221b966469c3e36d912bf82)
1 /*
2  * CDDL HEADER START
3  *
4  * This file and its contents are supplied under the terms of the
5  * Common Development and Distribution License ("CDDL"), version 1.0.
6  * You may only use this file in accordance with the terms of version
7  * 1.0 of the CDDL.
8  *
9  * A full copy of the text of the CDDL should have accompanied this
10  * source.  A copy of the CDDL is also available via the Internet at
11  * http://www.illumos.org/license/CDDL.
12  *
13  * CDDL HEADER END
14  */
15 
16 /*
17  * Copyright (c) 2016 by Delphix. All rights reserved.
18  */
19 
20 #include <sys/lua/lua.h>
21 #include <sys/lua/lualib.h>
22 #include <sys/lua/lauxlib.h>
23 
24 #include <zfs_prop.h>
25 
26 #include <sys/dsl_prop.h>
27 #include <sys/dsl_synctask.h>
28 #include <sys/dsl_dataset.h>
29 #include <sys/dsl_dir.h>
30 #include <sys/dmu_objset.h>
31 #include <sys/mntent.h>
32 #include <sys/sunddi.h>
33 #include <sys/zap.h>
34 #include <sys/zcp.h>
35 #include <sys/zcp_iter.h>
36 #include <sys/zcp_global.h>
37 #include <sys/zcp_prop.h>
38 #include <sys/zfs_ioctl.h>
39 #include <sys/zfs_znode.h>
40 #include <sys/zvol.h>
41 
42 #ifdef _KERNEL
43 #include <sys/zfs_quota.h>
44 #include <sys/zfs_vfsops.h>
45 #endif
46 
47 static int
48 get_objset_type(dsl_dataset_t *ds, zfs_type_t *type)
49 {
50 	int error;
51 	objset_t *os;
52 	error = dmu_objset_from_ds(ds, &os);
53 	if (error != 0)
54 		return (error);
55 	if (ds->ds_is_snapshot) {
56 		*type = ZFS_TYPE_SNAPSHOT;
57 	} else {
58 		switch (os->os_phys->os_type) {
59 		case DMU_OST_ZFS:
60 			*type = ZFS_TYPE_FILESYSTEM;
61 			break;
62 		case DMU_OST_ZVOL:
63 			*type = ZFS_TYPE_VOLUME;
64 			break;
65 		default:
66 			return (EINVAL);
67 		}
68 	}
69 	return (0);
70 }
71 
72 /*
73  * Returns the string name of ds's type in str (a buffer which should be
74  * at least 12 bytes long).
75  */
76 static int
77 get_objset_type_name(dsl_dataset_t *ds, char *str)
78 {
79 	zfs_type_t type = ZFS_TYPE_INVALID;
80 	int error = get_objset_type(ds, &type);
81 	if (error != 0)
82 		return (error);
83 	switch (type) {
84 	case ZFS_TYPE_SNAPSHOT:
85 		(void) strlcpy(str, "snapshot", ZAP_MAXVALUELEN);
86 		break;
87 	case ZFS_TYPE_FILESYSTEM:
88 		(void) strlcpy(str, "filesystem", ZAP_MAXVALUELEN);
89 		break;
90 	case ZFS_TYPE_VOLUME:
91 		(void) strlcpy(str, "volume", ZAP_MAXVALUELEN);
92 		break;
93 	default:
94 		return (EINVAL);
95 	}
96 	return (0);
97 }
98 
99 /*
100  * Determines the source of a property given its setpoint and
101  * property type. It pushes the source to the lua stack.
102  */
103 static void
104 get_prop_src(lua_State *state, const char *setpoint, zfs_prop_t prop)
105 {
106 	if (zfs_prop_readonly(prop) || (prop == ZFS_PROP_VERSION)) {
107 		lua_pushnil(state);
108 	} else {
109 		const char *src;
110 		if (strcmp("", setpoint) == 0) {
111 			src = "default";
112 		} else {
113 			src = setpoint;
114 		}
115 		(void) lua_pushstring(state, src);
116 	}
117 }
118 
119 /*
120  * Given an error encountered while getting properties, either longjmp's for
121  * a fatal error or pushes nothing to the stack for a non fatal one.
122  */
123 static int
124 zcp_handle_error(lua_State *state, const char *dataset_name,
125     const char *property_name, int error)
126 {
127 	ASSERT3S(error, !=, 0);
128 	if (error == ENOENT) {
129 		return (0);
130 	} else if (error == EINVAL) {
131 		return (luaL_error(state,
132 		    "property '%s' is not a valid property on dataset '%s'",
133 		    property_name, dataset_name));
134 	} else if (error == EIO) {
135 		return (luaL_error(state,
136 		    "I/O error while retrieving property '%s' on dataset '%s'",
137 		    property_name, dataset_name));
138 	} else {
139 		return (luaL_error(state, "unexpected error %d while "
140 		    "retrieving property '%s' on dataset '%s'",
141 		    error, property_name, dataset_name));
142 	}
143 }
144 
145 /*
146  * Look up a user defined property in the zap object. If it exists, push it
147  * and the setpoint onto the stack, otherwise don't push anything.
148  */
149 static int
150 zcp_get_user_prop(lua_State *state, dsl_pool_t *dp, const char *dataset_name,
151     const char *property_name)
152 {
153 	int error;
154 	char *buf;
155 	char setpoint[ZFS_MAX_DATASET_NAME_LEN];
156 	/*
157 	 * zcp_dataset_hold will either successfully return the requested
158 	 * dataset or throw a lua error and longjmp out of the zfs.get_prop call
159 	 * without returning.
160 	 */
161 	dsl_dataset_t *ds = zcp_dataset_hold(state, dp, dataset_name, FTAG);
162 	if (ds == NULL)
163 		return (1); /* not reached; zcp_dataset_hold() longjmp'd */
164 
165 	buf = kmem_alloc(ZAP_MAXVALUELEN, KM_SLEEP);
166 	error = dsl_prop_get_ds(ds, property_name, 1, ZAP_MAXVALUELEN,
167 	    buf, setpoint);
168 	dsl_dataset_rele(ds, FTAG);
169 
170 	if (error != 0) {
171 		kmem_free(buf, ZAP_MAXVALUELEN);
172 		return (zcp_handle_error(state, dataset_name, property_name,
173 		    error));
174 	}
175 	(void) lua_pushstring(state, buf);
176 	(void) lua_pushstring(state, setpoint);
177 	kmem_free(buf, ZAP_MAXVALUELEN);
178 	return (2);
179 }
180 
181 /*
182  * Check if the property we're looking for is stored in the ds_dir. If so,
183  * return it in the 'val' argument. Return 0 on success and ENOENT and if
184  * the property is not present.
185  */
186 static int
187 get_dsl_dir_prop(dsl_dataset_t *ds, zfs_prop_t zfs_prop,
188     uint64_t *val)
189 {
190 	dsl_dir_t *dd = ds->ds_dir;
191 	mutex_enter(&dd->dd_lock);
192 	switch (zfs_prop) {
193 	case ZFS_PROP_USEDSNAP:
194 		*val = dsl_dir_get_usedsnap(dd);
195 		break;
196 	case ZFS_PROP_USEDCHILD:
197 		*val = dsl_dir_get_usedchild(dd);
198 		break;
199 	case ZFS_PROP_USEDDS:
200 		*val = dsl_dir_get_usedds(dd);
201 		break;
202 	case ZFS_PROP_USEDREFRESERV:
203 		*val = dsl_dir_get_usedrefreserv(dd);
204 		break;
205 	case ZFS_PROP_LOGICALUSED:
206 		*val = dsl_dir_get_logicalused(dd);
207 		break;
208 	default:
209 		mutex_exit(&dd->dd_lock);
210 		return (SET_ERROR(ENOENT));
211 	}
212 	mutex_exit(&dd->dd_lock);
213 	return (0);
214 }
215 
216 /*
217  * Check if the property we're looking for is stored at the dsl_dataset or
218  * dsl_dir level. If so, push the property value and source onto the lua stack
219  * and return 0. If it is not present or a failure occurs in lookup, return a
220  * non-zero error value.
221  */
222 static int
223 get_special_prop(lua_State *state, dsl_dataset_t *ds, const char *dsname,
224     zfs_prop_t zfs_prop)
225 {
226 	int error = 0;
227 	objset_t *os;
228 	uint64_t numval = 0;
229 	char *strval = kmem_alloc(ZAP_MAXVALUELEN, KM_SLEEP);
230 	char setpoint[ZFS_MAX_DATASET_NAME_LEN] =
231 	    "Internal error - setpoint not determined";
232 	zfs_type_t ds_type = ZFS_TYPE_INVALID;
233 	zprop_type_t prop_type = zfs_prop_get_type(zfs_prop);
234 	(void) get_objset_type(ds, &ds_type);
235 
236 	switch (zfs_prop) {
237 	case ZFS_PROP_REFRATIO:
238 		numval = dsl_get_refratio(ds);
239 		break;
240 	case ZFS_PROP_USED:
241 		numval = dsl_get_used(ds);
242 		break;
243 	case ZFS_PROP_CLONES: {
244 		nvlist_t *clones = fnvlist_alloc();
245 		error = get_clones_stat_impl(ds, clones);
246 		if (error == 0) {
247 			/* push list to lua stack */
248 			VERIFY0(zcp_nvlist_to_lua(state, clones, NULL, 0ULL));
249 			/* source */
250 			(void) lua_pushnil(state);
251 		}
252 		nvlist_free(clones);
253 		kmem_free(strval, ZAP_MAXVALUELEN);
254 		return (error);
255 	}
256 	case ZFS_PROP_COMPRESSRATIO:
257 		numval = dsl_get_compressratio(ds);
258 		break;
259 	case ZFS_PROP_CREATION:
260 		numval = dsl_get_creation(ds);
261 		break;
262 	case ZFS_PROP_REFERENCED:
263 		numval = dsl_get_referenced(ds);
264 		break;
265 	case ZFS_PROP_AVAILABLE:
266 		numval = dsl_get_available(ds);
267 		break;
268 	case ZFS_PROP_LOGICALREFERENCED:
269 		numval = dsl_get_logicalreferenced(ds);
270 		break;
271 	case ZFS_PROP_CREATETXG:
272 		numval = dsl_get_creationtxg(ds);
273 		break;
274 	case ZFS_PROP_GUID:
275 		numval = dsl_get_guid(ds);
276 		break;
277 	case ZFS_PROP_UNIQUE:
278 		numval = dsl_get_unique(ds);
279 		break;
280 	case ZFS_PROP_OBJSETID:
281 		numval = dsl_get_objsetid(ds);
282 		break;
283 	case ZFS_PROP_ORIGIN:
284 		dsl_dir_get_origin(ds->ds_dir, strval);
285 		break;
286 	case ZFS_PROP_USERACCOUNTING:
287 		error = dmu_objset_from_ds(ds, &os);
288 		if (error == 0)
289 			numval = dmu_objset_userspace_present(os);
290 		break;
291 	case ZFS_PROP_WRITTEN:
292 		error = dsl_get_written(ds, &numval);
293 		break;
294 	case ZFS_PROP_TYPE:
295 		error = get_objset_type_name(ds, strval);
296 		break;
297 	case ZFS_PROP_PREV_SNAP:
298 		error = dsl_get_prev_snap(ds, strval);
299 		break;
300 	case ZFS_PROP_NAME:
301 		dsl_dataset_name(ds, strval);
302 		break;
303 	case ZFS_PROP_MOUNTPOINT:
304 		error = dsl_get_mountpoint(ds, dsname, strval, setpoint);
305 		break;
306 	case ZFS_PROP_VERSION:
307 		/* should be a snapshot or filesystem */
308 		ASSERT(ds_type != ZFS_TYPE_VOLUME);
309 		error = dmu_objset_from_ds(ds, &os);
310 		/* look in the master node for the version */
311 		if (error == 0) {
312 			error = zap_lookup(os, MASTER_NODE_OBJ, ZPL_VERSION_STR,
313 			    sizeof (numval), 1, &numval);
314 		}
315 		break;
316 	case ZFS_PROP_DEFER_DESTROY:
317 		numval = dsl_get_defer_destroy(ds);
318 		break;
319 	case ZFS_PROP_USERREFS:
320 		numval = dsl_get_userrefs(ds);
321 		break;
322 	case ZFS_PROP_FILESYSTEM_COUNT:
323 		error = dsl_dir_get_filesystem_count(ds->ds_dir, &numval);
324 		(void) strlcpy(setpoint, "", ZFS_MAX_DATASET_NAME_LEN);
325 		break;
326 	case ZFS_PROP_SNAPSHOT_COUNT:
327 		error = dsl_dir_get_snapshot_count(ds->ds_dir, &numval);
328 		(void) strlcpy(setpoint, "", ZFS_MAX_DATASET_NAME_LEN);
329 		break;
330 	case ZFS_PROP_NUMCLONES:
331 		numval = dsl_get_numclones(ds);
332 		break;
333 	case ZFS_PROP_INCONSISTENT:
334 		numval = dsl_get_inconsistent(ds);
335 		break;
336 	case ZFS_PROP_IVSET_GUID:
337 		if (dsl_dataset_is_zapified(ds)) {
338 			error = zap_lookup(ds->ds_dir->dd_pool->dp_meta_objset,
339 			    ds->ds_object, DS_FIELD_IVSET_GUID,
340 			    sizeof (numval), 1, &numval);
341 		} else {
342 			error = ENOENT;
343 		}
344 		break;
345 	case ZFS_PROP_RECEIVE_RESUME_TOKEN: {
346 		char *token = get_receive_resume_token(ds);
347 		if (token != NULL) {
348 			(void) strlcpy(strval, token, ZAP_MAXVALUELEN);
349 			kmem_strfree(token);
350 		} else {
351 			error = ENOENT;
352 		}
353 		break;
354 	}
355 	case ZFS_PROP_VOLSIZE:
356 		ASSERT(ds_type == ZFS_TYPE_VOLUME ||
357 		    ds_type == ZFS_TYPE_SNAPSHOT);
358 		error = dmu_objset_from_ds(ds, &os);
359 		if (error == 0) {
360 			error = zap_lookup(os, ZVOL_ZAP_OBJ, "size",
361 			    sizeof (numval), 1, &numval);
362 		}
363 		if (error == 0)
364 			(void) strlcpy(setpoint, dsname,
365 			    ZFS_MAX_DATASET_NAME_LEN);
366 
367 		break;
368 	case ZFS_PROP_VOLBLOCKSIZE: {
369 		ASSERT(ds_type == ZFS_TYPE_VOLUME);
370 		dmu_object_info_t doi;
371 		error = dmu_objset_from_ds(ds, &os);
372 		if (error == 0) {
373 			error = dmu_object_info(os, ZVOL_OBJ, &doi);
374 			if (error == 0)
375 				numval = doi.doi_data_block_size;
376 		}
377 		break;
378 	}
379 
380 	case ZFS_PROP_KEYSTATUS:
381 	case ZFS_PROP_KEYFORMAT: {
382 		/* provide defaults in case no crypto obj exists */
383 		setpoint[0] = '\0';
384 		if (zfs_prop == ZFS_PROP_KEYSTATUS)
385 			numval = ZFS_KEYSTATUS_NONE;
386 		else
387 			numval = ZFS_KEYFORMAT_NONE;
388 
389 		nvlist_t *nvl, *propval;
390 		nvl = fnvlist_alloc();
391 		dsl_dataset_crypt_stats(ds, nvl);
392 		if (nvlist_lookup_nvlist(nvl, zfs_prop_to_name(zfs_prop),
393 		    &propval) == 0) {
394 			const char *source;
395 
396 			(void) nvlist_lookup_uint64(propval, ZPROP_VALUE,
397 			    &numval);
398 			if (nvlist_lookup_string(propval, ZPROP_SOURCE,
399 			    &source) == 0)
400 				strlcpy(setpoint, source, sizeof (setpoint));
401 		}
402 		nvlist_free(nvl);
403 		break;
404 	}
405 
406 	case ZFS_PROP_SNAPSHOTS_CHANGED:
407 		numval = dsl_dir_snap_cmtime(ds->ds_dir).tv_sec;
408 		break;
409 
410 	default:
411 		/* Did not match these props, check in the dsl_dir */
412 		error = get_dsl_dir_prop(ds, zfs_prop, &numval);
413 	}
414 	if (error != 0) {
415 		kmem_free(strval, ZAP_MAXVALUELEN);
416 		return (error);
417 	}
418 
419 	switch (prop_type) {
420 	case PROP_TYPE_NUMBER: {
421 		(void) lua_pushnumber(state, numval);
422 		break;
423 	}
424 	case PROP_TYPE_STRING: {
425 		(void) lua_pushstring(state, strval);
426 		break;
427 	}
428 	case PROP_TYPE_INDEX: {
429 		const char *propval;
430 		error = zfs_prop_index_to_string(zfs_prop, numval, &propval);
431 		if (error != 0) {
432 			kmem_free(strval, ZAP_MAXVALUELEN);
433 			return (error);
434 		}
435 		(void) lua_pushstring(state, propval);
436 		break;
437 	}
438 	}
439 	kmem_free(strval, ZAP_MAXVALUELEN);
440 
441 	/* Push the source to the stack */
442 	get_prop_src(state, setpoint, zfs_prop);
443 	return (0);
444 }
445 
446 /*
447  * Look up a property and its source in the zap object. If the value is
448  * present and successfully retrieved, push the value and source on the
449  * lua stack and return 0. On failure, return a non-zero error value.
450  */
451 static int
452 get_zap_prop(lua_State *state, dsl_dataset_t *ds, zfs_prop_t zfs_prop)
453 {
454 	int error = 0;
455 	char setpoint[ZFS_MAX_DATASET_NAME_LEN];
456 	char *strval = kmem_alloc(ZAP_MAXVALUELEN, KM_SLEEP);
457 	uint64_t numval;
458 	const char *prop_name = zfs_prop_to_name(zfs_prop);
459 	zprop_type_t prop_type = zfs_prop_get_type(zfs_prop);
460 
461 	if (prop_type == PROP_TYPE_STRING) {
462 		/* Push value to lua stack */
463 		error = dsl_prop_get_ds(ds, prop_name, 1,
464 		    ZAP_MAXVALUELEN, strval, setpoint);
465 		if (error == 0)
466 			(void) lua_pushstring(state, strval);
467 	} else {
468 		error = dsl_prop_get_ds(ds, prop_name, sizeof (numval),
469 		    1, &numval, setpoint);
470 		if (error != 0)
471 			goto out;
472 #ifdef _KERNEL
473 		/* Fill in temporary value for prop, if applicable */
474 		(void) zfs_get_temporary_prop(ds, zfs_prop, &numval, setpoint);
475 #else
476 		kmem_free(strval, ZAP_MAXVALUELEN);
477 		return (luaL_error(state,
478 		    "temporary properties only supported in kernel mode",
479 		    prop_name));
480 #endif
481 		/* Push value to lua stack */
482 		if (prop_type == PROP_TYPE_INDEX) {
483 			const char *propval;
484 			error = zfs_prop_index_to_string(zfs_prop, numval,
485 			    &propval);
486 			if (error == 0)
487 				(void) lua_pushstring(state, propval);
488 		} else {
489 			if (error == 0)
490 				(void) lua_pushnumber(state, numval);
491 		}
492 	}
493 out:
494 	kmem_free(strval, ZAP_MAXVALUELEN);
495 	if (error == 0)
496 		get_prop_src(state, setpoint, zfs_prop);
497 	return (error);
498 }
499 
500 /*
501  * Determine whether property is valid for a given dataset
502  */
503 boolean_t
504 prop_valid_for_ds(dsl_dataset_t *ds, zfs_prop_t zfs_prop)
505 {
506 	zfs_type_t zfs_type = ZFS_TYPE_INVALID;
507 
508 	/* properties not supported */
509 	if ((zfs_prop == ZFS_PROP_ISCSIOPTIONS) ||
510 	    (zfs_prop == ZFS_PROP_MOUNTED))
511 		return (B_FALSE);
512 
513 	/* if we want the origin prop, ds must be a clone */
514 	if ((zfs_prop == ZFS_PROP_ORIGIN) && (!dsl_dir_is_clone(ds->ds_dir)))
515 		return (B_FALSE);
516 
517 	int error = get_objset_type(ds, &zfs_type);
518 	if (error != 0)
519 		return (B_FALSE);
520 	return (zfs_prop_valid_for_type(zfs_prop, zfs_type, B_FALSE));
521 }
522 
523 /*
524  * Look up a given dataset property. On success return 2, the number of
525  * values pushed to the lua stack (property value and source). On a fatal
526  * error, longjmp. On a non fatal error push nothing.
527  */
528 static int
529 zcp_get_system_prop(lua_State *state, dsl_pool_t *dp, const char *dataset_name,
530     zfs_prop_t zfs_prop)
531 {
532 	int error;
533 	/*
534 	 * zcp_dataset_hold will either successfully return the requested
535 	 * dataset or throw a lua error and longjmp out of the zfs.get_prop call
536 	 * without returning.
537 	 */
538 	dsl_dataset_t *ds = zcp_dataset_hold(state, dp, dataset_name, FTAG);
539 	if (ds == NULL)
540 		return (1); /* not reached; zcp_dataset_hold() longjmp'd */
541 
542 	/* Check that the property is valid for the given dataset */
543 	const char *prop_name = zfs_prop_to_name(zfs_prop);
544 	if (!prop_valid_for_ds(ds, zfs_prop)) {
545 		dsl_dataset_rele(ds, FTAG);
546 		return (0);
547 	}
548 
549 	/* Check if the property can be accessed directly */
550 	error = get_special_prop(state, ds, dataset_name, zfs_prop);
551 	if (error == 0) {
552 		dsl_dataset_rele(ds, FTAG);
553 		/* The value and source have been pushed by get_special_prop */
554 		return (2);
555 	}
556 	if (error != ENOENT) {
557 		dsl_dataset_rele(ds, FTAG);
558 		return (zcp_handle_error(state, dataset_name,
559 		    prop_name, error));
560 	}
561 
562 	/* If we were unable to find it, look in the zap object */
563 	error = get_zap_prop(state, ds, zfs_prop);
564 	dsl_dataset_rele(ds, FTAG);
565 	if (error != 0) {
566 		return (zcp_handle_error(state, dataset_name,
567 		    prop_name, error));
568 	}
569 	/* The value and source have been pushed by get_zap_prop */
570 	return (2);
571 }
572 
573 #ifdef _KERNEL
574 static zfs_userquota_prop_t
575 get_userquota_prop(const char *prop_name)
576 {
577 	zfs_userquota_prop_t type;
578 	/* Figure out the property type ({user|group}{quota|used}) */
579 	for (type = 0; type < ZFS_NUM_USERQUOTA_PROPS; type++) {
580 		if (strncmp(prop_name, zfs_userquota_prop_prefixes[type],
581 		    strlen(zfs_userquota_prop_prefixes[type])) == 0)
582 			break;
583 	}
584 	return (type);
585 }
586 
587 /*
588  * Given the name of a zfs_userquota_prop, this function determines the
589  * prop type as well as the numeric group/user ids based on the string
590  * following the '@' in the property name. On success, returns 0. On failure,
591  * returns a non-zero error.
592  * 'domain' must be free'd by caller using kmem_strfree()
593  */
594 static int
595 parse_userquota_prop(const char *prop_name, zfs_userquota_prop_t *type,
596     char **domain, uint64_t *rid)
597 {
598 	char *cp, *end, *domain_val;
599 
600 	*type = get_userquota_prop(prop_name);
601 	if (*type >= ZFS_NUM_USERQUOTA_PROPS)
602 		return (EINVAL);
603 
604 	*rid = 0;
605 	cp = strchr(prop_name, '@') + 1;
606 	if (strncmp(cp, "S-1-", 4) == 0) {
607 		/*
608 		 * It's a numeric SID (eg "S-1-234-567-89") and we want to
609 		 * separate the domain id and the rid
610 		 */
611 		int domain_len = strrchr(cp, '-') - cp;
612 		domain_val = kmem_alloc(domain_len + 1, KM_SLEEP);
613 		(void) strlcpy(domain_val, cp, domain_len + 1);
614 		cp += domain_len + 1;
615 
616 		(void) ddi_strtoll(cp, &end, 10, (longlong_t *)rid);
617 		if (*end != '\0') {
618 			kmem_strfree(domain_val);
619 			return (EINVAL);
620 		}
621 	} else {
622 		/* It's only a user/group ID (eg "12345"), just get the rid */
623 		domain_val = NULL;
624 		(void) ddi_strtoll(cp, &end, 10, (longlong_t *)rid);
625 		if (*end != '\0')
626 			return (EINVAL);
627 	}
628 	*domain = domain_val;
629 	return (0);
630 }
631 
632 /*
633  * Look up {user|group}{quota|used} property for given dataset. On success
634  * push the value (quota or used amount) and the setpoint. On failure, push
635  * a lua error.
636  */
637 static int
638 zcp_get_userquota_prop(lua_State *state, dsl_pool_t *dp,
639     const char *dataset_name, const char *prop_name)
640 {
641 	zfsvfs_t *zfvp;
642 	zfsvfs_t *zfsvfs;
643 	int error;
644 	zfs_userquota_prop_t type;
645 	char *domain;
646 	uint64_t rid, value = 0;
647 	objset_t *os;
648 
649 	dsl_dataset_t *ds = zcp_dataset_hold(state, dp, dataset_name, FTAG);
650 	if (ds == NULL)
651 		return (1); /* not reached; zcp_dataset_hold() longjmp'd */
652 
653 	error = parse_userquota_prop(prop_name, &type, &domain, &rid);
654 	if (error == 0) {
655 		error = dmu_objset_from_ds(ds, &os);
656 		if (error == 0) {
657 			zfsvfs = kmem_zalloc(sizeof (zfsvfs_t), KM_SLEEP);
658 			error = zfsvfs_create_impl(&zfvp, zfsvfs, os);
659 			if (error == 0) {
660 				error = zfs_userspace_one(zfvp, type, domain,
661 				    rid, &value);
662 				zfsvfs_free(zfvp);
663 			}
664 		}
665 		if (domain != NULL)
666 			kmem_strfree(domain);
667 	}
668 	dsl_dataset_rele(ds, FTAG);
669 
670 	if ((value == 0) && ((type == ZFS_PROP_USERQUOTA) ||
671 	    (type == ZFS_PROP_GROUPQUOTA)))
672 		error = SET_ERROR(ENOENT);
673 	if (error != 0) {
674 		return (zcp_handle_error(state, dataset_name,
675 		    prop_name, error));
676 	}
677 
678 	(void) lua_pushnumber(state, value);
679 	(void) lua_pushstring(state, dataset_name);
680 	return (2);
681 }
682 #endif
683 
684 /*
685  * Determines the name of the snapshot referenced in the written property
686  * name. Returns snapshot name in snap_name, a buffer that must be at least
687  * as large as ZFS_MAX_DATASET_NAME_LEN
688  */
689 static void
690 parse_written_prop(const char *dataset_name, const char *prop_name,
691     char *snap_name)
692 {
693 	ASSERT(zfs_prop_written(prop_name));
694 	const char *name = prop_name + ZFS_WRITTEN_PROP_PREFIX_LEN;
695 	if (strchr(name, '@') == NULL) {
696 		(void) snprintf(snap_name, ZFS_MAX_DATASET_NAME_LEN, "%s@%s",
697 		    dataset_name, name);
698 	} else {
699 		(void) strlcpy(snap_name, name, ZFS_MAX_DATASET_NAME_LEN);
700 	}
701 }
702 
703 /*
704  * Look up written@ property for given dataset. On success
705  * push the value and the setpoint. If error is fatal, we will
706  * longjmp, otherwise push nothing.
707  */
708 static int
709 zcp_get_written_prop(lua_State *state, dsl_pool_t *dp,
710     const char *dataset_name, const char *prop_name)
711 {
712 	char snap_name[ZFS_MAX_DATASET_NAME_LEN];
713 	uint64_t used, comp, uncomp;
714 	dsl_dataset_t *old;
715 	int error = 0;
716 
717 	parse_written_prop(dataset_name, prop_name, snap_name);
718 	dsl_dataset_t *new = zcp_dataset_hold(state, dp, dataset_name, FTAG);
719 	if (new == NULL)
720 		return (1); /* not reached; zcp_dataset_hold() longjmp'd */
721 
722 	error = dsl_dataset_hold(dp, snap_name, FTAG, &old);
723 	if (error != 0) {
724 		dsl_dataset_rele(new, FTAG);
725 		return (zcp_dataset_hold_error(state, dp, snap_name,
726 		    error));
727 	}
728 	error = dsl_dataset_space_written(old, new,
729 	    &used, &comp, &uncomp);
730 
731 	dsl_dataset_rele(old, FTAG);
732 	dsl_dataset_rele(new, FTAG);
733 
734 	if (error != 0) {
735 		return (zcp_handle_error(state, dataset_name,
736 		    snap_name, error));
737 	}
738 	(void) lua_pushnumber(state, used);
739 	(void) lua_pushstring(state, dataset_name);
740 	return (2);
741 }
742 
743 static int zcp_get_prop(lua_State *state);
744 static const zcp_lib_info_t zcp_get_prop_info = {
745 	.name = "get_prop",
746 	.func = zcp_get_prop,
747 	.pargs = {
748 	    { .za_name = "dataset", .za_lua_type = LUA_TSTRING },
749 	    { .za_name = "property", .za_lua_type =  LUA_TSTRING },
750 	    {NULL, 0}
751 	},
752 	.kwargs = {
753 	    {NULL, 0}
754 	}
755 };
756 
757 static int
758 zcp_get_prop(lua_State *state)
759 {
760 	const char *dataset_name;
761 	const char *property_name;
762 	dsl_pool_t *dp = zcp_run_info(state)->zri_pool;
763 	const zcp_lib_info_t *libinfo = &zcp_get_prop_info;
764 
765 	zcp_parse_args(state, libinfo->name, libinfo->pargs, libinfo->kwargs);
766 
767 	dataset_name = lua_tostring(state, 1);
768 	property_name = lua_tostring(state, 2);
769 
770 	/* User defined property */
771 	if (zfs_prop_user(property_name)) {
772 		return (zcp_get_user_prop(state, dp,
773 		    dataset_name, property_name));
774 	}
775 	/* userspace property */
776 	if (zfs_prop_userquota(property_name)) {
777 #ifdef _KERNEL
778 		return (zcp_get_userquota_prop(state, dp,
779 		    dataset_name, property_name));
780 #else
781 		return (luaL_error(state,
782 		    "user quota properties only supported in kernel mode",
783 		    property_name));
784 #endif
785 	}
786 	/* written@ property */
787 	if (zfs_prop_written(property_name)) {
788 		return (zcp_get_written_prop(state, dp,
789 		    dataset_name, property_name));
790 	}
791 
792 	zfs_prop_t zfs_prop = zfs_name_to_prop(property_name);
793 	/* Valid system property */
794 	if (zfs_prop != ZPROP_INVAL) {
795 		return (zcp_get_system_prop(state, dp, dataset_name,
796 		    zfs_prop));
797 	}
798 
799 	/* Invalid property name */
800 	return (luaL_error(state,
801 	    "'%s' is not a valid property", property_name));
802 }
803 
804 int
805 zcp_load_get_lib(lua_State *state)
806 {
807 	lua_pushcclosure(state, zcp_get_prop_info.func, 0);
808 	lua_setfield(state, -2, zcp_get_prop_info.name);
809 
810 	return (1);
811 }
812