Rooba
/
pupy
mirror of https://github.com/n1nj4sec/pupy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Write nice messages about connection status

This commit is contained in:
Oleksii Shevchuk 2016-11-21 21:35:56 +02:00
parent 288aa030bf
commit fbfb9c966f
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pupy/pupylib/PupyServer.py
View File

@ -35,6 +35,7 @@ import marshal
import network.conf import network.conf
import rpyc import rpyc
import shlex import shlex
import socket
try: try:
import ConfigParser as configparser import ConfigParser as configparser
@ -270,8 +271,14 @@ class PupyServer(threading.Thread):
except LauncherError as e: except LauncherError as e:
launcher.arg_parser.print_usage() launcher.arg_parser.print_usage()
return return
stream=launcher.iterate().next()
self.handler.display_info("Connecting ...") try:
stream=launcher.iterate().next()
except socket.error as e:
self.handler.display_error("Couldn't connect to pupy: {}".format(e))
return
self.handler.display_success("Connected. Starting session")
bgsrv=PupyConnectionThread(PupyService.PupyBindService, rpyc.Channel(stream), config={}) bgsrv=PupyConnectionThread(PupyService.PupyBindService, rpyc.Channel(stream), config={})
bgsrv.start() bgsrv.start()