Nyaaori/rockbox
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity
rockbox/utils/nwztools/scsitools/scsitool.c
Amaury Pouly b9923df170 imxtools/nwztools: fix compilation for windows 
There is a windows port of the sg_utils library for scsi pass-
through. This little changes make it compile under mingw. A better
fix would be to implement direct ioctl on both windows and linux
but that's already better than nothing

Change-Id: I0d77cd1bad69806a66f0590362f165f24fa240e9
2013-01-10 00:51:35 +00:00

736 lines
22 KiB
C
Raw Blame History

/***************************************************************************
* __________ __ ___.
* 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;
}
Reference in a new issue View git blame Copy permalink