#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <unistd.h>
#include "patcher.h"


//#######################################################################################################
void main(int argc, char* argv[]) {
  
FILE* in;
FILE* out;
uint8_t* buf;
uint32_t fsize;
int opt;
uint8_t outfilename[100];
int oflag=0,bflag=0;
uint32_t res;


// Разбор командной строки

while ((opt = getopt(argc, argv, "o:bh")) != -1) {
  switch (opt) {
   case 'h': 
     
printf("\n Программа для автоматического патча загрузчиков платформ Balong V7\n\n\
%s [ключи] <файл загрузчика usbloader>\n\n\
 Допустимы следующие ключи:\n\n\
-o file  - имя выходного файла. По умолчанию производится только проверка возможности патча\n\
-b       - добавить патч, отключающий проверку дефектных блоков\n\
\n",argv[0]);
    return;

   case 'o':
    strcpy(outfilename,optarg);
    oflag=1;
    break;

   case 'b':
     bflag=1;
     break;
     
   case '?':
   case ':':  
     return;
    
  }
}  

printf("\n Программа автоматической модификации загрузчиков Balong V7, (c) forth32");

 if (optind>=argc) {
    printf("\n - Не указано имя файла для загрузки\n - Для подсказки укажите ключ -h\n");
    return;
}  
    
in=fopen(argv[optind],"r");
if (in == 0) {
  printf("\n Ошибка открытия файла %s",argv[optind]);
  return;
}

// определяем размер файла
fseek(in,0,SEEK_END);
fsize=ftell(in);
rewind(in);

// выделяем буфер и читаем туда весь файл
buf=malloc(fsize);
fread(buf,1,fsize,in);
fclose(in);

res=pv7r2(buf, fsize);
if (res != 0)  printf("\n* Найдена сигнатура типа V7R2 по смещению %08x",res);
else {
   res=pv7r11(buf, fsize);
   if (res != 0)  printf("\n* Найдена сигнатура типа V7R11 по смещению %08x",res);
   else printf("\n! Сигнатура eraseall-патча не найдена");
}   
  
if (bflag) {
   res=perasebad(buf, fsize);
   if (res != 0) printf("\n* Найдена сигнатура isbad по смещению %08x",res);  
   else  printf("\n! Сигнатура isbad не найдена");  
}

if (oflag) {
  out=fopen(outfilename,"w");
  if (out != 0) {
    fwrite(buf,1,fsize,out);
    fclose(out);
  }
  else printf("\n Ошибка открытия выходного файла %s",outfilename);
}
free(buf);
printf("\n");
}