//-----------------------------------------------------------------------------
// A. Ozkal, 2020
//
// This code is licensed to you under the terms of the GNU GPL, version 2 or,
// at your option, any later version. See the LICENSE.txt file for the text of
// the license.
//-----------------------------------------------------------------------------
// main code for HF Mifare Ultralight read/simulation by Ave Ozkal
//-----------------------------------------------------------------------------

/* Several parts of this code is based on code by Craig Young from HF_YOUNG */

#include "standalone.h" // standalone definitions
#include "proxmark3_arm.h"
#include "appmain.h"
#include "fpgaloader.h"
#include "util.h"
#include "dbprint.h"

#include "ticks.h"  // SpinDelay
#include "mifareutil.h"
#include "iso14443a.h"

typedef struct {
    uint8_t uid[10];
    uint8_t uidlen;
    uint8_t atqa[2];
    uint8_t sak;
} PACKED card_clone_t;

void ModInfo(void) {
    DbpString("  HF Mifare Ultralight read/simulation by Ave Ozkal");
}

void RunMod(void) {
    StandAloneMode();
    Dbprintf("[=] AveUL (MF Ultralight read/emul) started");
    FpgaDownloadAndGo(FPGA_BITSTREAM_HF);

    // the main loop for your standalone mode
    for (;;) {
        WDT_HIT();

        // exit from RunMod,   send a usbcommand.
        if (data_available()) break;

        iso14a_card_select_t card;

        SpinDelay(500);
        iso14443a_setup(FPGA_HF_ISO14443A_READER_MOD);

        // 0 = search, 1 = read, 2 = emul
        int stage = 0;

        DbpString("Scanning...");
        for (;;) {
            // Was our button held down or pressed?
            int button_pressed = BUTTON_HELD(1000);

            if (button_pressed != BUTTON_NO_CLICK || data_available())
                break;
            else if (stage == 0) {
                if (!iso14443a_select_card(NULL, &card, NULL, true, 0, true)) {
                    continue;
                } else {
                    if (card.sak == 0x00 && card.atqa[0] == 0x44 && card.atqa[1] == 0 && card.uidlen == 7) {
                        DbpString("Found ultralight with UID: ");
                        Dbhexdump(7, card.uid, 0);
                        stage = 1;
                    }
                    else {
                        DbpString("Found non-ultralight card, ignoring");
                    }
                }
            }
            else if (stage == 1) {
                iso14443a_setup(FPGA_HF_ISO14443A_READER_LISTEN);
                iso14443a_select_card(NULL, NULL, NULL, true, 0, true);
                int i;
                bool read_successful = true;
                Dbprintf("Contents:");

                for (i = 0; i < 16; ++i)
                {
                    uint8_t dataout[4] = {0x00};
                    if (mifare_ultra_readblock(i, dataout)) {
                        // If there's an error reading, go back to stage 0
                        read_successful = false;
                        break;
                    }
                    // TODO: I'm not 100% on why I need to do + 14. The bin->eml results have 14 blocks of almost all 0 at start.
                    // and the bins just don't, so I'll admit that I am a little confused, but it works, so I won't question it much.
                    emlSetMem_xt(dataout, 14 + i, 1, 4);
                    Dbhexdump(4, dataout, 0);
                }

                if (read_successful) {
                    Dbprintf("Successfully loaded into emulator memory");
                    stage = 2;
                } else {
                    Dbprintf("Read failure, going back to stage 0.");
                    stage = 0;
                }
            }
            else if (stage == 2) {
                uint8_t flags = FLAG_7B_UID_IN_DATA;

                Dbprintf("Starting simulation, press pm3-button to stop and go back to scan");
                SimulateIso14443aTag(2, flags, card.uid);

                // Go back to stage 0 if user presses pm3-button
                stage = 0;
            }
        }
    }

    DbpString("[=] exiting");
    LEDsoff();
}