mirror of
https://github.com/facebookexperimental/reverie.git
synced 2024-11-28 09:24:18 +00:00
9a942d055f
Summary: Removes usage of the `associated_type_defaults` unstable feature, but does not actually disable the feature yet. This brings us one step closer to removing all usage of unstable nightly features, which will allow publishing a crate on https://crates.io. Reviewed By: rrnewton Differential Revision: D41388745 fbshipit-source-id: f347a577857a713fe3088a556cbf37ffe92e5553
104 lines
3.1 KiB
Rust
104 lines
3.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.
|
|
*/
|
|
|
|
//! An example that tracks thread pedigree using local state
|
|
use clap::Parser;
|
|
use reverie::syscalls::Syscall;
|
|
use reverie::Error;
|
|
use reverie::Guest;
|
|
use reverie::Pid;
|
|
use reverie::Tool;
|
|
use reverie_util::pedigree::Pedigree;
|
|
use reverie_util::CommonToolArguments;
|
|
use serde::Deserialize;
|
|
use serde::Serialize;
|
|
use tracing::debug;
|
|
use tracing::trace;
|
|
|
|
// TODO: Add handle pedigree forking, initialization, etc. to tool.
|
|
// This tool is NOT FUNCTIONAL in its current state.
|
|
|
|
#[derive(Debug, Serialize, Deserialize, Default, Clone)]
|
|
struct PedigreeLocal(Pedigree);
|
|
|
|
#[reverie::tool]
|
|
impl Tool for PedigreeLocal {
|
|
type GlobalState = ();
|
|
type ThreadState = PedigreeLocal;
|
|
|
|
fn new(pid: Pid, _cfg: &()) -> Self {
|
|
debug!("[pedigree] initialize pedigree for pid {}", pid);
|
|
PedigreeLocal(Pedigree::new())
|
|
}
|
|
|
|
fn init_thread_state(
|
|
&self,
|
|
_tid: Pid,
|
|
parent: Option<(Pid, &Self::ThreadState)>,
|
|
) -> Self::ThreadState {
|
|
if let Some((_, state)) = parent {
|
|
let mut parent = state.clone();
|
|
let child = parent.0.fork_mut();
|
|
trace!("child pedigree: {:?}", child);
|
|
PedigreeLocal(child)
|
|
} else {
|
|
PedigreeLocal(Pedigree::new())
|
|
}
|
|
}
|
|
|
|
async fn handle_syscall_event<T: Guest<Self>>(
|
|
&self,
|
|
guest: &mut T,
|
|
syscall: Syscall,
|
|
) -> Result<i64, Error> {
|
|
match syscall {
|
|
#[cfg(target_arch = "x86_64")]
|
|
Syscall::Fork(_) | Syscall::Vfork(_) => self.handle_fork(syscall, guest).await,
|
|
Syscall::Clone(_) => self.handle_fork(syscall, guest).await,
|
|
Syscall::Getpid(_) | Syscall::Getppid(_) | Syscall::Gettid(_) | Syscall::Getpgid(_) => {
|
|
let pid = guest.inject(syscall).await?;
|
|
let vpid = nix::unistd::Pid::try_from(&self.0).unwrap();
|
|
trace!("getpid returned {:?} vpid: {:?}", pid, vpid);
|
|
Ok(pid)
|
|
}
|
|
Syscall::Setpgid(_) => {
|
|
panic!("[pedigree] setpgid is not allowed.");
|
|
}
|
|
_ => guest.tail_inject(syscall).await,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl PedigreeLocal {
|
|
async fn handle_fork(
|
|
&self,
|
|
syscall: Syscall,
|
|
guest: &mut impl Guest<Self>,
|
|
) -> Result<i64, Error> {
|
|
let retval = guest.inject(syscall).await?;
|
|
let pedigree = guest.thread_state_mut().0.fork_mut();
|
|
trace!(
|
|
"got new pedigree: {:?} => {:x?}",
|
|
pedigree,
|
|
nix::unistd::Pid::try_from(&pedigree)
|
|
);
|
|
Ok(retval)
|
|
}
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), Error> {
|
|
let args = CommonToolArguments::from_args();
|
|
let log_guard = args.init_tracing();
|
|
let tracer = reverie_ptrace::TracerBuilder::<PedigreeLocal>::new(args.into())
|
|
.spawn()
|
|
.await?;
|
|
let (status, _global_state) = tracer.wait().await?;
|
|
drop(log_guard); // Flush logs before exiting.
|
|
status.raise_or_exit()
|
|
}
|