ADD: using @holiman's "ubs_poll_validate_length() function in some device-side...

[proxmark3-svn] / client / cmdhw.c

diff --git a/client/cmdhw.c b/client/cmdhw.c
index 5ec0aa601eb8024676220ca340a1431930fb861b..5f25241cafe1f91a80e1cc2a41c8936511901d52 100644 (file)
--- a/client/cmdhw.c
+++ b/client/cmdhw.c
@@ -23,9 +23,11 @@
 
 static int CmdHelp(const char *Cmd);
 
 
 static int CmdHelp(const char *Cmd);
 

-static void lookupChipID(uint32_t iChipID)
+static void lookupChipID(uint32_t iChipID, uint32_t mem_used)

 {
        char asBuff[100];
 {
        char asBuff[100];

+       uint32_t mem_avail = 0;
+       

        switch(iChipID)
        {
                case 0x270B0A40:
        switch(iChipID)
        {
                case 0x270B0A40:


@@ -103,37 +105,43 @@ static void lookupChipID(uint32_t iChipID)
        switch((iChipID&0xF00)>>8)
        {
                case 0:
        switch((iChipID&0xF00)>>8)
        {
                case 0:

-                       sprintf(asBuff,"None");
+                       mem_avail = 0;

                        break;
                case 1:
                        break;
                case 1:

-                       sprintf(asBuff,"8K bytes");
+                       mem_avail = 8;

                        break;
                case 2:
                        break;
                case 2:

-                       sprintf(asBuff,"16K bytes");
+                       mem_avail = 16;

                        break;
                case 3:
                        break;
                case 3:

-                       sprintf(asBuff,"32K bytes");
+                       mem_avail = 32;

                        break;
                case 5:
                        break;
                case 5:

-                       sprintf(asBuff,"64K bytes");
+                       mem_avail = 64;

                        break;
                case 7:
                        break;
                case 7:

-                       sprintf(asBuff,"128K bytes");
+                       mem_avail = 128;

                        break;
                case 9:
                        break;
                case 9:

-                       sprintf(asBuff,"256K bytes");
+                       mem_avail = 256;

                        break;
                case 10:
                        break;
                case 10:

-                       sprintf(asBuff,"512K bytes");
+                       mem_avail = 512;

                        break;
                case 12:
                        break;
                case 12:

-                       sprintf(asBuff,"1024K bytes");
+                       mem_avail = 1024;

                        break;
                case 14:
                        break;
                case 14:

-                       sprintf(asBuff,"2048K bytes");
+                       mem_avail = 2048;

                        break;
        }
                        break;
        }

-       PrintAndLog("Nonvolatile Program Memory Size: %s",asBuff);
+       PrintAndLog("Nonvolatile Program Memory Size: %dK bytes. Used: %d bytes (%2.0f\%). Free: %d bytes (%2.0f\%).", 
+                               mem_avail, 
+                               mem_used, 
+                               mem_avail == 0 ? 0 : (float)mem_used/(mem_avail*1024)*100,
+                               mem_avail*1024 - mem_used,
+                               mem_avail == 0 ? 0 : (float)(mem_avail*1024-mem_used)/(mem_avail*1024)*100
+                               );

        switch((iChipID&0xF000)>>12)
        {
                case 0:
        switch((iChipID&0xF000)>>12)
        {
                case 0:


@@ -396,12 +404,46 @@ int CmdTune(const char *Cmd)
 
 int CmdVersion(const char *Cmd)
 {
 
 int CmdVersion(const char *Cmd)
 {

+
+       clearCommandBuffer();

   UsbCommand c = {CMD_VERSION};
   UsbCommand c = {CMD_VERSION};

-  UsbCommand resp;
+       static UsbCommand resp = {0, {0, 0, 0}};
+
+       if (resp.arg[0] == 0 && resp.arg[1] == 0) { // no cached information available

   SendCommand(&c);
   SendCommand(&c);

-  if (WaitForResponseTimeout(CMD_ACK,&resp,1000)) {
-      lookupChipID(resp.arg[0]);
-  }
+       if (WaitForResponseTimeout(CMD_ACK,&resp,1000)) {
+               PrintAndLog("Prox/RFID mark3 RFID instrument");
+               PrintAndLog((char*)resp.d.asBytes);
+               lookupChipID(resp.arg[0], resp.arg[1]);
+       }
+       } else {
+               PrintAndLog("[[[ Cached information ]]]\n");
+               PrintAndLog("Prox/RFID mark3 RFID instrument");
+               PrintAndLog((char*)resp.d.asBytes);
+               lookupChipID(resp.arg[0], resp.arg[1]);
+               PrintAndLog("");
+       }
+       return 0;
+}
+
+int CmdStatus(const char *Cmd)
+{
+       UsbCommand c = {CMD_STATUS};
+       SendCommand(&c);
+       return 0;
+}
+
+int CmdPing(const char *Cmd)
+{
+       clearCommandBuffer();
+       UsbCommand resp;
+       UsbCommand c = {CMD_PING};
+       SendCommand(&c);
+       if (WaitForResponseTimeout(CMD_ACK,&resp,1000)) {
+               PrintAndLog("Ping successfull");
+       }else{
+               PrintAndLog("Ping failed");
+       }

   return 0;
 }
 
   return 0;
 }
 


@@ -418,6 +460,8 @@ static command_t CommandTable[] =
   {"setmux",        CmdSetMux,      0, "<loraw|hiraw|lopkd|hipkd> -- Set the ADC mux to a specific value"},
   {"tune",          CmdTune,        0, "Measure antenna tuning"},
   {"version",       CmdVersion,     0, "Show version information about the connected Proxmark"},
   {"setmux",        CmdSetMux,      0, "<loraw|hiraw|lopkd|hipkd> -- Set the ADC mux to a specific value"},
   {"tune",          CmdTune,        0, "Measure antenna tuning"},
   {"version",       CmdVersion,     0, "Show version information about the connected Proxmark"},

+       {"status",        CmdStatus,      0, "Show runtime status information about the connected Proxmark"},
+       {"ping",          CmdPing,        0, "Test if the pm3 is responsive"},

   {NULL, NULL, 0, NULL}
 };
 
   {NULL, NULL, 0, NULL}
 };