git.rot13.org Git - goodfet/blob - firmware/apps/avr/avr.c

   1 /*! \file avr.c
   2   \author Travis Goodspeed
   3   \brief AVR SPI Programmer
   4 */
   5
   6 #include "platform.h"
   7 #include "command.h"
   8
   9 #include <signal.h>
  10 #include <io.h>
  11 #include <iomacros.h>
  12
  13 #include "avr.h"
  14
  15 //! Setup the AVR pins.
  16 void avrsetup(){
  17   spisetup();
  18 }
  19
  20 //! Initialized an attached AVR.
  21 void avrconnect(){
  22   //set I/O pins
  23   avrsetup();
  24
  25   //Pulse !RST (SS) at least twice while CLK is low.
  26   CLRCLK;
  27   CLRSS;
  28
  29   SETSS;
  30   CLRCLK;
  31   delay(500);
  32   CLRSS;
  33   delay(500);
  34
  35   //Enable programming
  36   avr_prgen();
  37 }
  38
  39 //! Read and write an SPI byte with delays.
  40 unsigned char avrtrans8(unsigned char byte){
  41   register unsigned int bit;
  42   //This function came from the SPI Wikipedia article.
  43   //Minor alterations.
  44
  45   for (bit = 0; bit < 8; bit++) {
  46     /* write MOSI on trailing edge of previous clock */
  47     if (byte & 0x80)
  48       SETMOSI;
  49     else
  50       CLRMOSI;
  51     byte <<= 1;
  52
  53     delay(2);
  54     SETCLK;
  55
  56     /* read MISO on trailing edge */
  57     byte |= READMISO;
  58     delay(2);
  59     CLRCLK;
  60   }
  61
  62   return byte;
  63 }
  64
  65 //! Perform a 4-byte exchange.
  66 u8 avrexchange(u8 a, u8 b, u8 c, u8 d){
  67   avrtrans8(a);
  68   avrtrans8(b);
  69   if(avrtrans8(c)!=b){
  70     debugstr("AVR sync error, b not returned as c.");
  71     //Reconnect here?
  72   }
  73   return avrtrans8(d);
  74 }
  75
  76 //! Enable AVR programming mode.
  77 void avr_prgen(){
  78   avrexchange(0xAC, 0x53, 0, 0);
  79 }
  80
  81 //! Is the AVR ready or busy?
  82 u8 avr_isready(){
  83   return avrexchange(0xF0, 0, 0, 0);
  84 }
  85
  86 //! Read AVR device code.
  87 u8 avr_sig(u8 i){
  88   return avrexchange(0x30, //Read signature byte
  89               0x00,
  90               i&0x03,      //sig adr
  91               0x00         //don't care.
  92               );
  93 }
  94
  95 //! Erase an AVR device
  96 void avr_erase(){
  97   avrexchange(0xAC, 0x80, 0, 0);
  98 }
  99
 100 //! Read lock bits.
 101 u8 avr_lockbits(){
 102   return avrexchange(0x58, 0, 0, 0);
 103 }
 104
 105 //! Read a byte of EEPROM.
 106 u8 avr_peekeeprom(u16 adr){
 107   return avrexchange(0xA0, adr>>8, adr&0xFF, 0);
 108 }
 109 //! Read a byte of EEPROM.
 110 u8 avr_pokeeeprom(u16 adr, u8 val){
 111   return avrexchange(0xC0, adr>>8, adr&0xFF, val);
 112 }
 113
 114 //! Handles an AVR command.
 115 void avrhandle(unsigned char app,
 116                unsigned char verb,
 117                unsigned long len){
 118   unsigned long i;
 119   static u8 connected=0;
 120
 121   if(!avr_isready() && connected)
 122     debugstr("AVR is not yet ready.");
 123
 124   switch(verb){
 125   case READ:
 126   case WRITE:
 127     for(i=0;i<len;i++)
 128       cmddata[i]=avrtrans8(cmddata[i]);
 129     txdata(app,verb,len);
 130     break;
 131   case SETUP:
 132     avrsetup();
 133     txdata(app,verb,0);
 134     break;
 135   case START://returns device code
 136     avrconnect();
 137     //no break here
 138   case AVR_PEEKSIG:
 139     for(i=0;i<4;i++)
 140       cmddata[i]=avr_sig(i);
 141     txdata(app,verb,4);
 142     break;
 143   case AVR_ERASE:
 144     avr_erase();
 145     txdata(app,verb,0);
 146     break;
 147   case AVR_PEEKLOCK:
 148     cmddata[0]=avr_lockbits();
 149     txdata(app,verb,1);
 150     break;
 151
 152   case AVR_POKEEEPROM:
 153     avr_pokeeeprom(cmddataword[0], cmddata[2]);
 154     //no break here.
 155   case AVR_PEEKEEPROM:
 156     cmddata[0]=avr_peekeeprom(cmddataword[0]);
 157     txdata(app,verb,1);
 158     break;
 159   case PEEK:
 160   case POKE:
 161   default:
 162     debugstr("Verb unimplemented in AVR application.");
 163     txdata(app,NOK,0);
 164     break;
 165   }
 166 }
 167