Initial commit

cmd/cp2guest/cp2guest.go

@@ -0,0 +1,87 @@
+package main
+
+import (
+	"flag"
+	"git.mlody.eu/lmoskala/libguestd/pkg/libguestd"
+	"github.com/digitalocean/go-libvirt"
+	"log"
+	"net/url"
+	"os"
+	"time"
+)
+
+func main() {
+
+	libvirtUri := flag.String("libvirt", "qemu:///system", "libvirt connection URI")
+	domainName := flag.String("domain", "", "domain (instance) name")
+	Src := flag.String("src", "", "source file")
+	Dst := flag.String("dst", "", "destination file")
+	Reverse := flag.Bool("reverse", false, "reverse mode (guest to host)")
+
+	flag.Parse()
+
+	if *domainName == "" {
+		log.Fatalf("Domain name must be specified")
+	}
+	if *Src == "" {
+		log.Fatalf("Source file must be specified")
+	}
+	if *Dst == "" {
+		log.Fatalf("Destination file must be specified")
+	}
+	var data []byte
+	var err error
+	if !*Reverse {
+		data, err = os.ReadFile(*Src)
+		if err != nil {
+			log.Fatalf("error reading src file: %v\n", err)
+		}
+	}
+
+	uri, _ := url.Parse(*libvirtUri)
+	l, err := libvirt.ConnectToURI(uri)
+	if err != nil {
+		log.Fatalf("failed to connect: %v", err)
+	}
+
+	domains, _, err := l.ConnectListAllDomains(1, libvirt.ConnectListDomainsActive)
+	if err != nil {
+		log.Fatalf("failed to list all domains: %v", err)
+	}
+	for _, domain := range domains {
+		if domain.Name != *domainName {
+			continue
+		}
+		err = libguestd.GuestPing(l, domain)
+		if err != nil {
+			log.Printf("Error(GuestPing): %s", err)
+			continue
+		}
+		agentInfo, err := libguestd.GuestInfo(l, domain)
+		if err != nil {
+			log.Printf("Error(GuestInfo): %s", err)
+			continue
+		}
+		Start := time.Now()
+		if *Reverse {
+			data, err = libguestd.GuestFileRead(l, domain, agentInfo, *Src)
+		} else {
+			err = libguestd.GuestFileWrite(l, domain, agentInfo, *Dst, data)
+		}
+		if err != nil {
+			log.Printf("Error(GuestFileWrite): %s", err)
+		} else {
+			duration := time.Since(Start)
+			log.Printf("Copied %d bytes in %.1fs, (%.1f MiB/s)", len(data), duration.Seconds(), float64(len(data))/1024/1024/duration.Seconds())
+		}
+		if *Reverse {
+			err = os.WriteFile(*Dst, data, 0600)
+			if err != nil {
+				log.Printf("error writing file: %v", err)
+			}
+		}
+	}
+	if err = l.Disconnect(); err != nil {
+		log.Fatalf("failed to disconnect: %v", err)
+	}
+}