From df068274f4a383ce6c944168ce41f859f3bc22c9 Mon Sep 17 00:00:00 2001
From: Zach Eveland <zeveland@blacklabel-development.com>
Date: Tue, 30 Aug 2011 17:25:35 -0400
Subject: on a CDC or HID write() error, call setWriteError().  better handling
 of USB_Send errors in CDC.

---
 cores/arduino/CDC.cpp | 10 ++++++++--
 cores/arduino/HID.cpp |  8 ++++++--
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/cores/arduino/CDC.cpp b/cores/arduino/CDC.cpp
index b229cb5..0d04b9c 100644
--- a/cores/arduino/CDC.cpp
+++ b/cores/arduino/CDC.cpp
@@ -166,9 +166,15 @@ size_t Serial_::write(uint8_t c)
 	// open connection isn't broken cleanly (cable is yanked out, host dies
 	// or locks up, or host virtual serial port hangs)
 	if (_usbLineInfo.lineState > 0)	{
-		USB_Send(CDC_TX,&c,1);
-		return 1;
+		int r = USB_Send(CDC_TX,&c,1);
+		if (r > 0) {
+			return r;
+		} else {
+			setWriteError();
+			return 0;
+		}
 	}
+	setWriteError();
 	return 0;
 }
 
diff --git a/cores/arduino/HID.cpp b/cores/arduino/HID.cpp
index cea90fc..e3aadb5 100644
--- a/cores/arduino/HID.cpp
+++ b/cores/arduino/HID.cpp
@@ -398,11 +398,15 @@ size_t Keyboard_::write(uint8_t c)
 			_keyMap->charToKey(c,&keys);
 		else
 		{
-			if (c >= 128)
+			if (c >= 128) {
+				setWriteError();
 				return 0;
+			}
 			c = pgm_read_byte(_asciimap + c);
-			if (!c)
+			if (!c) {
+				setWriteError();
 				return 0;
+			}
 			if (c & 0x80)
 			{
 				keys.modifiers |= KEY_MODIFIER_LEFT_SHIFT;
-- 
cgit v1.2.3-18-g5258