1#![no_std]
10#![no_main]
11#![deny(missing_docs)]
12
13use core::ptr::addr_of_mut;
14
15use capsules_core::virtualizers::virtual_alarm::VirtualMuxAlarm;
16use components::gpio::GpioComponent;
17use kernel::capabilities;
18use kernel::component::Component;
19use kernel::hil::led::LedHigh;
20use kernel::platform::{KernelResources, SyscallDriverLookup};
21use kernel::process::ProcessArray;
22use kernel::scheduler::round_robin::RoundRobinSched;
23use kernel::{create_capability, debug, static_init};
24
25use stm32f429zi::chip_specs::Stm32f429Specs;
26use stm32f429zi::clocks::hsi::HSI_FREQUENCY_MHZ;
27use stm32f429zi::gpio::{AlternateFunction, Mode, PinId, PortId};
28use stm32f429zi::interrupt_service::Stm32f429ziDefaultPeripherals;
29
30pub mod io;
32
33const NUM_PROCS: usize = 4;
35
36type ChipHw = stm32f429zi::chip::Stm32f4xx<'static, Stm32f429ziDefaultPeripherals<'static>>;
37
38static mut PROCESSES: Option<&'static ProcessArray<NUM_PROCS>> = None;
40static mut CHIP: Option<&'static ChipHw> = None;
41static mut PROCESS_PRINTER: Option<&'static capsules_system::process_printer::ProcessPrinterText> =
42    None;
43
44const FAULT_RESPONSE: capsules_system::process_policies::PanicFaultPolicy =
46    capsules_system::process_policies::PanicFaultPolicy {};
47
48kernel::stack_size! {0x2000}
49
50type TemperatureSTMSensor = components::temperature_stm::TemperatureSTMComponentType<
51    capsules_core::virtualizers::virtual_adc::AdcDevice<'static, stm32f429zi::adc::Adc<'static>>,
52>;
53type TemperatureDriver = components::temperature::TemperatureComponentType<TemperatureSTMSensor>;
54
55struct STM32F429IDiscovery {
58    console: &'static capsules_core::console::Console<'static>,
59    ipc: kernel::ipc::IPC<{ NUM_PROCS as u8 }>,
60    led: &'static capsules_core::led::LedDriver<
61        'static,
62        LedHigh<'static, stm32f429zi::gpio::Pin<'static>>,
63        4,
64    >,
65    button: &'static capsules_core::button::Button<'static, stm32f429zi::gpio::Pin<'static>>,
66    adc: &'static capsules_core::adc::AdcVirtualized<'static>,
67    alarm: &'static capsules_core::alarm::AlarmDriver<
68        'static,
69        VirtualMuxAlarm<'static, stm32f429zi::tim2::Tim2<'static>>,
70    >,
71    temperature: &'static TemperatureDriver,
72    gpio: &'static capsules_core::gpio::GPIO<'static, stm32f429zi::gpio::Pin<'static>>,
73
74    scheduler: &'static RoundRobinSched<'static>,
75    systick: cortexm4::systick::SysTick,
76}
77
78impl SyscallDriverLookup for STM32F429IDiscovery {
80    fn with_driver<F, R>(&self, driver_num: usize, f: F) -> R
81    where
82        F: FnOnce(Option<&dyn kernel::syscall::SyscallDriver>) -> R,
83    {
84        match driver_num {
85            capsules_core::console::DRIVER_NUM => f(Some(self.console)),
86            capsules_core::led::DRIVER_NUM => f(Some(self.led)),
87            capsules_core::button::DRIVER_NUM => f(Some(self.button)),
88            capsules_core::adc::DRIVER_NUM => f(Some(self.adc)),
89            capsules_core::alarm::DRIVER_NUM => f(Some(self.alarm)),
90            capsules_extra::temperature::DRIVER_NUM => f(Some(self.temperature)),
91            kernel::ipc::DRIVER_NUM => f(Some(&self.ipc)),
92            capsules_core::gpio::DRIVER_NUM => f(Some(self.gpio)),
93            _ => f(None),
94        }
95    }
96}
97
98impl
99    KernelResources<
100        stm32f429zi::chip::Stm32f4xx<
101            'static,
102            stm32f429zi::interrupt_service::Stm32f429ziDefaultPeripherals<'static>,
103        >,
104    > for STM32F429IDiscovery
105{
106    type SyscallDriverLookup = Self;
107    type SyscallFilter = ();
108    type ProcessFault = ();
109    type Scheduler = RoundRobinSched<'static>;
110    type SchedulerTimer = cortexm4::systick::SysTick;
111    type WatchDog = ();
112    type ContextSwitchCallback = ();
113
114    fn syscall_driver_lookup(&self) -> &Self::SyscallDriverLookup {
115        self
116    }
117    fn syscall_filter(&self) -> &Self::SyscallFilter {
118        &()
119    }
120    fn process_fault(&self) -> &Self::ProcessFault {
121        &()
122    }
123    fn scheduler(&self) -> &Self::Scheduler {
124        self.scheduler
125    }
126    fn scheduler_timer(&self) -> &Self::SchedulerTimer {
127        &self.systick
128    }
129    fn watchdog(&self) -> &Self::WatchDog {
130        &()
131    }
132    fn context_switch_callback(&self) -> &Self::ContextSwitchCallback {
133        &()
134    }
135}
136
137unsafe fn setup_dma(
139    dma: &stm32f429zi::dma::Dma2,
140    dma_streams: &'static [stm32f429zi::dma::Stream<'static, stm32f429zi::dma::Dma2>; 8],
141    usart1: &'static stm32f429zi::usart::Usart<stm32f429zi::dma::Dma2>,
142) {
143    use stm32f429zi::dma::Dma2Peripheral;
144    use stm32f429zi::usart;
145
146    dma.enable_clock();
147
148    let usart1_tx_stream = &dma_streams[Dma2Peripheral::USART1_TX.get_stream_idx()];
149    let usart1_rx_stream = &dma_streams[Dma2Peripheral::USART1_RX.get_stream_idx()];
150
151    usart1.set_dma(
152        usart::TxDMA(usart1_tx_stream),
153        usart::RxDMA(usart1_rx_stream),
154    );
155
156    usart1_tx_stream.set_client(usart1);
157    usart1_rx_stream.set_client(usart1);
158
159    usart1_tx_stream.setup(Dma2Peripheral::USART1_TX);
160    usart1_rx_stream.setup(Dma2Peripheral::USART1_RX);
161
162    cortexm4::nvic::Nvic::new(Dma2Peripheral::USART1_TX.get_stream_irqn()).enable();
163    cortexm4::nvic::Nvic::new(Dma2Peripheral::USART1_RX.get_stream_irqn()).enable();
164}
165
166unsafe fn set_pin_primary_functions(
168    syscfg: &stm32f429zi::syscfg::Syscfg,
169    gpio_ports: &'static stm32f429zi::gpio::GpioPorts<'static>,
170) {
171    use kernel::hil::gpio::Configure;
172
173    syscfg.enable_clock();
174
175    gpio_ports.get_port_from_port_id(PortId::G).enable_clock();
176
177    gpio_ports.get_pin(PinId::PG14).map(|pin| {
179        pin.make_output();
180
181        kernel::debug::assign_gpios(Some(pin), None, None);
183    });
184
185    gpio_ports.get_port_from_port_id(PortId::A).enable_clock();
186
187    gpio_ports.get_pin(PinId::PA09).map(|pin| {
190        pin.set_mode(Mode::AlternateFunctionMode);
191        pin.set_alternate_function(AlternateFunction::AF7);
193    });
194    gpio_ports.get_pin(PinId::PA10).map(|pin| {
195        pin.set_mode(Mode::AlternateFunctionMode);
196        pin.set_alternate_function(AlternateFunction::AF7);
198    });
199
200    gpio_ports.get_pin(PinId::PA00).map(|pin| {
202        pin.enable_interrupt();
208    });
209    cortexm4::nvic::Nvic::new(stm32f429zi::nvic::EXTI0).enable(); gpio_ports.get_port_from_port_id(PortId::B).enable_clock();
217    gpio_ports.get_port_from_port_id(PortId::C).enable_clock();
218    gpio_ports.get_port_from_port_id(PortId::D).enable_clock();
219    gpio_ports.get_port_from_port_id(PortId::E).enable_clock();
220    gpio_ports.get_port_from_port_id(PortId::F).enable_clock();
221    gpio_ports.get_port_from_port_id(PortId::H).enable_clock();
223
224    gpio_ports.get_pin(PinId::PA03).map(|pin| {
226        pin.set_mode(stm32f429zi::gpio::Mode::AnalogMode);
227    });
228
229    gpio_ports.get_pin(PinId::PC00).map(|pin| {
231        pin.set_mode(stm32f429zi::gpio::Mode::AnalogMode);
232    });
233
234    gpio_ports.get_pin(PinId::PC03).map(|pin| {
236        pin.set_mode(stm32f429zi::gpio::Mode::AnalogMode);
237    });
238
239    gpio_ports.get_pin(PinId::PF03).map(|pin| {
241        pin.set_mode(stm32f429zi::gpio::Mode::AnalogMode);
242    });
243
244    gpio_ports.get_pin(PinId::PF05).map(|pin| {
246        pin.set_mode(stm32f429zi::gpio::Mode::AnalogMode);
247    });
248
249    gpio_ports.get_pin(PinId::PF10).map(|pin| {
251        pin.set_mode(stm32f429zi::gpio::Mode::AnalogMode);
252    });
253}
254
255unsafe fn setup_peripherals(tim2: &stm32f429zi::tim2::Tim2) {
257    cortexm4::nvic::Nvic::new(stm32f429zi::nvic::USART1).enable();
259
260    tim2.enable_clock();
262    tim2.start();
263    cortexm4::nvic::Nvic::new(stm32f429zi::nvic::TIM2).enable();
264}
265
266#[inline(never)]
272unsafe fn start() -> (
273    &'static kernel::Kernel,
274    STM32F429IDiscovery,
275    &'static stm32f429zi::chip::Stm32f4xx<'static, Stm32f429ziDefaultPeripherals<'static>>,
276) {
277    stm32f429zi::init();
278
279    let rcc = static_init!(stm32f429zi::rcc::Rcc, stm32f429zi::rcc::Rcc::new());
281    let clocks = static_init!(
282        stm32f429zi::clocks::Clocks<Stm32f429Specs>,
283        stm32f429zi::clocks::Clocks::new(rcc)
284    );
285    let syscfg = static_init!(
286        stm32f429zi::syscfg::Syscfg,
287        stm32f429zi::syscfg::Syscfg::new(clocks)
288    );
289    let exti = static_init!(
290        stm32f429zi::exti::Exti,
291        stm32f429zi::exti::Exti::new(syscfg)
292    );
293    let dma1 = static_init!(stm32f429zi::dma::Dma1, stm32f429zi::dma::Dma1::new(clocks));
294    let dma2 = static_init!(stm32f429zi::dma::Dma2, stm32f429zi::dma::Dma2::new(clocks));
295    let peripherals = static_init!(
296        Stm32f429ziDefaultPeripherals,
297        Stm32f429ziDefaultPeripherals::new(clocks, exti, dma1, dma2)
298    );
299
300    peripherals.init();
301    let base_peripherals = &peripherals.stm32f4;
302
303    setup_peripherals(&base_peripherals.tim2);
304
305    set_pin_primary_functions(syscfg, &base_peripherals.gpio_ports);
306
307    setup_dma(
308        dma2,
309        &base_peripherals.dma2_streams,
310        &base_peripherals.usart1,
311    );
312
313    let processes = components::process_array::ProcessArrayComponent::new()
315        .finalize(components::process_array_component_static!(NUM_PROCS));
316    PROCESSES = Some(processes);
317
318    let board_kernel = static_init!(kernel::Kernel, kernel::Kernel::new(processes.as_slice()));
320
321    let chip = static_init!(
322        stm32f429zi::chip::Stm32f4xx<Stm32f429ziDefaultPeripherals>,
323        stm32f429zi::chip::Stm32f4xx::new(peripherals)
324    );
325    CHIP = Some(chip);
326
327    base_peripherals.usart1.enable_clock();
334    let uart_mux = components::console::UartMuxComponent::new(&base_peripherals.usart1, 115200)
335        .finalize(components::uart_mux_component_static!());
336
337    (*addr_of_mut!(io::WRITER)).set_initialized();
338
339    let memory_allocation_capability = create_capability!(capabilities::MemoryAllocationCapability);
342    let process_management_capability =
343        create_capability!(capabilities::ProcessManagementCapability);
344
345    let console = components::console::ConsoleComponent::new(
347        board_kernel,
348        capsules_core::console::DRIVER_NUM,
349        uart_mux,
350    )
351    .finalize(components::console_component_static!());
352    components::debug_writer::DebugWriterComponent::new::<
354        <ChipHw as kernel::platform::chip::Chip>::ThreadIdProvider,
355    >(
356        uart_mux,
357        create_capability!(capabilities::SetDebugWriterCapability),
358    )
359    .finalize(components::debug_writer_component_static!());
360
361    let gpio_ports = &base_peripherals.gpio_ports;
365
366    let led = components::led::LedsComponent::new().finalize(components::led_component_static!(
367        LedHigh<'static, stm32f429zi::gpio::Pin>,
368        LedHigh::new(gpio_ports.get_pin(stm32f429zi::gpio::PinId::PG13).unwrap()),
369        LedHigh::new(gpio_ports.get_pin(stm32f429zi::gpio::PinId::PG14).unwrap()),
370        LedHigh::new(gpio_ports.get_pin(stm32f429zi::gpio::PinId::PB13).unwrap()),
371        LedHigh::new(gpio_ports.get_pin(stm32f429zi::gpio::PinId::PC05).unwrap()),
372    ));
373
374    let button = components::button::ButtonComponent::new(
376        board_kernel,
377        capsules_core::button::DRIVER_NUM,
378        components::button_component_helper!(
379            stm32f429zi::gpio::Pin,
380            (
381                gpio_ports.get_pin(stm32f429zi::gpio::PinId::PA00).unwrap(),
382                kernel::hil::gpio::ActivationMode::ActiveHigh,
383                kernel::hil::gpio::FloatingState::PullNone
384            )
385        ),
386    )
387    .finalize(components::button_component_static!(stm32f429zi::gpio::Pin));
388
389    let tim2 = &base_peripherals.tim2;
392    let mux_alarm = components::alarm::AlarmMuxComponent::new(tim2).finalize(
393        components::alarm_mux_component_static!(stm32f429zi::tim2::Tim2),
394    );
395
396    let alarm = components::alarm::AlarmDriverComponent::new(
397        board_kernel,
398        capsules_core::alarm::DRIVER_NUM,
399        mux_alarm,
400    )
401    .finalize(components::alarm_component_static!(stm32f429zi::tim2::Tim2));
402
403    let gpio = GpioComponent::new(
405        board_kernel,
406        capsules_core::gpio::DRIVER_NUM,
407        components::gpio_component_helper!(
408            stm32f429zi::gpio::Pin,
409            0 => gpio_ports.get_pin(PinId::PG09).unwrap(), 1 => gpio_ports.pins[6][14].as_ref().unwrap(), 2 => gpio_ports.pins[5][15].as_ref().unwrap(), 3 => gpio_ports.pins[4][13].as_ref().unwrap(), 4 => gpio_ports.pins[5][14].as_ref().unwrap(), 5 => gpio_ports.pins[4][11].as_ref().unwrap(), 6 => gpio_ports.pins[4][9].as_ref().unwrap(), 7 => gpio_ports.pins[5][13].as_ref().unwrap(), 8 => gpio_ports.pins[5][12].as_ref().unwrap(), 9 => gpio_ports.pins[3][15].as_ref().unwrap(), 10 => gpio_ports.pins[3][14].as_ref().unwrap(), 11 => gpio_ports.pins[0][7].as_ref().unwrap(),  12 => gpio_ports.pins[0][6].as_ref().unwrap(),  13 => gpio_ports.pins[0][5].as_ref().unwrap(),  14 => gpio_ports.pins[1][9].as_ref().unwrap(), 15 => gpio_ports.pins[1][8].as_ref().unwrap(), 16 => gpio_ports.pins[2][6].as_ref().unwrap(), 17 => gpio_ports.pins[1][15].as_ref().unwrap(), 18 => gpio_ports.pins[1][13].as_ref().unwrap(), 19 => gpio_ports.pins[1][12].as_ref().unwrap(), 20 => gpio_ports.pins[0][15].as_ref().unwrap(), 21 => gpio_ports.pins[2][7].as_ref().unwrap(), 26 => gpio_ports.pins[1][6].as_ref().unwrap(), 27 => gpio_ports.pins[1][2].as_ref().unwrap(), 28 => gpio_ports.pins[3][13].as_ref().unwrap(), 29 => gpio_ports.pins[3][12].as_ref().unwrap(), 30 => gpio_ports.pins[3][11].as_ref().unwrap(), 31 => gpio_ports.pins[4][2].as_ref().unwrap(), 33 => gpio_ports.pins[1][0].as_ref().unwrap(), 34 => gpio_ports.pins[4][0].as_ref().unwrap(), 35 => gpio_ports.pins[1][11].as_ref().unwrap(), 36 => gpio_ports.pins[1][10].as_ref().unwrap(), 37 => gpio_ports.pins[4][15].as_ref().unwrap(), 38 => gpio_ports.pins[4][14].as_ref().unwrap(), 39 => gpio_ports.pins[4][12].as_ref().unwrap(), 40 => gpio_ports.pins[4][10].as_ref().unwrap(), 41 => gpio_ports.pins[4][7].as_ref().unwrap(), 42 => gpio_ports.pins[4][8].as_ref().unwrap(), 43 => gpio_ports.pins[2][8].as_ref().unwrap(), 44 => gpio_ports.pins[2][9].as_ref().unwrap(), 45 => gpio_ports.pins[2][10].as_ref().unwrap(), 46 => gpio_ports.pins[2][11].as_ref().unwrap(), 47 => gpio_ports.pins[2][12].as_ref().unwrap(), 48 => gpio_ports.pins[3][2].as_ref().unwrap(), 49 => gpio_ports.pins[6][2].as_ref().unwrap(), 50 => gpio_ports.pins[6][3].as_ref().unwrap(), 51 => gpio_ports.pins[3][7].as_ref().unwrap(), 52 => gpio_ports.pins[3][6].as_ref().unwrap(), 53 => gpio_ports.pins[3][5].as_ref().unwrap(), 54 => gpio_ports.pins[3][4].as_ref().unwrap(), 55 => gpio_ports.pins[3][3].as_ref().unwrap(), 56 => gpio_ports.pins[4][2].as_ref().unwrap(), 57 => gpio_ports.pins[4][4].as_ref().unwrap(), 58 => gpio_ports.pins[4][5].as_ref().unwrap(), 59 => gpio_ports.pins[4][6].as_ref().unwrap(), 60 => gpio_ports.pins[4][3].as_ref().unwrap(), 61 => gpio_ports.pins[5][8].as_ref().unwrap(), 62 => gpio_ports.pins[5][7].as_ref().unwrap(), 63 => gpio_ports.pins[5][9].as_ref().unwrap(), 64 => gpio_ports.pins[6][1].as_ref().unwrap(), 65 => gpio_ports.pins[6][0].as_ref().unwrap(), 66 => gpio_ports.pins[3][1].as_ref().unwrap(), 67 => gpio_ports.pins[3][0].as_ref().unwrap(), 68 => gpio_ports.pins[5][0].as_ref().unwrap(), 69 => gpio_ports.pins[5][1].as_ref().unwrap(), 70 => gpio_ports.pins[5][2].as_ref().unwrap(), 71 => gpio_ports.pins[0][7].as_ref().unwrap()  ),
505    )
506    .finalize(components::gpio_component_static!(stm32f429zi::gpio::Pin));
507
508    let adc_mux = components::adc::AdcMuxComponent::new(&base_peripherals.adc1)
510        .finalize(components::adc_mux_component_static!(stm32f429zi::adc::Adc));
511
512    let temp_sensor = components::temperature_stm::TemperatureSTMComponent::new(
513        adc_mux,
514        stm32f429zi::adc::Channel::Channel18,
515        2.5,
516        0.76,
517    )
518    .finalize(components::temperature_stm_adc_component_static!(
519        stm32f429zi::adc::Adc
520    ));
521
522    let temp = components::temperature::TemperatureComponent::new(
523        board_kernel,
524        capsules_extra::temperature::DRIVER_NUM,
525        temp_sensor,
526    )
527    .finalize(components::temperature_component_static!(
528        TemperatureSTMSensor
529    ));
530
531    let adc_channel_0 =
532        components::adc::AdcComponent::new(adc_mux, stm32f429zi::adc::Channel::Channel3)
533            .finalize(components::adc_component_static!(stm32f429zi::adc::Adc));
534
535    let adc_channel_1 =
536        components::adc::AdcComponent::new(adc_mux, stm32f429zi::adc::Channel::Channel10)
537            .finalize(components::adc_component_static!(stm32f429zi::adc::Adc));
538
539    let adc_channel_2 =
540        components::adc::AdcComponent::new(adc_mux, stm32f429zi::adc::Channel::Channel13)
541            .finalize(components::adc_component_static!(stm32f429zi::adc::Adc));
542
543    let adc_channel_3 =
544        components::adc::AdcComponent::new(adc_mux, stm32f429zi::adc::Channel::Channel9)
545            .finalize(components::adc_component_static!(stm32f429zi::adc::Adc));
546
547    let adc_channel_4 =
548        components::adc::AdcComponent::new(adc_mux, stm32f429zi::adc::Channel::Channel15)
549            .finalize(components::adc_component_static!(stm32f429zi::adc::Adc));
550
551    let adc_channel_5 =
552        components::adc::AdcComponent::new(adc_mux, stm32f429zi::adc::Channel::Channel8)
553            .finalize(components::adc_component_static!(stm32f429zi::adc::Adc));
554
555    let adc_syscall =
556        components::adc::AdcVirtualComponent::new(board_kernel, capsules_core::adc::DRIVER_NUM)
557            .finalize(components::adc_syscall_component_helper!(
558                adc_channel_0,
559                adc_channel_1,
560                adc_channel_2,
561                adc_channel_3,
562                adc_channel_4,
563                adc_channel_5
564            ));
565
566    let process_printer = components::process_printer::ProcessPrinterTextComponent::new()
567        .finalize(components::process_printer_text_component_static!());
568    PROCESS_PRINTER = Some(process_printer);
569
570    let process_console = components::process_console::ProcessConsoleComponent::new(
572        board_kernel,
573        uart_mux,
574        mux_alarm,
575        process_printer,
576        Some(cortexm4::support::reset),
577    )
578    .finalize(components::process_console_component_static!(
579        stm32f429zi::tim2::Tim2
580    ));
581    let _ = process_console.start();
582
583    let scheduler = components::sched::round_robin::RoundRobinComponent::new(processes)
584        .finalize(components::round_robin_component_static!(NUM_PROCS));
585
586    let stm32f429i_discovery = STM32F429IDiscovery {
587        console,
588        ipc: kernel::ipc::IPC::new(
589            board_kernel,
590            kernel::ipc::DRIVER_NUM,
591            &memory_allocation_capability,
592        ),
593        adc: adc_syscall,
594        led,
595        temperature: temp,
596        button,
597        alarm,
598        gpio,
599
600        scheduler,
601        systick: cortexm4::systick::SysTick::new_with_calibration(
602            (HSI_FREQUENCY_MHZ * 1_000_000) as u32,
603        ),
604    };
605
606    debug!("Initialization complete. Entering main loop");
612
613    extern "C" {
615        static _sapps: u8;
617        static _eapps: u8;
619        static mut _sappmem: u8;
621        static _eappmem: u8;
623    }
624
625    kernel::process::load_processes(
626        board_kernel,
627        chip,
628        core::slice::from_raw_parts(
629            core::ptr::addr_of!(_sapps),
630            core::ptr::addr_of!(_eapps) as usize - core::ptr::addr_of!(_sapps) as usize,
631        ),
632        core::slice::from_raw_parts_mut(
633            core::ptr::addr_of_mut!(_sappmem),
634            core::ptr::addr_of!(_eappmem) as usize - core::ptr::addr_of!(_sappmem) as usize,
635        ),
636        &FAULT_RESPONSE,
637        &process_management_capability,
638    )
639    .unwrap_or_else(|err| {
640        debug!("Error loading processes!");
641        debug!("{:?}", err);
642    });
643
644    (board_kernel, stm32f429i_discovery, chip)
650}
651
652#[no_mangle]
654pub unsafe fn main() {
655    let main_loop_capability = create_capability!(capabilities::MainLoopCapability);
656
657    let (board_kernel, platform, chip) = start();
658    board_kernel.kernel_loop(&platform, chip, Some(&platform.ipc), &main_loop_capability);
659}