165 lines
3.8 KiB
Go
165 lines
3.8 KiB
Go
package vmware
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
// Workstation9Driver is a driver that can run VMware Workstation 9
|
|
// on non-Windows platforms.
|
|
type Workstation9Driver struct {
|
|
AppPath string
|
|
VdiskManagerPath string
|
|
VmrunPath string
|
|
}
|
|
|
|
func (d *Workstation9Driver) CompactDisk(diskPath string) error {
|
|
defragCmd := exec.Command(d.VdiskManagerPath, "-d", diskPath)
|
|
if _, _, err := d.runAndLog(defragCmd); err != nil {
|
|
return err
|
|
}
|
|
|
|
shrinkCmd := exec.Command(d.VdiskManagerPath, "-k", diskPath)
|
|
if _, _, err := d.runAndLog(shrinkCmd); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *Workstation9Driver) CreateDisk(output string, size string) error {
|
|
cmd := exec.Command(d.VdiskManagerPath, "-c", "-s", size, "-a", "lsilogic", "-t", "1", output)
|
|
if _, _, err := d.runAndLog(cmd); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *Workstation9Driver) IsRunning(vmxPath string) (bool, error) {
|
|
vmxPath, err := filepath.Abs(vmxPath)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
cmd := exec.Command(d.VmrunPath, "-T", "ws", "list")
|
|
stdout, _, err := d.runAndLog(cmd)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
for _, line := range strings.Split(stdout, "\n") {
|
|
if line == vmxPath {
|
|
return true, nil
|
|
}
|
|
}
|
|
|
|
return false, nil
|
|
}
|
|
|
|
func (d *Workstation9Driver) Start(vmxPath string, headless bool) error {
|
|
guiArgument := "gui"
|
|
if headless {
|
|
guiArgument = "nogui"
|
|
}
|
|
|
|
cmd := exec.Command(d.VmrunPath, "-T", "ws", "start", vmxPath, guiArgument)
|
|
if _, _, err := d.runAndLog(cmd); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *Workstation9Driver) Stop(vmxPath string) error {
|
|
cmd := exec.Command(d.VmrunPath, "-T", "ws", "stop", vmxPath, "hard")
|
|
if _, _, err := d.runAndLog(cmd); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *Workstation9Driver) Verify() error {
|
|
var err error
|
|
if d.AppPath == "" {
|
|
if d.AppPath, err = workstationFindVMware(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if d.VmrunPath == "" {
|
|
if d.VmrunPath, err = workstationFindVmrun(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if d.VdiskManagerPath == "" {
|
|
if d.VdiskManagerPath, err = workstationFindVdiskManager(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
log.Printf("VMware app path: %s", d.AppPath)
|
|
log.Printf("vmrun path: %s", d.VmrunPath)
|
|
log.Printf("vdisk-manager path: %s", d.VdiskManagerPath)
|
|
|
|
if _, err := os.Stat(d.AppPath); err != nil {
|
|
return fmt.Errorf("VMware application not found: %s", d.AppPath)
|
|
}
|
|
|
|
if _, err := os.Stat(d.VmrunPath); err != nil {
|
|
return fmt.Errorf("'vmrun' application not found: %s", d.VmrunPath)
|
|
}
|
|
|
|
if _, err := os.Stat(d.VdiskManagerPath); err != nil {
|
|
return fmt.Errorf("'vmrun' application not found: %s", d.VdiskManagerPath)
|
|
}
|
|
|
|
// Check to see if it APPEARS to be licensed.
|
|
if err := workstationCheckLicense(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *Workstation9Driver) ToolsIsoPath(flavor string) string {
|
|
return workstationToolsIsoPath(flavor)
|
|
}
|
|
|
|
func (d *Workstation9Driver) DhcpLeasesPath(device string) string {
|
|
return workstationDhcpLeasesPath(device)
|
|
}
|
|
|
|
func (d *Workstation9Driver) runAndLog(cmd *exec.Cmd) (string, string, error) {
|
|
var stdout, stderr bytes.Buffer
|
|
|
|
log.Printf("Executing: %s %v", cmd.Path, cmd.Args[1:])
|
|
cmd.Stdout = &stdout
|
|
cmd.Stderr = &stderr
|
|
err := cmd.Run()
|
|
|
|
stdoutString := strings.TrimSpace(stdout.String())
|
|
stderrString := strings.TrimSpace(stderr.String())
|
|
|
|
if _, ok := err.(*exec.ExitError); ok {
|
|
err = fmt.Errorf("VMware error: %s", stderrString)
|
|
}
|
|
|
|
log.Printf("stdout: %s", stdoutString)
|
|
log.Printf("stderr: %s", stderrString)
|
|
|
|
// Replace these for Windows, we only want to deal with Unix
|
|
// style line endings.
|
|
returnStdout := strings.Replace(stdout.String(), "\r\n", "\n", -1)
|
|
returnStderr := strings.Replace(stderr.String(), "\r\n", "\n", -1)
|
|
|
|
return returnStdout, returnStderr, err
|
|
}
|