Python网络编程Socket库

套接字（socket）

• 网络上的两个程序通过一个双向的通信连接实现数据的交换，这个连接的一端称为一个socket。
• 套接字是socket的通常叫法，用于描述IP地址和端口，是一个通信链的句柄，可以用来实现不同虚拟机或不同计算机之间的通信。
• Python中Socket库为操作系统的socket实现提供了一个Python接口。

1.socket协议类型

socket库中整合了多种协议类型

2.socket函数

• 服务器端Socket函数：socket库中的服务器端函数仅供服务器使用

• 客户端Socket函数：socket库中的客户端函数仅供客户端使用

• 公共Socket函数：socket库中的公共函数即可在服务器端使用也可在客户端使用，为通用函数

使用Socket进行TCP编程

TCP连接由客户端发起，服务器对连接进行响应

• 建立一个服务器，服务器进程需要绑定一个端口并监听来自其他客户端的连接。
• 若有客户端发起连接请求，服务器就与该客户端建立Socket连接，随后的通信就通过此Socket连接进行。
• 服务器依赖服务器地址，服务器端口，客户端地址，客户端端口这4项来唯一确定一个Socket连接。

一、服务器端TCP连接

建立服务器端的TCP连接，具体步骤如下：

1.在Python中创建一个基于IPv4和TCP协议的Socket

# 导入socket库及依赖库
import socket
import threading
import time
# 建立TCP连接的Socket
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

2.绑定监听的地址和端口，地址使用本机地址“127.0.0.1”或“local host”，使用大于1024的端口

# 绑定地址及监听端口
s.bind(('127.0.0.1', 6666))

3.调用listen方法开始监听端口，传入的参数指定等待连接的最大数量，设定为5（可以随意设置，建议不要太大）

# 调用listen方法监听端口
s.listen(5)
print('Wait for connection...')

4.创建一个tcp函数，该函数在连接建立后，服务器端首先发出一条表示连接成功的消息，然后等待客户端数据，再加上欢迎信息发送给客户端。若客户端发送exit字符串，则直接关闭连接

# 服务器端应答函数：
def tcp(sock, addr):
    print('Accept new connection from %s:%s...' % addr)
    sock.send(b'Success!')
    while True:
        data = sock.recv(1024)
        time.sleep(1)
        if not data or data.decode('utf-8') == 'exit':
            break
        sock.send(('Welcom! %s!' % data.decode('utf-8')).encode('utf-8'))
    sock.close()
print('Connection from %s:%s closed.' % addr)

5.通过一个循环接受来自客户端的连接，使用accept函数等待并返回一个客户端的连接，每个连接都分配一个新线程来处理

# 循环处理客户端连接
while True:
    # 接受来自客户端的新连接:
    sock, addr = s.accept()
# 创建新线程来处理TCP连接:
    t = threading.Thread(target=tcp args=(sock, addr))
    t.start()

二、客户端TCP连接

在服务器端TCP连接建立后，建立客户端TCP连接进行测试，具体步骤如下：

1.与服务器端的协议保持一致，也建立一个基于IPv4和TCP协议的Socket

# 导入socket库
import socket
# 建立TCP连接
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

2.与服务器端建立连接，连接的地址与端口需与服务器端保持一致

# 与服务器建立连接
s.connect(('127.0.0.1', 6666))

3.使用recv函数接受服务器提示信息，之后再使用send函数发送数据至服务器，可看到服务器返回的结果

# 接受服务器的连接成功提示信息
print(s.recv(1024).decode('utf-8'))
# 发送数据并接受服务器返回结果
for data in [b'Tom', b'Jerry', b'Spike']:
		s.send(data)
		print(s.recv(1024).decode('utf-8'))
# 发送退出信息断开连接
s.send(b'exit')
s.close()

使用Socket进行UDP编程

TCP建立的连接可靠，通信双方以流的形式互相传送数据。相对TCP协议，UDP则是面向无连接的协议

• 使用UDP协议时，无需建立连接的过程，仅需知道对方的IP地址及端口号，便可直接发送数据包，但无法保证能顺利传达到
• 虽然用UDP传输数据不可靠，但其传输速度比TCP快，对于不要求可靠到达的数据，就可以使用UDP协议
• UDP传输通常应用在通讯实时性要求更高于可靠性场景，例如网络游戏

使用Socket进行UDP编程

一、服务器端UDP连接

UDP连接与TCP连接类似，也分为服务器端和客户端，不同的是UDP连接无需调用listen方法，直接接受来自任何客户端的数据。

1.建立UDP连接，服务器端同样需要绑定地址与端口

# 导入socket库
import socket
# 建立UDP连接
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
# 绑定地址与端口
s.bind(('127.0.0.1', 6666))
print('set UDP on 6666...')

2.使用recvfrom方法返回数据及客户端的地址与端口，当服务器收到数据后，直接调用sendto把数据用UDP发给客户端

while True:
		# 接收来自任意客户端的数据:
		data, addr = s.recvfrom(1024)
		# 打印接受信息并回传欢迎信息
		print('Received from %s:%s.' % addr)
		s.sendto(b'Welcom! %s!' % data, addr)

二、客户端UDP连接

1.客户端使用UDP连接时同样需要先创建socket

# 导入socket库
import socket
# 建立UDP连接
s = socket.socket(socket.AF_INET, socket. SOCK_DGRAM)

2.之后无需使用connect方法，直接用sendto方法发送数据至服务器建立UDP连接，服务器端同样需要绑定地址与端口

# 发送数据并接受服务器回传数据
for data in [b'Tom', b'Jerry', b'Spike']:
		s.sendto(data, ('127.0.0.1', 99996666))
		print(s.recv(1024).decode('utf-8'))
s.close()

3.UDP连接与TCP连接可同时使用同一端口互不冲突，两者使用的端口是独立绑定的

小结

本文主要介绍了Python中的底层Socket库，及运用socket库建立TCP和UDP连接。

1.Socket库提供多种协议类型和函数，可用于建立TCP和UDP连接。

2.HTTP协议基于TCP协议进行客户端与服务器间的通讯，由客户端发起请求，服务器进行应答。