Add loongarch64 support

tikv · Nov 9, 2022 · 9c3c310 · 9c3c310
1 parent 383921f
commit 9c3c310
Show file tree

Hide file tree

Showing 3 changed files with 76 additions and 14 deletions.
diff --git a/src/backtrace/frame_pointer.rs b/src/backtrace/frame_pointer.rs
@@ -73,6 +73,9 @@ impl super::Trace for Trace {
             }
         };
 
+        #[cfg(all(target_arch = "loongarch64", target_os = "linux"))]
+        let frame_pointer = unsafe { (*ucontext).uc_mcontext.sc_regs[22] as usize };
+
         let mut frame_pointer = frame_pointer as *mut FramePointerLayout;
 
         let mut last_frame_pointer: *mut FramePointerLayout = null_mut();

diff --git a/src/backtrace/mod.rs b/src/backtrace/mod.rs
@@ -45,23 +45,39 @@ pub trait Trace {
 }
 
 #[cfg(not(all(
-    any(target_arch = "x86_64", target_arch = "aarch64"),
+    any(
+        target_arch = "x86_64",
+        target_arch = "aarch64",
+        target_arch = "loongarch64"
+    ),
     feature = "frame-pointer"
 )))]
 mod backtrace_rs;
 #[cfg(not(all(
-    any(target_arch = "x86_64", target_arch = "aarch64"),
+    any(
+        target_arch = "x86_64",
+        target_arch = "aarch64",
+        target_arch = "loongarch64"
+    ),
     feature = "frame-pointer"
 )))]
 pub use backtrace_rs::Trace as TraceImpl;
 
 #[cfg(all(
-    any(target_arch = "x86_64", target_arch = "aarch64"),
+    any(
+        target_arch = "x86_64",
+        target_arch = "aarch64",
+        target_arch = "loongarch64"
+    ),
     feature = "frame-pointer"
 ))]
 pub mod frame_pointer;
 #[cfg(all(
-    any(target_arch = "x86_64", target_arch = "aarch64"),
+    any(
+        target_arch = "x86_64",
+        target_arch = "aarch64",
+        target_arch = "loongarch64"
+    ),
     feature = "frame-pointer"
 ))]
 pub use frame_pointer::Trace as TraceImpl;
diff --git a/src/profiler.rs b/src/profiler.rs
@@ -9,7 +9,11 @@ use once_cell::sync::Lazy;
 use parking_lot::RwLock;
 use smallvec::SmallVec;
 
-#[cfg(any(target_arch = "x86_64", target_arch = "aarch64"))]
+#[cfg(any(
+    target_arch = "x86_64",
+    target_arch = "aarch64",
+    target_arch = "loongarch64"
+))]
 use findshlibs::{Segment, SharedLibrary, TargetSharedLibrary};
 
 use crate::backtrace::{Trace, TraceImpl};
@@ -29,14 +33,22 @@ pub struct Profiler {
 
     running: bool,
 
-    #[cfg(all(any(target_arch = "x86_64", target_arch = "aarch64")))]
+    #[cfg(all(any(
+        target_arch = "x86_64",
+        target_arch = "aarch64",
+        target_arch = "loongarch64"
+    )))]
     blocklist_segments: Vec<(usize, usize)>,
 }
 
 #[derive(Clone)]
 pub struct ProfilerGuardBuilder {
     frequency: c_int,
-    #[cfg(all(any(target_arch = "x86_64", target_arch = "aarch64")))]
+    #[cfg(all(any(
+        target_arch = "x86_64",
+        target_arch = "aarch64",
+        target_arch = "loongarch64"
+    )))]
     blocklist_segments: Vec<(usize, usize)>,
 }
 
@@ -45,7 +57,11 @@ impl Default for ProfilerGuardBuilder {
         ProfilerGuardBuilder {
             frequency: 99,
 
-            #[cfg(all(any(target_arch = "x86_64", target_arch = "aarch64")))]
+            #[cfg(all(any(
+                target_arch = "x86_64",
+                target_arch = "aarch64",
+                target_arch = "loongarch64"
+            )))]
             blocklist_segments: Vec::new(),
         }
     }
@@ -56,7 +72,11 @@ impl ProfilerGuardBuilder {
         Self { frequency, ..self }
     }
 
-    #[cfg(all(any(target_arch = "x86_64", target_arch = "aarch64")))]
+    #[cfg(all(any(
+        target_arch = "x86_64",
+        target_arch = "aarch64",
+        target_arch = "loongarch64"
+    )))]
     pub fn blocklist<T: AsRef<str>>(self, blocklist: &[T]) -> Self {
         let blocklist_segments = {
             let mut segments = Vec::new();
@@ -101,7 +121,11 @@ impl ProfilerGuardBuilder {
                 Err(Error::CreatingError)
             }
             Ok(profiler) => {
-                #[cfg(all(any(target_arch = "x86_64", target_arch = "aarch64")))]
+                #[cfg(all(any(
+                    target_arch = "x86_64",
+                    target_arch = "aarch64",
+                    target_arch = "loongarch64"
+                )))]
                 {
                     profiler.blocklist_segments = self.blocklist_segments;
                 }
@@ -234,7 +258,11 @@ impl Drop for ErrnoProtector {
 
 #[no_mangle]
 #[cfg_attr(
-    not(all(any(target_arch = "x86_64", target_arch = "aarch64"))),
+    not(all(any(
+        target_arch = "x86_64",
+        target_arch = "aarch64",
+        target_arch = "loongarch64"
+    ))),
     allow(unused_variables)
 )]
 extern "C" fn perf_signal_handler(
@@ -246,7 +274,11 @@ extern "C" fn perf_signal_handler(
 
     if let Some(mut guard) = PROFILER.try_write() {
         if let Ok(profiler) = guard.as_mut() {
-            #[cfg(any(target_arch = "x86_64", target_arch = "aarch64"))]
+            #[cfg(any(
+                target_arch = "x86_64",
+                target_arch = "aarch64",
+                target_arch = "loongarch64"
+            ))]
             if !ucontext.is_null() {
                 let ucontext: *mut libc::ucontext_t = ucontext as *mut libc::ucontext_t;
 
@@ -277,6 +309,9 @@ extern "C" fn perf_signal_handler(
                     }
                 };
 
+                #[cfg(all(target_arch = "loongarch64", target_os = "linux"))]
+                let addr = unsafe { (*ucontext).uc_mcontext.sc_pc as usize };
+
                 if profiler.is_blocklisted(addr) {
                     return;
                 }
@@ -324,12 +359,20 @@ impl Profiler {
             sample_counter: 0,
             running: false,
 
-            #[cfg(all(any(target_arch = "x86_64", target_arch = "aarch64")))]
+            #[cfg(all(any(
+                target_arch = "x86_64",
+                target_arch = "aarch64",
+                target_arch = "loongarch64"
+            )))]
             blocklist_segments: Vec::new(),
         })
     }
 
-    #[cfg(all(any(target_arch = "x86_64", target_arch = "aarch64")))]
+    #[cfg(all(any(
+        target_arch = "x86_64",
+        target_arch = "aarch64",
+        target_arch = "loongarch64"
+    )))]
     fn is_blocklisted(&self, addr: usize) -> bool {
         for libs in &self.blocklist_segments {
             if addr > libs.0 && addr < libs.1 {