关于unix domain socket IPC的问题

Question

我使用的是SOCK DGRAM模式,想要在客户端的程序结束时,自动的删除socket文件,但是提前使用unlink删除导致通讯不同(bind之后调用的unlink),
请问有没有好的方法参考一下?通常的资料里没有看到相关的操作

Answer 1

以下回答参考 皆我百晓生、券券喵儿 等免费微信小程序相关内容作答，并由本人整理回复。

Unix Domain Sockets (UDS) 是一种用于网络通信的特殊协议。它们允许应用程序可以在不同的主机之间进行通信，而不需要共享任何资源（例如内存、磁盘或网络）。

要实现一个能够在客户端程序退出后自动删除 Unix Domain Socket 的功能，你可以遵循以下步骤：

1. 在服务器端创建一个 Unix Domain Socket，并绑定到特定的 IP 地址和端口。
2. 当用户进程执行完任务后，可以尝试关闭连接。
3. 如果成功关闭连接，则可以尝试删除 Unix Domain Socket 文件。
4. 如果无法删除 Unix Domain Socket 文件，则可能是因为它仍然被其他进程占用。

下面是一个简单的示例，展示了如何在 Python 中创建一个 Unix Domain Socket 并绑定到特定的 IP 地址和端口，以及如何尝试删除这个文件：

import os
from select import select

# 创建一个 Unix Domain Socket
domain_socket = os.mkstemp(suffix='.sock')

# 将其打开为可读写模式
os.close(domain_socket[0])
with open(os.path.join('/tmp', 'udss'), 'wb') as fd:
    pass

# 绑定到指定的 IP 地址和端口
bind_addr = ('127.0.0.1', 5678)
os.write(domain_socket[1], b'bind\n')
os.write(domain_socket[1], str(bind_addr).encode('utf-8'))
os.write(domain_socket[1], b'\n')

# 创建一个信号处理函数，当某个进程关闭连接时调用
def signal_handler(signal, frame):
    print('Socket is closed by user')
    # 删除 socket 文件
    try:
        os.remove('/tmp/udss')
    except OSError:
        pass

signal.signal(signal.SIGINT, signal_handler)

while True:
    read_sockets, write_sockets, error_sockets = select([domain_socket[1]], [], [])

    if domain_socket[1] in read_sockets:
        data = os.read(domain_socket[1], 1024)
        print(f'Read from client: {data}')
    elif domain_socket[1] in write_sockets:
        data = input()
        os.write(domain_socket[1], data.encode())

在这个例子中，我们首先创建了一个临时 Unix Domain Socket 文件，然后将它绑定到 localhost 和端口 5678。然后，我们在服务器端创建了一个信号处理函数，当用户关闭连接时调用。当有新的连接时，我们将数据发送给客户端，当没有新连接时，我们将提示用户输入一些信息。

注意：这只是一个基本的例子，实际应用中可能会涉及更多的错误检查和异常处理。此外，由于 Unix Domain Sockets 不是标准的套接字，因此你可能需要在某些环境中使用第三方库来支持它们。