use eeprom for pin

2 files changed, 14 insertions, 4 deletions

diff --git a/Makefile b/Makefile
index 581165c..3f0732f 100644
--- a/Makefile
+++ b/Makefile
@@ -15,7 +15,7 @@ CC=gcc
 CXX=g++ -g
 LDXX=g++ -g
 
-love: kbc.flash
+love: kbc.flash kbc.eeprom
 
 %.flash: %.ld.o
 	$(AOBJCOPY) -O binary $< $@
diff --git a/kbc.c b/kbc.c
index 6a04e4a..81ba569 100644
--- a/kbc.c
+++ b/kbc.c
@@ -5,6 +5,7 @@
 #include <avr/interrupt.h>
 #include <avr/sleep.h>
 #include <avr/pgmspace.h>
+#include <avr/eeprom.h>
 #include <util/delay.h>
 
 #ifndef PIN
@@ -331,7 +332,7 @@ static uint8_t cntr = 0;
 static uint8_t error;
 #endif
 
-static const char passwd[sizeof(PIN) - 1] = PIN;
+static const EEMEM char passwd[sizeof(PIN) - 1] = PIN;
 static char code[sizeof(PIN)];
 
 static void state_enter(void)
@@ -543,17 +544,26 @@ static void handle_keypress(uint8_t data)
 	unlock = ascii == ENT || data == NUMPAD_ENTER;
 
 	if (lock || unlock) {
+		uint8_t eebyte, pos, ok = 1;
+
 		/* passwd:  a  b  c  d
 		 * code:   \0  a  b  c  d
 		 */
+
+		if (code[0])
+			ok = 0;
 #ifdef KILLSWITCH
-		if (code[0] || !memcmp("9164", code + 1, 4)) {
+		if (!memcmp("9164", code + 1, 4)) {
 			error = 0;
 			nextstate = STATE_ERROR;
 			return;
 		}
 #endif
-		if (code[0] || memcmp(passwd, code + 1, sizeof(passwd))) {
+		for (pos = 1; pos < sizeof(code); pos++)
+			if (code[pos] != eeprom_read_byte(passwd + pos - 1))
+				ok = 0;
+
+		if (!ok) {
 			dbg_wr(0x20);
 			nextstate = STATE_REJECT;
 		} else {