[Telemetry] Update the code that detect windows version to use CurrentMajorVersionNumber (Reland)

telemetry/telemetry/internal/platform/win_platform_backend.py

 # Copyright 2013 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import atexit
 import collections
 import contextlib
 import ctypes
 import logging
 import os
@@ skipping 18 matching lines @@
 try:
   import pywintypes  # pylint: disable=import-error
   import win32api  # pylint: disable=import-error
   from win32com.shell import shell  # pylint: disable=no-name-in-module
   from win32com.shell import shellcon  # pylint: disable=no-name-in-module
   import win32con  # pylint: disable=import-error
   import win32file  # pylint: disable=import-error
   import win32gui  # pylint: disable=import-error
   import win32pipe  # pylint: disable=import-error
   import win32process  # pylint: disable=import-error
+  try:
+    import winreg  # pylint: disable=import-error
+  except ImportError:
+    import _winreg as winreg  # pylint: disable=import-error
   import win32security  # pylint: disable=import-error
 except ImportError:
   pywintypes = None
   shell = None
   shellcon = None
   win32api = None
   win32con = None
   win32file = None
   win32gui = None
   win32pipe = None
   win32process = None
   win32security = None
+  winreg = None
 
 
 def _InstallWinRing0():
   """WinRing0 is used for reading MSRs."""
   executable_dir = os.path.dirname(sys.executable)
 
   python_is_64_bit = sys.maxsize > 2 ** 32
   dll_file_name = 'WinRing0x64.dll' if python_is_64_bit else 'WinRing0.dll'
   dll_path = os.path.join(executable_dir, dll_file_name)
 
@@ skipping 170 matching lines @@
   @decorators.Cache
   def GetOSVersionName(self):
     os_version = platform.uname()[3]
 
     if os_version.startswith('5.1.'):
       return os_version_module.XP
     if os_version.startswith('6.0.'):
       return os_version_module.VISTA
     if os_version.startswith('6.1.'):
       return os_version_module.WIN7
-    if os_version.startswith('6.2.'):
+    # The version of python.exe we commonly use (2.7) is only manifested as
+    # being compatible with Windows versions up to 8. Therefore Windows *lies*
+    # to python about the version number to keep it runnable on Windows 10.
+    key_name = r'Software\Microsoft\Windows NT\CurrentVersion'
+    key = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, key_name)
+    try:
+      value, _ = winreg.QueryValueEx(key, 'CurrentMajorVersionNumber')
+    except WindowsErrror:  # pylint: disable=undefined-variable
+      value = None
+    finally:
+      key.Close()
+    if value == 10:
+      return os_version_module.WIN10
+    elif os_version.startswith('6.2.'):
       return os_version_module.WIN8
-    if os_version.startswith('10.'):
-      return os_version_module.WIN10
-
-    raise NotImplementedError('Unknown win version %s.' % os_version)
+    elif os_version.startswith('6.3.'):
+      return os_version_module.WIN81
+    raise NotImplementedError(
+        'Unknown win version: %s, CurrentMajorVersionNumber: %s' %
+        (os_version, value))
 
   def CanFlushIndividualFilesFromSystemCache(self):
     return True
 
   def _GetWin32ProcessInfo(self, func, pid):
     mask = (win32con.PROCESS_QUERY_INFORMATION |
             win32con.PROCESS_VM_READ)
     handle = None
     try:
       handle = win32api.OpenProcess(mask, False, pid)
@@ skipping 154 matching lines @@
       return True
     hwnds = []
     win32gui.EnumWindows(find_chrome_windows, hwnds)
     if hwnds:
       for hwnd in hwnds:
         win32gui.SendMessage(hwnd, win32con.WM_CLOSE, 0, 0)
       return True
     else:
       logging.info('Did not find any windows owned by target process')
     return False