def grace_close_timeout(agent_id, user, timeout=120):
try:
time.sleep(timeout)
res, session = ra.session_exist(agent_id)
if res:
status = session[0]
if status == ra.Status.Timeout:
ra.db.edit_connection(agent_id=agent_id,
status=ra.Status.Closing)
remove_session(agent_id, user)
else:
logger.info('Session found for agent %s' % agent_id)
logger.info("No longer under 'timeout' status. Nothing to do.")
else:
logger.info('No session found for agent %s' % agent_id)
logger.info('Nothing to do.')
except Exception as e:
logger.error('Unable to close session from timeout thread.')
logger.exception(e)
def grace_close_timeout(agent_id, user, timeout=120):
try:
time.sleep(timeout)
res, session = ra.session_exist(agent_id)
if res:
status = session[0]
if status == ra.Status.Timeout:
ra.db.edit_connection(
agent_id=agent_id,
status=ra.Status.Closing
)
remove_session(agent_id, user)
else:
logger.info('Session found for agent %s' % agent_id)
logger.info("No longer under 'timeout' status. Nothing to do.")
else:
logger.info('No session found for agent %s' % agent_id)
logger.info('Nothing to do.')
except Exception as e:
logger.error('Unable to close session from timeout thread.')
logger.exception(e)
def terminate_rd_session(
agent_id=None,
user=Default.User,
customer=Default.Customer
):
if not agent_id:
return {
'pass': False,
'message': 'No agent id was provided.'
}
res, session = ra.session_exist(agent_id)
if res:
status, web_port = session
if status == ra.Status.Waiting:
return {
'pass': False,
'message': (
'Session still waiting for agent to respond.'
),
ra.RaKey.Status: status
}
if status == ra.Status.Timeout:
return {
'pass': False,
'message': (
'Session already in the process to be closed.'
),
ra.RaKey.Status: status,
}
return set_session_to_timeout(agent_id, user)
else:
message = "Could not find session for agent %s." % agent_id
logger.error(message)
return {
'pass': False,
'message': message,
'error': "Session not found."
}
def terminate_rd_session(agent_id=None,
user=Default.User,
customer=Default.Customer):
if not agent_id:
return {'pass': False, 'message': 'No agent id was provided.'}
res, session = ra.session_exist(agent_id)
if res:
status, web_port = session
if status == ra.Status.Waiting:
return {
'pass': False,
'message': ('Session still waiting for agent to respond.'),
ra.RaKey.Status: status
}
if status == ra.Status.Timeout:
return {
'pass': False,
'message': ('Session already in the process to be closed.'),
ra.RaKey.Status: status,
}
return set_session_to_timeout(agent_id, user)
else:
message = "Could not find session for agent %s." % agent_id
logger.error(message)
return {
'pass': False,
'message': message,
'error': "Session not found."
}
def new_rd_session(agent_id=None,
user=Default.User,
customer=Default.Customer):
if not agent_id:
return {'pass': False, 'message': 'No agent id was provided.'}
res, session = ra.session_exist(agent_id)
if res:
status, web_port = session
if status == ra.Status.Waiting:
return {
'pass': False,
'message':
('Session exist for this agent. Waiting for response.'),
ra.RaKey.Status: status
}
host = ra.get_hostname()
uri = ra.create_vnc_uri(host, web_port)
if status == ra.Status.Ready:
return {
'pass': False,
'message': ('Session exist for this agent. Using session.'),
ra.RaKey.Status: status,
ra.RaKey.WebPort: web_port,
ra.RaKey.Hostname: host,
ra.RaKey.Uri: uri
}
if status == ra.Status.Timeout:
ra.db.edit_connection(agent_id=agent_id, status=ra.Status.Ready)
return {
'pass': False,
'message': ('Session exist for this agent. Reusing.'),
ra.RaKey.Status: status,
ra.RaKey.WebPort: web_port,
ra.RaKey.Hostname: host,
ra.RaKey.Uri: uri
}
if status == ra.Status.Closing:
return {
'pass':
False,
'message': ('Session exist but is being closed.'
'Please try again in a few seconds.'),
ra.RaKey.Status:
status,
}
operation = RaOperation(ra.RaValue.RemoteDesktop,
agent_id,
username=user,
customer=customer,
uri=ra.RaUri.StartRemoteDesktop % agent_id,
method='POST')
port = None
tunnel_needed = True
if tunnel_needed:
params = None
port_range = list(ra.PortRange)
for p in ra.PortRange:
params = reverse_tunnel_params(port_range)
if ra.db.port_available(port=params[TunnelKey.HostPort]):
break
port_range.remove(p)
params = None
if params:
port = params[TunnelKey.HostPort]
ssh_port = params[TunnelKey.SSHPort]
operation.set_tunnel(host_port=port, ssh_port=ssh_port)
else:
return {
'pass': False,
'message': "Could not resolve host port for tunnel."
}
operation_id = save_operation(operation)
if operation_id:
result, msg = ra.db.save_connection(agent_id=agent_id,
host_port=port,
status=ra.Status.Waiting)
if result:
operation.operation_id = operation_id
ra.add_feedback(agent_id, ra.Status.Waiting)
store_in_agent_queue(operation)
return {
'pass': True,
'message': 'Remote desktop created. Waiting...'
}
else:
return {'pass': False, 'message': msg}
else:
return {
'pass': False,
'message': "Unable to save operation. Invalid operation ID."
}
def new_rd_session(
agent_id=None,
user=Default.User,
customer=Default.Customer
):
if not agent_id:
return {
'pass': False,
'message': 'No agent id was provided.'
}
res, session = ra.session_exist(agent_id)
if res:
status, web_port = session
if status == ra.Status.Waiting:
return {
'pass': False,
'message': (
'Session exist for this agent. Waiting for response.'
),
ra.RaKey.Status: status
}
host = ra.get_hostname()
uri = ra.create_vnc_uri(host, web_port)
if status == ra.Status.Ready:
return {
'pass': False,
'message': (
'Session exist for this agent. Using session.'
),
ra.RaKey.Status: status,
ra.RaKey.WebPort: web_port,
ra.RaKey.Hostname: host,
ra.RaKey.Uri: uri
}
if status == ra.Status.Timeout:
ra.db.edit_connection(
agent_id=agent_id,
status=ra.Status.Ready
)
return {
'pass': False,
'message': (
'Session exist for this agent. Reusing.'
),
ra.RaKey.Status: status,
ra.RaKey.WebPort: web_port,
ra.RaKey.Hostname: host,
ra.RaKey.Uri: uri
}
if status == ra.Status.Closing:
return {
'pass': False,
'message': (
'Session exist but is being closed.'
'Please try again in a few seconds.'
),
ra.RaKey.Status: status,
}
operation = RaOperation(
ra.RaValue.RemoteDesktop,
agent_id,
username=user,
customer=customer,
uri=ra.RaUri.StartRemoteDesktop % agent_id,
method='POST'
)
port = None
tunnel_needed = True
if tunnel_needed:
params = None
port_range = list(ra.PortRange)
for p in ra.PortRange:
params = reverse_tunnel_params(port_range)
if ra.db.port_available(
port=params[TunnelKey.HostPort]
):
break
port_range.remove(p)
params = None
if params:
port = params[TunnelKey.HostPort]
ssh_port = params[TunnelKey.SSHPort]
operation.set_tunnel(host_port=port, ssh_port=ssh_port)
else:
return {
'pass': False,
'message': "Could not resolve host port for tunnel."
}
operation_id = save_operation(operation)
if operation_id:
result, msg = ra.db.save_connection(
agent_id=agent_id,
host_port=port,
status=ra.Status.Waiting
)
if result:
operation.operation_id = operation_id
ra.add_feedback(agent_id, ra.Status.Waiting)
store_in_agent_queue(operation)
return {
'pass': True,
'message': 'Remote desktop created. Waiting...'
}
else:
return {
'pass': False,
'message': msg
}
else:
return {
'pass': False,
'message': "Unable to save operation. Invalid operation ID."
}
