-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtest_ssl.py
110 lines (91 loc) · 4.45 KB
/
test_ssl.py
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
# python 2.7
import json
import pytest
from v21.setting import STIX_CFG_SSL
from v21.cti_taxii2_client import Taxxi2Server, ApiRoot, Collection
from v21.cti_stix2_generator import _report_incident, get_config, _debug
config = {}
def trace(func):
def trace_func(*args, **kwargs):
ret = func(*args, **kwargs)
print(func.__name__)
return ret
return trace_func
@trace
def setup_module():
config_file = STIX_CFG_SSL
global config
config = get_config(config_file)
js = json.dumps(config, indent=4)
_debug('setup_module with config:\n{}'.format(js))
def test_server():
taxii2 = Taxxi2Server(config['protocol'], config['server'], config['port'], config['user'], config['password'],
config.get('verify', False), config['cert'], config['key'])
taxii2.show()
def test_api1_root():
taxii2 = Taxxi2Server(config['protocol'], config['server'], config['port'], config['user'], config['password'],
config.get('verify', False), config['cert'], config['key'])
api_root = taxii2.get_api_root('https://127.0.0.1:8080/api1/')
api_root.show()
def test_api1_root_ssl():
taxii2 = Taxxi2Server(config['protocol'], config['server'], config['port'], config['user'], config['password'],
config.get('verify', False), config['cert'], config['key'])
api_root = taxii2.get_api_root('https://127.0.0.1:444/api1/')
print(api_root)
api_root.show()
def test_api2_root():
taxii2 = Taxxi2Server(config['protocol'], config['server'], config['port'], config['user'], config['password'],
config.get('verify', False), config['cert'], config['key'])
api_root = taxii2.get_api_root('https://127.0.0.1:8080/api2/')
api_root.show()
def test_default_api_root():
taxii2 = Taxxi2Server(config['protocol'], config['server'], config['port'], config['user'], config['password'],
config.get('verify', False), config['cert'], config['key'])
api_root = taxii2.get_api_root()
api_root.show()
def test_collection_0():
taxii2 = Taxxi2Server(config['protocol'], config['server'], config['port'], config['user'], config['password'],
config.get('verify', False), config['cert'], config['key'])
api_root = taxii2.get_api_root()
collection_id = '472c94ae-3113-4e3e-a4dd-a9f4ac7471d4'
collection = api_root.get_collection(collection_id)
collection.show()
def test_collection_1():
taxii2 = Taxxi2Server(config['protocol'], config['server'], config['port'], config['user'], config['password'],
config.get('verify', False), config['cert'], config['key'])
api_root = taxii2.get_api_root()
collection_id = '52892447-4d7e-4f70-b94d-d7f22742ff63'
collection = api_root.get_collection(collection_id)
collection.show()
def test_collection_2():
taxii2 = Taxxi2Server(config['protocol'], config['server'], config['port'], config['user'], config['password'],
config.get('verify', False), config['cert'], config['key'])
api_root = taxii2.get_api_root()
collection_id = '64993447-4d7e-4f70-b94d-d7f33742ee63'
collection = api_root.get_collection(collection_id)
collection.show()
def test_collection_3():
taxii2 = Taxxi2Server(config['protocol'], config['server'], config['port'], config['user'], config['password'],
config.get('verify', False), config['cert'], config['key'])
api_root = taxii2.get_api_root()
collection_id = '91a7b528-80eb-42ed-a74d-c6fbd5a26116'
collection = api_root.get_collection(collection_id)
collection.show()
def test_collection_4():
taxii2 = Taxxi2Server(config['protocol'], config['server'], config['port'], config['user'], config['password'],
config.get('verify', False), config['cert'], config['key'])
taxii2.show()
api_root = taxii2.get_api_root()
collection_id = '365fed99-08fa-fdcd-a1b3-fb247eb41d01'
collection = api_root.get_collection(collection_id)
collection.show()
def test_report_incident():
# select id, type, aipmask, aport, vipmask, vport, protocol, sts, ets from incident where ...
incident = [2376151082326160871, 4, '10.95.7.111', 43262, '10.95.7.222', 445, 183, 1638310176, 1638310457]
_report_incident(incident, None, config)
# print out the results
taxii2 = Taxxi2Server(config['protocol'], config['server'], config['port'], config['user'], config['password'],
config.get('verify', False), config['cert'], config['key'])
api_root = taxii2.get_api_root()
collection = api_root.get_collection(config['collection_id'])
collection.show()