forked from berhsi/doorstatus
eb000bff46
better english in comments. the command line arguments are now processed by argparse. reading vom cli removed. fix bug in exception handling.
81 lines
2.6 KiB
Python
Executable file
81 lines
2.6 KiB
Python
Executable file
#!/usr/bin/python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# file: setstatus.py
|
|
# date: 26.07.2019
|
|
# email: berhsi@web.de
|
|
|
|
# client, that connects to the statusserver at port 10001 to update the
|
|
# krautspace door status. allowed arguments are 0 or 1.
|
|
|
|
import argparse
|
|
import socket
|
|
import ssl
|
|
import sys
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Set door status of
|
|
Krautspace.")
|
|
parser.add_argument("status_code", help="status to set", type=int,
|
|
choices=(0, 1))
|
|
args = parser.parse_args()
|
|
print("Status set to {}".format(bytes([args.status_code])))
|
|
|
|
HOST = 'localhost'
|
|
PORT = 10001
|
|
SERVER_NAME = 'server.status.kraut.space'
|
|
CLIENT_CERT = './certs/client.crt'
|
|
CLIENT_KEY = './certs/client.key'
|
|
SERVER_CERT = './certs/server.crt'
|
|
STATUS = None
|
|
RESPONSE = None
|
|
|
|
context = ssl.create_default_context(ssl.Purpose.SERVER_AUTH,
|
|
cafile=SERVER_CERT)
|
|
context.set_ciphers('EECDH+AESGCM') # only ciphers for tls 1.2 and 1.3
|
|
context.options |= getattr(ssl._ssl, 'OP_NO_COMPRESSION', 0)
|
|
context.load_cert_chain(certfile=CLIENT_CERT, keyfile=CLIENT_KEY)
|
|
print('SSL context created')
|
|
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM, 0) as mySocket:
|
|
print('Socket created')
|
|
try:
|
|
conn = context.wrap_socket(mySocket, server_side=False,
|
|
server_hostname=SERVER_NAME)
|
|
print('Connection wrapped with ssl.context')
|
|
conn.settimeout(5.0)
|
|
except Exception as e:
|
|
print('Context wrapper failed: {}'.format(e))
|
|
try:
|
|
conn.connect((HOST, PORT))
|
|
print('Connection established: {}'.format(conn.getpeercert()))
|
|
except socket.timeout:
|
|
print('Connection timeout')
|
|
except Exception as e:
|
|
print('Connection failed: {}'.format(e))
|
|
sys.exit(1)
|
|
|
|
try:
|
|
print('Send new status: {}'.format(STATUS))
|
|
conn.send(STATUS)
|
|
except Exception as e:
|
|
print('Error: {}'.format(e))
|
|
sys.exit(2)
|
|
|
|
try:
|
|
RESPONSE = conn.recv(1)
|
|
print('Server returns: {}'.format(RESPONSE))
|
|
if RESPONSE == STATUS:
|
|
print('Status sucessfull updated')
|
|
else:
|
|
print('Failed to update status')
|
|
print('Disconnect from server')
|
|
except Exception as e:
|
|
print('Error: {}'.format(e))
|
|
sys.exit(3)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|