CHG: "hf legic write" - now writes on the limits better.

[proxmark3-svn] / client / cmdhflegic.c

diff --git a/client/cmdhflegic.c b/client/cmdhflegic.c
index ab9df4875f655d39e915e2e23dd016dfd035f61b..b2d207e5856bb4cd41ff4b4fb474050ef6843e16 100644 (file)
--- a/client/cmdhflegic.c
+++ b/client/cmdhflegic.c
@@ -7,447 +7,1192 @@
 //-----------------------------------------------------------------------------
 // High frequency Legic commands
 //-----------------------------------------------------------------------------
 //-----------------------------------------------------------------------------
 // High frequency Legic commands
 //-----------------------------------------------------------------------------

-
-#include <stdio.h>
-#include <string.h>
-#include "proxmark3.h"
-#include "data.h"
-#include "ui.h"
-#include "cmdparser.h"

 #include "cmdhflegic.h"
 #include "cmdhflegic.h"

-#include "cmdmain.h"
-#include "util.h"
-#include "crc.h"
+

 static int CmdHelp(const char *Cmd);
 
 static int CmdHelp(const char *Cmd);
 

-int usage_legic_calccrc8(void){
-       PrintAndLog("Calculates the legic crc8 on the input hexbytes.");
+#define MAX_LENGTH 1024        
+
+int usage_legic_calccrc(void){
+       PrintAndLog("Calculates the legic crc8/crc16 on the given data.");

        PrintAndLog("There must be an even number of hexsymbols as input.");
        PrintAndLog("There must be an even number of hexsymbols as input.");

-       PrintAndLog("Usage:  hf legic crc8 <hexbytes>");
-       PrintAndLog("Options :");
-       PrintAndLog("  <hexbytes>   : hex bytes in a string");
+       PrintAndLog("Usage:  hf legic crc [h] d <data> u <uidcrc> c <8|16>");
+       PrintAndLog("Options:");
+       PrintAndLog("      h             : this help");
+       PrintAndLog("      d <data>      : (hex symbols) bytes to calculate crc over");
+       PrintAndLog("      u <uidcrc>    : MCC hexbyte");
+       PrintAndLog("      c <8|16>      : Crc type");

        PrintAndLog("");
        PrintAndLog("");

-       PrintAndLog("Sample  : hf legic crc8 deadbeef1122");
+       PrintAndLog("Samples:");
+       PrintAndLog("      hf legic crc b deadbeef1122");
+       PrintAndLog("      hf legic crc b deadbeef1122 u 9A c 16");

        return 0;
 }
        return 0;
 }

-
-int usage_legic_load(void){
-       PrintAndLog("It loads datasamples from the file `filename` to device memory");
-       PrintAndLog("Usage:  hf legic load <file name>");
-       PrintAndLog(" sample: hf legic load filename");
+int usage_legic_rdmem(void){   
+       PrintAndLog("Read data from a legic tag.");
+       PrintAndLog("Usage:  hf legic rdmem [h] <offset> <length> <IV>");
+       PrintAndLog("Options:");
+       PrintAndLog("      h             : this help");
+       PrintAndLog("      <offset>      : (hex) offset in data array to start download from");
+       PrintAndLog("      <length>      : (hex) number of bytes to read");
+       PrintAndLog("      <IV>          : (hex) (optional) Initialization vector to use. Must be odd and 7bits max");
+       PrintAndLog("");
+       PrintAndLog("Samples:");
+       PrintAndLog("      hf legic rdmem 0 16        - reads from byte[0] 0x16 bytes(system header)");
+       PrintAndLog("      hf legic rdmem 0 4 55      - reads from byte[0] 0x4 bytes with IV 0x55");
+       PrintAndLog("      hf legic rdmem 0 100 55    - reads 0x100 bytes with IV 0x55");
+       return 0;
+}
+int usage_legic_sim(void){
+       PrintAndLog("Simulates a LEGIC Prime tag. MIM22, MIM256, MIM1024 types can be emulated");
+       PrintAndLog("Use eload/esave to upload a dump into emulator memory");
+       PrintAndLog("Usage:  hf legic sim [h] <tagtype> <phase> <frame> <reqresp>");
+       PrintAndLog("Options:");
+       PrintAndLog("      h             : this help");
+       PrintAndLog("      <tagtype>     : 0 = MIM22");
+       PrintAndLog("                    : 1 = MIM256 (default)");
+       PrintAndLog("                    : 2 = MIM1024");       
+       PrintAndLog("      <phase>       : phase drift");
+       PrintAndLog("      <frame>       : frame drift");
+       PrintAndLog("      <reqresp>     : reqresp drift");
+       PrintAndLog("");
+       PrintAndLog("Samples:");
+       PrintAndLog("      hf legic sim");
+       PrintAndLog("      hf legic sim ");
+       return 0;
+}
+int usage_legic_write(void){
+       PrintAndLog("Write data to a LEGIC Prime tag. It autodetects tagsize to make sure size");
+       PrintAndLog("Usage:  hf legic write [h] o <offset> d <data (hex symbols)>");
+       PrintAndLog("Options:");
+       PrintAndLog("      h             : this help");
+       PrintAndLog("      o <offset>    : (hex) offset in data array to start writing");
+       //PrintAndLog("  <IV>          : (optional) Initialization vector to use (ODD and 7bits)");
+       PrintAndLog("      d <data>      : (hex symbols) bytes to write ");
+       PrintAndLog("");
+       PrintAndLog("Samples:");
+       PrintAndLog("      hf legic write o 10 d 11223344    - Write 0x11223344 starting from offset 0x10");
+       return 0;
+}
+int usage_legic_reader(void){
+       PrintAndLog("Read UID and type information from a legic tag.");
+       PrintAndLog("Usage:  hf legic reader [h]");
+       PrintAndLog("Options:");
+       PrintAndLog("      h             : this help");
+       PrintAndLog("");
+       PrintAndLog("Samples:");
+       PrintAndLog("      hf legic reader");
+       return 0;
+}
+int usage_legic_info(void){
+       PrintAndLog("Reads information from a legic prime tag.");
+       PrintAndLog("Shows systemarea, user areas etc");
+       PrintAndLog("Usage:  hf legic info [h]");
+       PrintAndLog("Options:");
+       PrintAndLog("      h             : this help");
+       PrintAndLog("");
+       PrintAndLog("Samples:");
+       PrintAndLog("      hf legic info");
+       return 0;
+}
+int usage_legic_dump(void){
+       PrintAndLog("Reads all pages from LEGIC Prime MIM22, MIM256, MIM1024");
+       PrintAndLog("and saves binary dump into the file `filename.bin` or `cardUID.bin`");
+       PrintAndLog("It autodetects card type.\n");     
+       PrintAndLog("Usage:  hf legic dump [h] o <filename w/o .bin>");
+       PrintAndLog("Options:");
+       PrintAndLog("      h             : this help");
+       PrintAndLog("      o <filename>  : filename w/o '.bin' to dump bytes");
+       PrintAndLog("");
+       PrintAndLog("Samples:");
+       PrintAndLog("      hf legic dump");
+       PrintAndLog("      hf legic dump o myfile");
+       return 0;
+}
+int usage_legic_restore(void){
+       PrintAndLog("Reads binary file and it autodetects card type and verifies that the file has the same size");
+       PrintAndLog("Then write the data back to card. All bytes except the first 7bytes [UID(4) MCC(1) DCF(2)]\n");
+       PrintAndLog("Usage:   hf legic restore [h] i <filename w/o .bin>");
+       PrintAndLog("Options:");
+       PrintAndLog("      h             : this help");
+       PrintAndLog("      i <filename>  : filename w/o '.bin' to restore bytes on to card from");
+       PrintAndLog("");
+       PrintAndLog("Samples:");
+       PrintAndLog("      hf legic restore i myfile");
+       return 0;
+}
+int usage_legic_eload(void){
+       PrintAndLog("It loads binary dump from the file `filename.bin`");
+       PrintAndLog("Usage:  hf legic eload [h] [card memory] <file name w/o `.bin`>");
+       PrintAndLog("Options:");
+       PrintAndLog("      h             : this help"); 
+       PrintAndLog("      [card memory] : 0 = MIM22");
+       PrintAndLog("                    : 1 = MIM256 (default)");
+       PrintAndLog("                    : 2 = MIM1024");
+       PrintAndLog("      <filename>    : filename w/o .bin to load"); 
+       PrintAndLog("");
+       PrintAndLog("Samples:");
+       PrintAndLog("      hf legic eload 2 myfile");
+       return 0;
+}
+int usage_legic_esave(void){
+       PrintAndLog("It saves binary dump into the file `filename.bin` or `cardID.bin`");
+       PrintAndLog(" Usage:  hf legic esave [h] [card memory] [file name w/o `.bin`]");
+       PrintAndLog("Options:");
+       PrintAndLog("      h             : this help");
+       PrintAndLog("      [card memory] : 0 = MIM22");
+       PrintAndLog("                    : 1 = MIM256 (default)");
+       PrintAndLog("                    : 2 = MIM1024");
+       PrintAndLog("      <filename>    : filename w/o .bin to load"); 
+       PrintAndLog("");
+       PrintAndLog("Samples:");        
+       PrintAndLog("      hf legic esave 2 myfile");

        return 0;
 }
 
 /*
  *  Output BigBuf and deobfuscate LEGIC RF tag data.
        return 0;
 }
 
 /*
  *  Output BigBuf and deobfuscate LEGIC RF tag data.

- *   This is based on information given in the talk held
+ *  This is based on information given in the talk held

  *  by Henryk Ploetz and Karsten Nohl at 26c3
  */
  *  by Henryk Ploetz and Karsten Nohl at 26c3
  */

-int CmdLegicDecode(const char *Cmd) {
-       // Index for the bytearray.
-       int i = 0;
-       int k = 0, segmentNum;
-       int segment_len = 0;
-       int segment_flag = 0;
+int CmdLegicInfo(const char *Cmd) {
+
+       char cmdp = param_getchar(Cmd, 0);
+       if ( cmdp == 'H' || cmdp == 'h' ) return usage_legic_info();
+
+       int i = 0, k = 0, segmentNum = 0, segment_len = 0, segment_flag = 0;
+       int crc = 0, wrp = 0, wrc = 0;

        uint8_t stamp_len = 0;
        uint8_t stamp_len = 0;

-       int crc = 0;
-       int wrp = 0;
-       int wrc = 0;
-       uint8_t data_buf[1200]; // receiver buffer,  should be 1024..
-       char token_type[4];
-
-       // copy data from proxmark into buffer
-       GetFromBigBuf(data_buf, sizeof(data_buf), 0);
+       uint8_t data[1024]; // receiver buffer
+       char token_type[5] = {0,0,0,0,0};
+       int dcf = 0;
+       int bIsSegmented = 0;
+
+       CmdLegicRdmem("0 22 55");
+       
+       // copy data from device
+       GetEMLFromBigBuf(data, sizeof(data), 0);

        if ( !WaitForResponseTimeout(CMD_ACK, NULL, 2000)){
                PrintAndLog("Command execute timeout");
                return 1;
        }
        
        // Output CDF System area (9 bytes) plus remaining header area (12 bytes)
        if ( !WaitForResponseTimeout(CMD_ACK, NULL, 2000)){
                PrintAndLog("Command execute timeout");
                return 1;
        }
        
        // Output CDF System area (9 bytes) plus remaining header area (12 bytes)

-       crc = data_buf[4];
-       uint32_t calc_crc =  CRC8Legic(data_buf, 4);    
+       crc = data[4];
+       uint32_t calc_crc =  CRC8Legic(data, 4);        

        
        PrintAndLog("\nCDF: System Area");
        PrintAndLog("------------------------------------------------------");
        PrintAndLog("MCD: %02x, MSN: %02x %02x %02x, MCC: %02x %s",
        
        PrintAndLog("\nCDF: System Area");
        PrintAndLog("------------------------------------------------------");
        PrintAndLog("MCD: %02x, MSN: %02x %02x %02x, MCC: %02x %s",

-               data_buf[0],
-               data_buf[1],
-               data_buf[2],
-               data_buf[3],
-               data_buf[4],
-               (calc_crc == crc) ? "OK":"Fail" 
+               data[0],
+               data[1],
+               data[2],
+               data[3],
+               data[4],
+               (calc_crc == crc) ? "OK":"Fail"

        );
  
        );
  

-       switch (data_buf[5] & 0x7f) {
-               case 0x00 ... 0x2f:
-                       strncpy(token_type, "IAM",sizeof(token_type));
-                       break;
-               case 0x30 ... 0x6f:
-                       strncpy(token_type, "SAM",sizeof(token_type));
-                       break;
-               case 0x70 ... 0x7f:
-                       strncpy(token_type, "GAM",sizeof(token_type));
-                       break;
-               default:
-                       strncpy(token_type, "???",sizeof(token_type));
-                       break;
-       }
+       // MCD = Manufacturer ID (should be list meaning something?)

 
 

-       stamp_len = 0xfc - data_buf[6];
+       token_type[0] = 0;
+       dcf = ((int)data[6] << 8) | (int)data[5];

 
 

-       PrintAndLog("DCF: %02x %02x, Token Type=%s (OLE=%01u), Stamp len=%02u",
-               data_buf[5],
-               data_buf[6],
-               token_type,
-               (data_buf[5]&0x80)>>7,
-               stamp_len
-       );
+       // New unwritten media?
+       if(dcf == 0xFFFF) {

 
 

-       PrintAndLog("WRP=%02u, WRC=%01u, RD=%01u, raw=%02x, SSC=%02x",
-               data_buf[7]&0x0f,
-               (data_buf[7]&0x70)>>4,
-               (data_buf[7]&0x80)>>7,
-               data_buf[7],
-               data_buf[8]
-       );
+               PrintAndLog("DCF: %d (%02x %02x), Token Type=NM (New Media)",
+                       dcf,
+                       data[5],
+                       data[6]
+               );
+       
+       } else if (dcf > 60000) {               // Master token?
+
+               int fl = 0;
+
+               if(data[6] == 0xec) {
+                       strncpy(token_type, "XAM", sizeof(token_type));
+                       fl = 1;
+                       stamp_len = 0x0c - (data[5] >> 4);
+               } else {
+                       switch (data[5] & 0x7f) {
+                       case 0x00 ... 0x2f:
+                               strncpy(token_type, "IAM", sizeof(token_type));
+                               fl = (0x2f - (data[5] & 0x7f)) + 1;
+                               break;
+                       case 0x30 ... 0x6f:
+                               strncpy(token_type, "SAM", sizeof(token_type));
+                               fl = (0x6f - (data[5] & 0x7f)) + 1;
+                               break;
+                       case 0x70 ... 0x7f:
+                               strncpy(token_type, "GAM", sizeof(token_type));
+                               fl = (0x7f - (data[5] & 0x7f)) + 1;
+                               break;
+                       }
+
+                       stamp_len = 0xfc - data[6];
+               }
+
+               PrintAndLog("DCF: %d (%02x %02x), Token Type=%s (OLE=%01u), OL=%02u, FL=%02u",
+                       dcf,
+                       data[5],
+                       data[6],
+                       token_type,
+                       (data[5] & 0x80 )>> 7,
+                       stamp_len,
+                       fl
+               );
+
+       } else {        // Is IM(-S) type of card...

 
 

-       PrintAndLog("Remaining Header Area");
-       PrintAndLog("%s", sprint_hex(data_buf+9, 13));
+               if(data[7] == 0x9F && data[8] == 0xFF) {
+                       bIsSegmented = 1;
+                       strncpy(token_type, "IM-S", sizeof(token_type));
+               } else {
+                       strncpy(token_type, "IM", sizeof(token_type));
+               }
+
+               PrintAndLog("DCF: %d (%02x %02x), Token Type=%s (OLE=%01u)",
+                       dcf,
+                       data[5],
+                       data[6],
+                       token_type,
+                       (data[5]&0x80) >> 7
+               );
+       }
+
+       // Makes no sence to show this on blank media...
+       if(dcf != 0xFFFF) {
+
+               if(bIsSegmented) {
+                       PrintAndLog("WRP=%02u, WRC=%01u, RD=%01u, SSC=%02x",
+                               data[7] & 0x0f,
+                               (data[7] & 0x70) >> 4,
+                               (data[7] & 0x80) >> 7,
+                               data[8]
+                       );
+               }
+
+               // Header area is only available on IM-S cards, on master tokens this data is the master token data itself
+               if(bIsSegmented || dcf > 60000) {
+                       if(dcf > 60000) {
+                               PrintAndLog("Master token data");
+                               PrintAndLog("%s", sprint_hex(data+8, 14));
+                       } else {
+                               PrintAndLog("Remaining Header Area");
+                               PrintAndLog("%s", sprint_hex(data+9, 13));
+                       }
+               }
+       }

        
        

-       uint8_t segCrcBytes[8] = {0x00};
+       uint8_t segCrcBytes[8] = {0,0,0,0,0,0,0,0};

        uint32_t segCalcCRC = 0;
        uint32_t segCRC = 0;
 
        uint32_t segCalcCRC = 0;
        uint32_t segCRC = 0;
 

-       // see if user area is xored or just zeros.
-       int numOfZeros = 0;
-       for (int index=22; index < 256; ++index){
-               if ( data_buf[index] == 0x00 )
-                       ++numOfZeros;
-       }
-       // if possible zeros is less then 60%, lets assume data is xored
-       // 256  - 22 (header) = 234
-       // 1024 - 22 (header) = 1002
-       int isXored = (numOfZeros*100/stamp_len) < 50;
-       PrintAndLog("is data xored?  %d  ( %d %)", isXored, (numOfZeros*100/stamp_len));
-
-       print_hex_break( data_buf, 33, 16);
+       // Data card?
+       if(dcf <= 60000) {

        
        

-       PrintAndLog("\nADF: User Area");
-       PrintAndLog("------------------------------------------------------");
-       i = 22;  
-       // 64 potential segements
-       // how to detect there is no segments?!?
-       for ( segmentNum=0; segmentNum<64; segmentNum++ ) {
-               segment_len = ((data_buf[i+1]^crc)&0x0f) * 256 + (data_buf[i]^crc);
-               segment_flag = ((data_buf[i+1]^crc)&0xf0)>>4;
-
-               wrp = (data_buf[i+2]^crc);
-               wrc = ((data_buf[i+3]^crc)&0x70)>>4;
-
-               bool hasWRC = (wrc > 0);
-               bool hasWRP = (wrp > wrc);
-               int wrp_len = (wrp - wrc);
-               int remain_seg_payload_len = (segment_len - wrp - 5);
+               PrintAndLog("\nADF: User Area");
+               PrintAndLog("------------------------------------------------------");
+
+               if(bIsSegmented) {
+
+                       // Data start point on segmented cards
+                       i = 22;  
+
+                       // decode segments
+                       for (segmentNum=1; segmentNum < 128; segmentNum++ )
+                       {
+                               segment_len = ((data[i+1] ^ crc) & 0x0f) * 256 + (data[i] ^ crc);
+                               segment_flag = ((data[i+1] ^ crc) & 0xf0) >> 4;
+                               wrp = (data[i+2] ^ crc);
+                               wrc = ((data[i+3] ^ crc) & 0x70) >> 4;
+
+                               bool hasWRC = (wrc > 0);
+                               bool hasWRP = (wrp > wrc);
+                               int wrp_len = (wrp - wrc);
+                               int remain_seg_payload_len = (segment_len - wrp - 5);

                
                

-               // validate segment-crc
-               segCrcBytes[0]=data_buf[0];                     //uid0
-               segCrcBytes[1]=data_buf[1];                     //uid1
-               segCrcBytes[2]=data_buf[2];                     //uid2
-               segCrcBytes[3]=data_buf[3];                     //uid3
-               segCrcBytes[4]=(data_buf[i]^crc);       //hdr0
-               segCrcBytes[5]=(data_buf[i+1]^crc); //hdr1
-               segCrcBytes[6]=(data_buf[i+2]^crc); //hdr2
-               segCrcBytes[7]=(data_buf[i+3]^crc); //hdr3
-
-               segCalcCRC = CRC8Legic(segCrcBytes, 8);
-               segCRC = data_buf[i+4]^crc;
-
-               PrintAndLog("Segment %02u \nraw header | 0x%02X 0x%02X 0x%02X 0x%02X \nSegment len: %u,  Flag: 0x%X (valid:%01u, last:%01u), WRP: %02u, WRC: %02u, RD: %01u, CRC: 0x%02X (%s)",
-                       segmentNum,
-                       data_buf[i]^crc,
-                       data_buf[i+1]^crc,
-                       data_buf[i+2]^crc,
-                       data_buf[i+3]^crc,
-                       segment_len, 
-                       segment_flag,
-                       (segment_flag & 0x4) >> 2,
-                       (segment_flag & 0x8) >> 3,
-                       wrp,
-                       wrc,
-                       ((data_buf[i+3]^crc) & 0x80) >> 7,
-                       segCRC,
-                       ( segCRC == segCalcCRC ) ? "OK" : "fail"
-               );
+                               // validate segment-crc
+                               segCrcBytes[0]=data[0];                 //uid0
+                               segCrcBytes[1]=data[1];                 //uid1
+                               segCrcBytes[2]=data[2];                 //uid2
+                               segCrcBytes[3]=data[3];                 //uid3
+                               segCrcBytes[4]=(data[i] ^ crc);   //hdr0
+                               segCrcBytes[5]=(data[i+1] ^ crc); //hdr1
+                               segCrcBytes[6]=(data[i+2] ^ crc); //hdr2
+                               segCrcBytes[7]=(data[i+3] ^ crc); //hdr3
+
+                               segCalcCRC = CRC8Legic(segCrcBytes, 8);
+                               segCRC = data[i+4] ^ crc;

 
 

-               i += 5;
+                               PrintAndLog("Segment %02u \nraw header | 0x%02X 0x%02X 0x%02X 0x%02X \nSegment len: %u,  Flag: 0x%X (valid:%01u, last:%01u), WRP: %02u, WRC: %02u, RD: %01u, CRC: 0x%02X (%s)",
+                                       segmentNum,
+                                       data[i] ^ crc,
+                                       data[i+1] ^ crc,
+                                       data[i+2] ^ crc,
+                                       data[i+3] ^ crc,
+                                       segment_len, 
+                                       segment_flag,
+                                       (segment_flag & 0x4) >> 2,
+                                       (segment_flag & 0x8) >> 3,
+                                       wrp,
+                                       wrc,
+                                       ((data[i+3]^crc) & 0x80) >> 7,
+                                       segCRC,
+                                       ( segCRC == segCalcCRC ) ? "OK" : "fail"
+                               );
+
+                               i += 5;

     
     

-               if ( hasWRC ) {
-                       PrintAndLog("WRC protected area:   (I %d | K %d| WRC %d)", i, k, wrc);
-                       PrintAndLog("\nrow  | data");
-                       PrintAndLog("-----+------------------------------------------------");
-                       // de-xor?  if not zero, assume it needs xoring.
-                       if ( isXored) {
-                               for ( k=i; k < wrc; ++k)
-                                       data_buf[k] ^= crc;
-                       }
-                       print_hex_break( data_buf+i, wrc, 16);
+                               if ( hasWRC ) {
+                                       PrintAndLog("WRC protected area:   (I %d | K %d| WRC %d)", i, k, wrc);
+                                       PrintAndLog("\nrow  | data");
+                                       PrintAndLog("-----+------------------------------------------------");
+
+                                       for ( k=i; k < (i + wrc); ++k)
+                                               data[k] ^= crc;
+
+                                       print_hex_break( data+i, wrc, 16);

                        
                        

-                       i += wrc;
-               }
+                                       i += wrc;
+                               }

     
     

-               if ( hasWRP ) {
-                       PrintAndLog("Remaining write protected area:  (I %d | K %d | WRC %d | WRP %d  WRP_LEN %d)",i, k, wrc, wrp, wrp_len);
-                       PrintAndLog("\nrow  | data");
-                       PrintAndLog("-----+------------------------------------------------");
+                               if ( hasWRP ) {
+                                       PrintAndLog("Remaining write protected area:  (I %d | K %d | WRC %d | WRP %d  WRP_LEN %d)",i, k, wrc, wrp, wrp_len);
+                                       PrintAndLog("\nrow  | data");
+                                       PrintAndLog("-----+------------------------------------------------");

 
 

-                       if (isXored) {
-                               for (k=i; k < wrp_len; ++k)
-                                       data_buf[k] ^= crc;
-                       }
+                                       for (k=i; k < (i+wrp_len); ++k)
+                                               data[k] ^= crc;

                        
                        

-                       print_hex_break( data_buf+i, wrp_len, 16);
+                                       print_hex_break( data+i, wrp_len, 16);

                        
                        

-                       i += wrp_len;
+                                       i += wrp_len;

                        
                        

-                       // does this one work?
-                       if( wrp_len == 8 )
-                               PrintAndLog("Card ID: %2X%02X%02X", data_buf[i-4]^crc, data_buf[i-3]^crc, data_buf[i-2]^crc);                   
-               }
+                                       // does this one work? (Answer: Only if KGH/BGH is used with BCD encoded card number! So maybe this will show just garbage...)
+                                       if( wrp_len == 8 )
+                                               PrintAndLog("Card ID: %2X%02X%02X", data[i-4]^crc, data[i-3]^crc, data[i-2]^crc);                       
+                               }

     
     

-               PrintAndLog("Remaining segment payload:  (I %d | K %d | Remain LEN %d)", i, k, remain_seg_payload_len);
-               PrintAndLog("\nrow  | data");
-               PrintAndLog("-----+------------------------------------------------");
-               if ( isXored ) {
-                       for ( k=i; k < remain_seg_payload_len; ++k)
-                               data_buf[k] ^= crc;
-               }
+                               PrintAndLog("Remaining segment payload:  (I %d | K %d | Remain LEN %d)", i, k, remain_seg_payload_len);
+                               PrintAndLog("\nrow  | data");
+                               PrintAndLog("-----+------------------------------------------------");
+
+                               for ( k=i; k < (i+remain_seg_payload_len); ++k)
+                                       data[k] ^= crc;

                
                

-               print_hex_break( data_buf+i, remain_seg_payload_len, 16);
+                               print_hex_break( data+i, remain_seg_payload_len, 16);

     
     

-               i += remain_seg_payload_len;
+                               i += remain_seg_payload_len;
+               
+                               PrintAndLog("-----+------------------------------------------------\n");
+
+                               // end with last segment
+                               if (segment_flag & 0x8) return 0;
+
+                       } // end for loop

                
                

-               PrintAndLog("-----+------------------------------------------------\n");
+               } else {

 
 

-               // end with last segment
-               if (segment_flag & 0x8) return 0;
+                       // Data start point on unsegmented cards
+                       i = 8;

 
 

-       } // end for loop
+                       wrp = data[7] & 0x0F;
+                       wrc = (data[7] & 0x70) >> 4;
+
+                       bool hasWRC = (wrc > 0);
+                       bool hasWRP = (wrp > wrc);
+                       int wrp_len = (wrp - wrc);
+                       int remain_seg_payload_len = (1024 - 22 - wrp); // Any chance to get physical card size here!?
+
+                       PrintAndLog("Unsegmented card - WRP: %02u, WRC: %02u, RD: %01u",
+                               wrp,
+                               wrc,
+                               (data[7] & 0x80) >> 7
+                       );
+
+                       if ( hasWRC ) {
+                               PrintAndLog("WRC protected area:   (I %d | WRC %d)", i, wrc);
+                               PrintAndLog("\nrow  | data");
+                               PrintAndLog("-----+------------------------------------------------");
+                               print_hex_break( data+i, wrc, 16);
+                               i += wrc;
+                       }
+    
+                       if ( hasWRP ) {
+                               PrintAndLog("Remaining write protected area:  (I %d | WRC %d | WRP %d | WRP_LEN %d)", i, wrc, wrp, wrp_len);
+                               PrintAndLog("\nrow  | data");
+                               PrintAndLog("-----+------------------------------------------------");
+                               print_hex_break( data + i, wrp_len, 16);
+                               i += wrp_len;
+                       
+                               // does this one work? (Answer: Only if KGH/BGH is used with BCD encoded card number! So maybe this will show just garbage...)
+                               if( wrp_len == 8 )
+                                       PrintAndLog("Card ID: %2X%02X%02X", data[i-4], data[i-3], data[i-2]);
+                       }
+    
+                       PrintAndLog("Remaining segment payload:  (I %d | Remain LEN %d)", i, remain_seg_payload_len);
+                       PrintAndLog("\nrow  | data");
+                       PrintAndLog("-----+------------------------------------------------");
+                       print_hex_break( data + i, remain_seg_payload_len, 16);
+                       i += remain_seg_payload_len;
+               
+                       PrintAndLog("-----+------------------------------------------------\n");
+               }
+       }

        return 0;
 }
 
        return 0;
 }
 

-int CmdLegicRFRead(const char *Cmd) {
-       int byte_count=0, offset=0;
-       sscanf(Cmd, "%i %i", &offset, &byte_count);
-       if(byte_count == 0) byte_count = -1;
-       if(byte_count + offset > 1024) byte_count = 1024 - offset;
+// params:
+// offset in data memory
+// number of bytes to read
+int CmdLegicRdmem(const char *Cmd) {

 
 

-       UsbCommand c= {CMD_READER_LEGIC_RF, {offset, byte_count, 0}};
+       char cmdp = param_getchar(Cmd, 0);
+       if ( cmdp == 'H' || cmdp == 'h' ) return usage_legic_rdmem();
+       
+       uint32_t offset = 0, len = 0, IV = 1;
+       sscanf(Cmd, "%x %x %x", &offset, &len, &IV);
+
+       // tagtype
+       legic_card_select_t card;
+       if (legic_get_type(&card)) {
+               PrintAndLog("Failed to identify tagtype");
+               return 1;
+       }
+
+       legic_print_type(card.cardsize, 0);
+
+       // OUT-OF-BOUNDS check
+       // UID 4 bytes can't be written to.
+       if ( len + offset >= card.cardsize ) {
+               len = card.cardsize - offset;
+               PrintAndLog("Out-of-bounds, Cardsize = %d, Trunc offset+len = %d", card.cardsize, len + offset);
+       }
+       
+       legic_chk_iv(&IV);
+       
+       UsbCommand c = {CMD_READER_LEGIC_RF, {offset, len, IV}};
+       clearCommandBuffer();
+       SendCommand(&c);
+       UsbCommand resp;
+       if ( !WaitForResponseTimeout(CMD_ACK, &resp, 3000) ) {
+               PrintAndLog("command execution time out");
+               return 1;
+       }
+
+       uint8_t isOK = resp.arg[0] & 0xFF;
+       uint16_t readlen = resp.arg[1];
+       if ( !isOK ) {
+               PrintAndLog("failed reading tag");
+               return 2;
+       }
+       
+       uint8_t *data = malloc(readlen);
+       if ( !data ){
+               PrintAndLog("Cannot allocate memory");
+               return 2;
+       }
+                       
+       if ( readlen != len )
+               PrintAndLog("Fail, only managed to read 0x%02X bytes", readlen);
+                       
+       // copy data from device
+       GetEMLFromBigBuf(data, readlen, 0);
+       if ( !WaitForResponseTimeout(CMD_ACK, NULL, 2500)){
+               PrintAndLog("Command execute timeout");
+               free(data);
+               return 1;
+       }
+       
+       PrintAndLog("\n ##  |  0  1  2  3  4  5  6  7  8  9  A  B  C  D  E  F 10 11 12 13 14 15 16 17 18 19 1A 1B 1C 1D 1E 1F");
+       PrintAndLog("-----+------------------------------------------------------------------------------------------------");
+       print_hex_break( data, readlen, 32);
+       free(data);
+       return 0;
+}
+
+// should say which tagtype
+// should load a tag to device mem.
+// int phase, int frame, int reqresp
+int CmdLegicRfSim(const char *Cmd) {
+       UsbCommand c = {CMD_SIMULATE_TAG_LEGIC_RF, {6,3,0}};
+       sscanf(Cmd, " %"lli" %"lli" %"lli, &c.arg[0], &c.arg[1], &c.arg[2]);

        clearCommandBuffer();
        SendCommand(&c);
        return 0;
 }
 
        clearCommandBuffer();
        SendCommand(&c);
        return 0;
 }
 

-int CmdLegicLoad(const char *Cmd) {
+int CmdLegicRfWrite(const char *Cmd) {
+
+       uint8_t *data = NULL;
+       uint8_t cmdp = 0;
+       bool errors = false;
+       int len = 0, bg, en;
+       uint32_t offset = 0, IV = 0x55;

        
        

-       char cmdp = param_getchar(Cmd, 0);
-       if ( cmdp == 'H' || cmdp == 'h' || cmdp == 0x00) return usage_legic_load();
+       while(param_getchar(Cmd, cmdp) != 0x00) {
+               switch(param_getchar(Cmd, cmdp)) {
+               case 'd':
+               case 'D':
+                       // peek at length of the input string so we can
+                       // figure out how many elements to malloc in "data"
+                       bg=en=0;
+                       if (param_getptr(Cmd, &bg, &en, cmdp+1)) {
+                               errors = true;
+                               break;
+                       }
+                       len = (en - bg + 1);

 
 

-       char filename[FILE_PATH_SIZE] = {0x00};
-       int len = strlen(Cmd);
+                       // check that user entered even number of characters
+                       // for hex data string
+                       if (len & 1) {
+                               errors = true;
+                               break;
+                       }
+                       
+                       // limit number of bytes to write. This is not a 'restore' command.
+                       if ( (len>>1) > 100 ){
+                               PrintAndLog("Max bound on 100bytes to write a one time.");
+                               PrintAndLog("Use the 'hf legic restore' command if you want to write the whole tag at once");
+                               errors = true;
+                       }
+
+                       // it's possible for user to accidentally enter "b" parameter
+                       // more than once - we have to clean previous malloc
+                       if (data)
+                               free(data);
+                       
+                       data = malloc(len >> 1);
+                       if ( data == NULL ) {
+                               PrintAndLog("Can't allocate memory. exiting");
+                               errors = true;
+                               break;
+                       }
+                       
+                       if (param_gethex(Cmd, cmdp+1, data, len)) {
+                               errors = true;
+                               break;
+                       }
+
+                       len >>= 1;      
+                       cmdp += 2;
+                       break;
+               case 'o':
+               case 'O':
+                       offset = param_get32ex(Cmd, cmdp+1, 4, 16);
+                       cmdp += 2;
+                       break;
+               case 'h':
+               case 'H':
+                       errors = true;
+                       break;
+               default:
+                       PrintAndLog("Unknown parameter '%c'", param_getchar(Cmd, cmdp));
+                       errors = true;
+                       break;
+               }
+               if (errors) break;
+       }
+       //Validations
+       if (errors){
+               if (data) 
+                       free(data);
+               return usage_legic_write();
+       }
+
+       // tagtype
+       legic_card_select_t card;
+       if (legic_get_type(&card)) {
+               PrintAndLog("Failed to identify tagtype");
+               return -1;
+       }
+
+       legic_print_type(card.cardsize, 0);

        
        

-       if (len > FILE_PATH_SIZE) {
-               PrintAndLog("Filepath too long (was %s bytes), max allowed is %s ", len, FILE_PATH_SIZE);
+       // OUT-OF-BOUNDS checks
+       // UID 4+1 bytes can't be written to.
+       if ( offset < 5 ) {
+               PrintAndLog("Out-of-bounds, bytes 0-1-2-3-4 can't be written to. Offset = %d", offset);
+               return -2;
+       }
+       
+       if ( len + offset >= card.cardsize ) {
+               PrintAndLog("Out-of-bounds, Cardsize = %d, [offset+len = %d ]", card.cardsize, len + offset);
+               return -2;
+       }
+
+       if (offset == 5 || offset == 6) {
+               PrintAndLog("############# DANGER ################");
+               PrintAndLog("# changing the DCF is irreversible  #");
+               PrintAndLog("#####################################");
+               PrintAndLog("do you really want to continue? y(es) n(o)");              
+               char answer;
+               sscanf("%c", &answer);
+               bool exit = !(answer == 'n' || answer == 'N');
+               if (exit)
+                       return 0;
+               printf("ICE DCF:  %c answer, %d\n", answer, exit);

                return 0;
        }
                return 0;
        }

-       memcpy(filename, Cmd, len);
-
-    FILE *f = fopen(filename, "r");
-    if(!f) {
-        PrintAndLog("couldn't open '%s'", Cmd);
-        return -1;
-    }
-       
-    char line[80]; 
-       int offset = 0; 
-       uint8_t data[USB_CMD_DATA_SIZE] = {0x00};
-       int index = 0;
-       int totalbytes = 0;
-    while ( fgets(line, sizeof(line), f) ) {
-        int res = sscanf(line, "%x %x %x %x %x %x %x %x", 
-            (unsigned int *)&data[index],
-                       (unsigned int *)&data[index + 1],
-                       (unsigned int *)&data[index + 2],
-                       (unsigned int *)&data[index + 3],
-            (unsigned int *)&data[index + 4],
-                       (unsigned int *)&data[index + 5],
-                       (unsigned int *)&data[index + 6],
-                       (unsigned int *)&data[index + 7]);
-                       
-        if(res != 8) {
-          PrintAndLog("Error: could not read samples");
-          fclose(f);
-          return -1;
-        }
-               index += res;
+       
+       legic_chk_iv(&IV);
+       
+       PrintAndLog("Writing to tag");
+
+       UsbCommand c = {CMD_WRITER_LEGIC_RF, {offset, len, IV}};
+       memcpy(c.d.asBytes, data, len); 
+       UsbCommand resp;
+       clearCommandBuffer();
+       SendCommand(&c);
+       
+       if (!WaitForResponseTimeout(CMD_ACK, &resp, 2000)) {
+               PrintAndLog("command execution time out");
+               return 1;
+       }
+       uint8_t isOK = resp.arg[0] & 0xFF;
+       if ( !isOK ) {
+               PrintAndLog("failed writing tag");
+               return 1;
+       }
+
+    return 0;
+}
+
+int CmdLegicCalcCrc(const char *Cmd){
+
+       uint8_t *data = NULL;
+       uint8_t cmdp = 0, uidcrc = 0, type=0;
+       bool errors = false;
+       int len = 0;
+       int bg, en;
+       
+       while(param_getchar(Cmd, cmdp) != 0x00) {
+               switch(param_getchar(Cmd, cmdp)) {
+               case 'b':
+               case 'B':
+                       // peek at length of the input string so we can
+                       // figure out how many elements to malloc in "data"
+                       bg=en=0;
+                       if (param_getptr(Cmd, &bg, &en, cmdp+1)) {
+                               errors = true;
+                               break;
+                       }
+                       len = (en - bg + 1);
+
+                       // check that user entered even number of characters
+                       // for hex data string
+                       if (len & 1) {
+                               errors = true;
+                               break;
+                       }
+
+                       // it's possible for user to accidentally enter "b" parameter
+                       // more than once - we have to clean previous malloc
+                       if (data) free(data);
+                       data = malloc(len >> 1);
+                       if ( data == NULL ) {
+                               PrintAndLog("Can't allocate memory. exiting");
+                               errors = true;
+                               break;
+                       }

                        
                        

-               if ( index == USB_CMD_DATA_SIZE ){
-//                     PrintAndLog("sent %d | %d | %d", index, offset, totalbytes);
-                       UsbCommand c = { CMD_DOWNLOADED_SIM_SAMPLES_125K, {offset, 0, 0}};
-                       memcpy(c.d.asBytes, data, sizeof(data));
-                       clearCommandBuffer();
-                       SendCommand(&c);
-                       if ( !WaitForResponseTimeout(CMD_ACK, NULL, 1500)){
-                               PrintAndLog("Command execute timeout");
-                               fclose(f);
-                               return 1;
+                       if (param_gethex(Cmd, cmdp+1, data, len)) {
+                               errors = true;
+                               break;

                        }
                        }

-                       offset += index;
-                       totalbytes += index;
-                       index = 0;
+
+                       len >>= 1;      
+                       cmdp += 2;
+                       break;
+               case 'u':
+               case 'U':                
+                       uidcrc = param_get8ex(Cmd, cmdp+1, 0, 16);
+                       cmdp += 2;
+                       break;
+               case 'c':
+               case 'C':
+                       type = param_get8ex(Cmd, cmdp+1, 0, 10);
+                       cmdp += 2;
+                       break;
+               case 'h':
+               case 'H':
+                       errors = true;
+                       break;
+               default:
+                       PrintAndLog("Unknown parameter '%c'", param_getchar(Cmd, cmdp));
+                       errors = true;
+                       break;

                }
                }

-    }
-    fclose(f);
+               if (errors) break;
+       }
+       //Validations
+       if (errors){
+               if (data) free(data);
+               return usage_legic_calccrc();
+       }
+       
+       switch (type){
+               case 16:
+                       PrintAndLog("Legic crc16: %X", CRC16Legic(data, len, uidcrc));
+                       break;
+               default:
+                       PrintAndLog("Legic crc8: %X",  CRC8Legic(data, len) );
+                       break;
+       }

        
        

-       // left over bytes?
-       if ( index != 0 ) {
-               UsbCommand c = { CMD_DOWNLOADED_SIM_SAMPLES_125K, {offset, 0, 0}};
-               memcpy(c.d.asBytes, data, 8);
+       if (data) free(data);
+       return 0;
+} 
+
+int legic_print_type(uint32_t tagtype, uint8_t spaces){
+       char spc[11] = "          ";
+       spc[10]=0x00;
+       char *spacer = spc + (10-spaces);
+
+       if ( tagtype == 22 )    
+               PrintAndLog("%sTYPE : MIM%d card (outdated)", spacer, tagtype);
+       else if ( tagtype == 256 )
+               PrintAndLog("%sTYPE : MIM%d card (234 bytes)", spacer, tagtype);
+       else if ( tagtype == 1024 )
+               PrintAndLog("%sTYPE : MIM%d card (1002 bytes)", spacer, tagtype);
+       else
+               PrintAndLog("%sTYPE : Unknown %06x", spacer, tagtype);
+       return 0;
+}
+int legic_get_type(legic_card_select_t *card){
+
+       if ( card == NULL ) return 1;
+
+       UsbCommand c = {CMD_LEGIC_INFO, {0,0,0}};
+       clearCommandBuffer();
+    SendCommand(&c);
+       UsbCommand resp;
+       if (!WaitForResponseTimeout(CMD_ACK, &resp, 500))
+               return 2;
+       
+       uint8_t isOK = resp.arg[0] & 0xFF;
+       if ( !isOK ) 
+               return 3;
+       
+       memcpy(card, (legic_card_select_t *)resp.d.asBytes, sizeof(legic_card_select_t));
+       return 0;
+}
+void legic_chk_iv(uint32_t *iv){
+       if ( (*iv & 0x7F) != *iv ){
+               *iv &= 0x7F;
+               PrintAndLog("Truncating IV to 7bits, %u", *iv);
+       }
+       // IV must be odd
+       if ( (*iv & 1) == 0 ){
+               *iv |= 0x01;  
+               PrintAndLog("LSB of IV must be SET %u", *iv);   
+       }
+}
+void legic_seteml(uint8_t *src, uint32_t offset, uint32_t numofbytes) {
+       size_t len = 0;
+       UsbCommand c = {CMD_LEGIC_ESET, {0, 0, 0}};     
+       for(size_t i = 0; i < numofbytes; i += USB_CMD_DATA_SIZE) {
+               
+               len = MIN((numofbytes - i), USB_CMD_DATA_SIZE);         
+               c.arg[0] = i; // offset
+               c.arg[1] = len; // number of bytes
+               memcpy(c.d.asBytes, src+i, len); 

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

-               if ( !WaitForResponseTimeout(CMD_ACK, NULL, 1500)){
-                               PrintAndLog("Command execute timeout");
-                               return 1;
-               }
-               totalbytes += index;            

        }
        }

-       
-    PrintAndLog("loaded %u samples", totalbytes);
-    return 0;

 }
 
 }
 

-int CmdLegicSave(const char *Cmd) {
-       int requested = 1024;
-       int offset = 0;
-       int delivered = 0;
-       char filename[FILE_PATH_SIZE];
-       uint8_t got[1024] = {0x00};
+int HFLegicReader(const char *Cmd, bool verbose) {

 
 

-       sscanf(Cmd, " %s %i %i", filename, &requested, &offset);
+       char cmdp = param_getchar(Cmd, 0);
+       if ( cmdp == 'H' || cmdp == 'h' ) return usage_legic_reader();
+       
+       legic_card_select_t card;
+       switch(legic_get_type(&card)){
+               case 1: 
+                       if ( verbose ) PrintAndLog("command execution time out"); 
+                       return 1;
+               case 2: 
+               case 3: 
+                       if ( verbose ) PrintAndLog("legic card select failed");
+                       return 2;
+               default: break;
+       }
+       PrintAndLog(" UID : %s", sprint_hex(card.uid, sizeof(card.uid)));
+       legic_print_type(card.cardsize, 0);
+       return 0;
+}
+int CmdLegicReader(const char *Cmd){
+       return HFLegicReader(Cmd, TRUE);
+}

 
 

-       /* If no length given save entire legic read buffer */
-       /* round up to nearest 8 bytes so the saved data can be used with legicload */
-       if (requested == 0)
-               requested = 1024;
+int CmdLegicDump(const char *Cmd){

 
 

-       if (requested % 8 != 0) {
-               int remainder = requested % 8;
-               requested = requested + 8 - remainder;
+       FILE *f;
+       char filename[FILE_PATH_SIZE] = {0x00};
+       char *fnameptr = filename;
+       size_t fileNlen = 0;
+       bool errors = false;
+       uint16_t dumplen;       
+       uint8_t cmdp = 0;
+       
+       memset(filename, 0, sizeof(filename));
+       
+       while(param_getchar(Cmd, cmdp) != 0x00) {
+               switch(param_getchar(Cmd, cmdp)) {
+                       case 'h':
+                       case 'H':
+                               return usage_legic_dump();
+                       case 'o':
+                       case 'O':
+                               fileNlen = param_getstr(Cmd, cmdp+1, filename);
+                               if (!fileNlen) 
+                                       errors = true; 
+                               if (fileNlen > FILE_PATH_SIZE-5) 
+                                       fileNlen = FILE_PATH_SIZE-5;
+                               cmdp += 2;
+                               break;
+                       default:
+                               PrintAndLog("Unknown parameter '%c'", param_getchar(Cmd, cmdp));
+                               errors = true;
+                               break;
+               }
+               if(errors) break;

        }
 
        }
 

-       if (offset + requested > sizeof(got)) {
-               PrintAndLog("Tried to read past end of buffer, <bytes> + <offset> > 1024");
-               return 0;
+       //Validations
+       if(errors) return usage_legic_dump();
+       
+       // tagtype
+       legic_card_select_t card;
+       if (legic_get_type(&card)) {
+               PrintAndLog("Failed to identify tagtype");
+               return -1;

        }
        }

+       dumplen = card.cardsize;
+       
+       legic_print_type(dumplen, 0);   
+       PrintAndLog("Reading tag memory...");

 
 

-       GetFromBigBuf(got, requested, offset);
-       if ( !WaitForResponseTimeout(CMD_ACK, NULL, 2000)){
-               PrintAndLog("Command execute timeout"); 
+       UsbCommand c = {CMD_READER_LEGIC_RF, {0x00, dumplen, 0x55}};
+       clearCommandBuffer();
+       SendCommand(&c);
+       UsbCommand resp;
+       if (!WaitForResponseTimeout(CMD_ACK, &resp, 3000)) {
+               PrintAndLog("Command execute time-out");

                return 1;
        }
                return 1;
        }

+               
+       uint8_t isOK = resp.arg[0] & 0xFF;
+       if ( !isOK ) {
+               PrintAndLog("Failed dumping tag data");
+               return 2;
+       }

 
 

-       FILE *f = fopen(filename, "w");
-       if(!f) {
-               PrintAndLog("couldn't open '%s'", Cmd+1);
-               return -1;
+       uint16_t readlen = resp.arg[1];
+       uint8_t *data = malloc(readlen);
+       if (!data) {
+               PrintAndLog("Fail, cannot allocate memory");
+               return 3;

        }
        }

+       memset(data, 0, readlen);

        
        

-       for (int j = 0; j < requested; j += 8) {
-               fprintf(f, "%02x %02x %02x %02x %02x %02x %02x %02x\n",
-                       got[j+0], got[j+1], got[j+2], got[j+3],
-                       got[j+4], got[j+5],     got[j+6], got[j+7]
-               );
-               delivered += 8;
-               if (delivered >= requested) break;
+       if ( readlen != dumplen )
+               PrintAndLog("Fail, only managed to read 0x%02X bytes of 0x%02X", readlen, dumplen);
+
+       // copy data from device
+       GetEMLFromBigBuf(data, readlen, 0);
+       if ( !WaitForResponseTimeout(CMD_ACK, NULL, 2500)) {
+               PrintAndLog("Fail, transfer from device time-out");
+               free(data);
+               return 4;
+       }
+
+       // user supplied filename?
+       if (fileNlen < 1)
+               sprintf(fnameptr,"%02X%02X%02X%02X.bin", data[0], data[1], data[2], data[3]);
+       else
+               sprintf(fnameptr + fileNlen,".bin");
+
+       if ((f = fopen(filename,"wb")) == NULL) { 
+               PrintAndLog("Could not create file name %s", filename);
+               if (data)
+                       free(data);
+               return 5;
+       }
+       fwrite(data, 1, readlen, f);
+       fclose(f);
+       free(data);
+       PrintAndLog("Wrote %d bytes to %s", readlen, filename);
+       return 0;
+}      
+
+int CmdLegicRestore(const char *Cmd){
+
+       FILE *f;
+       char filename[FILE_PATH_SIZE] = {0x00};
+       char *fnameptr = filename;
+       size_t fileNlen = 0;
+       bool errors = true;
+       uint16_t numofbytes;    
+       uint8_t cmdp = 0;
+       
+       memset(filename, 0, sizeof(filename));
+       
+       while(param_getchar(Cmd, cmdp) != 0x00) {
+               switch(param_getchar(Cmd, cmdp)) {
+                       case 'h':
+                       case 'H':
+                               errors = true;
+                               break;
+                       case 'i':
+                       case 'I':
+                               fileNlen = param_getstr(Cmd, cmdp+1, filename);
+                               if (!fileNlen) 
+                                       errors = true;
+                               else 
+                                       errors = false;
+                               
+                               if (fileNlen > FILE_PATH_SIZE-5) 
+                                       fileNlen = FILE_PATH_SIZE-5;
+                               cmdp += 2;                              
+                               break;
+                       default:
+                               PrintAndLog("Unknown parameter '%c'", param_getchar(Cmd, cmdp));
+                               errors = true;
+                               break;
+               }
+               if (errors) break;

        }
 
        }
 

+       //Validations
+       if(errors) return usage_legic_restore();
+       
+       // tagtype
+       legic_card_select_t card;
+       if (legic_get_type(&card)) {
+               PrintAndLog("Failed to identify tagtype");
+               return 1;
+       }
+       numofbytes = card.cardsize;     
+       
+       // set up buffer
+       uint8_t *data = malloc(numofbytes);
+       if (!data) {
+               PrintAndLog("Fail, cannot allocate memory");
+               return 2;               
+       }
+       memset(data, 0, numofbytes);
+       
+       legic_print_type(numofbytes, 0);        
+
+       // set up file
+       fnameptr += fileNlen;
+       sprintf(fnameptr, ".bin");
+
+       if ((f = fopen(filename,"rb")) == NULL) {
+               PrintAndLog("File %s not found or locked", filename);
+               return 3;
+       }       
+       
+       // verify size of dumpfile is the same as card. 
+       fseek(f, 0, SEEK_END); // seek to end of file
+       size_t filesize = ftell(f); // get current file pointer
+       fseek(f, 0, SEEK_SET); // seek back to beginning of file
+       
+       if ( filesize != numofbytes) {
+               PrintAndLog("Fail, filesize and cardsize is not equal. [%u != %u]", filesize, numofbytes);
+               free(data);
+               fclose(f);
+               return 4;
+       }
+
+       // load file
+       size_t bytes_read = fread(data, 1, numofbytes, f);
+       if ( bytes_read == 0){
+               PrintAndLog("File reading error");
+               free(data);
+               fclose(f);
+               return 2;
+       }

        fclose(f);
        fclose(f);

-       PrintAndLog("saved %u samples", delivered);
+
+       PrintAndLog("Restoring %s to card", filename);
+
+       // transfer to device
+       size_t len = 0;
+       UsbCommand c = {CMD_WRITER_LEGIC_RF, {0, 0, 0x55}};
+       UsbCommand resp;
+       for(size_t i = 7; i < numofbytes; i += USB_CMD_DATA_SIZE) {
+               
+               len = MIN((numofbytes - i), USB_CMD_DATA_SIZE);         
+               c.arg[0] = i; // offset
+               c.arg[1] = len; // number of bytes
+               memcpy(c.d.asBytes, data+i, len); 
+               PrintAndLog("offset %d | chunk %d | numofbytes %d", i, len, numofbytes);
+               clearCommandBuffer();
+               SendCommand(&c);
+       
+               if (!WaitForResponseTimeout(CMD_ACK, &resp, 2000)) {
+                       PrintAndLog("command execution time out");
+                       free(data);     
+                       return 1;
+               }
+               uint8_t isOK = resp.arg[0] & 0xFF;
+               if ( !isOK ) {
+                       PrintAndLog("failed writing tag [msg = %u]", resp.arg[1] & 0xFF);
+                       free(data);     
+                       return 1;
+               }
+               PrintAndLog("Wrote chunk %d - %d", i, len);
+       }       
+       
+       free(data);     
+       PrintAndLog("\nWrote %d bytes from file: %s to card", numofbytes, filename);

        return 0;
 }
 
        return 0;
 }
 

-//TODO: write a help text (iceman)
-int CmdLegicRfSim(const char *Cmd) {
-       UsbCommand c = {CMD_SIMULATE_TAG_LEGIC_RF, {6,3,0}};
-       sscanf(Cmd, " %"lli" %"lli" %"lli, &c.arg[0], &c.arg[1], &c.arg[2]);
-       clearCommandBuffer();
-       SendCommand(&c);
+int CmdLegicELoad(const char *Cmd) {
+       FILE * f;
+       char filename[FILE_PATH_SIZE];
+       char *fnameptr = filename;
+       int len, numofbytes;
+       int nameParamNo = 1;
+       
+       char cmdp = param_getchar(Cmd, 0);              
+       if ( cmdp == 'h' || cmdp == 'H' || cmdp == 0x00)
+               return usage_legic_eload();
+
+       switch (cmdp) {
+               case '0' : numofbytes = 22; break;
+               case '1' : 
+               case '\0': numofbytes = 256; break;
+               case '2' : numofbytes = 1024; break;
+               default  : numofbytes = 256;  nameParamNo = 0;break;
+       }
+
+       // set up buffer
+       uint8_t *data = malloc(numofbytes);
+       if (!data) {
+               PrintAndLog("Fail, cannot allocate memory");
+               return 3;               
+       }
+       memset(data, 0, numofbytes);
+       
+       // set up file
+       len = param_getstr(Cmd, nameParamNo, filename);
+       if (len > FILE_PATH_SIZE - 5) 
+               len = FILE_PATH_SIZE - 5;
+       fnameptr += len;
+       sprintf(fnameptr, ".bin");
+       
+       // open file
+       if ((f = fopen(filename,"rb")) == NULL) { 
+               PrintAndLog("File %s not found or locked", filename);
+               free(data);
+               return 1;
+       }
+
+       // load file
+       size_t bytes_read = fread(data, 1, numofbytes, f);
+       if ( bytes_read == 0){
+               PrintAndLog("File reading error");
+               free(data);
+               fclose(f);
+               return 2;
+       }
+       fclose(f);
+       
+       // transfer to device
+       legic_seteml(data, 0, numofbytes);
+               
+       free(data);     
+       PrintAndLog("\nLoaded %d bytes from file: %s  to emulator memory", numofbytes, filename);

        return 0;
 }
 
        return 0;
 }
 

-//TODO: write a help text (iceman)
-int CmdLegicRfWrite(const char *Cmd) {
-    UsbCommand c = {CMD_WRITER_LEGIC_RF};
-    int res = sscanf(Cmd, " 0x%"llx" 0x%"llx, &c.arg[0], &c.arg[1]);
-       if(res != 2) {
-               PrintAndLog("Please specify the offset and length as two hex strings");
-        return -1;
-    }
-       clearCommandBuffer();
-    SendCommand(&c);
-    return 0;
-}
+int CmdLegicESave(const char *Cmd) {
+       FILE *f;
+       char filename[FILE_PATH_SIZE];
+       char *fnameptr = filename;
+       int fileNlen, numofbytes, nameParamNo = 1;
+       
+       memset(filename, 0, sizeof(filename));

 
 

-int CmdLegicRfFill(const char *Cmd) {
-    UsbCommand cmd = {CMD_WRITER_LEGIC_RF};
-    int res = sscanf(Cmd, " 0x%"llx" 0x%"llx" 0x%"llx, &cmd.arg[0], &cmd.arg[1], &cmd.arg[2]);
-    if(res != 3) {
-        PrintAndLog("Please specify the offset, length and value as two hex strings");
-        return -1;
-    }
-
-    int i;
-    UsbCommand c = {CMD_DOWNLOADED_SIM_SAMPLES_125K, {0, 0, 0}};
-    for(i = 0; i < 48; i++) {
-               c.d.asBytes[i] = cmd.arg[2];
-    }
-       
-       for(i = 0; i < 22; i++) {
-               c.arg[0] = i*48;
-               SendCommand(&c);
-               WaitForResponse(CMD_ACK,NULL);
-       }
-       clearCommandBuffer();
-    SendCommand(&cmd);
-    return 0;
- }
+       char cmdp = param_getchar(Cmd, 0);
+       
+       if ( cmdp == 'h' || cmdp == 'H' || cmdp == 0x00)
+               return usage_legic_esave();

 
 

-int CmdLegicCalcCrc8(const char *Cmd){
+       switch (cmdp) {
+               case '0' : numofbytes = 22; break;
+               case '1' : 
+               case '\0': numofbytes = 256; break;
+               case '2' : numofbytes = 1024; break;
+               default  : numofbytes = 256; nameParamNo = 0; break;
+       }

 
 

-       int len =  strlen(Cmd); 
-       if (len & 1 ) return usage_legic_calccrc8(); 
+       fileNlen = param_getstr(Cmd, nameParamNo, filename);

        
        

-       // add 1 for null terminator.
-       uint8_t *data = malloc(len+1);
-       if ( data == NULL ) return 1;
+       if (fileNlen > FILE_PATH_SIZE - 5) 
+               fileNlen = FILE_PATH_SIZE - 5;
+
+       // set up buffer
+       uint8_t *data = malloc(numofbytes);
+       if (!data) {
+               PrintAndLog("Fail, cannot allocate memory");
+               return 3;               
+       }
+       memset(data, 0, numofbytes);

                
                

-       if (!param_gethex(Cmd, 0, data, len )) return usage_legic_calccrc8(); 
+       // download emulator memory
+       PrintAndLog("Reading emulator memory...");      
+       GetEMLFromBigBuf(data, numofbytes, 0);
+       if ( !WaitForResponseTimeout(CMD_ACK, NULL, 2500)) {
+               PrintAndLog("Fail, transfer from device time-out");
+               free(data);
+               return 4;
+       }
+
+       // user supplied filename?
+       if (fileNlen < 1)               
+               sprintf(fnameptr,"%02X%02X%02X%02X.bin", data[0], data[1], data[2], data[3]);
+       else
+               sprintf(fnameptr + fileNlen,".bin");

        
        

-       uint32_t checksum =  CRC8Legic(data, len/2);    
-       PrintAndLog("Bytes: %s || CRC8: %X", sprint_hex(data, len/2), checksum );
+       // open file
+       if ((f = fopen(filename,"wb")) == NULL) { 
+               PrintAndLog("Could not create file name %s", filename);
+               free(data);
+               return 1;
+       }
+       fwrite(data, 1, numofbytes, f);
+       fclose(f);

        free(data);
        free(data);

+       PrintAndLog("\nSaved %d bytes from emulator memory to file: %s", numofbytes, filename);

        return 0;
        return 0;

-} 
- 
+}
+
+int CmdLegicList(const char *Cmd) {
+       CmdHFList("legic");
+       return 0;
+}
+

 static command_t CommandTable[] =  {
 static command_t CommandTable[] =  {

-       {"help",        CmdHelp,        1, "This help"},
-       {"decode",      CmdLegicDecode, 0, "Display deobfuscated and decoded LEGIC RF tag data (use after hf legic reader)"},
-       {"read",        CmdLegicRFRead, 0, "[offset][length] -- read bytes from a LEGIC card"},
-       {"save",        CmdLegicSave,   0, "<filename> [<length>] -- Store samples"},
-       {"load",        CmdLegicLoad,   0, "<filename> -- Restore samples"},
-       {"sim",         CmdLegicRfSim,  0, "[phase drift [frame drift [req/resp drift]]] Start tag simulator (use after load or read)"},
-       {"write",       CmdLegicRfWrite,0, "<offset> <length> -- Write sample buffer (user after load or read)"},
-       {"fill",        CmdLegicRfFill, 0, "<offset> <length> <value> -- Fill/Write tag with constant value"},
-       {"crc8",        CmdLegicCalcCrc8, 1, "Calculate Legic CRC8 over given hexbytes"},
+       {"help",        CmdHelp,                        1, "This help"},
+       {"reader",      CmdLegicReader,         1, "LEGIC Prime Reader UID and tag info"},
+       {"info",        CmdLegicInfo,           0, "Display deobfuscated and decoded LEGIC Prime tag data"},
+       {"dump",        CmdLegicDump,           0, "Dump LEGIC Prime tag to binary file"},
+       {"restore", CmdLegicRestore,    0, "Restore a dump onto a LEGIC Prime tag"},
+       {"rdmem",       CmdLegicRdmem,          0, "Read bytes from a LEGIC Prime tag"},
+       {"sim",         CmdLegicRfSim,          0, "Start tag simulator"},
+       {"write",       CmdLegicRfWrite,        0, "Write data to a LEGIC Prime tag"},
+       {"crc",         CmdLegicCalcCrc,        1, "Calculate Legic CRC over given bytes"},     
+       {"eload",       CmdLegicELoad,          1, "Load binary dump to emulator memory"},
+       {"esave",       CmdLegicESave,          1, "Save emulator memory to binary file"},
+       {"list",        CmdLegicList,           1, "[Deprecated] List LEGIC history"},

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