rust/src/lib/run_program.rs

106 lines
3.0 KiB
Rust
Raw Normal View History

import str::sbuf;
import vec::vbuf;
2011-07-18 23:07:33 -05:00
export program;
export run_program;
export start_program;
export program_output;
export spawn_process;
2011-07-18 23:07:33 -05:00
native "rust" mod rustrt {
fn rust_run_program(vbuf argv, int in_fd, int out_fd, int err_fd) -> int;
}
fn arg_vec(str prog, vec[str] args) -> vec[sbuf] {
auto argptrs = [str::buf(prog)];
for (str arg in args) { vec::push[sbuf](argptrs, str::buf(arg)); }
vec::push[sbuf](argptrs, 0 as sbuf);
ret argptrs;
}
2011-07-18 23:07:33 -05:00
fn spawn_process(str prog, vec[str] args,
int in_fd, int out_fd, int err_fd) -> int {
// Note: we have to hold on to this vector reference while we hold a
// pointer to its buffer
auto argv = arg_vec(prog, args);
2011-07-18 23:07:33 -05:00
auto pid = rustrt::rust_run_program(vec::buf(argv),
in_fd, out_fd, err_fd);
ret pid;
}
fn run_program(str prog, vec[str] args) -> int {
ret os::waitpid(spawn_process(prog, args, 0, 0, 0));
}
type program =
obj {
fn get_id() -> int ;
fn input() -> io::writer ;
fn output() -> io::reader ;
fn close_input() ;
fn finish() -> int ;
};
2011-04-19 15:35:49 -05:00
fn start_program(str prog, vec[str] args) -> @program {
auto pipe_input = os::pipe();
auto pipe_output = os::pipe();
auto pid = spawn_process(prog, args, pipe_input.in, pipe_output.out, 0);
2011-07-18 23:07:33 -05:00
if (pid == -1) { fail; }
os::libc::close(pipe_input.in);
os::libc::close(pipe_output.out);
obj new_program(int pid,
mutable int in_fd,
os::libc::FILE out_file,
mutable bool finished) {
fn get_id() -> int { ret pid; }
fn input() -> io::writer {
ret io::new_writer(io::fd_buf_writer(in_fd, false));
}
fn output() -> io::reader {
ret io::new_reader(io::FILE_buf_reader(out_file, false));
}
fn close_input() {
auto invalid_fd = -1;
if (in_fd != invalid_fd) {
os::libc::close(in_fd);
in_fd = invalid_fd;
}
}
2011-04-19 15:35:49 -05:00
fn finish() -> int {
if (finished) { ret 0; }
finished = true;
self.close_input();
ret os::waitpid(pid);
}drop {
self.close_input();
if (!finished) {
os::waitpid(pid);
}
os::libc::fclose(out_file);
}
}
ret @new_program(pid, pipe_input.out, os::fd_FILE(pipe_output.in), false);
}
fn program_output(str prog, vec[str] args) -> rec(int status, str out) {
auto pr = start_program(prog, args);
pr.close_input();
auto out = pr.output();
auto buf = "";
while (!out.eof()) {
auto bytes = out.read_bytes(4096u);
buf += str::unsafe_from_bytes(bytes);
}
ret rec(status=pr.finish(), out=buf);
}
// Local Variables:
// mode: rust
// fill-column: 78;
// indent-tabs-mode: nil
// c-basic-offset: 4
// buffer-file-coding-system: utf-8-unix
// compile-command: "make -k -C $RBUILD 2>&1 | sed -e 's/\\/x\\//x:\\//g'";
// End: