socket

【一】socket套接字

1.简介

# 需求:编写一个cs架构的程序 实现数据交互
    思考:需要编写代码操作OSI七层 相当的复杂
  由于操作OSI七层是所有cs架构的程序都需要经历的过程 所以有固定的模块
  socket套接字是一门技术
  socket模块>>>:提供了快捷方式 不需要自己处理每一层
"""
以后我们写软件连socket的影子都看不到 因为被封装起来
socket是最底层的原理 很多框架都封装了 其实我们不需要深入研究
"""
# 位于应用层之下 传输层之上

2.基本使用

1.socket模块---》先服务端后客户端
 步骤：
 #服务端
 1.首先要有个socket对象server = socket.socket()
 2.绑定地址server.bind(('127.0.0.1', 8080)) 
 3.listen和accept对应TCP三次握手服务端的两个状态(建立连接)
    连接池可以等待几个客户端server.listen(5)
    建立成功通道sock, addr = server.accept()---》sock相当于一个通道，addr就是地址
 4.借助通道进行接收服务
 5.关闭通道，关闭服务器

 #客户端
 1.首先要有个socket对象client = socket.socket()
 2.链接服务端的地址和端口client.connect(('127.0.0.1', 8080)) 
 3.发送请求服务内容,接收
 4.关闭请求结束

    
​

# 服务端
import socket
​
​
server = socket.socket()  # 买手机
"""
通过查看源码得知 
括号内不写参数默认就是基于网络的遵循TCP协议的套接字
"""
server.bind(('127.0.0.1', 8080))  # 插电话卡
"""
服务端应该具备的特征
    固定的地址
    ...  
127.0.0.1是计算机的本地回环地址 只有当前计算机本身可以访问
"""
server.listen(5)  # 开机
"""
半连接池(暂且忽略 先直接写 后面讲)
"""
sock, addr = server.accept()  # 等待并接听电话  没有人来就原地等待(程序阻塞)
"""
listen和accept对应TCP三次握手服务端的两个状态
"""
print(addr)  # 客户端的地址
data = sock.recv(1024)  # 听别人说话
print(data.decode('utf8'))
sock.send('你好啊'.encode('utf8'))  # 回复别人说的话
"""
recv和send接收和发送的都是bytes类型的数据
"""
sock.close()  # 挂电话
server.close()  # 关机
​

# 客户端
import socket
​
​
client = socket.socket()  # 产生一个socket对象
client.connect(('127.0.0.1', 8080))  # 根据服务端的地址链接
​
client.send(b'hello sweet heart!!!')  # 给服务端发送消息
data = client.recv(1024)  # 接收服务端回复的消息
print(data.decode('utf8'))
​
client.close()  # 关闭客户端
​
###########################
服务端与客户端首次交互
    一边是recv那么另一边必须是send  两边不能相同 否则就'冷战'了
###########################

3.通信循环

1.先解决消息固定的问题
    利用input获取用户输入
2.再解决通信循环的问题
    将双方用于数据交互的代码循环起来
​
#服务端
while True:
    data = sock.recv(1024)  # 听别人说话
    print(data.decode('utf8'))
    msg = input('请回复消息>>>:').strip()
    sock.send(msg.encode('utf8'))  # 回复别人说的话
#客户端
while True:
    msg = input('请输入你需要发送的消息>>>:').strip()
    client.send(msg.encode('utf8'))  # 给服务端发送消息
    data = client.recv(1024)  # 接收服务端回复的消息
    print(data.decode('utf8'))

4.代码优化及链接循环

1.发送消息不能为空
    统计长度并判断即可
2.反复重启服务端可能会报错>>>:address in use
  这个错在苹果电脑报的频繁 windows频率较少
  from socket import SOL_SOCKET,SO_REUSEADDR
  server.setsockopt(SOL_SOCKET,SO_REUSEADDR,1) # 在bind前加
3.链接循环(服务多个客户端)
  """
  如果是windows 客户端异常退出之后服务端会直接报错
    处理方式
        异常处理
  如果是mac或linux 服务端会接收到一个空消息
    处理方式
        len判断
  """
  客户端如果异常断开 服务端代码应该重新回到accept等待新的客人
​
# 目前我们的服务端只能实现一次服务一个人 不能做到同事服务多个 学了并发才可以实现

4.1服务端

server = socket.socket()  # 买手机
"""
通过查看源码得知 
括号内不写参数默认就是基于网络的遵循TCP协议的套接字
"""
server.setsockopt(SOL_SOCKET,SO_REUSEADDR,1) #就是它，在bind前加
server.bind(('127.0.0.1', 8080))  # 插电话卡
"""
服务端应该具备的特征
    固定的地址
    ...  
127.0.0.1是计算机的本地回环地址 只有当前计算机本身可以访问
"""
server.listen(5)  # 开机
"""
半连接池(暂且忽略 先直接写 后面讲)
"""
while True:
    sock, addr = server.accept()  # 等待并接听电话  没有人来就原地等待(程序阻塞)
    """
    listen和accept对应TCP三次握手服务端的两个状态
    """
    print(addr)  # 客户端的地址
    while True:
        try:
            data = sock.recv(1024)  # 听别人说话
            if len(data) == 0:
                break
            print(data.decode('utf8'))
            msg = input('请回复消息>>>:').strip()
            if len(msg) == 0:
                msg = '太忙 暂无消息'
            sock.send(msg.encode('utf8'))  # 回复别人说的话
        except Exception:
            break
        """
        recv和send接收和发送的都是bytes类型的数据
        """

​

4.2客户端

import socket
client = socket.socket()  # 产生一个socket对象
client.connect(('127.0.0.1', 8080))  # 根据服务端的地址链接
​
while True:
    msg = input('请输入你需要发送的消息>>>:').strip()
    # 用户如果什么都不输 直接回车 那么不该往下走
    if len(msg) == 0:
        continue
    client.send(msg.encode('utf8'))  # 给服务端发送消息
    data = client.recv(1024)  # 接收服务端回复的消息
    print(data.decode('utf8'))
​
client.close()  # 关闭客户端
​

半连接池

listen(5)一次只能五个等待客户端
# py文件默认同一时间只能运行一次 如果想单独分开运行多次
​
# 半连接池
    设置的最大等待人数  >>>:  节省资源 提高效率

【二】黏包问题

1.原因：因为recv括号内我们不知道即将要接收的数据到底多大
如果每次接收的数据我们都能够精确的知道它的大小 那么肯定不会出现黏包
2.TCP协议的特点
    会将数据量比较小并且时间间隔比较短的数据整合到一起发送
  并且还会受制于recv括号内的数字大小(核心问题!!!)
  流式协议:跟水流一样不间断
3.如果能够精准的知道数据量多大 那么黏包问题就自动解决了!!!

解决黏包问题低配

方向:精准获取数据的大小

  """
  pack可以将任意长度的数字打包成固定长度
  unpack可以将固定长度的数字解包成打包之前数据真实的长度


  思路:
      1.先将真实数据打包成固定长度的包
      2.将固定长度的包先发给对方
      3.对方接收到包之后再解包获取真实数据长度
      4.接收真实数据长度
  """

# struct模块
	import struct

  data1 = 'hello world!'
  print(len(data1))  # 12
  res1 = struct.pack('i', len(data1))  # 第一个参数是格式 写i就可以了
  print(len(res1))  # 4
  ret1 = struct.unpack('i', res1)
  print(ret1)  # (12,)


  data2 = 'hello baby baby baby baby baby baby baby baby'
  print(len(data2))  # 45
  res2 = struct.pack('i', len(data2))
  print(len(res2))  # 4
  ret2 = struct.unpack('i', res2)
  print(ret2)  # (45,)

粘包升级(超大文件怎么传)

总体思路：
#服务端：
1.将文件名称，文件描述，文件大小等数据信息放到字典里
2.先打包字典--由于传输的数据只能是二进制数据，字典转为字符串就转不回来了，所以借助json,将字典转成字符串dumps,随后将他encode变成二进制数据
3.将刚刚的二进制字典进行打包struct.pack('i',len(字典的二进制数据))---》这时候是一个长度元组(4,),由于是i模式所以，肯定是4
4.先把长度传给客户端，也就是传报头
5.接着发送转换好的字典
6.最后用with open一行行传送真正的数据

#客户端：
1.接收报头也就是长度，我们固定了是4字节了，因为很明确的用了i模式服务端---client.recv(4)
2.解包解析出真实的字典长度---struct.unpack('i', 报头)[0]
3.接收字典数据，还原出字典信息，用json.loads
4.不要一次性接受文件的数据，可以选择一次接1024，也是用with open边接受边写入，接一次写一次，直到我没接收的大小到字典中给的文件的大小(这里借助while)

代码：
#服务端
import socket
import os
import struct
import json
server = socket.socket()
server.bind(('127.0.0.1', 8080))
server.listen(5)

conn, addr = server.accept()

data_dict = {
    'file_name':'xxx刺激.txt',
    'file_desc':'封控一个多月了',
    'file_size':os.path.getsize(r'04 struct模块.py')
}
# 1.先打包字典
dict_json_str = json.dumps(data_dict)
dict_bytes = dict_json_str.encode('utf8')
dict_package_header = struct.pack('i', len(dict_bytes))
# 2.发送报头(打包好后的长度)
conn.send(dict_package_header)
# 3.发送字典
conn.send(dict_bytes)
# 4.发送真实数据
with open(r'04 struct模块.py', 'rb') as f:
    for line in f:
        conn.send(line)

#客户端
import socket
import struct
import json

client = socket.socket()
client.connect(('127.0.0.1', 8080))

# 1.先接收固定长度的字典的报头
dict_header_len = client.recv(4)
# 2.解析出字典的真实长度
dict_real_len = struct.unpack('i', dict_header_len)[0]
# 3.接收字典数据
dict_data_bytes = client.recv(dict_real_len)
dict_data = json.loads(dict_data_bytes)
print(dict_data)
# 4.循环接收文件数据 不要一次性接收
recv_size = 0
with open(dict_data.get('file_name'),'wb') as f:
    while recv_size < dict_data.get('file_size'):
        data = client.recv(1024)
        recv_size += len(data)
        f.write(data)

【三】UDP协议

1.与tcp不同点
 a.server = socket.socket(type=socket.SOCK_DGRAM)  # 自己指定UDP协议(默认是TCP协议)
 b.收消息server.recvfrom(1024)
 c.发信息server.sendto(b'hello baby', addr)
 

#服务端
import socket
server = socket.socket(type=socket.SOCK_DGRAM)  # 自己指定UDP协议(默认是TCP协议)
server.bind(('127.0.0.1', 8080))
msg, addr = server.recvfrom(1024)
print('msg>>>:', msg.decode('utf8'))
print('addr>>>:', addr)
server.sendto(b'hello baby', addr)

#客户端
import socket
client = socket.socket(type=socket.SOCK_DGRAM)
server_addr = ('127.0.0.1', 8080)  # 查找通讯录
client.sendto(b'hello server baby', server_addr)
msg, addr = client.recvfrom(1024)
print('msg>>>:', msg.decode('utf8'))
print('addr>>>:', addr)