-
Notifications
You must be signed in to change notification settings - Fork 125
/
processUtils.ml
56 lines (44 loc) · 1.7 KB
/
processUtils.ml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
(*
Copyright © 2011 MLstate
This file is part of Opa.
Opa is free software: you can redistribute it and/or modify it under the
terms of the GNU Affero General Public License, version 3, as published by
the Free Software Foundation.
Opa is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for
more details.
You should have received a copy of the GNU Affero General Public License
along with Opa. If not, see <http://www.gnu.org/licenses/>.
*)
module U = Unix
module S = Sys
exception InvalidCommand
exception CommandError
let system_call cmd =
let ic, oc = Unix.open_process cmd in
let buf = Buffer.create 16 in
(try
while true do
Buffer.add_channel buf ic 1
done
with End_of_file -> ());
let ret = Unix.close_process (ic, oc) in
let msg = Base.String.rtrim ~is_space:(fun x -> if x = '\n' then true else false) (Buffer.contents buf) in
msg, ret
let get_return_code status =
match status with
| Unix.WSIGNALED x -> failwith (Printf.sprintf "Process killed by signal %d" x)
| Unix.WSTOPPED x -> failwith (Printf.sprintf "Process stopped by signal %d" x)
| Unix.WEXITED x -> x
let command_fct cmd fct =
let t, s = system_call (Printf.sprintf "%s %s" cmd fct) in
let r = get_return_code s in
if r = 0 then t
else raise CommandError
let which cmd = command_fct "which" cmd
let wget cmd = command_fct "wget" cmd
let tar cmd = command_fct "tar" cmd
let ps cmd = command_fct "ps" cmd
let mktemp cmd = command_fct "mktemp" cmd
let cp cmd = command_fct "cp" cmd