package bpfutil

import (
	"fmt"
	"os"
	"syscall"
	"unsafe"
)

const BPFFSPath = "/sys/fs/bpf"

var (
	FsMagicBPFFS     int32
	INSP_PINNED_PATH = fmt.Sprintf("%s/inspector", BPFFSPath) // nolint
)

func init() {
	// https://github.com/coreutils/coreutils/blob/v8.27/src/stat.c#L275
	// https://github.com/torvalds/linux/blob/v4.8/include/uapi/linux/magic.h#L80
	magic := uint32(0xCAFE4A11)
	// 0xCAFE4A11 overflows an int32, which is what's expected by Statfs_t.Type in 32bit platforms.
	// To avoid conditional compilation for all 32bit/64bit platforms, we use an unsafe cast
	FsMagicBPFFS = *(*int32)(unsafe.Pointer(&magic))
}

// IsMountedAt checks if the BPF fs is mounted already in the custom location
func IsMountedAt(mountpoint string) (bool, error) {
	var data syscall.Statfs_t
	if err := syscall.Statfs(mountpoint, &data); err != nil {
		return false, fmt.Errorf("cannot statfs %q: %v", mountpoint, err)
	}
	return int32(data.Type) == FsMagicBPFFS, nil
}

// IsMounted checks if the BPF fs is mounted already in the default location
func IsMounted() (bool, error) {
	return IsMountedAt(BPFFSPath)
}

// MountAt mounts the BPF fs in the custom location (if not already mounted)
func MountAt(mountpoint string) error {
	mounted, err := IsMountedAt(mountpoint)
	if err != nil {
		return err
	}
	if mounted {
		return nil
	}
	if err := syscall.Mount(mountpoint, mountpoint, "bpf", 0, ""); err != nil {
		return fmt.Errorf("error mounting %q: %v", mountpoint, err)
	}
	return nil
}

// Mount mounts the BPF fs in the default location (if not already mounted)
func Mount() error {
	err := MountAt(BPFFSPath)
	if err != nil {
		return err
	}

	// ensure inspector pinned path
	err = os.Mkdir(INSP_PINNED_PATH, 0755)
	if err != nil {
		return err
	}

	return nil
}
