nvexplorer.cpp

//-------------- Редактор двоичных образов NVRAM ----------------------------
#include "nvexplorer.h"
#include "sio.h"
#include "hexeditor.h"

//**************************************************
//* Конструктор класса
//**************************************************
nvexplorer::nvexplorer(uint8_t* xsrcdata, uint32_t srclen) : QMainWindow() {
 
uint32_t pos;
uint32_t i;
QString str;
QFont font;

// сохраняем параметры буфера с данными
srcdata=xsrcdata;
plen=srclen;

// создаем локальныую копию для редактирования
pdata=new uint8_t[plen];
memcpy(pdata,srcdata,plen);

// настройки геометрии окна
show();  
setAttribute(Qt::WA_DeleteOnClose);

config=new QSettings("forth32","qhuaweiflash",this);
QRect rect=config->value("/config/NvExplorerRect").toRect();
if (rect != QRect(0,0,0,0)) setGeometry(rect);
// выводим окно на передний план
setFocus();
raise();
activateWindow();

// Заголовок окна
setWindowTitle("Редактирование образа NVRAM");


// Главное меню
menubar = new QMenuBar(this);
setMenuBar(menubar);

menu_file = new QMenu("Файл",menubar);
menubar->addAction(menu_file->menuAction());

menu_edit = new QMenu("Правка",menubar);
menubar->addAction(menu_edit->menuAction());

menu_view = new QMenu("Вид",menubar);
menubar->addAction(menu_view->menuAction());

// тулбар
toolbar=new QToolBar(this);
addToolBar(toolbar);

// Центральный виджет
central=new QWidget(this);
setCentralWidget(central);

// основной компоновщик
vlm=new QVBoxLayout(central);

// Загружаем заголовок nv

memcpy(&nvhd, pdata, sizeof(nvhd));

// if (nvhd.magicnum != FILE_MAGIC_NUM) {
//   QMessageBox::critical(0,"Ошибка","Ошибка структуры образа NVRAM - неправильная сигнатура заголовка");
//   delete this;
// }

// Определяем тип CRC
switch (nvhd.crcflag) {
  case 0:
    crcmode=0;
    break;
    
  case 1:  
    crcmode=1;
    break;
    
  case 8:
    crcmode=2;
    break;
    
  default:
    crcmode=-1;
    break;
    
}
//----- Читаем каталог файлов

pos=nvhd.ctrl_size; // смещение до начала данных (конец управляющих структур)

// размер описателя файла в образе nvram
uint32_t fcsize=sizeof(struct nv_file)-4;
// смещение до описателя текущего файла
uint32_t fsoffset;

// вынимаем все описатели файлов
for(i=0;i<nvhd.file_num;i++) {
 fsoffset=i*fcsize; 
 memcpy(&flist[i],pdata+nvhd.file_offset+fsoffset,fcsize); 
 // вычисляем смещение до данных файла
 flist[i].offset=pos;
 pos+=flist[i].size;
}

// получаем смещение до поля CRC
crcoff=pos;

//----- Читаем каталог ячеек
itemlist=new struct nv_item[nvhd.item_size];
memcpy(itemlist,pdata+nvhd.item_offset,nvhd.item_size);

// Вычисляем максимальный размер ячейки
// for(i=0;i<nvhd.item_count;i++) 
//  if (maxitemlen < itemlist[i].len) maxitemlen = itemlist[i].len;
  
// Создаем таблицу nvram
nvtable=new QTableWidget(nvhd.item_count,5,central);

// заголовок таблицы
QStringList plst;
plst << "NVID" << "Размер" <<"Компонент" <<"Имя" <<"Содержимое";
nvtable->setHorizontalHeaderLabels(plst);

// выводим список ячеек в таблицу
QTableWidgetItem* cell;


for(i=0;i<nvhd.item_count;i++) {
  // id ячейки
  str.setNum(itemlist[i].id);
  cell=new QTableWidgetItem(str);
  cell->setFlags(Qt::ItemIsSelectable|Qt::ItemIsUserCheckable|Qt::ItemIsEnabled);
  font=cell->font();
  font.setBold(true);
  cell->setFont(font);
  nvtable->setItem(i,0,cell);

  // размер ячейки
  str.setNum(itemlist[i].len);
  cell=new QTableWidgetItem(str);
  cell->setFlags(Qt::ItemIsSelectable|Qt::ItemIsUserCheckable|Qt::ItemIsEnabled);
  nvtable->setItem(i,1,cell);

  // компонент
  int fid=itemlist[i].file_id;
  str.sprintf("%1i:%s",fid,flist[fileidx(fid)].name);
  cell=new QTableWidgetItem(str);
  cell->setFlags(Qt::ItemIsSelectable|Qt::ItemIsUserCheckable|Qt::ItemIsEnabled);
  nvtable->setItem(i,2,cell);

  // имя
  str=find_desc(itemlist[i].id);
  cell=new QTableWidgetItem(str);
  cell->setFlags(Qt::ItemIsSelectable|Qt::ItemIsUserCheckable|Qt::ItemIsEnabled);
  nvtable->setItem(i,3,cell);

  // Содержимое  
  datacell(i);
}

// ширина колонок
for(i=0;i<4;i++) {
   nvtable->resizeColumnToContents(i);
}  
// расширяем поле ID для лучшей читаемости
nvtable->setColumnWidth(0,nvtable->columnWidth(0)+5);
// расширяем поле содержимого до максимума
nvtable->horizontalHeader()->setSectionResizeMode(4, QHeaderView::Stretch);

// Вводим таблицу в компоновщик
vlm->addWidget(nvtable,3);

//-----------------------------------------------------------------------------------------------------------------------------------------------
// пункты меню
menu_file->addAction(QIcon::fromTheme("document-save"),"Сохранить",this,SLOT(save_all()),QKeySequence::Save);
toolbar->addAction(QIcon::fromTheme("document-save"),"Сохранить",this,SLOT(save_all()));
menu_file->addSeparator();
menu_file->addAction("Выход",this,SLOT(close()),QKeySequence("Esc"));

toolbar->addSeparator();

menu_edit->addAction(QIcon(":/icon_hex.png"),"Редактировать ячейку",this,SLOT(edititem()),QKeySequence("F2"));
toolbar->addAction(QIcon(":/icon_hex.png"),"Редактировать ячейку",this,SLOT(edititem()));

menu_edit->addAction(QIcon(":/icon_extract.png"),"Извлечь ячейку в файл",this,SLOT(extract_item()),QKeySequence("F11"));
toolbar->addAction(QIcon(":/icon_extract.png"),"Извлечь ячейку в файл",this,SLOT(extract_item()));

menu_edit->addAction(QIcon::fromTheme("object-flip-vertical"),"Загрузить ячейку из файла",this,SLOT(replace_item()),0);
toolbar->addAction(QIcon::fromTheme("object-flip-vertical"),"Загрузить ячейку из файла",this,SLOT(replace_item()));

menu_view->addAction(QIcon::fromTheme("zoom-in"),"Увеличить шрифт",this,SLOT(zoomin()),QKeySequence("Ctrl++"));
toolbar->addAction(QIcon::fromTheme("zoom-in"),"Увеличить шрифт",this,SLOT(zoomin()));
menu_view->addAction(QIcon::fromTheme("zoom-out"),"Уменьшить шрифт",this,SLOT(zoomout()),QKeySequence("Ctrl+-"));
toolbar->addAction(QIcon::fromTheme("zoom-out"),"Уменьшить шрифт",this,SLOT(zoomout()));

connect(nvtable,SIGNAL(cellActivated(int,int)),SLOT(edititem()));


}

//**********************************************************************
//*  ДЕструктор класса
//**********************************************************************
nvexplorer::~nvexplorer() {

int reply;

if (changed) {
  reply=QMessageBox::warning(this,"Запись данных","Содержимое NVRAM изменено, сохранить?",QMessageBox::Ok | QMessageBox::Cancel);
  if (reply == QMessageBox::Ok) {
    // сохранение данных
    save_all();
  }
}  

// геометрия главного окна
QRect rect=geometry();
config->setValue("/config/NvExplorerRect",rect);

delete nvtable;
delete [] itemlist;
delete [] pdata;
}
//**********************************************************************
//*  Ввод в таблицу содержимого ячейки
//**********************************************************************
void nvexplorer::datacell(int row) {
  
char dstr[10];
QString str;
uint32_t j;
uint32_t off=itemoff_idx(row);
QTableWidgetItem* cell;

uint32_t itemlen=itemlist[row].len;

for(j=0;j<itemlen;j++) {
  sprintf(dstr,"%02X ",*((uint8_t*)(pdata+off+j))&0xff);
  str.append(dstr);
}
    
cell=new QTableWidgetItem(str);
cell->setFlags(Qt::ItemIsSelectable|Qt::ItemIsUserCheckable|Qt::ItemIsEnabled);
QFont font=cell->font();
font.setFixedPitch(true);
cell->setFont(font);
nvtable->setItem(row,4,cell);
}


//**********************************************************************
//* Увеличение/уменьшение шрифта
//**********************************************************************
void nvexplorer::zoom (int dir) {
  
QFont font;
int row,col;

for(row=0;row<nvtable->rowCount();row++) {
  for(col=0;col<nvtable->columnCount();col++) {
    font=nvtable->item(row,col)->font();
    font.setPointSize(font.pointSize()+dir);
    nvtable->item(row,col)->setFont(font);
  }
}
// ширина колонок
for(col=0;col<4;col++) {
   nvtable->resizeColumnToContents(col);
}  
// расширяем поле ID для лучшей читаемости
nvtable->setColumnWidth(0,nvtable->columnWidth(0)+5);

}

//**********************************************************************
//* Слоты zoom
//**********************************************************************
void nvexplorer::zoomin() { zoom(1); }
void nvexplorer::zoomout() { zoom(-1); }
    

//**********************************************************************
//* Редактор ячеек
//**********************************************************************
void nvexplorer::edititem() {
 
QString title;  
  
int row=nvtable->currentRow();
uint32_t len=itemlist[row].len;
int res;

// загружаем данные в буфер для редактирования
QByteArray hexcup((char*)(pdata+itemoff_idx(row)),len);

// панель диалога
QDialog* qd=new QDialog;
QVBoxLayout* vlm=new QVBoxLayout(qd);

// заголовок
title.sprintf("Редактирование ячейки %i",itemlist[row].id);
config=new QSettings("forth32","qhuaweiflash",this);
qd->setWindowTitle(title);

// размер окна
QRect rect=config->value("/config/ItemEditorRect").toRect();
if (rect != QRect(0,0,0,0)) qd->setGeometry(rect);
else qd->resize(625,625);


// HEX-редактор
QHexEdit* dhex=new QHexEdit(qd);

// Настройка внешнего вида редактора
dhex->setAddressWidth(3);
dhex->setOverwriteMode(true);
dhex->setHexCaps(true);
dhex->setHighlighting(true);

// Загрузка данных в редактор
dhex->setData(hexcup);

dhex->setCursorPosition(0);
dhex->show();
dhex->setReadOnly(false);

vlm->addWidget(dhex);

// комментарии к шорткутам
QLabel* lbl=new QLabel("Enter - сохранить изменения,   Esc - отменить",qd);
vlm->addWidget(lbl);

// кнопки save и cancel
QDialogButtonBox* butt=new QDialogButtonBox(QDialogButtonBox::Save|QDialogButtonBox::Cancel,Qt::Horizontal,qd);
butt->button(QDialogButtonBox::Save)->setShortcut(QKeySequence(Qt::Key_Return));
butt->button(QDialogButtonBox::Cancel)->setShortcut(QKeySequence(Qt::Key_Escape));
connect(butt, SIGNAL(accepted()), qd, SLOT(accept()));
connect(butt, SIGNAL(rejected()), qd, SLOT(reject()));
vlm->addWidget(butt);

res=qd->exec();
if (res == QDialog::Accepted) {
  // изменения приняты
  if (dhex->isModified()) {
   hexcup=dhex->data();
   memcpy(pdata+itemoff_idx(row),hexcup.data(),len);
   changed_item(row);
  } 
}
// геометрия окна
rect=qd->geometry();
config->setValue("/config/ItemEditorRect",rect);

delete qd;
}



//**********************************************************************
//* Пост-обработка модификации ячеек
//**********************************************************************
void nvexplorer::changed_item(int row) {

QString title;  

// пересчитываем индивидуальную CRC - для файлов образов прошивок это не требуется
//   if (crcmode == 2) restore_item_crc(row);

// Вводим звездочку в заголовок
if (!changed) {
    // вводим звездочку в заголовок
    title=windowTitle();
    title.append(" *");
    setWindowTitle(title);
    changed=true;
}  
// перерисовываем строку данных в таблице
datacell(row);
}

//**********************************************************************
//* Сохранение всех изменений обратно в исходный буфер
//**********************************************************************
void nvexplorer::save_all() {

QString str;  
int pos;  

// пересчитываем блочную CRC
if (crcmode == 1) recalc_crc();

// копируем весь массив наружу
memcpy(srcdata,pdata,plen);
  
// удаляем звездочку из заголовка
str=windowTitle();
pos=str.indexOf('*');
if (pos != -1) {
  str.truncate(pos-1);
  setWindowTitle(str);
}  

changed=false;
}

//**********************************************************************
//* Извлечение ячейки в файл
//**********************************************************************
void nvexplorer::extract_item() {

QString filename;

int row=nvtable->currentRow();
filename.sprintf("nvitem-%05i.bin",itemlist[row].id);
filename=QFileDialog::getSaveFileName(this,"Имя сохраняемого файла",filename,"All files (*.*)");
if (filename.isEmpty()) return;
  
QFile out(filename,this);
if (!out.open(QIODevice::WriteOnly)) {
    QMessageBox::critical(0,"Ошибка","Ошибка создания файла");
    return;
}
out.write((char*)(pdata+itemoff_idx(row)),itemlist[row].len);
out.close();
}

//**********************************************************************
//* Загрузка ячейки из файла
//**********************************************************************
void nvexplorer::replace_item() {

QString filename;
QString str;

int row=nvtable->currentRow();
filename.sprintf("nvitem-%05i.bin",itemlist[row].id);
filename=QFileDialog::getOpenFileName(this,"Имя файла",filename,"All files (*.*)");
if (filename.isEmpty()) return;
  
QFile in(filename,this);
if (!in.open(QIODevice::ReadOnly)) {
    QMessageBox::critical(0,"Ошибка","Ошибка открытия файла");
    return;
}

if (in.size() != itemlist[row].len) {
    in.close();
    str.sprintf("Размер файла (%i) не совпадает с размером ячейки (%i)",(uint32_t)in.size(),(uint32_t)itemlist[row].len);
    QMessageBox::critical(0,"Ошибка",str);
    return;
}  


in.read((char*)(pdata+itemoff_idx(row)),itemlist[row].len);
in.close();
changed_item(row);

}