git.rot13.org Git - goodfet/blob - client/GoodFETMSP430.py

   1 #!/usr/bin/env python
   2 # GoodFET Client Library
   3 #
   4 # (C) 2009 Travis Goodspeed <travis at radiantmachines.com>
   5 #
   6 # Presently being rewritten.
   7
   8 import sys, time, string, cStringIO, struct, glob, serial, os;
   9
  10 from GoodFET import GoodFET;
  11
  12 class GoodFETMSP430(GoodFET):
  13     MSP430APP=0x11;  #Changed by inheritors.
  14     CoreID=0;
  15     DeviceID=0;
  16     JTAGID=0;
  17     MSP430ident=0;
  18     def MSP430setup(self):
  19         """Move the FET into the MSP430 JTAG application."""
  20         self.writecmd(self.MSP430APP,0x10,0,None);
  21
  22     def MSP430stop(self):
  23         """Stop debugging."""
  24         self.writecmd(self.MSP430APP,0x21,0,self.data);
  25
  26     def MSP430coreid(self):
  27         """Get the Core ID."""
  28         self.writecmd(self.MSP430APP,0xF0);
  29         CoreID=ord(self.data[0])+(ord(self.data[1])<<8);
  30         return CoreID;
  31     def MSP430deviceid(self):
  32         """Get the Core ID."""
  33         self.writecmd(self.MSP430APP,0xF1);
  34         DeviceID=(
  35             ord(self.data[0])+(ord(self.data[1])<<8)+
  36             (ord(self.data[2])<<16)+(ord(self.data[3])<<24));
  37         return DeviceID;
  38     def MSP430peek(self,adr):
  39         """Read a word at an address."""
  40         self.data=[adr&0xff, (adr&0xff00)>>8,
  41                    (adr&0xff0000)>>16,(adr&0xff000000)>>24,
  42                    ];
  43         self.writecmd(self.MSP430APP,0x02,4,self.data,1);
  44         return ord(self.data[0])+(ord(self.data[1])<<8);
  45     def MSP430peekblock(self,adr,blocks=1):
  46         """Grab a few block from an SPI Flash ROM.  Block size is unknown"""
  47         data=[adr&0xff, (adr&0xff00)>>8,
  48                    (adr&0xff0000)>>16,(adr&0xff000000)>>24,
  49                    blocks];
  50
  51         self.writecmd(self.MSP430APP,0x02,5,data,blocks);
  52         return self.data;
  53
  54     def MSP430poke(self,adr,val):
  55         """Write the contents of memory at an address."""
  56         self.data=[adr&0xff, (adr&0xff00)>>8,
  57                    (adr&0xff0000)>>16,(adr&0xff000000)>>24,
  58                    val&0xff, (val&0xff00)>>8];
  59         self.writecmd(self.MSP430APP,0x03,6,self.data);
  60         return ord(self.data[0])+(ord(self.data[1])<<8);
  61     def MSP430start(self):
  62         """Start debugging."""
  63         self.writecmd(self.MSP430APP,0x20,0,self.data);
  64         self.JTAGID=ord(self.data[0]);
  65         #print "Identified as %02x." % self.JTAGID;
  66         if(not (self.JTAGID==0x89 or self.JTAGID==0x91)):
  67             print "Error, misidentified as %02x." % self.JTAGID;
  68
  69     def MSP430haltcpu(self):
  70         """Halt the CPU."""
  71         self.writecmd(self.MSP430APP,0xA0,0,self.data);
  72     def MSP430releasecpu(self):
  73         """Resume the CPU."""
  74         self.writecmd(self.MSP430APP,0xA1,0,self.data);
  75     def MSP430shiftir8(self,ins):
  76         """Shift the 8-bit Instruction Register."""
  77         data=[ins];
  78         self.writecmd(self.MSP430APP,0x80,1,data);
  79         return ord(self.data[0]);
  80     def MSP430shiftdr16(self,dat):
  81         """Shift the 16-bit Data Register."""
  82         data=[dat&0xFF,(dat&0xFF00)>>8];
  83         self.writecmd(self.MSP430APP,0x81,2,data);
  84         return ord(self.data[0])#+(ord(self.data[1])<<8);
  85     def MSP430setinstrfetch(self):
  86         """Set the instruction fetch mode."""
  87         self.writecmd(self.MSP430APP,0xC1,0,self.data);
  88         return self.data[0];
  89     def MSP430ident(self):
  90         """Grab self-identification word from 0x0FF0 as big endian."""
  91         ident=0x00;
  92         if(self.JTAGID==0x89):
  93             i=self.MSP430peek(0x0ff0);
  94             ident=((i&0xFF00)>>8)+((i&0xFF)<<8)
  95
  96         if(self.JTAGID==0x91):
  97             i=self.MSP430peek(0x1A04);
  98             ident=((i&0xFF00)>>8)+((i&0xFF)<<8)
  99             #ident=0x0091;
 100
 101         return ident;
 102     def MSP430test(self):
 103         """Test MSP430 JTAG.  Requires that a chip be attached."""
 104         if self.MSP430ident()==0xffff:
 105             print "Is anything connected?";
 106         print "Testing RAM from 1c00 to 1d00.";
 107         for a in range(0x1c00,0x1d00):
 108             self.MSP430poke(a,0);
 109             if(self.MSP430peek(a)!=0):
 110                 print "Fault at %06x" % a;
 111             self.MSP430poke(a,0xffff);
 112             if(self.MSP430peek(a)!=0xffff):
 113                 print "Fault at %06x" % a;
 114         print "RAM Test Complete."
 115         for a in range(1,5):
 116             print "Identity %04x" % self.MSP430ident();
 117
 118     def MSP430flashtest(self):
 119         self.MSP430masserase();
 120         i=0x2500;
 121         while(i<0xFFFF):
 122             if(self.MSP430peek(i)!=0xFFFF):
 123                 print "ERROR: Unerased flash at %04x."%i;
 124             self.MSP430writeflash(i,0xDEAD);
 125             i+=2;
 126     def MSP430masserase(self):
 127         """Erase MSP430 flash memory."""
 128         self.writecmd(self.MSP430APP,0xE3,0,None);
 129     def MSP430writeflash(self,adr,val):
 130         """Write a word of flash memory."""
 131         if(self.MSP430peek(adr)!=0xFFFF):
 132             print "FLASH ERROR: %04x not clear." % adr;
 133         data=[adr&0xFF,(adr&0xFF00)>>8,val&0xFF,(val&0xFF00)>>8];
 134         self.writecmd(self.MSP430APP,0xE1,4,data);
 135         rval=ord(self.data[0])+(ord(self.data[1])<<8);
 136         if(val!=rval):
 137             print "FLASH WRITE ERROR AT %04x.  Found %04x, wrote %04x." % (adr,rval,val);
 138
 139     def MSP430dumpbsl(self):
 140         self.MSP430dumpmem(0xC00,0xfff);
 141     def MSP430dumpallmem(self):
 142         self.MSP430dumpmem(0x200,0xffff);
 143     def MSP430dumpmem(self,begin,end):
 144         i=begin;
 145         while i<end:
 146             print "%04x %04x" % (i, self.MSP430peek(i));
 147             i+=2;