diff --git a/doorlockd-new/doorlockd.py b/doorlockd-new/doorlockd.py new file mode 100755 index 0000000..d1e1591 --- /dev/null +++ b/doorlockd-new/doorlockd.py @@ -0,0 +1,280 @@ +#!/usr/bin/env python3 + +import argparse +import logging +import sys + +from enum import Enum +from threading import Thread +from time import sleep +from random import sample +from serial import Serial + +from flask import Flask, render_template, request, Markup +from flask_bootstrap import Bootstrap +from flask_socketio import SocketIO +from flask_wtf import FlaskForm + +from wtforms import PasswordField, StringField, SubmitField +from wtforms.validators import DataRequired, Length + +__author__ = 'Ralf Ramsauer' +__copyright = 'Copyright (c) Ralf Ramsauer, 2018' +__license__ = 'GPLv2' +__email__ = 'ralf@binary-kitchen.de' +__status__ = 'Development' +__maintainer__ = 'Ralf Ramsauer' +__version__ = '0.01a' + +log_level = logging.DEBUG +date_fmt = '%Y-%m-%d %H:%M:%S' +log_fmt = '%(asctime)-15s %(levelname)-8s %(message)s' +log = logging.getLogger() + +default_serial = '/dev/ttyS0' +default_ldap_uri = 'ldaps://ldap1.binary.kitchen/ ' \ + 'ldaps://ldap2.binary.kitchen/ ' \ + 'ldaps://ldapm.binary.kitchen/' +default_binddn = 'cn=%s,ou=people,dc=binary-kitchen,dc=de' + +html_title = 'Binary Kitchen Doorlock (%s - v%s)' % (__status__, __version__) + +webapp = Flask(__name__) +webapp.config['SECRET_KEY'] = 'foobar' +socketio = SocketIO(webapp, async_mode=None) +Bootstrap(webapp) + +# copied from sudo +eperm_insults = { + 'Wrong! You cheating scum!', + 'And you call yourself a Rocket Scientist!', + 'No soap, honkie-lips.', + 'Where did you learn to type?', + 'Are you on drugs?', + 'My pet ferret can type better than you!', + 'You type like i drive.', + 'Do you think like you type?', + 'Your mind just hasn\'t been the same since the electro-shock, has it?', + 'Maybe if you used more than just two fingers...', + 'BOB says: You seem to have forgotten your passwd, enter another!', + 'stty: unknown mode: doofus', + 'I can\'t hear you -- I\'m using the scrambler.', + 'The more you drive -- the dumber you get.', + 'Listen, broccoli brains, I don\'t have time to listen to this trash.', + 'I\'ve seen penguins that can type better than that.', + 'Have you considered trying to match wits with a rutabaga?', + 'You speak an infinite deal of nothing', +} + + +def choose_insult(): + return(sample(eperm_insults, 1)[0]) + + +class AuthMethod(Enum): + LDAP_USER_PW = 1 + + +class DoorState(Enum): + Close = 1 + Open = 2 + + def to_img(self): + led = 'red' + if self == DoorState.Open: + led = 'green' + return '' % led + + def to_html(self): + if self == DoorState.Open: + return 'Offen' + return 'Zu' + + +class LogicResponse(Enum): + Success = 0 + Perm = 1 + AlreadyLocked = 2 + AlreadyOpen = 3 + Inval = 4 + LDAP = 5 + + def to_html(self): + if self == LogicResponse.Success: + return 'Yo, passt.' + elif self == LogicResponse.Perm: + return choose_insult() + elif self == LogicResponse.AlreadyLocked: + return 'Narf. Schon zu.' + elif self == LogicResponse.AlreadyOpen: + return 'Schon offen, treten Sie ein!' + elif self == LogicResponse.Inval: + return 'Das was du willst geht nicht.' + elif self == LogicResponse.LDAP: + return 'Moep! Geh LDAP fixen!' + + return 'Bitte spezifizieren Sie.' + + +class DoorHandler: + state = DoorState.Close + + def __init__(self, device): + self.serial = Serial(device, baudrate=9600, bytesize=8, parity='N', + stopbits=1, timeout=1) + self.thread = Thread(target=self.thread_worker) + self.thread.start() + + def send_command(self, cmd): + print('sending cmd %c' % cmd) + + def clear_buffer(self): + print('clearing buffer...') + + def thread_worker(self): + while True: + self.clear_buffer() + + if self.state == DoorState.Open: + self.send_command('u') + elif self.state == DoorState.Close: + self.send_command('l') + sleep(1) + + def open(self): + if self.state == DoorState.Open: + return LogicResponse.AlreadyOpen + + self.state = DoorState.Open + return LogicResponse.Success + + def close(self): + if self.state == DoorState.Close: + return LogicResponse.AlreadyLocked + + self.state = DoorState.Close + return LogicResponse.Success + + def request(self, state): + if state == DoorState.Close: + return self.close() + elif state == DoorState.Open: + return self.open() + + +class Logic: + def __init__(self, device): + self.door_handler = DoorHandler(device) + + def _try_auth_ldap(self, user, password): + log.info('Trying to LDAP auth (user, password) as user %s', user) + return LogicResponse.Success + return LogicResponse.LDAP + + def try_auth(self, credentials): + method = credentials[0] + if method == AuthMethod.LDAP_USER_PW: + return self._try_auth_ldap(credentials[1], credentials[2]) + + return LogicResponse.Inval + + def _request(self, state, credentials): + err = self.try_auth(credentials) + if err != LogicResponse.Success: + return err + return self.door_handler.request(state) + + def request(self, state, credentials): + err = self._request(state, credentials) + return err + + @property + def state(self): + return self.door_handler.state + + +class AuthenticationForm(FlaskForm): + username = StringField('Username', [Length(min=4, max=25)]) + password = PasswordField('Password', [DataRequired()]) + open = SubmitField('Open') + close = SubmitField('Close') + + def __init__(self, *args, **kwargs): + FlaskForm.__init__(self, *args, **kwargs) + self.desired_state = DoorState.Close + + def validate(self): + if not FlaskForm.validate(self): + return False + + if self.open.data: + self.desired_state = DoorState.Open + + return True + + +def emit_status(message=None): + led = logic.state.to_img() + if message is None: + message = logic.state.to_html() + else: + message = message.to_html() + + socketio.emit('status', {'led': led, 'message': message}) + + +@socketio.on('connect') +def on_connect(): + emit_status() + + +@webapp.route('/display') +def display(): + return render_template('display.html') + + +@webapp.route('/', methods=['GET', 'POST']) +def home(): + authentication_form = AuthenticationForm() + response = None + + if request.method == 'POST' and authentication_form.validate(): + user = authentication_form.username.data + password = authentication_form.password.data + credentials = AuthMethod.LDAP_USER_PW, user, password + + log.info('Incoming request from %s' % user) + desired_state = authentication_form.desired_state + log.info(' desired state: %s' % desired_state) + log.info(' current state: %s' % logic.state) + response = logic.request(desired_state, credentials) + log.info(' response: %s' % response) + + # Don't trust python, zero credentials + user = password = credentials = None + + emit_status(response) + + return render_template('index.html', + authentication_form=authentication_form, + response=response, + state_text=logic.state.to_html(), + state_img=Markup(logic.state.to_img()), + title=html_title) + + +if __name__ == '__main__': + parser = argparse.ArgumentParser('doorlockd', 'Binary Kitchen doorlockd') + parser.add_argument('-s', '--serial', default=default_serial, type=str) + args = parser.parse_args() + + logging.basicConfig(level=log_level, stream=sys.stdout, + format=log_fmt, datefmt=date_fmt) + log.info('Starting doorlockd') + log.info('Using serial port: %s' % args.serial) + + logic = Logic(args.serial) + + socketio.run(webapp, port=8080) + + sys.exit(0) diff --git a/doorlockd-new/static/led-green.png b/doorlockd-new/static/led-green.png new file mode 100644 index 0000000..34e0080 Binary files /dev/null and b/doorlockd-new/static/led-green.png differ diff --git a/doorlockd-new/static/led-red.png b/doorlockd-new/static/led-red.png new file mode 100644 index 0000000..7d6bb6c Binary files /dev/null and b/doorlockd-new/static/led-red.png differ diff --git a/doorlockd-new/static/logo.svg b/doorlockd-new/static/logo.svg new file mode 100644 index 0000000..a70b25c --- /dev/null +++ b/doorlockd-new/static/logo.svg @@ -0,0 +1,81 @@ + + + + diff --git a/doorlockd-new/templates/display.html b/doorlockd-new/templates/display.html new file mode 100644 index 0000000..30479ea --- /dev/null +++ b/doorlockd-new/templates/display.html @@ -0,0 +1,25 @@ +{% extends "layout.html" %} + +{% block scripts %} + {{ super() }} + + +{% endblock %} + +{% block content %} +