0

我在调用 Python 中设置为“全局”的变量时遇到问题。

我在脚本开头设置变量，然后在函数内部修改变量，然后在脚本结束时回显变量的值，以及我将变量设置为的值function 似乎没有执行该功能。

脚本（用于扫描 PC 的前 1000 个端口的简单端口扫描器）。此外，我使用进程设置脚本的原因是因为我同时调用多个端口部分......但是我没有包含所有功能，因为它是一个相当长的脚本。

#!/usr/bin/env python
import socket
import subprocess
import sys
from datetime import datetime
from multiprocessing import Process
import random
import datetime

# Clear the screen
#subprocess.call('clear', shell=True)

# Ask for input
#remoteServer    = raw_input("Enter a remote host to scan: ")
remoteServer = sys.argv[1]
remoteServerIP  = socket.gethostbyname(remoteServer)

global openports
openports = []

def ports1():
    global openports
    for port in random.sample(range(1,1000), 999):
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sock.settimeout(.01)
        result = sock.connect_ex((remoteServerIP, port))
        if result == 0:
            openports.append(format(port))
        sock.close()


if __name__ == '__main__':
    p1 = Process(target = ports1)
    p1.start()
    p1.join()
    print openports

我知道变量正在设置，因为如果我openports从函数内部回显变量，它将返回正确的数据，但是当我在 name==main 部分中调用函数后回显它时，它返回为空细绳。

4

2 回答 2

3

发生这种情况是因为您使用单独的进程来查询端口。该进程openports中的变量被更新，而不是父进程中的变量！

有多种可能的解决方案。

您可以ports1()在原始过程中调用该函数，因为无论如何您都在等待它。

如果要并行化扫描仪，可以使用 amultiprocessing.Pool进行端口扫描。池中调用的函数可以返回打开端口或None关闭端口的端口号。例如：

from multiprocessing import Pool
import random
import socket
import sys

remoteServer = sys.argv[1]
remoteServerIP  = socket.gethostbyname(remoteServer)

ports = random.sample(range(1,1000), 999)

def testport(num):
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.settimeout(.01)
    result = sock.connect_ex((remoteServerIP, port))
    sock.close()
    if result == 0:
        return num
    return None

p = Pool()
openports = p.map(testport, ports)
openports = [prt for prt in openports if prt is not None]
于 2015-12-24T16:13:43.160 回答
0

尝试global从您定义它的第一个位置删除关键字。我相信您可以在那里将其定义为列表，然后global在更改之前在函数内使用关键字。打印出来时是否会给您正确的结果？

于 2015-12-24T16:15:07.937 回答