/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright (c) 2021 Daniel Austin <freebsd-ports@dan.me.uk>
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/param.h>
#include <sys/errno.h>
#include <err.h>
#include <libutil.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "mpsutil.h"

static int set_ncq(int ac, char **av);

MPS_TABLE(top, set);

static int
set_ncq(int ac, char **av)
{
	MPI2_CONFIG_PAGE_HEADER header;
	MPI2_CONFIG_PAGE_IO_UNIT_1 *iounit1;
	MPI2_CONFIG_REQUEST req;
	MPI2_CONFIG_REPLY reply;
	int error, fd;

	bzero(&req, sizeof(req));
	bzero(&header, sizeof(header));
	bzero(&reply, sizeof(reply));

	fd = mps_open(mps_unit);
	if (fd < 0) {
		error = errno;
		warn("mps_open");
		return (error);
	}

	error = mps_read_config_page_header(fd, MPI2_CONFIG_PAGETYPE_IO_UNIT, 1, 0,
		&header, NULL);
	if (error) {
			error = errno;
			warn("Failed to get IOUNIT page 1 header");
			return (error);
	}

	iounit1 = mps_read_config_page(fd, MPI2_CONFIG_PAGETYPE_IO_UNIT, 1, 0, NULL);
	if (iounit1 == NULL) {
		error = errno;
		warn("Failed to get IOUNIT page 1 info");
		return (error);
	}

	if (ac == 1) {
		/* just show current setting */
		printf("SATA Native Command Queueing is currently: %s\n",
			((iounit1->Flags & MPI2_IOUNITPAGE1_NATIVE_COMMAND_Q_DISABLE) == 0) ?
			"ENABLED" : "DISABLED");
	} else if (ac == 2) {
		if (!strcasecmp(av[1], "enable") || !strcmp(av[1], "1")) {
			iounit1->Flags &= ~MPI2_IOUNITPAGE1_NATIVE_COMMAND_Q_DISABLE;
		} else if (!strcasecmp(av[1], "disable") || !strcmp(av[1], "0")) {
			iounit1->Flags |= MPI2_IOUNITPAGE1_NATIVE_COMMAND_Q_DISABLE;
		} else {
			free(iounit1);
			error = EINVAL;
			warn("set ncq: Only 'enable' and 'disable' allowed.");
			return (EINVAL);
		}
		req.Function = MPI2_FUNCTION_CONFIG;
		req.Action = MPI2_CONFIG_ACTION_PAGE_WRITE_CURRENT;
		req.ExtPageLength = 0;
		req.ExtPageType = 0;
		req.Header = header;
		req.PageAddress = 0;
		if (mps_pass_command(fd, &req, sizeof(req) - sizeof(req.PageBufferSGE), &reply, sizeof(reply),
			NULL, 0, iounit1, sizeof(iounit1), 30) != 0) {
				free(iounit1);
				error = errno;
				warn("Failed to update config page");
		                return (error);
		}
		if (!IOC_STATUS_SUCCESS(reply.IOCStatus)) {
			free(iounit1);
			error = errno;
			warn("%s", mps_ioc_status(reply.IOCStatus));
			return (error);
		}
		printf("NCQ setting accepted.  It may not take effect until the controller is reset.\n");
	} else {
		free(iounit1);
		errno = EINVAL;
		warn("set ncq: too many arguments");
		return (EINVAL);
	}
	free(iounit1);

	close(fd);
	return (0);
}

MPS_COMMAND(set, ncq, set_ncq, "[enable|disable]", "set SATA NCQ function")