File: process_util.cc

package info (click to toggle)
chromium 139.0.7258.127-1
  • links: PTS, VCS
  • area: main
  • in suites:
  • size: 6,122,068 kB
  • sloc: cpp: 35,100,771; ansic: 7,163,530; javascript: 4,103,002; python: 1,436,920; asm: 946,517; xml: 746,709; pascal: 187,653; perl: 88,691; sh: 88,436; objc: 79,953; sql: 51,488; cs: 44,583; fortran: 24,137; makefile: 22,147; tcl: 15,277; php: 13,980; yacc: 8,984; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (84 lines) | stat: -rw-r--r-- 3,001 bytes parent folder | download | duplicates (3)
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifdef UNSAFE_BUFFERS_BUILD
// TODO(crbug.com/390223051): Remove C-library calls to fix the errors.
#pragma allow_unsafe_libc_calls
#endif

#include "remoting/host/base/process_util.h"

#include <string.h>

#include <string>

#include "base/files/file_util.h"
#include "base/logging.h"
#include "base/notimplemented.h"
#include "base/strings/string_util.h"
#include "base/strings/stringprintf.h"
#include "build/build_config.h"

#if BUILDFLAG(IS_WIN)
#include <windows.h>

#include "base/win/scoped_handle.h"
#endif

namespace remoting {

base::FilePath GetProcessImagePath(base::ProcessId pid) {
#if BUILDFLAG(IS_LINUX)
  // See: https://man7.org/linux/man-pages/man5/procfs.5.html. It says the
  // symlink will have "(deleted)" appended to the original pathname, but in
  // practice it also has a preceding space.
  static constexpr char kExeDeletedSuffix[] = " (deleted)";
  // We don't get the image path from the command line, since it's spoofable by
  // the process itself.
  base::FilePath process_exe_path(
      base::StringPrintf("/proc/%" CrPRIdPid "/exe", pid));
  base::FilePath process_image_path;
  base::ReadSymbolicLink(process_exe_path, &process_image_path);
  // TODO(yuweih): See if we can run this method on a worker thread that allows
  // blocking and use base::PathExists() instead.
  bool path_exists = access(process_image_path.value().c_str(), F_OK) == 0;
  if (!path_exists) {
    std::string process_base_name = process_image_path.BaseName().value();
    if (!base::EndsWith(process_base_name, kExeDeletedSuffix,
                        base::CompareCase::INSENSITIVE_ASCII)) {
      LOG(ERROR) << "Unexpected process image path: " << process_image_path
                 << ". Path doesn't exist and doesn't end with '"
                 << kExeDeletedSuffix << "'";
      return base::FilePath();
    }
    VLOG(1) << "Process image for PID " << pid
            << " no longer exists in its original path.";
    return process_image_path.DirName().Append(process_base_name.substr(
        0, process_base_name.size() - strlen(kExeDeletedSuffix)));
  }
  return process_image_path;
#elif BUILDFLAG(IS_WIN)
  base::win::ScopedHandle process_handle(
      OpenProcess(PROCESS_QUERY_LIMITED_INFORMATION, false, pid));
  if (!process_handle.is_valid()) {
    PLOG(ERROR) << "OpenProcess failed";
    return base::FilePath();
  }
  std::array<wchar_t, MAX_PATH + 1> buffer;
  DWORD size = buffer.size();
  if (!QueryFullProcessImageName(process_handle.Get(), 0, buffer.data(),
                                 &size)) {
    PLOG(ERROR) << "QueryFullProcessImageName failed";
    return base::FilePath();
  }
  DCHECK_GT(size, 0u);
  DCHECK_LT(size, buffer.size());
  return base::FilePath(base::FilePath::StringViewType(buffer.data(), size));
#else
  NOTIMPLEMENTED();
  return base::FilePath();
#endif
}

}  // namespace remoting