int
main(int argc, char * argv[])
{
bool last, got_stdin, is_reg;
bool want_file = false;
bool verbose_given = false;
bool version_given = false;
int res, c, len, k, n, rsp_len, resid, act_len, din_len, verb;
int sg_fd = -1;
int infd = -1;
int do_help = 0;
int ret = 0;
uint32_t gen_code = 0;
const char * device_name = NULL;
const char * file_name = NULL;
uint8_t * dmp = NULL;
uint8_t * dip = NULL;
uint8_t * free_dip = NULL;
char * cp;
char ebuff[EBUFF_SZ];
struct stat a_stat;
struct dout_buff_t dout;
struct opts_t opts;
struct opts_t * op;
const struct mode_s * mp;
op = &opts;
memset(op, 0, sizeof(opts));
memset(&dout, 0, sizeof(dout));
din_len = DEF_DIN_LEN;
while (1) {
int option_index = 0;
c = getopt_long(argc, argv, "b:dehi:I:l:m:No:s:S:t:vV", long_options,
&option_index);
if (c == -1)
break;
switch (c) {
case 'b':
op->bpw = sg_get_num(optarg);
if (op->bpw < 0) {
pr2serr("argument to '--bpw' should be in a positive "
"number\n");
return SG_LIB_SYNTAX_ERROR;
}
if ((cp = strchr(optarg, ','))) {
if (0 == strncmp("act", cp + 1, 3))
op->bpw_then_activate = true;
}
break;
case 'd':
op->dry_run = true;
break;
case 'e':
op->ealsd = true;
break;
case 'h':
case '?':
++do_help;
break;
case 'i':
op->mc_id = sg_get_num_nomult(optarg);
if ((op->mc_id < 0) || (op->mc_id > 255)) {
pr2serr("argument to '--id' should be in the range 0 to "
"255\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 'I':
file_name = optarg;
break;
case 'l':
op->mc_len = sg_get_num(optarg);
if (op->mc_len < 0) {
pr2serr("bad argument to '--length'\n");
return SG_LIB_SYNTAX_ERROR;
}
op->mc_len_given = true;
break;
case 'm':
if (isdigit(*optarg)) {
op->mc_mode = sg_get_num_nomult(optarg);
if ((op->mc_mode < 0) || (op->mc_mode > 255)) {
pr2serr("argument to '--mode' should be in the range 0 "
"to 255\n");
return SG_LIB_SYNTAX_ERROR;
}
} else {
len = strlen(optarg);
for (mp = mode_arr; mp->mode_string; ++mp) {
if (0 == strncmp(mp->mode_string, optarg, len)) {
op->mc_mode = mp->mode;
break;
}
}
if (! mp->mode_string) {
print_modes();
return SG_LIB_SYNTAX_ERROR;
}
}
break;
case 'N':
op->mc_non = true;
break;
case 'o':
op->mc_offset = sg_get_num(optarg);
if (op->mc_offset < 0) {
pr2serr("bad argument to '--offset'\n");
return SG_LIB_SYNTAX_ERROR;
}
if (0 != (op->mc_offset % 4)) {
pr2serr("'--offset' value needs to be a multiple of 4\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 's':
op->mc_skip = sg_get_num(optarg);
if (op->mc_skip < 0) {
pr2serr("bad argument to '--skip'\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 'S':
op->mc_subenc = sg_get_num_nomult(optarg);
if ((op->mc_subenc < 0) || (op->mc_subenc > 255)) {
pr2serr("expected argument to '--subenc' to be 0 to 255\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 't':
op->mc_tlen = sg_get_num(optarg);
if (op->mc_tlen < 0) {
pr2serr("bad argument to '--tlength'\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 'v':
verbose_given = true;
++op->verbose;
break;
case 'V':
version_given = true;
break;
default:
pr2serr("unrecognised option code 0x%x ??\n", c);
usage();
return SG_LIB_SYNTAX_ERROR;
}
}
if (do_help) {
if (do_help > 1) {
usage();
pr2serr("\n");
print_modes();
} else
usage();
return 0;
}
if (optind < argc) {
if (NULL == device_name) {
device_name = argv[optind];
++optind;
}
if (optind < argc) {
for (; optind < argc; ++optind)
pr2serr("Unexpected extra argument: %s\n", argv[optind]);
usage();
return SG_LIB_SYNTAX_ERROR;
}
}
#ifdef DEBUG
pr2serr("In DEBUG mode, ");
if (verbose_given && version_given) {
pr2serr("but override: '-vV' given, zero verbose and continue\n");
verbose_given = false;
version_given = false;
op->verbose = 0;
} else if (! verbose_given) {
pr2serr("set '-vv'\n");
op->verbose = 2;
} else
pr2serr("keep verbose=%d\n", op->verbose);
#else
if (verbose_given && version_given)
pr2serr("Not in DEBUG mode, so '-vV' has no special action\n");
#endif
if (version_given) {
pr2serr(ME "version: %s\n", version_str);
return 0;
}
if (NULL == device_name) {
pr2serr("missing device name!\n\n");
usage();
return SG_LIB_SYNTAX_ERROR;
}
switch (op->mc_mode) {
case MODE_DNLD_MC_OFFS:
case MODE_DNLD_MC_OFFS_SAVE:
case MODE_DNLD_MC_OFFS_DEFER:
want_file = true;
break;
case MODE_DNLD_STATUS:
case MODE_ACTIVATE_MC:
case MODE_ABORT_MC:
want_file = false;
break;
default:
pr2serr("%s: mc_mode=0x%x, continue for now\n", __func__,
op->mc_mode);
break;
}
if ((op->mc_len > 0) && (op->bpw > op->mc_len)) {
pr2serr("trim chunk size (CS) to be the same as LEN\n");
op->bpw = op->mc_len;
}
if ((op->mc_offset > 0) && (op->bpw > 0)) {
op->mc_offset = 0;
pr2serr("WARNING: --offset= ignored (set back to 0) when --bpw= "
"argument given (and > 0)\n");
}
#ifdef SG_LIB_WIN32
#ifdef SG_LIB_WIN32_DIRECT
if (op->verbose > 4)
pr2serr("Initial win32 SPT interface state: %s\n",
scsi_pt_win32_spt_state() ? "direct" : "indirect");
scsi_pt_win32_direct(SG_LIB_WIN32_DIRECT /* SPT pt interface */);
#endif
#endif
sg_fd = sg_cmds_open_device(device_name, false /* rw */, op->verbose);
if (sg_fd < 0) {
if (op->verbose)
pr2serr(ME "open error: %s: %s\n", device_name,
safe_strerror(-sg_fd));
ret = sg_convert_errno(-sg_fd);
goto fini;
}
if (file_name && (! want_file))
pr2serr("ignoring --in=FILE option\n");
else if (file_name) {
got_stdin = (0 == strcmp(file_name, "-"));
if (got_stdin)
infd = STDIN_FILENO;
else {
if ((infd = open(file_name, O_RDONLY)) < 0) {
ret = sg_convert_errno(errno);
snprintf(ebuff, EBUFF_SZ,
ME "could not open %s for reading", file_name);
perror(ebuff);
goto fini;
} else if (sg_set_binary_mode(infd) < 0)
perror("sg_set_binary_mode");
}
if ((0 == fstat(infd, &a_stat)) && S_ISREG(a_stat.st_mode)) {
is_reg = true;
if (0 == op->mc_len) {
if (op->mc_skip >= a_stat.st_size) {
pr2serr("skip exceeds file size of %d bytes\n",
(int)a_stat.st_size);
ret = SG_LIB_FILE_ERROR;
goto fini;
}
op->mc_len = (int)(a_stat.st_size) - op->mc_skip;
}
} else {
is_reg = false;
if (0 == op->mc_len)
op->mc_len = DEF_XFER_LEN;
}
if (op->mc_len > MAX_XFER_LEN) {
pr2serr("file size or requested length (%d) exceeds "
"MAX_XFER_LEN of %d bytes\n", op->mc_len,
MAX_XFER_LEN);
ret = SG_LIB_FILE_ERROR;
goto fini;
}
if (NULL == (dmp = (uint8_t *)malloc(op->mc_len))) {
pr2serr(ME "out of memory to hold microcode read from FILE\n");
ret = SG_LIB_CAT_OTHER;
goto fini;
}
/* Don't remember why this is preset to 0xff, from write_buffer */
memset(dmp, 0xff, op->mc_len);
if (op->mc_skip > 0) {
if (! is_reg) {
if (got_stdin)
pr2serr("Can't skip on stdin\n");
else
pr2serr(ME "not a 'regular' file so can't apply skip\n");
ret = SG_LIB_FILE_ERROR;
goto fini;
}
if (lseek(infd, op->mc_skip, SEEK_SET) < 0) {
ret = sg_convert_errno(errno);
snprintf(ebuff, EBUFF_SZ, ME "couldn't skip to "
"required position on %s", file_name);
perror(ebuff);
goto fini;
}
}
res = read(infd, dmp, op->mc_len);
if (res < 0) {
ret = sg_convert_errno(errno);
snprintf(ebuff, EBUFF_SZ, ME "couldn't read from %s",
file_name);
perror(ebuff);
goto fini;
}
if (res < op->mc_len) {
if (op->mc_len_given) {
pr2serr("tried to read %d bytes from %s, got %d bytes\n",
op->mc_len, file_name, res);
pr2serr("pad with 0xff bytes and continue\n");
} else {
if (op->verbose) {
pr2serr("tried to read %d bytes from %s, got %d "
"bytes\n", op->mc_len, file_name, res);
pr2serr("will send %d bytes", res);
if ((op->bpw > 0) && (op->bpw < op->mc_len))
pr2serr(", %d bytes per WRITE BUFFER command\n",
op->bpw);
else
pr2serr("\n");
}
op->mc_len = res;
}
}
if (! got_stdin)
close(infd);
infd = -1;
} else if (want_file) {
pr2serr("need --in=FILE option with given mode\n");
ret = SG_LIB_CONTRADICT;
goto fini;
}
if (op->mc_tlen < op->mc_len)
op->mc_tlen = op->mc_len;
if (op->mc_non && (MODE_DNLD_STATUS == op->mc_mode)) {
pr2serr("Do nothing because '--non' given so fetching the Download "
"microcode status\ndpage might be dangerous\n");
goto fini;
}
dip = sg_memalign(din_len, 0, &free_dip, op->verbose > 3);
if (NULL == dip) {
pr2serr(ME "out of memory (data-in buffer)\n");
ret = SG_LIB_CAT_OTHER;
goto fini;
}
verb = (op->verbose > 1) ? op->verbose - 1 : 0;
/* Fetch Download microcode status dpage for generation code ++ */
if (op->dry_run) {
n = sizeof(dummy_rd_resp);
n = (n < din_len) ? n : din_len;
memcpy(dip, dummy_rd_resp, n);
resid = din_len - n;
res = 0;
} else
res = sg_ll_receive_diag_v2(sg_fd, true /* pcv */,
DPC_DOWNLOAD_MICROCODE, dip, din_len,
0 /*default timeout */, &resid, true,
verb);
if (0 == res) {
rsp_len = sg_get_unaligned_be16(dip + 2) + 4;
act_len = din_len - resid;
if (rsp_len > din_len) {
pr2serr("<<< warning response buffer too small [%d but need "
"%d]>>>\n", din_len, rsp_len);
rsp_len = din_len;
}
if (rsp_len > act_len) {
pr2serr("<<< warning response too short [actually got %d but "
"need %d]>>>\n", act_len, rsp_len);
rsp_len = act_len;
}
if (rsp_len < 8) {
pr2serr("Download microcode status dpage too short\n");
ret = SG_LIB_CAT_OTHER;
goto fini;
}
if ((op->verbose > 2) || (op->dry_run && op->verbose))
pr2serr("rec diag(ini): rsp_len=%d, num_sub-enc=%u "
"rec_gen_code=%u\n", rsp_len, dip[1],
sg_get_unaligned_be32(dip + 4));
} else {
ret = res;
goto fini;
}
gen_code = sg_get_unaligned_be32(dip + 4);
if (MODE_DNLD_STATUS == op->mc_mode) {
show_download_mc_sdg(dip, rsp_len, gen_code);
goto fini;
} else if (! want_file) { /* ACTIVATE and ABORT */
res = send_then_receive(sg_fd, gen_code, 0, NULL, 0, &dout, dip,
din_len, true, op);
ret = res;
goto fini;
}
res = 0;
if (op->bpw > 0) {
for (k = 0, last = false; k < op->mc_len; k += n) {
n = op->mc_len - k;
if (n > op->bpw)
n = op->bpw;
else
last = true;
if (op->verbose)
pr2serr("bpw loop: mode=0x%x, id=%d, off_off=%d, len=%d, "
"last=%d\n", op->mc_mode, op->mc_id, k, n, last);
res = send_then_receive(sg_fd, gen_code, k, dmp + k, n, &dout,
dip, din_len, last, op);
if (res)
break;
}
if (op->bpw_then_activate && (0 == res)) {
op->mc_mode = MODE_ACTIVATE_MC;
if (op->verbose)
pr2serr("sending Activate deferred microcode [0xf]\n");
res = send_then_receive(sg_fd, gen_code, 0, NULL, 0, &dout,
dip, din_len, true, op);
}
} else {
if (op->verbose)
pr2serr("single: mode=0x%x, id=%d, offset=%d, len=%d\n",
op->mc_mode, op->mc_id, op->mc_offset, op->mc_len);
res = send_then_receive(sg_fd, gen_code, 0, dmp, op->mc_len, &dout,
dip, din_len, true, op);
}
if (res)
ret = res;
fini:
if ((infd >= 0) && (! got_stdin))
close(infd);
if (dmp)
free(dmp);
if (dout.free_doutp)
free(dout.free_doutp);
if (free_dip)
free(free_dip);
if (sg_fd >= 0) {
res = sg_cmds_close_device(sg_fd);
if (res < 0) {
pr2serr("close error: %s\n", safe_strerror(-res));
if (0 == ret)
ret = sg_convert_errno(-res);
}
}
if (0 == op->verbose) {
if (! sg_if_can2stderr("sg_ses_mocrocode failed: ", ret))
pr2serr("Some error occurred, try again with '-v' "
"or '-vv' for more information\n");
}
return (ret >= 0) ? ret : SG_LIB_CAT_OTHER;
}
int
main(int argc, char * argv[])
{
bool do_long = false;
bool o_readonly = false;
bool do_raw = false;
bool verbose_given = false;
bool version_given = false;
int res, c, len, k;
int sg_fd = -1;
int do_help = 0;
int do_hex = 0;
int rb_id = 0;
int rb_len = 4;
int rb_mode = 0;
int rb_mode_sp = 0;
int resid = 0;
int verbose = 0;
int ret = 0;
int64_t ll;
uint64_t rb_offset = 0;
const char * device_name = NULL;
uint8_t * resp;
const struct mode_s * mp;
while (1) {
int option_index = 0;
c = getopt_long(argc, argv, "hHi:l:Lm:o:rRS:vV", long_options,
&option_index);
if (c == -1)
break;
switch (c) {
case 'h':
case '?':
++do_help;
break;
case 'H':
++do_hex;
break;
case 'i':
rb_id = sg_get_num(optarg);
if ((rb_id < 0) || (rb_id > 255)) {
pr2serr("argument to '--id' should be in the range 0 to "
"255\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 'l':
rb_len = sg_get_num(optarg);
if (rb_len < 0) {
pr2serr("bad argument to '--length'\n");
return SG_LIB_SYNTAX_ERROR;
}
if (rb_len > 0xffffff) {
pr2serr("argument to '--length' must be <= 0xffffff\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 'L':
do_long = true;
break;
case 'm':
if (isdigit(*optarg)) {
rb_mode = sg_get_num(optarg);
if ((rb_mode < 0) || (rb_mode > 31)) {
pr2serr("argument to '--mode' should be in the range 0 "
"to 31\n");
return SG_LIB_SYNTAX_ERROR;
}
} else {
len = strlen(optarg);
for (mp = modes; mp->mode_string; ++mp) {
if (0 == strncmp(mp->mode_string, optarg, len)) {
rb_mode = mp->mode;
break;
}
}
if (NULL == mp->mode_string) {
print_modes();
return SG_LIB_SYNTAX_ERROR;
}
}
break;
case 'o':
ll = sg_get_llnum(optarg);
if (ll < 0) {
pr2serr("bad argument to '--offset'\n");
return SG_LIB_SYNTAX_ERROR;
}
rb_offset = ll;
break;
case 'r':
do_raw = true;
break;
case 'R':
o_readonly = true;
break;
case 'S':
rb_mode_sp = sg_get_num(optarg);
if ((rb_mode_sp < 0) || (rb_mode_sp > 7)) {
pr2serr("expected argument to '--specific' to be 0 to 7\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 'v':
verbose_given = true;
++verbose;
break;
case 'V':
version_given = true;
break;
default:
pr2serr("unrecognised option code 0x%x ??\n", c);
usage();
return SG_LIB_SYNTAX_ERROR;
}
}
if (do_help) {
if (do_help > 1) {
usage();
pr2serr("\n");
print_modes();
} else
usage();
return 0;
}
if (optind < argc) {
if (NULL == device_name) {
device_name = argv[optind];
++optind;
}
if (optind < argc) {
for (; optind < argc; ++optind)
pr2serr("Unexpected extra argument: %s\n", argv[optind]);
usage();
return SG_LIB_SYNTAX_ERROR;
}
}
#ifdef DEBUG
pr2serr("In DEBUG mode, ");
if (verbose_given && version_given) {
pr2serr("but override: '-vV' given, zero verbose and continue\n");
verbose_given = false;
version_given = false;
verbose = 0;
} else if (! verbose_given) {
pr2serr("set '-vv'\n");
verbose = 2;
} else
pr2serr("keep verbose=%d\n", verbose);
#else
if (verbose_given && version_given)
pr2serr("Not in DEBUG mode, so '-vV' has no special action\n");
#endif
if (version_given) {
pr2serr("version: %s\n", version_str);
return 0;
}
if (NULL == device_name) {
pr2serr("Missing device name!\n\n");
usage();
return SG_LIB_SYNTAX_ERROR;
}
len = rb_len ? rb_len : 8;
resp = (uint8_t *)malloc(len);
if (NULL == resp) {
pr2serr("unable to allocate %d bytes on the heap\n", len);
return SG_LIB_CAT_OTHER;
}
memset(resp, 0, len);
if (do_raw) {
if (sg_set_binary_mode(STDOUT_FILENO) < 0) {
perror("sg_set_binary_mode");
ret = SG_LIB_FILE_ERROR;
goto fini;
}
}
#ifdef SG_LIB_WIN32
#ifdef SG_LIB_WIN32_DIRECT
if (verbose > 4)
pr2serr("Initial win32 SPT interface state: %s\n",
scsi_pt_win32_spt_state() ? "direct" : "indirect");
scsi_pt_win32_direct(SG_LIB_WIN32_DIRECT /* SPT pt interface */);
#endif
#endif
sg_fd = sg_cmds_open_device(device_name, o_readonly, verbose);
if (sg_fd < 0) {
if (verbose)
pr2serr("open error: %s: %s\n", device_name,
safe_strerror(-sg_fd));
ret = sg_convert_errno(-sg_fd);
goto fini;
}
if (do_long)
res = sg_ll_read_buffer_16(sg_fd, rb_mode, rb_mode_sp, rb_id,
rb_offset, resp, rb_len, &resid, true,
verbose);
else if (rb_offset > 0xffffff) {
pr2serr("--offset value is too large for READ BUFFER(10), try "
"--16\n");
ret = SG_LIB_SYNTAX_ERROR;
goto fini;
} else
res = sg_ll_read_buffer_10(sg_fd, rb_mode, rb_mode_sp, rb_id,
(uint32_t)rb_offset, resp, rb_len, &resid,
true, verbose);
if (0 != res) {
char b[80];
ret = res;
if (res > 0) {
sg_get_category_sense_str(res, sizeof(b), b, verbose);
pr2serr("Read buffer(%d) failed: %s\n", (do_long ? 16 : 10), b);
}
goto fini;
}
if (resid > 0)
rb_len -= resid; /* got back less than requested */
if (rb_len > 0) {
if (do_raw)
dStrRaw(resp, rb_len);
else if (do_hex || (rb_len < 4))
hex2stdout((const uint8_t *)resp, rb_len, ((do_hex > 1) ? 0 : 1));
else {
switch (rb_mode) {
case MODE_DESCRIPTOR:
k = sg_get_unaligned_be24(resp + 1);
printf("OFFSET BOUNDARY: %d, Buffer offset alignment: "
"%d-byte\n", resp[0], (1 << resp[0]));
printf("BUFFER CAPACITY: %d (0x%x)\n", k, k);
break;
case MODE_ECHO_BDESC:
k = sg_get_unaligned_be16(resp + 2) & 0x1fff;
printf("EBOS:%d\n", resp[0] & 1 ? 1 : 0);
printf("Echo buffer capacity: %d (0x%x)\n", k, k);
break;
default:
hex2stdout((const uint8_t *)resp, rb_len,
(verbose > 1 ? 0 : 1));
break;
}
}
}
fini:
if (resp)
free(resp);
if (sg_fd >= 0) {
res = sg_cmds_close_device(sg_fd);
if (res < 0) {
pr2serr("close error: %s\n", safe_strerror(-res));
if (0 == ret)
ret = sg_convert_errno(-res);
}
}
if (0 == verbose) {
if (! sg_if_can2stderr("sg_read_buffer failed: ", ret))
pr2serr("Some error occurred, try again with '-v' "
"or '-vv' for more information\n");
}
return (ret >= 0) ? ret : SG_LIB_CAT_OTHER;
}
int
main(int argc, char * argv[])
{
int sg_fd, res, c, len, k, n, got_stdin, is_reg, rsp_len, verb, last;
int infd = -1;
int do_help = 0;
const char * device_name = NULL;
const char * file_name = NULL;
unsigned char * dmp = NULL;
unsigned char * dip = NULL;
char * cp;
char ebuff[EBUFF_SZ];
struct stat a_stat;
struct dout_buff_t dout;
struct opts_t opts;
struct opts_t * op;
const struct mode_s * mp;
uint32_t gen_code = 0;
int ret = 0;
op = &opts;
memset(op, 0, sizeof(opts));
memset(&dout, 0, sizeof(dout));
while (1) {
int option_index = 0;
c = getopt_long(argc, argv, "b:hi:I:l:m:No:s:S:t:vV", long_options,
&option_index);
if (c == -1)
break;
switch (c) {
case 'b':
op->bpw = sg_get_num(optarg);
if (op->bpw < 0) {
pr2serr("argument to '--bpw' should be in a positive "
"number\n");
return SG_LIB_SYNTAX_ERROR;
}
if ((cp = strchr(optarg, ','))) {
if (0 == strncmp("act", cp + 1, 3))
++op->bpw_then_activate;
}
break;
case 'h':
case '?':
++do_help;
break;
case 'i':
op->mc_id = sg_get_num(optarg);
if ((op->mc_id < 0) || (op->mc_id > 255)) {
pr2serr("argument to '--id' should be in the range 0 to "
"255\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 'I':
file_name = optarg;
break;
case 'l':
op->mc_len = sg_get_num(optarg);
if (op->mc_len < 0) {
pr2serr("bad argument to '--length'\n");
return SG_LIB_SYNTAX_ERROR;
}
op->mc_len_given = 1;
break;
case 'm':
if (isdigit(*optarg)) {
op->mc_mode = sg_get_num(optarg);
if ((op->mc_mode < 0) || (op->mc_mode > 255)) {
pr2serr("argument to '--mode' should be in the range 0 "
"to 255\n");
return SG_LIB_SYNTAX_ERROR;
}
} else {
len = strlen(optarg);
for (mp = mode_arr; mp->mode_string; ++mp) {
if (0 == strncmp(mp->mode_string, optarg, len)) {
op->mc_mode = mp->mode;
break;
}
}
if (! mp->mode_string) {
print_modes();
return SG_LIB_SYNTAX_ERROR;
}
}
break;
case 'N':
++op->mc_non;
break;
case 'o':
op->mc_offset = sg_get_num(optarg);
if (op->mc_offset < 0) {
pr2serr("bad argument to '--offset'\n");
return SG_LIB_SYNTAX_ERROR;
}
if (0 != (op->mc_offset % 4)) {
pr2serr("'--offset' value needs to be a multiple of 4\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 's':
op->mc_skip = sg_get_num(optarg);
if (op->mc_skip < 0) {
pr2serr("bad argument to '--skip'\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 'S':
op->mc_subenc = sg_get_num(optarg);
if ((op->mc_subenc < 0) || (op->mc_subenc > 255)) {
pr2serr("expected argument to '--subenc' to be 0 to 255\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 't':
op->mc_tlen = sg_get_num(optarg);
if (op->mc_tlen < 0) {
pr2serr("bad argument to '--tlength'\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 'v':
++op->verbose;
break;
case 'V':
pr2serr(ME "version: %s\n", version_str);
return 0;
default:
pr2serr("unrecognised option code 0x%x ??\n", c);
usage();
return SG_LIB_SYNTAX_ERROR;
}
}
if (do_help) {
if (do_help > 1) {
usage();
pr2serr("\n");
print_modes();
} else
usage();
return 0;
}
if (optind < argc) {
if (NULL == device_name) {
device_name = argv[optind];
++optind;
}
if (optind < argc) {
for (; optind < argc; ++optind)
pr2serr("Unexpected extra argument: %s\n", argv[optind]);
usage();
return SG_LIB_SYNTAX_ERROR;
}
}
if (NULL == device_name) {
pr2serr("missing device name!\n");
usage();
return SG_LIB_SYNTAX_ERROR;
}
if ((op->mc_len > 0) && (op->bpw > op->mc_len)) {
pr2serr("trim chunk size (CS) to be the same as LEN\n");
op->bpw = op->mc_len;
}
#ifdef SG_LIB_WIN32
#ifdef SG_LIB_WIN32_DIRECT
if (op->verbose > 4)
pr2serr("Initial win32 SPT interface state: %s\n",
scsi_pt_win32_spt_state() ? "direct" : "indirect");
scsi_pt_win32_direct(SG_LIB_WIN32_DIRECT /* SPT pt interface */);
#endif
#endif
sg_fd = sg_cmds_open_device(device_name, 0 /* rw */, op->verbose);
if (sg_fd < 0) {
pr2serr(ME "open error: %s: %s\n", device_name,
safe_strerror(-sg_fd));
return SG_LIB_FILE_ERROR;
}
if (file_name && ((MODE_DNLD_STATUS == op->mc_mode) ||
(MODE_ACTIVATE_MC == op->mc_mode)))
pr2serr("ignoring --in=FILE option\n");
else if (file_name) {
got_stdin = (0 == strcmp(file_name, "-")) ? 1 : 0;
if (got_stdin)
infd = STDIN_FILENO;
else {
if ((infd = open(file_name, O_RDONLY)) < 0) {
snprintf(ebuff, EBUFF_SZ,
ME "could not open %s for reading", file_name);
perror(ebuff);
ret = SG_LIB_FILE_ERROR;
goto fini;
} else if (sg_set_binary_mode(infd) < 0)
perror("sg_set_binary_mode");
}
if ((0 == fstat(infd, &a_stat)) && S_ISREG(a_stat.st_mode)) {
is_reg = 1;
if (0 == op->mc_len) {
if (op->mc_skip >= a_stat.st_size) {
pr2serr("skip exceeds file size of %d bytes\n",
(int)a_stat.st_size);
ret = SG_LIB_FILE_ERROR;
goto fini;
}
op->mc_len = (int)(a_stat.st_size) - op->mc_skip;
}
} else {
is_reg = 0;
if (0 == op->mc_len)
op->mc_len = DEF_XFER_LEN;
}
if (op->mc_len > MAX_XFER_LEN) {
pr2serr("file size or requested length (%d) exceeds "
"MAX_XFER_LEN of %d bytes\n", op->mc_len,
MAX_XFER_LEN);
ret = SG_LIB_FILE_ERROR;
goto fini;
}
if (NULL == (dmp = (unsigned char *)malloc(op->mc_len))) {
pr2serr(ME "out of memory (to hold microcode)\n");
ret = SG_LIB_CAT_OTHER;
goto fini;
}
/* Don't remember why this is preset to 0xff, from write_buffer */
memset(dmp, 0xff, op->mc_len);
if (op->mc_skip > 0) {
if (! is_reg) {
if (got_stdin)
pr2serr("Can't skip on stdin\n");
else
pr2serr(ME "not a 'regular' file so can't apply skip\n");
ret = SG_LIB_FILE_ERROR;
goto fini;
}
if (lseek(infd, op->mc_skip, SEEK_SET) < 0) {
snprintf(ebuff, EBUFF_SZ, ME "couldn't skip to "
"required position on %s", file_name);
perror(ebuff);
ret = SG_LIB_FILE_ERROR;
goto fini;
}
}
res = read(infd, dmp, op->mc_len);
if (res < 0) {
snprintf(ebuff, EBUFF_SZ, ME "couldn't read from %s",
file_name);
perror(ebuff);
ret = SG_LIB_FILE_ERROR;
goto fini;
}
if (res < op->mc_len) {
if (op->mc_len_given) {
pr2serr("tried to read %d bytes from %s, got %d bytes\n",
op->mc_len, file_name, res);
pr2serr("pad with 0xff bytes and continue\n");
} else {
if (op->verbose) {
pr2serr("tried to read %d bytes from %s, got %d "
"bytes\n", op->mc_len, file_name, res);
pr2serr("will send %d bytes", res);
if ((op->bpw > 0) && (op->bpw < op->mc_len))
pr2serr(", %d bytes per WRITE BUFFER command\n",
op->bpw);
else
pr2serr("\n");
}
op->mc_len = res;
}
}
if (! got_stdin)
close(infd);
infd = -1;
} else if (! ((MODE_DNLD_STATUS == op->mc_mode) ||
(MODE_ACTIVATE_MC == op->mc_mode))) {
pr2serr("need --in=FILE option with given mode\n");
ret = SG_LIB_SYNTAX_ERROR;
goto fini;
}
if (op->mc_tlen < op->mc_len)
op->mc_tlen = op->mc_len;
if (op->mc_non && (MODE_DNLD_STATUS == op->mc_mode)) {
pr2serr("Do nothing because '--non' given so fetching the Download "
"microcode status\ndpage might be dangerous\n");
goto fini;
}
if (NULL == (dip = (unsigned char *)malloc(DEF_DI_LEN))) {
pr2serr(ME "out of memory (data-in buffer)\n");
ret = SG_LIB_CAT_OTHER;
goto fini;
}
memset(dip, 0, DEF_DI_LEN);
verb = (op->verbose > 1) ? op->verbose - 1 : 0;
/* Fetch Download microcode status dpage for generation code ++ */
res = sg_ll_receive_diag(sg_fd, 1 /* pcv */, DPC_DOWNLOAD_MICROCODE, dip,
DEF_DI_LEN, 1, verb);
if (0 == res) {
rsp_len = sg_get_unaligned_be16(dip + 2) + 4;
if (rsp_len > DEF_DI_LEN) {
pr2serr("<<< warning response buffer too small [%d but need "
"%d]>>>\n", DEF_DI_LEN, rsp_len);
rsp_len = DEF_DI_LEN;
}
if (rsp_len < 8) {
pr2serr("Download microcode status dpage too short\n");
ret = SG_LIB_CAT_OTHER;
goto fini;
}
} else {
ret = res;
goto fini;
}
gen_code = sg_get_unaligned_be32(dip + 4);
if (MODE_DNLD_STATUS == op->mc_mode) {
ses_download_code_sdg(dip, rsp_len, gen_code);
goto fini;
} else if (MODE_ACTIVATE_MC == op->mc_mode) {
res = send_then_receive(sg_fd, gen_code, 0, NULL, 0, &dout, dip, 1,
op);
ret = res;
goto fini;
}
res = 0;
if (op->bpw > 0) {
for (k = 0, last = 0; k < op->mc_len; k += n) {
n = op->mc_len - k;
if (n > op->bpw)
n = op->bpw;
else
last = 1;
if (op->verbose)
pr2serr("bpw loop: mode=0x%x, id=%d, off_off=%d, len=%d, "
"last=%d\n", op->mc_mode, op->mc_id, k, n, last);
res = send_then_receive(sg_fd, gen_code, k, dmp + k, n, &dout,
dip, last, op);
if (res)
break;
}
if (op->bpw_then_activate && (0 == res)) {
op->mc_mode = MODE_ACTIVATE_MC;
if (op->verbose)
pr2serr("sending Activate deferred microcode [0xf]\n");
res = send_then_receive(sg_fd, gen_code, 0, NULL, 0, &dout,
dip, 1, op);
}
} else {
if (op->verbose)
pr2serr("single: mode=0x%x, id=%d, offset=%d, len=%d\n",
op->mc_mode, op->mc_id, op->mc_offset, op->mc_len);
res = send_then_receive(sg_fd, gen_code, 0, dmp, op->mc_len, &dout,
dip, 1, op);
}
if (res)
ret = res;
fini:
if ((infd >= 0) && (! got_stdin))
close(infd);
if (dmp)
free(dmp);
if (dout.doutp)
free(dout.doutp);
res = sg_cmds_close_device(sg_fd);
if (res < 0) {
pr2serr("close error: %s\n", safe_strerror(-res));
if (0 == ret)
return SG_LIB_FILE_ERROR;
}
if (ret && (0 == op->verbose)) {
if (SG_LIB_CAT_INVALID_OP == ret)
pr2serr("%sRECEIVE DIAGNOSTIC RESULTS command not supported\n",
((MODE_DNLD_STATUS == op->mc_mode) ?
"" : "SEND DIAGNOSTIC or "));
else if (ret > 0)
pr2serr("Failed, exit status %d\n", ret);
else if (ret < 0)
pr2serr("Some error occurred\n");
}
return (ret >= 0) ? ret : SG_LIB_CAT_OTHER;
}
int
main(int argc, char * argv[])
{
int sg_fd, k, num, rsp_len, res, rsp_buff_size, pg;
int read_in_len = 0;
int ret = 0;
struct opts_t opts;
struct opts_t * op;
unsigned char * rsp_buff = NULL;
const char * cp;
unsigned char * read_in = NULL;
op = &opts;
memset(op, 0, sizeof(opts));
op->maxlen = DEF_ALLOC_LEN;
op->page_code = -1;
res = process_cl(op, argc, argv);
if (res)
return SG_LIB_SYNTAX_ERROR;
if (op->do_help) {
if (op->opt_new)
usage();
else
usage_old();
return 0;
}
if (op->do_version) {
fprintf(stderr, "Version string: %s\n", version_str);
return 0;
}
rsp_buff_size = op->maxlen;
if (NULL == op->device_name) {
if (op->do_list) {
list_page_codes();
return 0;
}
fprintf(stderr, "No DEVICE argument given\n");
if (op->opt_new)
usage();
else
usage_old();
return SG_LIB_SYNTAX_ERROR;
}
if (op->do_raw) {
read_in = (unsigned char *)calloc(op->maxlen, 1);
if (NULL == read_in) {
fprintf(stderr, "unable to allocate %d bytes\n", op->maxlen);
return SG_LIB_CAT_OTHER;
}
if (build_diag_page(op->raw_arg, read_in, &read_in_len, op->maxlen)) {
if (op->opt_new) {
printf("Bad sequence after '--raw=' option\n");
usage();
} else {
printf("Bad sequence after '-raw=' option\n");
usage_old();
}
return SG_LIB_SYNTAX_ERROR;
}
}
if ((op->do_doff || op->do_uoff) && (! op->do_deftest)) {
if (op->opt_new) {
printf("setting --doff or --uoff only useful when -t is set\n");
usage();
} else {
printf("setting -doff or -uoff only useful when -t is set\n");
usage_old();
}
return SG_LIB_SYNTAX_ERROR;
}
if ((op->do_selftest > 0) && op->do_deftest) {
if (op->opt_new) {
printf("either set --selftest=SF or --test (not both)\n");
usage();
} else {
printf("either set -s=SF or -t (not both)\n");
usage_old();
}
return SG_LIB_SYNTAX_ERROR;
}
if (op->do_raw) {
if ((op->do_selftest > 0) || op->do_deftest || op->do_extdur ||
op->do_list) {
if (op->opt_new) {
printf("'--raw=' cannot be used with self-tests, '-e' or "
"'-l'\n");
usage();
} else {
printf("'-raw=' cannot be used with self-tests, '-e' or "
"'-l'\n");
usage_old();
}
return SG_LIB_SYNTAX_ERROR;
}
if (! op->do_pf) {
if (op->opt_new)
printf(">>> warning, '--pf' probably should be used with "
"'--raw='\n");
else
printf(">>> warning, '-pf' probably should be used with "
"'-raw='\n");
}
}
#ifdef SG_LIB_WIN32
#ifdef SG_LIB_WIN32_DIRECT
if (op->do_verbose > 4)
fprintf(stderr, "Initial win32 SPT interface state: %s\n",
scsi_pt_win32_spt_state() ? "direct" : "indirect");
if (op->maxlen >= 16384)
scsi_pt_win32_direct(SG_LIB_WIN32_DIRECT /* SPT pt interface */);
#endif
#endif
if ((sg_fd = sg_cmds_open_device(op->device_name, 0 /* rw */,
op->do_verbose)) < 0) {
fprintf(stderr, ME "error opening file: %s: %s\n", op->device_name,
safe_strerror(-sg_fd));
return SG_LIB_FILE_ERROR;
}
rsp_buff = (unsigned char *)calloc(op->maxlen, 1);
if (NULL == rsp_buff) {
fprintf(stderr, "unable to allocate %d bytes (2)\n", op->maxlen);
return SG_LIB_CAT_OTHER;
}
if (op->do_extdur) {
res = do_modes_0a(sg_fd, rsp_buff, 32, 1, 0, op->do_verbose);
if (0 == res) {
/* Assume mode sense(10) response without block descriptors */
num = sg_get_unaligned_be16(rsp_buff) - 6;
if (num >= 0xc) {
int secs;
secs = sg_get_unaligned_be16(rsp_buff + 18);
#ifdef SG_LIB_MINGW
printf("Expected extended self-test duration=%d seconds "
"(%g minutes)\n", secs, secs / 60.0);
#else
printf("Expected extended self-test duration=%d seconds "
"(%.2f minutes)\n", secs, secs / 60.0);
#endif
} else
printf("Extended self-test duration not available\n");
} else {
ret = res;
printf("Extended self-test duration (mode page 0xa) failed\n");
goto err_out9;
}
} else if ((op->do_list) || (op->page_code >= 0x0)) {
pg = op->page_code;
if (pg < 0)
res = do_senddiag(sg_fd, 0, 1 /* pf */, 0, 0, 0, rsp_buff, 4, 1,
op->do_verbose);
else
res = 0;
if (0 == res) {
if (0 == sg_ll_receive_diag(sg_fd, (pg >= 0x0),
((pg >= 0x0) ? pg : 0), rsp_buff,
rsp_buff_size, 1, op->do_verbose)) {
rsp_len = sg_get_unaligned_be16(rsp_buff + 2) + 4;
if (op->do_hex > 1)
dStrHex((const char *)rsp_buff, rsp_len,
(2 == op->do_hex) ? 0 : -1);
else if (pg < 0x1) {
printf("Supported diagnostic pages response:\n");
if (op->do_hex)
dStrHex((const char *)rsp_buff, rsp_len, 1);
else {
for (k = 0; k < (rsp_len - 4); ++k) {
cp = find_page_code_desc(rsp_buff[k + 4]);
printf(" 0x%02x %s\n", rsp_buff[k + 4],
(cp ? cp : "<unknown>"));
}
}
} else {
cp = find_page_code_desc(pg);
if (cp)
printf("%s diagnostic page [0x%x] response in "
"hex:\n", cp, pg);
else
printf("diagnostic page 0x%x response in hex:\n", pg);
dStrHex((const char *)rsp_buff, rsp_len, 1);
}
} else {
ret = res;
fprintf(stderr, "RECEIVE DIAGNOSTIC RESULTS command "
"failed\n");
goto err_out9;
}
} else {
ret = res;
goto err_out;
}
} else if (op->do_raw) {
res = do_senddiag(sg_fd, 0, op->do_pf, 0, 0, 0, read_in,
read_in_len, 1, op->do_verbose);
if (res) {
ret = res;
goto err_out;
}
} else {
res = do_senddiag(sg_fd, op->do_selftest, op->do_pf, op->do_deftest,
op->do_doff, op->do_uoff, NULL, 0, 1,
op->do_verbose);
if (0 == res) {
if ((5 == op->do_selftest) || (6 == op->do_selftest))
printf("Foreground self-test returned GOOD status\n");
else if (op->do_deftest && (! op->do_doff) && (! op->do_uoff))
printf("Default self-test returned GOOD status\n");
} else {
ret = res;
goto err_out;
}
}
res = sg_cmds_close_device(sg_fd);
if ((res < 0) && (0 == ret))
return SG_LIB_SYNTAX_ERROR;
return (ret >= 0) ? ret : SG_LIB_CAT_OTHER;
err_out:
if (SG_LIB_CAT_UNIT_ATTENTION == res)
fprintf(stderr, "SEND DIAGNOSTIC, unit attention\n");
else if (SG_LIB_CAT_ABORTED_COMMAND == res)
fprintf(stderr, "SEND DIAGNOSTIC, aborted command\n");
else if (SG_LIB_CAT_NOT_READY == res)
fprintf(stderr, "SEND DIAGNOSTIC, device not "
"ready\n");
else
fprintf(stderr, "SEND DIAGNOSTIC command, failed\n");
err_out9:
if (op->do_verbose < 2)
fprintf(stderr, " try again with '-vv' for more information\n");
res = sg_cmds_close_device(sg_fd);
if ((res < 0) && (0 == ret))
return SG_LIB_FILE_ERROR;
if (read_in)
free(read_in);
if (rsp_buff)
free(rsp_buff);
return (ret >= 0) ? ret : SG_LIB_CAT_OTHER;
}
int
main(int argc, char * argv[])
{
int res, c, len, k;
int sg_fd = -1;
int do_help = 0;
int do_hex = 0;
int do_long = 0;
int o_readonly = 0;
int rb_id = 0;
int rb_len = 4;
int rb_mode = 0;
int rb_mode_sp = 0;
int64_t ll;
uint64_t rb_offset = 0;
int do_raw = 0;
int resid = 0;
int verbose = 0;
int ret = 0;
const char * device_name = NULL;
unsigned char * resp;
const struct mode_s * mp;
while (1) {
int option_index = 0;
c = getopt_long(argc, argv, "hHi:l:Lm:o:rRS:vV", long_options,
&option_index);
if (c == -1)
break;
switch (c) {
case 'h':
case '?':
++do_help;
break;
case 'H':
++do_hex;
break;
case 'i':
rb_id = sg_get_num(optarg);
if ((rb_id < 0) || (rb_id > 255)) {
fprintf(stderr, "argument to '--id' should be in the range "
"0 to 255\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 'l':
rb_len = sg_get_num(optarg);
if (rb_len < 0) {
fprintf(stderr, "bad argument to '--length'\n");
return SG_LIB_SYNTAX_ERROR;
}
if (rb_len > 0xffffff) {
fprintf(stderr, "argument to '--length' must be <= "
"0xffffff\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 'L':
++do_long;
break;
case 'm':
if (isdigit(*optarg)) {
rb_mode = sg_get_num(optarg);
if ((rb_mode < 0) || (rb_mode > 31)) {
fprintf(stderr, "argument to '--mode' should be in the "
"range 0 to 31\n");
return SG_LIB_SYNTAX_ERROR;
}
} else {
len = strlen(optarg);
for (mp = modes; mp->mode_string; ++mp) {
if (0 == strncmp(mp->mode_string, optarg, len)) {
rb_mode = mp->mode;
break;
}
}
if (NULL == mp) {
print_modes();
return SG_LIB_SYNTAX_ERROR;
}
}
break;
case 'o':
ll = sg_get_llnum(optarg);
if (ll < 0) {
fprintf(stderr, "bad argument to '--offset'\n");
return SG_LIB_SYNTAX_ERROR;
}
rb_offset = ll;
break;
case 'r':
++do_raw;
break;
case 'R':
++o_readonly;
break;
case 'S':
rb_mode_sp = sg_get_num(optarg);
if ((rb_mode_sp < 0) || (rb_mode_sp > 7)) {
fprintf(stderr, "expected argument to '--specific' to be 0 "
"to 7\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 'v':
++verbose;
break;
case 'V':
fprintf(stderr, "version: %s\n", version_str);
return 0;
default:
fprintf(stderr, "unrecognised option code 0x%x ??\n", c);
usage();
return SG_LIB_SYNTAX_ERROR;
}
}
if (do_help) {
if (do_help > 1) {
usage();
fprintf(stderr, "\n");
print_modes();
} else
usage();
return 0;
}
if (optind < argc) {
if (NULL == device_name) {
device_name = argv[optind];
++optind;
}
if (optind < argc) {
for (; optind < argc; ++optind)
fprintf(stderr, "Unexpected extra argument: %s\n",
argv[optind]);
usage();
return SG_LIB_SYNTAX_ERROR;
}
}
if (NULL == device_name) {
fprintf(stderr, "missing device name!\n");
usage();
return SG_LIB_SYNTAX_ERROR;
}
if (rb_len > 0) {
resp = (unsigned char *)malloc(rb_len);
if (NULL == resp) {
fprintf(stderr, "unable to allocate %d bytes on the heap\n",
rb_len);
return SG_LIB_CAT_OTHER;
}
memset(resp, 0, rb_len);
} else
resp = NULL;
if (do_raw) {
if (sg_set_binary_mode(STDOUT_FILENO) < 0) {
perror("sg_set_binary_mode");
ret = SG_LIB_FILE_ERROR;
goto fini;
}
}
#ifdef SG_LIB_WIN32
#ifdef SG_LIB_WIN32_DIRECT
if (verbose > 4)
fprintf(stderr, "Initial win32 SPT interface state: %s\n",
scsi_pt_win32_spt_state() ? "direct" : "indirect");
scsi_pt_win32_direct(SG_LIB_WIN32_DIRECT /* SPT pt interface */);
#endif
#endif
sg_fd = sg_cmds_open_device(device_name, o_readonly, verbose);
if (sg_fd < 0) {
fprintf(stderr, "open error: %s: %s\n", device_name,
safe_strerror(-sg_fd));
ret = SG_LIB_FILE_ERROR;
goto fini;
}
if (do_long)
res = ll_read_buffer_16(sg_fd, rb_mode, rb_mode_sp, rb_id, rb_offset,
resp, rb_len, &resid, 1, verbose);
else if (rb_offset > 0xffffff) {
fprintf(stderr, "--offset value is too large for READ BUFFER(10), "
"try --16\n");
ret = SG_LIB_SYNTAX_ERROR;
goto fini;
} else
res = ll_read_buffer_10(sg_fd, rb_mode, rb_mode_sp, rb_id,
(uint32_t)rb_offset, resp, rb_len, &resid, 1,
verbose);
if (0 != res) {
char b[80];
ret = res;
if (res > 0) {
sg_get_category_sense_str(res, sizeof(b), b, verbose);
fprintf(stderr, "Read buffer(%d) failed: %s\n",
(do_long ? 16 : 10), b);
}
goto fini;
}
if (resid > 0)
rb_len -= resid; /* got back less than requested */
if (rb_len > 0) {
if (do_raw)
dStrRaw((const char *)resp, rb_len);
else if (do_hex || (rb_len < 4))
dStrHex((const char *)resp, rb_len, ((do_hex > 1) ? 0 : 1));
else {
switch (rb_mode) {
case MODE_DESCRIPTOR:
k = (resp[1] << 16) | (resp[2] << 8) | resp[3];
printf("OFFSET BOUNDARY: %d, Buffer offset alignment: "
"%d-byte\n", resp[0], (1 << resp[0]));
printf("BUFFER CAPACITY: %d (0x%x)\n", k, k);
break;
case MODE_ECHO_BDESC:
k = ((resp[2] & 0x1F) << 8) | resp[3];
printf("EBOS:%d\n", resp[0] & 1 ? 1 : 0);
printf("Echo buffer capacity: %d (0x%x)\n", k, k);
break;
default:
dStrHex((const char *)resp, rb_len, (verbose > 1 ? 0 : 1));
break;
}
}
}
fini:
if (resp)
free(resp);
if (sg_fd >= 0) {
res = sg_cmds_close_device(sg_fd);
if (res < 0) {
fprintf(stderr, "close error: %s\n", safe_strerror(-res));
if (0 == ret)
return SG_LIB_FILE_ERROR;
}
}
return (ret >= 0) ? ret : SG_LIB_CAT_OTHER;
}
int
main(int argc, char * argv[])
{
bool bpw_then_activate = false;
bool dry_run = false;
bool got_stdin = false;
bool wb_len_given = false;
int sg_fd, infd, res, c, len, k, n;
int bpw = 0;
int do_help = 0;
int ret = 0;
int verbose = 0;
int wb_id = 0;
int wb_len = 0;
int wb_mode = 0;
int wb_offset = 0;
int wb_skip = 0;
int wb_timeout = DEF_PT_TIMEOUT;
int wb_mspec = 0;
const char * device_name = NULL;
const char * file_name = NULL;
unsigned char * dop = NULL;
char * cp;
const struct mode_s * mp;
char ebuff[EBUFF_SZ];
while (1) {
int option_index = 0;
c = getopt_long(argc, argv, "b:dhi:I:l:m:o:rs:S:t:vV", long_options,
&option_index);
if (c == -1)
break;
switch (c) {
case 'b':
bpw = sg_get_num(optarg);
if (bpw < 0) {
pr2serr("argument to '--bpw' should be in a positive "
"number\n");
return SG_LIB_SYNTAX_ERROR;
}
if ((cp = strchr(optarg, ','))) {
if (0 == strncmp("act", cp + 1, 3))
bpw_then_activate = true;
}
break;
case 'd':
dry_run = true;
break;
case 'h':
case '?':
++do_help;
break;
case 'i':
wb_id = sg_get_num(optarg);
if ((wb_id < 0) || (wb_id > 255)) {
pr2serr("argument to '--id' should be in the range 0 to "
"255\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 'I':
file_name = optarg;
break;
case 'l':
wb_len = sg_get_num(optarg);
if (wb_len < 0) {
pr2serr("bad argument to '--length'\n");
return SG_LIB_SYNTAX_ERROR;
}
wb_len_given = true;
break;
case 'm':
if (isdigit(*optarg)) {
wb_mode = sg_get_num(optarg);
if ((wb_mode < 0) || (wb_mode > 31)) {
pr2serr("argument to '--mode' should be in the range 0 "
"to 31\n");
return SG_LIB_SYNTAX_ERROR;
}
} else {
len = strlen(optarg);
for (mp = mode_arr; mp->mode_string; ++mp) {
if (0 == strncmp(mp->mode_string, optarg, len)) {
wb_mode = mp->mode;
break;
}
}
if (! mp->mode_string) {
print_modes();
return SG_LIB_SYNTAX_ERROR;
}
}
break;
case 'o':
wb_offset = sg_get_num(optarg);
if (wb_offset < 0) {
pr2serr("bad argument to '--offset'\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 'r': /* --read-stdin and --raw (previous name) */
file_name = "-";
break;
case 's':
wb_skip = sg_get_num(optarg);
if (wb_skip < 0) {
pr2serr("bad argument to '--skip'\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 'S':
wb_mspec = sg_get_num(optarg);
if ((wb_mspec < 0) || (wb_mspec > 7)) {
pr2serr("expected argument to '--specific' to be 0 to 7\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 't':
wb_timeout = sg_get_num(optarg);
if (wb_timeout < 0) {
pr2serr("Invalid argument to '--timeout'\n");
return SG_LIB_SYNTAX_ERROR;
}
break;
case 'v':
++verbose;
break;
case 'V':
pr2serr(ME "version: %s\n", version_str);
return 0;
default:
pr2serr("unrecognised option code 0x%x ??\n", c);
usage();
return SG_LIB_SYNTAX_ERROR;
}
}
if (do_help) {
if (do_help > 1) {
usage();
pr2serr("\n");
print_modes();
} else
usage();
return 0;
}
if (optind < argc) {
if (NULL == device_name) {
device_name = argv[optind];
++optind;
}
if (optind < argc) {
for (; optind < argc; ++optind)
pr2serr("Unexpected extra argument: %s\n", argv[optind]);
usage();
return SG_LIB_SYNTAX_ERROR;
}
}
if (NULL == device_name) {
pr2serr("missing device name!\n");
usage();
return SG_LIB_SYNTAX_ERROR;
}
if ((wb_len > 0) && (bpw > wb_len)) {
pr2serr("trim chunk size (CS) to be the same as LEN\n");
bpw = wb_len;
}
#ifdef SG_LIB_WIN32
#ifdef SG_LIB_WIN32_DIRECT
if (verbose > 4)
pr2serr("Initial win32 SPT interface state: %s\n",
scsi_pt_win32_spt_state() ? "direct" : "indirect");
scsi_pt_win32_direct(SG_LIB_WIN32_DIRECT /* SPT pt interface */);
#endif
#endif
sg_fd = sg_cmds_open_device(device_name, false /* rw */, verbose);
if (sg_fd < 0) {
pr2serr(ME "open error: %s: %s\n", device_name,
safe_strerror(-sg_fd));
return SG_LIB_FILE_ERROR;
}
if (file_name || (wb_len > 0)) {
if (0 == wb_len)
wb_len = DEF_XFER_LEN;
if (NULL == (dop = (unsigned char *)malloc(wb_len))) {
pr2serr(ME "out of memory\n");
ret = SG_LIB_SYNTAX_ERROR;
goto err_out;
}
memset(dop, 0xff, wb_len);
if (file_name) {
got_stdin = (0 == strcmp(file_name, "-"));
if (got_stdin) {
if (wb_skip > 0) {
pr2serr("Can't skip on stdin\n");
ret = SG_LIB_FILE_ERROR;
goto err_out;
}
infd = STDIN_FILENO;
} else {
if ((infd = open(file_name, O_RDONLY)) < 0) {
snprintf(ebuff, EBUFF_SZ,
ME "could not open %s for reading", file_name);
perror(ebuff);
ret = SG_LIB_FILE_ERROR;
goto err_out;
} else if (sg_set_binary_mode(infd) < 0)
perror("sg_set_binary_mode");
if (wb_skip > 0) {
if (lseek(infd, wb_skip, SEEK_SET) < 0) {
snprintf(ebuff, EBUFF_SZ, ME "couldn't skip to "
"required position on %s", file_name);
perror(ebuff);
close(infd);
ret = SG_LIB_FILE_ERROR;
goto err_out;
}
}
}
res = read(infd, dop, wb_len);
if (res < 0) {
snprintf(ebuff, EBUFF_SZ, ME "couldn't read from %s",
file_name);
perror(ebuff);
if (! got_stdin)
close(infd);
ret = SG_LIB_FILE_ERROR;
goto err_out;
}
if (res < wb_len) {
if (wb_len_given) {
pr2serr("tried to read %d bytes from %s, got %d bytes\n",
wb_len, file_name, res);
pr2serr("pad with 0xff bytes and continue\n");
} else {
if (verbose) {
pr2serr("tried to read %d bytes from %s, got %d "
"bytes\n", wb_len, file_name, res);
pr2serr("will write %d bytes", res);
if ((bpw > 0) && (bpw < wb_len))
pr2serr(", %d bytes per WRITE BUFFER command\n",
bpw);
else
pr2serr("\n");
}
wb_len = res;
}
}
if (! got_stdin)
close(infd);
}
}
res = 0;
if (bpw > 0) {
for (k = 0; k < wb_len; k += n) {
n = wb_len - k;
if (n > bpw)
n = bpw;
if (verbose)
pr2serr("sending write buffer, mode=0x%x, mspec=%d, id=%d, "
" offset=%d, len=%d\n", wb_mode, wb_mspec, wb_id,
wb_offset + k, n);
if (dry_run) {
if (verbose)
pr2serr("skipping WRITE BUFFER command due to "
"--dry-run\n");
res = 0;
} else
res = sg_ll_write_buffer_v2(sg_fd, wb_mode, wb_mspec, wb_id,
wb_offset + k, dop + k, n,
wb_timeout, true, verbose);
if (res)
break;
}
if (bpw_then_activate) {
if (verbose)
pr2serr("sending Activate deferred microcode [0xf]\n");
if (dry_run) {
if (verbose)
pr2serr("skipping WRITE BUFFER(ACTIVATE) command due to "
"--dry-run\n");
res = 0;
} else
res = sg_ll_write_buffer_v2(sg_fd, MODE_ACTIVATE_MC,
0 /* buffer_id */,
0 /* buffer_offset */, 0,
NULL, 0, wb_timeout, true,
verbose);
}
} else {
if (verbose)
pr2serr("sending single write buffer, mode=0x%x, mpsec=%d, "
"id=%d, offset=%d, len=%d\n", wb_mode, wb_mspec, wb_id,
wb_offset, wb_len);
if (dry_run) {
if (verbose)
pr2serr("skipping WRITE BUFFER(all in one) command due to "
"--dry-run\n");
res = 0;
} else
res = sg_ll_write_buffer_v2(sg_fd, wb_mode, wb_mspec, wb_id,
wb_offset, dop, wb_len, wb_timeout,
true, verbose);
}
if (0 != res) {
char b[80];
ret = res;
sg_get_category_sense_str(res, sizeof(b), b, verbose);
pr2serr("Write buffer failed: %s\n", b);
}
err_out:
if (dop)
free(dop);
res = sg_cmds_close_device(sg_fd);
if (res < 0) {
pr2serr("close error: %s\n", safe_strerror(-res));
if (0 == ret)
return SG_LIB_FILE_ERROR;
}
return (ret >= 0) ? ret : SG_LIB_CAT_OTHER;
}
