Initial commit

cmd/guestrun/guestrun.go

@@ -0,0 +1,77 @@
+package main
+
+import (
+	"encoding/base64"
+	"flag"
+	"fmt"
+	"git.mlody.eu/lmoskala/libguestd/pkg/libguestd"
+	"github.com/digitalocean/go-libvirt"
+	"log"
+	"net/url"
+)
+
+func main() {
+	libvirtUri := flag.String("libvirt", "qemu:///system", "libvirt connection URI")
+	domainName := flag.String("domain", "", "domain (instance) name")
+	Cmd := flag.String("cmd", "", "Command (will be run as /bin/sh -c ...)")
+
+	flag.Parse()
+
+	if *domainName == "" {
+		log.Fatalf("Domain name must be specified")
+	}
+	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
+		}
+
+		pid, err := libguestd.GuestExecStart(l, domain, agentInfo, "/bin/sh", []string{"-c", *Cmd}, []string{}, []byte{})
+		if err != nil {
+			log.Printf("Error(GuestExecStart): %s", err)
+			continue
+		}
+		log.Printf("PID: %d", pid)
+
+		res, err := libguestd.GuestWaitForCommand(l, domain, agentInfo, pid)
+		if err != nil {
+			log.Printf("Error(GuestWaitForCommand): %s", err)
+		}
+		log.Printf("Exit status: %d", res.ExitCode)
+		log.Printf("Stdout follows:")
+		dec, _ := base64.StdEncoding.DecodeString(res.OutDataB64)
+		fmt.Println(string(dec))
+		if res.OutTruncated {
+			log.Printf("[Output was truncated]")
+		}
+		log.Printf("Stderr follows:")
+		dec, _ = base64.StdEncoding.DecodeString(res.ErrDataB64)
+		fmt.Println(string(dec))
+		if res.ErrTruncated {
+			log.Printf("[Output was truncated]")
+		}
+
+	}
+	if err = l.Disconnect(); err != nil {
+		log.Fatalf("failed to disconnect: %v", err)
+	}
+}