optimization UDOL creation. does not affect on functionality.

[proxmark3-svn] / armsrc / mifareutil.c

diff --git a/armsrc/mifareutil.c b/armsrc/mifareutil.c
index fdead5eecef4def266d00ab0d3be890467dcf99a..38ca934aa01687b542744965981e6eae886a6a1a 100644 (file)
--- a/armsrc/mifareutil.c
+++ b/armsrc/mifareutil.c
@@ -9,13 +9,15 @@
 // Work with mifare cards.\r
 //-----------------------------------------------------------------------------\r
 \r
 // Work with mifare cards.\r
 //-----------------------------------------------------------------------------\r
 \r

-#include <string.h>\r

 #include "mifareutil.h"\r
 #include "mifareutil.h"\r

+\r
+#include <string.h>\r
+#include <stdbool.h>\r
+\r

 #include "proxmark3.h"\r
 #include "apps.h"\r
 #include "util.h"\r
 #include "parity.h"\r
 #include "proxmark3.h"\r
 #include "apps.h"\r
 #include "util.h"\r
 #include "parity.h"\r

-\r

 #include "iso14443crc.h"\r
 #include "iso14443a.h"\r
 #include "crapto1/crapto1.h"\r
 #include "iso14443crc.h"\r
 #include "iso14443a.h"\r
 #include "crapto1/crapto1.h"\r


@@ -24,23 +26,27 @@
 int MF_DBGLEVEL = MF_DBG_ALL;\r
 \r
 // crypto1 helpers\r
 int MF_DBGLEVEL = MF_DBG_ALL;\r
 \r
 // crypto1 helpers\r

-void mf_crypto1_decrypt(struct Crypto1State *pcs, uint8_t *data, int len){\r
+void mf_crypto1_decryptEx(struct Crypto1State *pcs, uint8_t *data_in, int len, uint8_t *data_out){\r

        uint8_t bt = 0;\r
        int i;\r
        \r
        if (len != 1) {\r
                for (i = 0; i < len; i++)\r
        uint8_t bt = 0;\r
        int i;\r
        \r
        if (len != 1) {\r
                for (i = 0; i < len; i++)\r

-                       data[i] = crypto1_byte(pcs, 0x00, 0) ^ data[i];\r
+                       data_out[i] = crypto1_byte(pcs, 0x00, 0) ^ data_in[i];\r

        } else {\r
                bt = 0;\r
                for (i = 0; i < 4; i++)\r
        } else {\r
                bt = 0;\r
                for (i = 0; i < 4; i++)\r

-                       bt |= (crypto1_bit(pcs, 0, 0) ^ BIT(data[0], i)) << i;\r
+                       bt |= (crypto1_bit(pcs, 0, 0) ^ BIT(data_in[0], i)) << i;\r

                                \r
                                \r

-               data[0] = bt;\r
+               data_out[0] = bt;\r

        }\r
        return;\r
 }\r
 \r
        }\r
        return;\r
 }\r
 \r

+void mf_crypto1_decrypt(struct Crypto1State *pcs, uint8_t *data, int len){\r
+       mf_crypto1_decryptEx(pcs, data, len, data);\r
+}\r
+\r

 void mf_crypto1_encrypt(struct Crypto1State *pcs, uint8_t *data, uint16_t len, uint8_t *par) {\r
        uint8_t bt = 0;\r
        int i;\r
 void mf_crypto1_encrypt(struct Crypto1State *pcs, uint8_t *data, uint16_t len, uint8_t *par) {\r
        uint8_t bt = 0;\r
        int i;\r


@@ -581,6 +587,19 @@ uint8_t FirstBlockOfSector(uint8_t sectorNo)
                \r
 }\r
 \r
                \r
 }\r
 \r

+uint8_t SectorTrailer(uint8_t blockNo)\r
+{\r
+       if (blockNo < 32*4) {\r
+               return (blockNo | 0x03);\r
+       } else {\r
+               return (blockNo | 0x0f);\r
+       }\r
+}\r
+\r
+bool IsSectorTrailer(uint8_t blockNo)\r
+{\r
+       return (blockNo == SectorTrailer(blockNo));\r
+}\r

 \r
 // work with emulator memory\r
 void emlSetMem(uint8_t *data, int blockNum, int blocksCount) {\r
 \r
 // work with emulator memory\r
 void emlSetMem(uint8_t *data, int blockNum, int blocksCount) {\r


@@ -764,3 +783,125 @@ int mifare_desfire_des_auth2(uint32_t uid, uint8_t *key, uint8_t *blockData){
        }\r
        return 1;\r
 }\r
        }\r
        return 1;\r
 }\r

+\r
+//-----------------------------------------------------------------------------\r
+// MIFARE check keys\r
+//\r
+//-----------------------------------------------------------------------------\r
+// one key check\r
+int MifareChkBlockKey(uint8_t *uid, uint32_t *cuid, uint8_t *cascade_levels, uint64_t ui64Key, uint8_t blockNo, uint8_t keyType, uint8_t debugLevel) {\r
+\r
+       struct Crypto1State mpcs = {0, 0};\r
+       struct Crypto1State *pcs;\r
+       pcs = &mpcs;\r
+\r
+       // Iceman: use piwi's faster nonce collecting part in hardnested.\r
+       if (*cascade_levels == 0) { // need a full select cycle to get the uid first\r
+               iso14a_card_select_t card_info;\r
+               if(!iso14443a_select_card(uid, &card_info, cuid, true, 0, true)) {\r
+                       if (debugLevel >= 1)    Dbprintf("ChkKeys: Can't select card");\r
+                       return  1;\r
+               }\r
+               switch (card_info.uidlen) {\r
+                       case 4 : *cascade_levels = 1; break;\r
+                       case 7 : *cascade_levels = 2; break;\r
+                       case 10: *cascade_levels = 3; break;\r
+                       default: break;\r
+               }\r
+       } else { // no need for anticollision. We can directly select the card\r
+               if(!iso14443a_select_card(uid, NULL, NULL, false, *cascade_levels, true)) {\r
+                       if (debugLevel >= 1)    Dbprintf("ChkKeys: Can't select card (UID) lvl=%d", *cascade_levels);\r
+                       return  1;\r
+               }\r
+       }\r
+       \r
+       if(mifare_classic_auth(pcs, *cuid, blockNo, keyType, ui64Key, AUTH_FIRST)) {\r
+//             SpinDelayUs(AUTHENTICATION_TIMEOUT); // it not needs because mifare_classic_auth have timeout from iso14a_set_timeout()\r
+               return 2;\r
+       } else {\r
+/*             // let it be here. it like halt command, but maybe it will work in some strange cases\r
+               uint8_t dummy_answer = 0;\r
+               ReaderTransmit(&dummy_answer, 1, NULL);\r
+               int timeout = GetCountSspClk() + AUTHENTICATION_TIMEOUT;                        \r
+               // wait for the card to become ready again\r
+               while(GetCountSspClk() < timeout) {};\r
+*/\r
+               // it needs after success authentication\r
+               mifare_classic_halt(pcs, *cuid);\r
+       }\r
+       \r
+       return 0;\r
+}\r
+\r
+// multi key check\r
+int MifareChkBlockKeys(uint8_t *keys, uint8_t keyCount, uint8_t blockNo, uint8_t keyType, uint8_t debugLevel) {\r
+       uint8_t uid[10];\r
+       uint32_t cuid = 0;\r
+       uint8_t cascade_levels = 0;\r
+       uint64_t ui64Key = 0;\r
+\r
+       int retryCount = 0;\r
+       for (uint8_t i = 0; i < keyCount; i++) {\r
+\r
+               // Allow button press / usb cmd to interrupt device\r
+               if (BUTTON_PRESS() && !usb_poll_validate_length()) { \r
+                       Dbprintf("ChkKeys: Cancel operation. Exit...");\r
+                       return -2;\r
+               }\r
+\r
+               ui64Key = bytes_to_num(keys + i * 6, 6);\r
+               int res = MifareChkBlockKey(uid, &cuid, &cascade_levels, ui64Key, blockNo, keyType, debugLevel);\r
+               \r
+               // can't select\r
+               if (res == 1) {\r
+                       retryCount++;\r
+                       if (retryCount >= 5) {\r
+                               Dbprintf("ChkKeys: block=%d key=%d. Can't select. Exit...", blockNo, keyType);\r
+                               return -1;\r
+                       }\r
+                       --i; // try the same key once again\r
+\r
+                       SpinDelay(20);\r
+//                     Dbprintf("ChkKeys: block=%d key=%d. Try the same key once again...", blockNo, keyType);\r
+                       continue;\r
+               }\r
+               \r
+               // can't authenticate\r
+               if (res == 2) {\r
+                       retryCount = 0;\r
+                       continue; // can't auth. wrong key.\r
+               }\r
+\r
+               return i + 1;\r
+       }\r
+       \r
+       return 0;\r
+}\r
+\r
+// multisector multikey check\r
+int MifareMultisectorChk(uint8_t *keys, uint8_t keyCount, uint8_t SectorCount, uint8_t keyType, uint8_t debugLevel, TKeyIndex *keyIndex) {\r
+       int res = 0;\r
+       \r
+//     int clk = GetCountSspClk();\r
+\r
+       for(int sc = 0; sc < SectorCount; sc++){\r
+               WDT_HIT();\r
+\r
+               int keyAB = keyType;\r
+               do {\r
+                       res = MifareChkBlockKeys(keys, keyCount, FirstBlockOfSector(sc), keyAB & 0x01, debugLevel);\r
+                       if (res < 0){\r
+                               return res;\r
+                       }\r
+                       if (res > 0){\r
+                               (*keyIndex)[keyAB & 0x01][sc] = res;\r
+                       }\r
+               } while(--keyAB > 0);\r
+       }\r
+       \r
+//     Dbprintf("%d %d", GetCountSspClk() - clk, (GetCountSspClk() - clk)/(SectorCount*keyCount*(keyType==2?2:1)));\r
+       \r
+       return 0;\r
+}\r
+\r
+\r