pkg/utils/cp/cp.go

package cp

import (
	"fmt"
	"io"
	"os"

	ctrl "sigs.k8s.io/controller-runtime"
)

var utilLogger = ctrl.Log.WithName("cp-util")

var (
	EKS_CLI_BINARY    = "aws-eks-na-cli"
	EKS_V6_CLI_BINARY = "aws-eks-na-cli-v6"
)

func cp(src, dst string) error {
	sourceFileStat, err := os.Stat(src)
	if err != nil {
		return err
	}

	if !sourceFileStat.Mode().IsRegular() {
		return fmt.Errorf("%s is not a regular file", src)
	}

	source, err := os.Open(src)
	if err != nil {
		return err
	}
	defer source.Close()

	destination, err := os.Create(dst)
	if err != nil {
		return err
	}
	defer destination.Close()
	_, err = io.Copy(destination, source)
	return err
}

func CopyFile(src, dst string) (err error) {
	dstTmp := fmt.Sprintf("%s.tmp", dst)
	if err := cp(src, dstTmp); err != nil {
		return fmt.Errorf("failed to copy file: %s", err)
	}

	err = os.Rename(dstTmp, dst)
	if err != nil {
		return fmt.Errorf("failed to rename file: %s", err)
	}

	si, err := os.Stat(src)
	if err != nil {
		return fmt.Errorf("failed to stat file: %s", err)
	}
	err = os.Chmod(dst, si.Mode())
	if err != nil {
		return fmt.Errorf("failed to chmod file: %s", err)
	}

	return nil
}

func InstallBPFBinaries(pluginBins []string, hostCNIBinPath string) error {
	utilLogger.Info("Let's install BPF Binaries on to the host path.....")
	for _, plugin := range pluginBins {
		targetPlugin := plugin

		// CLI binary should always refer to aws-eks-na-cli
		if plugin == EKS_V6_CLI_BINARY {
			targetPlugin = EKS_CLI_BINARY
		}

		target := fmt.Sprintf("%s%s", hostCNIBinPath, targetPlugin)
		source := fmt.Sprintf("%s", plugin)
		utilLogger.Info("Installing BPF Binary..", "target", target, "source", source)

		if err := CopyFile(source, target); err != nil {
			utilLogger.Info("Failed to install", "target", target, "error", err)
		}
		utilLogger.Info("Successfully installed - ", "binary", target)
	}
	return nil
}