Gracefully handle lack of key/cert files.
This commit is contained in:
parent
808edc594a
commit
5ddc43e77a
10
agena.py
10
agena.py
|
@ -7,13 +7,12 @@ import subprocess
|
||||||
import socket
|
import socket
|
||||||
import socketserver
|
import socketserver
|
||||||
import ssl
|
import ssl
|
||||||
|
import sys
|
||||||
import tempfile
|
import tempfile
|
||||||
import urllib.parse
|
import urllib.parse
|
||||||
|
|
||||||
HOST, PORT = "localhost", 1965
|
HOST, PORT = "localhost", 1965
|
||||||
|
|
||||||
context = ssl.create_default_context(ssl.Purpose.CLIENT_AUTH)
|
|
||||||
context.load_cert_chain(certfile="cert.pem", keyfile="key.pem")
|
|
||||||
|
|
||||||
class AgenaHandler(socketserver.BaseRequestHandler):
|
class AgenaHandler(socketserver.BaseRequestHandler):
|
||||||
|
|
||||||
|
@ -190,6 +189,13 @@ class AgenaHandler(socketserver.BaseRequestHandler):
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|
||||||
|
if not (os.path.exists("cert.pem") and os.path.exists("key.pem")):
|
||||||
|
print("Couldn't find cert.pem and/or key.pem. :(")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
context = ssl.create_default_context(ssl.Purpose.CLIENT_AUTH)
|
||||||
|
context.load_cert_chain(certfile="cert.pem", keyfile="key.pem")
|
||||||
|
|
||||||
agena = socketserver.TCPServer((HOST, PORT), AgenaHandler)
|
agena = socketserver.TCPServer((HOST, PORT), AgenaHandler)
|
||||||
try:
|
try:
|
||||||
agena.serve_forever()
|
agena.serve_forever()
|
||||||
|
|
Loading…
Reference in New Issue