mirror of
https://github.com/Orange-OpenSource/hurl.git
synced 2024-11-30 10:56:09 +03:00
62 lines
1.7 KiB
Python
Executable File
62 lines
1.7 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# usage: ./server.py <port> <cert_file> <Client certificate authentication>
|
|
# Start the server with or without client certificate authentication
|
|
import flask
|
|
import sys
|
|
import ssl
|
|
|
|
app1 = flask.Flask("SSL Server")
|
|
|
|
|
|
@app1.route("/hello")
|
|
def hello():
|
|
return "Hello World!"
|
|
|
|
|
|
def start_server(port, cert_file, use_client_certificate_authentication):
|
|
ssl_context = get_ssl_context(cert_file, use_client_certificate_authentication)
|
|
app1.run(port=port, ssl_context=ssl_context)
|
|
|
|
|
|
def get_ssl_context(cert_file, use_client_certificate_authentication):
|
|
ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLSv1_2)
|
|
if use_client_certificate_authentication:
|
|
ssl_context.verify_mode = ssl.CERT_REQUIRED
|
|
ssl_context.load_verify_locations("ssl/ca/cert.pem")
|
|
ssl_context.load_cert_chain(cert_file, "ssl/server/key.pem")
|
|
return ssl_context
|
|
|
|
|
|
def print_usage_and_exit():
|
|
print(
|
|
"usage: ./server.py <port> <cert_file> <use_client_certificate_authentication>"
|
|
)
|
|
sys.exit(1)
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) < 4:
|
|
print_usage_and_exit()
|
|
|
|
port = int(sys.argv[1])
|
|
cert_file = sys.argv[2]
|
|
if sys.argv[3] == "true":
|
|
use_client_certificate_authentication = True
|
|
elif sys.argv[3] == "false":
|
|
use_client_certificate_authentication = False
|
|
else:
|
|
print_usage_and_exit()
|
|
|
|
print("Starting SSL Server")
|
|
print(" port: " + str(port))
|
|
print(" cert file: " + cert_file)
|
|
print(
|
|
" using client Certificate Authentication: "
|
|
+ ("yes" if use_client_certificate_authentication else "no")
|
|
)
|
|
start_server(port, cert_file, use_client_certificate_authentication)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|