hakurei/internal/pkg/testdata/main.go

//go:build testtool

package main

import (
	"errors"
	"log"
	"net"
	"os"
	"path"
	"slices"
	"strings"
	"syscall"

	"hakurei.app/container/fhs"
	"hakurei.app/container/vfs"
)

func main() {
	log.SetFlags(0)
	log.SetPrefix("testtool: ")

	var hostNet bool
	wantArgs := []string{"testtool"}
	if len(os.Args) == 2 {
		hostNet = true
		log.SetPrefix("testtool(net): ")
		wantArgs = []string{"testtool", "net"}
	}
	if !slices.Equal(os.Args, wantArgs) {
		log.Fatalf("Args: %q, want %q", os.Args, wantArgs)
	}

	var overlayRoot bool
	wantEnv := []string{"HAKUREI_TEST=1"}
	if len(os.Environ()) == 2 {
		overlayRoot = true
		log.SetPrefix("testtool(overlay root): ")
		wantEnv = []string{"HAKUREI_TEST=1", "HAKUREI_ROOT=1"}
	}
	if !slices.Equal(wantEnv, os.Environ()) {
		log.Fatalf("Environ: %q, want %q", os.Environ(), wantEnv)
	}

	var overlayWork bool
	const (
		wantExec     = "/opt/bin/testtool"
		wantExecWork = "/work/bin/testtool"
	)
	if got, err := os.Executable(); err != nil {
		log.Fatalf("Executable: error = %v", err)
	} else if got != wantExec {
		switch got {
		case wantExecWork:
			overlayWork = true
			log.SetPrefix("testtool(overlay work): ")

		default:
			log.Fatalf("Executable: %q, want %q", got, wantExec)
		}
	}

	wantHostname := "cure"
	if hostNet {
		wantHostname += "-net"
	}

	if hostname, err := os.Hostname(); err != nil {
		log.Fatalf("Hostname: error = %v", err)
	} else if hostname != wantHostname {
		log.Fatalf("Hostname: %q, want %q", hostname, wantHostname)
	}

	var m *vfs.MountInfo
	if f, err := os.Open(fhs.Proc + "self/mountinfo"); err != nil {
		log.Fatalf("Open: error = %v", err)
	} else {
		err = vfs.NewMountInfoDecoder(f).Decode(&m)
		closeErr := f.Close()
		if err != nil {
			log.Fatalf("Decode: error = %v", err)
		}
		if closeErr != nil {
			log.Fatalf("Close: error = %v", err)
		}
	}

	if _, err := net.Dial("tcp", "127.0.0.1:0"); err == nil {
		log.Fatal("Dial unexpectedly succeeded")
	} else if hostNet {
		if !errors.Is(err, syscall.ECONNREFUSED) {
			log.Fatalf("Dial: error = %v", err)
		}
	} else {
		if !errors.Is(err, syscall.ENETUNREACH) {
			log.Fatalf("Dial: error = %v", err)
		}
	}

	const checksumEmptyDir = "MGWmEfjut2QE2xPJwTsmUzpff4BN_FEnQ7T0j7gvUCCiugJQNwqt9m151fm9D1yU"
	ident := "U2cbgVgEtjfRuvHfE1cQnZ3t8yoexULQyo_VLgvxAVJSsobMcNaFIsuDWtmt7kzK"
	log.Println(m)
	next := func() { m = m.Next; log.Println(m) }

	if overlayRoot {
		ident = "5ey2wpmMpj483YYa7ZZQciYLA2cx3_l167JCqWW4Pd-5DVp81dj9EsBtVTwYptF6"

		if m.Root != "/" || m.Target != "/" ||
			m.Source != "overlay" || m.FsType != "overlay" {
			log.Fatal("unexpected root mount entry")
		}
		var lowerdir string
		for _, o := range strings.Split(m.FsOptstr, ",") {
			const lowerdirKey = "lowerdir="
			if strings.HasPrefix(o, lowerdirKey) {
				lowerdir = o[len(lowerdirKey):]
			}
		}
		if path.Base(lowerdir) != checksumEmptyDir {
			log.Fatal("unexpected artifact checksum")
		}
	} else {
		if hostNet {
			ident = "QdsJhGgnk5N2xdUNGcndXQxFKifxf1V_2t9X8CQ-pDcg24x6mGJC_BiLfGbs6Qml"
		}

		if m.Root != "/sysroot" || m.Target != "/" {
			log.Fatal("unexpected root mount entry")
		}

		next()
		if path.Base(m.Root) != "OLBgp1GsljhM2TJ-sbHjaiH9txEUvgdDTAzHv2P24donTt6_529l-9Ua0vFImLlb" {
			log.Fatal("unexpected file artifact checksum")
		}

		next()
		if path.Base(m.Root) != checksumEmptyDir {
			log.Fatal("unexpected artifact checksum")
		}
	}

	next() // testtool artifact

	next()
	if overlayWork {
		ident = "5ynpgDp5mcAmbFndOESXjQnOy_RG6furIxDpQz6c0bImBBLMAI8FouNbAB0uNs6f"
		if m.Root != "/" || m.Target != "/work" ||
			m.Source != "overlay" || m.FsType != "overlay" {
			log.Fatal("unexpected work mount entry")
		}
	} else {
		if path.Base(m.Root) != ident || m.Target != "/work" {
			log.Fatal("unexpected work mount entry")
		}
	}

	next()
	if path.Base(m.Root) != ident || m.Target != "/tmp" {
		log.Fatal("unexpected temp mount entry")
	}

	next()
	if m.Root != "/" || m.Target != "/proc" || m.Source != "proc" || m.FsType != "proc" {
		log.Fatal("unexpected proc mount entry")
	}

	next()
	if m.Root != "/" || m.Target != "/dev" || m.Source != "devtmpfs" || m.FsType != "tmpfs" {
		log.Fatal("unexpected dev mount entry")
	}

	for i := 0; i < 9; i++ { // private /dev entries
		next()
	}

	if m.Next != nil {
		log.Println("unexpected extra mount entries")
		for m.Next != nil {
			next()
		}
		os.Exit(1)
	}

	checkData := []byte{0}
	if hostNet {
		checkData = []byte("net")
	}
	if err := os.WriteFile("check", checkData, 0400); err != nil {
		log.Fatal(err)
	}
}