Socket 通讯原理

什么是 Socket？

Socket 的中文翻译过来就是“套接字”。套接字是什么，咱们先来看看它的英文含义：插座。

Socket 就像一个电话插座，负责连通两端的电话，进行点对点通讯，让电话能够进行通讯，端口就像插座上的孔，端口不能同时被其余进程占用。而咱们创建链接就像把插头插在这个插座上，建立一个 Socket 实例开始监听后，这个电话插座就时刻监听着消息的传入，谁拨通我这个“IP 地址和端口”，我就接通谁。

实际上，Socket 是在应用层和传输层之间的一个抽象层，它把 TCP/IP 层复杂的操做抽象为几个简单的接口，供应用层调用实现进程在网络中的通讯。Socket 起源于 UNIX，在 UNIX 一切皆文件的思想下，进程间通讯就被冠名为文件描述符（file descriptor），Socket 是一种“打开—读/写—关闭”模式的实现，服务器和客户端各自维护一个“文件”，在创建链接打开后，能够向文件写入内容供对方读取或者读取对方内容，通信结束时关闭文件。

另外咱们常常说到的Socket 所在位置以下图：

Socket 通讯过程

Socket 保证了不一样计算机之间的通讯，也就是网络通讯。对于网站，通讯模型是服务器与客户端之间的通讯。两端都创建了一个 Socket 对象，而后经过 Socket 对象对数据进行传输。一般服务器处于一个无限循环，等待客户端的链接。

一图胜千言，下面是面向链接的 TCP 时序图：

客户端过程：

客户端的过程比较简单，建立 Socket，链接服务器，将 Socket 与远程主机链接（注意：只有 TCP 才有“链接”的概念，一些 Socket 好比 UDP、ICMP 和 ARP 没有“链接”的概念），发送数据，读取响应数据，直到数据交换完毕，关闭链接，结束 TCP 对话。

import socket
import sys

if __name__ == '__main__':
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # 建立 Socket 链接
    sock.connect(('127.0.0.1', 8001))  # 链接服务器
    while True:
        data = input('Please input data:')
        if not data:
            break
        try:
            sock.sendall(data)
        except socket.error as e:
            print('Send Failed...', e)
            sys.exit(0)
        print('Send Successfully')

        res = sock.recv(4096)  # 获取服务器返回的数据，还能够用 recvfrom()、recv_into() 等
        print(res)
    sock.close()

sock.sendall(data)

这里也可用 send() 方法：不一样在于 sendall() 在返回前会尝试发送全部数据，而且成功时返回 None，而 send() 则返回发送的字节数量，失败时都抛出异常。

服务端过程：

咱再来聊聊服务端的过程，服务端先初始化 Socket，创建流式套接字，与本机地址及端口进行绑定，而后通知 TCP，准备好接收链接，调用 accept() 阻塞，等待来自客户端的链接。若是这时客户端与服务器创建了链接，客户端发送数据请求，服务器接收请求并处理请求，而后把响应数据发送给客户端，客户端读取数据，直到数据交换完毕。最后关闭链接，交互结束。

import socket
import sys

if __name__ == '__main__':
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # 建立 Socket 链接（TCP）
    print('Socket Created')

    try:
        sock.bind(('127.0.0.1', 8001))  # 配置 Socket，绑定 IP 地址和端口号
    except socket.error as e:
        print('Bind Failed...', e)
        sys.exit(0)

    sock.listen(5)  # 设置最大容许链接数，各链接和 Server 的通讯遵循 FIFO 原则

    while True:  # 循环轮询 Socket 状态，等待访问
        conn, addr = sock.accept()
        try:
            conn.settimeout(10)  # 若是请求超过 10 秒没有完成，就终止操做

            # 若是要同时处理多个链接，则下面的语句块应该用多线程来处理
            while True:  # 得到一个链接，而后开始循环处理这个链接发送的信息
                data = conn.recv(1024)
                print('Get value ' + data, end='\n\n')
                if not data:
                    print('Exit Server', end='\n\n')
                    break
                conn.sendall('OK')  # 返回数据
        except socket.timeout:  # 创建链接后，该链接在设定的时间内没有数据发来，就会引起超时
            print('Time out')

        conn.close()  # 当一个链接监听循环退出后，链接能够关掉
    sock.close()

conn, addr = sock.accept()

调用 accept() 时，Socket 会进入waiting状态。客户端请求链接时，方法创建链接并返回服务器。accept() 返回一个含有两个元素的元组 (conn, addr)。第一个元素 conn 是新的 Socket 对象，服务器必须经过它与客户端通讯；第二个元素 addr 是客户端的 IP 地址及端口。

data = conn.recv(1024)

接下来是处理阶段，服务器和客户端经过 send() 和 recv() 通讯（传输数据）。
服务器调用 send()，并采用字符串形式向客户端发送信息，send() 返回已发送的字符个数。
服务器调用 recv() 从客户端接收信息。调用 recv() 时，服务器必须指定一个整数，它对应于可经过本次方法调用来接收的最大数据量。recv() 在接收数据时会进入blocked状态，最后返回一个字符串，用它表示收到的数据。若是发送的数据量超过了 recv() 所容许的，数据会被截短。多余的数据将缓冲于接收端，之后调用 recv() 时，会继续读剩余的字节，若是有多余的数据会从缓冲区删除（以及自上次调用 recv() 以来，客户端可能发送的其它任何数据）。传输结束，服务器调用 Socket 的 close() 关闭链接。

从 TCP 链接的视角看 Socket 过程：

TCP 三次握手的 Socket 过程：

1. 服务器调用 socket()、bind()、listen() 完成初始化后，调用 accept() 阻塞等待；
2. 客户端 Socket 对象调用 connect() 向服务器发送了一个 SYN 并阻塞；
3. 服务器完成了第一次握手，即发送 SYN 和 ACK 应答；
4. 客户端收到服务端发送的应答以后，从 connect() 返回，再发送一个 ACK 给服务器；
5. 服务器 Socket 对象接收客户端第三次握手 ACK 确认，此时服务端从 accept() 返回，创建链接。

接下来就是两个端的链接对象互相收发数据。

TCP 四次挥手的 Socket 过程：

1. 某个应用进程调用 close() 主动关闭，发送一个 FIN；
2. 另外一端接收到 FIN 后被动执行关闭，并发送 ACK 确认；
3. 以后被动执行关闭的应用进程调用 close() 关闭 Socket，并也发送一个 FIN；
4. 接收到这个 FIN 的一端向另外一端 ACK 确认。

总结：

上面的代码简单地演示了 Socket 的基本函数使用，其实无论有多复杂的网络程序，这些基本函数都会用到。上面的服务端代码只有处理完一个客户端请求才会去处理下一个客户端的请求，这样的服务器处理能力很弱，而实际中服务器都须要有并发处理能力，为了达到并发处理，服务器就须要 fork 一个新的进程或者线程去处理请求。