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

Question

我现在正在尝试制作一个包含多个网络扫描工具的工具。使用 os 模块我将代码发送到控制台以查找我的 ip + CIDR 并且它正在工作但我无法让它返回数字作为我可以使用的东西（即字符串或 int）它总是只返回'0'作为我的IP地址。*#!/usr/bin/env python3import osimport subprocessdef 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(finalroot@kali:/home/kali# ./recon.py -q 10.0.2.15/240最后我希望输出是nmap -sS 10.50.0.2代替最后的 0

Answer 1

试试这个脚本：

#!/usr/bin/env python3

import os

import subprocess

def 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 subprocess

result = subprocess.run(["nmap", "-sS"], capture_output=True)

print(result.stdout)

result将保存运行进程的结果，包括退出代码和写入流的内容。如果您想要的数据是nmap打印到 的数据stdout，您可以简单地检查该属性。