Files
cs53-portfolio/lab-02-security-as-a-system/scenario/web_service.py
T

72 lines
1.5 KiB
Python

#!/usr/bin/env python3
import os
from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
from service_common import record_event
HOST = os.environ["MINICORP_SERVICE_HOST"]
PORT = int(os.environ["MINICORP_SERVICE_PORT"])
SERVICE_ID = os.environ["MINICORP_SERVICE_ID"]
SERVICE_TOKEN = os.environ["MINICORP_SERVICE_TOKEN"]
EVIDENCE_LOG = os.environ["MINICORP_EVIDENCE_LOG"]
class WebHandler(BaseHTTPRequestHandler):
def do_GET(self):
if self.path != "/":
self.send_error(404)
return
source = self.client_address[0]
record_event(
EVIDENCE_LOG,
service="web",
service_token=SERVICE_TOKEN,
source=source,
port=PORT,
)
body = (
"MiniCorp Employee Web Portal\n"
f"service_id: {SERVICE_ID}\n"
"environment: production\n"
"status: healthy\n"
).encode("utf-8")
self.send_response(200)
self.send_header(
"Content-Type",
"text/plain; charset=utf-8",
)
self.send_header(
"Content-Length",
str(len(body)),
)
self.end_headers()
self.wfile.write(body)
def log_message(self, format, *args):
pass
def main():
server = ThreadingHTTPServer(
(HOST, PORT),
WebHandler,
)
print(
f"MiniCorp web service listening on {HOST}:{PORT}"
)
server.serve_forever()
if __name__ == "__main__":
main()