Python tcp传输代码实例解析

时间：2020-07-31 python教程查看: 740

socket是编程语言对tcp/ip协议的封装，使用其可以在两个主机之间传输文件，下面直接进入主题，下面示例是最简单的tcp传输代码示例，先启动server再启动client。需要注意python3socket只能传输byte类型，具体如何转换可以复习字符编码章节相关知识

sever端代码

# -*- coding:utf-8 -*-
import socket
server = socket.socket()
server.bind(("0.0.0.0", 8888))
server.listen(10)
while True: #循环监听连接过来的客户端
  conn,addr = server.accept()
  while True: #每个客户端连接过来以后持续交互，当客户端端断开后在和下一个客户端交互
    try:
      rec_data = conn.recv(4096) #4096接收缓冲区大小
      if len(rec_data) == 0:
        break
      print("rec_data:", rec_data.decode(encoding="utf-8"))
      conn.send(rec_data.upper())
    except ConnectionResetError as e:
      print("一个客户端关闭了连接")
      break
server.close()

client端代码

import socket
client = socket.socket()
client.connect(("127.0.0.1", 8888))
while True:
  msg = input()
  if len(msg) == 0:
    continue
  client.send(msg.encode(encoding="utf-8")) #如果发送中文需要对其进行encode，接收也需要decode；英文前面直接加b指定类型
  client_rec = client.recv(4096)
  print("client_rec:", client_rec)
client.close()

粘包解决方法：对于两条紧挨的send语句，会存在上一条send的内容与下一条send的内容混在一起的情况，这个是因为send只是将数据发送给os，具体os如何发送数据无法控制，使用下面方法能够解决两条send语句之间发送报文粘包问题

server端代码

import socket, os
server = socket.socket()
server.bind(("0.0.0.0", 8888))
server.listen(10)
while True: #循环监听连接过来的客户端
  conn,addr = server.accept()
  while True: #收到客户端发过来的get filename如果filename存在就将文件发出否则返回文件不存在
    try:
      rec_data = conn.recv(1024) #4096接收缓冲区大小
      if len(rec_data) == 0:
        break
      rec_str_list = rec_data.decode(encoding="utf-8").split(" ")
      filename = rec_str_list[-1]
      if os.path.isfile(filename):#判断文件在当前路径下存在句将文件传给客户端
        file_data = open(filename,encoding="utf-8").read()
        conn.send(str(len(file_data)).encode(encoding="utf-8")) #将文件长度传给客户端
        print(conn.recv(1024).decode(encoding="utf-8")) #客户端收到文件长度返回确认信息，server收到确认信息后再开始传送数据
        conn.sendall(file_data.encode(encoding="utf-8"))#将数据全部发个客户端，客户端判断数据接收完成后发送接收完成的信息
        print(conn.recv(1024).decode(encoding="utf-8")) #接收客户端接收完成的确认指令，避免两个send直接连在一起产生粘包
      else:#如果文件名不存在则返回文件不存在
        conn.send(b"file is not exit")
    except ConnectionResetError as e:
      print("一个客户端关闭了连接")
      break
server.close()

client端代码

import socket, os
#输入get 文件名，当文件存在server家目录下则返回文件，客户端将接收文件内容并打印，
client = socket.socket()
client.connect(("127.0.0.1", 8888))
while True:
  msg = input()
  if len(msg) == 0:
    continue
  client.send(msg.encode(encoding="utf-8")) #如果发送中文需要对其进行encode，接收也需要decode；英文前面直接加b指定类型
  client_rec = client.recv(1024).decode(encoding="utf-8")
  print("client_rec", client_rec)
  if client_rec == "file is not exit": #如果文件不存在则用户打印文件不存在，用户继续输入
    print(client_rec)
    continue
  client.send(client_rec.encode(encoding="utf-8"))#文件存在则将文件大小发回去
  rec_len = int(client_rec)
  print("rec_len", rec_len)
  file_len = 0
  new_file = open("new_file","a",encoding="utf-8")
  while file_len < rec_len:#开始循环接收文件直到接收的文件大小与server端发过来的大小一样,将接收到的数据写入新文件
    rec_data_tmp = client.recv(1024)
    file_len += len(rec_data_tmp)
    print("receive date len", file_len)
    new_file.write(rec_data_tmp.decode(encoding="utf-8"))
  client.send("receive all".encode(encoding="utf-8"))

client.close()

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持python博客。

展开全文