xref: /illumos-gate/usr/src/test/zfs-tests/tests/functional/channel_program/synctask_core/change_key.c (revision d8f839f9)

/*
 * CDDL HEADER START
 *
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 *
 * CDDL HEADER END
 */

/*
 * Copyright 2020 Joyent, Inc.
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/debug.h>
#include <sys/fs/zfs.h>
#include <libzfs_core.h>
#include <libnvpair.h>

const char prog[] =
	"arg = ... \n"
	"fs = arg[\"dataset\"]\n"
	"hexkey = arg[\"" ZPOOL_HIDDEN_ARGS "\"][\"key\"]\n"
	"err = zfs.sync.change_key(fs, hexkey, 'hex')\n"
	"msg = \"changing key on \" .. fs .. \" err=\" .. err\n"
	"return msg";

/*
 * Get the pool name from a dataset. This is crude but good enough
 * for a test.
 */
static char *
get_pool(const char *dataset)
{
	char *res = strdup(dataset);

	if (res == NULL)
		abort();

	char *p = strchr(res, '/');

	if (p != NULL)
		*p = '\0';

	return (res);
}

int
main(int argc, char *argv[])
{
	const char *dataset = argv[1];
	const char *key = argv[2];
	char *pool = NULL;
	nvlist_t *args = fnvlist_alloc();
	nvlist_t *hidden_args = fnvlist_alloc();
	nvlist_t *result = NULL;
	int ret = 0;

	if (argc != 3) {
		(void) fprintf(stderr, "Usage: %s dataset key\n", argv[0]);
		exit(2);
	}

	VERIFY0(libzfs_core_init());

	pool = get_pool(dataset);

	fnvlist_add_string(args, "dataset", dataset);
	fnvlist_add_string(hidden_args, "key", key);
	fnvlist_add_nvlist(args, ZPOOL_HIDDEN_ARGS, hidden_args);

	ret = lzc_channel_program(pool, prog, ZCP_DEFAULT_INSTRLIMIT,
	    ZCP_DEFAULT_MEMLIMIT, args, &result);

	(void) printf("lzc_channel_program returned %d", ret);
	if (ret != 0)
		(void) printf(" (%s)", strerror(ret));
	(void) fputc('\n', stdout);

	dump_nvlist(result, 5);

	nvlist_free(args);
	nvlist_free(hidden_args);
	nvlist_free(result);
	free(pool);

	libzfs_core_fini();

	return (ret);
}