python socket网络编程

一、简介

socket模块展示了快速创建TCP和UDP服务器及客户端，使用原始套接字等所必须的代码。为了攻击进入或者保持控制目标主机，socket模块是我们必须使用的模块。

开始编写客户端代码

#  === TCP 客户端程序 client.py ===
#导入socket库里面所有的类
from socket import *

#设置ip地址及端口
IP = '127.0.0.1'
SERVER_PORT = 5000
#buffer length 为传输数据的最大长度
BUFLEN = 1024

# 实例化一个socket对象，指明协议
# 参数 AF_INET 表示该socket网络层使用IPv4协议
# 参数 SOCK_STREAM 表示该socket传输层使用TCP协议
dataSocket = socket(AF_INET, SOCK_STREAM)

# 连接服务端socket
dataSocket.connect((IP, SERVER_PORT))

while True:
    # 从终端读入用户输入的字符串
    toSend = input('>>> ')
    if  toSend =='exit':
        break
    # 发送消息，也要编码为 bytes
    dataSocket.send(toSend.encode())

    # 等待接收服务端的消息
    recved = dataSocket.recv(BUFLEN)
    # 如果返回空bytes，表示对方关闭了连接
    if not recved:
        break
    # 打印读取的信息
    print(recved.decode())

dataSocket.close()

在上面代码中，我们对原始套接字做了一定的假设：
首先假设连接总是能成功的，不会出现异常或者报错；
第二个假设是客户端首先发送数据到服务端（相反的情况是，在长连接的情况下，服务端会主动给客户端发送数据并等待客户端的响应。）；
第三个假设是服务端每次都能及时得返回请求（不会出现服务端错误500，503等）。
作为简单示范例子就不涉及上述情况了，后面会说。

然后就是服务端代码

#  === TCP 服务端程序 server.py ===

# 导入socket 库
from socket import *

# 主机地址为空字符串，表示绑定本机所有网络接口ip地址
# 等待客户端来连接
IP = ''
# 端口号
PORT = 5000
# 定义一次从socket缓冲区最多读入512个字节数据
BUFLEN = 512

# 实例化一个socket对象
# 参数 AF_INET 表示该socket网络层使用IP协议
# 参数 SOCK_STREAM 表示该socket传输层使用TCP协议
listenSocket = socket(AF_INET, SOCK_STREAM)

# socket绑定地址和端口
listenSocket.bind((IP, PORT))


# 使socket处于监听状态，等待客户端的连接请求
# 参数 8 表示 最多接受多少个等待连接的客户端
listenSocket.listen(8)
print(f'服务端启动成功，在{PORT}端口等待客户端连接...')

dataSocket, addr = listenSocket.accept()
print('接受一个客户端连接:', addr)

while True:
    # 尝试读取对方发送的消息
    # BUFLEN 指定从接收缓冲里最多读取多少字节
    recved = dataSocket.recv(BUFLEN)

    # 如果返回空bytes，表示对方关闭了连接
    # 退出循环，结束消息收发
    if not recved:
        break
    # 读取的字节数据是bytes类型，需要解码为字符串
    info = recved.decode()
    print(f'收到对方信息： {info}')
    # 发送的数据类型必须是bytes，所以要编码
    dataSocket.send(f'服务端接收到了信息 {info}'.encode())
# 服务端也调用close()关闭socket
dataSocket.close()
listenSocket.close()

注意：启动的时候，服务端一定要先于客户端启动。

在客户端输入传输的内容

服务端显示已收到

创建UDP客户端

创建UDP客户端的时候，只需要做三个改动；
1、删除dataSocket.connect((IP, SERVER_PORT))
因为UDP为无连接状态的传输协议，所以不需要调用connect方法
2、send() 方法改为 sendto()
3、接收信息的时候调用dataSocket.recvfrom(4096) 而不是 dataSocket.recv(BUFLEN)
括号内为数据包的最大长度