707ab9ebda2e4f6c712bddc3bf58476e87a7e4f5
[goodfet] / 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