@@ -442,6 +442,13 @@ def new(config):
442442 "when" : lambda answers : answers ["windows_server_one" ]
443443 and answers ["windows_server_one_dc" ],
444444 },
445+ {
446+ "type" : "confirm" ,
447+ "message" : "should we install a MITRE Caldera agent on the windows server" ,
448+ "name" : "windows_server_one_caldera_agent" ,
449+ "default" : False ,
450+ "when" : lambda answers : answers ["windows_server_one" ],
451+ },
445452 ]
446453
447454 answers = questionary .prompt (questions )
@@ -463,6 +470,8 @@ def new(config):
463470 if "windows_server_one_bad_blood" in answers :
464471 if answers ["windows_server_one_bad_blood" ]:
465472 configuration ["windows_servers" ][0 ]["bad_blood" ] = "1"
473+ if answers ["windows_server_one_caldera_agent" ]:
474+ configuration ["windows_servers" ][0 ]["install_caldera_agent" ] = "1"
466475
467476 questions = [
468477 {
@@ -493,6 +502,13 @@ def new(config):
493502 "default" : False ,
494503 "when" : lambda answers : answers ["windows_server_two" ],
495504 },
505+ {
506+ "type" : "confirm" ,
507+ "message" : "should we install a MITRE Caldera agent on the windows server" ,
508+ "name" : "windows_server_two_caldera_agent" ,
509+ "default" : False ,
510+ "when" : lambda answers : answers ["windows_server_two" ],
511+ },
496512 ]
497513
498514 answers = questionary .prompt (questions )
@@ -510,6 +526,8 @@ def new(config):
510526 configuration ["windows_servers" ][1 ]["join_domain" ] = "1"
511527 if answers ["windows_server_two_red_team_tools" ]:
512528 configuration ["windows_servers" ][1 ]["install_red_team_tools" ] = "1"
529+ if answers ["windows_server_two_caldera_agent" ]:
530+ configuration ["windows_servers" ][1 ]["install_caldera_agent" ] = "1"
513531
514532 questions = [
515533 {
@@ -518,6 +536,13 @@ def new(config):
518536 "name" : "linux_server" ,
519537 "default" : False ,
520538 },
539+ {
540+ "type" : "confirm" ,
541+ "message" : "should we install a MITRE Caldera agent on the linux server" ,
542+ "name" : "linux_server_caldera_agent" ,
543+ "default" : False ,
544+ "when" : lambda answers : answers ["linux_server" ],
545+ },
521546 {
522547 "type" : "confirm" ,
523548 "message" : "shall we build a kali linux machine" ,
@@ -558,6 +583,13 @@ def new(config):
558583 "name" : "phantom_installer" ,
559584 "when" : lambda answers : answers ["phantom" ],
560585 },
586+ {
587+ "type" : "confirm" ,
588+ "message" : "shall we build a MITRE Caldera server for attack simulation" ,
589+ "name" : "caldera_server" ,
590+ "default" : False ,
591+ "when" : lambda answers : configuration ["general" ]["cloud_provider" ] == "aws" ,
592+ },
561593 ]
562594
563595 answers = questionary .prompt (questions )
@@ -569,6 +601,9 @@ def new(config):
569601 "hostname" : "ar-linux" ,
570602 }
571603 )
604+ if "linux_server_caldera_agent" in answers :
605+ if answers ["linux_server_caldera_agent" ]:
606+ configuration ["linux_servers" ][0 ]["install_caldera_agent" ] = "1"
572607
573608 if configuration ["general" ]["cloud_provider" ] == "aws" :
574609 if answers ["kali_machine" ]:
@@ -586,6 +621,10 @@ def new(config):
586621 if answers ["snort_server" ]:
587622 configuration ["snort_server" ] = dict ()
588623 configuration ["snort_server" ]["snort_server" ] = "1"
624+
625+ if answers ["caldera_server" ]:
626+ configuration ["caldera_server" ] = dict ()
627+ configuration ["caldera_server" ]["caldera_server" ] = "1"
589628
590629 if answers ["phantom" ]:
591630 configuration ["phantom_server" ] = dict ()
0 commit comments