123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105 |
- package system
- import (
- "syscall"
- "unsafe"
- "github.com/Sirupsen/logrus"
- )
- var (
- ntuserApiset = syscall.NewLazyDLL("ext-ms-win-ntuser-window-l1-1-0")
- procGetVersionExW = modkernel32.NewProc("GetVersionExW")
- )
- // 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 = syscall.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
- }
- // 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
- }
- // 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 := syscall.UTF16PtrFromString(commandLine)
- if err != nil {
- return nil, err
- }
- argv, err := syscall.CommandLineToArgv(argsPtr, &argc)
- if err != nil {
- return nil, err
- }
- defer syscall.LocalFree(syscall.Handle(uintptr(unsafe.Pointer(argv))))
- newArgs := make([]string, argc)
- for i, v := range (*argv)[:argc] {
- newArgs[i] = string(syscall.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
- }
|