mirror of https://github.com/docker/docs.git
115 lines
2.4 KiB
Go
115 lines
2.4 KiB
Go
package hyperv
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"errors"
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"fmt"
|
|
|
|
"github.com/docker/machine/libmachine/log"
|
|
)
|
|
|
|
var powershell string
|
|
|
|
var (
|
|
ErrPowerShellNotFound = errors.New("Powershell was not found in the path")
|
|
ErrNotAdministrator = errors.New("Hyper-v commands have to be run as an Administrator")
|
|
ErrNotInstalled = errors.New("Hyper-V PowerShell Module is not available")
|
|
)
|
|
|
|
func init() {
|
|
powershell, _ = exec.LookPath("powershell.exe")
|
|
}
|
|
|
|
func cmdOut(args ...string) (string, error) {
|
|
args = append([]string{"-NoProfile", "-NonInteractive"}, args...)
|
|
cmd := exec.Command(powershell, args...)
|
|
log.Debugf("[executing ==>] : %v %v", powershell, strings.Join(args, " "))
|
|
var stdout bytes.Buffer
|
|
var stderr bytes.Buffer
|
|
cmd.Stdout = &stdout
|
|
cmd.Stderr = &stderr
|
|
err := cmd.Run()
|
|
log.Debugf("[stdout =====>] : %s", stdout.String())
|
|
log.Debugf("[stderr =====>] : %s", stderr.String())
|
|
return stdout.String(), err
|
|
}
|
|
|
|
func cmd(args ...string) error {
|
|
_, err := cmdOut(args...)
|
|
return err
|
|
}
|
|
|
|
func parseLines(stdout string) []string {
|
|
resp := []string{}
|
|
|
|
s := bufio.NewScanner(strings.NewReader(stdout))
|
|
for s.Scan() {
|
|
resp = append(resp, s.Text())
|
|
}
|
|
|
|
return resp
|
|
}
|
|
|
|
func hypervAvailable() error {
|
|
stdout, err := cmdOut("@(Get-Command Get-VM).ModuleName")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
resp := parseLines(stdout)
|
|
if resp[0] != "Hyper-V" {
|
|
return ErrNotInstalled
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func isAdministrator() (bool, error) {
|
|
hypervAdmin := isHypervAdministrator()
|
|
|
|
if hypervAdmin {
|
|
return true, nil
|
|
}
|
|
|
|
windowsAdmin, err := isWindowsAdministrator()
|
|
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
return windowsAdmin, nil
|
|
}
|
|
|
|
func isHypervAdministrator() bool {
|
|
stdout, err := cmdOut(`@([Security.Principal.WindowsPrincipal][Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole("Hyper-V Administrators")`)
|
|
if err != nil {
|
|
log.Debug(err)
|
|
return false
|
|
}
|
|
|
|
resp := parseLines(stdout)
|
|
return resp[0] == "True"
|
|
}
|
|
|
|
func isWindowsAdministrator() (bool, error) {
|
|
stdout, err := cmdOut(`@([Security.Principal.WindowsPrincipal][Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole] "Administrator")`)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
resp := parseLines(stdout)
|
|
return resp[0] == "True", nil
|
|
}
|
|
|
|
func quote(text string) string {
|
|
return fmt.Sprintf("'%s'", text)
|
|
}
|
|
|
|
func toMb(value int) string {
|
|
return fmt.Sprintf("%dMB", value)
|
|
}
|