// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"). You may
// not use this file except in compliance with the License. A copy of the
// License is located at
//
//	http://aws.amazon.com/apache2.0/
//
// or in the "license" file accompanying this file. This file is distributed
// on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
// express or implied. See the License for the specific language governing
// permissions and limitations under the License.

package main

import (
	"context"
	"os"
	"path/filepath"
	"syscall"
	"testing"

	"github.com/firecracker-microvm/firecracker-containerd/proto"

	"github.com/firecracker-microvm/firecracker-go-sdk"
	models "github.com/firecracker-microvm/firecracker-go-sdk/client/models"
	"github.com/sirupsen/logrus"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"github.com/firecracker-microvm/firecracker-containerd/config"
	"github.com/firecracker-microvm/firecracker-containerd/internal"
)

func TestBuildJailedRootHandler(t *testing.T) {
	internal.RequiresRoot(t)
	dir := t.TempDir()
	ctx := context.Background()

	kernelImagePath := filepath.Join(dir, "kernel-image")
	kernelImageFd, err := os.OpenFile(kernelImagePath, os.O_CREATE, 0600)
	require.NoError(t, err, "failed to create kernel image")
	defer kernelImageFd.Close()

	rootDrivePath := filepath.Join(dir, "root-drive")
	rootDriveFd, err := os.OpenFile(rootDrivePath, os.O_CREATE, 0600)
	require.NoError(t, err, "failed to create kernel image")
	defer rootDriveFd.Close()

	firecrackerPath := filepath.Join(dir, "firecracker")
	firecrackerFd, err := os.OpenFile(firecrackerPath, os.O_CREATE, 0600)
	require.NoError(t, err, "failed to create firecracker")
	defer firecrackerFd.Close()

	l := logrus.NewEntry(logrus.New())
	runcConfig := runcJailerConfig{
		OCIBundlePath:  dir,
		RuncBinPath:    "bin-path",
		RuncConfigPath: "./firecracker-runc-config.json.example",
		UID:            123,
		GID:            456,
	}
	vmID := "foo"
	jailer, err := newRuncJailer(ctx, l, vmID, runcConfig, []*proto.FirecrackerDriveMount{})
	require.NoError(t, err, "failed to create runc jailer")

	cfg := config.Config{
		FirecrackerBinaryPath: firecrackerPath,
		KernelImagePath:       kernelImagePath,
		RootDrive:             rootDrivePath,
	}
	machineConfig := firecracker.Config{
		SocketPath:      "/path/to/api.socket",
		KernelImagePath: kernelImagePath,
		Drives: []models.Drive{
			{
				PathOnHost:   firecracker.String(rootDrivePath),
				IsRootDevice: firecracker.Bool(true),
				IsReadOnly:   firecracker.Bool(true),
			},
		},
	}
	handler := jailer.BuildJailedRootHandler(&cfg, &machineConfig, vmID)

	machine := firecracker.Machine{
		Cfg: machineConfig,
	}
	err = handler.Fn(ctx, &machine)
	assert.NoError(t, err, "jailed handler failed to run")

	_, err = os.Stat(filepath.Join(dir, "config.json"))
	assert.NoError(t, err, "failed to copy runc config")

	_, err = os.Stat(filepath.Join(dir, "rootfs"))
	assert.NoError(t, err, "failed to create rootfs")

	_, err = os.Stat(filepath.Join(dir, "rootfs", filepath.Base(kernelImagePath)))
	assert.NoError(t, err, "failed to create kernel image")

	_, err = os.Stat(filepath.Join(dir, "rootfs", filepath.Base(rootDrivePath)))
	assert.NoError(t, err, "failed to create root drive")
}

func TestMkdirAllWithPermissions(t *testing.T) {
	// requires isolation so we can change uid/gid of files
	internal.RequiresRoot(t)

	tmpdir := t.TempDir()

	existingPath := filepath.Join(tmpdir, "exists")
	existingMode := os.FileMode(0700)
	err := os.Mkdir(existingPath, existingMode)
	require.NoError(t, err, "failed to create existing part of test directory")

	nonExistingPath := filepath.Join(existingPath, "nonexistent")
	newMode := os.FileMode(0755)
	newuid := uint32(123)
	newgid := uint32(456)
	err = mkdirAllWithPermissions(nonExistingPath, newMode, newuid, newgid)
	require.NoError(t, err, "failed to mkdirAllWithPermissions")

	existingPathStat, err := os.Stat(existingPath)
	require.NoError(t, err, "failed to stat pre-existing path")
	assert.Equal(t, existingMode, existingPathStat.Mode().Perm())
	assert.Equal(t, uint32(os.Getuid()), existingPathStat.Sys().(*syscall.Stat_t).Uid)
	assert.Equal(t, uint32(os.Getgid()), existingPathStat.Sys().(*syscall.Stat_t).Gid)

	newlyCreatedPathStat, err := os.Stat(nonExistingPath)
	require.NoError(t, err, "failed to stat newly created path")
	assert.Equal(t, newMode, newlyCreatedPathStat.Mode().Perm())
	assert.Equal(t, newuid, newlyCreatedPathStat.Sys().(*syscall.Stat_t).Uid)
	assert.Equal(t, newgid, newlyCreatedPathStat.Sys().(*syscall.Stat_t).Gid)
}

func TestBindMountToJail(t *testing.T) {
	// The user must be root to call chown.
	internal.RequiresRoot(t)

	dir := t.TempDir()

	f, err := os.Create(filepath.Join(dir, "src1"))
	require.NoError(t, err)
	defer f.Close()

	j := &runcJailer{
		started: false,
		Config:  runcJailerConfig{OCIBundlePath: dir},
	}

	// Create the mount point. The mount point will be used by runc later.
	err = j.bindMountFileToJail(
		filepath.Join(dir, "src1"),
		filepath.Join(dir, "dst1"),
	)
	require.NoError(t, err)

	_, err = os.Stat(filepath.Join(dir, "dst1"))
	require.NoError(t, err)

	// Once runc has been started, it doesn't create a mount point and
	// let the caller know the method cannot be used.
	j.started = true
	err = j.bindMountFileToJail(
		filepath.Join(dir, "src2"),
		filepath.Join(dir, "not-found"),
	)
	require.Error(t, err)
}

func TestFifoHandler(t *testing.T) {
	// Because of chown(2).
	internal.RequiresRoot(t)
	ctx := context.Background()

	testcases := []struct {
		name        string
		logPath     string
		metricsPath string
	}{
		{
			"Different basename",
			"log.fifo",
			"metrics.fifo",
		},
		{
			"Same basename",
			"log/vmid.fifo",
			"metrics/vmid.fifo",
		},
	}

	for _, tc := range testcases {
		tc := tc
		t.Run(tc.name, func(t *testing.T) {
			dir := t.TempDir()

			logPath := filepath.Join(dir, tc.logPath)
			metricsPath := filepath.Join(dir, tc.metricsPath)

			err := os.MkdirAll(filepath.Dir(logPath), 0750)
			require.NoError(t, err)
			err = os.WriteFile(logPath, []byte("log"), 0644)
			require.NoError(t, err)

			err = os.MkdirAll(filepath.Dir(metricsPath), 0750)
			require.NoError(t, err)
			err = os.WriteFile(metricsPath, []byte("metrics"), 0644)
			require.NoError(t, err)

			j := runcJailer{
				Config: runcJailerConfig{
					OCIBundlePath: dir,
				},
			}
			err = os.Mkdir(j.RootPath(), 0750)
			require.NoError(t, err)

			handler := j.BuildLinkFifoHandler()
			machine := firecracker.Machine{Cfg: firecracker.Config{LogPath: logPath, MetricsPath: metricsPath}}
			err = handler.Fn(ctx, &machine)
			require.NoError(t, err)
		})
	}
}

func TestPrepareBindMount(t *testing.T) {
	// Because of chown(2).
	internal.RequiresRoot(t)

	t.Run("no mounts", func(t *testing.T) {
		j := &runcJailer{}
		err := j.prepareBindMounts([]*proto.FirecrackerDriveMount{})
		require.NoError(t, err)
	})

	dir, err := os.MkdirTemp("", t.Name())
	require.NoError(t, err)
	defer os.RemoveAll(dir)

	j := &runcJailer{Config: runcJailerConfig{
		OCIBundlePath: filepath.Join(dir, "bundle"),
		UID:           1234,
		GID:           5678,
	}}

	err = os.WriteFile(dir+"/foobar", []byte("hello"), 0700)
	require.NoError(t, err)

	testcases := []struct {
		name     string
		hostPath string
	}{
		{
			name:     "absolute path",
			hostPath: dir + "/foobar",
		},
		{
			name:     "use dots to access the original directory",
			hostPath: "/../../../../../.." + dir + "/foobar",
		},
	}
	for _, tc := range testcases {
		t.Run(tc.name, func(t *testing.T) {
			err = j.prepareBindMounts([]*proto.FirecrackerDriveMount{{
				HostPath:       tc.hostPath,
				FilesystemType: "ext4",
				VMPath:         "/mnt",
			}})
			require.NoError(t, err)
			stat, err := os.Stat(dir)
			require.NoError(t, err)

			s := stat.Sys().(*syscall.Stat_t)
			assert.Equal(t, 0, int(s.Uid), "UID")
			assert.Equal(t, 0, int(s.Gid), "GID")
		})
	}
}