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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
# SPDX-FileCopyrightText: 2018-2024 Greenbone AG
#
# SPDX-License-Identifier: GPL-3.0-or-later
#
import unittest
from collections import OrderedDict
from gvm.errors import RequiredArgument
from gvm.protocols.ospv1 import Osp
from .. import MockConnection
class OSPStartScanTestCase(unittest.TestCase):
def setUp(self):
self.connection = MockConnection()
self.osp = Osp(self.connection)
def test_start_scan(self):
scanner_params = OrderedDict()
scanner_params["key1"] = "value1"
targets = list()
_target1 = OrderedDict()
_target1["hosts"] = "localhost"
_target1["ports"] = "22,80"
targets.append(_target1)
_target2 = OrderedDict()
_target2["hosts"] = "192.168.10.1"
_target2["ports"] = "443"
_smb = OrderedDict()
_smb["username"] = "username"
_smb["password"] = "pass"
_smb["port"] = "port"
_smb["type"] = "type"
_credential1 = {"smb": _smb}
_target2["credentials"] = _credential1
targets.append(_target2)
vts = OrderedDict()
vts["vt1"] = {"value_id": "value"}
vts["vt_groups"] = ["family=A", "family=B"]
self.osp.start_scan(
scan_id="123-456",
parallel=10,
scanner_params=scanner_params,
targets=targets,
vt_selection=vts,
)
self.connection.send.has_been_called_with(
b'<start_scan scan_id="123-456" parallel="10">'
b'<scanner_params key1="value1"/>'
b"<targets><target><hosts>localhost</hosts>"
b"<ports>22,80</ports></target>"
b"<target><hosts>192.168.10.1</hosts>"
b"<ports>443</ports>"
b"<credentials>"
b'<credential type="type" port="port" service="smb">'
b"<username>username</username>"
b"<password>pass</password>"
b"</credential></credentials>"
b"</target></targets><vt_selection>"
b'<vt_single id="vt1">'
b'<vt_value id="value_id">value</vt_value></vt_single>'
b'<vt_group filter="family=A"/><vt_group filter="family=B"/>'
b"</vt_selection></start_scan>"
)
def test_start_scan_without_target(self):
with self.assertRaises(RequiredArgument):
self.osp.start_scan()
def test_start_scan_legacy(self):
self.osp.start_scan(
scan_id="123-456", parallel=10, target="localhost", ports="22"
)
self.connection.send.has_been_called_with(
b'<start_scan scan_id="123-456" parallel="10" '
b'target="localhost" ports="22">'
b"<scanner_params/></start_scan>"
)
if __name__ == "__main__":
unittest.main()
|
