trezor-firmware/core/embed/rust/src/ui/model_tr/bootloader/mod.rs

use crate::{
    trezorhal::io::io_button_read,
    ui::{
        component::{Component, Never},
        display::{self, Font},
        model_tr::constant,
    },
};
use heapless::String;

mod confirm;
mod confirm_seed;
mod intro;
mod menu;
mod theme;
mod title;

use crate::ui::{
    component::{
        text::paragraphs::{Paragraph, ParagraphVecShort, Paragraphs, VecExt},
        Event, EventCtx,
    },
    constant::{screen, BACKLIGHT_NORMAL},
    display::{fade_backlight_duration, Color},
    event::ButtonEvent,
    geometry::{LinearPlacement, Offset, Rect},
    model_tr::{
        bootloader::{
            confirm::Confirm,
            confirm_seed::ConfirmSeed,
            intro::Intro,
            menu::Menu,
            theme::{BLD_BG, BLD_FG, ICON_ALERT, ICON_SPINNER, LOGO_EMPTY},
        },
        component::ResultScreen,
        theme::ICON_SUCCESS,
    },
    util::{from_c_array, from_c_str},
};

const SCREEN_ADJ: Rect = screen().split_top(64).0;

pub trait ReturnToC {
    fn return_to_c(self) -> u32;
}

impl ReturnToC for Never {
    fn return_to_c(self) -> u32 {
        unreachable!()
    }
}

impl ReturnToC for () {
    fn return_to_c(self) -> u32 {
        0
    }
}

fn button_eval() -> Option<ButtonEvent> {
    let event = io_button_read();
    if event == 0 {
        return None;
    }

    let event_type = event >> 24;
    let event_btn = event & 0xFFFFFF;

    let event = ButtonEvent::new(event_type, event_btn);

    if let Ok(event) = event {
        return Some(event);
    }
    None
}

fn run<F>(frame: &mut F) -> u32
where
    F: Component,
    F::Msg: ReturnToC,
{
    frame.place(SCREEN_ADJ);
    frame.paint();
    display::refresh();
    fade_backlight_duration(BACKLIGHT_NORMAL as _, 500);

    while button_eval().is_some() {}

    loop {
        let event = button_eval();
        if let Some(e) = event {
            let mut ctx = EventCtx::new();
            let msg = frame.event(&mut ctx, Event::Button(e));

            if let Some(message) = msg {
                return message.return_to_c();
            }

            frame.paint();
            display::refresh();
        }
    }
}

#[no_mangle]
extern "C" fn screen_install_confirm(
    vendor_str: *const cty::c_char,
    vendor_str_len: u8,
    version: *const cty::c_char,
    downgrade: bool,
    vendor: bool,
) -> u32 {
    let text = unwrap!(unsafe { from_c_array(vendor_str, vendor_str_len as usize) });
    let version = unwrap!(unsafe { from_c_str(version) });

    let title = if downgrade {
        "DOWNGRADE FIRMWARE"
    } else if vendor {
        "FW VENDOR CHANGE"
    } else {
        "UPDATE FIRMWARE"
    };

    let mut msg: String<30> = String::new();
    msg.push_str("Firmware version ").unwrap();
    msg.push_str(version).unwrap();

    let mut message = ParagraphVecShort::new();

    message.add(Paragraph::new(&theme::TEXT_NORMAL, msg.as_ref()));
    message.add(Paragraph::new(&theme::TEXT_NORMAL, "by"));
    message.add(Paragraph::new(&theme::TEXT_NORMAL, text));

    if vendor || downgrade {
        let mut m = ParagraphVecShort::new();
        m.add(Paragraph::new(&theme::TEXT_NORMAL, "Seed will be erased!"));
        let message2 = Paragraphs::new(m)
            .with_placement(LinearPlacement::vertical().align_at_center())
            .with_spacing(0);

        let mut frame = ConfirmSeed::new(
            BLD_BG,
            title,
            Paragraphs::new(message)
                .with_placement(LinearPlacement::vertical().align_at_center())
                .with_spacing(0),
            message2,
            "INSTALL",
            false,
        );
        run(&mut frame)
    } else {
        let mut frame = Confirm::new(
            BLD_BG,
            None,
            title,
            Paragraphs::new(message)
                .with_placement(LinearPlacement::vertical().align_at_center())
                .with_spacing(0),
            "INSTALL",
            false,
        );
        run(&mut frame)
    }
}

#[no_mangle]
extern "C" fn screen_wipe_confirm() -> u32 {
    let mut messages = ParagraphVecShort::new();

    // messages.add(Paragraph::new(
    //     &theme::TEXT_NORMAL,
    //     "Do you really want to wipe the device?",
    // ));
    messages.add(Paragraph::new(&theme::TEXT_NORMAL, "Seed will be erased!"));

    let message =
        Paragraphs::new(messages).with_placement(LinearPlacement::vertical().align_at_center());

    let mut frame = Confirm::new(BLD_BG, None, "WIPE TREZOR?", message, "WIPE", true);

    run(&mut frame)
}

#[no_mangle]
extern "C" fn screen_menu(_bld_version: *const cty::c_char) -> u32 {
    run(&mut Menu::new())
}

#[no_mangle]
extern "C" fn screen_intro(
    bld_version: *const cty::c_char,
    vendor_str: *const cty::c_char,
    vendor_str_len: u8,
    version: *const cty::c_char,
) -> u32 {
    let vendor = unwrap!(unsafe { from_c_array(vendor_str, vendor_str_len as usize) });
    let version = unwrap!(unsafe { from_c_str(version) });
    let bld_version = unwrap!(unsafe { from_c_str(bld_version) });

    run(&mut Intro::new(bld_version, vendor, version))
}

fn screen_progress(
    text: &str,
    text2: &str,
    progress: u16,
    initialize: bool,
    fg_color: Color,
    bg_color: Color,
    icon: Option<(&[u8], Color)>,
) -> u32 {
    if initialize {
        display::rect_fill(constant::screen(), bg_color);
    }

    let progress = if progress < 20 { 20 } else { progress };

    display::rect_rounded2_partial(
        Rect::new(
            screen().top_center() + Offset::new(-9, 3),
            screen().top_center() + Offset::new(9, 18 + 3),
        ),
        fg_color,
        bg_color,
        ((100_u32 * progress as u32) / 1000) as _,
        icon,
    );
    display::text_center(
        screen().center() + Offset::y(8),
        text,
        Font::BOLD,
        fg_color,
        bg_color,
    );
    display::text_center(
        screen().center() + Offset::y(20),
        text2,
        Font::BOLD,
        fg_color,
        bg_color,
    );

    // let loader_area = Rect::new(Point::new(5, HEIGHT - 16), Point::new(WIDTH - 5,
    // HEIGHT));
    //
    // let mut text = TextOverlay::new(text, Font::NORMAL);
    // text.place(loader_area.center() + Offset::y(Font::NORMAL.text_height() / 2));
    //
    // let fill_to = (loader_area.width() as u32 * progress as u32) / 1000;
    //
    // if let Some(icon) = icon {
    //     display::icon(screen().center(), icon.0, icon.1, BLD_BG);
    // }
    //
    // display::bar_with_text_and_fill(
    //     loader_area,
    //     Some(&text),
    //     fg_color,
    //     bg_color,
    //     0,
    //     fill_to as _,
    // );

    display::refresh();
    0
}

const INITIAL_INSTALL_LOADER_COLOR: Color = Color::rgb(0x4A, 0x90, 0xE2);

#[no_mangle]
extern "C" fn screen_install_progress(
    progress: u16,
    initialize: bool,
    _initial_setup: bool,
) -> u32 {
    screen_progress(
        "Installing",
        "Firmware",
        progress,
        initialize,
        BLD_FG,
        BLD_BG,
        Some((ICON_SUCCESS.0, BLD_FG)),
    )
}

#[no_mangle]
extern "C" fn screen_wipe_progress(progress: u16, initialize: bool) -> u32 {
    screen_progress(
        "Wiping",
        "Trezor",
        progress,
        initialize,
        BLD_FG,
        BLD_BG,
        Some((ICON_SUCCESS.0, BLD_FG)),
    )
}

#[no_mangle]
extern "C" fn screen_connect() -> u32 {
    let mut messages = ParagraphVecShort::new();

    messages.add(Paragraph::new(&theme::TEXT_NORMAL, "Waiting for host...").centered());

    let mut frame =
        Paragraphs::new(messages).with_placement(LinearPlacement::vertical().align_at_center());

    frame.place(SCREEN_ADJ);
    frame.paint();
    display::refresh();
    0
}

#[no_mangle]
extern "C" fn screen_wipe_success() -> u32 {
    let mut messages = ParagraphVecShort::new();
    messages.add(Paragraph::new(&theme::TEXT_BOLD, "Device wiped").centered());

    let m_top =
        Paragraphs::new(messages).with_placement(LinearPlacement::vertical().align_at_center());

    let mut messages = ParagraphVecShort::new();
    messages.add(Paragraph::new(&theme::TEXT_NORMAL, "PLEASE RECONNECT").centered());
    messages.add(Paragraph::new(&theme::TEXT_NORMAL, "THE DEVICE").centered());

    let m_bottom =
        Paragraphs::new(messages).with_placement(LinearPlacement::vertical().align_at_center());

    let mut frame = ResultScreen::new(BLD_FG, BLD_BG, ICON_SPINNER, m_top, m_bottom, true);
    frame.place(SCREEN_ADJ);
    frame.paint();
    display::refresh();
    0
}

#[no_mangle]
extern "C" fn screen_wipe_fail() -> u32 {
    let mut messages = ParagraphVecShort::new();
    messages.add(Paragraph::new(&theme::TEXT_BOLD, "Wipe failed").centered());

    let m_top =
        Paragraphs::new(messages).with_placement(LinearPlacement::vertical().align_at_center());

    let mut messages = ParagraphVecShort::new();
    messages.add(Paragraph::new(&theme::TEXT_NORMAL, "PLEASE RECONNECT").centered());
    messages.add(Paragraph::new(&theme::TEXT_NORMAL, "THE DEVICE").centered());

    let m_bottom =
        Paragraphs::new(messages).with_placement(LinearPlacement::vertical().align_at_center());

    let mut frame = ResultScreen::new(BLD_FG, BLD_BG, ICON_ALERT, m_top, m_bottom, true);
    frame.place(SCREEN_ADJ);
    frame.paint();
    display::refresh();
    0
}

#[no_mangle]
extern "C" fn screen_boot_empty(_firmware_present: bool) {
    display::icon(SCREEN_ADJ.center(), LOGO_EMPTY, BLD_FG, BLD_BG);
}

#[no_mangle]
extern "C" fn screen_install_fail() -> u32 {
    let mut messages = ParagraphVecShort::new();
    messages.add(Paragraph::new(&theme::TEXT_BOLD, "Install failed").centered());
    let m_top =
        Paragraphs::new(messages).with_placement(LinearPlacement::vertical().align_at_center());

    let mut messages = ParagraphVecShort::new();
    messages.add(Paragraph::new(&theme::TEXT_NORMAL, "PLEASE RECONNECT").centered());
    messages.add(Paragraph::new(&theme::TEXT_NORMAL, "THE DEVICE").centered());

    let m_bottom =
        Paragraphs::new(messages).with_placement(LinearPlacement::vertical().align_at_center());

    let mut frame = ResultScreen::new(BLD_FG, BLD_BG, ICON_ALERT, m_top, m_bottom, true);
    frame.place(SCREEN_ADJ);
    frame.paint();
    display::refresh();
    0
}

fn screen_install_success_bld(msg: &'static str, complete_draw: bool) -> u32 {
    let mut messages = ParagraphVecShort::new();
    messages.add(Paragraph::new(&theme::TEXT_BOLD, "Installed").centered());

    let m_top =
        Paragraphs::new(messages).with_placement(LinearPlacement::vertical().align_at_center());

    let mut messages = ParagraphVecShort::new();
    messages.add(Paragraph::new(&theme::TEXT_NORMAL, msg).centered());

    let m_bottom =
        Paragraphs::new(messages).with_placement(LinearPlacement::vertical().align_at_center());

    let mut frame = ResultScreen::new(BLD_FG, BLD_BG, ICON_SPINNER, m_top, m_bottom, complete_draw);
    frame.place(SCREEN_ADJ);
    frame.paint();
    display::refresh();
    0
}

fn screen_install_success_initial(msg: &'static str, complete_draw: bool) -> u32 {
    let mut messages = ParagraphVecShort::new();
    messages.add(Paragraph::new(&theme::TEXT_BOLD, "Installed").centered());

    let m_top =
        Paragraphs::new(messages).with_placement(LinearPlacement::vertical().align_at_center());

    let mut messages = ParagraphVecShort::new();
    messages.add(Paragraph::new(&theme::TEXT_NORMAL, msg).centered());

    let m_bottom =
        Paragraphs::new(messages).with_placement(LinearPlacement::vertical().align_at_center());

    let mut frame = ResultScreen::new(BLD_FG, BLD_BG, ICON_SPINNER, m_top, m_bottom, complete_draw);
    frame.place(SCREEN_ADJ);
    frame.paint();
    display::refresh();
    0
}

#[no_mangle]
extern "C" fn screen_install_success(
    reboot_msg: *const cty::c_char,
    initial_setup: bool,
    complete_draw: bool,
) -> u32 {
    let msg = unwrap!(unsafe { from_c_str(reboot_msg) });
    if initial_setup {
        screen_install_success_initial(msg, complete_draw)
    } else {
        screen_install_success_bld(msg, complete_draw)
    }
}

#[no_mangle]
extern "C" fn screen_welcome() -> u32 {
    let mut messages = ParagraphVecShort::new();
    messages.add(Paragraph::new(&theme::TEXT_NORMAL, "Get started with").centered());
    messages.add(Paragraph::new(&theme::TEXT_NORMAL, "your trezor at").centered());
    messages.add(Paragraph::new(&theme::TEXT_BOLD, "trezor.io/start").centered());
    let mut frame =
        Paragraphs::new(messages).with_placement(LinearPlacement::vertical().align_at_center());

    frame.place(SCREEN_ADJ);
    frame.paint();
    display::refresh();
    0
}