Set argv[0] of the buildbot master process to the name of the master.

scripts/master/master_utils.py

 # Copyright (c) 2012 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.
 
+import ctypes
+import ctypes.util
 import os
 import random
 import re
+import sys
 
 import buildbot
 from buildbot import interfaces, util
 from buildbot.buildslave import BuildSlave
 from buildbot.interfaces import IRenderable
 from buildbot.status import mail
 from buildbot.status.builder import BuildStatus
 from buildbot.status.status_push import HttpStatusPush
 from twisted.python import log
 from zope.interface import implements
@@ skipping 449 matching lines @@
     if 'port' in values and 'user' in values and 'password' in values:
       c['manhole'] = manhole.PasswordManhole(interface, values['user'],
                                             values['password'])
     elif 'port' in values:
       c['manhole'] = manhole.AuthorizedKeysManhole(interface,
           os.path.expanduser("~/.ssh/authorized_keys"))
 
   if active_master.buildbucket_bucket and active_master.service_account_path:
     SetupBuildbucket(c, active_master)
 
+  # TODO(dsansome): enable this on all masters if it works, remove completely
+  # if it doesn't.
+  if GetMastername() in {'chromium.infra', 'chromium.infra.cron'}:
+    SetMasterProcessName()
+
 
 def SetupBuildbucket(c, active_master):
   def params_hook(params, build):
     config_hook = c.get('buildbucket_params_hook')
     if callable(config_hook):
       config_hook(params, build)
 
     properties = params.setdefault('properties', {})
     properties.pop('requester', None)  # Ignore externally set requester.
     if build['created_by']:
@@ skipping 165 matching lines @@
   """
 
   def __call__(self, builder, slave_builders):
     if not slave_builders:
       return None
 
     preferred_slaves = [
         s for s in slave_builders
         if s.slave.properties.getProperty('preferred_builder') == builder.name]
     return random.choice(preferred_slaves or slave_builders)
+
+
+def SetMasterProcessName():
+  """Sets the name of this process to the name of the master.  Linux only."""
+
+  if sys.platform != 'linux2':
+    sys.exit(2)
+    return
+
+  SetCommandLine("master: %s" % GetMastername())
+
+
+def SetCommandLine(cmdline):
+  # Get the current commandline.
+  argc = ctypes.c_int()
+  argv = ctypes.POINTER(ctypes.c_char_p)()
+  ctypes.pythonapi.Py_GetArgcArgv(ctypes.byref(argc), ctypes.byref(argv))
+
+  # Calculate its length.
+  cmdlen = sum([len(argv[i]) for i in xrange(0, argc.value)]) + argc.value
+
+  # Pad the cmdline string to the required length.  If it's longer than the
+  # currentl commandline, truncate it.
+  if len(cmdline) >= cmdlen:
+    new_cmdline = ctypes.c_char_p(cmdline[:cmdlen-1] + '\0')
+  else:
+    new_cmdline = ctypes.c_char_p(cmdline.ljust(cmdlen, '\0'))
+
+  # Replace the old commandline.
+  libc = ctypes.CDLL(ctypes.util.find_library('c'))
+  libc.memcpy(argv.contents, new_cmdline, cmdlen)