444a41c93338947da170324f6e700089554fc9e8
[goodfet] / client / goodfet.cc
1 #!/usr/bin/env python
2
3 import sys;
4 import binascii;
5
6 from GoodFET import GoodFET;
7 from intelhex import IntelHex;
8
9
10
11 if(len(sys.argv)==1):
12     print "Usage: %s verb [objects]\n" % sys.argv[0];
13     print "%s test" % sys.argv[0];
14     print "%s dumpcode $foo.hex [0x$start 0x$stop]" % sys.argv[0];
15     print "%s erase" % sys.argv[0];
16     print "%s flash $foo.hex [0x$start 0x$stop]" % sys.argv[0];
17     print "%s verify $foo.hex [0x$start 0x$stop]" % sys.argv[0];
18     sys.exit();
19
20 #Initailize FET and set baud rate
21 client=GoodFET();
22 client.serInit("/dev/ttyUSB0")
23
24
25
26 #Connect to target
27 client.CCsetup();
28 client.CCstart();
29
30 if(sys.argv[1]=="test"):
31     client.CCtest();
32 if(sys.argv[1]=="dumpcode"):
33     f = sys.argv[2];
34     start=0x0000;
35     stop=0xFFFF;
36     if(len(sys.argv)>3):
37         start=int(sys.argv[3],16);
38     if(len(sys.argv)>4):
39         stop=int(sys.argv[4],16);
40     
41     print "Dumping code from %04x to %04x as %s." % (start,stop,f);
42     h = IntelHex(None);
43     i=start;
44     while i<=stop:
45         h[i>>1]=client.CCpeekcodebyte(i);
46         if(i%0x100==0):
47             print "Dumped %04x."%i;
48         i+=1;
49     h.write_hex_file(f);
50 if(sys.argv[1]=="dumpdata"):
51     f = sys.argv[2];
52     start=0xE000;
53     stop=0xFFFF;
54     if(len(sys.argv)>3):
55         start=int(sys.argv[3],16);
56     if(len(sys.argv)>4):
57         stop=int(sys.argv[4],16);
58     
59     print "Dumping data from %04x to %04x as %s." % (start,stop,f);
60     h = IntelHex(None);
61     i=start;
62     while i<=stop:
63         h[i]=client.CCpeekdatabyte(i);
64         if(i%0x100==0):
65             print "Dumped %04x."%i;
66         i+=1;
67     h.write_hex_file(f);
68 if(sys.argv[1]=="erase"):
69   print "Status: %s" % client.CCstatusstr();
70   client.CCchiperase();
71   print "Status: %s" %client.CCstatusstr();
72
73 if(sys.argv[1]=="flash"):
74     f=sys.argv[2];
75     start=0;
76     stop=0xFFFF;
77     if(len(sys.argv)>3):
78         start=int(sys.argv[3],16);
79     if(len(sys.argv)>4):
80         stop=int(sys.argv[4],16);
81     
82     h = IntelHex(f);
83     
84     client.MSP430masserase();
85     for i in h._buf.keys():
86         #print "%04x: %04x"%(i,h[i>>1]);
87         if(i>=start and i<=stop  and i&1==0):
88             client.MSP430writeflash(i,h[i>>1]);
89             if(i%0x100==0):
90                 print "%04x" % i;
91 if(sys.argv[1]=="writedata"):
92     f=sys.argv[2];
93     start=0;
94     stop=0xFFFF;
95     if(len(sys.argv)>3):
96         start=int(sys.argv[3],16);
97     if(len(sys.argv)>4):
98         stop=int(sys.argv[4],16);
99     
100     h = IntelHex(f);
101     
102     for i in h._buf.keys():
103         if(i>=start and i<=stop):
104             client.CCpokedatabyte(i,h[i]);
105             if(i%0x100==0):
106                 print "%04x" % i;
107 if(sys.argv[1]=="flashtest"):
108     client.MSP430flashtest();
109 if(sys.argv[1]=="verifycode"):
110     f=sys.argv[2];
111     start=0;
112     stop=0xFFFF;
113     if(len(sys.argv)>3):
114         start=int(sys.argv[3],16);
115     if(len(sys.argv)>4):
116         stop=int(sys.argv[4],16);
117     
118     h = IntelHex(f);
119     for i in h._buf.keys():
120         if(i>=start and i<=stop):
121             peek=client.MSP430peek(i)
122             if(h[i>>1]!=peek):
123                 print "ERROR at %04x, found %04x not %04x"%(i,peek,h[i>>1]);
124             if(i%0x100==0):
125                 print "%04x" % i;
126
127 client.CCstop();