/*      $NetBSD: wdc.c,v 1.6 2023/07/05 10:58:46 riastradh Exp $        */

/*-
 * Copyright (c) 2017 Netflix, Inc
 * All rights reserved.
 *
 * 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/cdefs.h>
#ifndef lint
__RCSID("$NetBSD: wdc.c,v 1.6 2023/07/05 10:58:46 riastradh Exp $");
#if 0
__FBSDID("$FreeBSD: head/sbin/nvmecontrol/wdc.c 329824 2018-02-22 13:32:31Z wma $");
#endif
#endif

#include <sys/param.h>
#include <sys/ioccom.h>
#include <sys/endian.h>

#include <ctype.h>
#include <err.h>
#include <fcntl.h>
#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "nvmectl.h"

#define WDC_NVME_TOC_SIZE       8

#define WDC_NVME_CAP_DIAG_OPCODE        0xe6
#define WDC_NVME_CAP_DIAG_CMD           0x0000

static void wdc_cap_diag(int argc, char *argv[]);

#define WDC_CAP_DIAG_USAGE      "wdc cap-diag [-o path-template]\n"

static const struct nvme_function wdc_funcs[] = {
        {"cap-diag",            wdc_cap_diag,           WDC_CAP_DIAG_USAGE},
        {NULL,                  NULL,                   NULL},
};

static void
wdc_append_serial_name(int fd, char *buf, size_t len, const char *suffix)
{
        struct nvm_identify_controller  cdata;
        char sn[sizeof(cdata.sn) + 1];
        char *walker;

        len -= strlen(buf);
        buf += strlen(buf);
        read_controller_data(fd, &cdata);
        memcpy(sn, cdata.sn, sizeof(cdata.sn));
        walker = sn + sizeof(cdata.sn) - 1;
        while (walker > sn && *walker == ' ')
                walker--;
        *++walker = '\0';
        snprintf(buf, len, "%s%s.bin", sn, suffix);
}

static void
wdc_get_data(int fd, uint32_t opcode, uint32_t len, uint32_t off, uint32_t cmd,
    uint8_t *buffer, size_t buflen)
{
        struct nvme_pt_command  pt;

        memset(&pt, 0, sizeof(pt));
        pt.cmd.opcode = opcode;
        pt.cmd.cdw10 = len / sizeof(uint32_t);  /* - 1 like all the others ??? */
        pt.cmd.cdw11 = off / sizeof(uint32_t);
        pt.cmd.cdw12 = cmd;
        pt.buf = buffer;
        pt.len = buflen;
        pt.is_read = 1;
//      printf("opcode %#x cdw10(len) %#x cdw11(offset?) %#x cdw12(cmd/sub) %#x buflen %zd\n",
//          (int)opcode, (int)cdw10, (int)cdw11, (int)cdw12, buflen);

        if (ioctl(fd, NVME_PASSTHROUGH_CMD, &pt) < 0)
                err(1, "wdc_get_data request failed");
        if (nvme_completion_is_error(&pt.cpl))
                errx(1, "wdc_get_data request returned error");
}

static void
wdc_do_dump(int fd, char *tmpl, const char *suffix, uint32_t opcode,
    uint32_t cmd, int len_off)
{
        int first;
        int fd2;
        uint8_t *buf;
        uint32_t len, offset;
        size_t resid;
        long page_size;

        page_size = sysconf(_SC_PAGESIZE);
        if (page_size <= 0)
                page_size = 4096;

        wdc_append_serial_name(fd, tmpl, MAXPATHLEN, suffix);

        /* XXX overwrite protection? */
        fd2 = open(tmpl, O_WRONLY | O_CREAT | O_TRUNC, 0644);
        if (fd2 < 0)
                err(1, "open %s", tmpl);
        buf = aligned_alloc(page_size, NVME_MAX_XFER_SIZE);
        if (buf == NULL)
                errx(1, "Can't get buffer to read dump");
        offset = 0;
        len = NVME_MAX_XFER_SIZE;
        first = 1;

        do {
                resid = len > NVME_MAX_XFER_SIZE ? NVME_MAX_XFER_SIZE : len;
                wdc_get_data(fd, opcode, resid, offset, cmd, buf, resid);
                if (first) {
                        len = be32dec(buf + len_off);
                        if (len == 0)
                                errx(1, "No data for %s", suffix);
                        if (memcmp("E6LG", buf, 4) != 0)
                                printf("Expected header of E6LG, found '%4.4s' instead\n",
                                    buf);
                        printf("Dumping %d bytes of version %d.%d log to %s\n", len,
                            buf[8], buf[9], tmpl);
                        /*
                         * Adjust amount to dump if total dump < 1MB,
                         * though it likely doesn't matter to the WDC
                         * analysis tools.
                         */
                        if (resid > len)
                                resid = len;
                        first = 0;
                }
                if (write(fd2, buf, resid) != (ssize_t)resid)
                        err(1, "write");
                offset += resid;
                len -= resid;
        } while (len > 0);
        free(buf);
        close(fd2);
}

__dead static void
wdc_cap_diag_usage(void)
{
        fprintf(stderr, "usage:\n");
        fprintf(stderr, "\t%s " WDC_CAP_DIAG_USAGE, getprogname());
        exit(1);
}

__dead static void
wdc_cap_diag(int argc, char *argv[])
{
        char path_tmpl[MAXPATHLEN];
        int ch, fd;

        path_tmpl[0] = '\0';
        while ((ch = getopt(argc, argv, "o:")) != -1) {
                switch ((char)ch) {
                case 'o':
                        strlcpy(path_tmpl, optarg, MAXPATHLEN);
                        break;
                default:
                        wdc_cap_diag_usage();
                }
        }
        /* Check that a controller was specified. */
        if (optind >= argc)
                wdc_cap_diag_usage();
        open_dev(argv[optind], &fd, 1, 1);

        wdc_do_dump(fd, path_tmpl, "cap_diag", WDC_NVME_CAP_DIAG_OPCODE,
            WDC_NVME_CAP_DIAG_CMD, 4);

        close(fd);

        exit(1);        
}

void
wdc(int argc, char *argv[])
{

        dispatch(argc, argv, wdc_funcs);
}