Python3 UNIX domain sockets使用代碼實現

本文轉載自查看原文 2020-02-03 14:37 1281 Linux

一、說明

背景一：大學的時候學網絡編程，經常看到說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()