python socket 编程

一、socket是什么？

Socket又称"套接字"，应用程序通常通过"套接字"向网络发出请求或者应答网络请求，使主机间或者一台计算机上的进程间可以通讯。socket通过各种函数实现数据封装，建立目标主机端口连接，在客户端的socket进行数据接收，将数据交付给目标主机。

二、socket的基本参数和函数介绍

1、Socket函数使用的格式为：socket(family,type[,protocol])

（1）参数一：family 指定应用程序使用的通信协议的协议族，对于TCP/IP协议族，该参数为AF_INET，以下为该参数的一些常用选项

Family参数	描述
socket.AF_UNIX	只能够用于单一的Unix系统进程间通信
socket.AF_INET	服务器之间网络通信
socket.AF_INET6	IPv6

（2）参数二：type 是要创建套接字的类型，以下为该参数的一些常用选项

Type参数	描述
socket.SOCK_STREAM	流式socket , 当使用TCP时选择此参数
socket.SOCK_DGRAM	数据报式socket ,当使用UDP时选择此参数
socket.SOCK_RAW	原始套接字，普通的套接字无法处理ICMP、IGMP等网络报文，而SOCK_RAW可以；其次，SOCK_RAW也可以处理特殊的IPv4报文；此外，利用原始套接字，可以通过IP_HDRINCL套接字选项由用户构造IP头。

（3）参数三：protocol 指明所要接收的协议类型，通常为0或者不填。以下为该参数的一些常用选项

Type参数	描述
socket.IPPROTO_RAW	相当于protocol=255，此时socket只能用来发送IP包，而不能接收任何的数据。发送的数据需要自己填充IP包头，并且自己计算校验和
socket.IPPROTO_IP	相当于protocol=0，此时用于接收任何的IP数据包。其中的校验和和协议分析由程序自己完成。

使用例子：   

创建TCP Socket：s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)

创建UDP Socket：s=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)

2、socket相关的函数

（1）服务端socket函数

　　　　s.bind(address)：将套接字绑定到地址，在AF_INET下，以tuple(host,prot)的方式传入，如s.bind((host,port))。

　　　　s.listen(backlog)：开始监听TCP传入连接。backlog指定在拒绝连接之前，操作系统可以挂起的最大连接数量。该值最少为1。

　　　　s.accept()：接受TCP连接并返回(conn,address)，其中conn是新的套接字对象，可以用来接收和发送数据，address是连接客户端的地址。(sock,addr = s.accept)。而socket.socket()的这个套接字只是用于监听。

　　（2）客户端socket函数

　　　　s.connect(address)：链接到address处的套接字，一般address的格式为tuple(host,port),如果链接出错，则返回socket.error错误。

　　　　s.connect_ex(address)：功能与s.connect(address)相同，但成功返回0，失败返回errno的值。

　　（3）公共Socket函数

　　　　s.send(string[,flag])：发送TCP数据，将string中的数据发送到连接到套接字，返回值是要发送到字节数量，该数量可能小于string的字节大小。

　　　　s.sendall(string[,flag])：完整发送TCP数据，将string中的数据发送到链接到套接字，但在返回之前尝试发送所有的数据。成功返回None，失败则抛出异常。

　　　　s.recv(bufsize[,flag])：接受TCP套接字的数据，数据以字符串的形式返回，bufsize指定要接收的最大数据量，flag提供有关消息的其它消息，通常可以忽略。

　　　　s.sendto(string[,flag],address)：发送UDP数据，将数据发送到套接字，address形式为tuple(hostadder,port),指定远程地址发送，返回值是发送到字节。

　　　　s.recvfrom(bufsize[,flag])：接受UDP套接字的数据，与recv()类似，但返回值是tuple(data,address)。其中data是包涵接受数据的字符串，address是发送数据的套接字地址。

　　　　s.close()：关闭套接字。

　　　　s.getpeername()：返回套接字的远程地址，返回值通常是一个tuple(hostaddr,port)。

　　　　s.getsockname()：返回套接字自己的地址，返回值通常是一个tuple(hostaddr,port)。

　　　　s.getsockopt(level,optname[,buflen]])：返回套接字选项的值。

　　　　s.gettimeout()：返回当前超时值，单位是秒，如果没有设置超时值则返回None。

　　　　s.setsockopt(level, option,value)：设置给定套接字选项的值。

　　　　s.settimeout(timeout)：设置套接字操作的超时时间，timeout是一个浮点数，单位是秒。值为None则表示永远不会超时。一般超时期应在刚创建套接字时设置。

　　　　s.setblocking(flag)：如果flag为0，则将套接字设置为非阻塞模式，否则将套接字设置为阻塞模式（默认值）。非阻塞模式下，如果调用recv()没有发现任何数据，或send()调用无法立即发送数据，那么将引发socket.error异常。

　　　　s.makefile()：创建一个与该套接字相关的文件。

　　　　s.fileno()：返回套接字的文件描述。

三、python socket实例

利用python写了一个tcp socket的通信实例，程序包括一个服务端和一个客户端，两者之间可以相互收发数据。

服务端代码：

import socket


class Server1:
    def __init__(self, host, port, lst_num):
        self.host = host
        self.port = port
        self.lst_num = lst_num

        # 参数表示选择TCP/IP协议族，使用TCP协议创建套接字
        self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        # 操作系统会在服务器socket被关闭或服务器进程终止后马上释放该服务器的端口
        self.server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        # 绑定ip和端口
        self.server.bind((self.host, self.port))

        self.server.listen(self.lst_num)  # 最多连接几个客户端
        print("---------Server start!---------")
        # 等待客户端连接连接，如果客户端连接，就会获取到客户端套接字连接和客户端地址
        self.conn, self.addr = self.server.accept()
        print("Success connect from ", self.addr)
        self.conn.send("Welcome connect!".encode())

    def send_message(self):
        message = input("server send:")
        self.conn.send(message.encode())

    def recv_message(self):
        data = self.conn.recv(1024)
        print("from {0}:".format(self.addr), data.decode('utf-8'))


if __name__ == '__main__':
    while True:  # 循环开启服务端服务，等待多个客户端的连接
        s = Server1('127.0.0.1', 62121, 5)

        '''循环对一个客户端接收消息;
        若是一个客户端异常断开，用try except捕获异常，
        防止服务端程序终止，不影响继续等待其他客户端连接。'''
        while True:
            try:
                s.recv_message()
            except Exception:
                break

        s.conn.close()  # 关闭一个客户端的连接
        s.server.close()  # 关闭套接字

客户端代码：

import socket


class Client1:
    def __init__(self, dst_host, dst_port, buf_size):
        self.dst_host = dst_host
        self.dst_port = dst_port
        self.buf_size = buf_size

        # 参数表示选择TCP/IP协议族，使用TCP协议创建套接字
        self.client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        # 操作系统会在客户端socket被关闭或客户端进程终止后马上释放该客户端的端口
        self.client.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        # 连接目标服务器
        self.client.connect((self.dst_host, self.dst_port))

    def send_message(self):
        message = input("client send:")
        self.client.send(message.encode())

    def recv_message(self):
        data = self.client.recv(self.buf_size)
        print("from ('{0}', {1}):".format(self.dst_host, self.dst_port), data.decode())


if __name__ == '__main__':
    c = Client1('127.0.0.1', 62121, 1024)
    c.recv_message()
    while True:  # 发送信息到服务端
        try:
            c.send_message()
        except Exception as info:
            print(info)
            break
    c.client.close()