| 798215af |
// +build !windows
|
| 28cf8fdd |
package main
import ( |
| ae0883ce |
"bufio" |
| 28cf8fdd |
"os/exec"
"strings"
"time"
|
| 28cda048 |
"github.com/docker/docker/pkg/stringid" |
| dc944ea7 |
"github.com/go-check/check" |
| 28cf8fdd |
"github.com/kr/pty"
)
// #9860 |
| dc944ea7 |
func (s *DockerSuite) TestAttachClosedOnContainerStop(c *check.C) { |
| 28cf8fdd |
cmd := exec.Command(dockerBinary, "run", "-dti", "busybox", "sleep", "2")
out, _, err := runCommandWithOutput(cmd)
if err != nil { |
| dc944ea7 |
c.Fatalf("failed to start container: %v (%v)", out, err) |
| 28cf8fdd |
}
|
| 475c6531 |
id := strings.TrimSpace(out) |
| 28cf8fdd |
if err := waitRun(id); err != nil { |
| dc944ea7 |
c.Fatal(err) |
| 28cf8fdd |
}
|
| 4203230c |
errChan := make(chan error) |
| 28cf8fdd |
go func() { |
| 4203230c |
defer close(errChan) |
| 28cf8fdd |
_, tty, err := pty.Open()
if err != nil { |
| 4203230c |
errChan <- err
return |
| 28cf8fdd |
}
attachCmd := exec.Command(dockerBinary, "attach", id)
attachCmd.Stdin = tty
attachCmd.Stdout = tty
attachCmd.Stderr = tty
if err := attachCmd.Run(); err != nil { |
| 4203230c |
errChan <- err
return |
| 28cf8fdd |
}
}()
waitCmd := exec.Command(dockerBinary, "wait", id)
if out, _, err = runCommandWithOutput(waitCmd); err != nil { |
| dc944ea7 |
c.Fatalf("error thrown while waiting for container: %s, %v", out, err) |
| 28cf8fdd |
}
select { |
| 4203230c |
case err := <-errChan:
c.Assert(err, check.IsNil) |
| 28cf8fdd |
case <-time.After(attachWait): |
| dc944ea7 |
c.Fatal("timed out without attach returning") |
| 28cf8fdd |
}
}
|
| dc944ea7 |
func (s *DockerSuite) TestAttachAfterDetach(c *check.C) { |
| 28cf8fdd |
name := "detachtest"
cpty, tty, err := pty.Open()
if err != nil { |
| dc944ea7 |
c.Fatalf("Could not open pty: %v", err) |
| 28cf8fdd |
}
cmd := exec.Command(dockerBinary, "run", "-ti", "--name", name, "busybox")
cmd.Stdin = tty
cmd.Stdout = tty
cmd.Stderr = tty
|
| 4203230c |
errChan := make(chan error) |
| 28cf8fdd |
go func() { |
| 4203230c |
errChan <- cmd.Run()
close(errChan) |
| 28cf8fdd |
}()
time.Sleep(500 * time.Millisecond) |
| a124bfae |
if err := waitRun(name); err != nil { |
| dc944ea7 |
c.Fatal(err) |
| a124bfae |
} |
| 28cf8fdd |
cpty.Write([]byte{16})
time.Sleep(100 * time.Millisecond)
cpty.Write([]byte{17})
|
| 4203230c |
select {
case err := <-errChan:
c.Assert(err, check.IsNil)
case <-time.After(5 * time.Second):
c.Fatal("timeout while detaching")
} |
| 28cf8fdd |
cpty, tty, err = pty.Open()
if err != nil { |
| dc944ea7 |
c.Fatalf("Could not open pty: %v", err) |
| 28cf8fdd |
}
cmd = exec.Command(dockerBinary, "attach", name)
cmd.Stdin = tty
cmd.Stdout = tty
cmd.Stderr = tty
|
| 6ef80577 |
if err := cmd.Start(); err != nil { |
| dc944ea7 |
c.Fatal(err) |
| 6ef80577 |
} |
| 28cf8fdd |
bytes := make([]byte, 10) |
| 6ef80577 |
var nBytes int
readErr := make(chan error, 1) |
| 28cf8fdd |
|
| 6ef80577 |
go func() {
time.Sleep(500 * time.Millisecond)
cpty.Write([]byte("\n"))
time.Sleep(500 * time.Millisecond) |
| 28cf8fdd |
|
| 6ef80577 |
nBytes, err = cpty.Read(bytes)
cpty.Close()
readErr <- err
}()
select {
case err := <-readErr: |
| 4203230c |
c.Assert(err, check.IsNil) |
| 6ef80577 |
case <-time.After(2 * time.Second): |
| dc944ea7 |
c.Fatal("timeout waiting for attach read") |
| 28cf8fdd |
}
|
| 6ef80577 |
if err := cmd.Wait(); err != nil { |
| dc944ea7 |
c.Fatal(err) |
| 28cf8fdd |
}
|
| 6ef80577 |
if !strings.Contains(string(bytes[:nBytes]), "/ #") { |
| dc944ea7 |
c.Fatalf("failed to get a new prompt. got %s", string(bytes[:nBytes])) |
| 6ef80577 |
} |
| 28cf8fdd |
} |
| ae0883ce |
// TestAttachDetach checks that attach in tty mode can be detached using the long container ID |
| dc944ea7 |
func (s *DockerSuite) TestAttachDetach(c *check.C) {
out, _ := dockerCmd(c, "run", "-itd", "busybox", "cat") |
| ae0883ce |
id := strings.TrimSpace(out)
if err := waitRun(id); err != nil { |
| dc944ea7 |
c.Fatal(err) |
| ae0883ce |
}
cpty, tty, err := pty.Open()
if err != nil { |
| dc944ea7 |
c.Fatal(err) |
| ae0883ce |
}
defer cpty.Close()
cmd := exec.Command(dockerBinary, "attach", id)
cmd.Stdin = tty
stdout, err := cmd.StdoutPipe()
if err != nil { |
| dc944ea7 |
c.Fatal(err) |
| ae0883ce |
}
defer stdout.Close()
if err := cmd.Start(); err != nil { |
| dc944ea7 |
c.Fatal(err) |
| ae0883ce |
}
if err := waitRun(id); err != nil { |
| dc944ea7 |
c.Fatalf("error waiting for container to start: %v", err) |
| ae0883ce |
}
if _, err := cpty.Write([]byte("hello\n")); err != nil { |
| dc944ea7 |
c.Fatal(err) |
| ae0883ce |
}
out, err = bufio.NewReader(stdout).ReadString('\n')
if err != nil { |
| dc944ea7 |
c.Fatal(err) |
| ae0883ce |
}
if strings.TrimSpace(out) != "hello" { |
| 3941623f |
c.Fatalf("expected 'hello', got %q", out) |
| ae0883ce |
}
// escape sequence
if _, err := cpty.Write([]byte{16}); err != nil { |
| dc944ea7 |
c.Fatal(err) |
| ae0883ce |
}
time.Sleep(100 * time.Millisecond)
if _, err := cpty.Write([]byte{17}); err != nil { |
| dc944ea7 |
c.Fatal(err) |
| ae0883ce |
}
ch := make(chan struct{})
go func() {
cmd.Wait()
ch <- struct{}{}
}()
running, err := inspectField(id, "State.Running")
if err != nil { |
| dc944ea7 |
c.Fatal(err) |
| ae0883ce |
}
if running != "true" { |
| 3941623f |
c.Fatal("expected container to still be running") |
| ae0883ce |
}
go func() { |
| dc944ea7 |
dockerCmd(c, "kill", id) |
| ae0883ce |
}()
select {
case <-ch:
case <-time.After(10 * time.Millisecond): |
| dc944ea7 |
c.Fatal("timed out waiting for container to exit") |
| ae0883ce |
}
} |
| 28cda048 |
// TestAttachDetachTruncatedID checks that attach in tty mode can be detached |
| dc944ea7 |
func (s *DockerSuite) TestAttachDetachTruncatedID(c *check.C) {
out, _ := dockerCmd(c, "run", "-itd", "busybox", "cat") |
| 28cda048 |
id := stringid.TruncateID(strings.TrimSpace(out))
if err := waitRun(id); err != nil { |
| dc944ea7 |
c.Fatal(err) |
| 28cda048 |
}
cpty, tty, err := pty.Open()
if err != nil { |
| dc944ea7 |
c.Fatal(err) |
| 28cda048 |
}
defer cpty.Close()
cmd := exec.Command(dockerBinary, "attach", id)
cmd.Stdin = tty
stdout, err := cmd.StdoutPipe()
if err != nil { |
| dc944ea7 |
c.Fatal(err) |
| 28cda048 |
}
defer stdout.Close()
if err := cmd.Start(); err != nil { |
| dc944ea7 |
c.Fatal(err) |
| 28cda048 |
}
if _, err := cpty.Write([]byte("hello\n")); err != nil { |
| dc944ea7 |
c.Fatal(err) |
| 28cda048 |
}
out, err = bufio.NewReader(stdout).ReadString('\n')
if err != nil { |
| dc944ea7 |
c.Fatal(err) |
| 28cda048 |
}
if strings.TrimSpace(out) != "hello" { |
| 3941623f |
c.Fatalf("expected 'hello', got %q", out) |
| 28cda048 |
}
// escape sequence
if _, err := cpty.Write([]byte{16}); err != nil { |
| dc944ea7 |
c.Fatal(err) |
| 28cda048 |
}
time.Sleep(100 * time.Millisecond)
if _, err := cpty.Write([]byte{17}); err != nil { |
| dc944ea7 |
c.Fatal(err) |
| 28cda048 |
}
ch := make(chan struct{})
go func() {
cmd.Wait()
ch <- struct{}{}
}()
running, err := inspectField(id, "State.Running")
if err != nil { |
| dc944ea7 |
c.Fatal(err) |
| 28cda048 |
}
if running != "true" { |
| 3941623f |
c.Fatal("expected container to still be running") |
| 28cda048 |
}
go func() { |
| dc944ea7 |
dockerCmd(c, "kill", id) |
| 28cda048 |
}()
select {
case <-ch:
case <-time.After(10 * time.Millisecond): |
| dc944ea7 |
c.Fatal("timed out waiting for container to exit") |
| 28cda048 |
}
} |