Removed old monolithic files
This commit is contained in:
1264
src/handler.py
1264
src/handler.py
File diff suppressed because it is too large
Load Diff
138
src/server.py
138
src/server.py
@@ -1,138 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
"""
|
||||
Main server module for the deception honeypot.
|
||||
Run this file to start the server.
|
||||
"""
|
||||
|
||||
import sys
|
||||
from http.server import HTTPServer
|
||||
|
||||
from config import get_config
|
||||
from tracker import AccessTracker
|
||||
from handler import Handler
|
||||
from logger import (
|
||||
initialize_logging,
|
||||
get_app_logger,
|
||||
get_access_logger,
|
||||
get_credential_logger,
|
||||
)
|
||||
from database import initialize_database
|
||||
from tasks_master import get_tasksmaster
|
||||
|
||||
|
||||
def print_usage():
|
||||
"""Print usage information"""
|
||||
print(f"Usage: {sys.argv[0]} [FILE]\n")
|
||||
print("FILE is file containing a list of webpage names to serve, one per line.")
|
||||
print("If no file is provided, random links will be generated.\n")
|
||||
print("Configuration:")
|
||||
print(" Configuration is loaded from a YAML file (default: config.yaml)")
|
||||
print("Set CONFIG_LOCATION environment variable to use a different file.\n")
|
||||
print("Example config.yaml structure:")
|
||||
print("server:")
|
||||
print("port: 5000")
|
||||
print("delay: 100")
|
||||
print("links:")
|
||||
print("min_length: 5")
|
||||
print("max_length: 15")
|
||||
print("min_per_page: 10")
|
||||
print("max_per_page: 15")
|
||||
print("canary:")
|
||||
print("token_url: null")
|
||||
print("token_tries: 10")
|
||||
print("dashboard:")
|
||||
print("secret_path: null # auto-generated if not set")
|
||||
print("database:")
|
||||
print('path: "data/krawl.db"')
|
||||
print("retention_days: 30")
|
||||
print("behavior:")
|
||||
print("probability_error_codes: 0")
|
||||
|
||||
|
||||
def main():
|
||||
"""Main entry point for the deception server"""
|
||||
if "-h" in sys.argv or "--help" in sys.argv:
|
||||
print_usage()
|
||||
exit(0)
|
||||
|
||||
config = get_config()
|
||||
|
||||
# Initialize logging with timezone
|
||||
initialize_logging()
|
||||
app_logger = get_app_logger()
|
||||
access_logger = get_access_logger()
|
||||
credential_logger = get_credential_logger()
|
||||
|
||||
# Initialize database for persistent storage
|
||||
try:
|
||||
initialize_database(config.database_path)
|
||||
app_logger.info(f"Database initialized at: {config.database_path}")
|
||||
except Exception as e:
|
||||
app_logger.warning(
|
||||
f"Database initialization failed: {e}. Continuing with in-memory only."
|
||||
)
|
||||
|
||||
tracker = AccessTracker(config.max_pages_limit, config.ban_duration_seconds)
|
||||
|
||||
Handler.config = config
|
||||
Handler.tracker = tracker
|
||||
Handler.counter = config.canary_token_tries
|
||||
Handler.app_logger = app_logger
|
||||
Handler.access_logger = access_logger
|
||||
Handler.credential_logger = credential_logger
|
||||
|
||||
if len(sys.argv) == 2:
|
||||
try:
|
||||
with open(sys.argv[1], "r") as f:
|
||||
Handler.webpages = f.readlines()
|
||||
|
||||
if not Handler.webpages:
|
||||
app_logger.warning(
|
||||
"The file provided was empty. Using randomly generated links."
|
||||
)
|
||||
Handler.webpages = None
|
||||
except IOError:
|
||||
app_logger.warning("Can't read input file. Using randomly generated links.")
|
||||
|
||||
# tasks master init
|
||||
tasks_master = get_tasksmaster()
|
||||
tasks_master.run_scheduled_tasks()
|
||||
|
||||
try:
|
||||
|
||||
banner = f"""
|
||||
|
||||
============================================================
|
||||
DASHBOARD AVAILABLE AT
|
||||
{config.dashboard_secret_path}
|
||||
============================================================
|
||||
"""
|
||||
app_logger.info(banner)
|
||||
app_logger.info(f"Starting deception server on port {config.port}...")
|
||||
if config.canary_token_url:
|
||||
app_logger.info(
|
||||
f"Canary token will appear after {config.canary_token_tries} tries"
|
||||
)
|
||||
else:
|
||||
app_logger.info(
|
||||
"No canary token configured (set CANARY_TOKEN_URL to enable)"
|
||||
)
|
||||
|
||||
server = HTTPServer(("0.0.0.0", config.port), Handler)
|
||||
app_logger.info("Server started. Use <Ctrl-C> to stop.")
|
||||
server.serve_forever()
|
||||
except KeyboardInterrupt:
|
||||
app_logger.info("Stopping server...")
|
||||
server.socket.close()
|
||||
app_logger.info("Server stopped")
|
||||
except Exception as e:
|
||||
app_logger.error(f"Error starting HTTP server on port {config.port}: {e}")
|
||||
app_logger.error(
|
||||
f"Make sure you are root, if needed, and that port {config.port} is open."
|
||||
)
|
||||
exit(1)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user