Function TestSSHServer_ClosesStdin

agent/agentssh/agentssh_test.go:407–491

(t *testing.T)

}

func TestSSHServer_ClosesStdin(t *testing.T) {
	t.Parallel()
	if runtime.GOOS == "windows" {
		t.Skip("bash doesn't exist on Windows")
	}

	ctx := testutil.Context(t, testutil.WaitMedium)
	logger := testutil.Logger(t)
	s, err := agentssh.NewServer(ctx, logger.Named("ssh-server"), prometheus.NewRegistry(), afero.NewMemMapFs(), agentexec.DefaultExecer, nil)
	require.NoError(t, err)
	logger = logger.Named("test")
	defer s.Close()
	err = s.UpdateHostSigner(42)
	assert.NoError(t, err)

	ln, err := net.Listen("tcp", "127.0.0.1:0")
	require.NoError(t, err)

	done := make(chan struct{})
	go func() {
		defer close(done)
		err := s.Serve(ln)
		assert.Error(t, err) // Server is closed.
	}()
	defer func() {
		err := s.Close()
		require.NoError(t, err)
		<-done
	}()

	c := sshClient(t, ln.Addr().String())

	sess, err := c.NewSession()
	require.NoError(t, err)
	stdout, err := sess.StdoutPipe()
	require.NoError(t, err)
	stdin, err := sess.StdinPipe()
	require.NoError(t, err)
	defer stdin.Close()

	dir := t.TempDir()
	err = os.MkdirAll(dir, 0o755)
	require.NoError(t, err)
	filePath := filepath.Join(dir, "result.txt")

	// the shell command `read` will block until data is written to stdin, or closed. It will return
	// exit code 1 if it hits EOF, which is what we want to test.
	cmdErrCh := make(chan error, 1)
	go func() {
		cmdErrCh <- sess.Start(fmt.Sprintf(`echo started; echo "read exit code: $(read && echo 0 || echo 1)" > %s`, filePath))
	}()

	cmdErr := testutil.RequireReceive(ctx, t, cmdErrCh)
	require.NoError(t, cmdErr)

	readCh := make(chan error, 1)
	go func() {
		buf := make([]byte, 8)