feat: docker setup container logging

Author: gregnr

client/main.go

@@ -57,8 +57,6 @@ func main() {
 		os.Exit(ExitSetupFailed)
 	}
 
-	fmt.Printf("Setting up interface %s\n", interfaceName)
-
 	links, err := netlink.LinkList()
 	if err != nil {
 		fmt.Printf("Could not list links: %v\n", err)
@@ -80,6 +78,8 @@ func main() {
 	linkAttrs := netlink.NewLinkAttrs()
 	linkAttrs.Name = interfaceName
 
+	fmt.Printf("Creating WireGuard interface %s\n", interfaceName)
+
 	wireguard := &netlink.Wireguard{LinkAttrs: linkAttrs}
 	err = netlink.LinkAdd(wireguard)
 	if err != nil {
@@ -95,6 +95,8 @@ func main() {
 		fmt.Printf("Could not parse host peer IPNet: %v\n", err)
 	}
 
+	fmt.Println("Assigning IP to WireGuard interface")
+
 	addr := netlink.Addr{IPNet: vmIpNet, Peer: hostIpNet}
 	netlink.AddrAdd(wireguard, &addr)
 
@@ -146,6 +148,8 @@ func main() {
 		},
 	}
 
+	fmt.Println("Configuring WireGuard device")
+
 	err = c.ConfigureDevice(interfaceName, wgtypes.Config{
 		PrivateKey: &vmPrivateKey,
 		Peers:      []wgtypes.PeerConfig{peer},
@@ -167,6 +171,8 @@ func main() {
 		os.Exit(ExitSetupFailed)
 	}
 
+	fmt.Println("Adding iptables NAT rule for host WireGuard IP")
+
 	// Add iptables NAT rule to translate incoming packet's
 	// source IP to the respective Docker network interface IP.
 	// Required to route reply packets back through correct

main.go

@@ -17,6 +17,7 @@ import (
 	"github.com/docker/docker/api/types/container"
 	"github.com/docker/docker/api/types/filters"
 	"github.com/docker/docker/client"
+	"github.com/docker/docker/pkg/stdcopy"
 	"golang.zx2c4.com/wireguard/conn"
 	"golang.zx2c4.com/wireguard/device"
 	"golang.zx2c4.com/wireguard/ipc"
@@ -314,5 +315,27 @@ func setupVm(
 		return fmt.Errorf("failed to start container: %w", err)
 	}
 
+	func() error {
+		reader, err := dockerCli.ContainerLogs(ctx, resp.ID, types.ContainerLogsOptions{
+			ShowStdout: true,
+			ShowStderr: true,
+			Follow:     true,
+		})
+		if err != nil {
+			return fmt.Errorf("failed to get logs for container %s: %w", resp.ID, err)
+		}
+
+		defer reader.Close()
+
+		_, err = stdcopy.StdCopy(os.Stdout, os.Stderr, reader)
+		if err != nil {
+			return err
+		}
+
+		return nil
+	}()
+
+	fmt.Println("Setup container complete")
+
 	return nil
 }