about summary refs log tree commit diff
path: root/src/lib/run_program.rs
diff options
context:
space:
mode:
authorMarijn Haverbeke <marijnh@gmail.com>2011-05-12 17:24:54 +0200
committerMarijn Haverbeke <marijnh@gmail.com>2011-05-12 21:30:44 +0200
commit3816e57fd2a8ab19e4ac6d4b3ddd5b49d5973ff2 (patch)
tree508982ed2f789aedd89eebd529343d9dc88b8e01 /src/lib/run_program.rs
parentb1d3364487eb647f2d7fcb412a260c960e38c73e (diff)
downloadrust-3816e57fd2a8ab19e4ac6d4b3ddd5b49d5973ff2.tar.gz
rust-3816e57fd2a8ab19e4ac6d4b3ddd5b49d5973ff2.zip
Downcase std modules again, move to :: for module dereferencing
This should be a snapshot transition.
Diffstat (limited to 'src/lib/run_program.rs')
-rw-r--r--src/lib/run_program.rs96
1 files changed, 96 insertions, 0 deletions
diff --git a/src/lib/run_program.rs b/src/lib/run_program.rs
new file mode 100644
index 00000000000..abcf1472067
--- /dev/null
+++ b/src/lib/run_program.rs
@@ -0,0 +1,96 @@
+import _str::sbuf;
+import _vec::vbuf;
+
+native "rust" mod rustrt {
+    fn rust_run_program(vbuf argv, int in_fd, int out_fd, int err_fd) -> int;
+}
+
+fn argvec(str prog, vec[str] args) -> vec[sbuf] {
+    auto argptrs = vec(_str::buf(prog));
+    for (str arg in args) {
+        _vec::push[sbuf](argptrs, _str::buf(arg));
+    }
+    _vec::push[sbuf](argptrs, 0 as sbuf);
+    ret argptrs;
+}
+
+fn run_program(str prog, vec[str] args) -> int {
+    auto pid = rustrt::rust_run_program(_vec::buf[sbuf](argvec(prog, args)),
+                                       0, 0, 0);
+    ret os::waitpid(pid);
+}
+
+type program =
+    state obj {
+        fn get_id() -> int;
+        fn input() -> io::writer;
+        fn output() -> io::reader;
+        fn close_input();
+        fn finish() -> int;
+    };
+
+fn start_program(str prog, vec[str] args) -> @program {
+    auto pipe_input = os::pipe();
+    auto pipe_output = os::pipe();
+    auto pid = rustrt::rust_run_program
+        (_vec::buf[sbuf](argvec(prog, args)),
+         pipe_input._0, pipe_output._1, 0);
+    if (pid == -1) {fail;}
+    os::libc::close(pipe_input._0);
+    os::libc::close(pipe_output._1);
+
+    state obj new_program(int pid,
+                          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() {
+            os::libc::close(in_fd);
+        }
+        fn finish() -> int {
+            if (finished) {ret 0;}
+            finished = true;
+            os::libc::close(in_fd);
+            ret os::waitpid(pid);
+        }
+        drop {
+            if (!finished) {
+                os::libc::close(in_fd);
+                os::waitpid(pid);
+            }
+            os::libc::fclose(out_file);
+        }
+    }
+    ret @new_program(pid, pipe_input._1,
+                     os::fd_FILE(pipe_output._0),
+                     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 .. 2>&1 | sed -e 's/\\/x\\//x:\\//g'";
+// End: