53 lines
1.6 KiB
C++
53 lines
1.6 KiB
C++
// Copyright (c) 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.
|
|
|
|
#include "base/process/process_handle.h"
|
|
|
|
#include <windows.h>
|
|
#include <tlhelp32.h>
|
|
|
|
#include "base/win/scoped_handle.h"
|
|
#include "base/win/windows_version.h"
|
|
|
|
namespace base {
|
|
|
|
ProcessId GetCurrentProcId() {
|
|
return ::GetCurrentProcessId();
|
|
}
|
|
|
|
ProcessHandle GetCurrentProcessHandle() {
|
|
return ::GetCurrentProcess();
|
|
}
|
|
|
|
ProcessId GetProcId(ProcessHandle process) {
|
|
if (process == base::kNullProcessHandle)
|
|
return 0;
|
|
// This returns 0 if we have insufficient rights to query the process handle.
|
|
// Invalid handles or non-process handles will cause a hard failure.
|
|
ProcessId result = GetProcessId(process);
|
|
CHECK(result != 0 || GetLastError() != ERROR_INVALID_HANDLE)
|
|
<< "process handle = " << process;
|
|
return result;
|
|
}
|
|
|
|
ProcessId GetParentProcessId(ProcessHandle process) {
|
|
ProcessId child_pid = GetProcId(process);
|
|
PROCESSENTRY32 process_entry;
|
|
process_entry.dwSize = sizeof(PROCESSENTRY32);
|
|
|
|
win::ScopedHandle snapshot(CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0));
|
|
if (snapshot.IsValid() && Process32First(snapshot.Get(), &process_entry)) {
|
|
do {
|
|
if (process_entry.th32ProcessID == child_pid)
|
|
return process_entry.th32ParentProcessID;
|
|
} while (Process32Next(snapshot.Get(), &process_entry));
|
|
}
|
|
|
|
// TODO(zijiehe): To match other platforms, -1 (UINT32_MAX) should be returned
|
|
// if |child_id| cannot be found in the |snapshot|.
|
|
return 0u;
|
|
}
|
|
|
|
} // namespace base
|