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 //! Read AVR device code.
  82 u8 avr_sig(u8 i){
  83   return avrexchange(0x30, //Read signature byte
  84               0x00,
  85               i&0x03,      //sig adr
  86               0x00         //don't care.
  87               );
  88 }
  89
  90 //! Handles an AVR command.
  91 void avrhandle(unsigned char app,
  92                unsigned char verb,
  93                unsigned long len){
  94   unsigned long i;
  95
  96
  97   switch(verb){
  98   case READ:
  99   case WRITE:
 100     for(i=0;i<len;i++)
 101       cmddata[i]=spitrans8(cmddata[i]);
 102     txdata(app,verb,len);
 103     break;
 104   case SETUP:
 105     avrsetup();
 106     txdata(app,verb,0);
 107     break;
 108   case START://returns device code
 109     avrconnect();
 110     //no break here
 111   case AVR_PEEKSIG:
 112     for(i=0;i<4;i++)
 113       cmddata[i]=avr_sig(i);
 114     txdata(app,verb,4);
 115     break;
 116   case PEEK:
 117   case POKE:
 118   default:
 119     debugstr("Verb unimplemented in AVR application.");
 120     txdata(app,NOK,0);
 121     break;
 122   }
 123 }
 124