More stuff in firmware main.c

God I am getting bad at writing these commit messages

1 files changed, 15 insertions, 0 deletions

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