From 731b52d816fc10c17cb703955c80a811a86e9b9c Mon Sep 17 00:00:00 2001
From: Michael Kuc <michaelkuc6@gmail.com>
Date: Sat, 22 Jun 2019 16:16:07 +0100
Subject: [PATCH] Fixed unidentified command. Fixed erroring.

---
 U2FMessage.cpp | 1 +
 U2F_CMD.cpp    | 4 +++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/U2FMessage.cpp b/U2FMessage.cpp
index dcb0b1b..b429d9c 100644
--- a/U2FMessage.cpp
+++ b/U2FMessage.cpp
@@ -184,4 +184,5 @@ void U2FMessage::error(const uint32_t tCID, const uint8_t tErr)
 	msg.cid = tCID;
 	msg.cmd = U2FHID_ERROR;
 	msg.data.push_back(tErr);
+	msg.write();
 }
diff --git a/U2F_CMD.cpp b/U2F_CMD.cpp
index 3de008a..7cdaf43 100644
--- a/U2F_CMD.cpp
+++ b/U2F_CMD.cpp
@@ -33,15 +33,17 @@ shared_ptr<U2F_CMD> U2F_CMD::get(const shared_ptr<U2FMessage> uMsg)
 			case U2FHID_PING:
 				return make_shared<U2F_Ping_CMD>(uMsg);
 			case U2FHID_MSG:
-					return U2F_Msg_CMD::generate(uMsg);
+				return U2F_Msg_CMD::generate(uMsg);
 			case U2FHID_INIT:
 				return make_shared<U2F_Init_CMD>(uMsg);
 			default:
+				U2FMessage::error(uMsg->cid, ERR_INVALID_CMD);
 				return {};
 		}
 	}
 	catch (runtime_error& ignored)
 	{
+		U2FMessage::error(uMsg->cid, ERR_OTHER);
 		return {};
 	}
 }