summaryrefslogtreecommitdiffstats
path: root/toolkit/components/maintenanceservice/tests/gtest/ServiceStartInteractiveOnly.cpp
blob: beff8e8e472bdb3c7b473a7782ddba995872152b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#include "gtest/gtest.h"

#include <stdio.h>
#include <windows.h>
#include <sddl.h>

#include "nsWindowsHelpers.h"

TEST(MaintenanceServiceTest, ServiceStartInteractiveOnly)
{
  // First, make a restricted token that excludes the Interactive group.
  SID_AND_ATTRIBUTES sid;
  DWORD SIDSize = SECURITY_MAX_SID_SIZE;
  sid.Sid = LocalAlloc(LMEM_FIXED, SIDSize);
  // Automatically free the SID when we are done with it.
  UniqueSidPtr uniqueSid(sid.Sid);
  ASSERT_TRUE(sid.Sid);

  BOOL success =
      CreateWellKnownSid(WinInteractiveSid, nullptr, sid.Sid, &SIDSize);
  ASSERT_TRUE(success);

  HANDLE primaryToken;
  success =
      OpenProcessToken(GetCurrentProcess(), TOKEN_ALL_ACCESS, &primaryToken);
  // Automatically close the token when we are done with it.
  nsAutoHandle uniquePrimaryToken(primaryToken);
  ASSERT_TRUE(success);

  HANDLE restrictedToken;
  success = CreateRestrictedToken(primaryToken, 0, 1, &sid, 0, nullptr, 0,
                                  nullptr, &restrictedToken);
  // Automatically close the token when we are done with it.
  nsAutoHandle uniqueRestrictedToken(restrictedToken);
  ASSERT_TRUE(success);

  success = ImpersonateLoggedOnUser(restrictedToken);
  ASSERT_TRUE(success);

  SC_HANDLE scmHandle =
      OpenSCManagerW(L"127.0.0.1", nullptr, SC_MANAGER_CONNECT);
  // Automatically close the SCM when we are done with it.
  nsAutoServiceHandle uniqueScmHandle(scmHandle);
  ASSERT_TRUE(scmHandle);

  SC_HANDLE serviceHandle =
      OpenServiceW(scmHandle, L"MozillaMaintenance", SERVICE_START);
  // Automatically close the SCM when we are done with it.
  nsAutoServiceHandle uniqueServiceHandle(serviceHandle);
  ASSERT_FALSE(serviceHandle);
  ASSERT_EQ(GetLastError(), static_cast<DWORD>(ERROR_ACCESS_DENIED));
}