/*************************************************************************** * __________ __ ___. * 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 #include #include #include #include #include #include #include #include #include #include #include #include #include "rbscsi.h" #include "misc.h" #include "para_noise.h" #include "nwz-db.h" bool g_debug = false; const char *g_force_series = NULL; char *g_out_prefix = NULL; rb_scsi_device_t g_dev; bool g_yes_i_want_a_brick = false; void help_us(bool unsupported, unsigned long model_id); 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) { struct rb_scsi_raw_cmd_t raw; raw.dir = RB_SCSI_NONE; if(flags & DO_READ) raw.dir = RB_SCSI_READ; if(flags & DO_WRITE) raw.dir = RB_SCSI_WRITE; raw.cdb_len = cdb_size; raw.cdb = cdb; raw.buf = buffer; raw.buf_len = *buf_size; raw.sense_len = *sense_size; raw.sense = sense; raw.tmo = 5; int ret = rb_scsi_raw_xfer(g_dev, &raw); *sense_size = raw.sense_len; *buf_size = raw.buf_len; return ret == RB_SCSI_OK || ret == RB_SCSI_SENSE ? raw.status : -ret; } int do_sense_analysis(int status, uint8_t *sense, int sense_size) { if(status != 0 && g_debug) { cprintf(GREY, "Status: %d\n", status); cprintf(GREY, "Sense:"); for(int i = 0; i < sense_size; i++) cprintf(GREY, " %02x", sense[i]); cprintf(GREY, "\n"); rb_scsi_decode_sense(g_dev, sense, sense_size); } return status; } /* * SCSI commands (see decode_scsi for more details) */ #define CMD_A3 0xa3 /* start a complicated, authenticated, session to do things */ #define CMD_A4 0xa4 /* start a complicated, authenticated, session to do things */ #define CMD_EMPR_DPCC 0xd7 #define CMD_DNK 0xdd #define CMD_DPCC 0xfb /* * DNK: command is in cdb[10], subcommand in cdb[11], cdb[7] must be 0xbc */ int do_dnk_cmd(bool read, uint32_t cmd, uint8_t sub_cmd, uint16_t arg, void *buffer, int *buffer_size) { uint8_t cdb[12] = {CMD_DNK, 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, read ? DO_READ : DO_WRITE, 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) #define DNK_HEX (1 << 3) struct dnk_prop_t { const char *name; const char *desc; uint8_t cmd; uint8_t subcmd; int size; unsigned flags; }; struct dnk_prop_t dnk_prop_list[] = { { "serial_num", "Serial number", 0x23, 1, 8, DNK_STRING}, { "storage_size", "Storage size(GB)", 0x23, 4, 4, DNK_EXACT_LENGTH | DNK_UINT32}, { "product_id", "Product ID", 0x23, 6, 12, DNK_STRING}, { "destination", "Destination", 0x23, 8, 4, DNK_EXACT_LENGTH | DNK_UINT32}, { "model_id", "Model ID", 0x23, 9, 4, DNK_EXACT_LENGTH | DNK_UINT32 | DNK_HEX}, { "ufn", "Update filename", 0x23, 21, 8, DNK_STRING}, { "kas", "Key and Signature", 0x23, 22, 64, DNK_STRING}, { "devinfo", "Devide Info", 0x12, 0, 64, DNK_HEX}, /* there are more obscure commands: * - 0x11 returns a 10-byte packet containing a 8-byte "LeftIdl8", scrambled * with para_noise (the 2-byte padding is random so that output is random * until unscrambled) * - 0x21 returns a 0x2b2 packet contaning a 0x2b0 "DNK", scrambled similarly * - 0x22 can write the DNK (sending scrambled data again) * - 0x23 has more subproperties: * - 5 is "eDKS" * - 7 is "ProductGroup" * - 10 is nvp properties (see get_dnk_nvp) (NOTE: nvp number vary by model here) * - 11 seems to read something from nvp and encrypt it with AES, not sure what * - many other read important/canonical entries of NVP (number does NOT vary model) * - 0x24 can write the same properties read by 0x23 * * This website has some background on Sony's acronym (DNK, EKB, ...): * https://wiki.physik.fu-berlin.de/linux-minidisc/doku.php?id=atracdownload-wiki */ }; #define NR_DNK_PROPS (sizeof(dnk_prop_list) / sizeof(dnk_prop_list[0])) uint16_t get_big_endian16(void *_buf) { uint8_t *buf = _buf; return buf[0] << 16 | buf[1]; } uint32_t get_big_endian32(void *_buf) { uint8_t *buf = _buf; return buf[0] << 24 | buf[1] << 16 | buf[2] << 8 | buf[3]; } void set_big_endian16(void *_buf, uint16_t val) { uint8_t *buf = _buf; buf[1] = val & 0xff; buf[0] = (val >> 8) & 0xff; } void set_big_endian32(void *_buf, uint32_t val) { uint8_t *buf = _buf; buf[3] = val & 0xff; buf[2] = (val >> 8) & 0xff; buf[1] = (val >> 16) & 0xff; buf[0] = (val >> 24) & 0xff; } uint32_t get_little_endian32(void *_buf) { uint8_t *buf = _buf; return buf[3] << 24 | buf[2] << 16 | buf[1] << 8 | buf[0]; } void set_little_endian32(void *_buf, uint32_t val) { uint8_t *buf = _buf; buf[0] = val & 0xff; buf[1] = (val >> 8) & 0xff; buf[2] = (val >> 16) & 0xff; buf[3] = (val >> 24) & 0xff; } int get_dnk_prop(int argc, const char* const* argv) { if(argc != 1 && argc != 4) { printf("You must specify a known property name or a full property specification:\n"); printf("Full usage: \n"); printf("Property usage: \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.desc = "Property"; 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 = malloc(prop.size + 1); int buffer_size = prop.size; int ret = do_dnk_cmd(true, prop.cmd, prop.subcmd, 0, buffer, &buffer_size); if(ret) { cprintf(GREY, "An error occured during request\n"); 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; cprintf(GREEN, "%s:", prop.desc); if(prop.flags & DNK_STRING) cprintf(YELLOW, " %s\n", buffer); else if(prop.flags & DNK_UINT32) { uint32_t val = get_big_endian32(buffer); if(prop.flags & DNK_HEX) cprintf(YELLOW, " 0x%x\n", val); else cprintf(YELLOW, " %u\n", val); } else { cprintf(YELLOW, "\n"); print_hex(buffer, buffer_size); } return 0; } /* get the model DID: code stolen from get_dnk_prop */ int get_model_id(unsigned long *model_id) { uint8_t mid_buf[4]; int mid_buf_size = sizeof(mid_buf); int ret = do_dnk_cmd(true, 0x23, 9, 0, mid_buf, &mid_buf_size); if(ret) { cprintf(RED, "Cannot get model ID from device: %d\n", ret); cprintf(RED, "Your device is most likely not compatible with this tool.\n"); return 2; } if(mid_buf_size != sizeof(mid_buf)) { cprintf(RED, "Cannot get model ID from device: device didn't send the expected amount of data\n"); cprintf(RED, "Your device is most likely not compatible with this tool.\n"); return 3; } *model_id = get_big_endian32(&mid_buf); return 0; } int get_model_and_series(int *model_index, int *series_index, unsigned long *model_id_out) { /* if the user forced the series, simply match by name, special for '?' which * prompts the list */ if(g_force_series) { cprintf(RED, "User forced series, auto-detection disabled\n"); *series_index = -1; *model_index = -1; for(int i = 0; i < NWZ_SERIES_COUNT; i++) if(strcmp(nwz_series[i].codename, g_force_series) == 0) *series_index = i; /* display list on error */ if(*series_index == -1) { cprintf(GREY, "Unrecognized series '%s', use -s ? to get a list of supported devices\n", g_force_series); return -1; } } else { /* we need to get the model ID */ unsigned long model_id; if(get_model_id(&model_id)) return 1; if(model_id_out) *model_id_out = model_id; *model_index = -1; for(int i = 0; i < NWZ_MODEL_COUNT; i++) if(nwz_model[i].mid == model_id) *model_index = i; if(*model_index == -1) { help_us(true, model_id); return 3; } *series_index = -1; for(int i = 0; i < NWZ_SERIES_COUNT; i++) for(int j = 0; j < nwz_series[i].mid_count; j++) if(nwz_series[i].mid[j] == model_id) *series_index = i; if(*series_index == -1) { help_us(true, model_id); return 3; } } cprintf_field("Model: ", "%s\n", *model_index == -1 ? "Unknown" : nwz_model[*model_index].name); cprintf_field("Series: ", "%s\n", *series_index == -1 ? "Unknown" : nwz_series[*series_index].name); return 0; } /* model IDs follow a pattern: the high 16-bit seem to encode the series and low 16-bit the size * (although this is not entirely reliable). Just try to find any device with the same high 16-bits * and return the series it belongs to. */ int guess_series_for_model(unsigned long model_id) { for(int i = 0; i < NWZ_SERIES_COUNT; i++) for(int j = 0; j < nwz_series[i].mid_count; j++) if(nwz_series[i].mid[j] >> 16 == model_id >> 16) return i; return -1; } /* Read nvp node, retrun nonzero on error, update size to actual length. The * index is the raw node number sent to the device */ int read_nvp_node(int node_index, void *buffer, size_t *size) { /* the returned data has a 4 byte header: * - byte 0/1 is the para_noise index, written as a 16bit big-endian number * - byte 2/3 is the node index, written as a 16-bit big-endian number * * NOTE: byte 0 is always 0 because the OF always picks small para_noise * indexes but I guess the actual encoding the one above */ int xfer_size = *size + 4; uint8_t *xfer_buf = malloc(xfer_size); int ret = do_dnk_cmd(true, 0x23, 10, node_index, xfer_buf, &xfer_size); if(ret) return ret; if(xfer_size <= 4) { free(xfer_buf); cprintf(GREY, "Device didn't send any data\n"); return 6; } if(get_big_endian16(xfer_buf + 2) != node_index) { free(xfer_buf); cprintf(GREY, "Device responded with invalid data\n"); return 1; } *size = xfer_size - 4; /* unscramble and copy */ for(int i = 4, idx = get_big_endian16(xfer_buf); i < xfer_size; i++, idx++) xfer_buf[i] ^= para_noise[idx % sizeof(para_noise)]; memcpy(buffer, xfer_buf + 4, *size); free(xfer_buf); return 0; } /* read nvp node, retrun nonzero on error */ int write_nvp_node(int node_index, void *buffer, int size) { /* the data buffer is prepended with a 4 byte header: * - byte 0/1 is the para_noise index, written as a 16bit big-endian number * - byte 2/3 is the node index, written as a 16-bit big-endian number */ int xfer_size = size + 4; uint8_t *xfer_buf = malloc(xfer_size); /* scramble, always use index 0 for para_noise */ set_big_endian16(xfer_buf, 0); /* para_noise index */ set_big_endian16(xfer_buf + 2, node_index); /* node index */ memcpy(xfer_buf + 4, buffer, size); for(int i = 4, idx = get_big_endian16(xfer_buf); i < xfer_size; i++, idx++) xfer_buf[i] ^= para_noise[idx % sizeof(para_noise)]; int ret = do_dnk_cmd(false, 0x24, 10, node_index, xfer_buf, &xfer_size); if(ret) return ret; if(xfer_size - 4 != (int)size) { free(xfer_buf); cprintf(GREY, "Wrong transfer size\n"); return 7; } free(xfer_buf); return 0; } int get_dnk_nvp(int argc, const char* const* argv) { if(argc != 1 && argc != 2) { printf("You must specify a known nvp node or a full node specification:\n"); printf("Node usage: \n"); printf("Node usage: \n"); printf("Nodes:\n"); for(unsigned i = 0; i < NWZ_NVP_COUNT; i++) printf(" %-6s%s\n", nwz_nvp[i].name, nwz_nvp[i].desc); printf("You can also specify a decimal or hexadecimal value directly\n"); return 1; } int series_index, model_index; int ret = get_model_and_series(&model_index, &series_index, NULL); if(ret) return ret; size_t size = 0; /* maybe user specified an explicit size */ if(argc == 2) { char *end; size = strtoul(argv[1], &end, 0); if(*end) { printf("Invalid user-specified size '%s'\n", argv[1]); return 5; } } /* find entry in NVP */ const char *node_name = argv[0]; const char *node_desc = NULL; int node_index = NWZ_NVP_INVALID; for(int i = 0; i < NWZ_NVP_COUNT; i++) if(strcmp(nwz_nvp[i].name, node_name) == 0) { if(nwz_series[series_index].nvp_index) node_index = (*nwz_series[series_index].nvp_index)[i]; if(node_index == NWZ_NVP_INVALID) { printf("This device doesn't have node '%s'\n", node_name); return 5; } node_desc = nwz_nvp[i].desc; /* if not overriden, try to get size from database */ if(size == 0) size = nwz_nvp[i].size; } /* if we can't find it, maybe check if it's a number */ if(node_index == NWZ_NVP_INVALID) { char *end; node_index = strtol(node_name, &end, 0); if(*end) node_index = NWZ_NVP_INVALID; /* string is not a number */ } if(node_index == NWZ_NVP_INVALID) { printf("I don't know about node '%s'\n", node_name); return 4; } /* if we don't have a size, take a big size to be sure */ if(size == 0) { size = 4096; printf("Note: node size unknown, trying to read %u bytes\n", (unsigned)size); } if(g_debug) printf("Asking device for %u bytes\n", (unsigned)size); /* take the size in the database as a hint of the size, but the device could * return less data */ uint8_t *buffer = malloc(size); ret = read_nvp_node(node_index, buffer, &size); if(ret != 0) { cprintf(GREY, "An error occured during request\n"); free(buffer); return ret; } cprintf(GREEN, "%s (node %d%s%s):\n", node_name, node_index, node_desc ? "," : "", node_desc ? node_desc : ""); print_hex(buffer, size); free(buffer); return 0; } int get_dnk_nvp_all() { int series_index, model_index; int ret = get_model_and_series(&model_index, &series_index, NULL); if(ret) return ret; for(int i = 0; i < NWZ_NVP_COUNT; i++) { int node_index = NWZ_NVP_INVALID; if(nwz_series[series_index].nvp_index) node_index = (*nwz_series[series_index].nvp_index)[i]; if(node_index == NWZ_NVP_INVALID) continue; /* device doesn't have this node */ /* this is suboptimal, it will query again the model on each request but there aren't * so many nodes anyway */ char buffer[10]; sprintf(buffer, "%d", node_index); cprintf(BLUE, "querying '%s'\n", nwz_nvp[i].name); get_dnk_nvp(1, &nwz_nvp[i].name); } return 0; } int get_dnk_nvp_multi(int argc, const char* const* argv) { if(argc == 0) { printf("You must specify one or more node names\n"); printf("usage: ..\n"); printf("Nodes:\n"); for(unsigned i = 0; i < NWZ_NVP_COUNT; i++) printf(" %-6s%s\n", nwz_nvp[i].name, nwz_nvp[i].desc); return 1; } /* special case for 'all' */ if(argc == 1 && strcmp(argv[0], "all") == 0) return get_dnk_nvp_all(); /* otherwise normal */ for(int i = 0; i < argc; i++) { cprintf(BLUE, "querying '%s'\n", argv[i]); get_dnk_nvp(1, &argv[i]); } return 0; } int set_dnk_nvp(int argc, const char* const* argv) { if(argc <= 1) { printf("NOTE: this command is potentially very dangerous!\n"); printf("\n"); printf("You must specify a known nvp node or a full node specification:\n"); printf("Usage: --yes-I-want-a-brick \n"); printf("Content must be a list of byte, in decimal or hexadecimal format, e.g. 10 0x30\n"); printf("Nodes:\n"); for(unsigned i = 0; i < NWZ_NVP_COUNT; i++) printf(" %-6s%s\n", nwz_nvp[i].name, nwz_nvp[i].desc); printf("You can also specify a decimal or hexadecimal value directly\n"); return 1; } int series_index, model_index; int ret = get_model_and_series(&model_index, &series_index, NULL); if(ret) return ret; if(!g_yes_i_want_a_brick) { cprintf(RED, "You must pass the option --yes-I-want-a-brick to show that you understand the risk\n"); return 1; } /* find entry in NVP */ const char *node_name = argv[0]; const char *node_desc = NULL; int node_index = NWZ_NVP_INVALID; for(int i = 0; i < NWZ_NVP_COUNT; i++) if(strcmp(nwz_nvp[i].name, node_name) == 0) { if(nwz_series[series_index].nvp_index) node_index = (*nwz_series[series_index].nvp_index)[i]; if(node_index == NWZ_NVP_INVALID) { cprintf(RED, "This device doesn't have node '%s'\n", node_name); return 5; } node_desc = nwz_nvp[i].desc; } /* if we can't find it, maybe check if it's a number */ if(node_index == NWZ_NVP_INVALID) { char *end; node_index = strtol(node_name, &end, 0); if(*end) node_index = NWZ_NVP_INVALID; /* string is not a number */ } if(node_index == NWZ_NVP_INVALID) { cprintf(RED, "I don't know about node '%s'\n", node_name); return 4; } /* build buffer */ int size = argc - 1; uint8_t *buffer = malloc(size); for(int i = 0; i < size; i++) { char *end; long val = strtol(argv[1 + i], &end, 0); if(val < 0 || val >= 256) { cprintf(RED, "value '%s' does not fit into a byte\n", argv[i + 1]); return 1; } buffer[i] = val; } if(g_debug) { cprintf(GREY, "Sending device the following buffer:\n"); print_hex(buffer, size); } ret = write_nvp_node(node_index, buffer, size); if(ret != 0) { cprintf(GREY, "An error occured during request\n"); free(buffer); return ret; } cprintf(GREEN, "Wrote %d bytes to %s (node %d%s%s):\n", size, node_name, node_index, node_desc ? "," : "", node_desc ? node_desc : ""); free(buffer); return 0; } struct dpcc_devinfo_t { uint8_t vendor_identification[8]; uint8_t product_identification[16]; uint8_t product_revision[4]; uint8_t product_sub_revision[4]; uint8_t storage_size[4]; uint8_t serial_number[16]; uint8_t vendor_specific[32]; } __attribute__((packed)); void dpcc_print_devinfo(void *buffer, int buf_size) { if(buf_size < sizeof(struct dpcc_devinfo_t)) { cprintf(GREY, "Cannot parse DEVINFO: buffer too small\n"); return; } struct dpcc_devinfo_t *devinfo = buffer; cprintf_field("Vendor identification: ", "%.8s\n", devinfo->vendor_identification); cprintf_field("Product identification: ", "%.16s\n", devinfo->product_identification); cprintf_field("Product revision: ", "%.4s\n", devinfo->product_revision); cprintf_field("Product sub revision: ", "%.4s\n", devinfo->product_sub_revision); cprintf_field("Storage size: ", "%.4s\n", devinfo->storage_size); cprintf_field("Serial number: ", "%.32s\n", devinfo->serial_number); cprintf_field("Vendor specific: ", "%.32s\n", devinfo->vendor_specific); } typedef void (*dpcc_print_func_t)(void *buffer, int buf_size); struct dpcc_prop_t { char *user_name; char name[8]; uint8_t cdb1; // flags: bit 0 means size flag (means size in paragraph) int size; dpcc_print_func_t print_func; }; struct dpcc_prop_t dpcc_prop_list[] = { { "dev_info", "DEVINFO", 0, 0x80, dpcc_print_devinfo }, /* there are more but they are very obscure */ }; #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) { memcpy((char *)(cdb + 3), prop->name, 7); 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, const char* const* argv) { if(argc != 1 && argc != 3) { printf("You must specify a known property name or a full property specification:\n"); printf("Full usage: \n"); printf("Property usage: \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) { 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 = malloc(prop.size); int buffer_size = prop.size; int ret = do_dpcc_cmd(0, &prop, buffer, &buffer_size); if(ret) { cprintf(GREY, "An error occured during request\n"); return ret; } if(buffer_size < prop.size) buffer[buffer_size] = 0; cprintf_field("Raw data: ", "%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, const char* const* argv) { (void) argc; (void )argv; void *buffer = malloc(32); int buffer_size = 32; int ret = do_dpcc_cmd(1, NULL, buffer, &buffer_size); if(ret) { cprintf(GREY, "An error occured during request\n"); 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, const char* const* argv) { (void) argc; (void )argv; uint8_t cdb[12] = {0xfc, 0, 0x20, 'd', 'b', 'm', 'n', 0, 0x80, 0, 0, 0}; char buffer[0x80]; int buffer_size = sizeof(buffer); 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) { cprintf(GREY, "An error occured during request\n"); return ret; } cprintf_field("Raw device info:", "\n"); print_hex(buffer, buffer_size); // the 16 first bytes are 'DEVINFO', 0x80, followed by zeroes dpcc_print_devinfo(buffer + 16, buffer_size - 16); return 0; } int get_dhp(int argc, const char* const* argv) { (void) argc; (void) argv; uint8_t cdb[12] = {0xfc, 0, 'D', 'd', 'h', 'p', 0, 0, 0, 0, 0, 0}; char *buffer = malloc(0x80); 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) { cprintf(GREY, "An error occured during request\n"); return ret; } buffer[buffer_size] = 0; cprintf_field("Destination/Headphones:", "\n"); print_hex(buffer, buffer_size); return 0; } int try_fw_upgrade(unsigned flags, int argc, const char* const* argv) { (void) argc; (void) argv; /* older devices may have used subcommand 3 instead of 4, but this is not * supported by any device I have seen */ uint8_t cdb[12] = {0xfc, 0, 0x04, 'd', 'b', 'm', 'n', 0, flags, 0, 0, 0}; char *buffer = malloc(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) { cprintf(GREY, "An error occured during request\n"); return ret; } /* the device is not supposed to send back anything but if it does, print it */ if(buffer_size > 0) { buffer[buffer_size] = 0; cprintf_field("Result:", "\n"); print_hex(buffer, buffer_size); } return 0; } int do_fw_upgrade(int argc, const char* const* argv) { if(!try_fw_upgrade(0x80, argc, argv)) return 0; cprintf(GREY, "Trying alternative firmware upgrade command...\n"); return try_fw_upgrade(0, argc, argv); } static struct { unsigned long dest; const char *name; } g_dest_list[] = { { 0, "J" }, { 1, "U" }, { 0x101, "U2" }, { 0x201, "U3" }, { 0x301, "CA" }, { 2, "CEV" }, { 0x102, "CE7" }, { 3, "CEW" }, { 0x103, "CEW2" }, { 4, "CN" }, { 5, "KR" }, { 6, "E" }, { 0x106, "MX" }, { 0x206, "E2" }, { 0x306, "MX3" }, { 7, "TW" }, }; #define DEST_COUNT (sizeof(g_dest_list) / sizeof(g_dest_list[0])) int do_dest(int argc, const char* const* argv) { /* it is possile to write any NVP node using the SCSI interface but only * give the user the possibility to write destination, because that's the * most useful one */ if(argc != 1 && argc != 3) { printf("Usage: get\n"); printf("Usage: set \n"); printf("Destination () can be either an integer or one of:\n"); for(size_t i = 0; i < DEST_COUNT; i++) printf(" %s\n", g_dest_list[i].name); printf("Sound pressure () can be be an integer, 'on' or 'off'\n"); return 1; } /* get model/series */ int model_index, series_index; unsigned long model_id; int ret = get_model_and_series(&model_index, &series_index, &model_id); if(ret) return ret; int shp_index = NWZ_NVP_INVALID; if(nwz_series[series_index].nvp_index) shp_index = (*nwz_series[series_index].nvp_index)[NWZ_NVP_SHP]; if(shp_index == NWZ_NVP_INVALID) { printf("This device doesn't have node 'shp'\n"); help_us(true, model_id); return 5; } /* in all cases, we need to read shp */ size_t size = nwz_nvp[NWZ_NVP_SHP].size; uint8_t *shp = malloc(size); ret = read_nvp_node(shp_index, shp, &size); if(ret != 0) { printf("Cannot read node 'shp'\n"); free(shp); return ret; } /* get */ if(strcmp(argv[0], "get") == 0) { if(argc != 1) { printf("Too many arguments for get\n"); free(shp); return 2; } const char *dst_name = "Unknown"; unsigned long dst = get_little_endian32(shp); for(size_t i = 0; i < DEST_COUNT; i++) if(dst == g_dest_list[i].dest) dst_name = g_dest_list[i].name; cprintf_field("Destination: ", "%s (%lx)\n", dst_name, dst); unsigned long sps = get_little_endian32(shp + 4); cprintf_field("Sound pressure: ", "%lu (%s)\n", sps, sps == 0 ? "off" : "on"); free(shp); } /* set */ if(strcmp(argv[0], "set") == 0) { if(argc != 3) { printf("Not enough arguments for set\n"); free(shp); return 2; } /* try to parse dest as integer */ char *end; unsigned long dst = strtoul(argv[1], &end, 0); if(*end) { /* assume string */ int index = -1; for(size_t i = 0; i < DEST_COUNT; i++) if(strcmp(argv[1], g_dest_list[i].name) == 0) index = i; if(index == -1) { printf("Unknown destination '%s'\n", argv[1]); free(shp); return 3; } dst = g_dest_list[index].dest; } /* try to parse sps as integer */ /* try to parse dest as integer */ unsigned long sps = strtoul(argv[2], &end, 0); if(*end) { /* assume string */ if(strcmp(argv[2], "on") == 0) sps = 1; else if(strcmp(argv[2], "off") == 0) sps = 0; else { printf("Unknown sound pressure setting '%s'\n", argv[2]); free(shp); return 3; } } set_little_endian32(shp, dst); set_little_endian32(shp + 4, sps); int ret = write_nvp_node(shp_index, shp, size); free(shp); if(ret != 0) printf("An error occured when writing node: %d\n", ret); else printf("Destination successfully changed.\nPlease RESET ALL SETTINGS on your device!\n"); return ret; } return 0; } int do_help_us(int argc, const char* const* argv) { unsigned long model_id; if(get_model_id(&model_id)) return 1; help_us(false, model_id); return 0; } typedef int (*cmd_fn_t)(int argc, const char* const* 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 }, { "set_dnk_nvp", "Set DNK NVP content", set_dnk_nvp}, { "get_dnk_nvp_multi", "Get several DNK NVP content", get_dnk_nvp_multi }, { "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 }, { "get_dhp", "Get destination headphones", get_dhp }, { "do_fw_upgrade", "Do a firmware upgrade", do_fw_upgrade }, { "dest_tool", "Get/Set destination and sound pressure regulation", do_dest }, { "help_us", "Provide useful information for developers to help us", do_help_us }, }; #define NR_CMDS (sizeof(cmd_list) / sizeof(cmd_list[0])) int process_cmd(const char *cmd, int argc, const char* const* 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:\n"); printf(" scsitool [options] list_devices\n"); printf(" scsitool [options] decode_scsi \n"); printf(" scsitool [options] [arguments]\n"); printf("Options:\n"); printf(" -o Set output prefix\n"); printf(" -h/--help Display this message\n"); printf(" -d/--debug Display debug messages\n"); printf(" -c/--no-color Disable color output\n"); printf(" -s/--series Force series (disable auto-detection, use '?' for the list)\n"); printf(" -a/--all List all devices, not just Sony (for list_devices)\n"); printf("Commands:\n"); printf(" %-20sList all connected Sony devices (experimental)\n", "list_devices"); for(unsigned i = 0; i < NR_CMDS; i++) printf(" %-20s%s\n", cmd_list[i].name, cmd_list[i].desc); exit(1); } void help_us(bool unsupported, unsigned long model_id) { if(unsupported) cprintf(RED, "Your device is not supported yet.\n"); cprintf(RED, "Please contact developers and send them the information below.\n"); cprintf(RED, "See https://www.rockbox.org/wiki/SonyNWDestTool#ReportDevice\n"); /* try to see if we know a device in the same series, so we can recommend a force action */ int series_idx = guess_series_for_model(model_id); if(series_idx >= 0) { cprintf(OFF, "It seems your devices belongs to the %s series.\n", nwz_series[series_idx].name); cprintf(OFF, "You can try to re-run this tool with the option -s %s\n", nwz_series[series_idx].codename); } cprintf(BLUE, "-------------------[ Paste information below ]-------------------\n"); cprintf_field("Model ID: ", "%#lx\n", model_id); get_dev_info(0, NULL); get_dhp(0, NULL); } static const char *get_model_name(uint32_t model_id) { int index = -1; for(int i = 0; i < NWZ_MODEL_COUNT; i++) if(nwz_model[i].mid == model_id) index = i; return index == -1 ? "Unknown" : nwz_model[index].name; } static bool is_sony_device(struct rb_scsi_devent_t *dev) { /* vendor must be SONY */ if(!dev->vendor || strcmp(dev->vendor, "SONY ") != 0) return false; /* model must be WALKMAN */ if(!dev->model || strcmp(dev->model, "WALKMAN ") != 0) return false; return true; } static int list_devices(bool list_all) { printf("%s device(s) found:\n", list_all ? "SCSI" : "Sony"); struct rb_scsi_devent_t *dev = rb_scsi_list(); if(dev == NULL) { cprintf(GREY, "Could not get the list of devices.\n"); return 1; } bool empty_list = true; for(struct rb_scsi_devent_t *p = dev; p->scsi_path; p++) { /* filter non-Sony devices */ if(!list_all && !is_sony_device(p)) continue; printf("%-20s%-20s%s %s %s\n", p->scsi_path, p->block_path, p->vendor, p->model, p->rev); empty_list = false; } if(empty_list) cprintf(GREY, "No devices found.\n"); rb_scsi_free_list(dev); return 0; } inline uint8_t xdigit2val(char c) { if('0' <= c && c <= '9') return c - '0'; else if('a' <= c && c <= 'f') return c - 'a' + 10; else if('A' <= c && c <= 'F') return c - 'A' + 10; else return 255; } static int decode_scsi_a3(uint8_t *cdb, int cdb_len) { cprintf_field("Opcode: ", "0xa3\n"); cprintf(RED, "Unimplemented\n"); return 0; } static int decode_scsi_a4(uint8_t *cdb, int cdb_len) { cprintf_field("Opcode: ", "0xa4\n"); cprintf(RED, "Unimplemented\n"); return 0; } static int decode_scsi_empr_dpcc(uint8_t *cdb, int cdb_len) { cprintf_field("Opcode: ", "%#x (EMPR DPCC)\n", cdb[0]); cprintf(RED, "Unimplemented\n"); return 0; } static int decode_scsi_dnk(uint8_t *cdb, int cdb_len) { /* * DNK (Device Node Key) commands are 12 bytes long and the CDB is of the form * DD 00 00 00 xx xx 00 BC ll ll xx yy * where * - xxxx is the argument (16 bits) * - llll is the length (16 bits) * - xx is the command * - yy is the subcommand */ cprintf_field("Opcode: ", "%#x (DNK)\n", cdb[0]); if(cdb_len != 12 || cdb[7] != 0xbc) { cprintf(GREY, "Invalid length for a DNK command\n"); return 1; } uint8_t cmd = cdb[10]; uint8_t subcmd = cdb[11]; uint16_t length = cdb[8] << 8 | cdb[9]; uint16_t arg = cdb[4] << 8 | cdb[5]; const char *cmd_str = "Unknown"; const char *subcmd_str = "Unknown"; switch(cmd) { case 0x11: cmd_str = "Read Leaf ID"; subcmd_str = "Unused"; break; case 0x12: cmd_str = "Get DEVINFO"; subcmd_str = "Unused"; break; case 0x21: cmd_str = "Report DNK"; subcmd_str = "Unused"; break; case 0x23: cmd_str = "Read Regs"; switch(subcmd) { case 1: subcmd_str = "Serial Num"; break; case 4: subcmd_str = "Storage Size"; break; case 5: subcmd_str = "DKS"; break; case 6: subcmd_str = "Product ID"; break; case 7: subcmd_str = "Product Group Scramble"; break; case 8: subcmd_str = "Destination"; break; case 9: subcmd_str = "Model ID"; break; case 0xa: subcmd_str = "NVP"; break; case 0xb: subcmd_str = "Marlin/Starfish stuff"; break; case 0xc: subcmd_str = "Unclear/Constant"; break; case 0xd: subcmd_str = "Secure Clock"; break; case 0xe: subcmd_str = "AAD ICV"; break; case 0xf: subcmd_str = "EMPR ICV"; break; case 0x10: subcmd_str = "Test Mode Flags"; break; case 0x11: subcmd_str = "Getty Mode Flags"; break; case 0x12: subcmd_str = "Key Mode Debug/Release"; break; case 0x13: subcmd_str = "System Information"; break; case 0x14: subcmd_str = "Random Data?"; break; case 0x15: subcmd_str = "Update Filename"; break; case 0x16: subcmd_str = "Key and Signature"; break; case 0x17: subcmd_str = "Bluetooth Parameters"; break; case 0x18: subcmd_str = "EMPR 0"; break; case 0x19: subcmd_str = "EMPR 1"; break; case 0x1a: subcmd_str = "EMPR 2"; break; case 0x1b: subcmd_str = "Color Variation"; break; case 0x1c: subcmd_str = "MTM Sec?"; break; case 0x1d: subcmd_str = "Slacker Time"; break; case 0x1f: subcmd_str = "Slacker ID File"; break; case 0x20: subcmd_str = "FM Parameters"; break; case 0x21: subcmd_str = "Speaker Ship Info"; break; case 0x22: subcmd_str = "BTMW Factory Pair Info"; break; case 0x23: subcmd_str = "U-boot Password"; break; case 0x24: subcmd_str = "Noise Cancel Driver Parameter"; break; case 0x25: subcmd_str = "Bluetooth PSKey"; break; case 0x27: subcmd_str = "VID/PID"; break; } default: break; } cprintf_field("Command: ", "%#x (%s)\n", cmd, cmd_str); cprintf_field("Sub-cmd: ", "%#x (%s)\n", subcmd, subcmd_str); cprintf_field("Argument: ", "%#x\n", arg); cprintf_field("Length: ", "%#x\n", length); return 0; } static int decode_scsi_dpcc(uint8_t *cdb, int cdb_len) { cprintf_field("Opcode: ", "%#x (DPCC)\n", cdb[0]); cprintf(RED, "Unimplemented\n"); return 0; } static int decode_scsi_fc(uint8_t *cdb, int cdb_len) { cprintf_field("Opcode: ", "0xfc\n"); if(cdb[3] == 'd' && cdb[4] == 'b' && cdb[5] == 'm' && cdb[6] == 'n') { uint8_t cmd = cdb[2]; uint8_t flags = cdb[8]; const char *cmd_name = "Unknown"; if(cmd == 0x04) cmd_name = "Firmware Upgrade"; if(cmd == 0x20) cmd_name = "Get Device Info"; cprintf(BLUE, "Device request:\n"); cprintf_field(" Command: ", "%#x (%s)\n", cmd, cmd_name); cprintf_field(" Flags(?): ", "%#x (Unknown)\n", flags); } return 0; } static int decode_scsi(int argc, const char* const* argv) { /* we need to parse the arguments, we support either as one big hexdump: * fc002064626d6e0080000000 * or as a sequence of hex bytes: * fc 00 20 64 62 6d 6e 00 80 00 00 00 */ if(argc == 0) { cprintf(GREY, "You need to specify the CDB to decode\n"); return 1; } #define MAX_CDB 16 uint8_t cdb[MAX_CDB]; int cdb_len; if(argc > MAX_CDB) { cprintf(GREY, "This does not look like a CDB (more than %d bytes)\n", MAX_CDB); return 1; } if(argc == 1) { /* allow the string to start with 0x */ const char *str = argv[0]; if(str[0] == '0' && str[1] == 'x') str += 2; cdb_len = strlen(str); if(cdb_len % 2) { cprintf(GREY, "The CDB must contain an even number of hex digits!\n"); return 1; } cdb_len /= 2; for(int i = 0; i < cdb_len; i++) { if(!isxdigit(str[2 * i]) || !isxdigit(str[2 * i + 1])) { cprintf(GREY, "The CDB must contain hex digits!\n"); return 1; } cdb[i] = xdigit2val(str[2 * i]) << 4 | xdigit2val(str[2 * i + 1]); } } else { cprintf(GREY, "unimplemented\n"); return 1; } cprintf(GREEN, "CDB: "); print_hex(cdb, cdb_len); if(cdb[0] == CMD_A3) return decode_scsi_a3(cdb, cdb_len); if(cdb[0] == CMD_A4) return decode_scsi_a4(cdb, cdb_len); if(cdb[0] == CMD_EMPR_DPCC) return decode_scsi_empr_dpcc(cdb, cdb_len); if(cdb[0] == CMD_DNK) return decode_scsi_dnk(cdb, cdb_len); if(cdb[0] == CMD_DPCC) return decode_scsi_dpcc(cdb, cdb_len); if(cdb[0] == 0xfc) return decode_scsi_fc(cdb, cdb_len); else { cprintf(RED, "I cannot decode this SCSI command\n"); } return 0; } int main(int argc, char **argv) { bool list_all = false; while(1) { static struct option long_options[] = { {"help", no_argument, 0, 'h'}, {"debug", no_argument, 0, 'd'}, {"no-color", no_argument, 0, 'c'}, {"series", required_argument, 0, 's'}, {"all", no_argument, 0, 'a'}, {"yes-I-want-a-brick", no_argument, 0, -2}, {0, 0, 0, 0} }; int c = getopt_long(argc, argv, "?dcfo:s:a", long_options, NULL); if(c == -1) break; switch(c) { case -2: g_yes_i_want_a_brick = true; break; case -1: break; case 'c': enable_color(false); break; case 'd': g_debug = true; break; case 'h': usage(); break; case 'o': g_out_prefix = optarg; break; case 's': g_force_series = optarg; break; case 'a': list_all = true; break; default: abort(); } } /* list devices */ if(g_force_series && strcmp(g_force_series, "?") == 0) { cprintf(OFF, "Series list:\n"); for(int i = 0; i < NWZ_SERIES_COUNT; i++) { printf(" %-10s %s (", nwz_series[i].codename, nwz_series[i].name); for(int j = 0; j < nwz_series[i].mid_count; j++) { if(j != 0) printf(", "); printf("%s", get_model_name(nwz_series[i].mid[j])); } printf(")\n"); } return 0; } /* special list_devices/decode_scsi command */ if(argc >= optind + 1 && strcmp(argv[optind], "list_devices") == 0) return list_devices(list_all); if(argc >= optind + 1 && strcmp(argv[optind], "decode_scsi") == 0) return decode_scsi(argc - optind - 1, (const char * const *)(argv + optind + 1)); if(argc - optind < 2) { usage(); return 1; } int ret = 0; int flags = 0; if(g_debug) flags |= RB_SCSI_DEBUG; g_dev = rb_scsi_open(argv[optind], flags, NULL, NULL); if(g_dev == 0) { cprintf(GREY, "Cannot open device\n"); ret = 1; goto Lend; } ret = process_cmd(argv[optind + 1], argc - optind - 2, (const char * const *)(argv + optind + 2)); rb_scsi_close(g_dev); Lend: color(OFF); return ret; }