egress-proxies-operator/internal/provider/kvm/conn_test.go

package kvm

import (
	"bytes"
	"context"
	"crypto/ed25519"
	"crypto/rand"
	"encoding/pem"
	"errors"
	"fmt"
	"io"
	"net"
	"os"
	"path/filepath"
	"strings"
	"sync"
	"sync/atomic"
	"testing"
	"time"

	libvirt "github.com/digitalocean/go-libvirt"
	"github.com/go-logr/logr"
	"golang.org/x/crypto/ssh"
	"golang.org/x/crypto/ssh/knownhosts"

	"gitea.home.hrajfrisbee.cz/kacerr/egress-proxies-operator/internal/provider"
)

// --- hostPool over a fake dialer ---

type fakeClient struct {
	done   chan struct{}
	closed atomic.Bool
}

func (c *fakeClient) close() {
	if c.closed.CompareAndSwap(false, true) {
		close(c.done)
	}
}

type fakeDialer struct {
	mu      sync.Mutex
	dials   int
	err     error
	delay   time.Duration
	clients []*fakeClient
	// api is what the dialled client exposes; nil for pool-only tests.
	api libvirtAPI
}

func (d *fakeDialer) dial(ctx context.Context, _ provider.KVMHost) (*hostClient, error) {
	if d.delay > 0 {
		select {
		case <-time.After(d.delay):
		case <-ctx.Done():
			return nil, ctx.Err()
		}
	}
	d.mu.Lock()
	defer d.mu.Unlock()
	d.dials++
	if d.err != nil {
		return nil, d.err
	}
	fc := &fakeClient{done: make(chan struct{})}
	d.clients = append(d.clients, fc)
	return &hostClient{api: d.api, done: fc.done, close: fc.close}, nil
}

func (d *fakeDialer) dialCount() int {
	d.mu.Lock()
	defer d.mu.Unlock()
	return d.dials
}

var testHost = provider.KVMHost{Name: "nas1", Address: "nas1:22", User: "virt", Socket: defaultSocket, Pool: "default", Bridge: "br0"}

var discard = logr.Discard()

func noop(libvirtAPI) error { return nil }

func TestHostPool_call_reusesConnection(t *testing.T) {
	t.Parallel()
	d := &fakeDialer{}
	p := newHostPool(d.dial)
	for range 3 {
		if err := p.call(context.Background(), discard, testHost, noop); err != nil {
			t.Fatalf("call() error = %v", err)
		}
	}
	if d.dialCount() != 1 {
		t.Errorf("dials = %d, want 1", d.dialCount())
	}
}

func TestHostPool_call_redialsAfterDisconnect(t *testing.T) {
	t.Parallel()
	d := &fakeDialer{}
	p := newHostPool(d.dial)
	if err := p.call(context.Background(), discard, testHost, noop); err != nil {
		t.Fatalf("call() error = %v", err)
	}
	d.clients[0].close()
	if err := p.call(context.Background(), discard, testHost, noop); err != nil {
		t.Fatalf("second call() error = %v", err)
	}
	if d.dialCount() != 2 {
		t.Errorf("dials = %d, want 2 after the first connection dropped", d.dialCount())
	}
}

// Only a transport error is evidence about the link. A daemon-reported
// libvirt error, or a verdict fn reached locally, must not cost the
// connection.
func TestHostPool_call_dropsConnectionOnlyOnTransportError(t *testing.T) {
	t.Parallel()
	tests := []struct {
		name        string
		err         error
		wantDropped bool
	}{
		{"transport error", io.ErrUnexpectedEOF, true},
		{"libvirt error", lvErr(libvirt.ErrNoDomain), false},
		{"locally classified error", fmt.Errorf("%w: creating iso writer: read-only file system", provider.ErrPermanent), false},
	}
	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()
			d := &fakeDialer{}
			p := newHostPool(d.dial)
			log, snapshot := captureLogger(1)
			err := p.call(context.Background(), log, testHost, func(libvirtAPI) error { return tc.err })
			if !errors.Is(err, tc.err) {
				t.Fatalf("call() error = %v, want %v passed through", err, tc.err)
			}
			if got := d.clients[0].closed.Load(); got != tc.wantDropped {
				t.Errorf("connection closed = %v, want %v", got, tc.wantDropped)
			}
			joined := strings.Join(snapshot(), "\n")
			if logged := strings.Contains(joined, `"msg"="libvirt connection dropped"`) && strings.Contains(joined, `"reason"="transport error"`); logged != tc.wantDropped {
				t.Errorf("transport drop logged = %v, want %v:\n%s", logged, tc.wantDropped, joined)
			}
			if err := p.call(context.Background(), discard, testHost, noop); err != nil {
				t.Fatalf("next call() error = %v", err)
			}
			wantDials := 1
			if tc.wantDropped {
				wantDials = 2
			}
			if d.dialCount() != wantDials {
				t.Errorf("dials = %d, want %d", d.dialCount(), wantDials)
			}
		})
	}
}

func TestHostPool_call_dialErrorPropagates(t *testing.T) {
	t.Parallel()
	d := &fakeDialer{err: errors.New("ssh: handshake failed")}
	p := newHostPool(d.dial)
	err := p.call(context.Background(), discard, testHost, noop)
	if err == nil || !strings.Contains(err.Error(), "nas1") || !strings.Contains(err.Error(), "handshake failed") {
		t.Errorf("call() error = %v, want the dial error naming the host", err)
	}
}

func TestHostPool_call_cancelAbandonsBlockedCall(t *testing.T) {
	t.Parallel()
	d := &fakeDialer{}
	p := newHostPool(d.dial)
	ctx, cancel := context.WithCancel(context.Background())
	released := make(chan struct{})
	go func() {
		time.Sleep(20 * time.Millisecond)
		cancel()
	}()
	err := p.call(ctx, discard, testHost, func(libvirtAPI) error {
		// Simulates an RPC stuck on a hung hypervisor: it only returns once
		// the pool closes the connection out from under it.
		<-d.clients[0].done
		close(released)
		return io.EOF
	})
	if !errors.Is(err, context.Canceled) {
		t.Fatalf("call() error = %v, want context.Canceled", err)
	}
	select {
	case <-released:
	case <-time.After(time.Second):
		t.Fatal("blocked call was not released by closing the connection")
	}
	if err := p.call(context.Background(), discard, testHost, noop); err != nil {
		t.Fatalf("call() after cancel = %v", err)
	}
	if d.dialCount() != 2 {
		t.Errorf("dials = %d, want 2", d.dialCount())
	}
}

func TestHostPool_call_concurrentCallersShareOneDial(t *testing.T) {
	t.Parallel()
	d := &fakeDialer{delay: 30 * time.Millisecond}
	p := newHostPool(d.dial)
	var wg sync.WaitGroup
	for range 8 {
		wg.Go(func() {
			if err := p.call(context.Background(), discard, testHost, noop); err != nil {
				t.Errorf("call() error = %v", err)
			}
		})
	}
	wg.Wait()
	if d.dialCount() != 1 {
		t.Errorf("dials = %d, want 1", d.dialCount())
	}
}

func TestHostPool_call_hostsAreIndependent(t *testing.T) {
	t.Parallel()
	d := &fakeDialer{}
	p := newHostPool(d.dial)
	other := testHost
	other.Name = "lab2"
	for _, h := range []provider.KVMHost{testHost, other, testHost, other} {
		if err := p.call(context.Background(), discard, h, noop); err != nil {
			t.Fatalf("call(%s) error = %v", h.Name, err)
		}
	}
	if d.dialCount() != 2 {
		t.Errorf("dials = %d, want one per host", d.dialCount())
	}
}

// A dial to a dead host holds the host's slot for as long as it takes; a
// caller with a deadline must not wait past it.
func TestHostPool_get_waitingRespectsContext(t *testing.T) {
	t.Parallel()
	release := make(chan struct{})
	dialing := make(chan struct{})
	var once sync.Once
	dial := func(context.Context, provider.KVMHost) (*hostClient, error) {
		once.Do(func() { close(dialing) })
		<-release
		return nil, errors.New("ssh dial: connection timed out")
	}
	p := newHostPool(dial)
	slow := make(chan error, 1)
	go func() { slow <- p.call(context.Background(), discard, testHost, noop) }()
	<-dialing
	ctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond)
	defer cancel()
	start := time.Now()

	err := p.call(ctx, discard, testHost, noop)

	if !errors.Is(err, context.DeadlineExceeded) {
		t.Errorf("call() error = %v, want the caller's deadline", err)
	}
	if waited := time.Since(start); waited > time.Second {
		t.Errorf("call() waited %s behind another caller's dial, want about 50ms", waited)
	}
	close(release)
	if err := <-slow; err == nil {
		t.Error("the blocked dial succeeded, want its error")
	}
}

func TestHostPool_probe_abandonsWithoutClosing(t *testing.T) {
	t.Parallel()
	d := &fakeDialer{}
	p := newHostPool(d.dial)
	unblock := make(chan struct{})
	ctx, cancel := context.WithTimeout(context.Background(), 20*time.Millisecond)
	defer cancel()

	err := p.probe(ctx, discard, testHost, func(libvirtAPI) error {
		<-unblock // a daemon that stopped answering
		return nil
	})

	if !errors.Is(err, context.DeadlineExceeded) {
		t.Fatalf("probe() error = %v, want DeadlineExceeded", err)
	}
	if d.clients[0].closed.Load() {
		t.Error("probe closed the shared connection when it gave up")
	}
	if err := p.probe(context.Background(), discard, testHost, noop); !errors.Is(err, errProbeBusy) {
		t.Errorf("probe() while the abandoned one still waits = %v, want errProbeBusy", err)
	}
	if err := p.call(context.Background(), discard, testHost, noop); err != nil {
		t.Errorf("call() on the kept connection = %v", err)
	}
	close(unblock)
	deadline := time.Now().Add(time.Second)
	for {
		err := p.probe(context.Background(), discard, testHost, noop)
		if err == nil {
			break
		}
		if !errors.Is(err, errProbeBusy) || time.Now().After(deadline) {
			t.Fatalf("probe() after the abandoned one returned = %v, want nil", err)
		}
		time.Sleep(time.Millisecond)
	}
	if d.dialCount() != 1 {
		t.Errorf("dials = %d, want 1: nothing should have redialed", d.dialCount())
	}
}

func TestHostPool_probe_dropsConnectionOnTransportError(t *testing.T) {
	t.Parallel()
	d := &fakeDialer{}
	p := newHostPool(d.dial)

	err := p.probe(context.Background(), discard, testHost, func(libvirtAPI) error { return io.ErrUnexpectedEOF })

	if !errors.Is(err, io.ErrUnexpectedEOF) {
		t.Fatalf("probe() error = %v, want io.ErrUnexpectedEOF passed through", err)
	}
	if !d.clients[0].closed.Load() {
		t.Error("a transport error during a probe left the connection cached")
	}
}

// --- SSH dial against an in-process server ---

func newKeyPair(t *testing.T) (ssh.Signer, []byte) {
	t.Helper()
	_, priv, err := ed25519.GenerateKey(rand.Reader)
	if err != nil {
		t.Fatal(err)
	}
	signer, err := ssh.NewSignerFromKey(priv)
	if err != nil {
		t.Fatal(err)
	}
	block, err := ssh.MarshalPrivateKey(priv, "")
	if err != nil {
		t.Fatal(err)
	}
	return signer, pem.EncodeToMemory(block)
}

// testSSHServer accepts one public key and serves
// direct-streamlocal@openssh.com channels by dialing the requested unix
// path, which is exactly what OpenSSH does for ssh.Client.Dial("unix").
type testSSHServer struct {
	addr    string
	hostKey ssh.PublicKey

	mu    sync.Mutex
	paths []string
}

func startSSHServer(t *testing.T, clientKey ssh.PublicKey, answerGlobalRequests bool) *testSSHServer {
	t.Helper()
	hostSigner, _ := newKeyPair(t)
	cfg := &ssh.ServerConfig{
		PublicKeyCallback: func(_ ssh.ConnMetadata, key ssh.PublicKey) (*ssh.Permissions, error) {
			if bytes.Equal(key.Marshal(), clientKey.Marshal()) {
				return nil, nil
			}
			return nil, errors.New("unknown public key")
		},
	}
	cfg.AddHostKey(hostSigner)
	ln, err := net.Listen("tcp", "127.0.0.1:0")
	if err != nil {
		t.Fatal(err)
	}
	t.Cleanup(func() { ln.Close() })
	s := &testSSHServer{addr: ln.Addr().String(), hostKey: hostSigner.PublicKey()}
	go func() {
		for {
			nc, err := ln.Accept()
			if err != nil {
				return
			}
			go s.serve(nc, cfg, answerGlobalRequests)
		}
	}()
	return s
}

func (s *testSSHServer) serve(nc net.Conn, cfg *ssh.ServerConfig, answerGlobalRequests bool) {
	_, chans, reqs, err := ssh.NewServerConn(nc, cfg)
	if err != nil {
		return
	}
	if answerGlobalRequests {
		go ssh.DiscardRequests(reqs)
	} else {
		go func() {
			for range reqs { //nolint:revive // deliberately never replies, to stall keepalives
			}
		}()
	}
	for newCh := range chans {
		if newCh.ChannelType() != "direct-streamlocal@openssh.com" {
			_ = newCh.Reject(ssh.UnknownChannelType, "test server only forwards unix sockets")
			continue
		}
		var payload struct {
			Path      string
			Reserved0 string
			Reserved1 uint32
		}
		if err := ssh.Unmarshal(newCh.ExtraData(), &payload); err != nil {
			_ = newCh.Reject(ssh.ConnectionFailed, err.Error())
			continue
		}
		s.mu.Lock()
		s.paths = append(s.paths, payload.Path)
		s.mu.Unlock()
		target, err := net.Dial("unix", payload.Path)
		if err != nil {
			_ = newCh.Reject(ssh.ConnectionFailed, err.Error())
			continue
		}
		ch, chReqs, err := newCh.Accept()
		if err != nil {
			target.Close()
			continue
		}
		go ssh.DiscardRequests(chReqs)
		go func() { _, _ = io.Copy(ch, target); ch.Close() }()
		go func() { _, _ = io.Copy(target, ch); target.Close() }()
	}
}

func (s *testSSHServer) requestedPaths() []string {
	s.mu.Lock()
	defer s.mu.Unlock()
	return append([]string(nil), s.paths...)
}

// startEchoSocket serves a unix socket that echoes everything back, standing
// in for libvirt-sock.
func startEchoSocket(t *testing.T) string {
	t.Helper()
	path := filepath.Join(t.TempDir(), "libvirt-sock")
	ln, err := net.Listen("unix", path)
	if err != nil {
		t.Fatal(err)
	}
	t.Cleanup(func() { ln.Close() })
	go func() {
		for {
			c, err := ln.Accept()
			if err != nil {
				return
			}
			go func() { _, _ = io.Copy(c, c); c.Close() }()
		}
	}()
	return path
}

type sshFixture struct {
	server  *testSSHServer
	cfg     provider.KVMConfig
	host    provider.KVMHost
	keyFile string
}

func newSSHFixture(t *testing.T, answerGlobalRequests bool) *sshFixture {
	t.Helper()
	clientSigner, clientPEM := newKeyPair(t)
	server := startSSHServer(t, clientSigner.PublicKey(), answerGlobalRequests)
	dir := t.TempDir()
	keyFile := filepath.Join(dir, "id_ed25519")
	if err := os.WriteFile(keyFile, clientPEM, 0o600); err != nil {
		t.Fatal(err)
	}
	khFile := filepath.Join(dir, "known_hosts")
	line := knownhosts.Line([]string{server.addr}, server.hostKey)
	if err := os.WriteFile(khFile, []byte(line+"\n"), 0o600); err != nil {
		t.Fatal(err)
	}
	return &sshFixture{
		server:  server,
		keyFile: keyFile,
		cfg:     provider.KVMConfig{SSHKeyFile: keyFile, KnownHostsFile: khFile},
		host:    provider.KVMHost{Name: "nas1", Address: server.addr, User: "virt", Socket: startEchoSocket(t)},
	}
}

func mustSSHAuth(t *testing.T, cfg provider.KVMConfig) *sshAuth {
	t.Helper()
	auth, err := newSSHAuth(cfg)
	if err != nil {
		t.Fatalf("newSSHAuth() error = %v", err)
	}
	return auth
}

func TestSSHAuth_dialSSH_forwardsUnixSocket(t *testing.T) {
	t.Parallel()
	f := newSSHFixture(t, true)
	auth := mustSSHAuth(t, f.cfg)
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()
	client, err := auth.dialSSH(ctx, f.host)
	if err != nil {
		t.Fatalf("dialSSH() error = %v", err)
	}
	defer client.Close()

	conn, err := unixForward{client: client, path: f.host.Socket}.Dial()
	if err != nil {
		t.Fatalf("unixForward.Dial() error = %v", err)
	}
	defer conn.Close()
	if _, err := conn.Write([]byte("ping")); err != nil {
		t.Fatal(err)
	}
	buf := make([]byte, 4)
	if _, err := io.ReadFull(conn, buf); err != nil || string(buf) != "ping" {
		t.Fatalf("echo through the forward = %q, %v; want ping", buf, err)
	}
	if paths := f.server.requestedPaths(); len(paths) != 1 || paths[0] != f.host.Socket {
		t.Errorf("server saw forward requests %v, want [%s]", paths, f.host.Socket)
	}
}

func TestSSHAuth_dialSSH_rejectsUnknownHostKey(t *testing.T) {
	t.Parallel()
	f := newSSHFixture(t, true)
	impostor, _ := newKeyPair(t)
	line := knownhosts.Line([]string{f.server.addr}, impostor.PublicKey())
	if err := os.WriteFile(f.cfg.KnownHostsFile, []byte(line+"\n"), 0o600); err != nil {
		t.Fatal(err)
	}
	auth := mustSSHAuth(t, f.cfg)
	client, err := auth.dialSSH(context.Background(), f.host)
	if err == nil {
		client.Close()
		t.Fatal("dialSSH() succeeded against a host whose key is not in known_hosts")
	}
	var keyErr *knownhosts.KeyError
	if !errors.As(err, &keyErr) {
		t.Errorf("dialSSH() error = %v, want a knownhosts.KeyError", err)
	}
}

func TestSSHAuth_dialSSH_insecureIgnoreHostKey(t *testing.T) {
	t.Parallel()
	f := newSSHFixture(t, true)
	f.cfg.KnownHostsFile = ""
	f.cfg.InsecureIgnoreHostKey = true
	client, err := mustSSHAuth(t, f.cfg).dialSSH(context.Background(), f.host)
	if err != nil {
		t.Fatalf("dialSSH() error = %v", err)
	}
	client.Close()
}

func TestSSHAuth_dialSSH_rejectsWrongClientKey(t *testing.T) {
	t.Parallel()
	f := newSSHFixture(t, true)
	_, otherPEM := newKeyPair(t)
	if err := os.WriteFile(f.keyFile, otherPEM, 0o600); err != nil {
		t.Fatal(err)
	}
	client, err := mustSSHAuth(t, f.cfg).dialSSH(context.Background(), f.host)
	if err == nil {
		client.Close()
		t.Fatal("dialSSH() succeeded with a key the server does not know")
	}
	if !strings.Contains(err.Error(), "ssh handshake") {
		t.Errorf("dialSSH() error = %v, want an auth failure from the handshake", err)
	}
}

func TestSSHAuth_dialSSH_cancelledContext(t *testing.T) {
	t.Parallel()
	f := newSSHFixture(t, true)
	ctx, cancel := context.WithCancel(context.Background())
	cancel()
	if client, err := mustSSHAuth(t, f.cfg).dialSSH(ctx, f.host); err == nil {
		client.Close()
		t.Fatal("dialSSH() succeeded with an already-cancelled context")
	}
}

func TestNewSSHAuth_missingFiles(t *testing.T) {
	t.Parallel()
	_, keyPEM := newKeyPair(t)
	keyFile := filepath.Join(t.TempDir(), "key")
	if err := os.WriteFile(keyFile, keyPEM, 0o600); err != nil {
		t.Fatal(err)
	}
	tests := []struct {
		name    string
		cfg     provider.KVMConfig
		wantSub string
	}{
		{"key file missing", provider.KVMConfig{SSHKeyFile: "/nonexistent/key", InsecureIgnoreHostKey: true}, "sshKeyFile"},
		{"key file garbage", provider.KVMConfig{SSHKeyFile: keyFile + ".bad", InsecureIgnoreHostKey: true}, "sshKeyFile"},
		{"known_hosts missing", provider.KVMConfig{SSHKeyFile: keyFile, KnownHostsFile: "/nonexistent/kh"}, "knownHostsFile"},
	}
	if err := os.WriteFile(keyFile+".bad", []byte("not a key"), 0o600); err != nil {
		t.Fatal(err)
	}
	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()
			_, err := newSSHAuth(tc.cfg)
			if err == nil || !strings.Contains(err.Error(), tc.wantSub) {
				t.Errorf("newSSHAuth() error = %v, want one mentioning %s", err, tc.wantSub)
			}
		})
	}
}

// The kvm-ssh Secret is an optional mount, so the provider has to be
// constructible without it and has to start working once it appears — the
// alternative was a constructor error that crash-looped the whole manager.
func TestLazySSH_load_healsWhenTheKeyAppears(t *testing.T) {
	t.Parallel()
	f := newSSHFixture(t, true)
	keyPEM, err := os.ReadFile(f.keyFile)
	if err != nil {
		t.Fatal(err)
	}
	if err := os.Remove(f.keyFile); err != nil {
		t.Fatal(err)
	}
	auth := newLazySSH(f.cfg).load

	if _, err := auth(); err == nil {
		t.Fatal("loading with no key file succeeded, want an error")
	} else if class := classify(err); class != provider.ErrTransient {
		t.Errorf("class = %v, want ErrTransient so the proxy retries once the Secret lands", class)
	}
	if err := os.WriteFile(f.keyFile, keyPEM, 0o600); err != nil {
		t.Fatal(err)
	}
	loaded, err := auth()
	if err != nil {
		t.Fatalf("loading after the key appeared: %v", err)
	}
	// Cached from here on: the file may vanish again (a Secret rotated out)
	// without breaking connections that already work.
	if err := os.Remove(f.keyFile); err != nil {
		t.Fatal(err)
	}
	if again, err := auth(); err != nil || again != loaded {
		t.Errorf("auth() = %v, %v; want the cached value reused", again, err)
	}
}

func TestLazySSH_load_unparseableKeyIsPermanent(t *testing.T) {
	t.Parallel()
	f := newSSHFixture(t, true)
	if err := os.WriteFile(f.keyFile, []byte("not a key"), 0o600); err != nil {
		t.Fatal(err)
	}
	_, err := newLazySSH(f.cfg).load()
	if classify(err) != provider.ErrPermanent {
		t.Errorf("class = %v, want ErrPermanent: retrying cannot fix a key that does not parse", classify(err))
	}
}

// The dialer must authenticate with the material the lazy loader produced,
// not a stale or empty one. The echo socket is not a libvirt server, so the
// call gets as far as the libvirt handshake and no further; what matters is
// that it is past SSH by then.
func TestLazySSH_dial_authenticatesOverSSH(t *testing.T) {
	t.Parallel()
	f := newSSHFixture(t, true)
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	_, err := newLazySSH(f.cfg).dial(ctx, f.host)
	if err == nil {
		t.Fatal("dial to an echo socket succeeded, want a libvirt handshake error")
	}
	if strings.Contains(err.Error(), "sshKeyFile") || strings.Contains(err.Error(), "ssh handshake") {
		t.Errorf("dial failed before libvirt: %v", err)
	}
	if paths := f.server.requestedPaths(); len(paths) != 1 || paths[0] != f.host.Socket {
		t.Errorf("server saw forward requests %v, want [%s]", paths, f.host.Socket)
	}
}

// New must not fail on missing key material, or one misconfigured provider
// aborts registry.Build and the manager exits before any controller starts.
func TestNew_missingSSHMaterialStillBuilds(t *testing.T) {
	t.Parallel()
	p, err := New(context.Background(), provider.ProviderConfig{
		Name: "kvm-home", Type: "kvm",
		KVM: &provider.KVMConfig{
			SSHKeyFile:     filepath.Join(t.TempDir(), "absent"),
			KnownHostsFile: filepath.Join(t.TempDir(), "absent"),
			Hosts:          []provider.KVMHost{{Name: "nas1", Address: "nas1:22", User: "virt", Bridge: "br0"}},
		},
	})
	if err != nil || p == nil {
		t.Fatalf("New() = %v, %v; want a usable provider", p, err)
	}
	if _, err := p.Get(context.Background(), "hosts/nas1/domains/proxy-x"); provider.Class(err) != provider.ErrTransient {
		t.Errorf("Get class = %v, want ErrTransient", provider.Class(err))
	}
}

// The health check reports where a connection broke, so every stage of a
// real dial must come back tagged.
func TestLazySSH_dial_tagsTheFailingStage(t *testing.T) {
	t.Parallel()
	tests := []struct {
		name       string
		mutate     func(t *testing.T, f *sshFixture)
		want       dialStage
		wantReason string
	}{
		{
			name:       "key material missing",
			mutate:     func(t *testing.T, f *sshFixture) { f.cfg.SSHKeyFile = filepath.Join(t.TempDir(), "absent") },
			want:       stageSSHMaterial,
			wantReason: reasonSSHMaterialUnavailable,
		},
		{
			name: "nothing listening",
			mutate: func(t *testing.T, f *sshFixture) {
				ln, err := net.Listen("tcp", "127.0.0.1:0")
				if err != nil {
					t.Fatal(err)
				}
				f.host.Address = ln.Addr().String()
				ln.Close()
			},
			want:       stageTCP,
			wantReason: provider.ReasonUnreachable,
		},
		{
			name: "host key not in known_hosts",
			mutate: func(t *testing.T, f *sshFixture) {
				impostor, _ := newKeyPair(t)
				line := knownhosts.Line([]string{f.server.addr}, impostor.PublicKey())
				if err := os.WriteFile(f.cfg.KnownHostsFile, []byte(line+"\n"), 0o600); err != nil {
					t.Fatal(err)
				}
			},
			want:       stageSSHHandshake,
			wantReason: reasonSSHHandshakeFailed,
		},
		{
			name: "client key rejected",
			mutate: func(t *testing.T, f *sshFixture) {
				_, otherPEM := newKeyPair(t)
				if err := os.WriteFile(f.keyFile, otherPEM, 0o600); err != nil {
					t.Fatal(err)
				}
			},
			want:       stageSSHHandshake,
			wantReason: reasonSSHHandshakeFailed,
		},
		{
			name:       "libvirt socket missing",
			mutate:     func(_ *testing.T, f *sshFixture) { f.host.Socket = "/nonexistent/libvirt-sock" },
			want:       stageLibvirtConnect,
			wantReason: reasonLibvirtConnectFailed,
		},
	}
	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()
			f := newSSHFixture(t, true)
			tc.mutate(t, f)
			ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
			defer cancel()

			_, err := newLazySSH(f.cfg).dial(ctx, f.host)

			var de *dialError
			if !errors.As(err, &de) || de.stage != tc.want {
				t.Fatalf("dial() error = %v, want a %s dialError", err, tc.want)
			}
			if got := checkReason(err); got != tc.wantReason {
				t.Errorf("checkReason() = %s, want %s", got, tc.wantReason)
			}
		})
	}
}

func TestKeepalive_closesClientWhenServerStalls(t *testing.T) {
	t.Parallel()
	f := newSSHFixture(t, false)
	client, err := mustSSHAuth(t, f.cfg).dialSSH(context.Background(), f.host)
	if err != nil {
		t.Fatalf("dialSSH() error = %v", err)
	}
	defer client.Close()
	stop := make(chan struct{})
	defer close(stop)
	go keepalive(client, 20*time.Millisecond, stop)

	closed := make(chan struct{})
	go func() { _ = client.Wait(); close(closed) }()
	select {
	case <-closed:
	case <-time.After(2 * time.Second):
		t.Fatal("client still open after keepalives went unanswered")
	}
}

func TestKeepalive_keepsHealthyClientOpen(t *testing.T) {
	t.Parallel()
	f := newSSHFixture(t, true)
	client, err := mustSSHAuth(t, f.cfg).dialSSH(context.Background(), f.host)
	if err != nil {
		t.Fatalf("dialSSH() error = %v", err)
	}
	defer client.Close()
	stop := make(chan struct{})
	defer close(stop)
	go keepalive(client, 10*time.Millisecond, stop)
	time.Sleep(60 * time.Millisecond)
	fwd := unixForward{client: client, path: f.host.Socket}
	conn, err := fwd.Dial()
	if err != nil {
		t.Fatalf("forward after several keepalives failed: %v", err)
	}
	conn.Close()
}

func TestHostPool_call_tracesAtV1(t *testing.T) {
	t.Parallel()
	d := &fakeDialer{api: readyHost()}
	p := newHostPool(d.dial)
	log, snapshot := captureLogger(1)
	err := p.call(context.Background(), log, testHost, func(api libvirtAPI) error {
		_, err := api.StoragePoolLookupByName("default")
		return err
	})
	if err != nil {
		t.Fatalf("call() error = %v", err)
	}
	joined := strings.Join(snapshot(), "\n")
	for _, want := range []string{`"msg"="libvirt host connected"`, `"address"="nas1:22"`, `"rpc"="virStoragePoolLookupByName"`, `"pool"="default"`, "libvirt "} {
		if !strings.Contains(joined, want) {
			t.Errorf("output missing %s:\n%s", want, joined)
		}
	}
}

func TestHostPool_call_noTracingBelowV1(t *testing.T) {
	t.Parallel()
	inner := readyHost()
	d := &fakeDialer{api: inner}
	p := newHostPool(d.dial)
	log, snapshot := captureLogger(0)
	err := p.call(context.Background(), log, testHost, func(api libvirtAPI) error {
		if api != libvirtAPI(inner) {
			t.Errorf("fn received %T, want the bare fake when V(1) is off", api)
		}
		return nil
	})
	if err != nil {
		t.Fatalf("call() error = %v", err)
	}
	if lines := snapshot(); len(lines) != 0 {
		t.Errorf("verbosity 0 logged %d lines:\n%s", len(lines), strings.Join(lines, "\n"))
	}
}

func TestHostPool_call_logsCancelDropReason(t *testing.T) {
	t.Parallel()
	d := &fakeDialer{}
	p := newHostPool(d.dial)
	log, snapshot := captureLogger(1)
	ctx, cancel := context.WithCancel(context.Background())
	cancel()
	released := make(chan struct{})
	err := p.call(ctx, log, testHost, func(libvirtAPI) error {
		<-d.clients[0].done
		close(released)
		return io.EOF
	})
	if !errors.Is(err, context.Canceled) {
		t.Fatalf("call() error = %v, want context.Canceled", err)
	}
	<-released
	joined := strings.Join(snapshot(), "\n")
	if !strings.Contains(joined, `"reason"="context cancelled"`) {
		t.Errorf("output missing the cancel drop line:\n%s", joined)
	}
	if strings.Contains(joined, `"reason"="transport error"`) {
		t.Errorf("cancel was also reported as a transport error:\n%s", joined)
	}
}