//go:build linux

package sandbox

import (
	"fmt"
	"io"
	"os"
	"os/exec"
	"syscall"
)

var (
	applyUnixSocketBlockFilter  = ApplyUnixSocketBlock
	applyLinuxNetworkDenyFilter = ApplyLinuxNetworkDeny
)

func RunLinuxSandboxHelper(args []string, stderr io.Writer) int {
	config, err := ParseLinuxSandboxHelperArgs(args)
	if err != nil {
		fmt.Fprintln(stderr, LinuxSandboxHelperName+": "+err.Error())
		return 2
	}
	if config.ApplySeccompThenExec {
		return runLinuxSandboxInnerStage(config, stderr)
	}
	if config.UseLandlock {
		return runLinuxSandboxLandlockStage(config, stderr)
	}
	helperPath, err := os.Executable()
	if err != nil {
		fmt.Fprintln(stderr, LinuxSandboxHelperName+": resolve helper path: "+err.Error())
		return 125
	}
	bwrapArgs, err := BuildLinuxSandboxBwrapArgs(LinuxSandboxBwrapOptions{
		Config:     config,
		HelperPath: helperPath,
	})
	if err != nil {
		fmt.Fprintln(stderr, LinuxSandboxHelperName+": "+err.Error())
		return 125
	}
	bwrapPath, err := findBubblewrapExecutable()
	if err != nil {
		fmt.Fprintln(stderr, LinuxSandboxHelperName+": bubblewrap is not available: "+err.Error())
		return 125
	}
	if err := syscall.Exec(bwrapPath, append([]string{"bwrap"}, bwrapArgs...), os.Environ()); err != nil {
		fmt.Fprintln(stderr, LinuxSandboxHelperName+": exec bubblewrap: "+err.Error())
		return 126
	}
	return 126
}

func findBubblewrapExecutable() (string, error) {
	if path, err := exec.LookPath("bwrap"); err == nil && path != "" {
		return path, nil
	}
	for _, candidate := range []string{"/usr/bin/bwrap", "/bin/bwrap"} {
		if executableRegularFile(candidate) {
			return candidate, nil
		}
	}
	return "", exec.ErrNotFound
}

func runLinuxSandboxInnerStage(config LinuxSandboxHelperConfig, stderr io.Writer) int {
	if config.UseLandlock {
		fmt.Fprintln(stderr, LinuxSandboxHelperName+": inner seccomp stage is incompatible with Landlock mode")
		return 2
	}
	if shouldUnshareLinuxNetwork(config.PermissionProfile.Network) {
		if err := applyLinuxNetworkDenyFilter(); err != nil {
			fmt.Fprintln(stderr, LinuxSandboxHelperName+": apply network deny: "+err.Error())
			return 125
		}
	}
	if config.BlockUnixSockets {
		if err := applyUnixSocketBlockFilter(); err != nil {
			fmt.Fprintln(stderr, LinuxSandboxHelperName+": warning: "+err.Error()+"; running without the Unix-socket filter")
		}
	}
	binary, err := exec.LookPath(config.Command[0])
	if err != nil {
		fmt.Fprintln(stderr, LinuxSandboxHelperName+": "+err.Error())
		return 127
	}
	if err := syscall.Exec(binary, config.Command, os.Environ()); err != nil {
		fmt.Fprintln(stderr, LinuxSandboxHelperName+": exec command: "+err.Error())
		return 126
	}
	return 126
}

func runLinuxSandboxLandlockStage(config LinuxSandboxHelperConfig, stderr io.Writer) int {
	if err := ApplyLandlockFilesystemProfile(config.PermissionProfile, config.CommandCWD); err != nil {
		fmt.Fprintln(stderr, LinuxSandboxHelperName+": apply Landlock: "+err.Error())
		return 125
	}
	binary, err := exec.LookPath(config.Command[0])
	if err != nil {
		fmt.Fprintln(stderr, LinuxSandboxHelperName+": "+err.Error())
		return 127
	}
	if err := syscall.Exec(binary, config.Command, linuxHelperSandboxEnvironment(config.PermissionProfile, os.Environ())); err != nil {
		fmt.Fprintln(stderr, LinuxSandboxHelperName+": exec command: "+err.Error())
		return 126
	}
	return 126
}