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
22/*
23 * Copyright (c) 2009, 2010, Oracle and/or its affiliates. All rights reserved.
24 */
25
26#include <sys/zfs_context.h>
27#include <sys/spa.h>
28#include <sys/zio.h>
29#include <sys/ddt.h>
30#include <sys/zap.h>
31#include <sys/dmu_tx.h>
32
33int ddt_zap_leaf_blockshift = 12;
34int ddt_zap_indirect_blockshift = 12;
35
36static int
37ddt_zap_create(objset_t *os, uint64_t *objectp, dmu_tx_t *tx, boolean_t prehash)
38{
39	zap_flags_t flags = ZAP_FLAG_HASH64 | ZAP_FLAG_UINT64_KEY;
40
41	if (prehash)
42		flags |= ZAP_FLAG_PRE_HASHED_KEY;
43
44	*objectp = zap_create_flags(os, 0, flags, DMU_OT_DDT_ZAP,
45	    ddt_zap_leaf_blockshift, ddt_zap_indirect_blockshift,
46	    DMU_OT_NONE, 0, tx);
47
48	return (*objectp == 0 ? ENOTSUP : 0);
49}
50
51static int
52ddt_zap_destroy(objset_t *os, uint64_t object, dmu_tx_t *tx)
53{
54	return (zap_destroy(os, object, tx));
55}
56
57static int
58ddt_zap_lookup(objset_t *os, uint64_t object, ddt_entry_t *dde)
59{
60	uchar_t cbuf[sizeof (dde->dde_phys) + 1];
61	uint64_t one, csize;
62	int error;
63
64	error = zap_length_uint64(os, object, (uint64_t *)&dde->dde_key,
65	    DDT_KEY_WORDS, &one, &csize);
66	if (error)
67		return (error);
68
69	ASSERT(one == 1);
70	ASSERT(csize <= sizeof (cbuf));
71
72	error = zap_lookup_uint64(os, object, (uint64_t *)&dde->dde_key,
73	    DDT_KEY_WORDS, 1, csize, cbuf);
74	if (error)
75		return (error);
76
77	ddt_decompress(cbuf, dde->dde_phys, csize, sizeof (dde->dde_phys));
78
79	return (0);
80}
81
82static void
83ddt_zap_prefetch(objset_t *os, uint64_t object, ddt_entry_t *dde)
84{
85	(void) zap_prefetch_uint64(os, object, (uint64_t *)&dde->dde_key,
86	    DDT_KEY_WORDS);
87}
88
89static int
90ddt_zap_update(objset_t *os, uint64_t object, ddt_entry_t *dde, dmu_tx_t *tx)
91{
92	uchar_t cbuf[sizeof (dde->dde_phys) + 1];
93	uint64_t csize;
94
95	csize = ddt_compress(dde->dde_phys, cbuf,
96	    sizeof (dde->dde_phys), sizeof (cbuf));
97
98	return (zap_update_uint64(os, object, (uint64_t *)&dde->dde_key,
99	    DDT_KEY_WORDS, 1, csize, cbuf, tx));
100}
101
102static int
103ddt_zap_remove(objset_t *os, uint64_t object, ddt_entry_t *dde, dmu_tx_t *tx)
104{
105	return (zap_remove_uint64(os, object, (uint64_t *)&dde->dde_key,
106	    DDT_KEY_WORDS, tx));
107}
108
109static int
110ddt_zap_walk(objset_t *os, uint64_t object, ddt_entry_t *dde, uint64_t *walk)
111{
112	zap_cursor_t zc;
113	zap_attribute_t za;
114	int error;
115
116	zap_cursor_init_serialized(&zc, os, object, *walk);
117	if ((error = zap_cursor_retrieve(&zc, &za)) == 0) {
118		uchar_t cbuf[sizeof (dde->dde_phys) + 1];
119		uint64_t csize = za.za_num_integers;
120		ASSERT(za.za_integer_length == 1);
121		error = zap_lookup_uint64(os, object, (uint64_t *)za.za_name,
122		    DDT_KEY_WORDS, 1, csize, cbuf);
123		ASSERT(error == 0);
124		if (error == 0) {
125			ddt_decompress(cbuf, dde->dde_phys, csize,
126			    sizeof (dde->dde_phys));
127			dde->dde_key = *(ddt_key_t *)za.za_name;
128		}
129		zap_cursor_advance(&zc);
130		*walk = zap_cursor_serialize(&zc);
131	}
132	zap_cursor_fini(&zc);
133	return (error);
134}
135
136static int
137ddt_zap_count(objset_t *os, uint64_t object, uint64_t *count)
138{
139
140	return (zap_count(os, object, count));
141}
142
143const ddt_ops_t ddt_zap_ops = {
144	"zap",
145	ddt_zap_create,
146	ddt_zap_destroy,
147	ddt_zap_lookup,
148	ddt_zap_prefetch,
149	ddt_zap_update,
150	ddt_zap_remove,
151	ddt_zap_walk,
152	ddt_zap_count,
153};
154