Python3 UNIX domain sockets使用代码实现

一、说明

背景一：大学的时候学网络编程，经常看到说socket有AF_UNIX、AF_INET和AF_INET6三个协议族，AF_UNIX一般不用、AF_INET是IPv4的、AF_INET6是IPv6的。基于这种说教一直以来写网络编程，上来协议族就写AF_INET，AF_UNIX是什么怎么用一直没深究。

背景二：毕业后多接触Linux，也经常用netstat看端口监听情况，在较长一段时间内也不懂“netstat -ltnp”、"netstat -unp"，上来就是“netstat -anp”。这样导致的问题就是在最后总是有“Active UNIX domain sockets”一项，而且经常占很长的一个版面，要往前拉很久才能看到前面的tcp和udp。

背景三：今天早上看《Linux内核源代码情景分析》的进程间通信章节，发现AF_UNIX和UNIX domain sockets竟是一个东西，同时解决了两个困惑，真想击节称叹。

二、使用代码实现

2.1 服务端示例代码

import socket

class SocketServer:
    def __init__(self):
        # 常规tcp监听写法
        # server_address = (‘127.0.0.1‘, 9999)
        # socket_family = socket.AF_INET
        # socket_type = socket.SOCK_STREAM

        # unix domain sockets 监听写法
        server_address = ‘/tmp/uds_socket‘
        socket_family = socket.AF_UNIX
        socket_type = socket.SOCK_STREAM

        # 其他代码完全一样
        self.sock = socket.socket(socket_family, socket_type)
        self.sock.bind(server_address)
        self.sock.listen(1)
        print(f"listening on ‘{server_address}‘.")
        pass

    def wait_and_deal_client_connect(self):
        while True:
            connection, client_address = self.sock.accept()
            data = connection.recv(1024)
            print(f"recv data from client ‘{client_address}‘: {data.decode()}")
            connection.sendall("hello client".encode())

    def __del__(self):
        self.sock.close()

if __name__ == "__main__":
    socket_server_obj = SocketServer()
    socket_server_obj.wait_and_deal_client_connect()

2.2 客户端示例代码

import socket

class SocketClient:
    def __init__(self):
        pass

    def connect_to_server(self):
        # 常规tcp连接写法
        # server_address = (‘127.0.0.1‘, 9999)
        # socket_family = socket.AF_INET
        # socket_type = socket.SOCK_STREAM

        # unix domain sockets 连接写法
        server_address = ‘/tmp/uds_socket‘
        socket_family = socket.AF_UNIX
        socket_type = socket.SOCK_STREAM

        # 其他代码完全一样
        sock = socket.socket(socket_family, socket_type)
        sock.connect(server_address)
        sock.sendall("hello server".encode())
        data = sock.recv(1024)
        print(f"recv data from server ‘{server_address}‘: {data.decode()}")
        sock.close()

if __name__ == "__main__":
    socket_client_obj = SocketClient()
    socket_client_obj.connect_to_server()