adding option to return result to the agent

pypykatz_server/server/pypysocketserver.py

@@ -13,10 +13,11 @@
 logging.basicConfig(level=10)
 
 class ThreadedPYPYSocketServer:
-	def __init__(self, ip, port, resultQ):
+	def __init__(self, ip, port, resultQ, send_results = False):
 		self.listen_ip = ip
 		self.listen_port = port
 		self.resultQ = resultQ
+		self.send_results = send_results
 
 		self.ssock = None
 
@@ -52,6 +53,12 @@ def handle(self, soc, addr):
 
 			cmd = PYPYCMD()
 			cmd.cmdtype = PYPYCMDType.END
+			if self.send_results == True:
+				data = ""
+				for luid in mimi.logon_sessions:
+					data += str(mimi.logon_sessions[luid])
+
+				cmd.params.append(data.encode())
 			transport.send(cmd)
 			print('[+] Client finished!')
 

server.py

@@ -18,6 +18,7 @@
 	socket_group.add_argument('-l', '--listen-ip', default = '0.0.0.0', help='IP address to listen on. Default 0.0.0.0')
 	socket_group.add_argument('-p', '--listen-port', type=int, default = 54545, help = 'Port to listen on. Default 54545')
 	socket_group.add_argument('-o', '--out-dir', default = 'creds', help = 'Directory to stroe credentials')
+	socket_group.add_argument('-r', '--return-data', action='store_true', help = 'Return data to the client after sucsessul dump')
 
 	args = parser.parse_args()
 
@@ -26,5 +27,5 @@
 	r.daemon = True
 	r.start()
 	print('[+] Starting server...')
-	server = ThreadedPYPYSocketServer(args.listen_ip, args.listen_port, resQ)
+	server = ThreadedPYPYSocketServer(args.listen_ip, args.listen_port, resQ, args.return_data)
 	server.run()