kernel_windows.go 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  1. // +build windows
  2. package kernel
  3. import (
  4. "fmt"
  5. "unsafe"
  6. "golang.org/x/sys/windows"
  7. )
  8. // VersionInfo holds information about the kernel.
  9. type VersionInfo struct {
  10. kvi string // Version of the kernel (e.g. 6.1.7601.17592 -> 6)
  11. major int // Major part of the kernel version (e.g. 6.1.7601.17592 -> 1)
  12. minor int // Minor part of the kernel version (e.g. 6.1.7601.17592 -> 7601)
  13. build int // Build number of the kernel version (e.g. 6.1.7601.17592 -> 17592)
  14. }
  15. func (k *VersionInfo) String() string {
  16. return fmt.Sprintf("%d.%d %d (%s)", k.major, k.minor, k.build, k.kvi)
  17. }
  18. // GetKernelVersion gets the current kernel version.
  19. func GetKernelVersion() (*VersionInfo, error) {
  20. var (
  21. h windows.Handle
  22. dwVersion uint32
  23. err error
  24. )
  25. KVI := &VersionInfo{"Unknown", 0, 0, 0}
  26. if err = windows.RegOpenKeyEx(windows.HKEY_LOCAL_MACHINE,
  27. windows.StringToUTF16Ptr(`SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\`),
  28. 0,
  29. windows.KEY_READ,
  30. &h); err != nil {
  31. return KVI, err
  32. }
  33. defer windows.RegCloseKey(h)
  34. var buf [1 << 10]uint16
  35. var typ uint32
  36. n := uint32(len(buf) * 2) // api expects array of bytes, not uint16
  37. if err = windows.RegQueryValueEx(h,
  38. windows.StringToUTF16Ptr("BuildLabEx"),
  39. nil,
  40. &typ,
  41. (*byte)(unsafe.Pointer(&buf[0])),
  42. &n); err != nil {
  43. return KVI, err
  44. }
  45. KVI.kvi = windows.UTF16ToString(buf[:])
  46. // Important - docker.exe MUST be manifested for this API to return
  47. // the correct information.
  48. if dwVersion, err = windows.GetVersion(); err != nil {
  49. return KVI, err
  50. }
  51. KVI.major = int(dwVersion & 0xFF)
  52. KVI.minor = int((dwVersion & 0XFF00) >> 8)
  53. KVI.build = int((dwVersion & 0xFFFF0000) >> 16)
  54. return KVI, nil
  55. }