Commit Graph

10262 Commits

Author SHA1 Message Date
Shreya Malviya 41e7cfb768 Common: Remove agent sub-configuration schemas 2022-09-02 12:05:08 +05:30
Shreya Malviya f2c9952685 Common: Remove old sub-configurations using marshmallow 2022-09-02 12:05:08 +05:30
Shreya Malviya 1311fd5d27 Common: Remove old AgentConfiguration using marshmallow 2022-09-02 12:05:08 +05:30
Shreya Malviya 6d29829808 UT: Modify tests to use new pydantic configurations
TODO: Fix error handling and some assertions (tuple/list stuff)
2022-09-02 12:05:08 +05:30
Shreya Malviya 275237c3f7 Common: Inherit from MutableInfectionMonkeyBaseModel where missing in new pydantic configurations 2022-09-02 12:05:08 +05:30
Shreya Malviya 2f05d22780 Common: Create AgentConfiguration using pydantic 2022-09-02 12:05:08 +05:30
Shreya Malviya b8914101a6 Common: Remove unused imports in agent_sub_configurations.py 2022-09-02 12:05:08 +05:30
Shreya Malviya 501f752248 Common: Create PropagationConfiguration using pydantic 2022-09-02 12:05:08 +05:30
Shreya Malviya 602604e408 Common: Create ExploitationConfiguration using pydantic 2022-09-02 12:05:08 +05:30
Shreya Malviya b74f90fe9b Common: Create ExploitationOptionsConfiguration using pydantic 2022-09-02 12:05:08 +05:30
Shreya Malviya 153c3e9b9f Common: Create NetworkScanConfiguration using pydantic 2022-09-02 12:05:08 +05:30
Shreya Malviya 63a8e81c74 Common: Create TCPScanConfiguration using pydantic 2022-09-02 12:05:07 +05:30
Shreya Malviya 858b7650ac Common: Create ICMPScanConfiguration using pydantic 2022-09-02 12:05:07 +05:30
Shreya Malviya e7b0e6babf Common: Create ScanTargetConfiguration using pydantic 2022-09-02 12:05:07 +05:30
Shreya Malviya 520183f42d Common: Create PluginConfiguration using pydantic 2022-09-02 12:05:07 +05:30
Shreya Malviya 554a180fbd Common: Create CustomPBAConfiguration using pydantic 2022-09-02 12:05:07 +05:30
Mike Salvatore 765ce858fb Agent: Rename IAgentRepository -> IAgentBinaryRepository
PR #2239
2022-09-01 19:03:37 -04:00
Kekoa Kaaikala 62e4fb89bb Agent: Inject dependencies into TCPRelay 2022-09-01 19:28:06 +00:00
Kekoa Kaaikala 46573c9f2d UT: Add tests for RelayUserHandler 2022-09-01 19:24:13 +00:00
Kekoa Kaaikala b1c1d2bd39 Agent: Allow callback registration on TCPConnectionHandler 2022-09-01 19:19:05 +00:00
Kekoa Kaaikala ef39a8afd1 Agent: Add TCPPipeSpawner class 2022-09-01 19:15:23 +00:00
Mike Salvatore ccffd1eaee Project: Update vulture_allowlist.py for IAgentRepository changes 2022-09-01 12:43:45 -04:00
Mike Salvatore 3ab800f119 Island: Add docstrings to the Agent model 2022-09-01 12:25:29 -04:00
Mike Salvatore 693ce9e486 Island: Implement MongoAgentRepository.reset() 2022-09-01 12:25:29 -04:00
Mike Salvatore d854eb7576 Island: Implement MongoAgentRepository.upsert_agent() 2022-09-01 12:13:01 -04:00
Kekoa Kaaikala 0e869462b5 Agent: Refactor TCPRelay
Integrate TCPConnectionHandler and RelayUserHandler into TCPRelay
Remove TCPProxy
2022-09-01 15:06:47 +00:00
Kekoa Kaaikala d6931a6414 Agent: Clean up RelayUserHandler
Add get_potential_users and remove pipe handling
2022-09-01 15:03:16 +00:00
Mike Salvatore 1745b76122 Island: Implement MongoAgentRepository.get_running_agents() 2022-09-01 10:38:01 -04:00
Mike Salvatore ba228e56b6 Island: Implement MongoAgentRepository.get_agent_by_id() 2022-09-01 10:21:40 -04:00
Mike Salvatore 6f285ba80c Island: Rename `id` parameter to `agent_id` in IAgentRepository 2022-09-01 10:21:40 -04:00
Mike Salvatore f00268c4a1 Island: Add IAgentRepository.reset() 2022-09-01 10:21:40 -04:00
Mike Salvatore 1eefbdb2b1 Island: Allow Agent.parent_id to be None 2022-09-01 10:21:40 -04:00
Mike Salvatore 22701fc0a3 Island: Redefine IAgentRepository 2022-09-01 10:21:40 -04:00
Mike Salvatore f79d2fd8a6 Island: Add AgentID type alias 2022-09-01 10:21:40 -04:00
Ilija Lazoroski 15dee35aa4 Agent: Add RelayUserHandler 2022-09-01 16:02:27 +02:00
Kekoa Kaaikala 1f723b174e Agent: Add TCPConnectionHandler 2022-09-01 13:54:17 +00:00
Kekoa Kaaikala 2e7be823a1 Agent: Revert old TcpProxy and create a new one 2022-09-01 13:14:07 +00:00
Ilija Lazoroski c9b7f924a3 UT: Fix ExploitingInterceptingTelemetryMessenger tests 2022-09-01 13:58:14 +02:00
Ilija Lazoroski eee06467f6 Agent: Add TODO to rethink on_user_disconnected 2022-09-01 13:44:39 +02:00
Ilija Lazoroski 508b811939 Agent: Rename on_user_connected to add_relay_user 2022-09-01 13:41:50 +02:00
Ilija Lazoroski 43a1ddb714 Agent: Rename on_potential_new_user to add_potential_user 2022-09-01 13:38:31 +02:00
Ilija Lazoroski a754963222 Agent: Remove on_user_disconnected in TCPRelay 2022-09-01 13:32:07 +02:00
Ilija Lazoroski 6e0dc0ca13 Agent: Use super().__init__(...) in TCPRelay 2022-09-01 13:27:40 +02:00
Ilija Lazoroski 9fae6cca20 Agent: Use dictionary for relay and potential users in TCPRelay 2022-09-01 12:56:26 +02:00
Ilija Lazoroski 9a3afb051d Agent: Use IPv4Address for RelayUser.address 2022-09-01 12:10:21 +02:00
Ilija Lazoroski 1d394bbd2e Agent: Fix docstrings in TCPRelay 2022-09-01 11:39:25 +02:00
Ilija Lazoroski d478e99227 Agent: Make TCPRelay non-optional in ExploitInterceptingTelemetryMessenger 2022-09-01 11:29:03 +02:00
Ilija Lazoroski ecdf04a85e Agent: Change relay control message
'-' to 'infection-monkey-relay-control-message: -'
2022-09-01 10:35:29 +02:00
Ilija Lazoroski 94fba0fdf0 Agent: Rename RelayUser.time to last_update_time 2022-09-01 10:17:13 +02:00
Ilija Lazoroski 302cdaac70 Agent: Use event.wait() when running TCPRelay 2022-09-01 09:57:46 +02:00