93 lines
2.5 KiB
C++
93 lines
2.5 KiB
C++
|
// Copyright (c) 2011 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.
|
||
|
|
|
||
|
|
#include "base/process/process.h"
|
||
|
|
|
||
|
|
#include "base/logging.h"
|
||
|
|
#include "base/memory/scoped_ptr.h"
|
||
|
|
#include "base/win/windows_version.h"
|
||
|
|
|
||
|
|
namespace base {
|
||
|
|
|
||
|
|
void Process::Close() {
|
||
|
|
if (!process_)
|
||
|
|
return;
|
||
|
|
|
||
|
|
// Don't call CloseHandle on a pseudo-handle.
|
||
|
|
if (process_ != ::GetCurrentProcess())
|
||
|
|
::CloseHandle(process_);
|
||
|
|
|
||
|
|
process_ = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
void Process::Terminate(int result_code) {
|
||
|
|
if (!process_)
|
||
|
|
return;
|
||
|
|
|
||
|
|
// Call NtTerminateProcess directly, without going through the import table,
|
||
|
|
// which might have been hooked with a buggy replacement by third party
|
||
|
|
// software. http://crbug.com/81449.
|
||
|
|
HMODULE module = GetModuleHandle(L"ntdll.dll");
|
||
|
|
typedef UINT (WINAPI *TerminateProcessPtr)(HANDLE handle, UINT code);
|
||
|
|
TerminateProcessPtr terminate_process = reinterpret_cast<TerminateProcessPtr>(
|
||
|
|
GetProcAddress(module, "NtTerminateProcess"));
|
||
|
|
terminate_process(process_, result_code);
|
||
|
|
}
|
||
|
|
|
||
|
|
bool Process::IsProcessBackgrounded() const {
|
||
|
|
if (!process_)
|
||
|
|
return false; // Failure case.
|
||
|
|
DWORD priority = GetPriority();
|
||
|
|
if (priority == 0)
|
||
|
|
return false; // Failure case.
|
||
|
|
return ((priority == BELOW_NORMAL_PRIORITY_CLASS) ||
|
||
|
|
(priority == IDLE_PRIORITY_CLASS));
|
||
|
|
}
|
||
|
|
|
||
|
|
bool Process::SetProcessBackgrounded(bool value) {
|
||
|
|
if (!process_)
|
||
|
|
return false;
|
||
|
|
// Vista and above introduce a real background mode, which not only
|
||
|
|
// sets the priority class on the threads but also on the IO generated
|
||
|
|
// by it. Unfortunately it can only be set for the calling process.
|
||
|
|
DWORD priority;
|
||
|
|
if ((base::win::GetVersion() >= base::win::VERSION_VISTA) &&
|
||
|
|
(process_ == ::GetCurrentProcess())) {
|
||
|
|
priority = value ? PROCESS_MODE_BACKGROUND_BEGIN :
|
||
|
|
PROCESS_MODE_BACKGROUND_END;
|
||
|
|
} else {
|
||
|
|
priority = value ? BELOW_NORMAL_PRIORITY_CLASS : NORMAL_PRIORITY_CLASS;
|
||
|
|
}
|
||
|
|
|
||
|
|
return (::SetPriorityClass(process_, priority) != 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
ProcessId Process::pid() const {
|
||
|
|
if (process_ == 0)
|
||
|
|
return 0;
|
||
|
|
|
||
|
|
return GetProcId(process_);
|
||
|
|
}
|
||
|
|
|
||
|
|
bool Process::is_current() const {
|
||
|
|
return process_ == GetCurrentProcess();
|
||
|
|
}
|
||
|
|
|
||
|
|
// static
|
||
|
|
Process Process::Current() {
|
||
|
|
return Process(::GetCurrentProcess());
|
||
|
|
}
|
||
|
|
|
||
|
|
// static
|
||
|
|
bool Process::CanBackgroundProcesses() {
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
int Process::GetPriority() const {
|
||
|
|
DCHECK(process_);
|
||
|
|
return ::GetPriorityClass(process_);
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace base
|