From bf678ec404a41c31c29f44ef821f9b69c44f4e61 Mon Sep 17 00:00:00 2001
From: Jenna Fox <a@creativepony.com>
Date: Sun, 23 Sep 2012 18:22:23 +1000
Subject: More stuff in firmware main.c

God I am getting bad at writing these commit messages
---
 firmware/main.c | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/firmware/main.c b/firmware/main.c
index 6518d9d..ae32cf7 100644
--- a/firmware/main.c
+++ b/firmware/main.c
@@ -442,6 +442,21 @@ uchar usbFunctionWrite(uchar *data, uchar length) {
     }
     
     do {
+        // remember vectors or the tinyvector table 
+        if (currentAddress == RESET_VECTOR_OFFSET * 2) {
+            vectorTemp[0] = *(short *)data;
+        }
+        
+        if (currentAddress == USBPLUS_VECTOR_OFFSET * 2) {
+            vectorTemp[1] = *(short *)data;
+        }
+        
+        // make sure we don't write over the bootloader!
+        if (currentAddress >= BOOTLOADER_ADDRESS - 6) {
+            __boot_page_fill_clear();
+            return isLast;
+        }
+        
         writeWordToPageBuffer(*(uint16_t *) data);
         currentAddress += 2; // advance progmem address
         data += 2; // advance data pointer
-- 
cgit v1.2.3