blob: 5e9f4d0b8300a0f21873662411144bd94e0fadfe (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
from __future__ import annotations
import subprocess
import faker
from generalresearch.managers.network.tool_run import ToolRunManager
from generalresearch.models.network.definitions import IPProtocol
from generalresearch.models.network.nmap.execute import execute_nmap
from generalresearch.models.network.nmap.result import NmapResult, PortState
from generalresearch.models.network.tool_run import NmapRun, Status, ToolClass, ToolName
fake = faker.Faker()
def resolve(host: str):
return subprocess.check_output(["dig", host, "+short"]).decode().strip()
def test_execute_nmap_scanme(toolrun_manager: ToolRunManager):
ip = resolve("scanme.nmap.org")
run: NmapRun = execute_nmap(
ip=ip, top_ports=None, ports="20-30", enable_advanced=False
)
assert run.tool_name == ToolName.NMAP
assert run.tool_class == ToolClass.PORT_SCAN
assert run.ip == ip
assert isinstance(run.parsed, NmapResult)
result = run.parsed
port22 = result._port_index[(IPProtocol.TCP, 22)]
assert port22.state == PortState.OPEN
toolrun_manager.create_nmap_run(run)
|
