palera1n/src/wx/ExploitPanel.cpp

#ifdef WITH_GUI

#include "ExploitPanel.hpp"

#include <wx/wx.h>

#include <thread>
#include <atomic>

#include "AppFrame.hpp"
#include "../m8/run.h"

static const char* stage_to_string(int stage)
{
    switch (stage) {
        case STAGE_PREPARE: return "Checking if device is ready";
        case STAGE_RESET:   return "Checking if device is ready";
        case STAGE_SETUP:   return "Setting up the exploit (this is the heap spray)";
        case STAGE_SPRAY:   return "Setting up the exploit (this is the heap spray)";
        case STAGE_PATCH:   return "Right before trigger (this is the real bug setup)";
        case STAGE_PONGO:   return "Booting...";
        case STAGE_JAILBREAK: return "Booting...";
        case STAGE_DONE:    return "Done.";
        default:            return "Working...";
    }
}

static int stage_to_progress(int stage)
{
    switch (stage) {
        case STAGE_PREPARE: return 5;
        case STAGE_RESET:   return 15;
        case STAGE_SETUP:   return 30;
        case STAGE_SPRAY:   return 50;
        case STAGE_PATCH:   return 70;
        case STAGE_PONGO:   return 90;
        case STAGE_JAILBREAK: return 95;
        case STAGE_DONE:    return 100;
        default:            return 0;
    }
}

ExploitPanel::ExploitPanel(MainFrame* frame, wxWindow* parent)
    : DevicePanel(frame, parent)
{
    auto* root = new wxBoxSizer(wxVERTICAL);

    auto* text = new wxStaticText(this, wxID_ANY,
        "Installing jailbreak, this will take a moment. If the device asks for a passcode\nplease enter it. Do not disconnect the device until finished.");

    m_statusText = new wxStaticText(this, wxID_ANY, "Idle.");
    m_progressBar = new wxGauge(this, wxID_ANY, 100);
    m_progressBar->SetValue(0);

    m_backButton = new wxButton(this, wxID_ANY, "Back");
    m_backButton->Disable();

    m_backButton->Bind(wxEVT_BUTTON, [this, frame](wxCommandEvent&) {
        m_statusText->SetLabel("Idle.");
        m_progressBar->SetValue(0);
        m_backButton->Disable();
        frame->ShowMain();
    });

    root->Add(text, 0, wxALL, 10);
    root->Add(m_statusText, 0, wxALL, 10);
    root->Add(m_progressBar, 0, wxALL | wxEXPAND, 20);

    root->AddStretchSpacer();
    root->Add(m_backButton, 0, wxALL | wxALIGN_RIGHT, 10);

    SetSizer(root);
}

void ExploitPanel::SetDeviceState(const DeviceState& state)
{
    DevicePanel::SetDeviceState(state);

    if (!IsShown())
        return;
}

void ExploitPanel::StartExploit()
{
    m_statusText->SetLabel("Starting...");
    m_backButton->Disable();

    std::thread([this]()
    {
        const auto deviceState = GetDeviceState();
        if (!deviceState.connected || deviceState.mode != DeviceMode::DFU)
            return;

        shared_t state{};
        atomic_init(&state.result, 0);
        atomic_init(&state.stop, false);
        atomic_init(&state.exploit_done, false);
        atomic_init(&state.pongo_done, false);
        atomic_init(&state.stage, STAGE_PREPARE);

        std::thread worker([&]() {
            exploit(&state);
        });

        worker.detach();

        while (true)
        {
            int stage = atomic_load(&state.stage);
            int progress = stage_to_progress(stage);

            wxTheApp->CallAfter([this, stage, progress]() {
                m_statusText->SetLabel(stage_to_string(stage));
                m_progressBar->SetValue(progress);
            });

            if (stage == STAGE_DONE) {
                wxTheApp->CallAfter([this]() {
                    m_backButton->Enable();
                });
                break;
            }

            std::this_thread::sleep_for(std::chrono::milliseconds(100));
        }
    }).detach();
}

#endif // WITH_GUI