Gecko [ v3.cesa.co.za ]

Name	Size	Permission	Date
__pycache__	[ DIR ]	drwxr-xr-x	2026-06-08 18:08
addlhelp	[ DIR ]	drwxr-xr-x	2026-06-08 18:08
commands	[ DIR ]	drwxr-xr-x	2026-06-08 18:08
data	[ DIR ]	drwxr-xr-x	2026-06-08 18:07
tests	[ DIR ]	drwxr-xr-x	2026-06-08 18:08
third_party	[ DIR ]	drwxr-xr-x	2026-06-08 18:08
utils	[ DIR ]	drwxr-xr-x	2026-06-08 18:08
vendored	[ DIR ]	drwxr-xr-x	2026-06-08 18:08
README	378 B	-rw-r--r--	1980-01-01 08:00
__init__.py	6.3 KB	-rw-r--r--	1980-01-01 08:00
__main__.py	34.63 KB	-rw-r--r--	1980-01-01 08:00
boto_resumable_upload.py	24.69 KB	-rw-r--r--	1980-01-01 08:00
boto_translation.py	85.52 KB	-rw-r--r--	1980-01-01 08:00
bucket_listing_ref.py	3.9 KB	-rw-r--r--	1980-01-01 08:00
cloud_api.py	40.97 KB	-rw-r--r--	1980-01-01 08:00
cloud_api_delegator.py	29.87 KB	-rw-r--r--	1980-01-01 08:00
command.py	109.98 KB	-rw-r--r--	1980-01-01 08:00
command_argument.py	4.35 KB	-rw-r--r--	1980-01-01 08:00
command_runner.py	23.21 KB	-rw-r--r--	1980-01-01 08:00
context_config.py	8.36 KB	-rw-r--r--	1980-01-01 08:00
cred_types.py	1.15 KB	-rw-r--r--	1980-01-01 08:00
cs_api_map.py	3.31 KB	-rw-r--r--	1980-01-01 08:00
daisy_chain_wrapper.py	14.39 KB	-rw-r--r--	1980-01-01 08:00
devshell_auth_plugin.py	1.69 KB	-rw-r--r--	1980-01-01 08:00
discard_messages_queue.py	1.29 KB	-rw-r--r--	1980-01-01 08:00
exception.py	5.18 KB	-rw-r--r--	1980-01-01 08:00
file_part.py	3.39 KB	-rw-r--r--	1980-01-01 08:00
gcs_json_api.py	104.02 KB	-rw-r--r--	1980-01-01 08:00
gcs_json_credentials.py	20.31 KB	-rw-r--r--	1980-01-01 08:00
gcs_json_media.py	28.5 KB	-rw-r--r--	1980-01-01 08:00
help_provider.py	4.26 KB	-rw-r--r--	1980-01-01 08:00
iamcredentials_api.py	11.72 KB	-rw-r--r--	1980-01-01 08:00
impersonation_credentials.py	2.51 KB	-rw-r--r--	1980-01-01 08:00
kms_api.py	13.24 KB	-rw-r--r--	1980-01-01 08:00
lazy_wrapper.py	2.26 KB	-rw-r--r--	1980-01-01 08:00
metrics.py	38.98 KB	-rw-r--r--	1980-01-01 08:00
metrics_reporter.py	4.61 KB	-rw-r--r--	1980-01-01 08:00
metrics_tuple.py	1.44 KB	-rw-r--r--	1980-01-01 08:00
name_expansion.py	29.97 KB	-rw-r--r--	1980-01-01 08:00
no_op_auth_plugin.py	1.19 KB	-rw-r--r--	1980-01-01 08:00
no_op_credentials.py	1.03 KB	-rw-r--r--	1980-01-01 08:00
parallel_tracker_file.py	12.14 KB	-rw-r--r--	1980-01-01 08:00
plurality_checkable_iterator.py	4.4 KB	-rw-r--r--	1980-01-01 08:00
progress_callback.py	6.08 KB	-rw-r--r--	1980-01-01 08:00
project_id.py	1.69 KB	-rw-r--r--	1980-01-01 08:00
pubsub_api.py	11.17 KB	-rw-r--r--	1980-01-01 08:00
resumable_streaming_upload.py	8.53 KB	-rw-r--r--	1980-01-01 08:00
seek_ahead_thread.py	5.11 KB	-rw-r--r--	1980-01-01 08:00
sig_handling.py	5.64 KB	-rw-r--r--	1980-01-01 08:00
storage_uri_builder.py	2.02 KB	-rw-r--r--	1980-01-01 08:00
storage_url.py	16.16 KB	-rw-r--r--	1980-01-01 08:00
tab_complete.py	10.77 KB	-rw-r--r--	1980-01-01 08:00
thread_message.py	13.71 KB	-rw-r--r--	1980-01-01 08:00
tracker_file.py	24.23 KB	-rw-r--r--	1980-01-01 08:00
tz_utc.py	2.2 KB	-rw-r--r--	1980-01-01 08:00
ui_controller.py	47.42 KB	-rw-r--r--	1980-01-01 08:00
wildcard_iterator.py	37.73 KB	-rw-r--r--	1980-01-01 08:00

Rename

# -*- coding: utf-8 -*-
# Copyright 2015 Google Inc. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Signal handling functions."""

from __future__ import absolute_import
from __future__ import print_function
from __future__ import division
from __future__ import unicode_literals

import logging
import os
import re
import signal
import sys
import traceback

from gslib import metrics
from gslib.exception import ControlCException
from gslib.utils.constants import UTF8
from gslib.utils.system_util import IS_WINDOWS

if IS_WINDOWS:
  import ctypes  # pylint: disable=g-import-not-at-top

# Maps from signal_num to list of signal handlers to call.
_non_final_signal_handlers = {}
# Maps from signal_num to the final signal handler (if any) that should be
# called for that signal.
_final_signal_handlers = {}

def RegisterSignalHandler(signal_num, handler, is_final_handler=False):
  """Registers a handler for signal signal_num.

Unlike calling signal.signal():
    - This function can be called from any thread (and will cause the handler to
      be run by the main thread when the signal is received).
    - Handlers are cumulative: When a given signal is received, all registered
      handlers will be executed (with the exception that only the last handler
      to register with is_final_handler=True will be called).

Handlers should make no ordering assumptions, other than that the last handler
  to register with is_final_handler=True will be called after all the other
  handlers.

Args:
    signal_num: The signal number with which to associate handler.
    handler: The handler.
    is_final_handler: Bool indicator whether handler should be called last among
                      all the handlers for this signal_num. The last handler to
                      register this way survives; other handlers registered with
                      is_final_handler=True will not be called when the signal
                      is received.
  Raises:
    RuntimeError: if attempt is made to register a signal_num not in
        GetCaughtSignals.
  """
  if signal_num not in GetCaughtSignals():
    raise RuntimeError('Attempt to register handler (%s) for signal %d, which '
                       'is not in GetCaughtSignals' % (handler, signal_num))
  if is_final_handler:
    _final_signal_handlers[signal_num] = handler
  else:
    _non_final_signal_handlers[signal_num].append(handler)

def _SignalHandler(signal_num, cur_stack_frame):
  """Global signal handler.

When a signal is caught we execute each registered handler for that signal.

Args:
    signal_num: Signal that was caught.
    cur_stack_frame: Unused.
  """
  if signal_num in _non_final_signal_handlers:
    for handler in _non_final_signal_handlers[signal_num]:
      handler(signal_num, cur_stack_frame)
  if signal_num in _final_signal_handlers:
    _final_signal_handlers[signal_num](signal_num, cur_stack_frame)

def InitializeSignalHandling():
  """Initializes global signal handling.

Sets up global signal handler for each signal we handle.
  """
  for signal_num in GetCaughtSignals():
    _non_final_signal_handlers[signal_num] = []
    # Make main signal handler catch the signal.
    signal.signal(signal_num, _SignalHandler)

def GetCaughtSignals():
  """Returns terminating signals that can be caught on this OS platform."""
  signals = [signal.SIGINT, signal.SIGTERM]
  if not IS_WINDOWS:
    # Windows doesn't have SIGQUIT.
    signals.append(signal.SIGQUIT)
  return signals

def KillProcess(pid):
  """Make best effort to kill the given process.

We ignore all exceptions so a caller looping through a list of processes will
  continue attempting to kill each, even if one encounters a problem.

Args:
    pid: The process ID.
  """
  try:
    # os.kill doesn't work in Python3 versions before 3.2.
    if IS_WINDOWS and ((3, 0) <= sys.version_info[:3] < (3, 2)):
      kernel32 = ctypes.windll.kernel32
      handle = kernel32.OpenProcess(1, 0, pid)
      kernel32.TerminateProcess(handle, 0)
    else:
      os.kill(pid, signal.SIGKILL)
  except:  # pylint: disable=bare-except
    pass

# pylint: disable=unused-argument
def MultithreadedMainSignalHandler(signal_num, cur_stack_frame):
  """Final signal handler for multi-threaded main process."""
  if signal_num == signal.SIGINT:
    if logging.getLogger().isEnabledFor(logging.DEBUG):
      stack_trace = ''.join(traceback.format_list(traceback.extract_stack()))
      err = ('DEBUG: Caught CTRL-C (signal %d) - Exception stack trace:\n'
             '    %s' % (signal_num, re.sub('\\n', '\n    ', stack_trace)))
      try:
        sys.stderr.write(err.encode(UTF8))
      except (UnicodeDecodeError, TypeError) as e:
        # Can happen when outputting invalid Unicode filenames.
        # or in python3, where stderr only accepts arguments of type
        # str, not bytes.
        sys.stderr.write(err)
    else:
      sys.stderr.write('Caught CTRL-C (signal %d) - exiting\n' % signal_num)

metrics.LogFatalError(exception=ControlCException())
  metrics.Shutdown()
  KillProcess(os.getpid())

def ChildProcessSignalHandler(signal_num, cur_stack_frame):
  """Final signal handler for child processes."""
  KillProcess(os.getpid())

# pylint: enable=unused-argument