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