oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update more tests to new actions schema

Browse Source
This commit is contained in:
Marek Wolan
2025-01-17 14:38:59 +00:00
parent 03cab0fcec
commit 858406c4a3
48 changed files with 351 additions and 413 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/assets/configs/action_penalty.yaml
View File

@@ -386,52 +386,52 @@ agents:
52: # old action num: 28
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 0
53: # old action num: 29
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 1
54: # old action num: 30
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 2
55: # old action num: 31
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 3
56: # old action num: 32
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 4
57: # old action num: 33
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 5
58: # old action num: 34
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 6
59: # old action num: 35
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 7
60: # old action num: 36
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 8
61: # old action num: 37
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 9
62: # old action num: 38
action: "host_nic_disable"

4
tests/assets/configs/basic_firewall.yaml
View File

@@ -39,10 +39,6 @@ agents:
node_name: client_2
application_name: WebBrowser
reward_function:
reward_components:
- type: DUMMY
agent_settings:
action_probabilities:
0: 0.4

3
tests/assets/configs/basic_switched_network.yaml
View File

@@ -51,9 +51,6 @@ agents:
0: 0.4
1: 0.6
- ref: defender
team: BLUE
type: ProxyAgent

20
tests/assets/configs/data_manipulation.yaml
View File

@@ -477,52 +477,52 @@ agents:
52: # old action num: 28
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 0
53: # old action num: 29
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 1
54: # old action num: 30
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 2
55: # old action num: 31
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 3
56: # old action num: 32
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 4
57: # old action num: 33
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 5
58: # old action num: 34
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 6
59: # old action num: 35
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 7
60: # old action num: 36
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 8
61: # old action num: 37
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 9
62: # old action num: 38
action: "host_nic_disable"

2
tests/assets/configs/dmz_network.yaml
View File

@@ -62,7 +62,7 @@ agents:
action: node_application_execute
options:
node_name: client_1
application_id: WebBrowser
application_name: WebBrowser
agent_settings:
action_probabilities:

32
tests/assets/configs/extended_config.yaml
View File

@@ -529,82 +529,82 @@ agents:
action: "host_nic_disable"
options:
node_name: domain_controller
nic_id: 0
nic_num: 1
63: # old action num: 39
action: "host_nic_enable"
options:
node_name: domain_controller
nic_id: 0
nic_num: 1
64: # old action num: 40
action: "host_nic_disable"
options:
node_name: web_server
nic_id: 0
nic_num: 1
65: # old action num: 41
action: "host_nic_enable"
options:
node_name: web_server
nic_id: 0
nic_num: 1
66: # old action num: 42
action: "host_nic_disable"
options:
node_name: database_server
nic_id: 0
nic_num: 1
67: # old action num: 43
action: "host_nic_enable"
options:
node_name: database_server
nic_id: 0
nic_num: 1
68: # old action num: 44
action: "host_nic_disable"
options:
node_name: backup_server
nic_id: 0
nic_num: 1
69: # old action num: 45
action: "host_nic_enable"
options:
node_name: backup_server
nic_id: 0
nic_num: 1
70: # old action num: 46
action: "host_nic_disable"
options:
node_name: security_suite
nic_id: 0
nic_num: 1
71: # old action num: 47
action: "host_nic_enable"
options:
node_name: security_suite
nic_id: 0
nic_num: 1
72: # old action num: 48
action: "host_nic_disable"
options:
node_name: security_suite
nic_id: 1
nic_num: 2
73: # old action num: 49
action: "host_nic_enable"
options:
node_name: security_suite
nic_id: 1
nic_num: 2
74: # old action num: 50
action: "host_nic_disable"
options:
node_name: client_1
nic_id: 0
nic_num: 1
75: # old action num: 51
action: "host_nic_enable"
options:
node_name: client_1
nic_id: 0
nic_num: 1
76: # old action num: 52
action: "host_nic_disable"
options:
node_name: client_2
nic_id: 0
nic_num: 1
77: # old action num: 53
action: "host_nic_enable"
options:
node_name: client_2
nic_id: 0
nic_num: 1

35
tests/assets/configs/firewall_actions_network.yaml
View File

@@ -109,12 +109,12 @@ agents:
position: 1
permission: PERMIT
src_ip: 192.168.0.10
dst_ip: 0.0.0.0
dst_ip: ALL
src_port: 80
dst_port: HTTP
protocol_name: TCP
src_wildcard: 0
dst_wildcard: 0
src_wildcard: NONE
dst_wildcard: NONE
2:
action: firewall_acl_remove_rule
options:
@@ -135,8 +135,8 @@ agents:
src_port: ARP
dst_port: DNS
protocol_name: ICMP
source_wildcard_id: 0
dest_wildcard_id: 0
src_wildcard: NONE
dst_wildcard: NONE
4:
action: firewall_acl_remove_rule
options:
@@ -157,8 +157,8 @@ agents:
src_port: HTTP
dst_port: HTTP
protocol_name: UDP
source_wildcard_id: 0
dest_wildcard_id: 0
src_wildcard: NONE
dst_wildcard: NONE
6:
action: firewall_acl_remove_rule
options:
@@ -179,8 +179,8 @@ agents:
src_port: HTTP
dst_port: HTTP
protocol_name: TCP
source_wildcard_id: 0
dest_wildcard_id: 0
src_wildcard: NONE
dst_wildcard: NONE
8:
action: firewall_acl_remove_rule
options:
@@ -201,8 +201,8 @@ agents:
src_port: POSTGRES_SERVER
dst_port: POSTGRES_SERVER
protocol_name: ICMP
source_wildcard_id: 0
dest_wildcard_id: 0
src_wildcard: NONE
dst_wildcard: NONE
10:
action: firewall_acl_remove_rule
options:
@@ -223,8 +223,8 @@ agents:
src_port: NONE
dst_port: NONE
protocol_name: none
source_wildcard_id: 0
dest_wildcard_id: 0
src_wildcard: NONE
dst_wildcard: NONE
12:
action: firewall_acl_remove_rule
options:
@@ -237,17 +237,14 @@ agents:
options:
type: network_port_disable
target_nodename: firewall
port_id: 3
port_num: 3
14:
action: network_port_enable
options:
type: network_port_enable
target_nodename: firewall
port_id: 3
agent_settings:
start_step: 5
frequency: 4
variance: 3
port_num: 3

28
tests/assets/configs/install_and_configure_apps.yaml
View File

@@ -43,34 +43,30 @@ agents:
action: configure_database_client
options:
node_name: client_1
config:
server_ip_address: 10.0.0.5
server_ip_address: 10.0.0.5
5:
action: configure_database_client
options:
node_name: client_1
config:
server_password: correct_password
server_password: correct_password
6:
action: c2_server_ransomware_configure
options:
node_name: client_2
config:
server_ip_address: 10.0.0.5
server_password: correct_password
payload: ENCRYPT
server_ip_address: 10.0.0.5
server_password: correct_password
payload: ENCRYPT
7:
action: configure_dos_bot
options:
node_name: client_3
config:
target_ip_address: 10.0.0.5
target_port: POSTGRES_SERVER
payload: DELETE
repeat: true
port_scan_p_of_success: 1.0
dos_intensity: 1.0
max_sessions: 1000
target_ip_address: 10.0.0.5
target_port: POSTGRES_SERVER
payload: DELETE
repeat: true
port_scan_p_of_success: 1.0
dos_intensity: 1.0
max_sessions: 1000
8:
action: node_application_install
options:

40
tests/assets/configs/multi_agent_session.yaml
View File

@@ -469,52 +469,52 @@ agents:
52: # old action num: 28
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 0
53: # old action num: 29
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 1
54: # old action num: 30
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 2
55: # old action num: 31
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 3
56: # old action num: 32
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 4
57: # old action num: 33
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 5
58: # old action num: 34
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 6
59: # old action num: 35
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 7
60: # old action num: 36
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 8
61: # old action num: 37
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 9
62: # old action num: 38
action: "host_nic_disable"
@@ -983,52 +983,52 @@ agents:
52: # old action num: 28
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 0
53: # old action num: 29
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 1
54: # old action num: 30
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 2
55: # old action num: 31
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 3
56: # old action num: 32
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 4
57: # old action num: 33
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 5
58: # old action num: 34
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 6
59: # old action num: 35
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 7
60: # old action num: 36
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 8
61: # old action num: 37
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 9
62: # old action num: 38
action: "host_nic_disable"

2
tests/assets/configs/nmap_ping_scan_red_agent_config.yaml
View File

@@ -28,7 +28,7 @@ agents:
0:
action: node_nmap_ping_scan
options:
node_name: client_1
source_node: client_1
target_ip_address: 192.168.1.0/24
show: False

32
tests/assets/configs/shared_rewards.yaml
View File

@@ -519,82 +519,82 @@ agents:
action: "host_nic_disable"
options:
node_name: domain_controller
nic_id: 0
nic_num: 1
63: # old action num: 39
action: "host_nic_enable"
options:
node_name: domain_controller
nic_id: 0
nic_num: 1
64: # old action num: 40
action: "host_nic_disable"
options:
node_name: web_server
nic_id: 0
nic_num: 1
65: # old action num: 41
action: "host_nic_enable"
options:
node_name: web_server
nic_id: 0
nic_num: 1
66: # old action num: 42
action: "host_nic_disable"
options:
node_name: database_server
nic_id: 0
nic_num: 1
67: # old action num: 43
action: "host_nic_enable"
options:
node_name: database_server
nic_id: 0
nic_num: 1
68: # old action num: 44
action: "host_nic_disable"
options:
node_name: backup_server
nic_id: 0
nic_num: 1
69: # old action num: 45
action: "host_nic_enable"
options:
node_name: backup_server
nic_id: 0
nic_num: 1
70: # old action num: 46
action: "host_nic_disable"
options:
node_name: security_suite
nic_id: 0
nic_num: 1
71: # old action num: 47
action: "host_nic_enable"
options:
node_name: security_suite
nic_id: 0
nic_num: 1
72: # old action num: 48
action: "host_nic_disable"
options:
node_name: security_suite
nic_id: 1
nic_num: 2
73: # old action num: 49
action: "host_nic_enable"
options:
node_name: security_suite
nic_id: 1
nic_num: 2
74: # old action num: 50
action: "host_nic_disable"
options:
node_name: client_1
nic_id: 0
nic_num: 1
75: # old action num: 51
action: "host_nic_enable"
options:
node_name: client_1
nic_id: 0
nic_num: 1
76: # old action num: 52
action: "host_nic_disable"
options:
node_name: client_2
nic_id: 0
nic_num: 1
77: # old action num: 53
action: "host_nic_enable"
options:
node_name: client_2
nic_id: 0
nic_num: 1
reward_function:
reward_components:

27
tests/assets/configs/test_application_install.yaml
View File

@@ -472,52 +472,52 @@ agents:
52: # old action num: 28
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 0
53: # old action num: 29
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 1
54: # old action num: 30
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 2
55: # old action num: 31
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 3
56: # old action num: 32
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 4
57: # old action num: 33
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 5
58: # old action num: 34
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 6
59: # old action num: 35
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 7
60: # old action num: 36
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 8
61: # old action num: 37
action: "router_acl_remove_rule"
options:
target_router_hostname: router_1
target_router: router_1
position: 9
62: # old action num: 38
action: "host_nic_disable"
@@ -618,14 +618,13 @@ agents:
action: node_application_execute
options:
node_name: domain_controller
application_id: 0
application_name: DoSBot
82:
action: configure_dos_bot
options:
node_name: domain_controller
config:
target_ip_address: 192.168.1.14
target_port: POSTGRES_SERVER
target_ip_address: 192.168.1.14
target_port: POSTGRES_SERVER
reward_function:
reward_components: