1111from starlette .staticfiles import StaticFiles
1212
1313from empire .server .api .middleware import EmpireCORSMiddleware
14+ from empire .server .api .v2 .admin import admin_api
15+ from empire .server .api .v2 .agent import agent_api , agent_file_api , agent_task_api
16+ from empire .server .api .v2 .bypass import bypass_api
17+ from empire .server .api .v2 .credential import credential_api
18+ from empire .server .api .v2 .download import download_api
19+ from empire .server .api .v2 .host import host_api , process_api
20+ from empire .server .api .v2 .ip import ip_api
21+ from empire .server .api .v2 .listener import listener_api , listener_template_api
22+ from empire .server .api .v2 .meta import meta_api
23+ from empire .server .api .v2 .module import module_api
24+ from empire .server .api .v2 .obfuscation import obfuscation_api
25+ from empire .server .api .v2 .plugin import plugin_api , plugin_registry_api , plugin_task_api
26+ from empire .server .api .v2 .profile import profile_api
27+ from empire .server .api .v2 .stager import stager_api , stager_template_api
28+ from empire .server .api .v2 .tag import tag_api
29+ from empire .server .api .v2 .user import user_api
1430from empire .server .api .v2 .websocket .socketio import setup_socket_events
1531from empire .server .core .config .config_manager import empire_config
1632from empire .server .core .config .data_manager import sync_starkiller
@@ -71,27 +87,6 @@ def initialize(run: bool = True, cert_path=None): # noqa: PLR0915
7187 port = empire_config .api .port
7288 secure = empire_config .api .secure
7389
74- # Not pretty but allows us to use main_menu by delaying the import
75- from empire .server .api .v2 .admin import admin_api
76- from empire .server .api .v2 .agent import agent_api , agent_file_api , agent_task_api
77- from empire .server .api .v2 .bypass import bypass_api
78- from empire .server .api .v2 .credential import credential_api
79- from empire .server .api .v2 .download import download_api
80- from empire .server .api .v2 .host import host_api , process_api
81- from empire .server .api .v2 .ip import ip_api
82- from empire .server .api .v2 .listener import listener_api , listener_template_api
83- from empire .server .api .v2 .meta import meta_api
84- from empire .server .api .v2 .module import module_api
85- from empire .server .api .v2 .obfuscation import obfuscation_api
86- from empire .server .api .v2 .plugin import (
87- plugin_api ,
88- plugin_registry_api ,
89- plugin_task_api ,
90- )
91- from empire .server .api .v2 .profile import profile_api
92- from empire .server .api .v2 .stager import stager_api , stager_template_api
93- from empire .server .api .v2 .tag import tag_api
94- from empire .server .api .v2 .user import user_api
9590 from empire .server .server import main
9691
9792 @asynccontextmanager
@@ -106,29 +101,29 @@ async def lifespan(app: FastAPI):
106101
107102 app = FastAPI (lifespan = lifespan )
108103
109- app .include_router (listener_template_api .router )
110- app .include_router (listener_api .router )
111- app .include_router (stager_template_api .router )
112- app .include_router (stager_api .router )
104+ app .include_router (admin_api .router )
105+ app .include_router (agent_file_api .router )
113106 app .include_router (agent_task_api .router )
114107 app .include_router (agent_api .router )
115- app .include_router (agent_file_api .router )
116- app .include_router (user_api .router )
117- app .include_router (module_api .router )
118108 app .include_router (bypass_api .router )
119- app .include_router (obfuscation_api .router )
120- app .include_router (process_api .router )
121- app .include_router (profile_api .router )
122109 app .include_router (credential_api .router )
123- app .include_router (host_api .router )
124110 app .include_router (download_api .router )
111+ app .include_router (host_api .router )
112+ app .include_router (ip_api .router )
113+ app .include_router (listener_api .router )
114+ app .include_router (listener_template_api .router )
125115 app .include_router (meta_api .router )
116+ app .include_router (module_api .router )
117+ app .include_router (obfuscation_api .router )
118+ app .include_router (plugin_registry_api .router )
126119 app .include_router (plugin_task_api .router )
127120 app .include_router (plugin_api .router )
128- app .include_router (plugin_registry_api .router )
121+ app .include_router (process_api .router )
122+ app .include_router (profile_api .router )
123+ app .include_router (stager_api .router )
124+ app .include_router (stager_template_api .router )
129125 app .include_router (tag_api .router )
130- app .include_router (ip_api .router )
131- app .include_router (admin_api .router )
126+ app .include_router (user_api .router )
132127
133128 app .add_middleware (
134129 EmpireCORSMiddleware ,
@@ -149,9 +144,6 @@ async def lifespan(app: FastAPI):
149144 sio = socketio .AsyncServer (
150145 async_mode = "asgi" ,
151146 cors_allowed_origins = "*" ,
152- # logger=True,
153- # engineio_logger=True,
154- # https://github.com/miguelgrinberg/flask-socketio/issues/274#issuecomment-231206374
155147 json = MyJsonWrapper ,
156148 )
157149 sio_app = socketio .ASGIApp (
@@ -179,7 +171,6 @@ async def lifespan(app: FastAPI):
179171 lifespan = "on" ,
180172 ssl_keyfile = f"{ cert_path } /empire-priv.key" ,
181173 ssl_certfile = f"{ cert_path } /empire-chain.pem" ,
182- # log_level="info",
183174 )
184175 else :
185176 uvicorn .run (
@@ -188,7 +179,6 @@ async def lifespan(app: FastAPI):
188179 port = port ,
189180 log_config = None ,
190181 lifespan = "on" ,
191- # log_level="info",
192182 )
193183
194184 return app
0 commit comments