mirror of
https://github.com/forth32/balongflash
synced 2024-11-03 15:40:15 +00:00
295 lines
9.4 KiB
C
295 lines
9.4 KiB
C
#include <stdio.h>
|
||
#include <stdlib.h>
|
||
#include <string.h>
|
||
#include <sys/types.h>
|
||
#include <sys/stat.h>
|
||
#include <fcntl.h>
|
||
#include <strings.h>
|
||
#include <termios.h>
|
||
#include <unistd.h>
|
||
#include <arpa/inet.h>
|
||
|
||
#include "hdlcio.h"
|
||
|
||
//@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@
|
||
|
||
void main(int argc, char* argv[]) {
|
||
|
||
struct termios sioparm;
|
||
unsigned int i,j,res,opt,npart=0,iolen,part,blk,blksize;
|
||
FILE* in;
|
||
FILE* out;
|
||
|
||
struct {
|
||
unsigned int offset;
|
||
unsigned int size;
|
||
unsigned int code;
|
||
}ptable[100];
|
||
|
||
unsigned char buf[4096];
|
||
unsigned char devname[50]="/dev/ttyUSB0";
|
||
unsigned char replybuf[4096];
|
||
unsigned char datamodecmd[]="AT^DATAMODE";
|
||
unsigned char resetcmd[]="AT^RESET";
|
||
unsigned char OKrsp[]={0x0d, 0x0a, 0x4f, 0x4b, 0x0d, 0x0a};
|
||
unsigned int dpattern=0xa55aaa55;
|
||
unsigned int mflag=0,eflag=0,rflag=0;
|
||
unsigned char filename [100];
|
||
|
||
unsigned char cmdver[7]={0x0c}; // версия протокола
|
||
unsigned char cmddone[7]={0x1}; // команда выхода из HDLC
|
||
unsigned char cmd_reset[7]={0xa}; // команда выхода из HDLC
|
||
unsigned char cmd_dload_init[15]={0x41}; // команда начала раздела
|
||
unsigned char cmd_data_packet[11000]={0x42}; // команда начала пакета
|
||
unsigned char cmd_dload_end[30]={0x43}; // команда конца раздела
|
||
// Коды типов разделов
|
||
struct {
|
||
char name[20];
|
||
int code;
|
||
} pcodes[]={
|
||
{"M3Boot",0x20000},
|
||
{"Ptable",0x10000},
|
||
{"Fastboot",0x110000},
|
||
{"Kernel",0x30000},
|
||
{"VxWorks",0x40000},
|
||
{"M3Image",0x50000},
|
||
{"DSP",0x60000},
|
||
{"NVRAM",0x70000},
|
||
{"System",0x590000},
|
||
{"APP",0x5a0000},
|
||
{"CD_WUI_VER",0xa0000},
|
||
{"CDROMISO",0xb0000},
|
||
{"WEBUI",0x5b0000},
|
||
{0,0}
|
||
};
|
||
//-d - попытка переключить модем из режима HDLC в АТ-режим\n\
|
||
|
||
while ((opt = getopt(argc, argv, "hp:mer")) != -1) {
|
||
switch (opt) {
|
||
case 'h':
|
||
|
||
printf("\n Утилита предназначена для аварийной USB-загрузки модемов E3372S\n\n\
|
||
%s [ключи] <имя файла для загрузки>\n\n\
|
||
Допустимы следующие ключи:\n\n\
|
||
-p <tty> - последовательный порт для общения с загрузчиком (по умолчанию /dev/ttyUSB0\n\
|
||
-m - вывести карту файла прошивки и завершить работу\n\
|
||
-e - разобрать файл прошивки на разделы и завершить работу\n\
|
||
-r - перезагрузить модем после прошивки\n\
|
||
\n",argv[0]);
|
||
return;
|
||
|
||
case 'p':
|
||
strcpy(devname,optarg);
|
||
break;
|
||
|
||
case 'm':
|
||
mflag=1;
|
||
break;
|
||
|
||
case 'r':
|
||
rflag=1;
|
||
break;
|
||
|
||
case 'e':
|
||
eflag=1;
|
||
break;
|
||
|
||
case '?':
|
||
case ':':
|
||
return;
|
||
}
|
||
}
|
||
|
||
if (optind>=argc) {
|
||
printf("\n - Не указано имя файла для загрузки\n");
|
||
return;
|
||
}
|
||
|
||
in=fopen(argv[optind],"r");
|
||
if (in == 0) {
|
||
printf("\n Ошибка открытия %s",argv[optind]);
|
||
return;
|
||
}
|
||
|
||
// Поиск разделов внутри файла
|
||
printf("\n Разбираем файл прошвки...");
|
||
while (fread(&i,1,4,in) == 4) {
|
||
if (i != dpattern) continue; // ищем разделитель
|
||
ptable[npart].offset=ftell(in); // позиция начала раздела
|
||
fread(buf,1,96,in); // заголовок
|
||
ptable[npart].offset+=*((unsigned int*)&buf[0])-4; // выкидываем из смещения длину заголовка+КС
|
||
ptable[npart].size=*((unsigned int*)&buf[20]); // размер раздела
|
||
ptable[npart].code=*((unsigned int*)&buf[16]); // тип раздела
|
||
npart++;
|
||
}
|
||
if (npart == 0) {
|
||
printf("\nРазделы не найдены!");
|
||
return ;
|
||
}
|
||
printf(" найдено %i разделов",npart);
|
||
//------ Режим вывода карты файла прошивки
|
||
|
||
if (mflag) {
|
||
printf("\n Таблица разделов, найденных в файле:\n\n ## Смещение Размер Имя\n-------------------------------------");
|
||
for (i=0;i<npart;i++) {
|
||
printf("\n %02i %08x %8i ",i,ptable[i].offset,ptable[i].size);
|
||
for(j=0;pcodes[j].code != 0;j++) {
|
||
if(pcodes[j].code == ptable[i].code) break;
|
||
}
|
||
if (pcodes[j].code != 0) printf("%s",pcodes[j].name);
|
||
else printf("Неопределенный раздел - %08x",ptable[i].code);
|
||
}
|
||
printf("\n");
|
||
return;
|
||
}
|
||
|
||
|
||
//------- Режим разрезания файла прошивки
|
||
if (eflag) {
|
||
printf("\n Выделение разделов из файла прошивки:\n\n ## Смещение Размер Имя\n-------------------------------------");
|
||
for (i=0;i<npart;i++) {
|
||
printf("\n %02i %08x %8i ",i,ptable[i].offset,ptable[i].size);
|
||
for(j=0;pcodes[j].code != 0;j++) {
|
||
if(pcodes[j].code == ptable[i].code) break;
|
||
}
|
||
if (pcodes[j].code != 0) {
|
||
printf("%s",pcodes[j].name);
|
||
sprintf(filename,"%02i-%s",i,pcodes[j].name);
|
||
}
|
||
else {
|
||
printf("Неопределенный раздел - %08x",ptable[i].code);
|
||
sprintf(filename,"%02i-U%08x",i,ptable[i].code);
|
||
}
|
||
out=fopen(filename,"w");
|
||
fseek(in,ptable[i].offset,SEEK_SET); // встаем на начало
|
||
for(j=0;j<ptable[i].size;j+=4) {
|
||
fread(buf,4,1,in);
|
||
fwrite(buf,4,1,out);
|
||
}
|
||
fclose(out);
|
||
}
|
||
printf("\n");
|
||
return;
|
||
}
|
||
|
||
// Настройка SIO
|
||
|
||
if (!open_port(devname)) {
|
||
#ifndef WIN32
|
||
printf("\n - Последовательный порт %s не открывается\n", devname);
|
||
#else
|
||
printf("\n - Последовательный порт COM%s не открывается\n", devname);
|
||
#endif
|
||
return;
|
||
}
|
||
|
||
|
||
tcsetattr(siofd, TCSANOW, &sioparm);
|
||
tcflush(siofd,TCIOFLUSH); // очистка выходного буфера
|
||
|
||
// выходим из режима HDLC - если модем уже был в нем
|
||
port_timeout(1);
|
||
send_cmd(cmddone,1,replybuf);
|
||
|
||
|
||
|
||
// Входим в HDLC-режим
|
||
printf("\n Входим в режим HDLC...");
|
||
port_timeout(100);
|
||
write(siofd,datamodecmd,strlen(datamodecmd));
|
||
res=read(siofd,replybuf,6);
|
||
if (res != 6) {
|
||
printf("\n Неправильный ответ на DATAMODE\n");
|
||
dump(replybuf,res,0);
|
||
return;
|
||
}
|
||
if (memcmp(replybuf,OKrsp,6) != 0) {
|
||
printf("\n Неправильный ответ на DATAMODE\n");
|
||
dump(replybuf,res,0);
|
||
return;
|
||
}
|
||
|
||
printf("ok");
|
||
iolen=send_cmd(cmdver,1,replybuf);
|
||
if ((iolen == 0)||(replybuf[1] != 0x0d)) {
|
||
printf("\n Ошибка команды GET_VERSION\n");
|
||
if (iolen != 0) dump(replybuf,iolen,0);
|
||
return;
|
||
}
|
||
i=replybuf[2];
|
||
replybuf[3+i]=0;
|
||
printf("\n Версия протокола: %s",replybuf+3);
|
||
printf("\n");
|
||
|
||
|
||
|
||
// Главный цикл записи разделов
|
||
for(part=0;part<npart;part++) {
|
||
printf("\n Записываем раздел %i -",part);
|
||
// выводим имя раздела
|
||
for(j=0;pcodes[j].code != 0;j++) {
|
||
if(pcodes[j].code == ptable[part].code) break;
|
||
}
|
||
if (pcodes[j].code != 0) {
|
||
printf("%s\n",pcodes[j].name);
|
||
}
|
||
else {
|
||
printf("код %08x\n",ptable[npart].code);
|
||
}
|
||
|
||
// заполняем командный пакет
|
||
*((unsigned int*)&cmd_dload_init[1])=htonl(ptable[part].code);
|
||
*((unsigned int*)&cmd_dload_init[5])=htonl(ptable[part].size);
|
||
// отсылаем команду
|
||
iolen=send_cmd(cmd_dload_init,12,replybuf);
|
||
if ((iolen == 0) || (replybuf[1] != 2)) {
|
||
printf("\n Заголовок раздела не принят, код ошибки = %02x\n",replybuf[3]);
|
||
// dump(cmd_dload_init,13,0);
|
||
return;
|
||
}
|
||
|
||
|
||
// Поблочный цикл
|
||
blksize=4096;
|
||
fseek(in,ptable[part].offset,SEEK_SET);
|
||
for(blk=0;blk<((ptable[part].size+4095)/4096);blk++) {
|
||
printf("\r Блок %i из %i",blk,(ptable[part].size+4095)/4096); fflush(stdout);
|
||
res=ptable[part].size+ptable[part].offset-ftell(in); // размер оставшегося куска до конца файла
|
||
if (res<4096) blksize=res; // корректируем размер последнего блока
|
||
*(unsigned int*)&cmd_data_packet[1]=htonl(blk+1); // # пакета
|
||
*(unsigned short*)&cmd_data_packet[5]=htons(blksize); // размер блока
|
||
fread(cmd_data_packet+7,1,blksize,in); // читаем очередной кусок раздела в буфер команды
|
||
iolen=send_cmd(cmd_data_packet,blksize+7,replybuf); // отсылаем команду
|
||
if ((iolen == 0) || (replybuf[1] != 2)) {
|
||
printf("\n Блок %i раздела не принят, код ошибки = %02x\n",blk,replybuf[3]);
|
||
// dump(cmd_data_packet,blksize+7,0);
|
||
return;
|
||
}
|
||
}
|
||
|
||
// Закрытие потока
|
||
*((unsigned int*)&cmd_dload_end[1])=htonl(ptable[part].size);
|
||
*((unsigned int*)&cmd_dload_end[8])=htonl(ptable[part].code);
|
||
iolen=send_cmd(cmd_dload_end,24,replybuf); // отсылаем команду
|
||
if ((iolen == 0) || (replybuf[1] != 2)) {
|
||
printf("\n Ошибка закрытия раздела, код ошибки = %02x\n",replybuf[3]);
|
||
dump(replybuf,iolen,0);
|
||
printf("\nИсходная команда:");
|
||
dump(cmd_data_packet,24,0);
|
||
return;
|
||
}
|
||
|
||
}
|
||
// Конец главного цикла
|
||
printf("\n");
|
||
|
||
|
||
port_timeout(1);
|
||
// выходим из режима HDLC и перезагружаемся
|
||
if (rflag) {
|
||
send_cmd(cmd_reset,1,replybuf);
|
||
write(siofd,resetcmd,strlen(resetcmd));
|
||
}
|
||
else send_cmd(cmddone,1,replybuf);
|
||
}
|