From bad9b58ce3e3770e12b9053c670b22c95aba8b45 Mon Sep 17 00:00:00 2001
From: Cristian Maglie <c.maglie@arduino.cc>
Date: Thu, 1 Oct 2015 12:35:24 +0200
Subject: [PUSB] Fixed checks on return values

---
 libraries/HID/HID.cpp | 20 ++++++++++++--------
 libraries/HID/HID.h   |  4 ++--
 2 files changed, 14 insertions(+), 10 deletions(-)

(limited to 'libraries')

diff --git a/libraries/HID/HID.cpp b/libraries/HID/HID.cpp
index a871415..a109574 100644
--- a/libraries/HID/HID.cpp
+++ b/libraries/HID/HID.cpp
@@ -35,19 +35,23 @@ int HID_::getInterface(uint8_t* interfaceNum)
 	return USB_SendControl(0, &hidInterface, sizeof(hidInterface));
 }
 
-int HID_::getDescriptor(int8_t t)
+int HID_::getDescriptor(int8_t type)
 {
-	if (HID_REPORT_DESCRIPTOR_TYPE == t) {
+	if (HID_REPORT_DESCRIPTOR_TYPE == type) {
 		HIDDescriptorListNode* current = rootNode;
 		int total = 0;
-		while(current != NULL) {
-			total += USB_SendControl(TRANSFER_PGM,current->data,current->length);
+		while (current != NULL) {
+			int res = USB_SendControl(TRANSFER_PGM, current->data, current->length);
+			if (res == -1)
+				return -1;
+			total += res;
 			current = current->next;
 		}
 		return total;
-	} else {
-		return 0;
 	}
+
+	// Ignored
+	return 0;
 }
 
 void HID_::AppendDescriptor(HIDDescriptorListNode *node)
@@ -71,9 +75,9 @@ void HID_::SendReport(uint8_t id, const void* data, int len)
 	USB_Send(endpoint() | TRANSFER_RELEASE,data,len);
 }
 
-bool HID_::setup(USBSetup& setup, uint8_t i)
+bool HID_::setup(USBSetup& setup, uint8_t interfaceNum)
 {
-	if (interface() != i) {
+	if (interface() != interfaceNum) {
 		return false;
 	} else {
 		uint8_t r = setup.bRequest;
diff --git a/libraries/HID/HID.h b/libraries/HID/HID.h
index f951229..ed08cf0 100644
--- a/libraries/HID/HID.h
+++ b/libraries/HID/HID.h
@@ -83,8 +83,8 @@ public:
 protected:
   // Implementation of the PUSBListNode
   int getInterface(uint8_t* interfaceNum);
-  int getDescriptor(int8_t t);
-  bool setup(USBSetup& setup, uint8_t i);
+  int getDescriptor(int8_t type);
+  bool setup(USBSetup& setup, uint8_t interfaceNum);
 
 private:
   HIDDescriptor hidInterface;
-- 
cgit v1.2.3-18-g5258