/*
 * 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.
 */

/*
 * Copyright 2024 Oxide Computer Company
 */

/*
 * Perform basic NVMe identify tests on a device. This goes through and issues
 * an identify controller on a device and then attempts to identify each
 * namespace present. If namespace management is present, then attempt to get
 * the generic namespace information. We only attempt to get namespace
 * information via the namespace fd for ns 1.
 */

#include <err.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/stdbool.h>

#include "nvme_ioctl_util.h"

static void
basic_id_determine_ns_sup(const void *data, uint32_t *nnsp, bool *hasns)
{
	const nvme_identify_ctrl_t *ctrl = data;
	*nnsp = ctrl->id_nn;
	*hasns = ctrl->id_oacs.oa_nsmgmt != 0;
}

int
main(void)
{
	int ret = EXIT_SUCCESS;
	int fd = nvme_ioctl_test_get_fd(0);
	nvme_ioctl_identify_t id;
	void *data;
	bool hasns;
	uint32_t nns;

	if ((data = malloc(NVME_IDENTIFY_BUFSIZE)) == NULL) {
		err(EXIT_FAILURE, "TEST FAILED: could not initialize identify "
		    "data buffer");
	}

	(void) memset(&id, 0, sizeof (id));
	id.nid_cns = NVME_IDENTIFY_CTRL;
	id.nid_data = (uintptr_t)data;

	if (ioctl(fd, NVME_IOC_IDENTIFY, &id) != 0) {
		err(EXIT_FAILURE, "TEST FAILED: cannot proceed with tests "
		    "due to failure to issue basic identify controller ioctl");
	} else if (id.nid_common.nioc_drv_err != NVME_IOCTL_E_OK) {
		errx(EXIT_FAILURE, "TEST FAILED: cannot proceed with tests "
		    "due to failure to obtain basic identify controller data, "
		    "got 0x%x", id.nid_common.nioc_drv_err);
	}

	basic_id_determine_ns_sup(data, &nns, &hasns);
	if (nns == 0) {
		errx(EXIT_FAILURE, "TEST FAILED: somehow identified that "
		    "zero namespaces exist");
	}
	(void) printf("TEST PASSED: successfully retrieved identify controller "
	    "data\n");

	for (uint32_t i = 1; i < nns; i++) {
		(void) memset(&id, 0, sizeof (id));
		id.nid_common.nioc_nsid = i;
		id.nid_cns = NVME_IDENTIFY_NSID;
		id.nid_data = (uintptr_t)data;

		if (ioctl(fd, NVME_IOC_IDENTIFY, &id) != 0) {
			warn("TEST FAILED: failed to issue identify namespace "
			    "0x%x ioctl", i);
			ret = EXIT_FAILURE;
		} else if (id.nid_common.nioc_drv_err != NVME_IOCTL_E_OK) {
			warnx("TEST FAILED: failed to obtain identify "
			    "namespace data for ns 0x%x, got error 0x%x", i,
			    id.nid_common.nioc_drv_err);
			ret = EXIT_FAILURE;
		} else {
			(void) printf("TEST PASSED: retrieved identify "
			    "namespace 0x%x data\n", i);
		}
	}

	if (hasns) {
		(void) memset(&id, 0, sizeof (id));
		id.nid_common.nioc_nsid = NVME_NSID_BCAST;
		id.nid_cns = NVME_IDENTIFY_NSID;
		id.nid_data = (uintptr_t)data;

		if (ioctl(fd, NVME_IOC_IDENTIFY, &id) != 0) {
			warn("TEST FAILED: failed to issue identify common "
			    "namespace ioctl");
			ret = EXIT_FAILURE;
		} else if (id.nid_common.nioc_drv_err != NVME_IOCTL_E_OK) {
			warnx("TEST FAILED: failed to obtain common identify "
			    "namespace data, got error 0x%x",
			    id.nid_common.nioc_drv_err);
			ret = EXIT_FAILURE;
		} else {
			(void) printf("TEST PASSED: retrieved common identify "
			    "namespace data\n");
		}
	}

	VERIFY0(close(fd));
	fd = nvme_ioctl_test_get_fd(1);
	(void) memset(&id, 0, sizeof (id));
	id.nid_cns = NVME_IDENTIFY_NSID;
	id.nid_data = (uintptr_t)data;

	if (ioctl(fd, NVME_IOC_IDENTIFY, &id) != 0) {
		warn("TEST FAILED: failed to issue identify namespace ioctl "
		    "on ns fd");
		ret = EXIT_FAILURE;
	} else if (id.nid_common.nioc_drv_err != NVME_IOCTL_E_OK) {
		warnx("TEST FAILED: failed to obtain identify namespace data "
		    "on ns fd, got error 0x%x", id.nid_common.nioc_drv_err);
		ret = EXIT_FAILURE;
	} else {
		(void) printf("TEST PASSED: retrieved identify namespace data"
		    "on ns fd\n");
	}
	VERIFY0(close(fd));

	return (ret);
}