syscall: restore signal mask after setting foreground process group

Fixes #37217

Change-Id: I0151bb77fc4c4552d1b19c31d784943b72f84b80
Reviewed-on: https://go-review.googlesource.com/c/go/+/313653
Trust: Ian Lance Taylor <iant@golang.org>
Run-TryBot: Ian Lance Taylor <iant@golang.org>
TryBot-Result: Go Bot <gobot@golang.org>
Reviewed-by: Tobias Klauser <tobias.klauser@gmail.com>
This commit is contained in:
Ian Lance Taylor 2021-04-26 17:27:58 -07:00
parent 39844971fb
commit 7d22c2181b
5 changed files with 56 additions and 9 deletions

View file

@ -167,11 +167,13 @@ func TestPgid(t *testing.T) {
func TestForeground(t *testing.T) {
signal.Ignore(syscall.SIGTTIN, syscall.SIGTTOU)
defer signal.Reset()
tty, err := os.OpenFile("/dev/tty", os.O_RDWR, 0)
if err != nil {
t.Skipf("Can't test Foreground. Couldn't open /dev/tty: %s", err)
}
defer tty.Close()
// This should really be pid_t, however _C_int (aka int32) is generally
// equivalent.
@ -216,8 +218,45 @@ func TestForeground(t *testing.T) {
if errno != 0 {
t.Fatalf("TIOCSPGRP failed with error code: %s", errno)
}
}
signal.Reset()
func TestForegroundSignal(t *testing.T) {
tty, err := os.OpenFile("/dev/tty", os.O_RDWR, 0)
if err != nil {
t.Skipf("couldn't open /dev/tty: %s", err)
}
defer tty.Close()
ch1 := make(chan os.Signal, 1)
ch2 := make(chan bool)
signal.Notify(ch1, syscall.SIGTTIN, syscall.SIGTTOU)
defer signal.Stop(ch1)
go func() {
cmd := create(t)
cmd.proc.SysProcAttr = &syscall.SysProcAttr{
Ctty: int(tty.Fd()),
Foreground: true,
}
cmd.Start()
cmd.Stop()
close(ch2)
}()
timer := time.NewTimer(30 * time.Second)
defer timer.Stop()
for {
select {
case sig := <-ch1:
t.Errorf("unexpected signal %v", sig)
case <-ch2:
// Success.
return
case <-timer.C:
t.Fatal("timed out waiting for child process")
}
}
}
// Test a couple of cases that SysProcAttr can't handle. Issue 29458.