git.zerfleddert.de Git - proxmark3-svn/blob - linux/flasher.c

   1 #include <usb.h>
   2 #include <stdio.h>
   3 #include <unistd.h>
   4 #include <stdlib.h>
   5 #include <strings.h>
   6 #include <string.h>
   7 #include <errno.h>
   8 #include <ctype.h>
   9
  10 #include "translate.h"
  11 #include "../winsrc/prox.h"
  12 #include "proxmark3.h"
  13
  14 static DWORD ExpectedAddr;
  15 static BYTE QueuedToSend[256];
  16 static BOOL AllWritten;
  17
  18 static void FlushPrevious(void)
  19 {
  20         UsbCommand c;
  21         memset(&c, 0, sizeof(c));
  22
  23         printf("expected = %08x flush, ", ExpectedAddr);
  24
  25         int i;
  26         for(i = 0; i < 240; i += 48) {
  27                 c.cmd = CMD_SETUP_WRITE;
  28                 memcpy(c.d.asBytes, QueuedToSend+i, 48);
  29                 c.ext1 = (i/4);
  30                 SendCommand(&c, TRUE);
  31         }
  32
  33         c.cmd = CMD_FINISH_WRITE;
  34         c.ext1 = (ExpectedAddr-1) & (~255);
  35         printf("c.ext1 = %08x\r", c.ext1);
  36         memcpy(c.d.asBytes, QueuedToSend+240, 16);
  37         SendCommand(&c, TRUE);
  38
  39         AllWritten = TRUE;
  40 }
  41
  42 static void GotByte(DWORD where, BYTE which)
  43 {
  44         AllWritten = FALSE;
  45
  46         if(where != ExpectedAddr) {
  47                 printf("bad: got at %08x, expected at %08x\n", where, ExpectedAddr);
  48                 exit(-1);
  49         }
  50         QueuedToSend[where & 255] = which;
  51         ExpectedAddr++;
  52
  53         if((where & 255) == 255) {
  54                 // we have completed a full page
  55                 FlushPrevious();
  56         }
  57 }
  58
  59 static int HexVal(int c)
  60 {
  61         c = tolower(c);
  62         if(c >= '0' && c <= '9') {
  63                 return c - '0';
  64         } else if(c >= 'a' && c <= 'f') {
  65                 return (c - 'a') + 10;
  66         } else {
  67                 printf("bad hex digit '%c'\n", c);
  68                 exit(-1);
  69         }
  70 }
  71
  72 static BYTE HexByte(char *s)
  73 {
  74         return (HexVal(s[0]) << 4) | HexVal(s[1]);
  75 }
  76
  77 static void LoadFlashFromSRecords(char *file, int addr)
  78 {
  79         ExpectedAddr = addr;
  80
  81         FILE *f = fopen(file, "r");
  82         if(!f) {
  83                 printf("couldn't open file\n");
  84                 exit(-1);
  85         }
  86
  87         char line[512];
  88         while(fgets(line, sizeof(line), f)) {
  89                 if(memcmp(line, "S3", 2)==0) {
  90                         char *s = line + 2;
  91                         int len = HexByte(s) - 5;
  92                         s += 2;
  93
  94                         char addrStr[9];
  95                         memcpy(addrStr, s, 8);
  96                         addrStr[8] = '\0';
  97                         DWORD addr;
  98                         sscanf(addrStr, "%x", &addr);
  99                         s += 8;
 100
 101                         int i;
 102                         for(i = 0; i < len; i++) {
 103                                 while((addr+i) > ExpectedAddr) {
 104                                         GotByte(ExpectedAddr, 0xff);
 105                                 }
 106                                 GotByte(addr+i, HexByte(s));
 107                                 s += 2;
 108                         }
 109                 }
 110         }
 111
 112         if(!AllWritten) FlushPrevious();
 113
 114         fclose(f);
 115         printf("\ndone.\n");
 116 }
 117
 118 int main(int argc, char **argv) {
 119         unsigned int addr = 0;
 120         BOOL fastflash = 0, flashboth = 0;
 121         UsbCommand c;
 122
 123         if (argc != 3 && ! ((argc == 4 && (*argv[3] == 'f' || strcmp(argv[1], "both") == 0) ) || (argc == 5 && *argv[4] == 'f'))) {
 124                 fprintf(stderr,"Usage: %s {bootrom|os|fpga} image.s19 [f]ast\n", argv[0]);
 125                 fprintf(stderr,"       %s {both} osimage.s19 fpgaimage.s19 [f]ast\n", argv[0]);
 126                 exit(EXIT_FAILURE);
 127         }
 128
 129         if (!strcmp(argv[1],"bootrom")) {
 130                 addr = 0;
 131         } else if (!strcmp(argv[1],"os")) {
 132                 addr = FLASH_ADDR_OS;
 133         } else if (!strcmp(argv[1],"fpga")) {
 134                 addr = FLASH_ADDR_FPGA;
 135         } else if (!strcmp(argv[1],"both")) {
 136                 flashboth = 1;
 137         } else {
 138                 fprintf(stderr,"Unknown action '%s'!\n", argv[1]);
 139                 exit(EXIT_FAILURE);
 140         }
 141
 142         if((argc == 4 && *argv[3] == 'f') || (argc == 5 && *argv[4] == 'f')) {
 143                 fastflash= 1;
 144                 fprintf(stderr,"Fastflash - device already in FLASH mode...\n");
 145         }
 146
 147         usb_init();
 148
 149         fprintf(stderr,"Waiting for Proxmark to appear on USB...\n");
 150         while(!(devh=OpenProxmark(0))) { sleep(1); }
 151         fprintf(stderr,"Found...\n");
 152
 153         if(!fastflash){
 154                 fprintf(stderr,"Entering flash-mode...\n");
 155                 bzero(&c, sizeof(c));
 156                 c.cmd = CMD_START_FLASH;
 157                 SendCommand(&c, FALSE);
 158                 CloseProxmark();
 159                 sleep(1);
 160
 161                 fprintf(stderr,"Waiting for Proxmark to reappear on USB...\n");
 162                 fprintf(stderr,"(Press and hold down button NOW if your bootloader requires it)\n");
 163                 while(!(devh=OpenProxmark(0))) { sleep(1); }
 164                 fprintf(stderr,"Found...\n");
 165         }
 166
 167         if(!flashboth)
 168                 LoadFlashFromSRecords(argv[2], addr);
 169         else {
 170                 fprintf(stderr,"Flashing os...\n");
 171                 LoadFlashFromSRecords(argv[2],FLASH_ADDR_OS);
 172                 fprintf(stderr,"Flashing fpga...\n");
 173                 LoadFlashFromSRecords(argv[3],FLASH_ADDR_FPGA);
 174         }
 175
 176         bzero(&c, sizeof(c));
 177         c.cmd = CMD_HARDWARE_RESET;
 178         SendCommand(&c, FALSE);
 179
 180         CloseProxmark();
 181
 182         fprintf(stderr,"Have a nice day!\n");
 183
 184         return 0;
 185 }