Source release 16.3.0

oemcrypto/test/oec_device_features.cpp

@@ -8,58 +8,15 @@
 
 #include <stdio.h>
 
-#ifdef _WIN32
-# include <windows.h>
-#else
-# include <sys/types.h>
-# include <unistd.h>
-#endif
-
 #include <cstring>
 
 #include "oec_test_data.h"
+#include "test_sleep.h"
 
 namespace wvoec {
 
 DeviceFeatures global_features;
 
-bool CanChangeTime() {
-#ifdef _WIN32
-  LUID desired_id;
-  if (!LookupPrivilegeValue(nullptr, SE_SYSTEMTIME_NAME, &desired_id))
-    return false;
-  HANDLE token;
-  if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ALL_ACCESS, &token))
-    return false;
-  std::unique_ptr<void, decltype(&CloseHandle)> safe_token(token, &CloseHandle);
-
-  // This queries all the permissions given to the token to determine if we can
-  // change the system time.  Note this is subtly different from PrivilegeCheck
-  // as that only checks "enabled" privileges; even with admin rights, the
-  // privilege is default disabled, even when granted.
-
-  DWORD size = 0;
-  // Determine how big we need to allocate first.
-  GetTokenInformation(token, TokenPrivileges, nullptr, 0, &size);
-  // Since TOKEN_PRIVILEGES uses a variable-length array, we need to use malloc
-  std::unique_ptr<TOKEN_PRIVILEGES, decltype(&free)> privileges(
-      (TOKEN_PRIVILEGES*)malloc(size), &free);
-  if (privileges && GetTokenInformation(token, TokenPrivileges,
-                                        privileges.get(), size, &size)) {
-    for (int i = 0; i < privileges->PrivilegeCount; i++) {
-      if (privileges->Privileges[i].Luid.HighPart == desired_id.HighPart &&
-          privileges->Privileges[i].Luid.LowPart == desired_id.LowPart) {
-        return true;
-      }
-    }
-  }
-
-  return false;
-#else
-  return getuid() == 0;
-#endif
-}
-
 void DeviceFeatures::Initialize() {
   if (initialized_) return;
   uses_keybox = false;
@@ -188,8 +145,11 @@ std::string DeviceFeatures::RestrictFilter(const std::string& initial_filter) {
   // clang-format on
   // Some tests may require root access. If user is not root, filter these tests
   // out.
-  if (!CanChangeTime()) {
+  if (!wvcdm::TestSleep::CanChangeSystemTime()) {
+    printf("Filtering out TimeRollbackPrevention.\n");
     FilterOut(&filter, "*TimeRollbackPrevention*");
+  } else {
+    printf("Can change time. I will run TimeRollbackPrevention.\n");
   }
   // Performance tests take a long time.  Filter them out if they are not
   // specifically requested.