操作系统基础

操作系统:(Operating System,简称 OS)是管理和控制计算机硬件与软件资源的计算机程序,是直接运行在“裸机”上的最基本的系统软件,任何其他软件都必须在操作系统的支持下才能运行。
python网络编程 - 图1
操作系统应该分成两部分功能

  1. 隐藏了丑陋的硬件调用接口(键盘、鼠标、音箱等等怎么实现的,就不需要你管了),为应用程序员提供调用硬件资源的更好,更简单,更清晰的模型(系统调用接口)。应用程序员有了这些接口后,就不用再考虑操作硬件的细节,专心开发自己的应用程序即可。
    例如:操作系统提供了文件这个抽象概念,对文件的操作就是对磁盘的操作,有了文件我们无需再去考虑关于磁盘的读写控制(比如控制磁盘转动,移动磁头读写数据等细节),
  2. 将应用程序对硬件资源的竞态请求变得有序化
    例如:很多应用软件其实是共享一套计算机硬件,比方说有可能有三个应用程序同时需要申请打印机来输出内容,那么 a 程序竞争到了打印机资源就打印,然后可能是 b 竞争到打印机资源,也可能是 c,这就导致了无序,打印机可能打印一段 a 的内容然后又去打印 c…, 操作系统的一个功能就是将这种无序变得有序。

socket

回顾一下五层通讯流程
python网络编程 - 图2
但实际上从传输层开始以及以下,都是操作系统帮咱们完成的,下面的各种包头封装的过程
python网络编程 - 图3
Socket 又称为套接字,它是应用层与 TCP/IP 协议族通信的中间软件抽象层,它是一组接口。在设计模式中,Socket 其实就是一个门面模式,它把复杂的 TCP/IP 协议族隐藏在 Socket 接口后面,对用户来说,一组简单的接口就是全部,让 Socket 去组织数据,以符合指定的协议。当我们使用不同的协议进行通信时就得使用不同的接口,还得处理不同协议的各种细节,这就增加了开发的难度,软件也不易于扩展(就像我们开发一套公司管理系统一样,报账、会议预定、请假等功能不需要单独写系统,而是一个系统上多个功能接口,不需要知道每个功能如何去实现的)。于是 UNIX BSD 就发明了 socket 这种东西,socket 屏蔽了各个协议的通信细节,使得程序员无需关注协议本身,直接使用 socket 提供的接口来进行互联的不同主机间的进程的通信。这就好比操作系统给我们提供了使用底层硬件功能的系统调用,通过系统调用我们可以方便的使用磁盘(文件操作),使用内存,而无需自己去进行磁盘读写,内存管理。socket 其实也是一样的东西,就是提供了 tcp/ip 协议的抽象,对外提供了一套接口,同过这个接口就可以统一、方便的使用 tcp/ip 协议的功能了。
其实站在你的角度上看,socket 就是一个模块。我们通过调用模块中已经实现的方法建立两个进程之间的连接和通信。也有人将 socket 说成 ip+port,因为 ip 是用来标识互联网中的一台主机的位置,而 port 是用来标识这台机器上的一个应用程序。 所以我们只要确立了 ip 和 port 就能找到一个应用程序,并且使用 socket 模块来与之通信。

套接字家族

套接字起源于 20 世纪 70 年代加利福尼亚大学伯克利分校版本的 Unix, 即人们所说的 BSD Unix。 因此, 有时人们也把套接字称为“伯克利套接字”或“BSD 套接字”。一开始, 套接字被设计用在同 一台主机上多个应用程序之间的通讯。这也被称进程间通讯, 或 IPC。套接字有两种(或者称为有两个种族), 分别是基于文件型的和基于网络型的。

基于文件类型的套接字家族

套接字家族的名字:AF_UNIX
unix 一切皆文件,基于文件的套接字调用的就是底层的文件系统来取数据,两个套接字进程运行在同一机器,可以通过访问同一个文件系统间接完成通信

基于网络类型的套接字家族

套接字家族的名字:AF_INET
(还有 AF_INET6 被用于 ipv6,还有一些其他的地址家族,不过,他们要么是只用于某个平台,要么就是已经被废弃,或者是很少被使用,或者是根本没有实现,所有地址家族中,AF_INET 是使用最广泛的一个,python 支持很多种地址家族,但是由于我们只关心网络编程,所以大部分时候我么只使用 AF_INET)

套接字的工作流程(基于 TCP 和 UDP 两个协议)

TCP 和 UDP 对比

  • TCP(Transmission Control Protocol)
    • 可靠的、面向连接的协议(eg: 打电话)、传输效率低全双工通信(发送缓存 & 接收缓存)、面向字节流。使用 TCP 的应用:Web 浏览器;文件传输程序。
  • UDP(User Datagram Protocol)
    • 不可靠的、无连接的服务,传输效率高(发送前时延小),一对一、一对多、多对一、多对多、面向报文(数据包),尽最大努力服务,无拥塞控制。使用 UDP 的应用:域名系统 (DNS);视频流;IP 语音(VoIP)。

TCP 协议下的 socket

python网络编程 - 图4
服务器端先初始化 Socket,然后与端口绑定(bind),对端口进行监听(listen),调用 accept 阻塞,等待客户端连接。在这时如果有个客户端初始化一个 Socket,然后连接服务器(connect),如果连接成功,这时客户端与服务器端的连接就建立了。客户端发送数据请求,服务器端接收请求并处理请求,然后把回应数据发送给客户端,客户端读取数据,最后关闭连接,一次交互结束

  1. import socket
  2. socket.socket(socket_family,socket_type,protocal=0)
  3. # socket_family 可以是 AF_UNIX 或 AF_INET。socket_type 可以是 SOCK_STREAM 或 SOCK_DGRAM。protocol 一般不填,默认值为 0。
  4. # 获取tcp/ip套接字
  5. tcpSock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  6. # 获取udp/ip套接字
  7. udpSock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
  8. # 由于 socket 模块中有太多的属性。我们在这里破例使用了'from module import *'语句。使用 'from socket import *',我们就把 socket 模块里的所有属性都带到我们的命名空间里了,这样能 大幅减短我们的代码。
  9. # 例如
  10. tcpSock = socket(AF_INET, SOCK_STREAM)

Python
Copy
服务端套接字函数

s.bind() 绑定 (主机, 端口号) 到套接字
s.listen() 开始 TCP 监听
s.accept() 被动接受 TCP 客户的连接,(阻塞式)等待连接的到来

客户端套接字函数

s.connect() 主动初始化 TCP 服务器连接
s.connect_ex() connect()函数的扩展版本, 出错时返回出错码, 而不是抛出异常

公共用途的套接字函数

s.recv() 接收 TCP 数据
s.send() 发送 TCP 数据(send 在待发送数据量大于己端缓存区剩余空间时, 数据丢失, 不会发完)
s.sendall() 发送完整的 TCP 数据(本质就是循环调用 send,sendall 在待发送数据量大于己端缓存区剩余空间时, 数据不丢失, 循环调用 send 直到发完)
s.recvfrom() 接收 UDP 数据
s.sendto() 发送 UDP 数据
s.getpeername() 连接到当前套接字的远端的地址
s.getsockname() 当前套接字的地址
s.getsockopt() 返回指定套接字的参数
s.setsockopt() 设置指定套接字的参数
s.close() 关闭套接字

面向锁的套接字方法

s.setblocking() 设置套接字的阻塞与非阻塞模式
s.settimeout() 设置阻塞套接字操作的超时时间
s.gettimeout() 得到阻塞套接字操作的超时时间

面向文件的套接字的函数

s.fileno() 套接字的文件描述符
s.makefile() 创建一个与该套接字相关的文件

第一版,单个客户端与服务端通信
服务端

  1. import socket
  2. phone = socket.socket(socket.AF_INET,socket.SOCK_STREAM) # 买电话
  3. phone.bind(('127.0.0.1',8080)) # 0 ~ 65535 1024之前系统分配好的端口 绑定电话卡
  4. phone.listen(5) # 同一时刻有5个请求,但是可以有N多个链接。 开机。
  5. conn, client_addr = phone.accept() # 接电话
  6. print(conn, client_addr, sep='\n')
  7. from_client_data = conn.recv(1024) # 一次接收的最大限制 bytes
  8. print(from_client_data.decode('utf-8'))
  9. conn.send(from_client_data.upper())
  10. conn.close() # 挂电话
  11. phone.close() # 关机

Python
Copy
客户端

  1. import socket
  2. phone = socket.socket(socket.AF_INET,socket.SOCK_STREAM) # 买电话
  3. phone.connect(('127.0.0.1',8080)) # 与客户端建立连接, 拨号
  4. phone.send('hello'.encode('utf-8'))
  5. from_server_data = phone.recv(1024)
  6. print(from_server_data)
  7. phone.close() # 挂电话

Python
Copy
第二版,通信循环
服务端

  1. import socket
  2. phone = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
  3. phone.bind(('127.0.0.1',8080))
  4. phone.listen(5)
  5. conn,client_addr = phone.accept()
  6. print(conn,client_addr,sep='\n')
  7. while 1:
  8. try:
  9. from_client_data = conn.recv(1024)
  10. print(from_client_data.decode('utf-8'))
  11. conn.send(from_client_data.upper())
  12. except ConnectionResetError:
  13. break
  14. conn.close()
  15. phone.close()

Python
Copy
客户端

  1. import socket
  2. phone = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
  3. phone.connect(('127.0.0.1',8080))
  4. while 1:
  5. client_data = input('>>> ')
  6. phone.send(client_data.encode('utf-8'))
  7. from_server_data = phone.recv(1024)
  8. print(from_server_data.decode('utf-8'))
  9. phone.close()

Python
Copy
第三版, 通信,连接循环
服务端

  1. import socket
  2. phone = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
  3. phone.bind(('127.0.0.1',8080))
  4. phone.listen(5)
  5. while 1:
  6. conn,client_addr = phone.accept()
  7. print(conn,client_addr,sep='\n')
  8. while 1:
  9. try:
  10. from_client_data = conn.recv(1024)
  11. print(from_client_data.decode('utf-8'))
  12. conn.send(from_client_data.upper())
  13. except:
  14. break
  15. conn.close()
  16. phone.close()

Python
Copy
客户端

  1. import socket
  2. phone = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
  3. phone.connect(('127.0.0.1',8080))
  4. while 1:
  5. client_data = input('>>> ')
  6. phone.send(client_data.encode('utf-8'))
  7. if client_data == 'q':break
  8. from_server_data = phone.recv(1024)
  9. print(from_server_data.decode('utf-8'))
  10. phone.close()

Python
Copy
远程执行命令的示例:

  1. import socket
  2. import subprocess
  3. phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  4. phone.bind(('127.0.0.1', 8080))
  5. phone.listen(5)
  6. while 1: # 循环连接客户端
  7. conn, client_addr = phone.accept()
  8. print(client_addr)
  9. while 1:
  10. try:
  11. cmd = conn.recv(1024)
  12. ret = subprocess.Popen(cmd.decode('utf-8'), shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
  13. correct_msg = ret.stdout.read()
  14. error_msg = ret.stderr.read()
  15. conn.send(correct_msg + error_msg)
  16. except ConnectionResetError:
  17. break
  18. conn.close()
  19. phone.close()

Python
Copy
客户端

  1. import socket
  2. phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 买电话
  3. phone.connect(('127.0.0.1', 8080)) # 与客户端建立连接, 拨号
  4. while 1:
  5. cmd = input('>>>')
  6. phone.send(cmd.encode('utf-8'))
  7. from_server_data = phone.recv(1024)
  8. print(from_server_data.decode('gbk'))
  9. phone.close() # 挂电话

Python
Copy

UDP 协议下的 socket

udp 是无链接的,先启动哪一端都不会报错
python网络编程 - 图5
服务器端先初始化 Socket,然后与端口绑定 (bind),recvform 接收消息,这个消息有两项,消息内容和对方客户端的地址,然后回复消息时也要带着你收到的这个客户端的地址,发送回去,最后关闭连接,一次交互结束
服务端

  1. import socket
  2. udp_sk = socket.socket(type=socket.SOCK_DGRAM) #创建一个服务器的套接字
  3. udp_sk.bind(('127.0.0.1',9000)) #绑定服务器套接字
  4. msg,addr = udp_sk.recvfrom(1024)
  5. print(msg)
  6. udp_sk.sendto(b'hi',addr) # 对话(接收与发送)
  7. udp_sk.close() # 关闭服务器套接字

Python
Copy
客户端

  1. import socket
  2. ip_port=('127.0.0.1',9000)
  3. udp_sk=socket.socket(type=socket.SOCK_DGRAM)
  4. udp_sk.sendto(b'hello',ip_port)
  5. back_msg,addr=udp_sk.recvfrom(1024)
  6. print(back_msg.decode('utf-8'),addr)

Python
Copy
类似于 qq 聊天的代码示例
服务端

  1. import socket
  2. ip_port=('127.0.0.1',8081)
  3. udp_server_sock=socket.socket(socket.AF_INET,socket.SOCK_DGRAM) #DGRAM:datagram 数据报文的意思,象征着UDP协议的通信方式
  4. udp_server_sock.bind(ip_port)#你对外提供服务的端口就是这一个,所有的客户端都是通过这个端口和你进行通信的
  5. while True:
  6. qq_msg,addr=udp_server_sock.recvfrom(1024)# 阻塞状态,等待接收消息
  7. print('来自[%s:%s]的一条消息:\033[1;34;43m%s\033[0m' %(addr[0],addr[1],qq_msg.decode('utf-8')))
  8. back_msg=input('回复消息: ').strip()
  9. udp_server_sock.sendto(back_msg.encode('utf-8'),addr)

Python
Copy
客户端

  1. import socket
  2. BUFSIZE=1024
  3. udp_client_socket=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
  4. qq_name_dic={
  5. 'taibai':('127.0.0.1',8081),
  6. 'Jedan':('127.0.0.1',8081),
  7. 'Jack':('127.0.0.1',8081),
  8. 'John':('127.0.0.1',8081),
  9. }
  10. while True:
  11. while 1:
  12. qq_name=input('请选择聊天对象: ').strip()
  13. if qq_name not in qq_name_dic:
  14. print('没有这个聊天对象')
  15. continue
  16. break
  17. while True:
  18. msg='发给'+ qq_name + ': ' + input('请输入消息,回车发送,输入q结束和他的聊天: ').strip()
  19. if msg == 'q':break
  20. if not msg:continue
  21. udp_client_socket.sendto(msg.encode('utf-8'),qq_name_dic[qq_name])# 必须带着自己的地址,这就是UDP不一样的地方,不需要建立连接,但是要带着自己的地址给服务端,否则服务端无法判断是谁给我发的消息,并且不知道该把消息回复到什么地方,因为我们之间没有建立连接通道
  22. back_msg,addr=udp_client_socket.recvfrom(BUFSIZE)# 同样也是阻塞状态,等待接收消息
  23. print('来自[%s:%s]的一条消息:\033[1;34;43m%s\033[0m' %(addr[0],addr[1],back_msg.decode('utf-8')))
  24. udp_client_socket.close()

Python
Copy
自制时间服务器
服务端

  1. from socket import *
  2. from time import strftime
  3. import time
  4. ip_port = ('127.0.0.1', 9000)
  5. bufsize = 1024
  6. tcp_server = socket(AF_INET, SOCK_DGRAM)
  7. tcp_server.setsockopt(SOL_SOCKET,SO_REUSEADDR,1)
  8. tcp_server.bind(ip_port)
  9. while True:
  10. msg, addr = tcp_server.recvfrom(bufsize)
  11. print('===>', msg.decode('utf-8'))
  12. stru_time = time.localtime() #当前的结构化时间
  13. if not msg:
  14. time_fmt = '%Y-%m-%d %X'
  15. else:
  16. time_fmt = msg.decode('utf-8')
  17. back_msg = strftime(time_fmt,stru_time)
  18. print(back_msg,type(back_msg))
  19. tcp_server.sendto(back_msg.encode('utf-8'), addr)
  20. tcp_server.close()

Python
Copy
客户端

  1. from socket import *
  2. ip_port=('127.0.0.1',9000)
  3. bufsize=1024
  4. tcp_client=socket(AF_INET,SOCK_DGRAM)
  5. while True:
  6. msg=input('请输入时间格式(例%Y %m %d)>>: ').strip()
  7. tcp_client.sendto(msg.encode('utf-8'),ip_port)
  8. data=tcp_client.recv(bufsize)
  9. print('当前日期:',str(data,encoding='utf-8'))

Python
Copy

粘包

python网络编程 - 图6
每个 socket 被创建后,都会分配两个缓冲区,输入缓冲区和输出缓冲区。
write()/send() 并不立即向网络中传输数据,而是先将数据写入缓冲区中,再由 TCP 协议将数据从缓冲区发送到目标机器。一旦将数据写入到缓冲区,函数就可以成功返回,不管它们有没有到达目标机器,也不管它们何时被发送到网络,这些都是 TCP 协议负责的事情。
TCP 协议独立于 write()/send() 函数,数据有可能刚被写入缓冲区就发送到网络,也可能在缓冲区中不断积压,多次写入的数据被一次性发送到网络,这取决于当时的网络情况、当前线程是否空闲等诸多因素,不由程序员控制。
read()/recv() 函数也是如此,也从输入缓冲区中读取数据,而不是直接从网络中读取。
这些 I / O 缓冲区特性可整理如下:

  1. I/ O 缓冲区在每个 TCP 套接字中单独存在;
  2. I/ O 缓冲区在创建套接字时自动生成;
  3. 即使关闭套接字也会继续传送输出缓冲区中遗留的数据;
  4. 关闭套接字将丢失输入缓冲区中的数据。

两种情况下会发生粘包

  1. 接收方没有及时接收缓冲区的包,造成多个包接收(客户端发送了一段数据,服务端只收了一小部分,服务端下次再收的时候还是从缓冲区拿上次遗留的数据,产生粘包)

服务端

  1. import socket
  2. import subprocess
  3. phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  4. phone.bind(('127.0.0.1', 8080))
  5. phone.listen(5)
  6. while 1: # 循环连接客户端
  7. conn, client_addr = phone.accept()
  8. print(client_addr)
  9. while 1:
  10. try:
  11. cmd = conn.recv(1024)
  12. ret = subprocess.Popen(cmd.decode('utf-8'), shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
  13. correct_msg = ret.stdout.read()
  14. error_msg = ret.stderr.read()
  15. conn.send(correct_msg + error_msg)
  16. except ConnectionResetError:
  17. break
  18. conn.close()
  19. phone.close()

Python
Copy
客户端

  1. import socket
  2. phone = socket.socket(socket.AF_INET,socket.SOCK_STREAM) # 买电话
  3. phone.connect(('127.0.0.1',8080)) # 与客户端建立连接, 拨号
  4. while 1:
  5. cmd = input('>>>')
  6. phone.send(cmd.encode('utf-8'))
  7. from_server_data = phone.recv(1024)
  8. print(from_server_data.decode('gbk'))
  9. phone.close()

Python
Copy

  1. 发送端需要等缓冲区满才发送出去,造成粘包(发送数据时间间隔很短,数据也很小,会合到一起,产生粘包)

服务端

  1. import socket
  2. phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  3. phone.bind(('127.0.0.1', 8080))
  4. phone.listen(5)
  5. conn, client_addr = phone.accept()
  6. frist_data = conn.recv(1024)
  7. print('1:',frist_data.decode('utf-8')) # 1: helloworld
  8. second_data = conn.recv(1024)
  9. print('2:',second_data.decode('utf-8'))
  10. conn.close()
  11. phone.close()

Python
Copy
客户端

  1. import socket
  2. phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  3. phone.connect(('127.0.0.1', 8080))
  4. phone.send(b'hello')
  5. phone.send(b'world')
  6. phone.close()

Python
Copy

粘包的解决方案

struct 模块

该模块可以把一个类型,如数字,转成固定长度的 bytes
python网络编程 - 图7

  1. import struct
  2. # 将一个数字转化成等长度的bytes类型。
  3. ret = struct.pack('i', 183346)
  4. print(ret, type(ret), len(ret))
  5. # 通过unpack反解回来
  6. ret1 = struct.unpack('i',ret)[0]
  7. print(ret1, type(ret1), len(ret1))
  8. # 但是通过struct 处理不能处理太大

Python
Copy
方案一:

  1. import socket
  2. import subprocess
  3. import struct
  4. phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  5. phone.bind(('127.0.0.1', 8080))
  6. phone.listen(5)
  7. while 1:
  8. conn, client_addr = phone.accept()
  9. print(client_addr)
  10. while 1:
  11. try:
  12. cmd = conn.recv(1024)
  13. ret = subprocess.Popen(cmd.decode('utf-8'), shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
  14. correct_msg = ret.stdout.read()
  15. error_msg = ret.stderr.read()
  16. # 1 制作固定报头
  17. total_size = len(correct_msg) + len(error_msg)
  18. header = struct.pack('i', total_size)
  19. # 2 发送报头
  20. conn.send(header)
  21. # 发送真实数据:
  22. conn.send(correct_msg)
  23. conn.send(error_msg)
  24. except ConnectionResetError:
  25. break
  26. conn.close()
  27. phone.close()
  28. # 但是low版本有问题:
  29. # 1,报头不只有总数据大小,而是还应该有MD5数据,文件名等等一些数据。
  30. # 2,通过struct模块直接数据处理,不能处理太大。

Python
Copy
客户端

  1. import socket
  2. import struct
  3. phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  4. phone.connect(('127.0.0.1', 8080))
  5. while 1:
  6. cmd = input('>>>').strip()
  7. if not cmd: continue
  8. phone.send(cmd.encode('utf-8'))
  9. # 1,接收固定报头
  10. header = phone.recv(4)
  11. # 2,解析报头
  12. total_size = struct.unpack('i', header)[0]
  13. # 3,根据报头信息,接收真实数据
  14. recv_size = 0
  15. res = b''
  16. while recv_size < total_size:
  17. recv_data = phone.recv(1024)
  18. res += recv_data
  19. recv_size += len(recv_data)
  20. print(res.decode('gbk'))
  21. phone.close()

Python
Copy
方案二:可自定制报头

  1. 整个流程的大致解释:
  2. 我们可以把报头做成字典,字典里包含将要发送的真实数据的描述信息(大小啊之类的),然后json序列化,然后用struck将序列化后的数据长度打包成4个字节。
  3. 我们在网络上传输的所有数据 都叫做数据包,数据包里的所有数据都叫做报文,报文里面不止有你的数据,还有ip地址、mac地址、端口号等等,其实所有的报文都有报头,这个报头是协议规定的,看一下
  4. 发送时:
  5. 先发报头长度
  6. 再编码报头内容然后发送
  7. 最后发真实内容
  8. 接收时:
  9. 先手报头长度,用struct取出来
  10. 根据取出的长度收取报头内容,然后解码,反序列化
  11. 从反序列化的结果中取出待取数据的描述信息,然后去取真实的数据内容

服务端

  1. import socket
  2. import subprocess
  3. import struct
  4. import json
  5. phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  6. phone.bind(('127.0.0.1', 8080))
  7. phone.listen(5)
  8. while 1:
  9. conn, client_addr = phone.accept()
  10. print(client_addr)
  11. while 1:
  12. try:
  13. cmd = conn.recv(1024)
  14. ret = subprocess.Popen(cmd.decode('utf-8'), shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
  15. correct_msg = ret.stdout.read()
  16. error_msg = ret.stderr.read()
  17. # 1 制作固定报头
  18. total_size = len(correct_msg) + len(error_msg)
  19. header_dict = {
  20. 'md5': 'fdsaf2143254f',
  21. 'file_name': 'f1.txt',
  22. 'total_size': total_size,
  23. }
  24. header_dict_json = json.dumps(header_dict) # str
  25. bytes_headers = header_dict_json.encode('utf-8')
  26. header_size = len(bytes_headers)
  27. header = struct.pack('i', header_size)
  28. # 2 发送报头长度
  29. conn.send(header)
  30. # 3 发送报头
  31. conn.send(bytes_headers)
  32. # 4 发送真实数据:
  33. conn.send(correct_msg)
  34. conn.send(error_msg)
  35. except ConnectionResetError:
  36. break
  37. conn.close()
  38. phone.close()

Python
Copy
客户端

  1. import socket
  2. import struct
  3. import json
  4. phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  5. phone.connect(('127.0.0.1', 8080))
  6. while 1:
  7. cmd = input('>>>').strip()
  8. if not cmd: continue
  9. phone.send(cmd.encode('utf-8'))
  10. # 1,接收固定报头
  11. header_size = struct.unpack('i', phone.recv(4))[0]
  12. # 2,解析报头长度
  13. header_bytes = phone.recv(header_size)
  14. header_dict = json.loads(header_bytes.decode('utf-8'))
  15. # 3,收取报头
  16. total_size = header_dict['total_size']
  17. # 3,根据报头信息,接收真实数据
  18. recv_size = 0
  19. res = b''
  20. while recv_size < total_size:
  21. recv_data = phone.recv(1024)
  22. res += recv_data
  23. recv_size += len(recv_data)
  24. print(res.decode('gbk'))
  25. phone.close()

Python
Copy
FTP 上传下载文件的代码(简单版)
服务端

  1. import socket
  2. import struct
  3. import json
  4. sk = socket.socket()
  5. # buffer = 4096 # 当双方的这个接收发送的大小比较大的时候,就像这个4096,就会丢数据,这个等我查一下再告诉大家,改小了就ok的,在linux上也是ok的。
  6. buffer = 1024 #每次接收数据的大小
  7. sk.bind(('127.0.0.1',8090))
  8. sk.listen()
  9. conn,addr = sk.accept()
  10. #接收
  11. head_len = conn.recv(4)
  12. head_len = struct.unpack('i',head_len)[0] #解包
  13. json_head = conn.recv(head_len).decode('utf-8') #反序列化
  14. head = json.loads(json_head)
  15. filesize = head['filesize']
  16. with open(head['filename'],'wb') as f:
  17. while filesize:
  18. if filesize >= buffer: #>=是因为如果刚好等于的情况出现也是可以的。
  19. content = conn.recv(buffer)
  20. f.write(content)
  21. filesize -= buffer
  22. else:
  23. content = conn.recv(buffer)
  24. f.write(content)
  25. break
  26. conn.close()
  27. sk.close()

Python
Copy
客户端

  1. import os
  2. import json
  3. import socket
  4. import struct
  5. sk = socket.socket()
  6. sk.connect(('127.0.0.1',8090))
  7. buffer = 1024 #读取文件的时候,每次读取的大小
  8. head = {
  9. 'filepath':r'C:\Users\Aaron\Desktop\新建文件夹', #需要下载的文件路径,也就是文件所在的文件夹
  10. 'filename':'config', #改成上面filepath下的一个文件
  11. 'filesize':None,
  12. }
  13. file_path = os.path.join(head['filepath'],head['filename'])
  14. filesize = os.path.getsize(file_path)
  15. head['filesize'] = filesize
  16. # json_head = json.dumps(head,ensure_ascii=False) #字典转换成字符串
  17. json_head = json.dumps(head) #字典转换成字符串
  18. bytes_head = json_head.encode('utf-8') #字符串转换成bytes类型
  19. print(json_head)
  20. print(bytes_head)
  21. #计算head的长度,因为接收端先接收我们自己定制的报头,对吧
  22. head_len = len(bytes_head) #报头长度
  23. pack_len = struct.pack('i',head_len)
  24. print(head_len)
  25. print(pack_len)
  26. sk.send(pack_len) #先发送报头长度
  27. sk.send(bytes_head) #再发送bytes类型的报头
  28. #即便是视频文件,也是可以按行来读取的,也可以readline,也可以for循环,但是读取出来的数据大小就不固定了,影响效率,有可能读的比较小,也可能很大,像视频文件一般都是一行的二进制字节流。
  29. #所有我们可以用read,设定一个一次读取内容的大小,一边读一边发,一边收一边写
  30. with open(file_path,'rb') as f:
  31. while filesize:
  32. if filesize >= buffer: #>=是因为如果刚好等于的情况出现也是可以的。
  33. content = f.read(buffer) #每次读取出来的内容
  34. sk.send(content)
  35. filesize -= buffer #每次减去读取的大小
  36. else: #那么说明剩余的不够一次读取的大小了,那么只要把剩下的读取出来发送过去就行了
  37. content = f.read(filesize)
  38. sk.send(content)
  39. break
  40. sk.close()