socket网络编程实践

目的：采用socket API编程，完成两个树莓派之间、或者树莓派与Ubuntu系统之间的网络文字通信（或聊天）

(一) 客户端和服务器端的连接

server代码

import socket           # 导入socket库
import time             # 导入time库
HOST =  'host'       # 连接本地服务器
PORT = 8001               # 设置端口号
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)       # 选择IPv4地址以及TCP协议
sock.bind((HOST, PORT))          # 绑定端口
sock.listen(5)                   # 监听这个端口，可连接最多5个设备
while True:  
    connection, address = sock.accept()              # 接受客户端的连接请求
    try:  
        connection.settimeout(10)                      # 设置10s时限
        buf = connection.recv(1024)                    # 接收数据实例化
        if buf:                                 # 接收成功
            print('connection success!')
            connection.send(b'welcome to server!')        # 发送消息，b为bytes类型
        else:                                   # 接收失败
            connection.send(b'please go out!')  
    except socket.timeout:                      # 超时
        print ('time out')
connection.close()                  # 关闭连接

Client代码

import socket
import time
 
HOST = 'host'  #服务器的ip
PORT = 8001
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  
sock.connect((HOST, PORT))  
time.sleep(2)  
sock.send(b'1')          # 发送信息
print (sock.recv(1024).decode())        # 打印接收消息，并且译码
sock.close()                      #关闭连接

先运行服务器端，再运行客户端

(二) 进行实时聊天

server端代码

import socket
host = 'host'
port = 12345
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.bind((host,port))
s.listen(1)
sock,addr = s.accept()
print('Connection built')
info = sock.recv(1024).decode()
while info != 'exit':
  print('client:'+info)
  send_mes = input()
  sock.send(send_mes.encode())
  if send_mes =='exit':
    break
  info = sock.recv(1024).decode()
sock.close()

client端代码

import socket
s= socket.socket()
host = '192.168.43.46'
port = 12345
s.connect((host,port))
print('Linked')
info = ''
while info != 'exit':
  print('server:'+info)
  send_mes=input()
  s.send(send_mes.encode())
  if send_mes =='exit':
    break
  info = s.recv(1024).decode()
s.close()

双方互相发送消息时，只能依次轮流发送，一方不可连续发送多条消息

这个代码可以直接通过teamviewer的文件传输传输到树莓派，同样可以成功运行，完成树莓派和PC之间的通信

(三) 用完成网络图像文件（或实时视频）的传输。

server端代码

import socket
import os
import hashlib
server = socket.socket()
server.bind(("192.168.43.168", 6969))
server.listen(5)  # 监听

print("监听开始..")
while True:
    conn, addr = server.accept()  # 等待连接
    print("conn:", conn, "\naddr:", addr)  # conn连接实例
    while True:
        data = conn.recv(1024)  # 接收
        if not data:  # 客户端已断开
            print("客户端断开连接")
            break
        print("收到的命令：", data.decode("utf-8"))
        cmd, filename = data.decode("utf-8").split(" ")
        if cmd =="get":
            if os.path.isfile(filename):  # 判断文件存在
                # 1.先发送文件大小，让客户端准备接收
                size = os.stat(filename).st_size  #获取文件大小
                conn.send(str(size).encode("utf-8"))  # 发送数据长度
                print("发送的大小：", size)
                # 2.发送文件内容
                conn.recv(1024)  # 接收确认
                m = hashlib.md5()
                f = open(filename, "rb")
                for line in f:
                    conn.send(line)  # 发送数据
                    m.update(line)
                f.close()
                # 3.发送md5值进行校验
                md5 = m.hexdigest()
                conn.send(md5.encode("utf-8"))  # 发送md5值
                print("md5:", md5)
server.close()

client端代码

import socket
import os
import hashlib
client = socket.socket()  # 生成socket连接对象
ip_port =("192.168.43.168", 6969)  # 地址和端口号
client.connect(ip_port)  # 连接
print("服务器已连接")
while True:
    content = input(">>")
    if len(content)==0: continue  # 如果传入空字符会阻塞
    if content.startswith("get"):
        client.send(content.encode("utf-8"))  # 传送和接收都是bytes类型
        # 1.先接收长度，建议8192
        server_response = client.recv(1024)
        file_size = int(server_response.decode("utf-8"))
        print("接收到的大小：", file_size)
        # 2.接收文件内容
        client.send("准备好接收".encode("utf-8"))  # 接收确认
        filename = "new" + content.split(" ")[1]
        f = open(filename, "wb")
        received_size = 0
        m = hashlib.md5()
        while received_size < file_size:
            size = 0  # 准确接收数据大小，解决粘包
            if file_size - received_size > 1024: # 多次接收
                size = 1024
            else:  # 最后一次接收完毕
                size = file_size - received_size
            data = client.recv(size)  # 多次接收内容，接收大数据
            data_len = len(data)
            received_size += data_len
            print("已接收：", int(received_size/file_size*100), "%")

            m.update(data)
            f.write(data)
        f.close()
        print("实际接收的大小:", received_size)  # 解码
       # 3.md5值校验
        md5_sever = client.recv(1024).decode("utf-8")
        md5_client = m.hexdigest()
        print("服务器发来的md5:", md5_sever)
        print("接收文件的md5:", md5_client)
        if md5_sever == md5_client:
            print("MD5值校验成功")
        else:
            print("MD5值校验失败")
client.close()

1. 在树莓派上运行client
2. pc上运行server
3. 客户端上运行后，输入get 文件名（该文件需要和运行客户端在同一个文件夹下）就可以进行文件传输了**

树莓派上接收到的图片和原图片