server_utils.py - mozsearch

Enable keyboard shortcuts

# Copyright 2021 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 contextlib

import json

import os

import socket

# Use a unix abstract domain socket:

# https://man7.org/linux/man-pages/man7/unix.7.html#:~:text=abstract:

SOCKET_ADDRESS = '\0chromium_build_server_socket'

BUILD_SERVER_ENV_VARIABLE = 'INVOKED_BY_BUILD_SERVER'

def MaybeRunCommand(name, argv, stamp_file):

  """Returns True if the command was successfully sent to the build server."""

  # When the build server runs a command, it sets this environment variable.

  # This prevents infinite recursion where the script sends a request to the

  # build server, then the build server runs the script, and then the script

  # sends another request to the build server.

  if BUILD_SERVER_ENV_VARIABLE in os.environ:

    return False

  with contextlib.closing(socket.socket(socket.AF_UNIX)) as sock:

    try:

      sock.connect(SOCKET_ADDRESS)

      sock.sendall(

          json.dumps({

              'name': name,

              'cmd': argv,

              'cwd': os.getcwd(),

              'stamp_file': stamp_file,

          }).encode('utf8'))

    except socket.error as e:

      # [Errno 111] Connection refused. Either the server has not been started

      #             or the server is not currently accepting new connections.

      if e.errno == 111:

        return False

      raise e

  return True