mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-21 14:20:23 -04:00
128 lines
3.8 KiB
Go
128 lines
3.8 KiB
Go
package system // import "github.com/docker/docker/pkg/system"
|
|
|
|
import (
|
|
"fmt"
|
|
"unsafe"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"golang.org/x/sys/windows"
|
|
)
|
|
|
|
var (
|
|
ntuserApiset = windows.NewLazyDLL("ext-ms-win-ntuser-window-l1-1-0")
|
|
procGetVersionExW = modkernel32.NewProc("GetVersionExW")
|
|
procGetProductInfo = modkernel32.NewProc("GetProductInfo")
|
|
)
|
|
|
|
// OSVersion is a wrapper for Windows version information
|
|
// https://msdn.microsoft.com/en-us/library/windows/desktop/ms724439(v=vs.85).aspx
|
|
type OSVersion struct {
|
|
Version uint32
|
|
MajorVersion uint8
|
|
MinorVersion uint8
|
|
Build uint16
|
|
}
|
|
|
|
// https://msdn.microsoft.com/en-us/library/windows/desktop/ms724833(v=vs.85).aspx
|
|
type osVersionInfoEx struct {
|
|
OSVersionInfoSize uint32
|
|
MajorVersion uint32
|
|
MinorVersion uint32
|
|
BuildNumber uint32
|
|
PlatformID uint32
|
|
CSDVersion [128]uint16
|
|
ServicePackMajor uint16
|
|
ServicePackMinor uint16
|
|
SuiteMask uint16
|
|
ProductType byte
|
|
Reserve byte
|
|
}
|
|
|
|
// GetOSVersion gets the operating system version on Windows. Note that
|
|
// docker.exe must be manifested to get the correct version information.
|
|
func GetOSVersion() OSVersion {
|
|
var err error
|
|
osv := OSVersion{}
|
|
osv.Version, err = windows.GetVersion()
|
|
if err != nil {
|
|
// GetVersion never fails.
|
|
panic(err)
|
|
}
|
|
osv.MajorVersion = uint8(osv.Version & 0xFF)
|
|
osv.MinorVersion = uint8(osv.Version >> 8 & 0xFF)
|
|
osv.Build = uint16(osv.Version >> 16)
|
|
return osv
|
|
}
|
|
|
|
func (osv OSVersion) ToString() string {
|
|
return fmt.Sprintf("%d.%d.%d", osv.MajorVersion, osv.MinorVersion, osv.Build)
|
|
}
|
|
|
|
// IsWindowsClient returns true if the SKU is client
|
|
// @engine maintainers - this function should not be removed or modified as it
|
|
// is used to enforce licensing restrictions on Windows.
|
|
func IsWindowsClient() bool {
|
|
osviex := &osVersionInfoEx{OSVersionInfoSize: 284}
|
|
r1, _, err := procGetVersionExW.Call(uintptr(unsafe.Pointer(osviex)))
|
|
if r1 == 0 {
|
|
logrus.Warnf("GetVersionExW failed - assuming server SKU: %v", err)
|
|
return false
|
|
}
|
|
const verNTWorkstation = 0x00000001
|
|
return osviex.ProductType == verNTWorkstation
|
|
}
|
|
|
|
// IsIoTCore returns true if the currently running image is based off of
|
|
// Windows 10 IoT Core.
|
|
// @engine maintainers - this function should not be removed or modified as it
|
|
// is used to enforce licensing restrictions on Windows.
|
|
func IsIoTCore() bool {
|
|
var returnedProductType uint32
|
|
r1, _, err := procGetProductInfo.Call(6, 1, 0, 0, uintptr(unsafe.Pointer(&returnedProductType)))
|
|
if r1 == 0 {
|
|
logrus.Warnf("GetProductInfo failed - assuming this is not IoT: %v", err)
|
|
return false
|
|
}
|
|
const productIoTUAP = 0x0000007B
|
|
const productIoTUAPCommercial = 0x00000083
|
|
return returnedProductType == productIoTUAP || returnedProductType == productIoTUAPCommercial
|
|
}
|
|
|
|
// Unmount is a platform-specific helper function to call
|
|
// the unmount syscall. Not supported on Windows
|
|
func Unmount(dest string) error {
|
|
return nil
|
|
}
|
|
|
|
// CommandLineToArgv wraps the Windows syscall to turn a commandline into an argument array.
|
|
func CommandLineToArgv(commandLine string) ([]string, error) {
|
|
var argc int32
|
|
|
|
argsPtr, err := windows.UTF16PtrFromString(commandLine)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
argv, err := windows.CommandLineToArgv(argsPtr, &argc)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer windows.LocalFree(windows.Handle(uintptr(unsafe.Pointer(argv))))
|
|
|
|
newArgs := make([]string, argc)
|
|
for i, v := range (*argv)[:argc] {
|
|
newArgs[i] = string(windows.UTF16ToString((*v)[:]))
|
|
}
|
|
|
|
return newArgs, nil
|
|
}
|
|
|
|
// HasWin32KSupport determines whether containers that depend on win32k can
|
|
// run on this machine. Win32k is the driver used to implement windowing.
|
|
func HasWin32KSupport() bool {
|
|
// For now, check for ntuser API support on the host. In the future, a host
|
|
// may support win32k in containers even if the host does not support ntuser
|
|
// APIs.
|
|
return ntuserApiset.Load() == nil
|
|
}
|