libgo/go/syscall/exec_unix_test.go

   1 // Copyright 2015 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.
   4
   5 // +build aix darwin dragonfly freebsd hurd linux netbsd openbsd solaris
   6
   7 package syscall_test
   8
   9 import (
  10         "internal/testenv"
  11         "io"
  12         "os"
  13         "os/exec"
  14         "os/signal"
  15         "runtime"
  16         "syscall"
  17         "testing"
  18         "unsafe"
  19 )
  20
  21 type command struct {
  22         pipe io.WriteCloser
  23         proc *exec.Cmd
  24         test *testing.T
  25 }
  26
  27 func (c *command) Info() (pid, pgrp int) {
  28         pid = c.proc.Process.Pid
  29
  30         pgrp, err := syscall.Getpgid(pid)
  31         if err != nil {
  32                 c.test.Fatal(err)
  33         }
  34
  35         return
  36 }
  37
  38 func (c *command) Start() {
  39         if err := c.proc.Start(); err != nil {
  40                 c.test.Fatal(err)
  41         }
  42 }
  43
  44 func (c *command) Stop() {
  45         c.pipe.Close()
  46         if err := c.proc.Wait(); err != nil {
  47                 c.test.Fatal(err)
  48         }
  49 }
  50
  51 func create(t *testing.T) *command {
  52         testenv.MustHaveExec(t)
  53
  54         proc := exec.Command("cat")
  55         stdin, err := proc.StdinPipe()
  56         if err != nil {
  57                 t.Fatal(err)
  58         }
  59
  60         return &command{stdin, proc, t}
  61 }
  62
  63 func parent() (pid, pgrp int) {
  64         return syscall.Getpid(), syscall.Getpgrp()
  65 }
  66
  67 func TestZeroSysProcAttr(t *testing.T) {
  68         ppid, ppgrp := parent()
  69
  70         cmd := create(t)
  71
  72         cmd.Start()
  73         defer cmd.Stop()
  74
  75         cpid, cpgrp := cmd.Info()
  76
  77         if cpid == ppid {
  78                 t.Fatalf("Parent and child have the same process ID")
  79         }
  80
  81         if cpgrp != ppgrp {
  82                 t.Fatalf("Child is not in parent's process group")
  83         }
  84 }
  85
  86 func TestSetpgid(t *testing.T) {
  87         ppid, ppgrp := parent()
  88
  89         cmd := create(t)
  90
  91         cmd.proc.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
  92         cmd.Start()
  93         defer cmd.Stop()
  94
  95         cpid, cpgrp := cmd.Info()
  96
  97         if cpid == ppid {
  98                 t.Fatalf("Parent and child have the same process ID")
  99         }
 100
 101         if cpgrp == ppgrp {
 102                 t.Fatalf("Parent and child are in the same process group")
 103         }
 104
 105         if cpid != cpgrp {
 106                 t.Fatalf("Child's process group is not the child's process ID")
 107         }
 108 }
 109
 110 func TestPgid(t *testing.T) {
 111         ppid, ppgrp := parent()
 112
 113         cmd1 := create(t)
 114
 115         cmd1.proc.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
 116         cmd1.Start()
 117         defer cmd1.Stop()
 118
 119         cpid1, cpgrp1 := cmd1.Info()
 120
 121         if cpid1 == ppid {
 122                 t.Fatalf("Parent and child 1 have the same process ID")
 123         }
 124
 125         if cpgrp1 == ppgrp {
 126                 t.Fatalf("Parent and child 1 are in the same process group")
 127         }
 128
 129         if cpid1 != cpgrp1 {
 130                 t.Fatalf("Child 1's process group is not its process ID")
 131         }
 132
 133         cmd2 := create(t)
 134
 135         cmd2.proc.SysProcAttr = &syscall.SysProcAttr{
 136                 Setpgid: true,
 137                 Pgid:    cpgrp1,
 138         }
 139         cmd2.Start()
 140         defer cmd2.Stop()
 141
 142         cpid2, cpgrp2 := cmd2.Info()
 143
 144         if cpid2 == ppid {
 145                 t.Fatalf("Parent and child 2 have the same process ID")
 146         }
 147
 148         if cpgrp2 == ppgrp {
 149                 t.Fatalf("Parent and child 2 are in the same process group")
 150         }
 151
 152         if cpid2 == cpgrp2 {
 153                 t.Fatalf("Child 2's process group is its process ID")
 154         }
 155
 156         if cpid1 == cpid2 {
 157                 t.Fatalf("Child 1 and 2 have the same process ID")
 158         }
 159
 160         if cpgrp1 != cpgrp2 {
 161                 t.Fatalf("Child 1 and 2 are not in the same process group")
 162         }
 163 }
 164
 165 func TestForeground(t *testing.T) {
 166         if runtime.GOOS == "hurd" {
 167                 t.Skip("skipping; TestForeground: fails on GNU/Hurd")
 168         }
 169         signal.Ignore(syscall.SIGTTIN, syscall.SIGTTOU)
 170
 171         tty, err := os.OpenFile("/dev/tty", os.O_RDWR, 0)
 172         if err != nil {
 173                 t.Skipf("Can't test Foreground. Couldn't open /dev/tty: %s", err)
 174         }
 175
 176         fpgrp := syscall.Pid_t(0)
 177
 178         errno := syscall.Ioctl(tty.Fd(), syscall.TIOCGPGRP, uintptr(unsafe.Pointer(&fpgrp)))
 179         if errno != 0 {
 180                 t.Fatalf("TIOCGPGRP failed with error code: %s", errno)
 181         }
 182
 183         if fpgrp == 0 {
 184                 t.Fatalf("Foreground process group is zero")
 185         }
 186
 187         ppid, ppgrp := parent()
 188
 189         cmd := create(t)
 190
 191         cmd.proc.SysProcAttr = &syscall.SysProcAttr{
 192                 Ctty:       int(tty.Fd()),
 193                 Foreground: true,
 194         }
 195         cmd.Start()
 196
 197         cpid, cpgrp := cmd.Info()
 198
 199         if cpid == ppid {
 200                 t.Fatalf("Parent and child have the same process ID")
 201         }
 202
 203         if cpgrp == ppgrp {
 204                 t.Fatalf("Parent and child are in the same process group")
 205         }
 206
 207         if cpid != cpgrp {
 208                 t.Fatalf("Child's process group is not the child's process ID")
 209         }
 210
 211         cmd.Stop()
 212
 213         errno = syscall.Ioctl(tty.Fd(), syscall.TIOCSPGRP, uintptr(unsafe.Pointer(&fpgrp)))
 214         if errno != 0 {
 215                 t.Fatalf("TIOCSPGRP failed with error code: %s", errno)
 216         }
 217
 218         signal.Reset()
 219 }