rockbox/utils/nwztools/scsitools/scsitool.c

/***************************************************************************
 *             __________               __   ___.
 *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
 *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
 *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
 *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
 *                     \/            \/     \/    \/            \/
 * $Id$
 *
 * Copyright (C) 2012 Amaury Pouly
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
 * KIND, either express or implied.
 *
 ****************************************************************************/
#include <stdio.h>
#include <stdint.h>
#include <stdbool.h>
#include <stdlib.h>
#include <stddef.h>
#include <string.h>
#include <getopt.h>
#include <stdarg.h>
#include <ctype.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#ifndef _WIN32
#include <scsi/scsi.h>
#endif
#include <scsi/sg_lib.h>
#include <scsi/sg_pt.h>
#include "misc.h"
#include "para_noise.h"

/* the windows port doesn't have scsi.h and GOOD */
#ifndef GOOD
#define GOOD                 0x00
#endif

bool g_debug = false;
bool g_force = false;
char *g_out_prefix = NULL;
int g_dev_fd = 0;

#define let_the_force_flow(x) do { if(!g_force) return x; } while(0)
#define continue_the_force(x) if(x) let_the_force_flow(x)

#define check_field(v_exp, v_have, str_ok, str_bad) \
    if((v_exp) != (v_have)) \
    { cprintf(RED, str_bad); let_the_force_flow(__LINE__); } \
    else { cprintf(RED, str_ok); }

#define errorf(...) do { cprintf(GREY, __VA_ARGS__); return __LINE__; } while(0)

#if 0
void *buffer_alloc(int sz)
{
#ifdef SG_LIB_MINGW
    unsigned psz = getpagesize();
#else
    unsigned psz = sysconf(_SC_PAGESIZE); /* was getpagesize() */
#endif
    void *buffer = malloc(sz + psz);
    return (void *)(((ptrdiff_t)(buffer + psz - 1)) & ~(psz - 1));
}
#else
void *buffer_alloc(int sz)
{
    return malloc(sz);
}
#endif

static void print_hex(void *_buffer, int buffer_size)
{
    uint8_t *buffer = _buffer;
    for(int i = 0; i < buffer_size; i += 16)
    {
        for(int j = 0; j < 16; j++)
        {
            if(i + j < buffer_size)
                cprintf(YELLOW, " %02x", buffer[i + j]);
            else
                cprintf(YELLOW, "   ");
        }
        printf(" ");
        for(int j = 0; j < 16; j++)
        {
            if(i + j < buffer_size)
                cprintf(RED, "%c", isprint(buffer[i + j]) ? buffer[i + j] : '.');
            else
                cprintf(RED, " ");
        }
        printf("\n");
    }
}

/* Do read */
#define DO_READ     (1 << 1)
/* Do write */
#define DO_WRITE    (1 << 2)

/* returns <0 on error and status otherwise */
int do_scsi(uint8_t *cdb, int cdb_size, unsigned flags, void *sense, int *sense_size, void *buffer, int *buf_size)
{
    char error[256];
    struct sg_pt_base *obj = construct_scsi_pt_obj();
    if(obj == NULL)
    {
        cprintf(GREY, "construct_scsi_pt_obj failed\n");
        return 1;
    }
    set_scsi_pt_cdb(obj, cdb, cdb_size);
    if(sense)
        set_scsi_pt_sense(obj, sense, *sense_size);
    if(flags & DO_READ)
        set_scsi_pt_data_in(obj, buffer, *buf_size);
    if(flags & DO_WRITE)
        set_scsi_pt_data_out(obj, buffer, *buf_size);
    int ret = do_scsi_pt(obj, g_dev_fd, 1, 0);
    switch(get_scsi_pt_result_category(obj))
    {
        case SCSI_PT_RESULT_SENSE:
        case SCSI_PT_RESULT_GOOD:
            ret = get_scsi_pt_status_response(obj);
            break;
        case SCSI_PT_RESULT_STATUS:
            cprintf(GREY, "Status error: %d (", get_scsi_pt_status_response(obj));
            sg_print_scsi_status(get_scsi_pt_status_response(obj));
            printf(")\n");
            break;
        case SCSI_PT_RESULT_TRANSPORT_ERR:
            cprintf(GREY, "Transport error: %s\n", get_scsi_pt_transport_err_str(obj, 256, error));
            ret = -2;
            break;
        case SCSI_PT_RESULT_OS_ERR:
            cprintf(GREY, "OS error: %s\n", get_scsi_pt_os_err_str(obj, 256, error));
            ret = -3;
            break;
        default:
            cprintf(GREY, "Unknown error\n");
            break;
    }

    if(sense)
        *sense_size = get_scsi_pt_sense_len(obj);
    if(flags & (DO_WRITE | DO_READ))
        *buf_size -= get_scsi_pt_resid(obj);
    
    destruct_scsi_pt_obj(obj);
    return ret;
}

int do_sense_analysis(int status, uint8_t *sense, int sense_size)
{
    if(status != GOOD || g_debug)
    {
        cprintf_field("Status:", " "); fflush(stdout);
        sg_print_scsi_status(status);
        cprintf_field("\nSense:", " "); fflush(stdout);
        sg_print_sense(NULL, sense, sense_size, 0);
    }
    if(status == GOOD)
        return 0;
    return status;
}

int do_dnk_cmd(uint32_t cmd, uint8_t sub_cmd, uint16_t arg, void *buffer, int *buffer_size)
{
    uint8_t cdb[12] = {0xdd, 0, 0, 0, 0, 0, 0, 0xbc, 0, 0, 0, 0};
    cdb[10] = cmd;
    cdb[11] = sub_cmd;
    cdb[8] = (*buffer_size) >> 8;
    cdb[9] = (*buffer_size) & 0xff;
    cdb[4] = (arg >> 8) & 0xff;
    cdb[5] = arg & 0xff;
    
    uint8_t sense[32];
    int sense_size = 32;
    
    int ret = do_scsi(cdb, 12, DO_READ, sense, &sense_size, buffer, buffer_size);
    if(ret < 0)
        return ret;
    ret = do_sense_analysis(ret, sense, sense_size);
    if(ret)
        return ret;
    return 0;
}

#define DNK_EXACT_LENGTH    (1 << 0)
#define DNK_STRING          (1 << 1)
#define DNK_UINT32          (1 << 2)

struct dnk_prop_t
{
    char *name;
    uint8_t cmd;
    uint8_t subcmd;
    int size;
    unsigned flags;
};

struct dnk_prop_t dnk_prop_list[] =
{
    { "serial_num", 0x23, 1, 8, DNK_STRING},
    { "model_id", 0x23, 4, 4, DNK_EXACT_LENGTH | DNK_UINT32},
    { "product_id", 0x23, 6, 12, DNK_STRING},
    { "destination", 0x23, 8, 4, DNK_EXACT_LENGTH | DNK_UINT32},
    { "model_id2", 0x23, 9, 4, DNK_EXACT_LENGTH | DNK_UINT32},
    { "dev_info", 0x12, 0, 64, DNK_STRING},
};

#define NR_DNK_PROPS   (sizeof(dnk_prop_list) / sizeof(dnk_prop_list[0]))

int get_dnk_prop(int argc, char **argv)
{
    if(argc != 1 && argc != 4)
    {
        printf("You must specify a known property name or a full property specification:\n");
        printf("Full usage: <cmd> <subcmd> <size> <flags>\n");
        printf("Property usage: <prop>\n");
        printf("Properties:");
        for(unsigned i = 0; i < NR_DNK_PROPS; i++)
            printf(" %s", dnk_prop_list[i].name);
        printf("\n");
        return 1;
    }

    struct dnk_prop_t prop;
    memset(&prop, 0, sizeof(prop));
    if(argc == 1)
    {
        for(unsigned i = 0; i < NR_DNK_PROPS; i++)
            if(strcmp(dnk_prop_list[i].name, argv[0]) == 0)
                prop = dnk_prop_list[i];
        if(prop.name == NULL)
        {
            cprintf(GREY, "Unknown property '%s'\n", argv[0]);
            return 1;
        }
    }
    else
    {
        prop.cmd = strtoul(argv[0], NULL, 0);
        prop.subcmd = strtoul(argv[1], NULL, 0);
        prop.size = strtoul(argv[2], NULL, 0);
        prop.flags = strtoul(argv[3], NULL, 0);
    }

    char *buffer = buffer_alloc(prop.size + 1);
    int buffer_size = prop.size;
    int ret = do_dnk_cmd(prop.cmd, prop.subcmd, 0, buffer, &buffer_size);
    if(ret)
        return ret;
    if(buffer_size == 0)
    {
        cprintf(GREY, "Device didn't send any data\n");
        return 1;
    }
    if((prop.flags & DNK_EXACT_LENGTH) && buffer_size != prop.size)
    {
        cprintf(GREY, "Device didn't send the expected amount of data\n");
        return 2;
    }
    buffer[buffer_size] = 0;
    if(prop.flags & DNK_STRING)
        cprintf_field("Property: ", "%s\n", buffer);
    else if(prop.flags & DNK_UINT32)
        cprintf_field("Property: ", "0x%x\n", *(uint32_t *)buffer);
    else
    {
        cprintf(GREEN, "Property:\n");
        print_hex(buffer, buffer_size);
    }
    return 0;
}

#define NVP_EXACT_LENGTH    (1 << 0)
#define NVP_STRING          (1 << 1)
#define NVP_UINT32          (1 << 2)

struct nvp_prop_t
{
    const char *name;
    const char *desc;
    uint8_t node;
    int size;
    unsigned flags;
};

#define NVP_ENTRY(node, name,desc,size,flag) { name, desc, node, size, flag }

struct nvp_prop_t nvp_prop_list[] =
{
    NVP_ENTRY(1, "bti", "boot image", 262144, 0),
    NVP_ENTRY(2, "hdi", "hold image", 262144, 0),
    NVP_ENTRY(3, "cng", "aad key", 704, 0),
    NVP_ENTRY(4, "ser", "serial number", 16, NVP_STRING),
    NVP_ENTRY(5, "app", "application parameter", 4096, 0),
    NVP_ENTRY(6, "eri", "update error image", 262144, 0),
    NVP_ENTRY(7, "dcc", "secure clock", 20, 0),
    NVP_ENTRY(8, "mdl", "middleware parameter", 8, 0),
    NVP_ENTRY(9, "fup", "firmware update flag", 4, 0),
    NVP_ENTRY(10, "bok", "beep ok flag", 4, 0),
    NVP_ENTRY(11, "shp", "ship information", 32, 0),
    NVP_ENTRY(12, "dba", "aad icv", 160, 0),
    NVP_ENTRY(13, "dbv", "empr key", 520, 0),
    NVP_ENTRY(14, "tr0", "EKB 0", 16384, 0),
    NVP_ENTRY(15, "tr1", "EKB 1", 16384, 0),
    NVP_ENTRY(16, "mid", "model id", 64, 0),
    NVP_ENTRY(17, "tst", "test mode flag", 4, 0),
    NVP_ENTRY(18, "gty", "getty mode flag", 4, 0),
    NVP_ENTRY(19, "fui", "update image", 262144, 0),
    NVP_ENTRY(20, "lbi", "low battery image", 262144, 0),
    NVP_ENTRY(21, "dor", "key mode (debug/release)", 4, 0),
    NVP_ENTRY(22, "edw", "quick shutdown flag", 4, 0),
    NVP_ENTRY(23, "ubp", "u-boot password", 32, 0),
    NVP_ENTRY(24, "syi", "system information", 4, 0),
    NVP_ENTRY(25, "exm", "exception monitor mode", 4, 0),
    NVP_ENTRY(26, "pcd", "product code", 5, 0),
    NVP_ENTRY(27, "btc", "battery calibration", 4, 0),
    NVP_ENTRY(28, "rnd", "wmt key", 64, 0),
    NVP_ENTRY(29, "ufn", "update file name", 8, 0),
    NVP_ENTRY(30, "sdp", "sound driver parameter", 64, 0),
    NVP_ENTRY(31, "ncp", "noise cancel driver parameter", 64, 0),
    NVP_ENTRY(32, "kas", "key and signature", 64, 0),
    NVP_ENTRY(33, "sfi", "starfish id", 64, 0),
    NVP_ENTRY(34, "rtc", "rtc alarm", 16, 0),
    NVP_ENTRY(35, "bpr", "bluetooth address", 2048, 0),
    NVP_ENTRY(36, "e00", "EMPR  0", 1024, 0),
    NVP_ENTRY(37, "e01", "EMPR  1", 1024, 0),
    NVP_ENTRY(38, "e02", "EMPR  2", 1024, 0),
    NVP_ENTRY(39, "e03", "EMPR  3", 1024, 0),
    NVP_ENTRY(40, "e04", "EMPR  4", 1024, 0),
    NVP_ENTRY(41, "e05", "EMPR  5", 1024, 0),
    NVP_ENTRY(42, "e06", "EMPR  6", 1024, 0),
    NVP_ENTRY(43, "e07", "EMPR  7", 1024, 0),
    NVP_ENTRY(44, "e08", "EMPR  8", 1024, 0),
    NVP_ENTRY(45, "e09", "EMPR  9", 1024, 0),
    NVP_ENTRY(46, "e10", "EMPR 10", 1024, 0),
    NVP_ENTRY(47, "e11", "EMPR 11", 1024, 0),
    NVP_ENTRY(48, "e12", "EMPR 12", 1024, 0),
    NVP_ENTRY(49, "e13", "EMPR 13", 1024, 0),
    NVP_ENTRY(50, "e14", "EMPR 14", 1024, 0),
    NVP_ENTRY(51, "e15", "EMPR 15", 1024, 0),
    NVP_ENTRY(52, "e16", "EMPR 16", 1024, 0),
    NVP_ENTRY(53, "e17", "EMPR 17", 1024, 0),
    NVP_ENTRY(54, "e18", "EMPR 18", 1024, 0),
    NVP_ENTRY(55, "e19", "EMPR 19", 1024, 0),
    NVP_ENTRY(56, "e20", "EMPR 20", 1024, 0),
    NVP_ENTRY(57, "e21", "EMPR 21", 1024, 0),
    NVP_ENTRY(58, "e22", "EMPR 22", 1024, 0),
    NVP_ENTRY(59, "e23", "EMPR 23", 1024, 0),
    NVP_ENTRY(60, "e24", "EMPR 24", 1024, 0),
    NVP_ENTRY(61, "e25", "EMPR 25", 1024, 0),
    NVP_ENTRY(62, "e26", "EMPR 26", 1024, 0),
    NVP_ENTRY(63, "e27", "EMPR 27", 1024, 0),
    NVP_ENTRY(64, "e28", "EMPR 28", 1024, 0),
    NVP_ENTRY(65, "e29", "EMPR 29", 1024, 0),
    NVP_ENTRY(66, "e30", "EMPR 30", 1024, 0),
    NVP_ENTRY(67, "e31", "EMPR 31", 1024, 0),
    NVP_ENTRY(68, "clv", "color variation", 4, 0),
    NVP_ENTRY(69, "slp", "time out to sleep", 4, 0),
    NVP_ENTRY(70, "ipt", "disable iptable flag", 4, 0),
    NVP_ENTRY(71, "mtm", "marlin time", 64, 0),
    NVP_ENTRY(72, "mcr", "marlin crl", 16384, 0),
    NVP_ENTRY(73, "mdk", "marlin device key", 33024, 0),
    NVP_ENTRY(74, "muk", "marlin user key", 24576, 0),
    NVP_ENTRY(75, "pts", "wifi protected setup", 4, 0),
    NVP_ENTRY(76, "skt", "slacker time", 16, 0),
    NVP_ENTRY(77, "mac", "wifi mac address", 6, 0),
    NVP_ENTRY(78, "apd", "application debug mode flag", 4, 0),
    NVP_ENTRY(79, "blf", "browser log mode flag", 4, 0),
    NVP_ENTRY(80, "hld", "hold mode", 4, 0),
    NVP_ENTRY(81, "skd", "slacker id file", 8224, 0),
    NVP_ENTRY(82, "fmp", "fm parameter", 16, 0),
    NVP_ENTRY(83, "sps", "speaker ship info", 4, 0),
    NVP_ENTRY(84, "msc", "mass storage class mode", 4, 0),
    NVP_ENTRY(85, "vrt", "europe vol regulation flag", 4, 0),
    NVP_ENTRY(86, "psk", "bluetooth pskey", 512, 0),
    NVP_ENTRY(87, "bml", "btmw log mode flag", 4, 0),
    NVP_ENTRY(88, "bfd", "btmw factory scdb", 512, 0),
    NVP_ENTRY(89, "bfp", "btmw factory pair info", 512, 0),
};

#define NR_NVP_PROPS   (sizeof(nvp_prop_list) / sizeof(nvp_prop_list[0]))

int get_dnk_nvp(int argc, char **argv)
{
    if(argc != 1 && argc != 3)
    {
        printf("You must specify a known nvp node or a full node specification:\n");
        printf("Full usage: <node> <size> <flags>\n");
        printf("Node usage: <node>\n");
        printf("Nodes:\n");
        for(unsigned i = 0; i < NR_NVP_PROPS; i++)
            printf("  %s\t%s\n", nvp_prop_list[i].name, nvp_prop_list[i].desc);
        return 1;
    }

    struct nvp_prop_t prop;
    memset(&prop, 0, sizeof(prop));
    if(argc == 1)
    {
        for(unsigned i = 0; i < NR_NVP_PROPS; i++)
            if(strcmp(nvp_prop_list[i].name, argv[0]) == 0)
                prop = nvp_prop_list[i];
        if(prop.name == NULL)
        {
            cprintf(GREY, "Unknown node '%s'\n", argv[0]);
            return 1;
        }
    }
    else
    {
        prop.node = strtoul(argv[0], NULL, 0);
        prop.size = strtoul(argv[1], NULL, 0);
        prop.flags = strtoul(argv[2], NULL, 0);
    }

    int buffer_size = prop.size + 4;
    uint8_t *buffer = buffer_alloc(buffer_size + 1);
    int ret = do_dnk_cmd(0x23, 10, prop.node, buffer, &buffer_size);
    if(ret)
        return ret;
    if(buffer_size <= 4)
    {
        cprintf(GREY, "Device didn't send any data\n");
        return 1;
    }
    if(buffer[0] != 0 || buffer[2] != 0 || buffer[3] != prop.node)
    {
        cprintf(GREY, "Device responded with invalid data\n");
        return 1;
    }

    for(int i = 4, idx = buffer[1]; i < buffer_size; i++, idx++)
        buffer[i] ^= para_noise[idx];
    buffer[buffer_size] = 0;
    buffer += 4;
    buffer_size -= 4;
    if((prop.flags & DNK_EXACT_LENGTH) && buffer_size != prop.size)
    {
        cprintf(GREY, "Device didn't send the expected amount of data\n");
        return 2;
    }
    
    if(prop.flags & DNK_STRING)
        cprintf_field("Node: ", "%s\n", buffer);
    else if(prop.flags & DNK_UINT32)
        cprintf_field("Node: ", "0x%x\n", *(uint32_t *)buffer);
    else
    {
        cprintf(GREEN, "Node:\n");
        print_hex(buffer, buffer_size);
    }
    return 0;
}

struct dpcc_prop_t
{
    char *user_name;
    char name[7];
    uint8_t cdb1;
    int size;
};

struct dpcc_prop_t dpcc_prop_list[] =
{
    { "dev_info", "DEVINFO", 0, 0x80 },
};

#define NR_DPCC_PROPS   (sizeof(dpcc_prop_list) / sizeof(dpcc_prop_list[0]))

int do_dpcc_cmd(uint32_t cmd, struct dpcc_prop_t *prop, void *buffer, int *buffer_size)
{
    uint8_t cdb[12] = {0xfb, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
    cdb[2] = cmd;
    if(cmd == 0)
    {
        strncpy((char *)(cdb + 3), prop->name, 7); // warning: erase cdb[10] !
        cdb[1] = prop->cdb1;
        if(prop->cdb1 & 1)
            cdb[10] = (*buffer_size + 15) / 16;
        else
            cdb[10] = *buffer_size;
    }

    uint8_t sense[32];
    int sense_size = 32;

    int ret = do_scsi(cdb, 12, DO_READ, sense, &sense_size, buffer, buffer_size);
    if(ret < 0)
        return ret;
    ret = do_sense_analysis(ret, sense, sense_size);
    if(ret)
        return ret;
    return 0;
}

int get_dpcc_prop(int argc, char **argv)
{
    if(argc != 1 && argc != 3)
    {
        printf("You must specify a known property name or a full property specification:\n");
        printf("Full usage: <prop code> <large> <size>\n");
        printf("Property usage: <prop>\n");
        printf("Properties:");
        for(unsigned i = 0; i < NR_DPCC_PROPS; i++)
            printf(" %s", dpcc_prop_list[i].user_name);
        printf("\n");
        return 1;
    }

    struct dpcc_prop_t prop;
    memset(&prop, 0, sizeof(prop));
    if(argc == 1)
    {
        for(unsigned i = 0; i < NR_DPCC_PROPS; i++)
            if(strcmp(dpcc_prop_list[i].user_name, argv[0]) == 0)
                prop = dpcc_prop_list[i];
        if(prop.user_name[0] == 0)
        {
            cprintf(GREY, "Unknown property '%s'\n", argv[0]);
            return 1;
        }
    }
    else
    {
        strncpy(prop.name, argv[0], 7);
        prop.cdb1 = strtoul(argv[1], NULL, 0);
        prop.size = strtoul(argv[2], NULL, 0);
    }

    char *buffer = buffer_alloc(prop.size);
    int buffer_size = prop.size;
    int ret = do_dpcc_cmd(0, &prop, buffer, &buffer_size);
    if(ret)
        return ret;
    if(buffer_size < prop.size)
        buffer[buffer_size] = 0;
    cprintf_field("Property: ", "%s\n", buffer);
    return 0;
}

struct user_timer_t
{
    uint16_t magic;
    uint8_t res[6];
    uint8_t year[2]; // bcd
    uint8_t month; // bcd
    uint8_t day; // bcd
    uint8_t hour; // bcd
    uint8_t min; // bcd
    uint8_t sec; // bcd
    uint8_t res2[17];
} __attribute__((packed));

int get_user_time(int argc, char **argv)
{
    (void) argc;
    (void )argv;

    void *buffer = buffer_alloc(32);
    int buffer_size = 32;
    int ret = do_dpcc_cmd(1, NULL, buffer, &buffer_size);
    if(ret)
        return ret;
    struct user_timer_t *time = buffer;
    cprintf_field("User Time: ", "%02x/%02x/%02x%02x %02x:%02x:%02x\n",
        time->day, time->month, time->year[0], time->year[1], time->hour,
        time->min, time->sec);
    return 0;
}

int get_dev_info(int argc, char **argv)
{
    (void) argc;
    (void )argv;
    uint8_t cdb[12] = {0xfc, 0, 0x20, 'd', 'b', 'm', 'n', 0, 0x80, 0, 0, 0};

    char *buffer = buffer_alloc(0x81);
    int buffer_size = 0x80;
    uint8_t sense[32];
    int sense_size = 32;

    int ret = do_scsi(cdb, 12, DO_READ, sense, &sense_size, buffer, &buffer_size);
    if(ret < 0)
        return ret;
    ret = do_sense_analysis(ret, sense, sense_size);
    if(ret)
        return ret;
    buffer[buffer_size] = 0;
    cprintf_field("Device Info:", "\n");
    print_hex(buffer, buffer_size);
    return 0;
}

int do_fw_upgrade(int argc, char **argv)
{
    (void) argc;
    (void )argv;
    uint8_t cdb[12] = {0xfc, 0, 0x04, 'd', 'b', 'm', 'n', 0, 0x80, 0, 0, 0};

    char *buffer = buffer_alloc(0x81);
    int buffer_size = 0x80;
    uint8_t sense[32];
    int sense_size = 32;

    int ret = do_scsi(cdb, 12, DO_READ, sense, &sense_size, buffer, &buffer_size);
    if(ret < 0)
        return ret;
    ret = do_sense_analysis(ret, sense, sense_size);
    if(ret)
        return ret;
    buffer[buffer_size] = 0;
    cprintf_field("Result:", "\n");
    print_hex(buffer, buffer_size);
    return 0;
}

typedef int (*cmd_fn_t)(int argc, char **argv);

struct cmd_t
{
    const char *name;
    const char *desc;
    cmd_fn_t fn;
};

struct cmd_t cmd_list[] =
{
    { "get_dnk_prop", "Get DNK property", get_dnk_prop },
    { "get_dnk_nvp", "Get DNK NVP content", get_dnk_nvp },
    { "get_dpcc_prop", "Get DPCC property", get_dpcc_prop },
    { "get_user_time", "Get user time", get_user_time },
    { "get_dev_info", "Get device info", get_dev_info },
    { "do_fw_upgrade", "Do a firmware upgrade", do_fw_upgrade },
};

#define NR_CMDS (sizeof(cmd_list) / sizeof(cmd_list[0]))

int process_cmd(const char *cmd, int argc, char **argv)
{
    for(unsigned i = 0; i < NR_CMDS; i++)
        if(strcmp(cmd_list[i].name, cmd) == 0)
            return cmd_list[i].fn(argc, argv);
    cprintf(GREY, "Unknown command '%s'\n", cmd);
    return 1;
}

static void usage(void)
{
    printf("Usage: scsitool [options] <dev> <command> [arguments]\n");
    printf("Options:\n");
    printf("  -o <prefix>\tSet output prefix\n");
    printf("  -f/--force\tForce to continue on errors\n");
    printf("  -?/--help\tDisplay this message\n");
    printf("  -d/--debug\tDisplay debug messages\n");
    printf("  -c/--no-color\tDisable color output\n");
    printf("Commands:\n");
    for(unsigned i = 0; i < NR_CMDS; i++)
        printf("  %s\t%s\n", cmd_list[i].name, cmd_list[i].desc);
    exit(1);
}

int main(int argc, char **argv)
{
    while(1)
    {
        static struct option long_options[] =
        {
            {"help", no_argument, 0, '?'},
            {"debug", no_argument, 0, 'd'},
            {"no-color", no_argument, 0, 'c'},
            {"force", no_argument, 0, 'f'},
            {0, 0, 0, 0}
        };

        int c = getopt_long(argc, argv, "?dcfo:", long_options, NULL);
        if(c == -1)
            break;
        switch(c)
        {
            case -1:
                break;
            case 'c':
                enable_color(false);
                break;
            case 'd':
                g_debug = true;
                break;
            case 'f':
                g_force = true;
                break;
            case '?':
                usage();
                break;
            case 'o':
                g_out_prefix = optarg;
                break;
            default:
                abort();
        }
    }

    if(argc - optind < 2)
    {
        usage();
        return 1;
    }

    int ret = 0;
    g_dev_fd = scsi_pt_open_device(argv[optind], false, true);
    if(g_dev_fd < 0)
    {
        cprintf(GREY, "Cannot open device: %m\n");
        ret = 1;
        goto Lend;
    }

    ret = process_cmd(argv[optind + 1], argc - optind - 2, argv + optind + 2);

    scsi_pt_close_device(g_dev_fd);
Lend:
    color(OFF);

    return ret;
}