1
0
mirror of https://github.com/golang/go synced 2024-11-25 22:07:58 -07:00

os, syscall: add ProcAttr type. Change StartProcess etc. to use it.

The Windows code is untested.

R=rsc, gri, brainman, rsc1
CC=golang-dev
https://golang.org/cl/4253052
This commit is contained in:
Roger Peppe 2011-03-15 14:41:19 -04:00 committed by Russ Cox
parent a34f1bbb22
commit aa55c05213
9 changed files with 115 additions and 54 deletions

View File

@ -32,7 +32,7 @@ func run(stdin []byte, argv []string) (stdout, stderr []byte, ok bool) {
if err != nil { if err != nil {
fatal("%s", err) fatal("%s", err)
} }
p, err := os.StartProcess(cmd, argv, os.Environ(), "", []*os.File{r0, w1, w2}) p, err := os.StartProcess(cmd, argv, &os.ProcAttr{Files: []*os.File{r0, w1, w2}})
if err != nil { if err != nil {
fatal("%s", err) fatal("%s", err)
} }

View File

@ -83,7 +83,7 @@ func exec(rw http.ResponseWriter, args []string) (status int) {
if *verbose { if *verbose {
log.Printf("executing %v", args) log.Printf("executing %v", args)
} }
p, err := os.StartProcess(bin, args, os.Environ(), *goroot, fds) p, err := os.StartProcess(bin, args, &os.ProcAttr{Files: fds, Dir: *goroot})
defer r.Close() defer r.Close()
w.Close() w.Close()
if err != nil { if err != nil {

View File

@ -1279,25 +1279,31 @@ func Attach(pid int) (Process, os.Error) {
return p, nil return p, nil
} }
// ForkExec forks the current process and execs argv0, stopping the // StartProcess forks the current process and execs argv0, stopping the
// new process after the exec syscall. See os.ForkExec for additional // new process after the exec syscall. See os.StartProcess for additional
// details. // details.
func ForkExec(argv0 string, argv []string, envv []string, dir string, fd []*os.File) (Process, os.Error) { func StartProcess(argv0 string, argv []string, attr *os.ProcAttr) (Process, os.Error) {
sysattr := &syscall.ProcAttr{
Dir: attr.Dir,
Env: attr.Env,
Ptrace: true,
}
p := newProcess(-1) p := newProcess(-1)
// Create array of integer (system) fds. // Create array of integer (system) fds.
intfd := make([]int, len(fd)) intfd := make([]int, len(attr.Files))
for i, f := range fd { for i, f := range attr.Files {
if f == nil { if f == nil {
intfd[i] = -1 intfd[i] = -1
} else { } else {
intfd[i] = f.Fd() intfd[i] = f.Fd()
} }
} }
sysattr.Files = intfd
// Fork from the monitor thread so we get the right tracer pid. // Fork from the monitor thread so we get the right tracer pid.
err := p.do(func() os.Error { err := p.do(func() os.Error {
pid, errno := syscall.PtraceForkExec(argv0, argv, envv, dir, intfd) pid, _, errno := syscall.StartProcess(argv0, argv, sysattr)
if errno != 0 { if errno != 0 {
return &os.PathError{"fork/exec", argv0, os.Errno(errno)} return &os.PathError{"fork/exec", argv0, os.Errno(errno)}
} }

View File

@ -105,7 +105,7 @@ func Run(name string, argv, envv []string, dir string, stdin, stdout, stderr int
} }
// Run command. // Run command.
c.Process, err = os.StartProcess(name, argv, envv, dir, fd[0:]) c.Process, err = os.StartProcess(name, argv, &os.ProcAttr{Dir: dir, Files: fd[:], Env: envv})
if err != nil { if err != nil {
goto Error goto Error
} }

View File

@ -99,7 +99,8 @@ func DateServer(rw http.ResponseWriter, req *http.Request) {
fmt.Fprintf(rw, "pipe: %s\n", err) fmt.Fprintf(rw, "pipe: %s\n", err)
return return
} }
p, err := os.StartProcess("/bin/date", []string{"date"}, os.Environ(), "", []*os.File{nil, w, w})
p, err := os.StartProcess("/bin/date", []string{"date"}, &os.ProcAttr{Files: []*os.File{nil, w, w}})
defer r.Close() defer r.Close()
w.Close() w.Close()
if err != nil { if err != nil {

View File

@ -21,27 +21,46 @@ func newProcess(pid, handle int) *Process {
return p return p
} }
// StartProcess starts a new process with the program, arguments, // ProcAttr holds the attributes that will be applied to a new process
// and environment specified by name, argv, and envv. The fd array specifies the // started by StartProcess.
// file descriptors to be set up in the new process: fd[0] will be Unix file type ProcAttr struct {
// descriptor 0 (standard input), fd[1] descriptor 1, and so on. A nil entry // If Dir is non-empty, the child changes into the directory before
// will cause the child to have no open file descriptor with that index. // creating the process.
// If dir is not empty, the child chdirs into the directory before execing the program. Dir string
func StartProcess(name string, argv []string, envv []string, dir string, fd []*File) (p *Process, err Error) { // If Env is non-nil, it gives the environment variables for the
if envv == nil { // new process in the form returned by Environ.
envv = Environ() // If it is nil, the result of Environ will be used.
Env []string
// Files specifies the open files inherited by the new process. The
// first three entries correspond to standard input, standard output, and
// standard error. An implementation may support additional entries,
// depending on the underlying operating system. A nil entry corresponds
// to that file being closed when the process starts.
Files []*File
}
// StartProcess starts a new process with the program, arguments and attributes
// specified by name, argv and attr.
func StartProcess(name string, argv []string, attr *ProcAttr) (p *Process, err Error) {
sysattr := &syscall.ProcAttr{
Dir: attr.Dir,
Env: attr.Env,
}
if sysattr.Env == nil {
sysattr.Env = Environ()
} }
// Create array of integer (system) fds. // Create array of integer (system) fds.
intfd := make([]int, len(fd)) intfd := make([]int, len(attr.Files))
for i, f := range fd { for i, f := range attr.Files {
if f == nil { if f == nil {
intfd[i] = -1 intfd[i] = -1
} else { } else {
intfd[i] = f.Fd() intfd[i] = f.Fd()
} }
} }
sysattr.Files = intfd
pid, h, e := syscall.StartProcess(name, argv, envv, dir, intfd) pid, h, e := syscall.StartProcess(name, argv, sysattr)
if e != 0 { if e != 0 {
return nil, &PathError{"fork/exec", name, Errno(e)} return nil, &PathError{"fork/exec", name, Errno(e)}
} }

View File

@ -409,25 +409,26 @@ func TestRename(t *testing.T) {
} }
} }
func TestForkExec(t *testing.T) { func TestStartProcess(t *testing.T) {
var cmd, adir, expect string var cmd, expect string
var args []string var args []string
r, w, err := Pipe() r, w, err := Pipe()
if err != nil { if err != nil {
t.Fatalf("Pipe: %v", err) t.Fatalf("Pipe: %v", err)
} }
attr := &ProcAttr{Files: []*File{nil, w, Stderr}}
if syscall.OS == "windows" { if syscall.OS == "windows" {
cmd = Getenv("COMSPEC") cmd = Getenv("COMSPEC")
args = []string{Getenv("COMSPEC"), "/c cd"} args = []string{Getenv("COMSPEC"), "/c cd"}
adir = Getenv("SystemRoot") attr.Dir = Getenv("SystemRoot")
expect = Getenv("SystemRoot") + "\r\n" expect = Getenv("SystemRoot") + "\r\n"
} else { } else {
cmd = "/bin/pwd" cmd = "/bin/pwd"
args = []string{"pwd"} args = []string{"pwd"}
adir = "/" attr.Dir = "/"
expect = "/\n" expect = "/\n"
} }
p, err := StartProcess(cmd, args, nil, adir, []*File{nil, w, Stderr}) p, err := StartProcess(cmd, args, attr)
if err != nil { if err != nil {
t.Fatalf("StartProcess: %v", err) t.Fatalf("StartProcess: %v", err)
} }
@ -751,7 +752,7 @@ func run(t *testing.T, cmd []string) string {
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
p, err := StartProcess("/bin/hostname", []string{"hostname"}, nil, "/", []*File{nil, w, Stderr}) p, err := StartProcess("/bin/hostname", []string{"hostname"}, &ProcAttr{Files: []*File{nil, w, Stderr}})
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }

View File

@ -96,13 +96,16 @@ func SetNonblock(fd int, nonblocking bool) (errno int) {
// no rescheduling, no malloc calls, and no new stack segments. // no rescheduling, no malloc calls, and no new stack segments.
// The calls to RawSyscall are okay because they are assembly // The calls to RawSyscall are okay because they are assembly
// functions that do not grow the stack. // functions that do not grow the stack.
func forkAndExecInChild(argv0 *byte, argv []*byte, envv []*byte, traceme bool, dir *byte, fd []int, pipe int) (pid int, err int) { func forkAndExecInChild(argv0 *byte, argv, envv []*byte, dir *byte, attr *ProcAttr, pipe int) (pid int, err int) {
// Declare all variables at top in case any // Declare all variables at top in case any
// declarations require heap allocation (e.g., err1). // declarations require heap allocation (e.g., err1).
var r1, r2, err1 uintptr var r1, r2, err1 uintptr
var nextfd int var nextfd int
var i int var i int
// guard against side effects of shuffling fds below.
fd := append([]int(nil), attr.Files...)
darwin := OS == "darwin" darwin := OS == "darwin"
// About to call fork. // About to call fork.
@ -128,13 +131,21 @@ func forkAndExecInChild(argv0 *byte, argv []*byte, envv []*byte, traceme bool, d
// Fork succeeded, now in child. // Fork succeeded, now in child.
// Enable tracing if requested. // Enable tracing if requested.
if traceme { if attr.Ptrace {
_, _, err1 = RawSyscall(SYS_PTRACE, uintptr(PTRACE_TRACEME), 0, 0) _, _, err1 = RawSyscall(SYS_PTRACE, uintptr(PTRACE_TRACEME), 0, 0)
if err1 != 0 { if err1 != 0 {
goto childerror goto childerror
} }
} }
// Session ID
if attr.Setsid {
_, _, err1 = RawSyscall(SYS_SETSID, 0, 0, 0)
if err1 != 0 {
goto childerror
}
}
// Chdir // Chdir
if dir != nil { if dir != nil {
_, _, err1 = RawSyscall(SYS_CHDIR, uintptr(unsafe.Pointer(dir)), 0, 0) _, _, err1 = RawSyscall(SYS_CHDIR, uintptr(unsafe.Pointer(dir)), 0, 0)
@ -220,28 +231,44 @@ childerror:
panic("unreached") panic("unreached")
} }
func forkExec(argv0 string, argv []string, envv []string, traceme bool, dir string, fd []int) (pid int, err int) {
type ProcAttr struct {
Setsid bool // Create session.
Ptrace bool // Enable tracing.
Dir string // Current working directory.
Env []string // Environment.
Files []int // File descriptors.
}
var zeroAttributes ProcAttr
func forkExec(argv0 string, argv []string, attr *ProcAttr) (pid int, err int) {
var p [2]int var p [2]int
var n int var n int
var err1 uintptr var err1 uintptr
var wstatus WaitStatus var wstatus WaitStatus
if attr == nil {
attr = &zeroAttributes
}
p[0] = -1 p[0] = -1
p[1] = -1 p[1] = -1
// Convert args to C form. // Convert args to C form.
argv0p := StringBytePtr(argv0) argv0p := StringBytePtr(argv0)
argvp := StringArrayPtr(argv) argvp := StringArrayPtr(argv)
envvp := StringArrayPtr(envv) envvp := StringArrayPtr(attr.Env)
var dirp *byte
if len(dir) > 0 {
dirp = StringBytePtr(dir)
}
if OS == "freebsd" && len(argv[0]) > len(argv0) { if OS == "freebsd" && len(argv[0]) > len(argv0) {
argvp[0] = argv0p argvp[0] = argv0p
} }
var dir *byte
if attr.Dir != "" {
dir = StringBytePtr(attr.Dir)
}
// Acquire the fork lock so that no other threads // Acquire the fork lock so that no other threads
// create new fds that are not yet close-on-exec // create new fds that are not yet close-on-exec
// before we fork. // before we fork.
@ -259,7 +286,7 @@ func forkExec(argv0 string, argv []string, envv []string, traceme bool, dir stri
} }
// Kick off child. // Kick off child.
pid, err = forkAndExecInChild(argv0p, argvp, envvp, traceme, dirp, fd, p[1]) pid, err = forkAndExecInChild(argv0p, argvp, envvp, dir, attr, p[1])
if err != 0 { if err != 0 {
error: error:
if p[0] >= 0 { if p[0] >= 0 {
@ -297,13 +324,14 @@ func forkExec(argv0 string, argv []string, envv []string, traceme bool, dir stri
} }
// Combination of fork and exec, careful to be thread safe. // Combination of fork and exec, careful to be thread safe.
func ForkExec(argv0 string, argv []string, envv []string, dir string, fd []int) (pid int, err int) { func ForkExec(argv0 string, argv []string, attr *ProcAttr) (pid int, err int) {
return forkExec(argv0, argv, envv, false, dir, fd) return forkExec(argv0, argv, attr)
} }
// PtraceForkExec is like ForkExec, but starts the child in a traced state. // StartProcess wraps ForkExec for package os.
func PtraceForkExec(argv0 string, argv []string, envv []string, dir string, fd []int) (pid int, err int) { func StartProcess(argv0 string, argv []string, attr *ProcAttr) (pid, handle int, err int) {
return forkExec(argv0, argv, envv, true, dir, fd) pid, err = forkExec(argv0, argv, attr)
return pid, 0, err
} }
// Ordinary exec. // Ordinary exec.
@ -314,9 +342,3 @@ func Exec(argv0 string, argv []string, envv []string) (err int) {
uintptr(unsafe.Pointer(&StringArrayPtr(envv)[0]))) uintptr(unsafe.Pointer(&StringArrayPtr(envv)[0])))
return int(err1) return int(err1)
} }
// StartProcess wraps ForkExec for package os.
func StartProcess(argv0 string, argv []string, envv []string, dir string, fd []int) (pid, handle int, err int) {
pid, err = forkExec(argv0, argv, envv, false, dir, fd)
return pid, 0, err
}

View File

@ -114,21 +114,33 @@ func SetNonblock(fd int, nonblocking bool) (errno int) {
return 0 return 0
} }
type ProcAttr struct {
Dir string
Env []string
Files []int
}
var zeroAttributes ProcAttr
// TODO(kardia): Add trace // TODO(kardia): Add trace
//The command and arguments are passed via the Command line parameter. //The command and arguments are passed via the Command line parameter.
func StartProcess(argv0 string, argv []string, envv []string, dir string, fd []int) (pid, handle int, err int) { func StartProcess(argv0 string, argv []string, attr *ProcAttr) (pid, handle int, err int) {
if len(fd) > 3 { if attr == nil {
attr = &zeroAttributes
}
if len(attr.Files) > 3 {
return 0, 0, EWINDOWS return 0, 0, EWINDOWS
} }
// CreateProcess will throw an error if the dir is not set to a valid dir // CreateProcess will throw an error if the dir is not set to a valid dir
// thus get the working dir if dir is empty. // thus get the working dir if dir is empty.
if len(dir) == 0 { dir := attr.Dir
if dir == "" {
if wd, ok := Getwd(); ok == 0 { if wd, ok := Getwd(); ok == 0 {
dir = wd dir = wd
} }
} }
fd := attr.Files
startupInfo := new(StartupInfo) startupInfo := new(StartupInfo)
processInfo := new(ProcessInformation) processInfo := new(ProcessInformation)
@ -180,7 +192,7 @@ func StartProcess(argv0 string, argv []string, envv []string, dir string, fd []i
nil, //ptr to struct lpThreadAttributes nil, //ptr to struct lpThreadAttributes
true, //bInheritHandles true, //bInheritHandles
CREATE_UNICODE_ENVIRONMENT, //Flags CREATE_UNICODE_ENVIRONMENT, //Flags
createEnvBlock(envv), //env block, NULL uses parent env createEnvBlock(attr.Env), //env block, NULL uses parent env
StringToUTF16Ptr(dir), StringToUTF16Ptr(dir),
startupInfo, startupInfo,
processInfo) processInfo)