def accept_connection(conn):
try:
sconn = StealthConn(conn, server=True)
# The sender is either going to chat to us or send a file
cmd = sconn.recv()
if cmd == b'ECHO':
echo_server(sconn)
elif cmd == b'FILE':
p2p_download_file(sconn)
except socket.error:
print("Connection closed unexpectedly")
def accept_connection(conn):
try:
sconn = StealthConn(conn, server=True)
# The sender is either going to chat to us or send a file
cmd, flag = sconn.recv()
if cmd == b'ECHO':
echo_server(sconn)
elif cmd == b'FILE':
p2p_download_file(sconn)
except socket.error:
print("Connection closed unexpectedly")
def accept_connection(conn):
try:
sconn = StealthConn(conn, server=True)
cmd = sconn.recv()
if cmd == b'ECHO': #Echo
echo_server(sconn)
elif cmd == b'FILE': #File download
p2p_download_file(sconn)
except socket.error:
print(Fore.CYAN + "[-] Connection closed unexpectedly\n" +
Style.RESET_ALL) #Something went wrong
def accept_connection(conn):
try:
sconn = StealthConn(conn, server=True)
# The sender is either going to chat to us or send a file
cmd = sconn.recv()
if cmd == b'ECHO':
echo_server(sconn)
elif cmd == b'FILE':
p2p_download_file(sconn)
else:
raise RuntimeError("Invalid command: {}".format(cmd))
except socket.error:
print("Connection closed unexpectedly")
def accept_connection(conn):
try:
sconn = StealthConn(conn, rsa_key, server=True)
# The sender is either going to chat to us or send a file
cmd = sconn.recv()
if cmd == b'ECHO':
echo_server(sconn)
elif cmd == b'FILE':
p2p_download_file(sconn)
except socket.error:
print("Connection closed unexpectedly")
except RuntimeError:
traceback.print_exc()
print("Authentication Failed")
def accept_connection(conn):
try:
f = open("serverPublic.pem", 'r')
rsaKey = RSA.importKey(f.read())
f.close()
sconn = StealthConn(conn, rsaKey, server=True)
# The sender is either going to chat to us or send a file
cmd = sconn.recv()
if cmd == b'ECHO':
echo_server(sconn)
elif cmd == b'FILE':
p2p_download_file(sconn)
except socket.error:
print("Connection closed unexpectedly")
except RuntimeError:
traceback.print_exc()
print("Authentication Failed")
def find_bot():
print("Finding another bot...")
port = 1337
conn = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
while 1:
if port == server_port:
# Don't connect to yourself, silly bot!
port += 1
else:
try:
print("Found bot on port %d" % port)
conn.connect(("localhost", port))
sconn = StealthConn(conn, client=True)
return sconn
except socket.error:
print("No bot was listening on port %d" % port)
port += 1
def find_bot(): #Find other bots listening on the network
print(Fore.GREEN + "[+] Locating other cyber-attack bots" +
Style.RESET_ALL)
port = 1337
conn = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
while 1:
if port == server_port: #Don't connect to yourself, you dingbat bot!
port += 1
else:
try:
conn.connect(("localhost", port))
print(Fore.MAGENTA + "[+] Found bot on port " + str(port) +
Style.RESET_ALL) #Aha, found a bot on the network
sconn = StealthConn(conn, client=True)
return sconn
except socket.error:
print(Fore.RED + "[-] No bot was listening on port " +
str(port) + Style.RESET_ALL) #Aww, no bot found
port += 1
def find_bot():
print("Finding another bot...")
port = 1337
conn = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
while 1:
if port == server_port:
# Don't connect to yourself, silly bot!
port += 1
else:
try:
conn.connect(("localhost", port))
print("Found bot on port %d" % port)
sconn = StealthConn(conn, rsa_key, client=True)
return sconn
except socket.error:
print("No bot was listening on port %d" % port)
port += 1
except RuntimeError:
traceback.print_exc()
print("Authentication Failed")
