From 91ede1ea08ae217ccd22cdc936e474cdc0a4466c Mon Sep 17 00:00:00 2001
From: Amaury Pouly <amaury.pouly@gmail.com>
Date: Sun, 18 Jun 2017 13:06:12 +0200
Subject: [PATCH] nwztools/scsitool: fix devinfo, add dhp

The devinfo request returned the raw data, now the tool prints the various
fields. Also add support for the dhp (destination/headphones/color ...): this
one is untested because it's only supported starting from A10 or A20. There is
still a problem with the dpcc prop: although it should work for DEVINFO, it does
not, despite the fact that the get_dev_info command works and is internally (on
the Sony) translated into a dpcc request. I keep the code just in case.

Change-Id: I5aa8ef4afb0b11d3c0ddfa3d38f3e737ee1aff66
---
 utils/nwztools/scsitools/scsitool.c | 75 ++++++++++++++++++++++++++---
 1 file changed, 69 insertions(+), 6 deletions(-)

diff --git a/utils/nwztools/scsitools/scsitool.c b/utils/nwztools/scsitools/scsitool.c
index 6c3527713e..a90729a3d5 100644
--- a/utils/nwztools/scsitools/scsitool.c
+++ b/utils/nwztools/scsitools/scsitool.c
@@ -512,17 +512,48 @@ int get_dnk_nvp(int argc, char **argv)
     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[7];
-    uint8_t cdb1;
+    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 },
+    { "dev_info", "DEVINFO", 0, 0x80, dpcc_print_devinfo },
     /* there are more but they are very obscure */
 };
 
@@ -534,7 +565,7 @@ int do_dpcc_cmd(uint32_t cmd, struct dpcc_prop_t *prop, void *buffer, int *buffe
     cdb[2] = cmd;
     if(cmd == 0)
     {
-        strncpy((char *)(cdb + 3), prop->name, 7); // warning: erase cdb[10] !
+        memcpy((char *)(cdb + 3), prop->name, 7);
         cdb[1] = prop->cdb1;
         if(prop->cdb1 & 1)
             cdb[10] = (*buffer_size + 15) / 16;
@@ -575,7 +606,7 @@ int get_dpcc_prop(int argc, char **argv)
         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)
+        if(prop.user_name == 0)
         {
             cprintf(GREY, "Unknown property '%s'\n", argv[0]);
             return 1;
@@ -641,7 +672,7 @@ int get_dev_info(int argc, char **argv)
     (void )argv;
     uint8_t cdb[12] = {0xfc, 0, 0x20, 'd', 'b', 'm', 'n', 0, 0x80, 0, 0, 0};
 
-    char *buffer = malloc(0x81);
+    char *buffer = malloc(0x80);
     int buffer_size = 0x80;
     uint8_t sense[32];
     int sense_size = 32;
@@ -651,9 +682,40 @@ int get_dev_info(int argc, char **argv)
         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("Device Info:", "\n");
+    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, char **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;
 }
@@ -839,6 +901,7 @@ struct cmd_t cmd_list[] =
     { "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 },
 };