socket_ftp_上传下载
admin
2023-07-14 11:24:29
0

#main.py
import socketserver,json,os,hashlib
class MyTcpHandler(socketserver.BaseRequestHandler):
   
def put(self,*args):
        cmd_dic=args[
0]
        filename=cmd_dic[
"filename"]
        filesize=cmd_dic[
"size"]
       
if os.path.isfile(filename):
            f=
open(filename+".new",'wb')
       
else:
            f=
open(filename,'wb')
       
self.request.send(b'200 ok')
        received_size=
0
       
m=hashlib.md5()
       
while received_size             data=self.request.recv(1024)
            m.update(data)
            f.write(data)
            received_size+=
len(data)
       
else:
            uploaded_file_md5=m.hexdigest()
           
self.request.send(b"ready recive your md5")
            client_file_md5=
self.request.recv(1024)
            
print("client_file_md5:",client_file_md5.decode())
           
print("uploaded_file_md5:",uploaded_file_md5)
           
print("file [%s] has uploaded..."%filename)
   
def get(self,*args):
        cmd_dic=args[
0]
        filename=cmd_dic[
"filename"]
       
if os.path.isfile(filename):
            f=
open(filename,'rb')
            filesize=os.stat(filename).st_size
           
self.request.send(str(filesize).encode())
           
self.request.recv(1024)
            m = hashlib.md5()
           
for line in f:
                m.update(line)
               
self.request.send(line)
           
else:
               
print("file md5:", m.hexdigest())
               
self.request.recv(1024)
               
self.request.send(m.hexdigest().encode("utf-8"))
               
print("file download  success...")
                f.close()
       
else:
           
print(filename,"is not exist")
   
def handle(self):
       
while True:
           
try:
               
self.data=self.request.recv(1024).strip()
                
print("{} wrote:".format(self.client_address[0]))
               
print(self.data.decode())
                cmd_dic=json.loads(
self.data.decode())
               
print("cmd_dic:",cmd_dic)
                action=cmd_dic[
"action"]
               
if hasattr(self,action):
                    func=(
getattr(self,action))
                    func(cmd_dic)
           
except ConnectionResetError as e:
               
print("error",e)
               
break
if
__name__=="__main__":
    HOST,PORT=
"localhost",5566
   
server=socketserver.ThreadingTCPServer((HOST,PORT),MyTcpHandler)
    server.serve_forever()
    server.allow_reuse_address()

#client.py
import socket,os,json,hashlib
class Ftpclient(object):
   
def __init__(self):
        self.client=socket.socket()
   
def help(self):
       
print("put/get/cd/pwd")
   
def connect(self,ip,port):
        self.client.connect((ip,port))
   
def interactive(self):
       
while True:
            cmd=
input("请输入要上传下载的文件:").strip()
           
if len(cmd)==0:continue
           
cmd_str=cmd.split()[0]
           
if hasattr(self,cmd_str):
                func=
getattr(self,cmd_str)
                func(cmd)
           
else:
                self.help()
   
def put(self,*args):
        cmd_split=args[
0].split()
       
if len(cmd_split)>1:
            filename=cmd_split[
1]
           
if os.path.isfile(filename):
                filesize=os.stat(filename).st_size
                msg_dic={
                   
"action":"put",
                   
"filename":filename,
                   
"size":filesize,
                   
"overridden":True
               
}
                self.client.send(json.dumps(msg_dic).encode(
"utf-8"))
               
server_response=self.client.recv(1024)
                m=hashlib.md5()
                f=
open(filename,'rb')
               
for line in f:
                    m.update(line)
                    self.client.send(line)
               
else:
                   
print("file md5:", m.hexdigest())
                    self.client.recv(
1024)
                    self.client.send(m.hexdigest().encode(
"utf-8"))
                   
print("file upload success...")
                    f.close()
           
else:
               
print(filename,"is not exist")
   
def get(self,*args):
        cmd_split=args[
0].split()
       
if len(cmd_split)>1:
            filename=cmd_split[
1]
            msg_dic={
                   
"action":"get",
                   
"filename":filename,
                   
"overridden":True
           
}
            self.client.send(json.dumps(msg_dic).encode(
"utf-8"))
            received_size =
0
           
m = hashlib.md5()
            f=
open(filename+".download",'wb')
            server_response=self.client.recv(
1024)
            file_total_size=
int(server_response.decode())
            self.client.send(
b"ready to recv file")
           
while received_size < file_total_size:
               
if file_total_size - received_size > 1024:
                    size=
1024
               
else:
                    size=file_total_size - received_size
                data = self.client.recv(size)
                m.update(data)
                f.write(data)
                received_size +=
len(data)
           
else:
                download_file_md5 = m.hexdigest()
                self.client.send(
b"ready recive your md5")
                servrer_file_md5 = self.client.recv(
1024)
               
print("server_file_md5:",servrer_file_md5.decode())
               
print("download_file_md5:",download_file_md5)
                
print("file [%s] has download..."%filename)
               
print("源文件和下载文件大小分别为:",received_size,file_total_size)
ftp=Ftpclient()
ftp.connect(
"localhost",5566)
ftp.interactive()

相关内容

热门资讯

黄河科技学院2026年招生简章 长按图片识别二维码或点击 “阅读原文” 查看电子招生简章。
医路起航,从“心” 开始!黄河... 6月1日上午,黄河科技学院附属医院2022级临床医学本科实习生入院岗前培训在大医讲堂顺利举办。院领导...
问题居然在实体卡槽上!美版iP... 6月2日消息,日前,又有博主提前把还没发布的iPhone 18 Pro电池参数给曝光了出来,根据爆料...
2026年618手机购机攻略:... 每年618都是手机换新的最佳窗口期,2026年大促叠加数码国补、平台满减、品牌降价三重福利,新机老机...
8.99万打穿底价!史无前例的... 科技的浪漫,不是将人类送入遥不可及的星辰,而是把曾经高不可攀的科幻,变成柴米油盐里的“论斤卖”。 如...
苹果首款折叠屏iPhone U... PChome 6月2日消息,据知名数码博主@i冰宇宙最新曝光的机模谍照,苹果首款折叠屏手机(内部命名...
特朗普:美伊暂停谈判?假新闻! 当地时间6月2日,美国总统特朗普否认伊朗已停止与调解方沟通,称美国与伊朗之间的对话一直在持续进行。特...
白宫记协晚宴改期至7月,特朗普... 当地时间6月2日,美国总统特朗普表示,此前因突发暴力事件而中断的白宫记协晚宴已确定改期至7月24日举...
特朗普,开掉“不听话的情报头子... 苦撑15个月后,美国国家情报总监加巴德即将离开白宫。她对辞职的解释是:丈夫近期确诊罕见骨癌,自己希望...
面粉染头模仿黄仁勋,农村青年走... “先赔偿5000元肖像侵权使用费,再删除所有视频,要么就起诉你。”这是黄仁勋的模仿者杨洋最近频繁收到...