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