代码如下:
import sys
import socket
import getopt
import threading
import subprocess
#Define global variables
listen = False
command = False
upload = False
execute = ""
target = ""
upload_destination = ""
port = 0
def usage():
print("SZY NET TOOL")
print("")
print("Usage: szynet.py -t target_host -p port")
print("-l --listen - listen on [host]:[port] for")
print(" incoming connections")
print("-e --execute=file_to_run - execute the given file upon")
print(" receiving a connection")
print("-c --command - initialize a command shell")
print("-u --upload=destination - upon receiving connection upload a")
print(" file and write to [destination]")
print("")
print("")
print("Examples:")
print("szynet.py -t 192.168.0.1 -p 5555 -l -c")
print("szynet.py -t 192.168.0.1 -p 5555 -l -u=c:\\target.exe")
print("szynet.py -t 192.168.0.1 -p 5555 -l -e=\"cat /etc/passwd\"")
print("echo 'ABCDEFGHI' | ./szynet.py -t 192.168.11.12 -p 135")
sys.exit(0)
def client_sender(buffer):
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
try:
#Connect to the target host
client.connect((target,port))
if len(buffer):
client.send(buffer.encode())
while True:
#Wait for data return
recv_len = 1
response = ""
while recv_len:
data = client.recv(4096).decode()
recv_len = len(data)
response = response + data
if recv_len < 4096:
break
print(response)
#Wait for more input
buffer = input("")
buffer += "\n"
#Send out
client.send(buffer.encode())
except:
print("Exception! Exiting.")
#close connection
client.close()
def server_loop():
global target
#If no target is defined, listen on all ports
if not len(target):
target = "0.0.0.0"
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind((target,port))
server.listen(5)
while True:
client_socket, addr = server.accept()
#Split a thread to handle the new client
client_thread = threading.Thread(target=client_handler,args=(client_socket,))
client_thread.start()
def run_command(command):
#line feed
command = command.rstrip()
#Run the command and return the output
try:
output = subprocess.getoutput(command)
except:
output = "Failed to execute command.\r\n"
#Send the output
return output
def client_handler(client_socket):
global upload
global execute
global command
#Check uploaded files
if len(upload_destination):
#Read all the characters and write down the target
file_buffer = ""
#Continue reading data until no data matches
while True:
data = client_socket.recv(1024)
if not data:
break
else:
file_buffer += data
#Take the data and write it out
try:
file_descriptor = open(upload_destination,"wb")
file_descriptor.write(file_buffer)
file_descriptor.close()
#Make sure the document is written
client_socket.send("Successfully saved file to {} \r\n".format(upload_destination).encode())
except:
client_socket.send("Failed to save file to {} \r\n".format(upload_destination).encode())
#Check command execution
if len(execute):
#execute command
output = run_command(execute)
client_socket.send(output.encode())
#If the command line is needed, we enter another loop
if command:
while True:
#Pop up a window
client_socket.send("<SZY:#> ".encode())
#Receive the file until a newline character is found
cmd_buffer = ""
while "\n" not in cmd_buffer:
cmd_buffer += client_socket.recv(1024).decode()
#Return command output
response = run_command(cmd_buffer)
#Return response data
client_socket.send(response.encode())
def main():
global listen
global port
global execute
global command
global upload_destination
global target
if not len(sys.argv[1:]):
usage()
#Read command line options
try:
opts, args = getopt.getopt(sys.argv[1:],"hle:t:p:cu",["help","listen","execute","target","port","command","upload"])
except getopt.GetoptError as err:
print(str(err))
usage()
for o,a in opts:
if o in ("-h","--help"):
usage()
elif o in ("-l","--listen"):
listen = True
elif o in ("-e","--commandshell"):
execute = True
elif o in ("-c","--commandshell"):
command = True
elif o in ("-u","--upload"):
upload_destination = a
elif o in ("-t","--target"):
target = a
elif o in ("-p","--port"):
port = int(a)
else:
assert False,"Unhandled Option"
#we are listening or just sending data from standard input
if not listen and len(target) and port > 0:
#Read memory data from the command line
#This will block, so you don't have to send ctrl-d when you send data to standard input
buffer = sys.stdin.read()
#send data
client_sender(buffer)
#We start listening and preparing to upload files and execute commands
#Place a bounce shell
#Depends on the command line options above
if listen:
server_loop()
main()
使用演示:
服务端:
客户端:
windows使用Ctrl+Z组合键发送EOF指令后执行命令。
linux使用Ctrl+D组合键发送EOF指令后执行命令。