content/common/set_process_title.cc

   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/files/file_path.h"
  23 #include "base/files/file_util.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 (base::EndsWith(title, kDeletedSuffix, base::CompareCase::SENSITIVE))
  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 base::CommandLine* command_line =
  74       base::CommandLine::ForCurrentProcess();
  75   for (size_t i = 1; i < command_line->argv().size(); ++i) {
  76     if (!title.empty())
  77       title += " ";
  78     title += command_line->argv()[i];
  79   }
  80   // Disable prepending argv[0] with '-' if we prepended it ourselves above.
  81   setproctitle(have_argv0 ? "-%s" : "%s", title.c_str());
  82 }
  83
  84 #else
  85
  86 // All other systems (basically Windows & Mac) have no need or way to implement
  87 // this function.
  88 void SetProcessTitleFromCommandLine(const char** /* main_argv */) {
  89 }
  90
  91 #endif
  92
  93 } // namespace content