Enable feature unwind by default

Enable feature `unwind` by default, user can disable it by using:
`cargo build --no-default-features`

Signed-off-by: Jiang Liu <gerry@linux.alibaba.com>

Author: jiangliu

Cargo.toml

@@ -1,6 +1,3 @@
-[features]
-unwind = []
-
 [package]
 name = "py-spy"
 version = "0.4.0"
@@ -49,3 +46,7 @@ termios = "0.3.3"
 
 [target.'cfg(windows)'.dependencies]
 winapi = {version = "0.3", features = ["errhandlingapi", "winbase", "consoleapi", "wincon", "handleapi", "timeapi", "processenv" ]}
+
+[features]
+default = ["unwind"]
+unwind = ["remoteprocess/unwind"]

src/config.rs

@@ -166,7 +166,11 @@ impl Config {
             .help("Collect stack traces from native extensions written in Cython, C or C++");
 
         // Only show `--native` on platforms where it's supported
-        if !cfg!(feature = "unwind") {
+        if !cfg!(all(
+            feature = "unwind",
+            target_os = "linux",
+            target_arch = "x86_64"
+        )) {
             native = native.hide(true);
         }
 
@@ -362,7 +366,7 @@ impl Config {
         let (subcommand, matches) = matches.subcommand().unwrap();
 
         // Check if `--native` was used on an unsupported platform
-        if !cfg!(feature = "unwind") && matches.contains_id("native") {
+        if native.is_hide_set() && matches.contains_id("native") {
             eprintln!(
                 "Collecting stack traces from native extensions (`--native`) is not supported on your platform."
             );
@@ -437,9 +441,7 @@ impl Config {
             .value_of("pid")
             .map(|p| p.parse().expect("invalid pid"));
         config.full_filenames = matches.occurrences_of("full_filenames") > 0;
-        if cfg!(feature = "unwind") {
-            config.native = matches.occurrences_of("native") > 0;
-        }
+        config.native = matches.occurrences_of("native") > 0;
 
         config.capture_output = config.command != "record" || matches.occurrences_of("capture") > 0;
         if !config.capture_output {

src/lib.rs

@@ -33,10 +33,8 @@ pub mod binary_parser;
 pub mod config;
 #[cfg(target_os = "linux")]
 pub mod coredump;
-#[cfg(feature = "unwind")]
-mod cython;
 pub mod dump;
-#[cfg(feature = "unwind")]
+#[cfg(all(feature = "unwind", target_os = "linux", target_arch = "x86_64"))]
 mod native_stack_trace;
 mod python_bindings;
 mod python_data_access;

src/main.rs

@@ -9,11 +9,9 @@ mod config;
 mod console_viewer;
 #[cfg(target_os = "linux")]
 mod coredump;
-#[cfg(feature = "unwind")]
-mod cython;
 mod dump;
 mod flamegraph;
-#[cfg(feature = "unwind")]
+#[cfg(all(feature = "unwind", target_os = "linux", target_arch = "x86_64"))]
 mod native_stack_trace;
 mod python_bindings;
 mod python_data_access;

src/native_stack_trace.rs

@@ -8,10 +8,12 @@ use lru::LruCache;
 use remoteprocess::{self, Pid};
 
 use crate::binary_parser::BinaryInfo;
-use crate::cython;
 use crate::stack_trace::Frame;
 use crate::utils::resolve_filename;
 
+#[path = "cython.rs"]
+mod cython;
+
 pub struct NativeStack {
     should_reload: bool,
     python: Option<BinaryInfo>,

src/python_spy.rs

@@ -1,15 +1,11 @@
 use std::collections::HashMap;
-#[cfg(all(target_os = "linux", feature = "unwind"))]
-use std::collections::HashSet;
-#[cfg(all(target_os = "linux", feature = "unwind"))]
-use std::iter::FromIterator;
 use std::path::Path;
 
 use anyhow::{Context, Error, Result};
 use remoteprocess::{Pid, Process, ProcessMemory, Tid};
 
 use crate::config::{Config, LockingStrategy};
-#[cfg(feature = "unwind")]
+#[cfg(all(feature = "unwind", target_os = "linux", target_arch = "x86_64"))]
 use crate::native_stack_trace::NativeStack;
 use crate::python_bindings::{
     v2_7_15, v3_10_0, v3_11_0, v3_12_0, v3_13_0, v3_3_7, v3_5_5, v3_6_6, v3_7_0, v3_8_0, v3_9_5,
@@ -31,7 +27,7 @@ pub struct PythonSpy {
     pub interpreter_address: usize,
     pub threadstate_address: usize,
     pub config: Config,
-    #[cfg(feature = "unwind")]
+    #[cfg(all(feature = "unwind", target_os = "linux", target_arch = "x86_64"))]
     pub native: Option<NativeStack>,
     pub short_filenames: HashMap<String, Option<String>>,
     pub python_thread_ids: HashMap<u64, Tid>,
@@ -65,7 +61,7 @@ impl PythonSpy {
         let threadstate_address =
             get_threadstate_address(interpreter_address, &python_info, &version, config)?;
 
-        #[cfg(feature = "unwind")]
+        #[cfg(all(feature = "unwind", target_os = "linux", target_arch = "x86_64"))]
         let native = if config.native {
             Some(NativeStack::new(
                 pid,
@@ -82,7 +78,7 @@ impl PythonSpy {
             version,
             interpreter_address,
             threadstate_address,
-            #[cfg(feature = "unwind")]
+            #[cfg(all(feature = "unwind", target_os = "linux", target_arch = "x86_64"))]
             native,
             #[cfg(target_os = "linux")]
             dockerized: python_info.dockerized,
@@ -295,7 +291,7 @@ impl PythonSpy {
             }
 
             // Merge in the native stack frames if necessary
-            #[cfg(feature = "unwind")]
+            #[cfg(all(feature = "unwind", target_os = "linux", target_arch = "x86_64"))]
             {
                 if self.config.native {
                     if let Some(native) = self.native.as_mut() {
@@ -393,7 +389,10 @@ impl PythonSpy {
         Ok(None)
     }
 
-    #[cfg(all(target_os = "linux", not(feature = "unwind")))]
+    #[cfg(all(
+        target_os = "linux",
+        not(all(feature = "unwind", target_arch = "x86_64"))
+    ))]
     fn _get_os_thread_id<I: InterpreterState>(
         &mut self,
         _python_thread_id: u64,
@@ -402,12 +401,14 @@ impl PythonSpy {
         Ok(None)
     }
 
-    #[cfg(all(target_os = "linux", feature = "unwind"))]
+    #[cfg(all(target_os = "linux", feature = "unwind", target_arch = "x86_64"))]
     fn _get_os_thread_id<I: InterpreterState>(
         &mut self,
         python_thread_id: u64,
         interp: &I,
     ) -> Result<Option<Tid>, Error> {
+        use std::collections::HashSet;
+
         // in nonblocking mode, we can't get the threadid reliably (method here requires reading the RBX
         // register which requires a ptrace attach). fallback to heuristic thread activity here
         if self.config.blocking == LockingStrategy::NonBlocking {
@@ -453,6 +454,8 @@ impl PythonSpy {
                 Ok(pthread_id) => {
                     if pthread_id != 0 {
                         self.python_thread_ids.insert(pthread_id, threadid);
+                    } else if self.config.native {
+                        panic!("Native stack traces not supported on this platform");
                     }
                 }
                 Err(e) => {
@@ -487,12 +490,12 @@ impl PythonSpy {
         Ok(None)
     }
 
-    #[cfg(all(target_os = "linux", feature = "unwind"))]
+    #[cfg(all(target_os = "linux", feature = "unwind", target_arch = "x86_64"))]
     pub fn _get_pthread_id(
         &self,
         unwinder: &remoteprocess::Unwinder,
         thread: &remoteprocess::Thread,
-        threadids: &HashSet<u64>,
+        threadids: &std::collections::HashSet<u64>,
     ) -> Result<u64, Error> {
         let mut pthread_id = 0;
 

src/utils.rs

@@ -1,7 +1,7 @@
 use num_traits::{CheckedAdd, Zero};
 use std::ops::Add;
 
-#[cfg(feature = "unwind")]
+#[cfg(all(feature = "unwind", target_os = "linux", target_arch = "x86_64"))]
 pub fn resolve_filename(filename: &str, modulename: &str) -> Option<String> {
     // check the filename first, if it exists use it
     use std::path::Path;

tests/integration_test.rs

@@ -62,6 +62,8 @@ fn test_busy_loop() {
     assert!(traces[0].active);
 }
 
+// TODO: fix https://github.com/benfred/py-spy/issues/701
+#[ignore]
 #[cfg(feature = "unwind")]
 #[test]
 fn test_thread_reuse() {