def recv_cfg(self, msg):
"""receive cfg/topology configuration"""
if msg.type != ConfigurationMessage.ident:
raise MessageProcessingError('unexpected message: %s' % msg)
self.logger.debug('got channel configuration')
# gw node name
hostname = _getshorthostname()
if not msg.gateway:
self.nodename = hostname
self.logger.warn('gw name not provided, using system hostname %s',
self.nodename)
else:
self.nodename = msg.gateway
self.logger.debug('using gateway node name %s', self.nodename)
if self.nodename.lower() != hostname.lower():
self.logger.debug('gw name %s does not match system hostname %s',
self.nodename, hostname)
# topology
task_self().topology = self.topology = msg.data_decode()
self.logger.debug('decoded propagation tree')
self.logger.debug('\n%s', self.topology)
self.setup = True
self._ack(msg)
def recv_cfg(self, msg):
"""receive cfg/topology configuration"""
if msg.type != ConfigurationMessage.ident:
raise MessageProcessingError('unexpected message: %s' % msg)
self.logger.debug('got channel configuration')
# gw node name
hostname = _getshorthostname()
if not msg.gateway:
self.nodename = hostname
self.logger.warn('gw name not provided, using system hostname %s',
self.nodename)
else:
self.nodename = msg.gateway
self.logger.debug('using gateway node name %s', self.nodename)
if self.nodename.lower() != hostname.lower():
self.logger.debug('gw name %s does not match system hostname %s',
self.nodename, hostname)
# topology
task_self().topology = self.topology = msg.data_decode()
self.logger.debug('decoded propagation tree')
self.logger.debug('\n%s', self.topology)
self.setup = True
self._ack(msg)
def gateway_main():
"""ClusterShell gateway entry point"""
host = _getshorthostname()
# configure root logger
logdir = os.path.expanduser(os.environ.get('CLUSTERSHELL_GW_LOG_DIR',
'/tmp'))
loglevel = os.environ.get('CLUSTERSHELL_GW_LOG_LEVEL', 'INFO')
try:
log_level = getattr(logging, loglevel.upper(), logging.INFO)
log_fmt = '%(asctime)s %(name)s %(levelname)s %(message)s'
logging.basicConfig(level=log_level, format=log_fmt,
filename=os.path.join(logdir, "%s.gw.log" % host))
except (IOError, OSError):
pass # logging failure is not fatal
logger = logging.getLogger(__name__)
sys.excepthook = gateway_excepthook
logger.debug('Starting gateway on %s', host)
logger.debug("environ=%s", os.environ)
set_nonblock_flag(sys.stdin.fileno())
set_nonblock_flag(sys.stdout.fileno())
set_nonblock_flag(sys.stderr.fileno())
task = task_self()
# Disable MsgTree buffering, it is enabled later when needed
task.set_default("stdout_msgtree", False)
task.set_default("stderr_msgtree", False)
if sys.stdin.isatty():
logger.critical('Gateway failure: sys.stdin.isatty() is True')
sys.exit(1)
gateway = GatewayChannel(task)
worker = StreamWorker(handler=gateway)
# Define worker._fanout to not rely on the engine's fanout, and use
# the special value FANOUT_UNLIMITED to always allow registration
worker._fanout = FANOUT_UNLIMITED
worker.set_reader(gateway.SNAME_READER, sys.stdin)
worker.set_writer(gateway.SNAME_WRITER, sys.stdout, retain=False)
# must stay disabled for now (see #274)
#worker.set_writer(gateway.SNAME_ERROR, sys.stderr, retain=False)
task.schedule(worker)
logger.debug('Starting task')
try:
task.resume()
logger.debug('Task performed')
except EngineAbortException as exc:
logger.debug('EngineAbortException')
except IOError as exc:
logger.debug('Broken pipe (%s)', exc)
raise
except Exception as exc:
logger.exception('Gateway failure: %s', exc)
logger.debug('-------- The End --------')
def gateway_main():
"""ClusterShell gateway entry point"""
host = _getshorthostname()
# configure root logger
logdir = os.path.expanduser(
os.environ.get('CLUSTERSHELL_GW_LOG_DIR', '/tmp'))
loglevel = os.environ.get('CLUSTERSHELL_GW_LOG_LEVEL', 'INFO')
try:
log_level = getattr(logging, loglevel.upper(), logging.INFO)
log_fmt = '%(asctime)s %(name)s %(levelname)s %(message)s'
logging.basicConfig(level=log_level,
format=log_fmt,
filename=os.path.join(logdir, "%s.gw.log" % host))
except (IOError, OSError):
pass # logging failure is not fatal
logger = logging.getLogger(__name__)
sys.excepthook = gateway_excepthook
logger.debug('Starting gateway on %s', host)
logger.debug("environ=%s", os.environ)
set_nonblock_flag(sys.stdin.fileno())
set_nonblock_flag(sys.stdout.fileno())
set_nonblock_flag(sys.stderr.fileno())
task = task_self()
# Disable MsgTree buffering, it is enabled later when needed
task.set_default("stdout_msgtree", False)
task.set_default("stderr_msgtree", False)
if sys.stdin.isatty():
logger.critical('Gateway failure: sys.stdin.isatty() is True')
sys.exit(1)
gateway = GatewayChannel(task)
worker = StreamWorker(handler=gateway)
# Define worker._fanout to not rely on the engine's fanout, and use
# the special value FANOUT_UNLIMITED to always allow registration
worker._fanout = FANOUT_UNLIMITED
worker.set_reader(gateway.SNAME_READER, sys.stdin)
worker.set_writer(gateway.SNAME_WRITER, sys.stdout, retain=False)
# must stay disabled for now (see #274)
#worker.set_writer(gateway.SNAME_ERROR, sys.stderr, retain=False)
task.schedule(worker)
logger.debug('Starting task')
try:
task.resume()
logger.debug('Task performed')
except EngineAbortException as exc:
logger.debug('EngineAbortException')
except IOError as exc:
logger.debug('Broken pipe (%s)', exc)
raise
except Exception as exc:
logger.exception('Gateway failure: %s', exc)
logger.debug('-------- The End --------')