python-socket编程（二）模块基础

一、socket模块

1.1 创建套接字对象

socket.socket([family[, type]])

• family: 套接字家族可以使用AF_UNIX或者AF_INET
• type: 套接字类型可以根据是面向连接的还是非连接分为SOCK_STREAM或SOCK_DGRAM

1.2 重用套接字

解决有时候地址被占用的问题

s.setsockopt(SQL_SOCKET,SO_REUSEADDR,1)

1.3 套接字对象内建方法

1.3.1 服务端

• s.bind()：绑定地址（host,port）到套接字， 在AF_INET下,以元组（host,port）的形式表示地址。
• s.listen(backlog=n)：开始TCP监听。backlog指定在拒绝连接之前，操作系统可以挂起的最大连接数量。该值至少为1，大部分应用程序设为5就可以了
• s.accept()：被动接受TCP客户端连接,(阻塞式)等待连接的到来

1.3.2 客户端

• s.connect()：主动初始化TCP服务器连接，。一般address的格式为元组（hostname,port），如果连接出错，返回socket.error错误
• s.connect_ex()：connect()函数的扩展版本,出错时返回出错码,而不是抛出异常

公共使用

• s.recv()：接收TCP数据，数据以字符串形式返回，bufsize指定要接收的最大数据量。flag提供有关消息的其他信息，通常可以忽略
• s.send()：发送TCP数据，将string中的数据发送到连接的套接字。返回值是要发送的字节数量，该数量可能小于string的字节大小
• s.sendall()：完整发送TCP数据，完整发送TCP数据。将string中的数据发送到连接的套接字，但在返回之前会尝试发送所有数据。成功返回None，失败则抛出异常
• s.recvfrom()：接收UDP数据，与recv()类似，但返回值是（data,address）。其中data是包含接收数据的字符串，address是发送数据的套接字地址
• s.sendto()：发送UDP数据，将数据发送到套接字，address是形式为（ipaddr，port）的元组，指定远程地址。返回值是发送的字节数
• s.close()：关闭套接字
• s.getpeername()：返回连接套接字的远程地址。返回值通常是元组（ipaddr,port）
• s.getsockname()：返回套接字自己的地址。通常是一个元组(ipaddr,port)
• s.setsockopt(level,optname,value)：设置给定套接字选项的值。
• s.getsockopt(level,optname[.buflen])：返回套接字选项的值。
• s.settimeout(timeout)：设置套接字操作的超时期，timeout是一个浮点数，单位是秒。值为None表示没有超时期。一般，超时期应该在刚创建套接字时设置，因为它们可能用于连接的操作（如connect()）
• s.gettimeout()：返回当前超时期的值，单位是秒，如果没有设置超时期，则返回None
• s.fileno()：返回套接字的文件描述符
• s.setblocking(flag)：如果flag为0，则将套接字设为非阻塞模式，否则将套接字设为阻塞模式（默认值）。非阻塞模式下，如果调用recv()没有发现任何数据，或send()调用无法立即发送数据，那么将引起socket.error异常
• s.makefile()：创建一个与该套接字相关连的文件

二、tcp简单实例

一对一模型

• 服务端

import socket

ip_port=('127.0.0.1',8001)
backlog=5
buffersize=1024

tcp_server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
tcp_server.bind(ip_port)
tcp_server.listen(backlog)

conn,addr = tcp_server.accept()
print(conn)
#<socket.socket fd=4, family=AddressFamily.AF_INET, type=SocketKind.SOCK_STREAM, proto=0, laddr=('127.0.0.1', 8001), raddr=('127.0.0.1', 57902)>
print(addr)
('127.0.0.1', 57902)
msg = conn.recv(buffersize).decode('utf-8')
print("客户端发来的消息是：%s"%msg)
conn.close()
tcp_server.close()

• 客户端

import socket

ip_port = ('127.0.0.1',8001)

tcp_client = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
tcp_client.connect(('127.0.0.1',8001))

msg = "你好，我是客户端"
tcp_client.send(msg.encode('utf-8'))

tcp_client.clsoe()

tcp连接的时候要简要说明，无论是客户端还是服务端，当发送消息的时候都会通过socket发送至对方主机的内存中的缓存区域（内核态内存），也就是buffer区域。例子中buffer设置为1024bytes，若发送的量大于这个数值，那当程序收取的时候最后还是只能收取1024bytes。

一对多模型

• 服务端

import socket

ip_port=('127.0.0.1',8001)
backlog=5
buffersize=1024


tcp_server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
tcp_server.bind(ip_port)
tcp_server.listen(backlog)
while True:
    conn,addr = tcp_server.accept()
    while True:
        msg = conn.recv(buffersize).decode('utf-8')
        if not msg:
            break
        # 当客户端断开连接的时候会一直循环收到空信息，因此使用此条退出循环
        print("客户端发来的消息是：%s"%msg
    conn.close()
    #当一个客户端断开连接，服务端也断开此次连接，循环进行下一次accept等待连接
tcp_server.close()

• 客户端

import socket

ip_port = ('127.0.0.1',8001)

tcp_client = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
tcp_client.connect(('127.0.0.1',8001))
while True:
    msg = input('-->:')
    if not msg:continue
    # 如果输入的信息为空则跳过此次发送
    tcp_client.send(msg.encode('utf-8'))
tcp_client.close()

• 我们用客户端的代码同时开启两个客户端
• 客户端1发送两条信息（1）hello （2）i am 1
• 客户端2发送两条信息（1）hello （2）i am 2
• 服务端会收到两条消息 “hello”和“ i am 1”
• 断开客户端1
• 服务端会收到一条消息 “helloi am 2”，这是因为服务端首先会将发客户端2的连接在backlog连接池挂起，客户端发送的消息会储存在连接池，服务器段一次收取一条，而tcp是数据流，所以客户端2发送的两条信息会粘连在一起，此为粘包现象。这个现象我们会在后续深入讨论。

三、UDP简单实例

• 服务端

import socket

ip_port=('127.0.0.1',8001)
backlog=5
buffersize=1024

udp_server = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
udp_server.bind(ip_port)
while True:
    data,addr = udp_server.recvfrom(buffersize)
    print("udp客户端发来的消息是:%s"%data.decode('utf-8'))

• 客户端

import socket

ip_port = ('127.0.0.1',8001)

udp_client = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)

while True:
    data = input("客户请输入数据:")
    udp_client.sendto(data.encode('utf-8'),ip_port)

使用tcp传输数据的时候，发送空数据会导致进程堵塞，而udp不会产生的这个问题