pwntools_http_server.py

import argparse
import sys

from pwnlib.tubes.listen import listen


def main(cli_arguments):
    l = listen(port=cli_arguments.port, bindaddr='localhost', typ='tcp')
    _ = l.wait_for_connection()

    request = l.recv()

    message = "request received:\n\t%s" % str(request)
    length = len(message)

    response = """HTTP/1.1 200 OK
    Content-Length: %s
    Content-Type: text/html

    %s""" % (length, message)

    l.send(response)

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument("port", help="The port to listen to.")
    cli_arguments = parser.parse_args()

    main(cli_arguments)