summaryrefslogtreecommitdiffstats
path: root/lock/fend.c
blob: 39854c60bef4198ba97186e99550b9e02e7fe8b7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
/* 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 "nrfm.h"
#include "uart.h"
#include "util.h"

#define LOCK_PIN      PD2
#define UNLOCK_PIN    PD3

#define RX_PIN        PD7
#define RX_DDR        DDRD
#define RX_PORT       PORTD
#define RX_PCIE       PCIE2
#define RX_PCINT      PCINT23
#define RX_PCMSK      PCMSK2
#define RX_PCINTVEC   PCINT2_vect

#define VCC_MIN       4000

#define LOCK_LED      PC3
#define UNLOCK_LED    PC4
#define BATLOW_LED    PC5
#define LED_DDR       DDRC
#define LED_PORT      PORTC

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_PIN);
	RX_PORT |= (1 << RX_PIN); 
	PCICR |= (1 << RX_PCIE);
	RX_PCMSK |= (1 << RX_PCINT);
}

static inline void init_btns(void)
{
	DDRD &= ~((1 << LOCK_PIN) | (1 << UNLOCK_PIN));
	PORTD |= ((1 << LOCK_PIN) | (1 << UNLOCK_PIN));

	EICRA = 0b00000000;
	EIMSK = (1 << INT0) | (1 << INT1);
}

static inline void init_leds(void) 
{
	LED_DDR |= (1 << LOCK_LED) | (1 << UNLOCK_LED);
	LED_DDR |= (1 << BATLOW_LED);

	LED_PORT &= ~(1 << LOCK_LED);
	LED_PORT &= ~(1 << UNLOCK_LED);
	LED_PORT &= ~(1 << BATLOW_LED);
}

int main(void)
{
	int i;
	uint8_t rxaddr[ADDRLEN] = { 194, 178, 82 };
	uint8_t txaddr[ADDRLEN] = { 194, 178, 83 };

	char buf[WDLEN], key[WDLEN];

	wdt_off();
	init_rx();
	init_leds();
	init_btns();

	uart_init();
	radio_init(rxaddr);
	radio_print_config();

	sei();
	radio_listen();

	for (;;) {
		if (!sync && (islock || isunlock)) {
			xor(KEY, SYN, buf, WDLEN);
			do {
				sync = radio_sendto(txaddr, buf, WDLEN);
				_delay_ms(50);
			} while (!sync);
		}

		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);
				} else if (isunlock) {
					isunlock = 0;
					xor(key, UNLOCK, buf, WDLEN);
				}
				radio_sendto(txaddr, buf, WDLEN);
			}
		}

		if (!sync) {
			if (getvcc() < VCC_MIN) {
				for (i = 0; i < 5; i++) {
					LED_PORT ^= (1 << BATLOW_LED);
					_delay_ms(100);
				}
			}

			radio_pwr_dwn();
			sleep_enable();	
			sleep_bod_disable();
			sleep_cpu();
			sleep_disable();
			radio_listen();
		}
	}
	return 0;
}

ISR(RX_PCINTVEC)
{
	rxd = 1;
}

ISR(INT0_vect)
{
	if (is_btn_pressed(PIND, LOCK_PIN)) {
		sync = 0;
		islock = 1;
	}
}

ISR(INT1_vect)
{
	if (is_btn_pressed(PIND, UNLOCK_PIN)) {
		sync = 0;
		isunlock = 1;
	}
}