Added support for adding codegen to intrrupt entry and exit (needed for RISC-V)

3 files changed, 32 insertions, 2 deletions

diff --git a/rtic-macros/src/codegen/async_dispatchers.rs b/rtic-macros/src/codegen/async_dispatchers.rs
index a12ad32..bdbfeaa 100644
--- a/rtic-macros/src/codegen/async_dispatchers.rs
+++ b/rtic-macros/src/codegen/async_dispatchers.rs
@@ -1,5 +1,11 @@
 use crate::syntax::ast::App;
-use crate::{analyze::Analysis, codegen::util};
+use crate::{
+    analyze::Analysis,
+    codegen::{
+        bindings::{interrupt_entry, interrupt_exit},
+        util,
+    },
+};
 use proc_macro2::TokenStream as TokenStream2;
 use quote::quote;
 
@@ -59,18 +65,25 @@ pub fn codegen(app: &App, analysis: &Analysis) -> TokenStream2 {
         if level > 0 {
             let doc = format!("Interrupt handler to dispatch async tasks at priority {level}");
             let attribute = &interrupts.get(&level).expect("UNREACHABLE").1.attrs;
+            let entry_stmts = interrupt_entry(app, analysis);
+            let exit_stmts = interrupt_exit(app, analysis);
+
             items.push(quote!(
                 #[allow(non_snake_case)]
                 #[doc = #doc]
                 #[no_mangle]
                 #(#attribute)*
                 unsafe fn #dispatcher_name() {
+                    #(#entry_stmts)*
+
                     /// The priority of this interrupt handler
                     const PRIORITY: u8 = #level;
 
                     rtic::export::run(PRIORITY, || {
                         #(#stmts)*
                     });
+
+                    #(#exit_stmts)*
                 }
             ));
         } else {
diff --git a/rtic-macros/src/codegen/bindings/cortex.rs b/rtic-macros/src/codegen/bindings/cortex.rs
index f028cee..d5eb12e 100644
--- a/rtic-macros/src/codegen/bindings/cortex.rs
+++ b/rtic-macros/src/codegen/bindings/cortex.rs
@@ -312,3 +312,11 @@ pub fn architecture_specific_analysis(app: &App, _: &SyntaxAnalysis) -> parse::R
 
     Ok(())
 }
+
+pub fn interrupt_entry(_app: &App, _analysis: &CodegenAnalysis) -> Vec<TokenStream2> {
+    vec![]
+}
+
+pub fn interrupt_exit(_app: &App, _analysis: &CodegenAnalysis) -> Vec<TokenStream2> {
+    vec![]
+}
diff --git a/rtic-macros/src/codegen/hardware_tasks.rs b/rtic-macros/src/codegen/hardware_tasks.rs
index 8a5a8f6..2c5254e 100644
--- a/rtic-macros/src/codegen/hardware_tasks.rs
+++ b/rtic-macros/src/codegen/hardware_tasks.rs
@@ -1,7 +1,10 @@
 use crate::syntax::{ast::App, Context};
 use crate::{
     analyze::Analysis,
-    codegen::{local_resources_struct, module, shared_resources_struct},
+    codegen::{
+        bindings::{interrupt_entry, interrupt_exit},
+        local_resources_struct, module, shared_resources_struct,
+    },
 };
 use proc_macro2::TokenStream as TokenStream2;
 use quote::quote;
@@ -17,6 +20,8 @@ pub fn codegen(app: &App, analysis: &Analysis) -> TokenStream2 {
         let priority = task.args.priority;
         let cfgs = &task.cfgs;
         let attrs = &task.attrs;
+        let entry_stmts = interrupt_entry(app, analysis);
+        let exit_stmts = interrupt_exit(app, analysis);
 
         mod_app.push(quote!(
             #[allow(non_snake_case)]
@@ -24,6 +29,8 @@ pub fn codegen(app: &App, analysis: &Analysis) -> TokenStream2 {
             #(#attrs)*
             #(#cfgs)*
             unsafe fn #symbol() {
+                #(#entry_stmts)*
+
                 const PRIORITY: u8 = #priority;
 
                 rtic::export::run(PRIORITY, || {
@@ -31,6 +38,8 @@ pub fn codegen(app: &App, analysis: &Analysis) -> TokenStream2 {
                         #name::Context::new()
                     )
                 });
+
+                #(#exit_stmts)*
             }
         ));