main.rs

#![no_std]
#![no_main]
// Needed for embassy macros
#![feature(type_alias_impl_trait)]

mod globals;
use crate::globals::println;

use embassy_executor::{task, Executor};
use embassy_futures::yield_now;
use riscv_rt::entry;
use static_cell::StaticCell;

#[entry]
fn main() -> ! {
    // self::globals::setup();

    let p = esp32c3_hal::pac::Peripherals::take().unwrap();

    // Disable the RTC and TIMG watchdog timers
    // Technically, the error still occurs even without this, but I've included it
    // for completeness.
    {
        use esp32c3_hal::{clock::ClockControl, prelude::*, timer::TimerGroup, Rtc};

        let system = p.SYSTEM.split();
        let clocks = ClockControl::boot_defaults(system.clock_control).freeze();

        let mut rtc = Rtc::new(p.RTC_CNTL);
        let timer_group0 = TimerGroup::new(p.TIMG0, &clocks);
        let mut wdt0 = timer_group0.wdt;
        let timer_group1 = TimerGroup::new(p.TIMG1, &clocks);
        let mut wdt1 = timer_group1.wdt;

        rtc.rwdt.disable();
        rtc.swd.disable();
        wdt0.disable();
        wdt1.disable();
    }

    println!("Booted");

    static EXECUTOR: StaticCell<Executor> = StaticCell::new();
    EXECUTOR.init(Executor::new()).run(move |spawner| {
        spawner.spawn(network_task()).unwrap();
        spawner.spawn(sensor_task()).unwrap();
    });
}

#[task]
async fn network_task() {
    println!("Started network_task");
    let mut i = 0;
    loop {
        println!("In main(), i was {}", i);
        i += 1;
        yield_now().await; // Yield to ensure fairness
        println!("After network yield");
    }
}

#[task]
async fn sensor_task() {
    println!("Started sensor_task");
    let mut i = 0;
    loop {
        println!("In data(), i was {}", i);
        i += 1;
        yield_now().await; // Yield to ensure fairness
        println!("after sensor yield");
    }
}

#[export_name = "ExceptionHandler"]
fn custom_exception_handler(trap_frame: &riscv_rt::TrapFrame) -> ! {
    let mcause = riscv::register::mcause::read();
    println!(
        "mcause bits: {:#b}, mcause cause: {:?}",
        mcause.bits(),
        mcause.cause(),
    );
    println!("return address register (ra): {:#x}", trap_frame.ra);
    println!("trap_frame: {:?}", trap_frame);

    loop {}
}