1 // Copyright 2011 The Go Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style
3 // license that can be found in the LICENSE file.
5 // +build aix darwin dragonfly freebsd netbsd openbsd solaris
13 type SysProcAttr
struct {
14 Chroot
string // Chroot.
15 Credential
*Credential
// Credential.
16 Ptrace
bool // Enable tracing.
17 Setsid
bool // Create session.
18 Setpgid
bool // Set process group ID to Pgid, or, if Pgid == 0, to new pid.
19 Setctty
bool // Set controlling terminal to fd Ctty
20 Noctty
bool // Detach fd 0 from controlling terminal
21 Ctty
int // Controlling TTY fd
22 Foreground
bool // Place child's process group in foreground. (Implies Setpgid. Uses Ctty as fd of controlling TTY)
23 Pgid
int // Child's process group ID if Setpgid.
26 // Implemented in runtime package.
27 func runtime_BeforeFork()
28 func runtime_AfterFork()
30 // Fork, dup fd onto 0..len(fd), and exec(argv0, argvv, envv) in child.
31 // If a dup or exec fails, write the errno error to pipe.
32 // (Pipe is close-on-exec so if exec succeeds, it will be closed.)
33 // In the child, this function must not acquire any locks, because
34 // they might have been locked at the time of the fork. This means
35 // no rescheduling, no malloc calls, and no new stack segments.
36 // For the same reason compiler does not race instrument it.
37 // The calls to RawSyscall are okay because they are assembly
38 // functions that do not grow the stack.
40 func forkAndExecInChild(argv0
*byte, argv
, envv
[]*byte, chroot
, dir
*byte, attr
*ProcAttr
, sys
*SysProcAttr
, pipe
int) (pid
int, err Errno
) {
41 // Declare all variables at top in case any
42 // declarations require heap allocation (e.g., err1).
50 // guard against side effects of shuffling fds below.
51 // Make sure that nextfd is beyond any currently open files so
52 // that we can't run the risk of overwriting any of them.
53 fd
:= make([]int, len(attr
.Files
))
54 nextfd
= len(attr
.Files
)
55 for i
, ufd
:= range attr
.Files
{
56 if nextfd
< int(ufd
) {
63 // About to call fork.
64 // No more allocation or calls of non-assembly functions.
78 // Fork succeeded, now in child.
80 // Enable tracing if requested.
82 err1
= raw_ptrace(_PTRACE_TRACEME
, 0, nil, nil)
97 if sys
.Setpgid || sys
.Foreground
{
98 // Place child in process group.
99 err1
= raw_setpgid(0, sys
.Pgid
)
106 pgrp
:= Pid_t(sys
.Pgid
)
111 // Place process group in foreground.
112 _
, err1
= raw_ioctl_ptr(sys
.Ctty
, TIOCSPGRP
, unsafe
.Pointer(&pgrp
))
120 err1
= raw_chroot(chroot
)
127 if cred
:= sys
.Credential
; cred
!= nil {
128 ngroups
:= len(cred
.Groups
)
130 err2
:= setgroups(0, nil)
137 groups
:= make([]Gid_t
, ngroups
)
138 for i
, v
:= range cred
.Groups
{
141 err2
:= setgroups(ngroups
, &groups
[0])
151 err2
:= Setgid(int(cred
.Gid
))
156 err2
= Setuid(int(cred
.Uid
))
165 err1
= raw_chdir(dir
)
171 // Pass 1: look for fd[i] < i and move those up above len(fd)
172 // so that pass 2 won't stomp on an fd it needs later.
174 err1
= raw_dup2(pipe
, nextfd
)
178 raw_fcntl(nextfd
, F_SETFD
, FD_CLOEXEC
)
182 for i
= 0; i
< len(fd
); i
++ {
183 if fd
[i
] >= 0 && fd
[i
] < int(i
) {
184 if nextfd
== pipe
{ // don't stomp on pipe
187 err1
= raw_dup2(fd
[i
], nextfd
)
191 raw_fcntl(nextfd
, F_SETFD
, FD_CLOEXEC
)
197 // Pass 2: dup fd[i] down onto i.
198 for i
= 0; i
< len(fd
); i
++ {
204 // dup2(i, i) won't clear close-on-exec flag on Linux,
205 // probably not elsewhere either.
206 _
, err1
= raw_fcntl(fd
[i
], F_SETFD
, 0)
212 // The new fd is created NOT close-on-exec,
213 // which is exactly what we want.
214 err1
= raw_dup2(fd
[i
], i
)
220 // By convention, we don't close-on-exec the fds we are
221 // started with, so if len(fd) < 3, close 0, 1, 2 as needed.
222 // Programs that know they inherit fds >= 3 will need
223 // to set them close-on-exec.
224 for i
= len(fd
); i
< 3; i
++ {
228 // Detach fd 0 from tty
230 _
, err1
= raw_ioctl(0, TIOCNOTTY
, 0)
236 // Set the controlling TTY to Ctty
242 _
, err1
= raw_ioctl(sys
.Ctty
, TIOCSCTTY
, 0)
249 err1
= raw_execve(argv0
, &argv
[0], &envv
[0])
252 // send error code on pipe
253 raw_write(pipe
, (*byte)(unsafe
.Pointer(&err1
)), int(unsafe
.Sizeof(err1
)))
259 // Try to open a pipe with O_CLOEXEC set on both file descriptors.
260 func forkExecPipe(p
[]int) error
{
265 _
, err
= fcntl(p
[0], F_SETFD
, FD_CLOEXEC
)
269 _
, err
= fcntl(p
[1], F_SETFD
, FD_CLOEXEC
)