Door lock with MOSFETs and and without RFM.HEADmaster

1 files changed, 198 insertions, 0 deletions

diff --git a/door_lock_rfm/client.c b/door_lock_rfm/client.c
new file mode 100644
index 0000000..91fe040
--- /dev/null
+++ b/door_lock_rfm/client.c
@@ -0,0 +1,198 @@
+/* Lock front, connected to the fingerprint scanner */
+
+#include <stdint.h>
+#include <stdlib.h>
+
+#include <avr/wdt.h>
+#include <avr/sleep.h>
+#include <avr/interrupt.h>
+#include <util/delay.h>
+
+#include "fpm.h"
+#include "nrfm.h"
+#include "util.h"
+
+#define LOCK_PIN      PD3
+#define UNLOCK_PIN    PD2
+#define ENROLL_PIN    PD4
+
+#define RX_IRQ_PIN    PC0
+#define RX_DDR        DDRC
+#define RX_PIN        PINC
+#define RX_ICR        PCICR 
+#define RX_IE         PCIE1
+#define RX_INT        PCINT8
+#define RX_MSK        PCMSK1
+#define RX_INTVEC     PCINT1_vect
+
+#define VCC_MIN       4000
+
+static volatile uint8_t rxd = 0;
+static volatile uint8_t sync = 0;
+static volatile uint8_t islock = 0;
+static volatile uint8_t isunlock = 0;
+
+static inline void wdt_off(void)
+{
+	cli();
+	wdt_reset();
+	MCUSR &= ~(1 << WDRF);
+	WDTCSR |= (1 << WDCE) | (1 << WDE);
+	WDTCSR = 0x00;
+}
+
+static inline void init_rx(void)
+{
+	RX_DDR &= ~(1 << RX_IRQ_PIN);
+	RX_ICR |= (1 << RX_IE);
+	RX_MSK |= (1 << RX_INT);
+}
+
+static inline void init_btns(void)
+{
+	DDRD &= ~((1 << LOCK_PIN) | (1 << UNLOCK_PIN) | (1 << ENROLL_PIN));
+	PORTD |= ((1 << LOCK_PIN) | (1 << UNLOCK_PIN) | (1 << ENROLL_PIN));
+
+	EICRA = 0b00000000;
+	EIMSK = (1 << INT0) | (1 << INT1);
+
+	PCICR |= (1 << PCIE2);
+	PCMSK2 |= (1 << PCINT20);
+}
+
+static inline void fpm_ok(void)
+{
+	fpm_led_on(BLUE);
+	_delay_ms(500);
+	fpm_led_off();
+}
+
+static inline void fpm_nok(void)
+{
+	fpm_led_on(RED);
+	_delay_ms(500);
+	fpm_led_off();
+}
+
+int main(void)
+{
+	int i, retries;
+	uint8_t rxaddr[ADDRLEN] = { 194, 178, 82 };
+	uint8_t txaddr[ADDRLEN] = { 194, 178, 83 };
+
+	char buf[WDLEN], key[WDLEN];
+
+	wdt_off();
+	led_init();
+
+	fpm_init();
+	if (fpm_get_count() == 0)
+		fpm_enroll();
+
+	init_rx();
+	init_btns();
+
+	radio_init(rxaddr);
+
+	sei();
+	radio_listen();
+
+	for (;;) {
+		if (!sync && (islock || isunlock)) {
+			if (isunlock) {
+				if (!fpm_match()) {
+					isunlock = 0;
+					fpm_nok();
+					continue;
+				}
+				else
+					fpm_ok();
+			}
+
+			xor(KEY, SYN, buf, WDLEN);
+			retries = 0;
+			do {
+				sync = radio_sendto(txaddr, buf, WDLEN);
+				retries++;
+			} while (!sync && retries < 2);
+			
+			if (!sync) {
+				islock = 0;
+				isunlock = 0;
+			}
+		}
+
+		if (rxd) {
+			radio_recv(buf, WDLEN);
+			rxd = 0;
+			if (sync && (islock || isunlock)) {
+				sync = 0;
+				xor(KEY, buf, key, WDLEN);
+				if (islock) {
+					islock = 0;
+					xor(key, LOCK, buf, WDLEN);
+					if (radio_sendto(txaddr, buf, WDLEN))
+						led_locked();
+				} else if (isunlock) {
+					isunlock = 0;
+					xor(key, UNLOCK, buf, WDLEN);
+					if (radio_sendto(txaddr, buf, WDLEN))
+						led_unlocked();
+				}
+			}
+		}
+
+		if (!sync) {
+			if (getvcc() < VCC_MIN) {
+				for (i = 0; i < 5; i++) {
+					led_bat();
+					_delay_ms(100);
+				}
+			}
+
+			radio_pwr_dwn();
+			sleep_bod_disable();
+			set_sleep_mode(SLEEP_MODE_PWR_DOWN);
+			sleep_mode();
+			radio_listen();
+		}
+	}
+	return 0;
+}
+
+ISR(RX_INTVEC)
+{
+	if (!(RX_PIN & (1 << RX_IRQ_PIN)))
+		rxd = 1;
+}
+
+ISR(INT0_vect)
+{
+	sync = 0;
+	isunlock = 1;
+}
+
+ISR(INT1_vect)
+{
+	if (is_btn_pressed(PIND, LOCK_PIN)) {
+		sync = 0;
+		islock = 1;
+	}
+}
+
+ISR(PCINT2_vect)
+{
+	uint16_t id;
+
+	if (is_btn_pressed(PIND, ENROLL_PIN)) {
+		id = fpm_match();
+		if (id == 1 || id == 2) {
+			fpm_ok();
+			_delay_ms(1000);
+			if (fpm_enroll())
+				fpm_ok();
+		} else
+			fpm_nok();
+	}
+}
+