OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "content/common/set_process_title.h" |
| 6 |
| 7 #include <stddef.h> |
| 8 |
| 9 #include "build/build_config.h" |
| 10 |
| 11 #if defined(OS_POSIX) && !defined(OS_MACOSX) && !defined(OS_SOLARIS) |
| 12 #include <limits.h> |
| 13 #include <stdlib.h> |
| 14 #include <unistd.h> |
| 15 |
| 16 #include <string> |
| 17 |
| 18 #include "base/command_line.h" |
| 19 #endif // defined(OS_POSIX) && !defined(OS_MACOSX) && !defined(OS_SOLARIS) |
| 20 |
| 21 #if defined(OS_LINUX) |
| 22 #include <sys/prctl.h> |
| 23 |
| 24 #include "base/files/file_path.h" |
| 25 #include "base/files/file_util.h" |
| 26 #include "base/process/process_metrics.h" |
| 27 #include "base/strings/string_util.h" |
| 28 #include "base/threading/platform_thread.h" |
| 29 // Linux/glibc doesn't natively have setproctitle(). |
| 30 #include "content/common/set_process_title_linux.h" |
| 31 #endif // defined(OS_LINUX) |
| 32 |
| 33 namespace content { |
| 34 |
| 35 // TODO(jrg): Find out if setproctitle or equivalent is available on Android. |
| 36 #if defined(OS_POSIX) && !defined(OS_MACOSX) && !defined(OS_SOLARIS) && \ |
| 37 !defined(OS_ANDROID) |
| 38 |
| 39 void SetProcessTitleFromCommandLine(const char** main_argv) { |
| 40 // Build a single string which consists of all the arguments separated |
| 41 // by spaces. We can't actually keep them separate due to the way the |
| 42 // setproctitle() function works. |
| 43 std::string title; |
| 44 bool have_argv0 = false; |
| 45 |
| 46 #if defined(OS_LINUX) |
| 47 DCHECK_EQ(base::PlatformThread::CurrentId(), getpid()); |
| 48 |
| 49 if (main_argv) |
| 50 setproctitle_init(main_argv); |
| 51 |
| 52 // In Linux we sometimes exec ourselves from /proc/self/exe, but this makes us |
| 53 // show up as "exe" in process listings. Read the symlink /proc/self/exe and |
| 54 // use the path it points at for our process title. Note that this is only for |
| 55 // display purposes and has no TOCTTOU security implications. |
| 56 base::FilePath target; |
| 57 base::FilePath self_exe(base::kProcSelfExe); |
| 58 if (base::ReadSymbolicLink(self_exe, &target)) { |
| 59 have_argv0 = true; |
| 60 title = target.value(); |
| 61 // If the binary has since been deleted, Linux appends " (deleted)" to the |
| 62 // symlink target. Remove it, since this is not really part of our name. |
| 63 const std::string kDeletedSuffix = " (deleted)"; |
| 64 if (base::EndsWith(title, kDeletedSuffix, base::CompareCase::SENSITIVE)) |
| 65 title.resize(title.size() - kDeletedSuffix.size()); |
| 66 |
| 67 // PR_SET_NAME is available in Linux 2.6.9 and newer. |
| 68 // When available at run time, this sets the short process name that shows |
| 69 // when the full command line is not being displayed in most process |
| 70 // listings. |
| 71 prctl(PR_SET_NAME, base::FilePath(title).BaseName().value().c_str()); |
| 72 } |
| 73 #endif // defined(OS_LINUX) |
| 74 |
| 75 const base::CommandLine* command_line = |
| 76 base::CommandLine::ForCurrentProcess(); |
| 77 for (size_t i = 1; i < command_line->argv().size(); ++i) { |
| 78 if (!title.empty()) |
| 79 title += " "; |
| 80 title += command_line->argv()[i]; |
| 81 } |
| 82 // Disable prepending argv[0] with '-' if we prepended it ourselves above. |
| 83 setproctitle(have_argv0 ? "-%s" : "%s", title.c_str()); |
| 84 } |
| 85 |
| 86 #else |
| 87 |
| 88 // All other systems (basically Windows & Mac) have no need or way to implement |
| 89 // this function. |
| 90 void SetProcessTitleFromCommandLine(const char** /* main_argv */) { |
| 91 } |
| 92 |
| 93 #endif |
| 94 |
| 95 } // namespace content |
OLD | NEW |