前提・実現したいこと
このプログラムはサーバーが複数のクライアントからコマンドを受信し、その結果を返すプログラムです。
このプログラムをリバースシェル対応にしたいです。
(現在 複数のクライアントから→サーバー)
(実現したいこと サーバーから→複数のクライアント)
該当のソースコード
server.py
import socket import threading import subprocess HOST = '192.168.0.26' PORT = 9998 clients = [] def remove_conection(con, address): """クライアントと接続を切る""" print('[切断]{}'.format(address)) con.close() clients.remove((con, address)) def server_start(): """サーバをスタートする""" sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sock.bind((HOST, PORT)) sock.listen(10) while True: con, address = sock.accept() print("[接続]{}".format(address)) clients.append((con, address)) handle_thread = threading.Thread(target=handler, args=(con, address), daemon=True) handle_thread.start() def handler(con, address): """クライアントからデータを受信する""" while True: try: data = con.recv(1024) except ConnectionResetError: remove_conection(con, address) break else: if not data: remove_conection(con, address) break else: print("[受信]{} - {}".format(address, data.decode("cp932"))) p = subprocess.run(data.decode("cp932"), shell=True, timeout = 5,stdout=subprocess.PIPE,stderr=subprocess.PIPE,stdin=subprocess.DEVNULL) data = p.stdout + p.stderr con.sendall(data) if __name__ == "__main__": server_start()
client.py
import socket import threading HOST = '192.168.0.26' PORT = 9998 def input_msg_roop(sock): """メッセージ入力を促し、サーバに送信する""" while True: msg = input('>>>') if msg == 'stop': sys.exit() elif msg: sock.send(msg.encode('cp932')) result = sock.recv(4096) print(result.decode("cp932")) def client_start(): """クライアントのスタート""" sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sock.connect((HOST, PORT)) try: input_msg_roop(sock) finally: sock.close() if __name__ == "__main__": client_start()
あなたの回答
tips
プレビュー