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
 115
 116 //! Handles an AVR command.
 117 void avrhandle(unsigned char app,
 118                unsigned char verb,
 119                unsigned long len){
 120   unsigned long i;
 121   static u8 connected=0;
 122
 123   if(!avr_isready() && connected)
 124     debugstr("AVR is not yet ready.");
 125
 126   switch(verb){
 127   case READ:
 128   case WRITE:
 129     for(i=0;i<len;i++)
 130       cmddata[i]=avrtrans8(cmddata[i]);
 131     txdata(app,verb,len);
 132     break;
 133   case SETUP:
 134     avrsetup();
 135     txdata(app,verb,0);
 136     break;
 137   case START://returns device code
 138     avrconnect();
 139     //no break here
 140   case AVR_PEEKSIG:
 141     for(i=0;i<4;i++)
 142       cmddata[i]=avr_sig(i);
 143     txdata(app,verb,4);
 144     break;
 145   case AVR_ERASE:
 146     avr_erase();
 147     txdata(app,verb,0);
 148     break;
 149   case AVR_PEEKLOCK:
 150     cmddata[0]=avr_lockbits();
 151     txdata(app,verb,1);
 152     break;
 153
 154   case AVR_POKEEEPROM:
 155     avr_pokeeeprom(cmddataword[0], cmddata[2]);
 156     //no break here.
 157   case AVR_PEEKEEPROM:
 158     cmddata[0]=avr_peekeeprom(cmddataword[0]);
 159     txdata(app,verb,1);
 160     break;
 161   case PEEK:
 162   case POKE:
 163   default:
 164     debugstr("Verb unimplemented in AVR application.");
 165     txdata(app,NOK,0);
 166     break;
 167   }
 168 }
 169