python3基于TCP实现CS架构文件传输
本文实例为大家分享了python3实现CS架构文件传输的具体代码,供大家参考,具体内容如下
1、目标:
基于tcp实现CS架构的文件传输
指令列表:(1)get:从服务器端下载文件
(2)put:向服务器端上传文件
(3)list:获得服务器端的目录
2、socket模块函数:
(1)send和sendall:send的作用是发送TCP数据,返回发送的数据大小。send函数不保证将所有数据全部发送,因此可能需要重复多次才能完成所有数据的发送。sendall的作用是发送完整的TCP数据,成功时返回None,失败时抛出异常
(2)bind:在服务器端使用,用于将socket绑定在一个特定的ip地址和端口上。在《UNIX网络编程》一书中提到,如果调用connect或者listen之前没有bind一个特定的端口,内核会为相应的套接字分配一个随机的端口。因此,在客户端调用connect函数之前不需要bind
(3)listen:通过参数设定服务器端最多可以接受几个客户端的连接,但是只有在完成与第一个客户端的传送后才会进行与第二个客户端的传送
3、代码:
(1)服务器端:
importsocket importos importsys importtime Host='127.0.0.1' Port=12000 Addr=(Host,Port) sockListener=socket.socket(socket.AF_INET,socket.SOCK_STREAM) sockListener.bind(Addr) sockListener.listen(5) defrecvfile(cliSocket): print('startreveivingfile...') msg='noproblem' msg_utf8=msg.encode(encoding="utf-8") cliSocket.send(msg_utf8) filename_utf8=clientSocket.recv(4096) filename=filename_utf8.decode() f=open(filename,'wb') whileTrue: data_utf8=cliSocket.recv(4096) data=data_utf8.decode() ifdata=='EOF': print('receivedfilesuccessfully') break f.write(data_utf8) f.close() defsendfile(cliSocket): print('startsendingfile...') msg='noproblem' msg_utf8=msg.encode(encoding="utf-8") cliSocket.send(msg_utf8) filename_utf8=cliSocket.recv(4096) filename=filename_utf8.decode() f=open(filename,'rb') whileTrue: data=f.read(4096) ifnotdata: break cliSocket.sendall(data) f.close() time.sleep(1) msg='EOF' msg_utf8=msg.encode(encoding="utf-8") cliSocket.sendall(msg_utf8) print('sentfilesuccessfully') defgetList(cliSocket): path=sys.path[0] every_file=os.listdir(path) forfilenameinevery_file: pathmsg=filename+'\n' pathmsg_utf8=pathmsg.encode(encoding="utf-8") cliSocket.sendall(pathmsg_utf8) time.sleep(1) msg='EOF' msg_utf8=msg.encode(encoding="utf-8") cliSocket.sendall(msg_utf8) print('allfilenameshavebeensend') whileTrue: print('waitingforconnection...') clientSocket,addr=sockListener.accept() print('...connectionfrom:',addr) whileTrue: msg_utf8=clientSocket.recv(4096) msg=msg_utf8.decode() ifmsg=='exit': print(addr,'closetheconnection') break elifmsg=='get': sendfile(clientSocket) elifmsg=='put': recvfile(clientSocket) elifmsg=='list': getList(clientSocket) else: print('clienterror!') break
(2)客户端:
importsocket importtime Host='127.0.0.1' Port=12000 Addr=(Host,Port) clientSocket=socket.socket(socket.AF_INET,socket.SOCK_STREAM) clientSocket.connect(Addr) defrecvfile(filename): print('startreceivingfile...') f=open(filename,'wb') filename_utf8=filename.encode(encoding="utf-8") clientSocket.sendall(filename_utf8) whileTrue: data_utf8=clientSocket.recv(4096) data=data_utf8.decode() ifdata=='EOF': print('receivefilesuccessfully') break f.write(data_utf8) f.close() defsendfile(filename): print('startsendingfile...') f=open(filename,'rb') filename_utf8=filename.encode(encoding="utf-8") clientSocket.sendall(filename_utf8) whileTrue: data=f.read(4096) ifnotdata: break clientSocket.sendall(data) f.close() time.sleep(1) endmsg='EOF' endmsg_utf8=endmsg.encode(encoding="utf-8") clientSocket.sendall(endmsg_utf8) print('sendfilesuccessfully') defconfirm(confirm_command): confirm_command_utf8=confirm_command.encode(encoding="utf-8") clientSocket.sendall(confirm_command_utf8) msg_utf8=clientSocket.recv(4096) msg=msg_utf8.decode() print('reveivemessage:',msg) ifmsg=='noproblem': returnTrue else: returnFalse defoperation1(filename): ifconfirm('get'): recvfile(filename) else: print('serveerror!') defoperation2(filename): ifconfirm('put'): sendfile(filename) else: print('serveerror!') defoperation3(act): ifact=='list': act_utf8=act.encode(encoding="utf-8") clientSocket.sendall(act_utf8) whileTrue: msg_utf8=clientSocket.recv(1024) msg=msg_utf8.decode() ifmsg=='EOF': break print(msg) else: print('wrongcommand!') try: whileTrue: command=input('>>>') ifnotcommand: continue elifcommand=='exit': command_utf8=command.encode(encoding="utf-8") clientSocket.sendall(command_utf8) print('theconnectionisclosed') break msg=command.split() iflen(msg)==2andmsg[0]=='get': operation1(msg[1]) eliflen(msg)==2andmsg[0]=='put': operation2(msg[1]) eliflen(msg)==1: operation3(msg[0]) else: print('wrongcommand!') exceptsocket.errorase: print('error:',e) print('anerrorcausestheconnectiontoclose!') clientSocket.close()
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持毛票票。