X-Git-Url: http://git.rot13.org/?p=goodfet;a=blobdiff_plain;f=firmware%2Fapps%2Favr%2Favr.c;h=577d87847d32aec4e0550b2efa65248ee3f71df9;hp=a8d9d341b82672cf199673e73c56cf610dd61d20;hb=ccbb5f9f7ae6fee33ba360030ecdaa76e7967024;hpb=43d125b5bca16f0941450bee6ba7bd94ab41dc72

diff --git a/firmware/apps/avr/avr.c b/firmware/apps/avr/avr.c
index a8d9d34..577d878 100644
--- a/firmware/apps/avr/avr.c
+++ b/firmware/apps/avr/avr.c
@@ -11,6 +11,7 @@
 #include <iomacros.h>
 
 #include "avr.h"
+#include "glitch.h"
 
 //! Setup the AVR pins.
 void avrsetup(){
@@ -28,9 +29,9 @@ void avrconnect(){
 
   SETSS;
   CLRCLK;
-  delay(500);
+  delay(10);
   CLRSS;
-  delay(500);
+  delay(10);
   
   //Enable programming
   avr_prgen();
@@ -67,7 +68,7 @@ u8 avrexchange(u8 a, u8 b, u8 c, u8 d){
   avrtrans8(a);
   avrtrans8(b);
   if(avrtrans8(c)!=b){
-    debugstr("AVR sync error, b not returned as c.");
+    //debugstr("AVR sync error, b not returned as c.");
     //Reconnect here?
   }
   return avrtrans8(d);
@@ -101,6 +102,11 @@ void avr_erase(){
 u8 avr_lockbits(){
   return avrexchange(0x58, 0, 0, 0);
 }
+//! Write lock bits.
+void avr_setlock(u8 bits){
+  avrexchange(0xAC,0xE0,0x00,
+	      bits);
+}
 
 //! Read a byte of EEPROM.
 u8 avr_peekeeprom(u16 adr){
@@ -129,8 +135,10 @@ void avrhandle(unsigned char app,
   unsigned int at;
   static u8 connected=0;
   
+  /*
   if(!avr_isready() && connected)
     debugstr("AVR is not yet ready.");
+  */
   
   switch(verb){
   case READ:
@@ -159,7 +167,10 @@ void avrhandle(unsigned char app,
     cmddata[0]=avr_lockbits();
     txdata(app,verb,1);
     break;
-
+  case AVR_POKELOCK:
+    avr_setlock(cmddata[0]);
+    txdata(app,verb,0);
+    break;
   case AVR_POKEEEPROM:
     avr_pokeeeprom(cmddataword[0], cmddata[2]);
     //no break here.