#2689 Initial Implementation of multi-port listeners.

src/primaite/notebooks/Command-&-Control-E2E-Demonstration.ipynb

@@ -133,6 +133,15 @@
     "          action: C2_SERVER_RANSOMWARE_LAUNCH\n",
     "          options:\n",
     "            node_id: 1\n",
+    "        7:\n",
+    "          action: CONFIGURE_C2_BEACON\n",
+    "          options:\n",
+    "            node_id: 0\n",
+    "            config:\n",
+    "              c2_server_ip_address: 192.168.10.21\n",
+    "              keep_alive_frequency: 10\n",
+    "              masquerade_protocol: TCP\n",
+    "              masquerade_port: DNS\n",
     "\n",
     "\n",
     "\n",
@@ -1045,6 +1054,127 @@
    "source": [
     "display_obs_diffs(pre_blue_action_obs, post_blue_action_obs, blue_env.game.step_counter)"
    ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## **Command and Control** | C2 Beacon Actions\n",
+    "\n",
+    "Before any C2 Server commands is able to accept any commands, it must first establish connection with a C2 beacon.\n",
+    "\n",
+    "This can be done by installing, configuring and then executing a C2 Beacon. "
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## **Command and Control** | Configurability \n",
+    "\n",
+    "TODO: Fleshout"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "with open(data_manipulation_config_path()) as f:\n",
+    "    cfg = yaml.safe_load(f)\n",
+    "    # removing all agents & adding the custom agent.\n",
+    "    cfg['agents'] = {}\n",
+    "    cfg['agents'] = c2_agent_yaml\n",
+    "    \n",
+    "\n",
+    "c2_config_env = PrimaiteGymEnv(env_config=cfg)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Installing the C2 Server"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "client_1: Computer = c2_config_env.game.simulation.network.get_node_by_hostname(\"client_1\")\n",
+    "client_1.software_manager.install(C2Server)\n",
+    "c2_server: C2Server = client_1.software_manager.software[\"C2Server\"]\n",
+    "c2_server.run()\n",
+    "client_1.software_manager.show()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Installing the C2 Beacon via NODE_APPLICATION_INSTALL"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "c2_config_env.step(1)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Configuring the C2 Beacon using different parameters:\n",
+    "\n",
+    "``` yaml\n",
+    "          action: CONFIGURE_C2_BEACON\n",
+    "          options:\n",
+    "            node_id: 0\n",
+    "            config:\n",
+    "              c2_server_ip_address: 192.168.10.21\n",
+    "              keep_alive_frequency: 10\n",
+    "              masquerade_protocol: TCP\n",
+    "              masquerade_port: DNS\n",
+    "```"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "c2_config_env.step(7)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Establishing connection to the C2 Server.\n",
+    "c2_config_env.step(3)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "web_server: Server = c2_config_env.game.simulation.network.get_node_by_hostname(\"web_server\")\n",
+    "c2_beacon: C2Beacon = web_server.software_manager.software[\"C2Beacon\"]\n",
+    "c2_beacon.show()\n",
+    "c2_server.show()"
+   ]
   }
  ],
  "metadata": {