fix the restip argument

CHANGELOG.md

@@ -7,6 +7,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+-   Fix the restip argument which wasn't being used (@Vinnybod)
+
 ## [5.4.2] - 2023-06-07
 
 -   Updated Starkiller to v2.3.2

empire/server/api/app.py

@@ -52,7 +52,7 @@ def load_starkiller(v2App):
     )
 
 
-def initialize(secure: bool = False, port: int = 1337):
+def initialize(secure: bool = False, ip: str = "0.0.0.0", port: int = 1337):
     # Not pretty but allows us to use main_menu by delaying the import
     from empire.server.api.v2.agent import agent_api, agent_file_api, agent_task_api
     from empire.server.api.v2.bypass import bypass_api
@@ -143,7 +143,7 @@ def shutdown_event():
     if not secure:
         uvicorn.run(
             v2App,
-            host="0.0.0.0",
+            host=ip,
             port=port,
             log_config=None,
             lifespan="on",

empire/server/server.py

@@ -173,6 +173,6 @@ def run(args):
             subprocess.call("./setup/cert.sh")
             time.sleep(3)
 
-        app.initialize(secure=args.secure_api, port=args.restport)
+        app.initialize(secure=args.secure_api, ip=args.restip, port=args.restport)
 
     sys.exit()