def _weirdify(self, client_sock, backend_sock, to_backend,
statsd_prefix, behavior, behavior_name):
"""This is where all the magic happens.
Depending the configuration, we will chose to either drop packets,
proxy them, wait a long time, etc, as defined in the configuration.
"""
if hasattr(client_sock, 'closed') and client_sock.closed:
raise ValueError("Client is gone")
if to_backend:
self.statsd_incr(statsd_prefix + 'to_backend')
dest = backend_sock
source = client_sock
else:
self.statsd_incr(statsd_prefix + 'to_client')
source = backend_sock
dest = client_sock
self._logger.debug('starting weirdify %s' % to_backend)
try:
# XXX cache this ?
args = self.settings['args']
behavior.update_settings(extract_settings(args, 'behavior',
behavior_name))
# calling the handler
return self.handler(source, dest, to_backend, behavior)
finally:
self._logger.debug('exiting weirdify %s' % to_backend)
def _weirdify(self, client_sock, backend_sock, to_backend, statsd_prefix,
behavior, behavior_name):
"""This is where all the magic happens.
Depending the configuration, we will chose to either drop packets,
proxy them, wait a long time, etc, as defined in the configuration.
"""
if hasattr(client_sock, 'closed') and client_sock.closed:
raise ValueError("Client is gone")
if to_backend:
self.statsd_incr(statsd_prefix + 'to_backend')
dest = backend_sock
source = client_sock
else:
self.statsd_incr(statsd_prefix + 'to_client')
source = backend_sock
dest = client_sock
self._logger.debug('starting weirdify %s' % to_backend)
try:
# XXX cache this ?
args = self.settings['args']
behavior.update_settings(
extract_settings(args, 'behavior', behavior_name))
# calling the handler
return self.handler(source, dest, to_backend, behavior)
finally:
self._logger.debug('exiting weirdify %s' % to_backend)
def __init__(self, proxy, backend, protocol='tcp', behaviors=None,
settings=None, statsd=None, logger=None, **kwargs):
self.settings = settings
cfg = self.settings.getsection('vaurien')
if behaviors is None:
behaviors = get_behaviors()
logger.info('Starting the Chaos TCP Server')
parsed_proxy = parse_address(proxy)
self.backend = backend
dest = parse_address(backend)
backlog = cfg.get('backlog', 8192)
StreamServer.__init__(self, parsed_proxy, backlog=backlog, **kwargs)
self.max_accept = 2000 # XXX option ?
self.pool_max_size = cfg.get('pool_max_size', 1000)
self.pool_timeout = cfg.get('pool_timeout', 30)
self.async_mode = not cfg.get('sync', False)
self._pool = FactoryPool(self._create_connection, self.pool_max_size,
self.pool_timeout)
self.dest = dest
self.running = True
self._statsd = statsd
self._logger = logger
self.behaviors = behaviors
self.behaviors.update(get_prefixed_sections(self.settings, 'behavior',
logger))
self.behavior = get_behaviors()['dummy']
self.behavior_name = 'dummy'
self.stay_connected = cfg.get('stay_connected', False)
self.timeout = cfg.get('timeout', 30)
self.protocol = cfg.get('protocol', protocol)
# creating the handler with the passed options
protocols = get_protocols()
protocols.update(get_prefixed_sections(self.settings, 'protocol',
logger))
self.handler = protocols[self.protocol]
# updating the handler settings
args = settings['args']
self.handler.update_settings(extract_settings(args, 'protocol',
self.protocol))
protocol_specific_settings = self.settings.getsection('protocol.%s' % self.protocol)
if protocol_specific_settings:
self.handler.update_settings(protocol_specific_settings)
self.handler.proxy = self
logger.info('Options:')
logger.info('* proxies from %s to %s' % (proxy, backend))
logger.info('* timeout: %d' % self.timeout)
logger.info('* stay_connected: %d' % self.stay_connected)
logger.info('* pool_max_size: %d' % self.pool_max_size)
logger.info('* pool_timeout: %d' % self.pool_timeout)
logger.info('* async_mode: %d' % self.async_mode)
def __init__(self, proxy, backend, protocol='tcp', behaviors=None,
settings=None, statsd=None, logger=None, **kwargs):
self.settings = settings
cfg = self.settings.getsection('vaurien')
if behaviors is None:
behaviors = get_behaviors()
logger.info('Starting the Chaos TCP Server')
parsed_proxy = parse_address(proxy)
self.backend = backend
dest = parse_address(backend)
backlog = cfg.get('backlog', 8192)
StreamServer.__init__(self, parsed_proxy, backlog=backlog, **kwargs)
self.max_accept = 2000 # XXX option ?
self.pool_max_size = cfg.get('pool_max_size', 1000)
self.pool_timeout = cfg.get('pool_timeout', 30)
self.async_mode = not cfg.get('sync', False)
self._pool = FactoryPool(self._create_connection, self.pool_max_size,
self.pool_timeout)
self.dest = dest
self.running = True
self._statsd = statsd
self._logger = logger
self.behaviors = behaviors
self.behaviors.update(get_prefixed_sections(self.settings, 'behavior',
logger))
self.behavior = get_behaviors()['dummy']
self.behavior_name = 'dummy'
self.stay_connected = cfg.get('stay_connected', False)
self.timeout = cfg.get('timeout', 30)
self.protocol = cfg.get('protocol', protocol)
# creating the handler with the passed options
protocols = get_protocols()
protocols.update(get_prefixed_sections(self.settings, 'protocol',
logger))
self.handler = protocols[self.protocol]
# updating the handler settings
args = settings['args']
self.handler.update_settings(extract_settings(args, 'protocol',
self.protocol))
protocol_specific_settings = self.settings.getsection('protocol.%s' % self.protocol)
if protocol_specific_settings:
self.handler.update_settings(protocol_specific_settings)
self.handler.proxy = self
logger.info('Options:')
logger.info('* proxies from %s to %s' % (proxy, backend))
logger.info('* timeout: %d' % self.timeout)
logger.info('* stay_connected: %d' % self.stay_connected)
logger.info('* pool_max_size: %d' % self.pool_max_size)
logger.info('* pool_timeout: %d' % self.pool_timeout)
logger.info('* async_mode: %d' % self.async_mode)
def __init__(
self, proxy, backend, protocol="tcp", behaviors=None, settings=None, statsd=None, logger=None, **kwargs
):
self.settings = settings
cfg = self.settings.getsection("vaurien")
if behaviors is None:
behaviors = get_behaviors()
logger.info("Starting the Chaos TCP Server")
parsed_proxy = parse_address(proxy)
self.backend = backend
dest = parse_address(backend)
backlog = cfg.get("backlog", 8192)
StreamServer.__init__(self, parsed_proxy, backlog=backlog, **kwargs)
self.max_accept = 2000 # XXX option ?
self.pool_max_size = cfg.get("pool_max_size", 1000)
self.pool_timeout = cfg.get("pool_timeout", 30)
self.async_mode = not cfg.get("sync", False)
self._pool = FactoryPool(self._create_connection, self.pool_max_size, self.pool_timeout)
self.dest = dest
self.running = True
self._statsd = statsd
self._logger = logger
self.behaviors = behaviors
self.behaviors.update(get_prefixed_sections(self.settings, "behavior", logger))
self.behavior = get_behaviors()["dummy"]
self.behavior_name = "dummy"
self.stay_connected = cfg.get("stay_connected", False)
self.timeout = cfg.get("timeout", 30)
self.protocol = cfg.get("protocol", protocol)
# creating the handler with the passed options
protocols = get_protocols()
protocols.update(get_prefixed_sections(self.settings, "protocol", logger))
self.handler = protocols[self.protocol]
# updating the handler settings
args = settings["args"]
self.handler.update_settings(extract_settings(args, "protocol", self.protocol))
protocol_specific_settings = self.settings.getsection("protocol.%s" % self.protocol)
if protocol_specific_settings:
self.handler.update_settings(protocol_specific_settings)
self.handler.proxy = self
logger.info("Options:")
logger.info("* proxies from %s to %s" % (proxy, backend))
logger.info("* timeout: %d" % self.timeout)
logger.info("* stay_connected: %d" % self.stay_connected)
logger.info("* pool_max_size: %d" % self.pool_max_size)
logger.info("* pool_timeout: %d" % self.pool_timeout)
logger.info("* async_mode: %d" % self.async_mode)
