ask for PIN in GetAddress and GetPublicKey messages

trezor · Mar 30, 2015 · 7c6d2fe · 7c6d2fe
1 parent f5fb0c3
commit 7c6d2fe
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/firmware/fsm.c b/firmware/fsm.c
@@ -268,6 +268,11 @@ void fsm_msgGetPublicKey(GetPublicKey *msg)
 {
 	RESP_INIT(PublicKey);
 
+	if (!protectPin(true)) {
+		layoutHome();
+		return;
+	}
+
 	const HDNode *node = fsm_getDerivedNode(msg->address_n, msg->address_n_count);
 	if (!node) return;
 
@@ -500,6 +505,11 @@ void fsm_msgGetAddress(GetAddress *msg)
 {
 	RESP_INIT(Address);
 
+	if (!protectPin(true)) {
+		layoutHome();
+		return;
+	}
+
 	const CoinType *coin = fsm_getCoin(msg->coin_name);
 	if (!coin) return;
 	const HDNode *node = fsm_getDerivedNode(msg->address_n, msg->address_n_count);