summaryrefslogtreecommitdiffstats
path: root/src/runtime/testdata/testprognet/signalexec.go
blob: 4a988ef6c1da2441b810cb3f526956fa2e71ef5d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
// Copyright 2017 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

// +build darwin dragonfly freebsd linux netbsd openbsd

// This is in testprognet instead of testprog because testprog
// must not import anything (like net, but also like os/signal)
// that kicks off background goroutines during init.

package main

import (
	"fmt"
	"os"
	"os/exec"
	"os/signal"
	"sync"
	"syscall"
	"time"
)

func init() {
	register("SignalDuringExec", SignalDuringExec)
	register("Nop", Nop)
}

func SignalDuringExec() {
	pgrp := syscall.Getpgrp()

	const tries = 10

	var wg sync.WaitGroup
	c := make(chan os.Signal, tries)
	signal.Notify(c, syscall.SIGWINCH)
	wg.Add(1)
	go func() {
		defer wg.Done()
		for range c {
		}
	}()

	for i := 0; i < tries; i++ {
		time.Sleep(time.Microsecond)
		wg.Add(2)
		go func() {
			defer wg.Done()
			cmd := exec.Command(os.Args[0], "Nop")
			cmd.Stdout = os.Stdout
			cmd.Stderr = os.Stderr
			if err := cmd.Run(); err != nil {
				fmt.Printf("Start failed: %v", err)
			}
		}()
		go func() {
			defer wg.Done()
			syscall.Kill(-pgrp, syscall.SIGWINCH)
		}()
	}

	signal.Stop(c)
	close(c)
	wg.Wait()

	fmt.Println("OK")
}

func Nop() {
	// This is just for SignalDuringExec.
}