69 lines
2.2 KiB
Rust
69 lines
2.2 KiB
Rust
// Copyright 2020 The ChromiumOS Authors
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
/// Very crude interactive console to allow the test host to run shell commands
|
|
/// in the guest and receive the output.
|
|
use std::fs::File;
|
|
use std::io;
|
|
use std::io::prelude::*;
|
|
use std::os::unix::process::ExitStatusExt;
|
|
use std::path::Path;
|
|
use std::process::Command;
|
|
use std::process::Stdio;
|
|
use std::str;
|
|
|
|
/// Device file to read from and write to.
|
|
const CONSOLE_FILE: &'static str = "/dev/ttyS1";
|
|
|
|
/// Line sent when we are ready to receive a command.
|
|
/// \x05 is the ENQ (enquiry) character, which is rarely used and 'should'
|
|
/// not appear in command output.
|
|
const READY_LINE: &'static str = "\x05READY";
|
|
|
|
/// Line sent containing the exit code of the program
|
|
/// \x05 is the ENQ (enquiry) character, which is rarely used and 'should'
|
|
/// not appear in command output.
|
|
const EXIT_CODE_LINE: &'static str = "\x05EXIT_CODE";
|
|
|
|
/// When ready to receive a command, the `READY_LINE` is written to `input`.
|
|
/// The received command is executed via /bin/sh and it's stdout is written
|
|
/// back to `output`, terminated by `EXIT_CODE_LINE ${exit_code}`.
|
|
fn listen(input: Box<dyn io::Read>, mut output: Box<dyn io::Write>) -> io::Result<()> {
|
|
let mut reader = io::BufReader::new(input);
|
|
loop {
|
|
writeln!(&mut output, "{}", READY_LINE).unwrap();
|
|
|
|
let mut command = String::new();
|
|
reader.read_line(&mut command)?;
|
|
if command.trim() == "exit" {
|
|
break;
|
|
}
|
|
|
|
println!("-> {:?}", command);
|
|
let result = Command::new("/bin/sh")
|
|
.args(&["-c", &command])
|
|
.stderr(Stdio::inherit())
|
|
.output()
|
|
.unwrap();
|
|
let exit_code = match result.status.code() {
|
|
Some(code) => code,
|
|
None => -result.status.signal().unwrap(),
|
|
};
|
|
|
|
output.write(&result.stdout)?;
|
|
println!("<- {}", exit_code);
|
|
writeln!(&mut output, "{EXIT_CODE_LINE} {exit_code}")?;
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
fn main() {
|
|
let path = Path::new(CONSOLE_FILE);
|
|
listen(
|
|
Box::new(File::open(path).unwrap()),
|
|
Box::new(File::create(path).unwrap()),
|
|
)
|
|
.unwrap();
|
|
}
|