网络编程-套接字(socket)

一、Socket（套接字）

★注意点：

① 127.0.0.1本机地址回环：只能识别自己，其他人无法访问

② send与recv对应，不要出现两边是相同的情况，recv是跟内存要数据，无需考虑

③ tcp特点是会将数据量比较小的并且时间间隔比较短的数据，一次性打包发送给对方

import socket

server = socket.socket()  # 买手机 不传参数默认用的就是TCP协议
server.bind(('127.0.0.1',8080))  # bind((host,port))  插电话卡  绑定ip和端口
server.listen(5)  # 开机    半连接池

conn, addr = server.accept()  # 接听电话  等着别人给你打电话    
data = conn.recv(1024)  # 听别人说话 接收1024个字节数据          
conn.send(b'hello baby~')  # 给别人回话

conn.close()  # 挂电话
server.close()  # 关机

import socket

client = socket.socket()  # 拿电话
client.connect(('127.0.0.1',8080))  # 拨号   写的是对方的ip和port

client.send(b'hello world!')  # 对别人说话
data = client.recv(1024)  # 听别人说话
print(data)

client.close()  # 挂电话

二、TCP粘包问题

　　我们基于TCP协议写一个远程命令操控

from socket import *
import subprocess

1024

tcp_socket_server=socket(AF_INET,SOCK_STREAM)
tcp_socket_server.setsockopt(SOL_SOCKET,SO_REUSEADDR,1)
tcp_socket_server.bind(('127.0.0.1',8888))
tcp_socket_server.listen(5)

while True:
    conn,addr=tcp_socket_server.accept()

    while True:
        cmd=conn.recv(1024)
        if len(cmd) == 0:break

        res=subprocess.Popen(cmd.decode('utf-8'),shell=True,
                         stdout=subprocess.PIPE,
                         stdin=subprocess.PIPE,
                         stderr=subprocess.PIPE)

        stderr=res.stderr.read()
        stdout=res.stdout.read()
        conn.send(stderr)
        conn.send(stdout)

import socket

s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
res=s.connect_ex(('127.0.0.1',8888))

while True:
    msg=input('>>: ').strip()
    if len(msg) == 0:continue
    if msg == 'quit':break

    s.send(msg.encode('utf-8'))
    act_res=s.recv(1024)

    print(act_res.decode('utf-8'),end='')

　　同时执行多条命令之后，得到的结果可能只有一部分，在执行其他命令的时候又接收到之前执行的另外一部分结果，这种现象就是粘包。

三、解决粘包问题

       问题的根源在于，接收端不知道发送端将要传送的字节流的长度，所以解决粘包的方法就是围绕，如何让发送端在发送数据前，把自己将要发送的字节流总大小让接收端知晓，然后接收端来一个while循环接收完所有数据。

import json
import socket
import struct
import subprocess

server = socket.socket()
server.bind(('127.0.0.1',8080))
server.listen(5)

while True:
    conn,addr = server.accept()
    while True:
        try:
            msg = conn.recv(1024).decode('utf-8')
            if not msg : break
            obj = subprocess.Popen(msg,shell = True,stdout = subprocess.PIPE,stderr = subprocess.PIPE)
            res = obj.stdout.read() + obj.stderr.read()
            dict = {'name':'Spenser','file_size':len(res),'info':'to be a better man'}
            json_dict = json.dumps(dict)
            #1.制作字典报头
            header = struct.pack('i',len(json_dict))
            #2.传送字典报头
            conn.send(header)
            #3.传送字典
            conn.send(json_dict.encode('utf-8'))
            #4.传送真实数据
            conn.send(res)
        except ConnectionResetError:
            break
    conn.close()

import socket
import struct
import json

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

while True:
    msg = input('>>>:').encode('utf-8')
    if len(msg) == 0:continue
    client.send(msg)
    # 1.先接受字典报头
    header_dict = client.recv(4)
    # 2.解析报头 获取字典的长度
    dict_size = struct.unpack('i',header_dict)[0]  # 解包的时候一定要加上索引0
    # 3.接收字典数据
    dict_bytes = client.recv(dict_size)
    dict_json = json.loads(dict_bytes.decode('utf-8'))
    # 4.从字典中获取信息
    print(dict_json)
    recv_size = 0
    real_data = b''
    while recv_size < dict_json.get('file_size'):  # real_size = 102400
        data = client.recv(1024)
        real_data += data
        recv_size += len(data)
    print(real_data.decode('gbk'))