From f55c677ba46a8243f077195ef4f39cde1babf560 Mon Sep 17 00:00:00 2001
From: tmk <nobody@nowhere>
Date: Sun, 10 Mar 2013 15:05:28 +0900
Subject: [PATCH] Fix watchdog in bootloader jump

- disable watchdog after watchdog reset
- clear WDRF after watchdog reset
---
 common/bootloader.c          | 27 +++++++++++++++------------
 keyboard/hhkb/Makefile.iwrap |  9 +++++++++
 keyboard/hhkb/Makefile.lufa  | 11 ++++++-----
 3 files changed, 30 insertions(+), 17 deletions(-)

diff --git a/common/bootloader.c b/common/bootloader.c
index 77fa1b30..02322041 100644
--- a/common/bootloader.c
+++ b/common/bootloader.c
@@ -11,15 +11,16 @@
 #endif
 
 
-/* Boot Section Size in bytes
- * Teensy halfKay   512
- * Atmel DFU loader 4096
- * LUFA bootloader  4096
+/* Boot Section Size in *BYTEs*
+ * Teensy   halfKay     512
+ * Teensy++ halfKay     1024
+ * Atmel DFU loader     4096
+ * LUFA bootloader      4096
  */
 #ifndef BOOT_SIZE
 #define BOOT_SIZE 512
 #endif
-#define FLASH_SIZE          (FLASHEND + 1)
+#define FLASH_SIZE          (FLASHEND + 1L)
 #define BOOTLOADER_START    (FLASH_SIZE - BOOT_SIZE)
 
 
@@ -58,13 +59,15 @@ void bootloader_jump_after_watchdog_reset(void) __attribute__ ((used, naked, sec
 void bootloader_jump_after_watchdog_reset(void)
 {
     if ((MCUSR & (1<<WDRF)) && reset_key == BOOTLOADER_RESET_KEY) {
-
-        #if defined(__AVR_ATmega168__) || defined(__AVR_ATmega168P__) || defined(__AVR_ATmega328P__)
-        // This makes custom USBasploader come up.
-        MCUSR = 0;
-        #endif
-
         reset_key = 0;
+
+        // My custom USBasploader requires this to come up.
+        MCUSR = 0;
+
+        // Seems like Teensy halfkay loader requires clearing WDRF and disabling watchdog.
+        MCUSR &= ~(1<<WDRF);
+        wdt_disable();
+
         ((void (*)(void))BOOTLOADER_START)();
     }
 }
@@ -138,4 +141,4 @@ void bootloader_jump(void) {
     // start Bootloader
     ((void (*)(void))BOOTLOADER_START)();
 }
-#endif     
+#endif
diff --git a/keyboard/hhkb/Makefile.iwrap b/keyboard/hhkb/Makefile.iwrap
index f87df0d2..3ec1a316 100644
--- a/keyboard/hhkb/Makefile.iwrap
+++ b/keyboard/hhkb/Makefile.iwrap
@@ -49,6 +49,15 @@ EXTRAKEY_ENABLE = yes	# Audio control and System control
 #NKRO_ENABLE = yes	# USB Nkey Rollover
 
 
+# Boot Section Size in *BYTEs*
+#   Teensy   halfKay    512
+#   Teensy++ halfKay    1024
+#   Atmel DFU loader    4096
+#   LUFA bootloader     4096
+#   USBaspLoader        2048
+OPT_DEFS += -DBOOT_SIZE=2048
+
+
 
 #---------------- Programming Options --------------------------
 AVRDUDE = avrdude
diff --git a/keyboard/hhkb/Makefile.lufa b/keyboard/hhkb/Makefile.lufa
index afd413d0..75c0310d 100644
--- a/keyboard/hhkb/Makefile.lufa
+++ b/keyboard/hhkb/Makefile.lufa
@@ -107,11 +107,12 @@ CONSOLE_ENABLE = yes	# Console for debug
 #PS2_MOUSE_ENABLE = yes	# PS/2 mouse(TrackPoint) support
 
 
-# Boot Section Size in bytes
-#   Teensy halfKay   512
-#   Atmel DFU loader 4096
-#   LUFA bootloader  4096
-#OPT_DEFS += -DBOOT_SIZE=4096
+# Boot Section Size in *BYTEs*
+#   Teensy   halfKay    512
+#   Teensy++ halfKay    1024
+#   Atmel DFU loader    4096
+#   LUFA bootloader     4096
+OPT_DEFS += -DBOOT_SIZE=1024
 
 
 # Search Path