1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00
v/vlib/os/process.v

318 lines
7.6 KiB
V
Raw Normal View History

2020-11-16 19:32:50 +03:00
module os
// ProcessState.not_started - the process has not yet started
// ProcessState.running - the process is currently running
// ProcessState.stopped - the process was running, but was stopped temporarily
// ProcessState.exited - the process has finished/exited
// ProcessState.aborted - the process was terminated by a signal
// ProcessState.closed - the process resources like opened file descriptors were freed/discarded, final state.
2020-11-16 19:32:50 +03:00
pub enum ProcessState {
not_started
running
stopped
exited
aborted
closed
2020-11-16 19:32:50 +03:00
}
[heap]
2020-11-16 19:32:50 +03:00
pub struct Process {
pub:
filename string // the process's command file path
2020-11-16 19:32:50 +03:00
pub mut:
pid int // the PID of the process
code int = -1
2020-11-16 19:32:50 +03:00
// the exit code of the process, != -1 *only* when status is .exited *and* the process was not aborted
status ProcessState = .not_started
2020-11-16 19:32:50 +03:00
// the current status of the process
err string // if the process fails, contains the reason why
2020-11-16 19:32:50 +03:00
args []string // the arguments that the command takes
env_is_custom bool // true, when the environment was customized with .set_environment
2021-02-07 07:19:05 +03:00
env []string // the environment with which the process was started (list of 'var=val')
use_stdio_ctl bool // when true, then you can use p.stdin_write(), p.stdout_slurp() and p.stderr_slurp()
2021-04-04 17:05:06 +03:00
use_pgroup bool // when true, the process will create a new process group, enabling .signal_pgkill()
stdio_fd [3]int // the stdio file descriptors for the child process, used only by the nix implementation
wdata voidptr // the WProcess; used only by the windows implementation
2020-11-16 19:32:50 +03:00
}
// new_process - create a new process descriptor
// NB: new does NOT start the new process.
// That is done because you may want to customize it first,
// by calling different set_ methods on it.
// In order to start it, call p.run() or p.wait()
pub fn new_process(filename string) &Process {
return &Process{
filename: filename
2021-04-04 17:05:06 +03:00
stdio_fd: [-1, -1, -1]!
2020-11-16 19:32:50 +03:00
}
}
// set_args - set the arguments for the new process
pub fn (mut p Process) set_args(pargs []string) {
2020-11-16 19:32:50 +03:00
if p.status != .not_started {
return
2020-11-16 19:32:50 +03:00
}
p.args = pargs
return
2020-11-16 19:32:50 +03:00
}
// set_environment - set a custom environment variable mapping for the new process
pub fn (mut p Process) set_environment(envs map[string]string) {
2020-11-16 19:32:50 +03:00
if p.status != .not_started {
return
2020-11-16 19:32:50 +03:00
}
p.env_is_custom = true
p.env = []string{}
for k, v in envs {
p.env << '$k=$v'
}
return
2020-11-16 19:32:50 +03:00
}
// run - starts the new process
pub fn (mut p Process) run() {
2020-11-16 19:32:50 +03:00
if p.status != .not_started {
return
2020-11-16 19:32:50 +03:00
}
p._spawn()
return
2020-11-16 19:32:50 +03:00
}
// signal_kill - kills the process, after that it is no longer running
pub fn (mut p Process) signal_kill() {
2020-11-16 19:32:50 +03:00
if p.status !in [.running, .stopped] {
return
2020-11-16 19:32:50 +03:00
}
p._signal_kill()
p.status = .aborted
return
2020-11-16 19:32:50 +03:00
}
2021-04-04 17:05:06 +03:00
// signal_pgkill - kills the whole process group
pub fn (mut p Process) signal_pgkill() {
if p.status !in [.running, .stopped] {
return
}
p._signal_pgkill()
return
}
2020-11-16 19:32:50 +03:00
// signal_stop - stops the process, you can resume it with p.signal_continue()
pub fn (mut p Process) signal_stop() {
2020-11-16 19:32:50 +03:00
if p.status != .running {
return
2020-11-16 19:32:50 +03:00
}
p._signal_stop()
p.status = .stopped
return
2020-11-16 19:32:50 +03:00
}
// signal_continue - tell a stopped process to continue/resume its work
pub fn (mut p Process) signal_continue() {
2020-11-16 19:32:50 +03:00
if p.status != .stopped {
return
2020-11-16 19:32:50 +03:00
}
p._signal_continue()
p.status = .running
return
2020-11-16 19:32:50 +03:00
}
// wait - wait for a process to finish.
// NB: You have to call p.wait(), otherwise a finished process
// would get to a zombie state, and its resources will not get
// released fully, until its parent process exits.
// NB: This call will block the calling process until the child
// process is finished.
pub fn (mut p Process) wait() {
2020-11-16 19:32:50 +03:00
if p.status == .not_started {
p._spawn()
}
if p.status !in [.running, .stopped] {
return
2020-11-16 19:32:50 +03:00
}
p._wait()
return
2020-11-16 19:32:50 +03:00
}
// close - free the OS resources associated with the process.
// Can be called multiple times, but will free the resources just once.
// This sets the process state to .closed, which is final.
pub fn (mut p Process) close() {
if p.status in [.not_started, .closed] {
return
}
p.status = .closed
$if !windows {
for i in 0 .. 3 {
if p.stdio_fd[i] != 0 {
fd_close(p.stdio_fd[i])
}
}
}
}
[unsafe]
pub fn (mut p Process) free() {
p.close()
unsafe {
p.filename.free()
p.err.free()
p.args.free()
p.env.free()
}
}
2020-11-16 19:32:50 +03:00
//
// _spawn - should not be called directly, but only by p.run()/p.wait() .
// It encapsulates the fork/execve mechanism that allows the
// asynchronous starting of the new child process.
fn (mut p Process) _spawn() int {
if !p.env_is_custom {
p.env = []string{}
current_environment := environ()
for k, v in current_environment {
p.env << '$k=$v'
}
}
mut pid := 0
$if windows {
pid = p.win_spawn_process()
} $else {
pid = p.unix_spawn_process()
}
p.pid = pid
p.status = .running
return 0
}
// is_alive - query whether the process p.pid is still alive
pub fn (mut p Process) is_alive() bool {
if p.status in [.running, .stopped] {
return p._is_alive()
}
return false
}
//
pub fn (mut p Process) set_redirect_stdio() {
2020-11-16 19:32:50 +03:00
p.use_stdio_ctl = true
return
2020-11-16 19:32:50 +03:00
}
pub fn (mut p Process) stdin_write(s string) {
p._check_redirection_call('stdin_write')
2021-04-04 17:05:06 +03:00
$if windows {
p.win_write_string(0, s)
} $else {
fd_write(p.stdio_fd[0], s)
}
2020-11-16 19:32:50 +03:00
}
2021-02-07 07:19:05 +03:00
// will read from stdout pipe, will only return when EOF (end of file) or data
// means this will block unless there is data
2020-11-16 19:32:50 +03:00
pub fn (mut p Process) stdout_slurp() string {
p._check_redirection_call('stdout_slurp')
2021-04-04 17:05:06 +03:00
$if windows {
return p.win_slurp(1)
} $else {
return fd_slurp(p.stdio_fd[1]).join('')
}
2020-11-16 19:32:50 +03:00
}
2021-02-07 07:19:05 +03:00
// read from stderr pipe, wait for data or EOF
2020-11-16 19:32:50 +03:00
pub fn (mut p Process) stderr_slurp() string {
p._check_redirection_call('stderr_slurp')
2021-04-04 17:05:06 +03:00
$if windows {
return p.win_slurp(2)
} $else {
return fd_slurp(p.stdio_fd[2]).join('')
}
2020-11-16 19:32:50 +03:00
}
2021-02-07 07:19:05 +03:00
// read from stdout, return if data or not
2020-11-16 19:32:50 +03:00
pub fn (mut p Process) stdout_read() string {
p._check_redirection_call('stdout_read')
2021-04-04 17:05:06 +03:00
$if windows {
s, _ := p.win_read_string(1, 4096)
return s
} $else {
s, _ := fd_read(p.stdio_fd[1], 4096)
return s
}
2020-11-16 19:32:50 +03:00
}
pub fn (mut p Process) stderr_read() string {
p._check_redirection_call('stderr_read')
2021-04-04 17:05:06 +03:00
$if windows {
s, _ := p.win_read_string(2, 4096)
return s
} $else {
s, _ := fd_read(p.stdio_fd[2], 4096)
return s
}
2020-11-16 19:32:50 +03:00
}
// _check_redirection_call - should be called just by stdxxx methods
fn (mut p Process) _check_redirection_call(fn_name string) {
if !p.use_stdio_ctl {
panic('Call p.set_redirect_stdio() before calling p.$fn_name')
}
if p.status == .not_started {
panic('Call p.${fn_name}() after you have called p.run()')
}
}
// _signal_stop - should not be called directly, except by p.signal_stop
fn (mut p Process) _signal_stop() {
$if windows {
p.win_stop_process()
} $else {
p.unix_stop_process()
}
}
// _signal_continue - should not be called directly, just by p.signal_continue
fn (mut p Process) _signal_continue() {
$if windows {
p.win_resume_process()
} $else {
p.unix_resume_process()
}
}
// _signal_kill - should not be called directly, except by p.signal_kill
fn (mut p Process) _signal_kill() {
$if windows {
p.win_kill_process()
} $else {
p.unix_kill_process()
}
}
2021-04-04 17:05:06 +03:00
// _signal_pgkill - should not be called directly, except by p.signal_pgkill
fn (mut p Process) _signal_pgkill() {
$if windows {
p.win_kill_pgroup()
} $else {
p.unix_kill_pgroup()
}
}
2020-11-16 19:32:50 +03:00
// _wait - should not be called directly, except by p.wait()
fn (mut p Process) _wait() {
$if windows {
p.win_wait()
} $else {
p.unix_wait()
}
}
// _is_alive - should not be called directly, except by p.is_alive()
fn (mut p Process) _is_alive() bool {
$if windows {
return p.win_is_alive()
} $else {
return p.unix_is_alive()
}
}