python基础 socket基础

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()