quickpassthrough/internal/configs/config_initramfstools.go

package configs

import (
	"bufio"
	"fmt"
	"os"
	"regexp"
	"strings"

	"github.com/HikariKnight/quickpassthrough/internal/common"
	"github.com/HikariKnight/quickpassthrough/pkg/fileio"
)

// Special function to read the header of a file (reads the first N lines)
func initramfs_readHeader(lines int, fileName string) string {
	// Open the file
	f, err := os.Open(fileName)
	common.ErrorCheck(err, fmt.Sprintf("Error opening %s", fileName))
	defer f.Close()

	header_re := regexp.MustCompile(`^#`)
	var header []string

	// Make a new scanner
	scanner := bufio.NewScanner(f)

	// Read the first N lines
	for i := 0; i < lines; i++ {
		scanner.Scan()
		if header_re.MatchString(scanner.Text()) {
			header = append(header, scanner.Text())
		}
	}

	// Return the header
	return fmt.Sprintf("%s\n", strings.Join(header, "\n"))
}

// Reads the system file and copies over the content while inserting the vfio modules
// Takes the config file as argument
func initramfs_addModules(conffile string) {
	// Make a regex to get the system path instead of the config path
	syspath_re := regexp.MustCompile(`^config`)

	// Make a regex to skip specific modules and comments
	skipmodules_re := regexp.MustCompile(`(^#|vendor-reset|vfio|vfio_pci|vfio_iommu_type1|vfio_virqfd)`)

	// Get the syspath
	syspath := syspath_re.ReplaceAllString(conffile, "")

	// Open the system file for reading
	sysfile, err := os.Open(syspath)
	common.ErrorCheck(err, fmt.Sprintf("Error opening file for reading %s", syspath))
	defer sysfile.Close()

	// Check if user has vendor-reset installed/enabled and make sure that is first
	content := fileio.ReadFile(syspath)
	if strings.Contains(content, "vendor-reset") {
		fileio.AppendContent("vendor-reset\n", conffile)
	}

	// Write the vfio modules
	fileio.AppendContent(
		fmt.Sprint(
			"# Added by quickpassthrough #\n",
			fmt.Sprintf(
				"%s\n",
				strings.Join(vfio_modules(), "\n"),
			),
			"#############################\n",
		),
		conffile,
	)

	// Scan the system file line by line
	scanner := bufio.NewScanner(sysfile)
	for scanner.Scan() {
		// If this is not a line we skip then
		if !skipmodules_re.MatchString(scanner.Text()) {
			// Add the module to our config
			fileio.AppendContent(fmt.Sprintf("%s\n", scanner.Text()), conffile)
		}
	}
}