xref: /illumos-gate/usr/src/cmd/zhack/zhack.c (revision ca0cc3918a1789fa839194af2a9245f801a06b1a)
153089ab7Seschrock /*
253089ab7Seschrock  * CDDL HEADER START
353089ab7Seschrock  *
453089ab7Seschrock  * The contents of this file are subject to the terms of the
553089ab7Seschrock  * Common Development and Distribution License (the "License").
653089ab7Seschrock  * You may not use this file except in compliance with the License.
753089ab7Seschrock  *
853089ab7Seschrock  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
953089ab7Seschrock  * or http://www.opensolaris.org/os/licensing.
1053089ab7Seschrock  * See the License for the specific language governing permissions
1153089ab7Seschrock  * and limitations under the License.
1253089ab7Seschrock  *
1353089ab7Seschrock  * When distributing Covered Code, include this CDDL HEADER in each
1453089ab7Seschrock  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
1553089ab7Seschrock  * If applicable, add the following below this CDDL HEADER, with the
1653089ab7Seschrock  * fields enclosed by brackets "[]" replaced with your own identifying
1753089ab7Seschrock  * information: Portions Copyright [yyyy] [name of copyright owner]
1853089ab7Seschrock  *
1953089ab7Seschrock  * CDDL HEADER END
2053089ab7Seschrock  */
2153089ab7Seschrock 
2253089ab7Seschrock /*
23*ca0cc391SMatthew Ahrens  * Copyright (c) 2011, 2015 by Delphix. All rights reserved.
24a7a845e4SSteven Hartland  * Copyright (c) 2013 Steven Hartland. All rights reserved.
2553089ab7Seschrock  */
2653089ab7Seschrock 
2753089ab7Seschrock /*
2853089ab7Seschrock  * zhack is a debugging tool that can write changes to ZFS pool using libzpool
2953089ab7Seschrock  * for testing purposes. Altering pools with zhack is unsupported and may
3053089ab7Seschrock  * result in corrupted pools.
3153089ab7Seschrock  */
3253089ab7Seschrock 
3353089ab7Seschrock #include <stdio.h>
3453089ab7Seschrock #include <stdlib.h>
3553089ab7Seschrock #include <ctype.h>
3653089ab7Seschrock #include <sys/zfs_context.h>
3753089ab7Seschrock #include <sys/spa.h>
3853089ab7Seschrock #include <sys/spa_impl.h>
3953089ab7Seschrock #include <sys/dmu.h>
4053089ab7Seschrock #include <sys/zap.h>
4153089ab7Seschrock #include <sys/zfs_znode.h>
4253089ab7Seschrock #include <sys/dsl_synctask.h>
4353089ab7Seschrock #include <sys/vdev.h>
4453089ab7Seschrock #include <sys/fs/zfs.h>
4553089ab7Seschrock #include <sys/dmu_objset.h>
4653089ab7Seschrock #include <sys/dsl_pool.h>
4753089ab7Seschrock #include <sys/zio_checksum.h>
4853089ab7Seschrock #include <sys/zio_compress.h>
4953089ab7Seschrock #include <sys/zfeature.h>
503b2aab18SMatthew Ahrens #include <sys/dmu_tx.h>
5153089ab7Seschrock #undef ZFS_MAXNAMELEN
5253089ab7Seschrock #undef verify
5353089ab7Seschrock #include <libzfs.h>
5453089ab7Seschrock 
5553089ab7Seschrock extern boolean_t zfeature_checks_disable;
5653089ab7Seschrock 
5753089ab7Seschrock const char cmdname[] = "zhack";
5853089ab7Seschrock libzfs_handle_t *g_zfs;
5953089ab7Seschrock static importargs_t g_importargs;
6053089ab7Seschrock static char *g_pool;
6153089ab7Seschrock static boolean_t g_readonly;
6253089ab7Seschrock 
6353089ab7Seschrock static void
6453089ab7Seschrock usage(void)
6553089ab7Seschrock {
6653089ab7Seschrock 	(void) fprintf(stderr,
6753089ab7Seschrock 	    "Usage: %s [-c cachefile] [-d dir] <subcommand> <args> ...\n"
6853089ab7Seschrock 	    "where <subcommand> <args> is one of the following:\n"
6953089ab7Seschrock 	    "\n", cmdname);
7053089ab7Seschrock 
7153089ab7Seschrock 	(void) fprintf(stderr,
7253089ab7Seschrock 	    "    feature stat <pool>\n"
7353089ab7Seschrock 	    "        print information about enabled features\n"
7453089ab7Seschrock 	    "    feature enable [-d desc] <pool> <feature>\n"
7553089ab7Seschrock 	    "        add a new enabled feature to the pool\n"
7653089ab7Seschrock 	    "        -d <desc> sets the feature's description\n"
7753089ab7Seschrock 	    "    feature ref [-md] <pool> <feature>\n"
7853089ab7Seschrock 	    "        change the refcount on the given feature\n"
7953089ab7Seschrock 	    "        -d decrease instead of increase the refcount\n"
8053089ab7Seschrock 	    "        -m add the feature to the label if increasing refcount\n"
8153089ab7Seschrock 	    "\n"
8253089ab7Seschrock 	    "    <feature> : should be a feature guid\n");
8353089ab7Seschrock 	exit(1);
8453089ab7Seschrock }
8553089ab7Seschrock 
8653089ab7Seschrock 
8753089ab7Seschrock static void
887fdd916cSGeorge Wilson fatal(spa_t *spa, void *tag, const char *fmt, ...)
8953089ab7Seschrock {
9053089ab7Seschrock 	va_list ap;
9153089ab7Seschrock 
927fdd916cSGeorge Wilson 	if (spa != NULL) {
937fdd916cSGeorge Wilson 		spa_close(spa, tag);
947fdd916cSGeorge Wilson 		(void) spa_export(g_pool, NULL, B_TRUE, B_FALSE);
957fdd916cSGeorge Wilson 	}
967fdd916cSGeorge Wilson 
9753089ab7Seschrock 	va_start(ap, fmt);
9853089ab7Seschrock 	(void) fprintf(stderr, "%s: ", cmdname);
9953089ab7Seschrock 	(void) vfprintf(stderr, fmt, ap);
10053089ab7Seschrock 	va_end(ap);
10153089ab7Seschrock 	(void) fprintf(stderr, "\n");
10253089ab7Seschrock 
10353089ab7Seschrock 	exit(1);
10453089ab7Seschrock }
10553089ab7Seschrock 
10653089ab7Seschrock /* ARGSUSED */
10753089ab7Seschrock static int
10853089ab7Seschrock space_delta_cb(dmu_object_type_t bonustype, void *data,
10953089ab7Seschrock     uint64_t *userp, uint64_t *groupp)
11053089ab7Seschrock {
11153089ab7Seschrock 	/*
11253089ab7Seschrock 	 * Is it a valid type of object to track?
11353089ab7Seschrock 	 */
11453089ab7Seschrock 	if (bonustype != DMU_OT_ZNODE && bonustype != DMU_OT_SA)
11553089ab7Seschrock 		return (ENOENT);
11653089ab7Seschrock 	(void) fprintf(stderr, "modifying object that needs user accounting");
11753089ab7Seschrock 	abort();
11853089ab7Seschrock 	/* NOTREACHED */
11953089ab7Seschrock }
12053089ab7Seschrock 
12153089ab7Seschrock /*
12253089ab7Seschrock  * Target is the dataset whose pool we want to open.
12353089ab7Seschrock  */
12453089ab7Seschrock static void
12553089ab7Seschrock import_pool(const char *target, boolean_t readonly)
12653089ab7Seschrock {
12753089ab7Seschrock 	nvlist_t *config;
12853089ab7Seschrock 	nvlist_t *pools;
12953089ab7Seschrock 	int error;
13053089ab7Seschrock 	char *sepp;
13153089ab7Seschrock 	spa_t *spa;
13253089ab7Seschrock 	nvpair_t *elem;
13353089ab7Seschrock 	nvlist_t *props;
13453089ab7Seschrock 	const char *name;
13553089ab7Seschrock 
13653089ab7Seschrock 	kernel_init(readonly ? FREAD : (FREAD | FWRITE));
13753089ab7Seschrock 	g_zfs = libzfs_init();
13853089ab7Seschrock 	ASSERT(g_zfs != NULL);
13953089ab7Seschrock 
14053089ab7Seschrock 	dmu_objset_register_type(DMU_OST_ZFS, space_delta_cb);
14153089ab7Seschrock 
14253089ab7Seschrock 	g_readonly = readonly;
14353089ab7Seschrock 
14453089ab7Seschrock 	/*
14553089ab7Seschrock 	 * If we only want readonly access, it's OK if we find
14653089ab7Seschrock 	 * a potentially-active (ie, imported into the kernel) pool from the
14753089ab7Seschrock 	 * default cachefile.
14853089ab7Seschrock 	 */
14953089ab7Seschrock 	if (readonly && spa_open(target, &spa, FTAG) == 0) {
15053089ab7Seschrock 		spa_close(spa, FTAG);
15153089ab7Seschrock 		return;
15253089ab7Seschrock 	}
15353089ab7Seschrock 
15453089ab7Seschrock 	g_importargs.unique = B_TRUE;
15553089ab7Seschrock 	g_importargs.can_be_active = readonly;
15653089ab7Seschrock 	g_pool = strdup(target);
15753089ab7Seschrock 	if ((sepp = strpbrk(g_pool, "/@")) != NULL)
15853089ab7Seschrock 		*sepp = '\0';
15953089ab7Seschrock 	g_importargs.poolname = g_pool;
16053089ab7Seschrock 	pools = zpool_search_import(g_zfs, &g_importargs);
16153089ab7Seschrock 
162a7a845e4SSteven Hartland 	if (nvlist_empty(pools)) {
16353089ab7Seschrock 		if (!g_importargs.can_be_active) {
16453089ab7Seschrock 			g_importargs.can_be_active = B_TRUE;
16553089ab7Seschrock 			if (zpool_search_import(g_zfs, &g_importargs) != NULL ||
16653089ab7Seschrock 			    spa_open(target, &spa, FTAG) == 0) {
1677fdd916cSGeorge Wilson 				fatal(spa, FTAG, "cannot import '%s': pool is "
1687fdd916cSGeorge Wilson 				    "active; run " "\"zpool export %s\" "
1697fdd916cSGeorge Wilson 				    "first\n", g_pool, g_pool);
17053089ab7Seschrock 			}
17153089ab7Seschrock 		}
17253089ab7Seschrock 
1737fdd916cSGeorge Wilson 		fatal(NULL, FTAG, "cannot import '%s': no such pool "
1747fdd916cSGeorge Wilson 		    "available\n", g_pool);
17553089ab7Seschrock 	}
17653089ab7Seschrock 
17753089ab7Seschrock 	elem = nvlist_next_nvpair(pools, NULL);
17853089ab7Seschrock 	name = nvpair_name(elem);
17953089ab7Seschrock 	verify(nvpair_value_nvlist(elem, &config) == 0);
18053089ab7Seschrock 
18153089ab7Seschrock 	props = NULL;
18253089ab7Seschrock 	if (readonly) {
18353089ab7Seschrock 		verify(nvlist_alloc(&props, NV_UNIQUE_NAME, 0) == 0);
18453089ab7Seschrock 		verify(nvlist_add_uint64(props,
18553089ab7Seschrock 		    zpool_prop_to_name(ZPOOL_PROP_READONLY), 1) == 0);
18653089ab7Seschrock 	}
18753089ab7Seschrock 
18853089ab7Seschrock 	zfeature_checks_disable = B_TRUE;
18953089ab7Seschrock 	error = spa_import(name, config, props, ZFS_IMPORT_NORMAL);
19053089ab7Seschrock 	zfeature_checks_disable = B_FALSE;
19153089ab7Seschrock 	if (error == EEXIST)
19253089ab7Seschrock 		error = 0;
19353089ab7Seschrock 
19453089ab7Seschrock 	if (error)
1957fdd916cSGeorge Wilson 		fatal(NULL, FTAG, "can't import '%s': %s", name,
1967fdd916cSGeorge Wilson 		    strerror(error));
19753089ab7Seschrock }
19853089ab7Seschrock 
19953089ab7Seschrock static void
20053089ab7Seschrock zhack_spa_open(const char *target, boolean_t readonly, void *tag, spa_t **spa)
20153089ab7Seschrock {
20253089ab7Seschrock 	int err;
20353089ab7Seschrock 
20453089ab7Seschrock 	import_pool(target, readonly);
20553089ab7Seschrock 
20653089ab7Seschrock 	zfeature_checks_disable = B_TRUE;
20753089ab7Seschrock 	err = spa_open(target, spa, tag);
20853089ab7Seschrock 	zfeature_checks_disable = B_FALSE;
20953089ab7Seschrock 
21053089ab7Seschrock 	if (err != 0)
2117fdd916cSGeorge Wilson 		fatal(*spa, FTAG, "cannot open '%s': %s", target,
2127fdd916cSGeorge Wilson 		    strerror(err));
21353089ab7Seschrock 	if (spa_version(*spa) < SPA_VERSION_FEATURES) {
2147fdd916cSGeorge Wilson 		fatal(*spa, FTAG, "'%s' has version %d, features not enabled",
2157fdd916cSGeorge Wilson 		    target, (int)spa_version(*spa));
21653089ab7Seschrock 	}
21753089ab7Seschrock }
21853089ab7Seschrock 
21953089ab7Seschrock static void
22053089ab7Seschrock dump_obj(objset_t *os, uint64_t obj, const char *name)
22153089ab7Seschrock {
22253089ab7Seschrock 	zap_cursor_t zc;
22353089ab7Seschrock 	zap_attribute_t za;
22453089ab7Seschrock 
22553089ab7Seschrock 	(void) printf("%s_obj:\n", name);
22653089ab7Seschrock 
22753089ab7Seschrock 	for (zap_cursor_init(&zc, os, obj);
22853089ab7Seschrock 	    zap_cursor_retrieve(&zc, &za) == 0;
22953089ab7Seschrock 	    zap_cursor_advance(&zc)) {
23053089ab7Seschrock 		if (za.za_integer_length == 8) {
23153089ab7Seschrock 			ASSERT(za.za_num_integers == 1);
23253089ab7Seschrock 			(void) printf("\t%s = %llu\n",
23353089ab7Seschrock 			    za.za_name, (u_longlong_t)za.za_first_integer);
23453089ab7Seschrock 		} else {
23553089ab7Seschrock 			ASSERT(za.za_integer_length == 1);
23653089ab7Seschrock 			char val[1024];
23753089ab7Seschrock 			VERIFY(zap_lookup(os, obj, za.za_name,
23853089ab7Seschrock 			    1, sizeof (val), val) == 0);
23953089ab7Seschrock 			(void) printf("\t%s = %s\n", za.za_name, val);
24053089ab7Seschrock 		}
24153089ab7Seschrock 	}
24253089ab7Seschrock 	zap_cursor_fini(&zc);
24353089ab7Seschrock }
24453089ab7Seschrock 
24553089ab7Seschrock static void
24653089ab7Seschrock dump_mos(spa_t *spa)
24753089ab7Seschrock {
24853089ab7Seschrock 	nvlist_t *nv = spa->spa_label_features;
24953089ab7Seschrock 
25053089ab7Seschrock 	(void) printf("label config:\n");
25153089ab7Seschrock 	for (nvpair_t *pair = nvlist_next_nvpair(nv, NULL);
25253089ab7Seschrock 	    pair != NULL;
25353089ab7Seschrock 	    pair = nvlist_next_nvpair(nv, pair)) {
25453089ab7Seschrock 		(void) printf("\t%s\n", nvpair_name(pair));
25553089ab7Seschrock 	}
25653089ab7Seschrock }
25753089ab7Seschrock 
25853089ab7Seschrock static void
25953089ab7Seschrock zhack_do_feature_stat(int argc, char **argv)
26053089ab7Seschrock {
26153089ab7Seschrock 	spa_t *spa;
26253089ab7Seschrock 	objset_t *os;
26353089ab7Seschrock 	char *target;
26453089ab7Seschrock 
26553089ab7Seschrock 	argc--;
26653089ab7Seschrock 	argv++;
26753089ab7Seschrock 
26853089ab7Seschrock 	if (argc < 1) {
26953089ab7Seschrock 		(void) fprintf(stderr, "error: missing pool name\n");
27053089ab7Seschrock 		usage();
27153089ab7Seschrock 	}
27253089ab7Seschrock 	target = argv[0];
27353089ab7Seschrock 
27453089ab7Seschrock 	zhack_spa_open(target, B_TRUE, FTAG, &spa);
27553089ab7Seschrock 	os = spa->spa_meta_objset;
27653089ab7Seschrock 
27753089ab7Seschrock 	dump_obj(os, spa->spa_feat_for_read_obj, "for_read");
27853089ab7Seschrock 	dump_obj(os, spa->spa_feat_for_write_obj, "for_write");
27953089ab7Seschrock 	dump_obj(os, spa->spa_feat_desc_obj, "descriptions");
28043466aaeSMax Grossman 	if (spa_feature_is_active(spa, SPA_FEATURE_ENABLED_TXG)) {
28143466aaeSMax Grossman 		dump_obj(os, spa->spa_feat_enabled_txg_obj, "enabled_txg");
28243466aaeSMax Grossman 	}
28353089ab7Seschrock 	dump_mos(spa);
28453089ab7Seschrock 
28553089ab7Seschrock 	spa_close(spa, FTAG);
28653089ab7Seschrock }
28753089ab7Seschrock 
28853089ab7Seschrock static void
2892acef22dSMatthew Ahrens zhack_feature_enable_sync(void *arg, dmu_tx_t *tx)
29053089ab7Seschrock {
2913b2aab18SMatthew Ahrens 	spa_t *spa = dmu_tx_pool(tx)->dp_spa;
2923b2aab18SMatthew Ahrens 	zfeature_info_t *feature = arg;
29353089ab7Seschrock 
2942acef22dSMatthew Ahrens 	feature_enable_sync(spa, feature, tx);
2952acef22dSMatthew Ahrens 
2964445fffbSMatthew Ahrens 	spa_history_log_internal(spa, "zhack enable feature", tx,
297*ca0cc391SMatthew Ahrens 	    "guid=%s flags=%x",
298*ca0cc391SMatthew Ahrens 	    feature->fi_guid, feature->fi_flags);
29953089ab7Seschrock }
30053089ab7Seschrock 
30153089ab7Seschrock static void
30253089ab7Seschrock zhack_do_feature_enable(int argc, char **argv)
30353089ab7Seschrock {
30453089ab7Seschrock 	char c;
30553089ab7Seschrock 	char *desc, *target;
30653089ab7Seschrock 	spa_t *spa;
30753089ab7Seschrock 	objset_t *mos;
30853089ab7Seschrock 	zfeature_info_t feature;
3092acef22dSMatthew Ahrens 	spa_feature_t nodeps[] = { SPA_FEATURE_NONE };
31053089ab7Seschrock 
31153089ab7Seschrock 	/*
31253089ab7Seschrock 	 * Features are not added to the pool's label until their refcounts
31353089ab7Seschrock 	 * are incremented, so fi_mos can just be left as false for now.
31453089ab7Seschrock 	 */
31553089ab7Seschrock 	desc = NULL;
31653089ab7Seschrock 	feature.fi_uname = "zhack";
317*ca0cc391SMatthew Ahrens 	feature.fi_flags = 0;
31853089ab7Seschrock 	feature.fi_depends = nodeps;
31943466aaeSMax Grossman 	feature.fi_feature = SPA_FEATURE_NONE;
32053089ab7Seschrock 
32153089ab7Seschrock 	optind = 1;
32253089ab7Seschrock 	while ((c = getopt(argc, argv, "rmd:")) != -1) {
32353089ab7Seschrock 		switch (c) {
32453089ab7Seschrock 		case 'r':
325*ca0cc391SMatthew Ahrens 			feature.fi_flags |= ZFEATURE_FLAG_READONLY_COMPAT;
32653089ab7Seschrock 			break;
32753089ab7Seschrock 		case 'd':
32853089ab7Seschrock 			desc = strdup(optarg);
32953089ab7Seschrock 			break;
33053089ab7Seschrock 		default:
33153089ab7Seschrock 			usage();
33253089ab7Seschrock 			break;
33353089ab7Seschrock 		}
33453089ab7Seschrock 	}
33553089ab7Seschrock 
33653089ab7Seschrock 	if (desc == NULL)
33753089ab7Seschrock 		desc = strdup("zhack injected");
33853089ab7Seschrock 	feature.fi_desc = desc;
33953089ab7Seschrock 
34053089ab7Seschrock 	argc -= optind;
34153089ab7Seschrock 	argv += optind;
34253089ab7Seschrock 
34353089ab7Seschrock 	if (argc < 2) {
34453089ab7Seschrock 		(void) fprintf(stderr, "error: missing feature or pool name\n");
34553089ab7Seschrock 		usage();
34653089ab7Seschrock 	}
34753089ab7Seschrock 	target = argv[0];
34853089ab7Seschrock 	feature.fi_guid = argv[1];
34953089ab7Seschrock 
35053089ab7Seschrock 	if (!zfeature_is_valid_guid(feature.fi_guid))
3517fdd916cSGeorge Wilson 		fatal(NULL, FTAG, "invalid feature guid: %s", feature.fi_guid);
35253089ab7Seschrock 
35353089ab7Seschrock 	zhack_spa_open(target, B_FALSE, FTAG, &spa);
35453089ab7Seschrock 	mos = spa->spa_meta_objset;
35553089ab7Seschrock 
3562acef22dSMatthew Ahrens 	if (zfeature_is_supported(feature.fi_guid))
3577fdd916cSGeorge Wilson 		fatal(spa, FTAG, "'%s' is a real feature, will not enable");
35853089ab7Seschrock 	if (0 == zap_contains(mos, spa->spa_feat_desc_obj, feature.fi_guid))
3597fdd916cSGeorge Wilson 		fatal(spa, FTAG, "feature already enabled: %s",
3607fdd916cSGeorge Wilson 		    feature.fi_guid);
36153089ab7Seschrock 
3623b2aab18SMatthew Ahrens 	VERIFY0(dsl_sync_task(spa_name(spa), NULL,
3637d46dc6cSMatthew Ahrens 	    zhack_feature_enable_sync, &feature, 5, ZFS_SPACE_CHECK_NORMAL));
36453089ab7Seschrock 
36553089ab7Seschrock 	spa_close(spa, FTAG);
36653089ab7Seschrock 
36753089ab7Seschrock 	free(desc);
36853089ab7Seschrock }
36953089ab7Seschrock 
37053089ab7Seschrock static void
3713b2aab18SMatthew Ahrens feature_incr_sync(void *arg, dmu_tx_t *tx)
37253089ab7Seschrock {
3733b2aab18SMatthew Ahrens 	spa_t *spa = dmu_tx_pool(tx)->dp_spa;
3743b2aab18SMatthew Ahrens 	zfeature_info_t *feature = arg;
3752acef22dSMatthew Ahrens 	uint64_t refcount;
37653089ab7Seschrock 
37743466aaeSMax Grossman 	VERIFY0(feature_get_refcount_from_disk(spa, feature, &refcount));
3782acef22dSMatthew Ahrens 	feature_sync(spa, feature, refcount + 1, tx);
3794445fffbSMatthew Ahrens 	spa_history_log_internal(spa, "zhack feature incr", tx,
38057221772SChristopher Siden 	    "guid=%s", feature->fi_guid);
38153089ab7Seschrock }
38253089ab7Seschrock 
38353089ab7Seschrock static void
3843b2aab18SMatthew Ahrens feature_decr_sync(void *arg, dmu_tx_t *tx)
38553089ab7Seschrock {
3863b2aab18SMatthew Ahrens 	spa_t *spa = dmu_tx_pool(tx)->dp_spa;
3873b2aab18SMatthew Ahrens 	zfeature_info_t *feature = arg;
3882acef22dSMatthew Ahrens 	uint64_t refcount;
38953089ab7Seschrock 
39043466aaeSMax Grossman 	VERIFY0(feature_get_refcount_from_disk(spa, feature, &refcount));
3912acef22dSMatthew Ahrens 	feature_sync(spa, feature, refcount - 1, tx);
3924445fffbSMatthew Ahrens 	spa_history_log_internal(spa, "zhack feature decr", tx,
39357221772SChristopher Siden 	    "guid=%s", feature->fi_guid);
39453089ab7Seschrock }
39553089ab7Seschrock 
39653089ab7Seschrock static void
39753089ab7Seschrock zhack_do_feature_ref(int argc, char **argv)
39853089ab7Seschrock {
39953089ab7Seschrock 	char c;
40053089ab7Seschrock 	char *target;
40153089ab7Seschrock 	boolean_t decr = B_FALSE;
40253089ab7Seschrock 	spa_t *spa;
40353089ab7Seschrock 	objset_t *mos;
40453089ab7Seschrock 	zfeature_info_t feature;
4052acef22dSMatthew Ahrens 	spa_feature_t nodeps[] = { SPA_FEATURE_NONE };
40653089ab7Seschrock 
40753089ab7Seschrock 	/*
40853089ab7Seschrock 	 * fi_desc does not matter here because it was written to disk
40953089ab7Seschrock 	 * when the feature was enabled, but we need to properly set the
41053089ab7Seschrock 	 * feature for read or write based on the information we read off
41153089ab7Seschrock 	 * disk later.
41253089ab7Seschrock 	 */
41353089ab7Seschrock 	feature.fi_uname = "zhack";
414*ca0cc391SMatthew Ahrens 	feature.fi_flags = 0;
41553089ab7Seschrock 	feature.fi_desc = NULL;
41653089ab7Seschrock 	feature.fi_depends = nodeps;
41743466aaeSMax Grossman 	feature.fi_feature = SPA_FEATURE_NONE;
41853089ab7Seschrock 
41953089ab7Seschrock 	optind = 1;
42053089ab7Seschrock 	while ((c = getopt(argc, argv, "md")) != -1) {
42153089ab7Seschrock 		switch (c) {
42253089ab7Seschrock 		case 'm':
423*ca0cc391SMatthew Ahrens 			feature.fi_flags |= ZFEATURE_FLAG_MOS;
42453089ab7Seschrock 			break;
42553089ab7Seschrock 		case 'd':
42653089ab7Seschrock 			decr = B_TRUE;
42753089ab7Seschrock 			break;
42853089ab7Seschrock 		default:
42953089ab7Seschrock 			usage();
43053089ab7Seschrock 			break;
43153089ab7Seschrock 		}
43253089ab7Seschrock 	}
43353089ab7Seschrock 	argc -= optind;
43453089ab7Seschrock 	argv += optind;
43553089ab7Seschrock 
43653089ab7Seschrock 	if (argc < 2) {
43753089ab7Seschrock 		(void) fprintf(stderr, "error: missing feature or pool name\n");
43853089ab7Seschrock 		usage();
43953089ab7Seschrock 	}
44053089ab7Seschrock 	target = argv[0];
44153089ab7Seschrock 	feature.fi_guid = argv[1];
44253089ab7Seschrock 
44353089ab7Seschrock 	if (!zfeature_is_valid_guid(feature.fi_guid))
4447fdd916cSGeorge Wilson 		fatal(NULL, FTAG, "invalid feature guid: %s", feature.fi_guid);
44553089ab7Seschrock 
44653089ab7Seschrock 	zhack_spa_open(target, B_FALSE, FTAG, &spa);
44753089ab7Seschrock 	mos = spa->spa_meta_objset;
44853089ab7Seschrock 
4492acef22dSMatthew Ahrens 	if (zfeature_is_supported(feature.fi_guid)) {
4502acef22dSMatthew Ahrens 		fatal(spa, FTAG,
4512acef22dSMatthew Ahrens 		    "'%s' is a real feature, will not change refcount");
4522acef22dSMatthew Ahrens 	}
45353089ab7Seschrock 
45453089ab7Seschrock 	if (0 == zap_contains(mos, spa->spa_feat_for_read_obj,
45553089ab7Seschrock 	    feature.fi_guid)) {
456*ca0cc391SMatthew Ahrens 		feature.fi_flags &= ~ZFEATURE_FLAG_READONLY_COMPAT;
45753089ab7Seschrock 	} else if (0 == zap_contains(mos, spa->spa_feat_for_write_obj,
45853089ab7Seschrock 	    feature.fi_guid)) {
459*ca0cc391SMatthew Ahrens 		feature.fi_flags |= ZFEATURE_FLAG_READONLY_COMPAT;
46053089ab7Seschrock 	} else {
4617fdd916cSGeorge Wilson 		fatal(spa, FTAG, "feature is not enabled: %s", feature.fi_guid);
46253089ab7Seschrock 	}
46353089ab7Seschrock 
4642acef22dSMatthew Ahrens 	if (decr) {
4652acef22dSMatthew Ahrens 		uint64_t count;
46643466aaeSMax Grossman 		if (feature_get_refcount_from_disk(spa, &feature,
46743466aaeSMax Grossman 		    &count) == 0 && count != 0) {
4682acef22dSMatthew Ahrens 			fatal(spa, FTAG, "feature refcount already 0: %s",
4692acef22dSMatthew Ahrens 			    feature.fi_guid);
4702acef22dSMatthew Ahrens 		}
4712acef22dSMatthew Ahrens 	}
47253089ab7Seschrock 
4733b2aab18SMatthew Ahrens 	VERIFY0(dsl_sync_task(spa_name(spa), NULL,
4747d46dc6cSMatthew Ahrens 	    decr ? feature_decr_sync : feature_incr_sync, &feature,
4757d46dc6cSMatthew Ahrens 	    5, ZFS_SPACE_CHECK_NORMAL));
47653089ab7Seschrock 
47753089ab7Seschrock 	spa_close(spa, FTAG);
47853089ab7Seschrock }
47953089ab7Seschrock 
48053089ab7Seschrock static int
48153089ab7Seschrock zhack_do_feature(int argc, char **argv)
48253089ab7Seschrock {
48353089ab7Seschrock 	char *subcommand;
48453089ab7Seschrock 
48553089ab7Seschrock 	argc--;
48653089ab7Seschrock 	argv++;
48753089ab7Seschrock 	if (argc == 0) {
48853089ab7Seschrock 		(void) fprintf(stderr,
48953089ab7Seschrock 		    "error: no feature operation specified\n");
49053089ab7Seschrock 		usage();
49153089ab7Seschrock 	}
49253089ab7Seschrock 
49353089ab7Seschrock 	subcommand = argv[0];
49453089ab7Seschrock 	if (strcmp(subcommand, "stat") == 0) {
49553089ab7Seschrock 		zhack_do_feature_stat(argc, argv);
49653089ab7Seschrock 	} else if (strcmp(subcommand, "enable") == 0) {
49753089ab7Seschrock 		zhack_do_feature_enable(argc, argv);
49853089ab7Seschrock 	} else if (strcmp(subcommand, "ref") == 0) {
49953089ab7Seschrock 		zhack_do_feature_ref(argc, argv);
50053089ab7Seschrock 	} else {
50153089ab7Seschrock 		(void) fprintf(stderr, "error: unknown subcommand: %s\n",
50253089ab7Seschrock 		    subcommand);
50353089ab7Seschrock 		usage();
50453089ab7Seschrock 	}
50553089ab7Seschrock 
50653089ab7Seschrock 	return (0);
50753089ab7Seschrock }
50853089ab7Seschrock 
50953089ab7Seschrock #define	MAX_NUM_PATHS 1024
51053089ab7Seschrock 
51153089ab7Seschrock int
51253089ab7Seschrock main(int argc, char **argv)
51353089ab7Seschrock {
51453089ab7Seschrock 	extern void zfs_prop_init(void);
51553089ab7Seschrock 
51653089ab7Seschrock 	char *path[MAX_NUM_PATHS];
51753089ab7Seschrock 	const char *subcommand;
51853089ab7Seschrock 	int rv = 0;
51953089ab7Seschrock 	char c;
52053089ab7Seschrock 
52153089ab7Seschrock 	g_importargs.path = path;
52253089ab7Seschrock 
52353089ab7Seschrock 	dprintf_setup(&argc, argv);
52453089ab7Seschrock 	zfs_prop_init();
52553089ab7Seschrock 
52653089ab7Seschrock 	while ((c = getopt(argc, argv, "c:d:")) != -1) {
52753089ab7Seschrock 		switch (c) {
52853089ab7Seschrock 		case 'c':
52953089ab7Seschrock 			g_importargs.cachefile = optarg;
53053089ab7Seschrock 			break;
53153089ab7Seschrock 		case 'd':
53253089ab7Seschrock 			assert(g_importargs.paths < MAX_NUM_PATHS);
53353089ab7Seschrock 			g_importargs.path[g_importargs.paths++] = optarg;
53453089ab7Seschrock 			break;
53553089ab7Seschrock 		default:
53653089ab7Seschrock 			usage();
53753089ab7Seschrock 			break;
53853089ab7Seschrock 		}
53953089ab7Seschrock 	}
54053089ab7Seschrock 
54153089ab7Seschrock 	argc -= optind;
54253089ab7Seschrock 	argv += optind;
54353089ab7Seschrock 	optind = 1;
54453089ab7Seschrock 
54553089ab7Seschrock 	if (argc == 0) {
54653089ab7Seschrock 		(void) fprintf(stderr, "error: no command specified\n");
54753089ab7Seschrock 		usage();
54853089ab7Seschrock 	}
54953089ab7Seschrock 
55053089ab7Seschrock 	subcommand = argv[0];
55153089ab7Seschrock 
55253089ab7Seschrock 	if (strcmp(subcommand, "feature") == 0) {
55353089ab7Seschrock 		rv = zhack_do_feature(argc, argv);
55453089ab7Seschrock 	} else {
55553089ab7Seschrock 		(void) fprintf(stderr, "error: unknown subcommand: %s\n",
55653089ab7Seschrock 		    subcommand);
55753089ab7Seschrock 		usage();
55853089ab7Seschrock 	}
55953089ab7Seschrock 
5607fdd916cSGeorge Wilson 	if (!g_readonly && spa_export(g_pool, NULL, B_TRUE, B_FALSE) != 0) {
5617fdd916cSGeorge Wilson 		fatal(NULL, FTAG, "pool export failed; "
56253089ab7Seschrock 		    "changes may not be committed to disk\n");
56353089ab7Seschrock 	}
56453089ab7Seschrock 
56553089ab7Seschrock 	libzfs_fini(g_zfs);
56653089ab7Seschrock 	kernel_fini();
56753089ab7Seschrock 
56853089ab7Seschrock 	return (rv);
56953089ab7Seschrock }
570