#!/usr/bin/env python3
"""Serves attacker.html on :8080 and victim.html on :8081 (cross-origin)."""
import http.server
import os
import socketserver
import threading

ROOT = os.path.dirname(os.path.abspath(__file__))


class Handler(http.server.SimpleHTTPRequestHandler):
    def __init__(self, *a, **kw):
        super().__init__(*a, directory=ROOT, **kw)

    def end_headers(self):
        # Permit cross-origin prerender of the victim.
        self.send_header("Supports-Loading-Mode", "credentialed-prerender")
        self.send_header("Cache-Control", "no-store")
        super().end_headers()


def serve(port):
    socketserver.TCPServer.allow_reuse_address = True
    with socketserver.TCPServer(("0.0.0.0", port), Handler) as httpd:
        print(f"[+] serving on http://0.0.0.0:{port}/")
        httpd.serve_forever()


if __name__ == "__main__":
    threading.Thread(target=serve, args=(8081,), daemon=True).start()
    serve(8080)