]> git.proxmox.com Git - mirror_zfs.git/blame - module/zfs/dsl_dir.c
Illumos #4347 ZPL can use dmu_tx_assign(TXG_WAIT)
[mirror_zfs.git] / module / zfs / dsl_dir.c
CommitLineData
34dc7c2f
BB
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/*
428870ff 22 * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
2e528b49 23 * Copyright (c) 2013 by Delphix. All rights reserved.
b1118acb 24 * Copyright (c) 2013 Martin Matuska. All rights reserved.
34dc7c2f
BB
25 */
26
34dc7c2f
BB
27#include <sys/dmu.h>
28#include <sys/dmu_objset.h>
29#include <sys/dmu_tx.h>
30#include <sys/dsl_dataset.h>
31#include <sys/dsl_dir.h>
32#include <sys/dsl_prop.h>
33#include <sys/dsl_synctask.h>
34#include <sys/dsl_deleg.h>
35#include <sys/spa.h>
428870ff 36#include <sys/metaslab.h>
34dc7c2f
BB
37#include <sys/zap.h>
38#include <sys/zio.h>
39#include <sys/arc.h>
40#include <sys/sunddi.h>
41#include "zfs_namecheck.h"
42
43static uint64_t dsl_dir_space_towrite(dsl_dir_t *dd);
34dc7c2f
BB
44
45/* ARGSUSED */
46static void
47dsl_dir_evict(dmu_buf_t *db, void *arg)
48{
49 dsl_dir_t *dd = arg;
1fde1e37 50 ASSERTV(dsl_pool_t *dp = dd->dd_pool;)
34dc7c2f
BB
51 int t;
52
53 for (t = 0; t < TXG_SIZE; t++) {
54 ASSERT(!txg_list_member(&dp->dp_dirty_dirs, dd, t));
55 ASSERT(dd->dd_tempreserved[t] == 0);
56 ASSERT(dd->dd_space_towrite[t] == 0);
57 }
58
34dc7c2f 59 if (dd->dd_parent)
13fe0198 60 dsl_dir_rele(dd->dd_parent, dd);
34dc7c2f
BB
61
62 spa_close(dd->dd_pool->dp_spa, dd);
63
64 /*
428870ff
BB
65 * The props callback list should have been cleaned up by
66 * objset_evict().
34dc7c2f
BB
67 */
68 list_destroy(&dd->dd_prop_cbs);
69 mutex_destroy(&dd->dd_lock);
70 kmem_free(dd, sizeof (dsl_dir_t));
71}
72
73int
13fe0198 74dsl_dir_hold_obj(dsl_pool_t *dp, uint64_t ddobj,
34dc7c2f
BB
75 const char *tail, void *tag, dsl_dir_t **ddp)
76{
77 dmu_buf_t *dbuf;
78 dsl_dir_t *dd;
79 int err;
80
13fe0198 81 ASSERT(dsl_pool_config_held(dp));
34dc7c2f
BB
82
83 err = dmu_bonus_hold(dp->dp_meta_objset, ddobj, tag, &dbuf);
13fe0198 84 if (err != 0)
34dc7c2f
BB
85 return (err);
86 dd = dmu_buf_get_user(dbuf);
87#ifdef ZFS_DEBUG
88 {
89 dmu_object_info_t doi;
90 dmu_object_info_from_db(dbuf, &doi);
91 ASSERT3U(doi.doi_type, ==, DMU_OT_DSL_DIR);
b128c09f 92 ASSERT3U(doi.doi_bonus_size, >=, sizeof (dsl_dir_phys_t));
34dc7c2f
BB
93 }
94#endif
34dc7c2f
BB
95 if (dd == NULL) {
96 dsl_dir_t *winner;
34dc7c2f 97
b8d06fca 98 dd = kmem_zalloc(sizeof (dsl_dir_t), KM_PUSHPAGE);
34dc7c2f
BB
99 dd->dd_object = ddobj;
100 dd->dd_dbuf = dbuf;
101 dd->dd_pool = dp;
102 dd->dd_phys = dbuf->db_data;
34dc7c2f
BB
103 mutex_init(&dd->dd_lock, NULL, MUTEX_DEFAULT, NULL);
104
105 list_create(&dd->dd_prop_cbs, sizeof (dsl_prop_cb_record_t),
106 offsetof(dsl_prop_cb_record_t, cbr_node));
107
428870ff
BB
108 dsl_dir_snap_cmtime_update(dd);
109
34dc7c2f 110 if (dd->dd_phys->dd_parent_obj) {
13fe0198 111 err = dsl_dir_hold_obj(dp, dd->dd_phys->dd_parent_obj,
34dc7c2f 112 NULL, dd, &dd->dd_parent);
13fe0198 113 if (err != 0)
b128c09f 114 goto errout;
34dc7c2f
BB
115 if (tail) {
116#ifdef ZFS_DEBUG
117 uint64_t foundobj;
118
119 err = zap_lookup(dp->dp_meta_objset,
120 dd->dd_parent->dd_phys->dd_child_dir_zapobj,
121 tail, sizeof (foundobj), 1, &foundobj);
122 ASSERT(err || foundobj == ddobj);
123#endif
124 (void) strcpy(dd->dd_myname, tail);
125 } else {
126 err = zap_value_search(dp->dp_meta_objset,
127 dd->dd_parent->dd_phys->dd_child_dir_zapobj,
128 ddobj, 0, dd->dd_myname);
129 }
13fe0198 130 if (err != 0)
b128c09f 131 goto errout;
34dc7c2f
BB
132 } else {
133 (void) strcpy(dd->dd_myname, spa_name(dp->dp_spa));
134 }
135
428870ff
BB
136 if (dsl_dir_is_clone(dd)) {
137 dmu_buf_t *origin_bonus;
138 dsl_dataset_phys_t *origin_phys;
139
140 /*
141 * We can't open the origin dataset, because
142 * that would require opening this dsl_dir.
143 * Just look at its phys directly instead.
144 */
145 err = dmu_bonus_hold(dp->dp_meta_objset,
146 dd->dd_phys->dd_origin_obj, FTAG, &origin_bonus);
13fe0198 147 if (err != 0)
428870ff
BB
148 goto errout;
149 origin_phys = origin_bonus->db_data;
150 dd->dd_origin_txg =
151 origin_phys->ds_creation_txg;
152 dmu_buf_rele(origin_bonus, FTAG);
153 }
154
34dc7c2f
BB
155 winner = dmu_buf_set_user_ie(dbuf, dd, &dd->dd_phys,
156 dsl_dir_evict);
157 if (winner) {
158 if (dd->dd_parent)
13fe0198 159 dsl_dir_rele(dd->dd_parent, dd);
34dc7c2f
BB
160 mutex_destroy(&dd->dd_lock);
161 kmem_free(dd, sizeof (dsl_dir_t));
162 dd = winner;
163 } else {
164 spa_open_ref(dp->dp_spa, dd);
165 }
166 }
167
168 /*
169 * The dsl_dir_t has both open-to-close and instantiate-to-evict
170 * holds on the spa. We need the open-to-close holds because
171 * otherwise the spa_refcnt wouldn't change when we open a
172 * dir which the spa also has open, so we could incorrectly
173 * think it was OK to unload/export/destroy the pool. We need
174 * the instantiate-to-evict hold because the dsl_dir_t has a
175 * pointer to the dd_pool, which has a pointer to the spa_t.
176 */
177 spa_open_ref(dp->dp_spa, tag);
178 ASSERT3P(dd->dd_pool, ==, dp);
179 ASSERT3U(dd->dd_object, ==, ddobj);
180 ASSERT3P(dd->dd_dbuf, ==, dbuf);
181 *ddp = dd;
182 return (0);
b128c09f
BB
183
184errout:
185 if (dd->dd_parent)
13fe0198 186 dsl_dir_rele(dd->dd_parent, dd);
b128c09f
BB
187 mutex_destroy(&dd->dd_lock);
188 kmem_free(dd, sizeof (dsl_dir_t));
189 dmu_buf_rele(dbuf, tag);
190 return (err);
34dc7c2f
BB
191}
192
193void
13fe0198 194dsl_dir_rele(dsl_dir_t *dd, void *tag)
34dc7c2f
BB
195{
196 dprintf_dd(dd, "%s\n", "");
197 spa_close(dd->dd_pool->dp_spa, tag);
198 dmu_buf_rele(dd->dd_dbuf, tag);
199}
200
201/* buf must be long enough (MAXNAMELEN + strlen(MOS_DIR_NAME) + 1 should do) */
202void
203dsl_dir_name(dsl_dir_t *dd, char *buf)
204{
205 if (dd->dd_parent) {
206 dsl_dir_name(dd->dd_parent, buf);
207 (void) strcat(buf, "/");
208 } else {
209 buf[0] = '\0';
210 }
211 if (!MUTEX_HELD(&dd->dd_lock)) {
212 /*
213 * recursive mutex so that we can use
214 * dprintf_dd() with dd_lock held
215 */
216 mutex_enter(&dd->dd_lock);
217 (void) strcat(buf, dd->dd_myname);
218 mutex_exit(&dd->dd_lock);
219 } else {
220 (void) strcat(buf, dd->dd_myname);
221 }
222}
223
29809a6c 224/* Calculate name length, avoiding all the strcat calls of dsl_dir_name */
34dc7c2f
BB
225int
226dsl_dir_namelen(dsl_dir_t *dd)
227{
228 int result = 0;
229
230 if (dd->dd_parent) {
231 /* parent's name + 1 for the "/" */
232 result = dsl_dir_namelen(dd->dd_parent) + 1;
233 }
234
235 if (!MUTEX_HELD(&dd->dd_lock)) {
236 /* see dsl_dir_name */
237 mutex_enter(&dd->dd_lock);
238 result += strlen(dd->dd_myname);
239 mutex_exit(&dd->dd_lock);
240 } else {
241 result += strlen(dd->dd_myname);
242 }
243
244 return (result);
245}
246
34dc7c2f
BB
247static int
248getcomponent(const char *path, char *component, const char **nextp)
249{
250 char *p;
13fe0198 251
9babb374 252 if ((path == NULL) || (path[0] == '\0'))
2e528b49 253 return (SET_ERROR(ENOENT));
34dc7c2f
BB
254 /* This would be a good place to reserve some namespace... */
255 p = strpbrk(path, "/@");
256 if (p && (p[1] == '/' || p[1] == '@')) {
257 /* two separators in a row */
2e528b49 258 return (SET_ERROR(EINVAL));
34dc7c2f
BB
259 }
260 if (p == NULL || p == path) {
261 /*
262 * if the first thing is an @ or /, it had better be an
263 * @ and it had better not have any more ats or slashes,
264 * and it had better have something after the @.
265 */
266 if (p != NULL &&
267 (p[0] != '@' || strpbrk(path+1, "/@") || p[1] == '\0'))
2e528b49 268 return (SET_ERROR(EINVAL));
34dc7c2f 269 if (strlen(path) >= MAXNAMELEN)
2e528b49 270 return (SET_ERROR(ENAMETOOLONG));
34dc7c2f
BB
271 (void) strcpy(component, path);
272 p = NULL;
273 } else if (p[0] == '/') {
13fe0198 274 if (p - path >= MAXNAMELEN)
2e528b49 275 return (SET_ERROR(ENAMETOOLONG));
34dc7c2f 276 (void) strncpy(component, path, p - path);
13fe0198 277 component[p - path] = '\0';
34dc7c2f
BB
278 p++;
279 } else if (p[0] == '@') {
280 /*
281 * if the next separator is an @, there better not be
282 * any more slashes.
283 */
284 if (strchr(path, '/'))
2e528b49 285 return (SET_ERROR(EINVAL));
13fe0198 286 if (p - path >= MAXNAMELEN)
2e528b49 287 return (SET_ERROR(ENAMETOOLONG));
34dc7c2f 288 (void) strncpy(component, path, p - path);
13fe0198 289 component[p - path] = '\0';
34dc7c2f 290 } else {
13fe0198 291 panic("invalid p=%p", (void *)p);
34dc7c2f
BB
292 }
293 *nextp = p;
294 return (0);
295}
296
297/*
13fe0198
MA
298 * Return the dsl_dir_t, and possibly the last component which couldn't
299 * be found in *tail. The name must be in the specified dsl_pool_t. This
300 * thread must hold the dp_config_rwlock for the pool. Returns NULL if the
301 * path is bogus, or if tail==NULL and we couldn't parse the whole name.
302 * (*tail)[0] == '@' means that the last component is a snapshot.
34dc7c2f
BB
303 */
304int
13fe0198 305dsl_dir_hold(dsl_pool_t *dp, const char *name, void *tag,
34dc7c2f
BB
306 dsl_dir_t **ddp, const char **tailp)
307{
fcf37ec6 308 char *buf;
13fe0198 309 const char *spaname, *next, *nextnext = NULL;
34dc7c2f
BB
310 int err;
311 dsl_dir_t *dd;
34dc7c2f 312 uint64_t ddobj;
34dc7c2f 313
ed2e1576 314 buf = kmem_alloc(MAXNAMELEN, KM_PUSHPAGE);
34dc7c2f 315 err = getcomponent(name, buf, &next);
13fe0198 316 if (err != 0)
fcf37ec6 317 goto error;
34dc7c2f 318
13fe0198
MA
319 /* Make sure the name is in the specified pool. */
320 spaname = spa_name(dp->dp_spa);
321 if (strcmp(buf, spaname) != 0) {
2e528b49 322 err = SET_ERROR(EINVAL);
13fe0198 323 goto error;
34dc7c2f
BB
324 }
325
13fe0198 326 ASSERT(dsl_pool_config_held(dp));
34dc7c2f 327
13fe0198
MA
328 err = dsl_dir_hold_obj(dp, dp->dp_root_dir_obj, NULL, tag, &dd);
329 if (err != 0) {
fcf37ec6 330 goto error;
34dc7c2f
BB
331 }
332
333 while (next != NULL) {
334 dsl_dir_t *child_ds;
335 err = getcomponent(next, buf, &nextnext);
13fe0198 336 if (err != 0)
34dc7c2f
BB
337 break;
338 ASSERT(next[0] != '\0');
339 if (next[0] == '@')
340 break;
341 dprintf("looking up %s in obj%lld\n",
342 buf, dd->dd_phys->dd_child_dir_zapobj);
343
344 err = zap_lookup(dp->dp_meta_objset,
345 dd->dd_phys->dd_child_dir_zapobj,
346 buf, sizeof (ddobj), 1, &ddobj);
13fe0198 347 if (err != 0) {
34dc7c2f
BB
348 if (err == ENOENT)
349 err = 0;
350 break;
351 }
352
13fe0198
MA
353 err = dsl_dir_hold_obj(dp, ddobj, buf, tag, &child_ds);
354 if (err != 0)
34dc7c2f 355 break;
13fe0198 356 dsl_dir_rele(dd, tag);
34dc7c2f
BB
357 dd = child_ds;
358 next = nextnext;
359 }
34dc7c2f 360
13fe0198
MA
361 if (err != 0) {
362 dsl_dir_rele(dd, tag);
fcf37ec6 363 goto error;
34dc7c2f
BB
364 }
365
366 /*
367 * It's an error if there's more than one component left, or
368 * tailp==NULL and there's any component left.
369 */
370 if (next != NULL &&
371 (tailp == NULL || (nextnext && nextnext[0] != '\0'))) {
372 /* bad path name */
13fe0198 373 dsl_dir_rele(dd, tag);
34dc7c2f 374 dprintf("next=%p (%s) tail=%p\n", next, next?next:"", tailp);
2e528b49 375 err = SET_ERROR(ENOENT);
34dc7c2f 376 }
13fe0198 377 if (tailp != NULL)
34dc7c2f 378 *tailp = next;
34dc7c2f 379 *ddp = dd;
fcf37ec6
BB
380error:
381 kmem_free(buf, MAXNAMELEN);
34dc7c2f
BB
382 return (err);
383}
384
34dc7c2f 385uint64_t
b128c09f
BB
386dsl_dir_create_sync(dsl_pool_t *dp, dsl_dir_t *pds, const char *name,
387 dmu_tx_t *tx)
34dc7c2f 388{
b128c09f 389 objset_t *mos = dp->dp_meta_objset;
34dc7c2f 390 uint64_t ddobj;
428870ff 391 dsl_dir_phys_t *ddphys;
34dc7c2f
BB
392 dmu_buf_t *dbuf;
393
394 ddobj = dmu_object_alloc(mos, DMU_OT_DSL_DIR, 0,
395 DMU_OT_DSL_DIR, sizeof (dsl_dir_phys_t), tx);
b128c09f
BB
396 if (pds) {
397 VERIFY(0 == zap_add(mos, pds->dd_phys->dd_child_dir_zapobj,
398 name, sizeof (uint64_t), 1, &ddobj, tx));
399 } else {
400 /* it's the root dir */
401 VERIFY(0 == zap_add(mos, DMU_POOL_DIRECTORY_OBJECT,
402 DMU_POOL_ROOT_DATASET, sizeof (uint64_t), 1, &ddobj, tx));
403 }
34dc7c2f
BB
404 VERIFY(0 == dmu_bonus_hold(mos, ddobj, FTAG, &dbuf));
405 dmu_buf_will_dirty(dbuf, tx);
428870ff 406 ddphys = dbuf->db_data;
34dc7c2f 407
428870ff 408 ddphys->dd_creation_time = gethrestime_sec();
b128c09f 409 if (pds)
428870ff
BB
410 ddphys->dd_parent_obj = pds->dd_object;
411 ddphys->dd_props_zapobj = zap_create(mos,
34dc7c2f 412 DMU_OT_DSL_PROPS, DMU_OT_NONE, 0, tx);
428870ff 413 ddphys->dd_child_dir_zapobj = zap_create(mos,
34dc7c2f 414 DMU_OT_DSL_DIR_CHILD_MAP, DMU_OT_NONE, 0, tx);
b128c09f 415 if (spa_version(dp->dp_spa) >= SPA_VERSION_USED_BREAKDOWN)
428870ff 416 ddphys->dd_flags |= DD_FLAG_USED_BREAKDOWN;
34dc7c2f
BB
417 dmu_buf_rele(dbuf, FTAG);
418
419 return (ddobj);
420}
421
b128c09f
BB
422boolean_t
423dsl_dir_is_clone(dsl_dir_t *dd)
34dc7c2f 424{
b128c09f
BB
425 return (dd->dd_phys->dd_origin_obj &&
426 (dd->dd_pool->dp_origin_snap == NULL ||
427 dd->dd_phys->dd_origin_obj !=
428 dd->dd_pool->dp_origin_snap->ds_object));
34dc7c2f
BB
429}
430
431void
432dsl_dir_stats(dsl_dir_t *dd, nvlist_t *nv)
433{
434 mutex_enter(&dd->dd_lock);
b128c09f
BB
435 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_USED,
436 dd->dd_phys->dd_used_bytes);
34dc7c2f
BB
437 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_QUOTA, dd->dd_phys->dd_quota);
438 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_RESERVATION,
439 dd->dd_phys->dd_reserved);
440 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_COMPRESSRATIO,
441 dd->dd_phys->dd_compressed_bytes == 0 ? 100 :
442 (dd->dd_phys->dd_uncompressed_bytes * 100 /
443 dd->dd_phys->dd_compressed_bytes));
24a64651
MA
444 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_LOGICALUSED,
445 dd->dd_phys->dd_uncompressed_bytes);
b128c09f
BB
446 if (dd->dd_phys->dd_flags & DD_FLAG_USED_BREAKDOWN) {
447 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_USEDSNAP,
448 dd->dd_phys->dd_used_breakdown[DD_USED_SNAP]);
449 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_USEDDS,
450 dd->dd_phys->dd_used_breakdown[DD_USED_HEAD]);
451 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_USEDREFRESERV,
452 dd->dd_phys->dd_used_breakdown[DD_USED_REFRSRV]);
453 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_USEDCHILD,
454 dd->dd_phys->dd_used_breakdown[DD_USED_CHILD] +
455 dd->dd_phys->dd_used_breakdown[DD_USED_CHILD_RSRV]);
456 }
34dc7c2f
BB
457 mutex_exit(&dd->dd_lock);
458
b128c09f 459 if (dsl_dir_is_clone(dd)) {
34dc7c2f
BB
460 dsl_dataset_t *ds;
461 char buf[MAXNAMELEN];
462
13fe0198 463 VERIFY0(dsl_dataset_hold_obj(dd->dd_pool,
b128c09f 464 dd->dd_phys->dd_origin_obj, FTAG, &ds));
34dc7c2f 465 dsl_dataset_name(ds, buf);
b128c09f 466 dsl_dataset_rele(ds, FTAG);
34dc7c2f
BB
467 dsl_prop_nvlist_add_string(nv, ZFS_PROP_ORIGIN, buf);
468 }
34dc7c2f
BB
469}
470
471void
472dsl_dir_dirty(dsl_dir_t *dd, dmu_tx_t *tx)
473{
474 dsl_pool_t *dp = dd->dd_pool;
475
476 ASSERT(dd->dd_phys);
477
13fe0198 478 if (txg_list_add(&dp->dp_dirty_dirs, dd, tx->tx_txg)) {
34dc7c2f
BB
479 /* up the hold count until we can be written out */
480 dmu_buf_add_ref(dd->dd_dbuf, dd);
481 }
482}
483
484static int64_t
485parent_delta(dsl_dir_t *dd, uint64_t used, int64_t delta)
486{
487 uint64_t old_accounted = MAX(used, dd->dd_phys->dd_reserved);
488 uint64_t new_accounted = MAX(used + delta, dd->dd_phys->dd_reserved);
489 return (new_accounted - old_accounted);
490}
491
492void
493dsl_dir_sync(dsl_dir_t *dd, dmu_tx_t *tx)
494{
495 ASSERT(dmu_tx_is_syncing(tx));
496
34dc7c2f 497 mutex_enter(&dd->dd_lock);
c99c9001 498 ASSERT0(dd->dd_tempreserved[tx->tx_txg&TXG_MASK]);
34dc7c2f
BB
499 dprintf_dd(dd, "txg=%llu towrite=%lluK\n", tx->tx_txg,
500 dd->dd_space_towrite[tx->tx_txg&TXG_MASK] / 1024);
501 dd->dd_space_towrite[tx->tx_txg&TXG_MASK] = 0;
34dc7c2f
BB
502 mutex_exit(&dd->dd_lock);
503
504 /* release the hold from dsl_dir_dirty */
505 dmu_buf_rele(dd->dd_dbuf, dd);
506}
507
508static uint64_t
509dsl_dir_space_towrite(dsl_dir_t *dd)
510{
511 uint64_t space = 0;
512 int i;
513
514 ASSERT(MUTEX_HELD(&dd->dd_lock));
515
516 for (i = 0; i < TXG_SIZE; i++) {
517 space += dd->dd_space_towrite[i&TXG_MASK];
518 ASSERT3U(dd->dd_space_towrite[i&TXG_MASK], >=, 0);
519 }
520 return (space);
521}
522
523/*
524 * How much space would dd have available if ancestor had delta applied
525 * to it? If ondiskonly is set, we're only interested in what's
526 * on-disk, not estimated pending changes.
527 */
528uint64_t
529dsl_dir_space_available(dsl_dir_t *dd,
530 dsl_dir_t *ancestor, int64_t delta, int ondiskonly)
531{
532 uint64_t parentspace, myspace, quota, used;
533
534 /*
535 * If there are no restrictions otherwise, assume we have
536 * unlimited space available.
537 */
538 quota = UINT64_MAX;
539 parentspace = UINT64_MAX;
540
541 if (dd->dd_parent != NULL) {
542 parentspace = dsl_dir_space_available(dd->dd_parent,
543 ancestor, delta, ondiskonly);
544 }
545
546 mutex_enter(&dd->dd_lock);
547 if (dd->dd_phys->dd_quota != 0)
548 quota = dd->dd_phys->dd_quota;
b128c09f 549 used = dd->dd_phys->dd_used_bytes;
34dc7c2f
BB
550 if (!ondiskonly)
551 used += dsl_dir_space_towrite(dd);
34dc7c2f
BB
552
553 if (dd->dd_parent == NULL) {
554 uint64_t poolsize = dsl_pool_adjustedsize(dd->dd_pool, FALSE);
555 quota = MIN(quota, poolsize);
556 }
557
558 if (dd->dd_phys->dd_reserved > used && parentspace != UINT64_MAX) {
559 /*
560 * We have some space reserved, in addition to what our
561 * parent gave us.
562 */
563 parentspace += dd->dd_phys->dd_reserved - used;
564 }
565
b128c09f
BB
566 if (dd == ancestor) {
567 ASSERT(delta <= 0);
568 ASSERT(used >= -delta);
569 used += delta;
570 if (parentspace != UINT64_MAX)
571 parentspace -= delta;
572 }
573
34dc7c2f
BB
574 if (used > quota) {
575 /* over quota */
576 myspace = 0;
34dc7c2f
BB
577 } else {
578 /*
579 * the lesser of the space provided by our parent and
580 * the space left in our quota
581 */
582 myspace = MIN(parentspace, quota - used);
583 }
584
585 mutex_exit(&dd->dd_lock);
586
587 return (myspace);
588}
589
590struct tempreserve {
591 list_node_t tr_node;
592 dsl_pool_t *tr_dp;
593 dsl_dir_t *tr_ds;
594 uint64_t tr_size;
595};
596
597static int
598dsl_dir_tempreserve_impl(dsl_dir_t *dd, uint64_t asize, boolean_t netfree,
599 boolean_t ignorequota, boolean_t checkrefquota, list_t *tr_list,
600 dmu_tx_t *tx, boolean_t first)
601{
602 uint64_t txg = tx->tx_txg;
603 uint64_t est_inflight, used_on_disk, quota, parent_rsrv;
428870ff 604 uint64_t deferred = 0;
34dc7c2f 605 struct tempreserve *tr;
428870ff 606 int retval = EDQUOT;
34dc7c2f
BB
607 int txgidx = txg & TXG_MASK;
608 int i;
609 uint64_t ref_rsrv = 0;
610
611 ASSERT3U(txg, !=, 0);
612 ASSERT3S(asize, >, 0);
613
614 mutex_enter(&dd->dd_lock);
615
616 /*
617 * Check against the dsl_dir's quota. We don't add in the delta
618 * when checking for over-quota because they get one free hit.
619 */
620 est_inflight = dsl_dir_space_towrite(dd);
621 for (i = 0; i < TXG_SIZE; i++)
622 est_inflight += dd->dd_tempreserved[i];
b128c09f 623 used_on_disk = dd->dd_phys->dd_used_bytes;
34dc7c2f
BB
624
625 /*
626 * On the first iteration, fetch the dataset's used-on-disk and
627 * refreservation values. Also, if checkrefquota is set, test if
628 * allocating this space would exceed the dataset's refquota.
629 */
630 if (first && tx->tx_objset) {
631 int error;
428870ff 632 dsl_dataset_t *ds = tx->tx_objset->os_dsl_dataset;
34dc7c2f
BB
633
634 error = dsl_dataset_check_quota(ds, checkrefquota,
635 asize, est_inflight, &used_on_disk, &ref_rsrv);
636 if (error) {
637 mutex_exit(&dd->dd_lock);
638 return (error);
639 }
640 }
641
642 /*
643 * If this transaction will result in a net free of space,
644 * we want to let it through.
645 */
646 if (ignorequota || netfree || dd->dd_phys->dd_quota == 0)
647 quota = UINT64_MAX;
648 else
649 quota = dd->dd_phys->dd_quota;
650
651 /*
428870ff
BB
652 * Adjust the quota against the actual pool size at the root
653 * minus any outstanding deferred frees.
34dc7c2f
BB
654 * To ensure that it's possible to remove files from a full
655 * pool without inducing transient overcommits, we throttle
656 * netfree transactions against a quota that is slightly larger,
657 * but still within the pool's allocation slop. In cases where
658 * we're very close to full, this will allow a steady trickle of
659 * removes to get through.
660 */
661 if (dd->dd_parent == NULL) {
428870ff 662 spa_t *spa = dd->dd_pool->dp_spa;
34dc7c2f 663 uint64_t poolsize = dsl_pool_adjustedsize(dd->dd_pool, netfree);
428870ff
BB
664 deferred = metaslab_class_get_deferred(spa_normal_class(spa));
665 if (poolsize - deferred < quota) {
666 quota = poolsize - deferred;
667 retval = ENOSPC;
34dc7c2f
BB
668 }
669 }
670
671 /*
672 * If they are requesting more space, and our current estimate
673 * is over quota, they get to try again unless the actual
674 * on-disk is over quota and there are no pending changes (which
675 * may free up space for us).
676 */
428870ff
BB
677 if (used_on_disk + est_inflight >= quota) {
678 if (est_inflight > 0 || used_on_disk < quota ||
679 (retval == ENOSPC && used_on_disk < quota + deferred))
680 retval = ERESTART;
34dc7c2f
BB
681 dprintf_dd(dd, "failing: used=%lluK inflight = %lluK "
682 "quota=%lluK tr=%lluK err=%d\n",
683 used_on_disk>>10, est_inflight>>10,
428870ff 684 quota>>10, asize>>10, retval);
34dc7c2f 685 mutex_exit(&dd->dd_lock);
2e528b49 686 return (SET_ERROR(retval));
34dc7c2f
BB
687 }
688
689 /* We need to up our estimated delta before dropping dd_lock */
690 dd->dd_tempreserved[txgidx] += asize;
691
692 parent_rsrv = parent_delta(dd, used_on_disk + est_inflight,
693 asize - ref_rsrv);
694 mutex_exit(&dd->dd_lock);
695
b8d06fca 696 tr = kmem_zalloc(sizeof (struct tempreserve), KM_PUSHPAGE);
34dc7c2f
BB
697 tr->tr_ds = dd;
698 tr->tr_size = asize;
699 list_insert_tail(tr_list, tr);
700
701 /* see if it's OK with our parent */
702 if (dd->dd_parent && parent_rsrv) {
703 boolean_t ismos = (dd->dd_phys->dd_head_dataset_obj == 0);
704
705 return (dsl_dir_tempreserve_impl(dd->dd_parent,
706 parent_rsrv, netfree, ismos, TRUE, tr_list, tx, FALSE));
707 } else {
708 return (0);
709 }
710}
711
712/*
713 * Reserve space in this dsl_dir, to be used in this tx's txg.
714 * After the space has been dirtied (and dsl_dir_willuse_space()
715 * has been called), the reservation should be canceled, using
716 * dsl_dir_tempreserve_clear().
717 */
718int
719dsl_dir_tempreserve_space(dsl_dir_t *dd, uint64_t lsize, uint64_t asize,
720 uint64_t fsize, uint64_t usize, void **tr_cookiep, dmu_tx_t *tx)
721{
722 int err;
723 list_t *tr_list;
724
725 if (asize == 0) {
726 *tr_cookiep = NULL;
727 return (0);
728 }
729
b8d06fca 730 tr_list = kmem_alloc(sizeof (list_t), KM_PUSHPAGE);
34dc7c2f
BB
731 list_create(tr_list, sizeof (struct tempreserve),
732 offsetof(struct tempreserve, tr_node));
733 ASSERT3S(asize, >, 0);
734 ASSERT3S(fsize, >=, 0);
735
736 err = arc_tempreserve_space(lsize, tx->tx_txg);
737 if (err == 0) {
738 struct tempreserve *tr;
739
b8d06fca 740 tr = kmem_zalloc(sizeof (struct tempreserve), KM_PUSHPAGE);
34dc7c2f
BB
741 tr->tr_size = lsize;
742 list_insert_tail(tr_list, tr);
743
744 err = dsl_pool_tempreserve_space(dd->dd_pool, asize, tx);
745 } else {
746 if (err == EAGAIN) {
63fd3c6c
AL
747 txg_delay(dd->dd_pool, tx->tx_txg,
748 MSEC2NSEC(10), MSEC2NSEC(10));
2e528b49 749 err = SET_ERROR(ERESTART);
34dc7c2f
BB
750 }
751 dsl_pool_memory_pressure(dd->dd_pool);
752 }
753
754 if (err == 0) {
755 struct tempreserve *tr;
756
b8d06fca 757 tr = kmem_zalloc(sizeof (struct tempreserve), KM_PUSHPAGE);
34dc7c2f
BB
758 tr->tr_dp = dd->dd_pool;
759 tr->tr_size = asize;
760 list_insert_tail(tr_list, tr);
761
762 err = dsl_dir_tempreserve_impl(dd, asize, fsize >= asize,
763 FALSE, asize > usize, tr_list, tx, TRUE);
764 }
765
13fe0198 766 if (err != 0)
34dc7c2f
BB
767 dsl_dir_tempreserve_clear(tr_list, tx);
768 else
769 *tr_cookiep = tr_list;
770
771 return (err);
772}
773
774/*
775 * Clear a temporary reservation that we previously made with
776 * dsl_dir_tempreserve_space().
777 */
778void
779dsl_dir_tempreserve_clear(void *tr_cookie, dmu_tx_t *tx)
780{
781 int txgidx = tx->tx_txg & TXG_MASK;
782 list_t *tr_list = tr_cookie;
783 struct tempreserve *tr;
784
785 ASSERT3U(tx->tx_txg, !=, 0);
786
787 if (tr_cookie == NULL)
788 return;
789
c65aa5b2 790 while ((tr = list_head(tr_list))) {
34dc7c2f
BB
791 if (tr->tr_dp) {
792 dsl_pool_tempreserve_clear(tr->tr_dp, tr->tr_size, tx);
793 } else if (tr->tr_ds) {
794 mutex_enter(&tr->tr_ds->dd_lock);
795 ASSERT3U(tr->tr_ds->dd_tempreserved[txgidx], >=,
796 tr->tr_size);
797 tr->tr_ds->dd_tempreserved[txgidx] -= tr->tr_size;
798 mutex_exit(&tr->tr_ds->dd_lock);
799 } else {
800 arc_tempreserve_clear(tr->tr_size);
801 }
802 list_remove(tr_list, tr);
803 kmem_free(tr, sizeof (struct tempreserve));
804 }
805
806 kmem_free(tr_list, sizeof (list_t));
807}
808
809static void
810dsl_dir_willuse_space_impl(dsl_dir_t *dd, int64_t space, dmu_tx_t *tx)
811{
812 int64_t parent_space;
813 uint64_t est_used;
814
815 mutex_enter(&dd->dd_lock);
816 if (space > 0)
817 dd->dd_space_towrite[tx->tx_txg & TXG_MASK] += space;
818
b128c09f 819 est_used = dsl_dir_space_towrite(dd) + dd->dd_phys->dd_used_bytes;
34dc7c2f
BB
820 parent_space = parent_delta(dd, est_used, space);
821 mutex_exit(&dd->dd_lock);
822
823 /* Make sure that we clean up dd_space_to* */
824 dsl_dir_dirty(dd, tx);
825
826 /* XXX this is potentially expensive and unnecessary... */
827 if (parent_space && dd->dd_parent)
828 dsl_dir_willuse_space_impl(dd->dd_parent, parent_space, tx);
829}
830
831/*
832 * Call in open context when we think we're going to write/free space,
833 * eg. when dirtying data. Be conservative (ie. OK to write less than
834 * this or free more than this, but don't write more or free less).
835 */
836void
837dsl_dir_willuse_space(dsl_dir_t *dd, int64_t space, dmu_tx_t *tx)
838{
839 dsl_pool_willuse_space(dd->dd_pool, space, tx);
840 dsl_dir_willuse_space_impl(dd, space, tx);
841}
842
843/* call from syncing context when we actually write/free space for this dd */
844void
b128c09f 845dsl_dir_diduse_space(dsl_dir_t *dd, dd_used_t type,
34dc7c2f
BB
846 int64_t used, int64_t compressed, int64_t uncompressed, dmu_tx_t *tx)
847{
848 int64_t accounted_delta;
a169a625
MA
849
850 /*
851 * dsl_dataset_set_refreservation_sync_impl() calls this with
852 * dd_lock held, so that it can atomically update
853 * ds->ds_reserved and the dsl_dir accounting, so that
854 * dsl_dataset_check_quota() can see dataset and dir accounting
855 * consistently.
856 */
b128c09f 857 boolean_t needlock = !MUTEX_HELD(&dd->dd_lock);
34dc7c2f
BB
858
859 ASSERT(dmu_tx_is_syncing(tx));
b128c09f 860 ASSERT(type < DD_USED_NUM);
34dc7c2f 861
a169a625
MA
862 dmu_buf_will_dirty(dd->dd_dbuf, tx);
863
b128c09f
BB
864 if (needlock)
865 mutex_enter(&dd->dd_lock);
866 accounted_delta = parent_delta(dd, dd->dd_phys->dd_used_bytes, used);
867 ASSERT(used >= 0 || dd->dd_phys->dd_used_bytes >= -used);
34dc7c2f
BB
868 ASSERT(compressed >= 0 ||
869 dd->dd_phys->dd_compressed_bytes >= -compressed);
870 ASSERT(uncompressed >= 0 ||
871 dd->dd_phys->dd_uncompressed_bytes >= -uncompressed);
b128c09f 872 dd->dd_phys->dd_used_bytes += used;
34dc7c2f
BB
873 dd->dd_phys->dd_uncompressed_bytes += uncompressed;
874 dd->dd_phys->dd_compressed_bytes += compressed;
b128c09f
BB
875
876 if (dd->dd_phys->dd_flags & DD_FLAG_USED_BREAKDOWN) {
877 ASSERT(used > 0 ||
878 dd->dd_phys->dd_used_breakdown[type] >= -used);
879 dd->dd_phys->dd_used_breakdown[type] += used;
880#ifdef DEBUG
d6320ddb
BB
881 {
882 dd_used_t t;
883 uint64_t u = 0;
884 for (t = 0; t < DD_USED_NUM; t++)
885 u += dd->dd_phys->dd_used_breakdown[t];
886 ASSERT3U(u, ==, dd->dd_phys->dd_used_bytes);
887 }
b128c09f
BB
888#endif
889 }
890 if (needlock)
891 mutex_exit(&dd->dd_lock);
34dc7c2f
BB
892
893 if (dd->dd_parent != NULL) {
b128c09f 894 dsl_dir_diduse_space(dd->dd_parent, DD_USED_CHILD,
34dc7c2f 895 accounted_delta, compressed, uncompressed, tx);
b128c09f
BB
896 dsl_dir_transfer_space(dd->dd_parent,
897 used - accounted_delta,
898 DD_USED_CHILD_RSRV, DD_USED_CHILD, tx);
34dc7c2f
BB
899 }
900}
901
b128c09f
BB
902void
903dsl_dir_transfer_space(dsl_dir_t *dd, int64_t delta,
904 dd_used_t oldtype, dd_used_t newtype, dmu_tx_t *tx)
905{
b128c09f
BB
906 ASSERT(dmu_tx_is_syncing(tx));
907 ASSERT(oldtype < DD_USED_NUM);
908 ASSERT(newtype < DD_USED_NUM);
909
910 if (delta == 0 || !(dd->dd_phys->dd_flags & DD_FLAG_USED_BREAKDOWN))
911 return;
912
a169a625
MA
913 dmu_buf_will_dirty(dd->dd_dbuf, tx);
914 mutex_enter(&dd->dd_lock);
b128c09f
BB
915 ASSERT(delta > 0 ?
916 dd->dd_phys->dd_used_breakdown[oldtype] >= delta :
917 dd->dd_phys->dd_used_breakdown[newtype] >= -delta);
918 ASSERT(dd->dd_phys->dd_used_bytes >= ABS(delta));
919 dd->dd_phys->dd_used_breakdown[oldtype] -= delta;
920 dd->dd_phys->dd_used_breakdown[newtype] += delta;
a169a625 921 mutex_exit(&dd->dd_lock);
b128c09f
BB
922}
923
13fe0198
MA
924typedef struct dsl_dir_set_qr_arg {
925 const char *ddsqra_name;
926 zprop_source_t ddsqra_source;
927 uint64_t ddsqra_value;
928} dsl_dir_set_qr_arg_t;
929
34dc7c2f 930static int
13fe0198 931dsl_dir_set_quota_check(void *arg, dmu_tx_t *tx)
34dc7c2f 932{
13fe0198
MA
933 dsl_dir_set_qr_arg_t *ddsqra = arg;
934 dsl_pool_t *dp = dmu_tx_pool(tx);
935 dsl_dataset_t *ds;
936 int error;
937 uint64_t towrite, newval;
34dc7c2f 938
13fe0198
MA
939 error = dsl_dataset_hold(dp, ddsqra->ddsqra_name, FTAG, &ds);
940 if (error != 0)
941 return (error);
942
943 error = dsl_prop_predict(ds->ds_dir, "quota",
944 ddsqra->ddsqra_source, ddsqra->ddsqra_value, &newval);
945 if (error != 0) {
946 dsl_dataset_rele(ds, FTAG);
947 return (error);
948 }
428870ff 949
13fe0198
MA
950 if (newval == 0) {
951 dsl_dataset_rele(ds, FTAG);
34dc7c2f 952 return (0);
13fe0198 953 }
34dc7c2f 954
13fe0198 955 mutex_enter(&ds->ds_dir->dd_lock);
34dc7c2f
BB
956 /*
957 * If we are doing the preliminary check in open context, and
958 * there are pending changes, then don't fail it, since the
959 * pending changes could under-estimate the amount of space to be
960 * freed up.
961 */
13fe0198 962 towrite = dsl_dir_space_towrite(ds->ds_dir);
34dc7c2f 963 if ((dmu_tx_is_syncing(tx) || towrite == 0) &&
13fe0198
MA
964 (newval < ds->ds_dir->dd_phys->dd_reserved ||
965 newval < ds->ds_dir->dd_phys->dd_used_bytes + towrite)) {
2e528b49 966 error = SET_ERROR(ENOSPC);
34dc7c2f 967 }
13fe0198
MA
968 mutex_exit(&ds->ds_dir->dd_lock);
969 dsl_dataset_rele(ds, FTAG);
970 return (error);
34dc7c2f
BB
971}
972
34dc7c2f 973static void
13fe0198 974dsl_dir_set_quota_sync(void *arg, dmu_tx_t *tx)
34dc7c2f 975{
13fe0198
MA
976 dsl_dir_set_qr_arg_t *ddsqra = arg;
977 dsl_pool_t *dp = dmu_tx_pool(tx);
978 dsl_dataset_t *ds;
979 uint64_t newval;
428870ff 980
13fe0198 981 VERIFY0(dsl_dataset_hold(dp, ddsqra->ddsqra_name, FTAG, &ds));
34dc7c2f 982
b1118acb
MM
983 if (spa_version(dp->dp_spa) >= SPA_VERSION_RECVD_PROPS) {
984 dsl_prop_set_sync_impl(ds, zfs_prop_to_name(ZFS_PROP_QUOTA),
985 ddsqra->ddsqra_source, sizeof (ddsqra->ddsqra_value), 1,
986 &ddsqra->ddsqra_value, tx);
34dc7c2f 987
b1118acb
MM
988 VERIFY0(dsl_prop_get_int_ds(ds,
989 zfs_prop_to_name(ZFS_PROP_QUOTA), &newval));
990 } else {
991 newval = ddsqra->ddsqra_value;
992 spa_history_log_internal_ds(ds, "set", tx, "%s=%lld",
993 zfs_prop_to_name(ZFS_PROP_QUOTA), (longlong_t)newval);
994 }
6f1ffb06 995
13fe0198
MA
996 dmu_buf_will_dirty(ds->ds_dir->dd_dbuf, tx);
997 mutex_enter(&ds->ds_dir->dd_lock);
998 ds->ds_dir->dd_phys->dd_quota = newval;
999 mutex_exit(&ds->ds_dir->dd_lock);
1000 dsl_dataset_rele(ds, FTAG);
34dc7c2f
BB
1001}
1002
1003int
428870ff 1004dsl_dir_set_quota(const char *ddname, zprop_source_t source, uint64_t quota)
34dc7c2f 1005{
13fe0198 1006 dsl_dir_set_qr_arg_t ddsqra;
428870ff 1007
13fe0198
MA
1008 ddsqra.ddsqra_name = ddname;
1009 ddsqra.ddsqra_source = source;
1010 ddsqra.ddsqra_value = quota;
428870ff 1011
13fe0198
MA
1012 return (dsl_sync_task(ddname, dsl_dir_set_quota_check,
1013 dsl_dir_set_quota_sync, &ddsqra, 0));
34dc7c2f
BB
1014}
1015
1016int
13fe0198 1017dsl_dir_set_reservation_check(void *arg, dmu_tx_t *tx)
34dc7c2f 1018{
13fe0198
MA
1019 dsl_dir_set_qr_arg_t *ddsqra = arg;
1020 dsl_pool_t *dp = dmu_tx_pool(tx);
1021 dsl_dataset_t *ds;
1022 dsl_dir_t *dd;
1023 uint64_t newval, used, avail;
1024 int error;
428870ff 1025
13fe0198
MA
1026 error = dsl_dataset_hold(dp, ddsqra->ddsqra_name, FTAG, &ds);
1027 if (error != 0)
1028 return (error);
1029 dd = ds->ds_dir;
34dc7c2f
BB
1030
1031 /*
1032 * If we are doing the preliminary check in open context, the
1033 * space estimates may be inaccurate.
1034 */
13fe0198
MA
1035 if (!dmu_tx_is_syncing(tx)) {
1036 dsl_dataset_rele(ds, FTAG);
34dc7c2f 1037 return (0);
13fe0198
MA
1038 }
1039
1040 error = dsl_prop_predict(ds->ds_dir,
1041 zfs_prop_to_name(ZFS_PROP_RESERVATION),
1042 ddsqra->ddsqra_source, ddsqra->ddsqra_value, &newval);
1043 if (error != 0) {
1044 dsl_dataset_rele(ds, FTAG);
1045 return (error);
1046 }
34dc7c2f
BB
1047
1048 mutex_enter(&dd->dd_lock);
b128c09f 1049 used = dd->dd_phys->dd_used_bytes;
34dc7c2f
BB
1050 mutex_exit(&dd->dd_lock);
1051
1052 if (dd->dd_parent) {
1053 avail = dsl_dir_space_available(dd->dd_parent,
1054 NULL, 0, FALSE);
1055 } else {
1056 avail = dsl_pool_adjustedsize(dd->dd_pool, B_FALSE) - used;
1057 }
1058
13fe0198
MA
1059 if (MAX(used, newval) > MAX(used, dd->dd_phys->dd_reserved)) {
1060 uint64_t delta = MAX(used, newval) -
d164b209
BB
1061 MAX(used, dd->dd_phys->dd_reserved);
1062
13fe0198
MA
1063 if (delta > avail ||
1064 (dd->dd_phys->dd_quota > 0 &&
1065 newval > dd->dd_phys->dd_quota))
2e528b49 1066 error = SET_ERROR(ENOSPC);
d164b209
BB
1067 }
1068
13fe0198
MA
1069 dsl_dataset_rele(ds, FTAG);
1070 return (error);
34dc7c2f
BB
1071}
1072
13fe0198 1073void
6f1ffb06 1074dsl_dir_set_reservation_sync_impl(dsl_dir_t *dd, uint64_t value, dmu_tx_t *tx)
34dc7c2f 1075{
34dc7c2f
BB
1076 uint64_t used;
1077 int64_t delta;
1078
1079 dmu_buf_will_dirty(dd->dd_dbuf, tx);
1080
1081 mutex_enter(&dd->dd_lock);
b128c09f 1082 used = dd->dd_phys->dd_used_bytes;
6f1ffb06
MA
1083 delta = MAX(used, value) - MAX(used, dd->dd_phys->dd_reserved);
1084 dd->dd_phys->dd_reserved = value;
34dc7c2f
BB
1085
1086 if (dd->dd_parent != NULL) {
1087 /* Roll up this additional usage into our ancestors */
b128c09f
BB
1088 dsl_dir_diduse_space(dd->dd_parent, DD_USED_CHILD_RSRV,
1089 delta, 0, 0, tx);
34dc7c2f 1090 }
b128c09f 1091 mutex_exit(&dd->dd_lock);
34dc7c2f
BB
1092}
1093
6f1ffb06 1094static void
13fe0198 1095dsl_dir_set_reservation_sync(void *arg, dmu_tx_t *tx)
6f1ffb06 1096{
13fe0198
MA
1097 dsl_dir_set_qr_arg_t *ddsqra = arg;
1098 dsl_pool_t *dp = dmu_tx_pool(tx);
1099 dsl_dataset_t *ds;
1100 uint64_t newval;
6f1ffb06 1101
13fe0198
MA
1102 VERIFY0(dsl_dataset_hold(dp, ddsqra->ddsqra_name, FTAG, &ds));
1103
b1118acb
MM
1104 if (spa_version(dp->dp_spa) >= SPA_VERSION_RECVD_PROPS) {
1105 dsl_prop_set_sync_impl(ds,
1106 zfs_prop_to_name(ZFS_PROP_RESERVATION),
1107 ddsqra->ddsqra_source, sizeof (ddsqra->ddsqra_value), 1,
1108 &ddsqra->ddsqra_value, tx);
13fe0198 1109
b1118acb
MM
1110 VERIFY0(dsl_prop_get_int_ds(ds,
1111 zfs_prop_to_name(ZFS_PROP_RESERVATION), &newval));
1112 } else {
1113 newval = ddsqra->ddsqra_value;
1114 spa_history_log_internal_ds(ds, "set", tx, "%s=%lld",
1115 zfs_prop_to_name(ZFS_PROP_RESERVATION),
1116 (longlong_t)newval);
1117 }
1118
13fe0198
MA
1119 dsl_dir_set_reservation_sync_impl(ds->ds_dir, newval, tx);
1120 dsl_dataset_rele(ds, FTAG);
6f1ffb06
MA
1121 }
1122
34dc7c2f 1123int
428870ff
BB
1124dsl_dir_set_reservation(const char *ddname, zprop_source_t source,
1125 uint64_t reservation)
34dc7c2f 1126{
13fe0198 1127 dsl_dir_set_qr_arg_t ddsqra;
428870ff 1128
13fe0198
MA
1129 ddsqra.ddsqra_name = ddname;
1130 ddsqra.ddsqra_source = source;
1131 ddsqra.ddsqra_value = reservation;
428870ff 1132
13fe0198
MA
1133 return (dsl_sync_task(ddname, dsl_dir_set_reservation_check,
1134 dsl_dir_set_reservation_sync, &ddsqra, 0));
34dc7c2f
BB
1135}
1136
1137static dsl_dir_t *
1138closest_common_ancestor(dsl_dir_t *ds1, dsl_dir_t *ds2)
1139{
1140 for (; ds1; ds1 = ds1->dd_parent) {
1141 dsl_dir_t *dd;
1142 for (dd = ds2; dd; dd = dd->dd_parent) {
1143 if (ds1 == dd)
1144 return (dd);
1145 }
1146 }
1147 return (NULL);
1148}
1149
1150/*
1151 * If delta is applied to dd, how much of that delta would be applied to
1152 * ancestor? Syncing context only.
1153 */
1154static int64_t
1155would_change(dsl_dir_t *dd, int64_t delta, dsl_dir_t *ancestor)
1156{
1157 if (dd == ancestor)
1158 return (delta);
1159
1160 mutex_enter(&dd->dd_lock);
b128c09f 1161 delta = parent_delta(dd, dd->dd_phys->dd_used_bytes, delta);
34dc7c2f
BB
1162 mutex_exit(&dd->dd_lock);
1163 return (would_change(dd->dd_parent, delta, ancestor));
1164}
1165
13fe0198
MA
1166typedef struct dsl_dir_rename_arg {
1167 const char *ddra_oldname;
1168 const char *ddra_newname;
1169} dsl_dir_rename_arg_t;
34dc7c2f 1170
13fe0198 1171/* ARGSUSED */
34dc7c2f 1172static int
13fe0198 1173dsl_valid_rename(dsl_pool_t *dp, dsl_dataset_t *ds, void *arg)
34dc7c2f 1174{
13fe0198
MA
1175 int *deltap = arg;
1176 char namebuf[MAXNAMELEN];
34dc7c2f 1177
13fe0198
MA
1178 dsl_dataset_name(ds, namebuf);
1179
1180 if (strlen(namebuf) + *deltap >= MAXNAMELEN)
2e528b49 1181 return (SET_ERROR(ENAMETOOLONG));
13fe0198
MA
1182 return (0);
1183}
1184
1185static int
1186dsl_dir_rename_check(void *arg, dmu_tx_t *tx)
1187{
1188 dsl_dir_rename_arg_t *ddra = arg;
1189 dsl_pool_t *dp = dmu_tx_pool(tx);
1190 dsl_dir_t *dd, *newparent;
1191 const char *mynewname;
1192 int error;
1193 int delta = strlen(ddra->ddra_newname) - strlen(ddra->ddra_oldname);
34dc7c2f 1194
13fe0198
MA
1195 /* target dir should exist */
1196 error = dsl_dir_hold(dp, ddra->ddra_oldname, FTAG, &dd, NULL);
1197 if (error != 0)
1198 return (error);
1199
1200 /* new parent should exist */
1201 error = dsl_dir_hold(dp, ddra->ddra_newname, FTAG,
1202 &newparent, &mynewname);
1203 if (error != 0) {
1204 dsl_dir_rele(dd, FTAG);
1205 return (error);
1206 }
1207
1208 /* can't rename to different pool */
1209 if (dd->dd_pool != newparent->dd_pool) {
1210 dsl_dir_rele(newparent, FTAG);
1211 dsl_dir_rele(dd, FTAG);
2e528b49 1212 return (SET_ERROR(ENXIO));
13fe0198
MA
1213 }
1214
1215 /* new name should not already exist */
1216 if (mynewname == NULL) {
1217 dsl_dir_rele(newparent, FTAG);
1218 dsl_dir_rele(dd, FTAG);
2e528b49 1219 return (SET_ERROR(EEXIST));
13fe0198
MA
1220 }
1221
1222 /* if the name length is growing, validate child name lengths */
1223 if (delta > 0) {
1224 error = dmu_objset_find_dp(dp, dd->dd_object, dsl_valid_rename,
1225 &delta, DS_FIND_CHILDREN | DS_FIND_SNAPSHOTS);
1226 if (error != 0) {
1227 dsl_dir_rele(newparent, FTAG);
1228 dsl_dir_rele(dd, FTAG);
1229 return (error);
1230 }
1231 }
34dc7c2f 1232
13fe0198 1233 if (newparent != dd->dd_parent) {
34dc7c2f
BB
1234 /* is there enough space? */
1235 uint64_t myspace =
b128c09f 1236 MAX(dd->dd_phys->dd_used_bytes, dd->dd_phys->dd_reserved);
34dc7c2f
BB
1237
1238 /* no rename into our descendant */
13fe0198
MA
1239 if (closest_common_ancestor(dd, newparent) == dd) {
1240 dsl_dir_rele(newparent, FTAG);
1241 dsl_dir_rele(dd, FTAG);
2e528b49 1242 return (SET_ERROR(EINVAL));
13fe0198 1243 }
34dc7c2f 1244
13fe0198
MA
1245 error = dsl_dir_transfer_possible(dd->dd_parent,
1246 newparent, myspace);
1247 if (error != 0) {
1248 dsl_dir_rele(newparent, FTAG);
1249 dsl_dir_rele(dd, FTAG);
1250 return (error);
1251 }
34dc7c2f
BB
1252 }
1253
13fe0198
MA
1254 dsl_dir_rele(newparent, FTAG);
1255 dsl_dir_rele(dd, FTAG);
34dc7c2f
BB
1256 return (0);
1257}
1258
1259static void
13fe0198 1260dsl_dir_rename_sync(void *arg, dmu_tx_t *tx)
34dc7c2f 1261{
13fe0198
MA
1262 dsl_dir_rename_arg_t *ddra = arg;
1263 dsl_pool_t *dp = dmu_tx_pool(tx);
1264 dsl_dir_t *dd, *newparent;
1265 const char *mynewname;
1266 int error;
34dc7c2f 1267 objset_t *mos = dp->dp_meta_objset;
34dc7c2f 1268
13fe0198
MA
1269 VERIFY0(dsl_dir_hold(dp, ddra->ddra_oldname, FTAG, &dd, NULL));
1270 VERIFY0(dsl_dir_hold(dp, ddra->ddra_newname, FTAG, &newparent,
1271 &mynewname));
34dc7c2f 1272
6f1ffb06 1273 /* Log this before we change the name. */
6f1ffb06 1274 spa_history_log_internal_dd(dd, "rename", tx,
13fe0198 1275 "-> %s", ddra->ddra_newname);
6f1ffb06 1276
13fe0198 1277 if (newparent != dd->dd_parent) {
b128c09f
BB
1278 dsl_dir_diduse_space(dd->dd_parent, DD_USED_CHILD,
1279 -dd->dd_phys->dd_used_bytes,
34dc7c2f
BB
1280 -dd->dd_phys->dd_compressed_bytes,
1281 -dd->dd_phys->dd_uncompressed_bytes, tx);
13fe0198 1282 dsl_dir_diduse_space(newparent, DD_USED_CHILD,
b128c09f 1283 dd->dd_phys->dd_used_bytes,
34dc7c2f
BB
1284 dd->dd_phys->dd_compressed_bytes,
1285 dd->dd_phys->dd_uncompressed_bytes, tx);
b128c09f
BB
1286
1287 if (dd->dd_phys->dd_reserved > dd->dd_phys->dd_used_bytes) {
1288 uint64_t unused_rsrv = dd->dd_phys->dd_reserved -
1289 dd->dd_phys->dd_used_bytes;
1290
1291 dsl_dir_diduse_space(dd->dd_parent, DD_USED_CHILD_RSRV,
1292 -unused_rsrv, 0, 0, tx);
13fe0198 1293 dsl_dir_diduse_space(newparent, DD_USED_CHILD_RSRV,
b128c09f
BB
1294 unused_rsrv, 0, 0, tx);
1295 }
34dc7c2f
BB
1296 }
1297
1298 dmu_buf_will_dirty(dd->dd_dbuf, tx);
1299
1300 /* remove from old parent zapobj */
13fe0198 1301 error = zap_remove(mos, dd->dd_parent->dd_phys->dd_child_dir_zapobj,
34dc7c2f 1302 dd->dd_myname, tx);
13fe0198 1303 ASSERT0(error);
34dc7c2f 1304
13fe0198
MA
1305 (void) strcpy(dd->dd_myname, mynewname);
1306 dsl_dir_rele(dd->dd_parent, dd);
1307 dd->dd_phys->dd_parent_obj = newparent->dd_object;
1308 VERIFY0(dsl_dir_hold_obj(dp,
1309 newparent->dd_object, NULL, dd, &dd->dd_parent));
34dc7c2f
BB
1310
1311 /* add to new parent zapobj */
13fe0198
MA
1312 VERIFY0(zap_add(mos, newparent->dd_phys->dd_child_dir_zapobj,
1313 dd->dd_myname, 8, 1, &dd->dd_object, tx));
1314
1315 dsl_prop_notify_all(dd);
34dc7c2f 1316
13fe0198
MA
1317 dsl_dir_rele(newparent, FTAG);
1318 dsl_dir_rele(dd, FTAG);
34dc7c2f
BB
1319}
1320
1321int
13fe0198 1322dsl_dir_rename(const char *oldname, const char *newname)
34dc7c2f 1323{
13fe0198 1324 dsl_dir_rename_arg_t ddra;
34dc7c2f 1325
13fe0198
MA
1326 ddra.ddra_oldname = oldname;
1327 ddra.ddra_newname = newname;
34dc7c2f 1328
13fe0198
MA
1329 return (dsl_sync_task(oldname,
1330 dsl_dir_rename_check, dsl_dir_rename_sync, &ddra, 3));
34dc7c2f
BB
1331}
1332
1333int
1334dsl_dir_transfer_possible(dsl_dir_t *sdd, dsl_dir_t *tdd, uint64_t space)
1335{
1336 dsl_dir_t *ancestor;
1337 int64_t adelta;
1338 uint64_t avail;
1339
1340 ancestor = closest_common_ancestor(sdd, tdd);
1341 adelta = would_change(sdd, -space, ancestor);
1342 avail = dsl_dir_space_available(tdd, ancestor, adelta, FALSE);
1343 if (avail < space)
2e528b49 1344 return (SET_ERROR(ENOSPC));
34dc7c2f
BB
1345
1346 return (0);
1347}
428870ff
BB
1348
1349timestruc_t
1350dsl_dir_snap_cmtime(dsl_dir_t *dd)
1351{
1352 timestruc_t t;
1353
1354 mutex_enter(&dd->dd_lock);
1355 t = dd->dd_snap_cmtime;
1356 mutex_exit(&dd->dd_lock);
1357
1358 return (t);
1359}
1360
1361void
1362dsl_dir_snap_cmtime_update(dsl_dir_t *dd)
1363{
1364 timestruc_t t;
1365
1366 gethrestime(&t);
1367 mutex_enter(&dd->dd_lock);
1368 dd->dd_snap_cmtime = t;
1369 mutex_exit(&dd->dd_lock);
1370}
c28b2279
BB
1371
1372#if defined(_KERNEL) && defined(HAVE_SPL)
1373EXPORT_SYMBOL(dsl_dir_set_quota);
1374EXPORT_SYMBOL(dsl_dir_set_reservation);
c28b2279 1375#endif