quickpassthrough/internal/ls_iommu_downloader/ls_iommu_downloader.go

package ls_iommu_downloader

import (
	"crypto/sha256"
	"encoding/hex"
	"encoding/json"
	"fmt"
	"io"
	"log"
	"net/http"
	"os"
	"strings"
	"time"

	"github.com/HikariKnight/ls-iommu/pkg/errorcheck"
	"github.com/HikariKnight/quickpassthrough/pkg/fileio"
	"github.com/HikariKnight/quickpassthrough/pkg/untar"
	"github.com/cavaliergopher/grab/v3"
)

// Generated from github API response using https://mholt.github.io/json-to-go/
type Response struct {
	URL       string `json:"url"`
	AssetsURL string `json:"assets_url"`
	UploadURL string `json:"upload_url"`
	HTMLURL   string `json:"html_url"`
	ID        int    `json:"id"`
	Author    struct {
		Login             string `json:"login"`
		ID                int    `json:"id"`
		NodeID            string `json:"node_id"`
		AvatarURL         string `json:"avatar_url"`
		GravatarID        string `json:"gravatar_id"`
		URL               string `json:"url"`
		HTMLURL           string `json:"html_url"`
		FollowersURL      string `json:"followers_url"`
		FollowingURL      string `json:"following_url"`
		GistsURL          string `json:"gists_url"`
		StarredURL        string `json:"starred_url"`
		SubscriptionsURL  string `json:"subscriptions_url"`
		OrganizationsURL  string `json:"organizations_url"`
		ReposURL          string `json:"repos_url"`
		EventsURL         string `json:"events_url"`
		ReceivedEventsURL string `json:"received_events_url"`
		Type              string `json:"type"`
		SiteAdmin         bool   `json:"site_admin"`
	} `json:"author"`
	NodeID          string    `json:"node_id"`
	TagName         string    `json:"tag_name"`
	TargetCommitish string    `json:"target_commitish"`
	Name            string    `json:"name"`
	Draft           bool      `json:"draft"`
	Prerelease      bool      `json:"prerelease"`
	CreatedAt       time.Time `json:"created_at"`
	PublishedAt     time.Time `json:"published_at"`
	Assets          []struct {
		URL      string `json:"url"`
		ID       int    `json:"id"`
		NodeID   string `json:"node_id"`
		Name     string `json:"name"`
		Label    string `json:"label"`
		Uploader struct {
			Login             string `json:"login"`
			ID                int    `json:"id"`
			NodeID            string `json:"node_id"`
			AvatarURL         string `json:"avatar_url"`
			GravatarID        string `json:"gravatar_id"`
			URL               string `json:"url"`
			HTMLURL           string `json:"html_url"`
			FollowersURL      string `json:"followers_url"`
			FollowingURL      string `json:"following_url"`
			GistsURL          string `json:"gists_url"`
			StarredURL        string `json:"starred_url"`
			SubscriptionsURL  string `json:"subscriptions_url"`
			OrganizationsURL  string `json:"organizations_url"`
			ReposURL          string `json:"repos_url"`
			EventsURL         string `json:"events_url"`
			ReceivedEventsURL string `json:"received_events_url"`
			Type              string `json:"type"`
			SiteAdmin         bool   `json:"site_admin"`
		} `json:"uploader"`
		ContentType        string    `json:"content_type"`
		State              string    `json:"state"`
		Size               int       `json:"size"`
		DownloadCount      int       `json:"download_count"`
		CreatedAt          time.Time `json:"created_at"`
		UpdatedAt          time.Time `json:"updated_at"`
		BrowserDownloadURL string    `json:"browser_download_url"`
	} `json:"assets"`
	TarballURL string `json:"tarball_url"`
	ZipballURL string `json:"zipball_url"`
	Body       string `json:"body"`
}

func CheckLsIOMMU() {
	// Check the API for releases
	resp, err := http.Get("https://api.github.com/repos/hikariknight/ls-iommu/releases/latest")
	errorcheck.ErrorCheck(err)

	// Close the response when function ends
	defer resp.Body.Close()

	// Get the response body
	body, err := io.ReadAll(resp.Body)
	errorcheck.ErrorCheck(err)

	var result Response
	if err := json.Unmarshal(body, &result); err != nil {
		fmt.Println("Cant decode JSON")
	}

	// Make the directory for ls-iommu if it does not exist
	path := "utils"
	if !fileio.FileExist(path) {
		err := os.Mkdir(path, os.ModePerm)
		errorcheck.ErrorCheck(err)
	}

	// Generate the download url
	downloadUrl := fmt.Sprintf(
		"https://github.com/HikariKnight/ls-iommu/releases/download/%s/ls-iommu_Linux_x86_64.tar.gz",
		result.TagName,
	)

	// Generate checksums.txt url
	checkSumsUrl := fmt.Sprintf(
		"https://github.com/HikariKnight/ls-iommu/releases/download/%s/checksums.txt",
		result.TagName,
	)

	fileName := fmt.Sprintf("%s/ls-iommu_Linux_x86_64.tar.gz", path)

	// Get the checksum data
	checksums, err := http.Get(checkSumsUrl)
	errorcheck.ErrorCheck(err)
	defer checksums.Body.Close()
	checksums_txt, err := io.ReadAll(checksums.Body)
	errorcheck.ErrorCheck(err)

	// Check if the tar.gz exists
	if !fileio.FileExist(fileName) {
		downloadNewVersion(path, fileName, downloadUrl)
		if checkSum(string(checksums_txt), fileName) {
			err = untar.Untar(fmt.Sprintf("%s/", path), fileName)
			errorcheck.ErrorCheck(err)
		}
	} else {
		if !checkSum(string(checksums_txt), fileName) {
			downloadNewVersion(path, fileName, downloadUrl)
			err = untar.Untar(fmt.Sprintf("%s/", path), fileName)
			errorcheck.ErrorCheck(err)
		}
	}
}

func checkSum(checksums string, fileName string) bool {
	r, err := os.Open(fileName)
	errorcheck.ErrorCheck(err)
	defer r.Close()

	hasher := sha256.New()
	if _, err := io.Copy(hasher, r); err != nil {
		log.Fatal(err)
	}
	value := hex.EncodeToString(hasher.Sum(nil))

	return strings.Contains(checksums, value)
}

func downloadNewVersion(path, fileName, downloadUrl string) {
	// Create a request
	grabClient := grab.NewClient()
	req, _ := grab.NewRequest(fileName, downloadUrl)

	// Remove old archive
	os.Remove(fileName)

	// Download ls-iommu
	download := grabClient.Do(req)

	// check for errors
	if err := download.Err(); err != nil {
		fmt.Fprintf(os.Stderr, "Download failed: %v\n", err)
		if !fileio.FileExist("utils/ls-iommu") {
			log.Fatal("If the above error is 404, then we could not communicate with the GitHub API\n Please manually download and extract ls-iommu to: utils/\nYou can download it from: https://github.com/HikariKnight/ls-iommu/releases")
		} else {
			fmt.Println("Existing ls-iommu binary detected in \"utils/\", will use that instead as the GitHub API did not respond.")
		}
	} else {
		fmt.Printf("Download saved to ./%v \n", download.Filename)
	}
}