使用 os 或 subprocess 模块返回字符串或整数

Question

我现在正在尝试制作一个包含多个网络扫描工具的工具。使用 os 模块我将代码发送到控制台以查找我的 ip + CIDR 并且它正在工作但我无法让它返回数字作为我可以使用的东西（即字符串或 int）它总是只返回'0'作为我的IP地址。

*

#!/usr/bin/env python3

import os

import subprocess

def quiet_scan():

    address = "ip addr | grep 'inet 10.*' | awk '{print $2}'"

    ipcidr = int(os.system(address))

    print(ipcidr)

    nmapCom = ('nmap -sS ' + str(ipcidr))

    print(nmapCom)

    final = (os.system(nmapCom))

    print(final

root@kali:/home/kali# ./recon.py -q 

10.0.2.15/24

0

最后我希望输出是nmap -sS 10.50.0.2代替最后的 0

Answer 1

试试这个脚本：#!/usr/bin/env python3import osimport subprocessdef run_cmd(cmd):    return subprocess.run(cmd, capture_output=True, shell=True).stdout.decode().strip()def quiet_scan():    address = run_cmd("ip addr | grep 'inet 10.*' | awk '{print $2}'")    print(address)    final = run_cmd(f"nmap -sS {address}")     print(final)quiet_scan()该函数run_cmd将 cmd 作为字符串，用 shell 运行它，然后解码结果和最后一个换行符。

Answer 2

我认为subprocess.run使用capture_outputflag 在这里会很好用：import subprocessresult = subprocess.run(["nmap", "-sS"], capture_output=True)print(result.stdout)result将保存运行进程的结果，包括退出代码和写入流的内容。如果您想要的数据是nmap打印到 的数据stdout，您可以简单地检查该属性。