| Index: chrome/installer/util/scoped_token_privilege.h
|
| diff --git a/chrome/installer/util/scoped_token_privilege.h b/chrome/installer/util/scoped_token_privilege.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..3def5eb6b8bdb814fec31745406e1f527b370169
|
| --- /dev/null
|
| +++ b/chrome/installer/util/scoped_token_privilege.h
|
| @@ -0,0 +1,46 @@
|
| +// Copyright 2017 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.
|
| +
|
| +#ifndef CHROME_INSTALLER_UTIL_SCOPED_TOKEN_PRIVILEGE_H_
|
| +#define CHROME_INSTALLER_UTIL_SCOPED_TOKEN_PRIVILEGE_H_
|
| +
|
| +#include <windows.h>
|
| +
|
| +#include "base/macros.h"
|
| +#include "base/win/scoped_handle.h"
|
| +
|
| +namespace installer {
|
| +
|
| +// This class is available for Windows only and will enable the privilege
|
| +// defined by |privilege_name| on the current process' token. The privilege will
|
| +// be disabled upon the ScopedTokenPrivilege's destruction (unless it was
|
| +// already enabled when the ScopedTokenPrivilege object was constructed).
|
| +// Some privileges might require admin rights to be enabled (check is_enabled()
|
| +// to know whether |privilege_name| was successfully enabled).
|
| +class ScopedTokenPrivilege {
|
| + public:
|
| + explicit ScopedTokenPrivilege(const wchar_t* privilege_name);
|
| + ~ScopedTokenPrivilege();
|
| +
|
| + // Always returns true unless the privilege could not be enabled.
|
| + bool is_enabled() const { return is_enabled_; }
|
| +
|
| + private:
|
| + // Always true unless the privilege could not be enabled.
|
| + bool is_enabled_;
|
| +
|
| + // A scoped handle to the current process' token. This will be closed
|
| + // preemptively should enabling the privilege fail in the constructor.
|
| + base::win::ScopedHandle token_;
|
| +
|
| + // The previous state of the privilege this object is responsible for. As set
|
| + // by AdjustTokenPrivileges() upon construction.
|
| + TOKEN_PRIVILEGES previous_privileges_;
|
| +
|
| + DISALLOW_IMPLICIT_CONSTRUCTORS(ScopedTokenPrivilege);
|
| +};
|
| +
|
| +} // namespace installer
|
| +
|
| +#endif // CHROME_INSTALLER_UTIL_SCOPED_TOKEN_PRIVILEGE_H_
|
|
|