Unix套接字凭证在Python中传递

时间:2011-11-02 15:14:40

标签: python linux sockets credentials

如何在Python中完成Unix套接字凭据传递?

1 个答案:

答案 0 :(得分:19)

关于这一主题的互联网搜索得出的结果令人惊讶。我想我会在这里发布问题并回答其他对这个主题感兴趣的人。

以下客户端和服务器应用程序演示了如何使用标准python解释器在Linux上完成此任务。不需要扩展,但由于使用了嵌入式常量,代码是特定于Linux的。

服务器:

#!/usr/bin/env python

import struct
from socket import socket, AF_UNIX, SOCK_STREAM, SOL_SOCKET

SO_PEERCRED = 17 # Pulled from /usr/include/asm-generic/socket.h

s = socket(AF_UNIX, SOCK_STREAM)

s.bind('/tmp/pass_cred')
s.listen(1)

conn, addr = s.accept()

creds = conn.getsockopt(SOL_SOCKET, SO_PEERCRED, struct.calcsize('3i'))

pid, uid, gid = struct.unpack('3i',creds)

print 'pid: %d, uid: %d, gid %d' % (pid, uid, gid)

客户端:

#!/usr/bin/env python

from socket import socket, AF_UNIX, SOCK_STREAM, SOL_SOCKET

SO_PASSCRED = 16 # Pulled from /usr/include/asm-generic/socket.h

s = socket(AF_UNIX, SOCK_STREAM)

s.setsockopt(SOL_SOCKET, SO_PASSCRED, 1)

s.connect('/tmp/pass_cred')

s.close()

不幸的是,SO_PEERCRED和SO_PASSCRED常量不会被python的套接字模块导出,因此必须手动输入它们。虽然这些值不太可能改变,但这是可能的。任何使用此方法的应用程序都应考虑这一点。