reverie/tests/backtrace.rs
Jason White cc40e95c91 Remove unnecessary Serialize/Deserialize impls for tools
Summary: Now that `Tool` doesn't require `Serialize` and `Deserialize`, we can remove these unnecessary derives. Hopefully this will have a slight improvement to compile-times.

Reviewed By: wkhughes

Differential Revision: D40958697

fbshipit-source-id: 88aa1f4ee2b953ba287d749c88d200c2887ccd46
2022-11-03 12:53:11 -07:00

71 lines
2.1 KiB
Rust

/*
* Copyright (c) Meta Platforms, Inc. and affiliates.
* All rights reserved.
*
* This source code is licensed under the BSD-style license found in the
* LICENSE file in the root directory of this source tree.
*/
//! Tests for getting backtraces from the guest.
#![cfg(not(sanitized))]
use reverie::syscalls::Syscall;
use reverie::Error;
use reverie::ExitStatus;
use reverie::Guest;
use reverie::Tool;
#[derive(Debug, Default, Clone)]
struct TestTool;
#[reverie::tool]
impl Tool for TestTool {
async fn handle_syscall_event<T: Guest<Self>>(
&self,
guest: &mut T,
syscall: Syscall,
) -> Result<i64, Error> {
if let Syscall::Getpid(_) = &syscall {
let backtrace = guest
.backtrace()
.expect("failed to get backtrace from guest");
// There's no guarantee our function is at the top of the stack, so
// we simply assert that it is *somewhere* in the stack.
assert!(
backtrace.iter().any(|frame| {
if let Some(symbol) = &frame.symbol {
// Due to name mangling, there won't be an exact match.
symbol.name.contains("funky_function")
} else {
// FIXME: The unwind library is currently broken on
// platform010. It is not able to find symbols for any
// stack frames. Change this back to `false` when that
// is fixed. For now, file and line numbers still work.
true
//false
}
}),
"guest backtrace did not contain our expected function:\n{}",
backtrace.pretty().unwrap()
);
}
Ok(guest.inject(syscall).await?)
}
}
#[inline(never)]
fn funky_function() {
let _ = unsafe { libc::getpid() };
}
#[test]
fn smoke() {
use reverie_ptrace::testing::test_fn;
let (output, _) = test_fn::<TestTool, _>(funky_function).unwrap();
assert_eq!(output.status, ExitStatus::Exited(0));
}