mirror of
https://github.com/fluencelabs/wasmer
synced 2025-03-30 14:31:07 +00:00
126 lines
3.6 KiB
Rust
126 lines
3.6 KiB
Rust
#![allow(dead_code)]
|
|
|
|
use super::file_descriptor::FileDescriptor;
|
|
use libc;
|
|
use std::io;
|
|
use std::io::BufReader;
|
|
use std::io::Read;
|
|
|
|
// A struct to hold the references to the base stdout and the captured one
|
|
pub struct StdioCapturer {
|
|
stdout_backup: libc::c_int,
|
|
stderr_backup: libc::c_int,
|
|
stdout_reader: libc::c_int,
|
|
stderr_reader: libc::c_int,
|
|
}
|
|
|
|
#[cfg(not(target_os = "windows"))]
|
|
use libc::{STDERR_FILENO, STDOUT_FILENO};
|
|
|
|
#[cfg(target_os = "windows")]
|
|
const _STDIN_FILENO: libc::c_int = 0;
|
|
#[cfg(target_os = "windows")]
|
|
const STDOUT_FILENO: libc::c_int = 1;
|
|
#[cfg(target_os = "windows")]
|
|
const STDERR_FILENO: libc::c_int = 2;
|
|
|
|
fn pipe(ptr: *mut libc::c_int) -> Result<libc::c_int, io::Error> {
|
|
#[cfg(not(target_os = "windows"))]
|
|
let result = unsafe { libc::pipe(ptr) };
|
|
#[cfg(target_os = "windows")]
|
|
let result = unsafe { libc::pipe(ptr, 1000, libc::O_TEXT) };
|
|
|
|
if result == -1 {
|
|
Err(std::io::Error::last_os_error())
|
|
} else {
|
|
Ok(result)
|
|
}
|
|
}
|
|
|
|
/// `dup` creates a new `fd`, make sure you `close` it when you're done with it!!
|
|
fn dup(oldfd: libc::c_int) -> Result<libc::c_int, io::Error> {
|
|
let result = unsafe { libc::dup(oldfd) };
|
|
|
|
if result == -1 {
|
|
Err(std::io::Error::last_os_error())
|
|
} else {
|
|
Ok(result)
|
|
}
|
|
}
|
|
|
|
fn dup2(oldfd: libc::c_int, newfd: libc::c_int) -> Result<libc::c_int, io::Error> {
|
|
let result = unsafe { libc::dup2(oldfd, newfd) };
|
|
|
|
if result == -1 {
|
|
Err(std::io::Error::last_os_error())
|
|
} else {
|
|
Ok(result)
|
|
}
|
|
}
|
|
|
|
fn close(fd: libc::c_int) -> Result<libc::c_int, io::Error> {
|
|
let result = unsafe { libc::close(fd) };
|
|
|
|
if result == -1 {
|
|
Err(std::io::Error::last_os_error())
|
|
} else {
|
|
Ok(result)
|
|
}
|
|
}
|
|
|
|
// Implementation inspired in
|
|
// https://github.com/rust-lang/rust/blob/7d52cbce6db83e4fc2d8706b4e4b9c7da76cbcf8/src/test/run-pass/issues/issue-30490.rs
|
|
// Currently only works in Unix systems (Mac, Linux)
|
|
impl StdioCapturer {
|
|
fn pipe() -> (libc::c_int, libc::c_int) {
|
|
let mut fds = [0; 2];
|
|
pipe(fds.as_mut_ptr()).unwrap();
|
|
|
|
(fds[0], fds[1])
|
|
}
|
|
|
|
pub fn new() -> Self {
|
|
let stdout_backup = dup(STDOUT_FILENO).unwrap();
|
|
let stderr_backup = dup(STDERR_FILENO).unwrap();
|
|
|
|
let (stdout_reader, stdout_writer) = Self::pipe();
|
|
let (stderr_reader, stderr_writer) = Self::pipe();
|
|
|
|
dup2(stdout_writer, STDOUT_FILENO).unwrap();
|
|
dup2(stderr_writer, STDERR_FILENO).unwrap();
|
|
|
|
// Make sure we close any duplicates of the writer end of the pipe,
|
|
// otherwise we can get stuck reading from the pipe which has open
|
|
// writers but no one supplying any input
|
|
close(stdout_writer).unwrap();
|
|
close(stderr_writer).unwrap();
|
|
|
|
StdioCapturer {
|
|
stdout_backup,
|
|
stderr_backup,
|
|
stdout_reader,
|
|
stderr_reader,
|
|
}
|
|
}
|
|
|
|
pub fn end(self) -> Result<(String, String), std::io::Error> {
|
|
dup2(self.stdout_backup, STDOUT_FILENO).unwrap();
|
|
dup2(self.stderr_backup, STDERR_FILENO).unwrap();
|
|
|
|
close(self.stdout_backup).unwrap();
|
|
close(self.stderr_backup).unwrap();
|
|
|
|
let fd = FileDescriptor::new(self.stdout_reader);
|
|
let mut reader = BufReader::new(fd);
|
|
let mut stdout_read = "".to_string();
|
|
let _ = reader.read_to_string(&mut stdout_read)?;
|
|
|
|
let fd = FileDescriptor::new(self.stderr_reader);
|
|
let mut reader = BufReader::new(fd);
|
|
let mut stderr_read = "".to_string();
|
|
let _ = reader.read_to_string(&mut stderr_read)?;
|
|
|
|
Ok((stdout_read, stderr_read))
|
|
}
|
|
}
|