在Python 3中查找给定套接字和inode的进程ID

时间:2013-02-02 22:30:56

标签: linux sockets python-3.x python-3.2 procfs

/ proc / net / tcp给我一个套接字的本地地址,端口和inode号(例如0.0.0.0:5432和9289)。

鉴于上述信息,我想找到特定过程的PID。

可以打开/ proc中的每个编号文件夹,然后使用shell命令检查符号链接以查找套接字/ inode编号,例如“$ sudo ls -l / proc / * / fd / 2&gt; / dev / null | grep插座”。但是,这似乎比必要的计算成本更高,因为任何给定系统上的<5%的进程都有开放的TCP套接字。

找到打开给定套接字的PID的最有效方法是什么?我更喜欢使用标准库,而我目前正在使用Python 3.2.3进行开发。

修改:从问题中删除了代码示例,因为它们现在包含在下面的答案中。

2 个答案:

答案 0 :(得分:5)

以下代码实现了原始目标:

def find_pid(inode):

    # get a list of all files and directories in /proc
    procFiles = os.listdir("/proc/")

    # remove the pid of the current python process
    procFiles.remove(str(os.getpid()))

    # set up a list object to store valid pids
    pids = []

    for f in procFiles:
        try:
            # convert the filename to an integer and back, saving the result to a list
            integer = int(f)
            pids.append(str(integer))
        except ValueError:
            # if the filename doesn't convert to an integer, it's not a pid, and we don't care about it
            pass

    for pid in pids:
        # check the fd directory for socket information
        fds = os.listdir("/proc/%s/fd/" % pid)
        for fd in fds:
            # save the pid for sockets matching our inode
            if ('socket:[%d]' % inode) == os.readlink("/proc/%s/fd/%s" % (pid, fd)):
                return pid

答案 1 :(得分:2)

我不知道如何在python中执行此操作,但您可以使用lsof(1)

lsof -i | awk -v sock=158384387 '$6 == sock{print $2}'

158384387是套接字的inode。然后使用subprocess.Popen从python中调用它。

如果要查看其他用户打开的套接字,则必须使用sudo(8)。