142 lines
2.7 KiB
Go
142 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"os/exec"
|
|
"path"
|
|
"time"
|
|
)
|
|
|
|
var DOCKER_PATH string = path.Join(os.Getenv("DOCKERPATH"), "docker")
|
|
|
|
// WARNING: this crashTest will 1) crash your host, 2) remove all containers
|
|
func runDaemon() (*exec.Cmd, error) {
|
|
os.Remove("/var/run/docker.pid")
|
|
exec.Command("rm", "-rf", "/var/lib/docker/containers")
|
|
cmd := exec.Command(DOCKER_PATH, "-d")
|
|
outPipe, err := cmd.StdoutPipe()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
errPipe, err := cmd.StderrPipe()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := cmd.Start(); err != nil {
|
|
return nil, err
|
|
}
|
|
go func() {
|
|
io.Copy(os.Stdout, outPipe)
|
|
}()
|
|
go func() {
|
|
io.Copy(os.Stderr, errPipe)
|
|
}()
|
|
return cmd, nil
|
|
}
|
|
|
|
func crashTest() error {
|
|
if err := exec.Command("/bin/bash", "-c", "while true; do true; done").Start(); err != nil {
|
|
return err
|
|
}
|
|
|
|
var endpoint string
|
|
if ep := os.Getenv("TEST_ENDPOINT"); ep == "" {
|
|
endpoint = "192.168.56.1:7979"
|
|
} else {
|
|
endpoint = ep
|
|
}
|
|
|
|
c := make(chan bool)
|
|
var conn io.Writer
|
|
|
|
go func() {
|
|
conn, _ = net.Dial("tcp", endpoint)
|
|
c <- false
|
|
}()
|
|
go func() {
|
|
time.Sleep(2 * time.Second)
|
|
c <- true
|
|
}()
|
|
<-c
|
|
|
|
restartCount := 0
|
|
totalTestCount := 1
|
|
for {
|
|
daemon, err := runDaemon()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
restartCount++
|
|
// time.Sleep(5000 * time.Millisecond)
|
|
var stop bool
|
|
go func() error {
|
|
stop = false
|
|
for i := 0; i < 100 && !stop; {
|
|
func() error {
|
|
if conn != nil {
|
|
fmt.Fprintf(conn, "%d\n", totalTestCount)
|
|
}
|
|
cmd := exec.Command(DOCKER_PATH, "run", "base", "echo", "hello", "world")
|
|
outPipe, err := cmd.StdoutPipe()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
inPipe, err := cmd.StdinPipe()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := cmd.Start(); err != nil {
|
|
return err
|
|
}
|
|
go func() {
|
|
io.Copy(os.Stdout, outPipe)
|
|
}()
|
|
// Expecting error, do not check
|
|
inPipe.Write([]byte("hello world!!!!!\n"))
|
|
go inPipe.Write([]byte("hello world!!!!!\n"))
|
|
go inPipe.Write([]byte("hello world!!!!!\n"))
|
|
inPipe.Close()
|
|
|
|
go func() error {
|
|
r := bufio.NewReader(outPipe)
|
|
if out, err := r.ReadString('\n'); err != nil {
|
|
return err
|
|
} else if out == "hello world\n" {
|
|
log.Printf("%d", i)
|
|
if conn != nil {
|
|
fmt.Fprintf(conn, "%d\n", totalTestCount)
|
|
}
|
|
i++
|
|
totalTestCount++
|
|
}
|
|
return nil
|
|
}()
|
|
if err := cmd.Wait(); err != nil {
|
|
return err
|
|
}
|
|
outPipe.Close()
|
|
return nil
|
|
}()
|
|
}
|
|
return nil
|
|
}()
|
|
time.Sleep(20 * time.Second)
|
|
stop = true
|
|
if err := daemon.Process.Kill(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func main() {
|
|
if err := crashTest(); err != nil {
|
|
log.Println(err)
|
|
}
|
|
}
|