#!/usr/bin/env python3
import http.server
import socketserver
import sys

PORT = 8080

class Handler(http.server.SimpleHTTPRequestHandler):
    def log_message(self, fmt, *args):
        sys.stderr.write("[server] %s\n" % (fmt % args))
        sys.stderr.flush()

    def do_GET(self):
        if self.path.startswith('/victim-saw'):
            sys.stderr.write("[server] *** VICTIM EXTENSION OBSERVED: %s\n" % self.path)
            sys.stderr.flush()
            self.send_response(200)
            self.send_header('Content-Type', 'text/plain')
            self.end_headers()
            self.wfile.write(b'ok')
            return
        if self.path.startswith('/probe'):
            self.send_response(200)
            self.send_header('Content-Type', 'text/plain')
            self.end_headers()
            self.wfile.write(b'probe')
            return
        return super().do_GET()

socketserver.TCPServer.allow_reuse_address = True
with socketserver.TCPServer(("127.0.0.1", PORT), Handler) as httpd:
    sys.stderr.write("[server] serving on 127.0.0.1:%d\n" % PORT)
    sys.stderr.flush()
    httpd.serve_forever()