#include "FireDogEditor.h" #include #include #include #include #include "config.h" #include "featurelibrary.h" #include "rule/rule.h" #include "partutil.h" #include "matchthread.h" #include "gui18n.h" #include "gui18nutil.h" using namespace firedog; //构造函数 FireDogEditor::FireDogEditor(QWidget *parent) : QssMainWindow(parent) { ui.setupUi(this); updateWindowTitle(false); this->loadingDialog = new LoadingDialog(this); //添加版本 QString fireDogVersion = FIREDOG_VERSION; QLabel* fireDogVersionLable = new QLabel("FireDog Version:" + fireDogVersion, this); QString fireDogFeatureLibraryVersion = FIREDOG_FEATURE_LIBRARY_VERSION; QLabel* fireDogFeatureLibraryVersionLable = new QLabel("Feature Library Version:" + fireDogFeatureLibraryVersion, this); statusBar()->addPermanentWidget(fireDogVersionLable); statusBar()->addPermanentWidget(fireDogFeatureLibraryVersionLable); //初始化 init(); } //初始化 void FireDogEditor::init() { QFile qss(":/qss/css/qss.css"); if (qss.open(QIODevice::ReadOnly)) { QByteArray qssBytes = qss.readAll(); //this->setStyleSheet(qssBytes); qApp->setStyleSheet(qssBytes); qApp->setFont(QFont("微软雅黑", 9)); } this->parseThread = new ParseThread(); //子窗口 fireDogFeatureInfoView = new FireDogFeatureInfo(this); QString fireDogFeatureInfoViewTitle = Gui18n::GetInstance()->GetConfig("feature-info-window-title", "Feature Editor"); fireDogFeatureInfoView->setWindowTitle(fireDogFeatureInfoViewTitle); fireDogFeatureRuleInfoView = new FireDogFeatureRuleInfo(this); QString fireDogFeatureRuleInfoViewTitle = Gui18n::GetInstance()->GetConfig("feature-rule-window-title", "Feature Rule Editor"); fireDogFeatureRuleInfoView->setWindowTitle(fireDogFeatureRuleInfoViewTitle); fireDogAbout = new FireDogAbout(this); QString fireDogAboutTitle = Gui18n::GetInstance()->GetConfig("about-window-title", "About"); fireDogAbout->setWindowTitle(fireDogAboutTitle); fireDogSwitchLanguageUi = new Ui::FireDogSwitchLanguageUi(this); //tab表-end================================================================== //特征库表格-start================================================================== this->featureLibraryTableModel = new BigDataTableModel(); QStringList featureLibraryTableModelTitle; QString ftableTitlteName = Gui18n::GetInstance()->GetConfig("feature-librarys-name", "Name"); QString ftableTitlteDescribe = Gui18n::GetInstance()->GetConfig("feature-librarys-describe", "Describe"); QString ftableTitlteAuthor = Gui18n::GetInstance()->GetConfig("feature-librarys-author", "Author"); featureLibraryTableModelTitle << ftableTitlteName << ftableTitlteDescribe << ftableTitlteAuthor; featureLibraryTableModel->setHeaders(featureLibraryTableModelTitle); ui.tableViewLibrary->setModel(this->featureLibraryTableModel); //不可编辑 ui.tableViewLibrary->setEditTriggers(QAbstractItemView::NoEditTriggers); ui.tableViewLibrary->setSelectionMode(QAbstractItemView::SingleSelection); ui.tableViewLibrary->setSelectionBehavior(QAbstractItemView::SelectRows); //一次滚动一个像素 ui.tableViewLibrary->setVerticalScrollMode(QAbstractItemView::ScrollPerPixel); ui.tableViewLibrary->setHorizontalScrollMode(QAbstractItemView::ScrollPerPixel); //可弹出右键菜单 ui.tableViewLibrary->setContextMenuPolicy(Qt::CustomContextMenu); //设置列宽 ui.tableViewLibrary->setColumnWidth(0, 150); ui.tableViewLibrary->setColumnWidth(1, 290); ui.tableViewLibrary->setColumnWidth(2, 150); //菜单 QString fltMenuAddNew = Gui18n::GetInstance()->GetConfig("feature-librarys-menu-addnew", "Add New"); QString fltMenuAddDelete = Gui18n::GetInstance()->GetConfig("feature-librarys-menu-delete", "Delete"); featureLibraryTableMenu = new QMenu(ui.tableViewLibrary); featureLibraryTableMenuAddAction = new QAction(); featureLibraryTableMenuAddAction->setText(QString(fltMenuAddNew)); featureLibraryTableMenu->addAction(featureLibraryTableMenuAddAction); featureLibraryTableMenuDelAction = new QAction(); featureLibraryTableMenuDelAction->setText(QString(fltMenuAddDelete)); featureLibraryTableMenu->addAction(featureLibraryTableMenuDelAction); //特征库表格-end================================================================== //详情-特征表格-start================================================================== this->featureLibraryInfoFeatureTableModel = new BigDataTableModel(); QString fftableTitlteKey = Gui18n::GetInstance()->GetConfig("feature-info-features-key", "Key"); QString fftableTitlteType = Gui18n::GetInstance()->GetConfig("feature-info-features-type", "Type"); QString fftableTitlteContent = Gui18n::GetInstance()->GetConfig("feature-info-features-content", "Content"); QStringList featureLibraryInfoFeatureTableModelTitle; featureLibraryInfoFeatureTableModelTitle << fftableTitlteKey << fftableTitlteType << fftableTitlteContent; featureLibraryInfoFeatureTableModel->setHeaders(featureLibraryInfoFeatureTableModelTitle); ui.tableViewLibraryInfoFeatures->setModel(this->featureLibraryInfoFeatureTableModel); //菜单 QString flitMenuAdd = Gui18n::GetInstance()->GetConfig("feature-info-features-menu-add", "Add"); QString flitMenuEdit = Gui18n::GetInstance()->GetConfig("feature-info-features-menu-edit", "Edit"); QString flitMenuAddDelete = Gui18n::GetInstance()->GetConfig("feature-info-features-menu-del", "Delete"); featureLibraryInfoFeatureTableMenu = new QMenu(ui.tableViewLibraryInfoFeatures); featureLibraryInfoFeatureTableMenuAddAction = new QAction(); featureLibraryInfoFeatureTableMenuAddAction->setText(QString(flitMenuAdd)); featureLibraryInfoFeatureTableMenu->addAction(featureLibraryInfoFeatureTableMenuAddAction); featureLibraryInfoFeatureTableMenuEditAction = new QAction(); featureLibraryInfoFeatureTableMenuEditAction->setText(QString(flitMenuEdit)); featureLibraryInfoFeatureTableMenu->addAction(featureLibraryInfoFeatureTableMenuEditAction); featureLibraryInfoFeatureTableMenuDelAction = new QAction(); featureLibraryInfoFeatureTableMenuDelAction->setText(QString(flitMenuAddDelete)); featureLibraryInfoFeatureTableMenu->addAction(featureLibraryInfoFeatureTableMenuDelAction); //不可编辑 ui.tableViewLibraryInfoFeatures->setEditTriggers(QAbstractItemView::NoEditTriggers); ui.tableViewLibraryInfoFeatures->setSelectionMode(QAbstractItemView::SingleSelection); ui.tableViewLibraryInfoFeatures->setSelectionBehavior(QAbstractItemView::SelectRows); //一次滚动一个像素 ui.tableViewLibraryInfoFeatures->setVerticalScrollMode(QAbstractItemView::ScrollPerPixel); ui.tableViewLibraryInfoFeatures->setHorizontalScrollMode(QAbstractItemView::ScrollPerPixel); //可弹出右键菜单 ui.tableViewLibraryInfoFeatures->setContextMenuPolicy(Qt::CustomContextMenu); //设置列宽 ui.tableViewLibraryInfoFeatures->setColumnWidth(0, 100); ui.tableViewLibraryInfoFeatures->setColumnWidth(1, 100); ui.tableViewLibraryInfoFeatures->setColumnWidth(2, 385); //详情-特征表格-end================================================================== //详情-规则-start================================================================== this->featureLibraryInfoRuleTreeModel = new QStandardItemModel(); ui.treeViewLibraryInfoRules->setModel(featureLibraryInfoRuleTreeModel); //可弹出右键菜单 ui.treeViewLibraryInfoRules->setContextMenuPolicy(Qt::CustomContextMenu); //菜单 QString flrtMenuAdd = Gui18n::GetInstance()->GetConfig("feature-info-rule-menu-addchild", "Add Child"); QString flrtMenuEdit = Gui18n::GetInstance()->GetConfig("feature-info-rule-menu-edit", "Edit"); QString flrtMenuAddDelete = Gui18n::GetInstance()->GetConfig("feature-info-rule-menu-del", "Delete"); featureLibraryInfoRuleTreeMenu = new QMenu(ui.treeViewLibraryInfoRules); featureLibraryInfoRuleTreeMenuAddAction = new QAction(); featureLibraryInfoRuleTreeMenuAddAction->setText(QString(flrtMenuAdd)); featureLibraryInfoRuleTreeMenu->addAction(featureLibraryInfoRuleTreeMenuAddAction); featureLibraryInfoRuleTreeMenuEditAction = new QAction(); featureLibraryInfoRuleTreeMenuEditAction->setText(QString(flrtMenuEdit)); featureLibraryInfoRuleTreeMenu->addAction(featureLibraryInfoRuleTreeMenuEditAction); featureLibraryInfoRuleTreeMenuDelAction = new QAction(); featureLibraryInfoRuleTreeMenuDelAction->setText(QString(flrtMenuAddDelete)); featureLibraryInfoRuleTreeMenu->addAction(featureLibraryInfoRuleTreeMenuDelAction); //详情-规则-end================================================================== //匹配-文件列表-start================================================================== this->matchingFilesTableModel = new BigDataTableModel(); QString mfilesTablePath = Gui18n::GetInstance()->GetConfig("matching-files-path", "Path"); QString mfilesTableSize = Gui18n::GetInstance()->GetConfig("matching-files-size", "Size"); QStringList matchingFilesTableModelTitle; matchingFilesTableModelTitle << mfilesTablePath << mfilesTableSize; matchingFilesTableModel->setHeaders(matchingFilesTableModelTitle); ui.tableViewFiles->setModel(this->matchingFilesTableModel); //菜单 QString mfilesTableMenuDel = Gui18n::GetInstance()->GetConfig("matching-files-menu-del", "Delete"); matchingFilesTableMenu = new QMenu(ui.tableViewFiles); matchingFilesTableDelAction = new QAction(); matchingFilesTableDelAction->setText(QString(mfilesTableMenuDel)); matchingFilesTableMenu->addAction(matchingFilesTableDelAction); //不可编辑 ui.tableViewFiles->setEditTriggers(QAbstractItemView::NoEditTriggers); ui.tableViewFiles->setSelectionMode(QAbstractItemView::SingleSelection); ui.tableViewFiles->setSelectionBehavior(QAbstractItemView::SelectRows); //一次滚动一个像素 ui.tableViewFiles->setVerticalScrollMode(QAbstractItemView::ScrollPerPixel); ui.tableViewFiles->setHorizontalScrollMode(QAbstractItemView::ScrollPerPixel); //可弹出右键菜单 ui.tableViewFiles->setContextMenuPolicy(Qt::CustomContextMenu); //设置列宽 //设置列宽 ui.tableViewFiles->setColumnWidth(0, 280); ui.tableViewFiles->setColumnWidth(1, 100); //匹配-文件列表-end================================================================== //匹配-结果列表-start================================================================== this->matchingResultTableModel = new BigDataTableModel(); QString mresultTableSource = Gui18n::GetInstance()->GetConfig("matching-result-table-source", "Source"); QString mresultTableFeatureName = Gui18n::GetInstance()->GetConfig("matching-result-table-feature-name", "Feature Name"); QString mresultTableFeatureDescribe = Gui18n::GetInstance()->GetConfig("matching-result-table-feature-describe", "Feature Describe"); QString mresultTableFeatureAuthor = Gui18n::GetInstance()->GetConfig("matching-result-table-feature-author", "Feature Author"); QStringList matchingResultTableTitle; matchingResultTableTitle << mresultTableSource << mresultTableFeatureName << mresultTableFeatureDescribe << mresultTableFeatureAuthor; matchingResultTableModel->setHeaders(matchingResultTableTitle); ui.tableViewSearchResult->setModel(this->matchingResultTableModel); //不可编辑 ui.tableViewSearchResult->setEditTriggers(QAbstractItemView::NoEditTriggers); ui.tableViewSearchResult->setSelectionMode(QAbstractItemView::SingleSelection); ui.tableViewSearchResult->setSelectionBehavior(QAbstractItemView::SelectRows); //一次滚动一个像素 ui.tableViewSearchResult->setVerticalScrollMode(QAbstractItemView::ScrollPerPixel); ui.tableViewSearchResult->setHorizontalScrollMode(QAbstractItemView::ScrollPerPixel); //可弹出右键菜单 ui.tableViewSearchResult->setContextMenuPolicy(Qt::CustomContextMenu); //设置列宽 //设置列宽 ui.tableViewSearchResult->setColumnWidth(0, 180); ui.tableViewSearchResult->setColumnWidth(1, 150); ui.tableViewSearchResult->setColumnWidth(2, 300); ui.tableViewSearchResult->setColumnWidth(3, 150); //匹配-结果列表-end================================================================== //绑定事件 connect(ui.actionOpen, &QAction::triggered, this, &FireDogEditor::slots_openFile); connect(ui.actionSave, &QAction::triggered, this, &FireDogEditor::slots_saveFile); connect(ui.actionSaveTo, &QAction::triggered, this, &FireDogEditor::slots_saveToFile); connect(ui.actionHomePage, &QAction::triggered, this, &FireDogEditor::slots_openGit); connect(ui.actionReportIssue, &QAction::triggered, this, &FireDogEditor::slots_openIssue); connect(ui.actionFireKylin, &QAction::triggered, this, &FireDogEditor::slots_openFireKylin); connect(ui.actionAbout, &QAction::triggered, this, &FireDogEditor::slots_about); connect(ui.actionLanguage, &QAction::triggered, this, &FireDogEditor::slots_settingLanguage); connect(ui.pushButtonSearchName, &QPushButton::clicked, this, &FireDogEditor::slots_featureLibraryTableSearch); //特征表格右键事件 connect(ui.tableViewLibrary, SIGNAL(customContextMenuRequested(QPoint)), this, SLOT(slots_featureTableOpenMenu(QPoint))); //特征详情表格右键事件 connect(ui.tableViewLibraryInfoFeatures, SIGNAL(customContextMenuRequested(QPoint)), this, SLOT(slots_featureInfoTableOpenMenu(QPoint))); //特征详情规则树右键事件 connect(ui.treeViewLibraryInfoRules, SIGNAL(customContextMenuRequested(QPoint)), this, SLOT(slots_featureInfoRuleTreeOpenMenu(QPoint))); //匹配文件列表右键事件 connect(ui.tableViewFiles, SIGNAL(customContextMenuRequested(QPoint)), this, SLOT(slots_matchFilesTableOpenMenu(QPoint))); //特征表格点击事件 connect(ui.tableViewLibrary->selectionModel(), SIGNAL(currentRowChanged(const QModelIndex&, const QModelIndex&)), this, SLOT(slots_selectFeatureTableEvent(const QModelIndex&, const QModelIndex&))); //解析线程 connect(this->parseThread, SIGNAL(parseBegin()), this, SLOT(slots_parseBinBegin())); connect(this->parseThread, SIGNAL(parseEnd(firedog::FeatureLibrary*,int)), this, SLOT(slots_parseBinEnd(firedog::FeatureLibrary*, int))); connect(this->featureLibraryTableMenuAddAction, &QAction::triggered, this, &FireDogEditor::slots_featureTableMenuAddEvent); connect(this->featureLibraryTableMenuDelAction, &QAction::triggered, this, &FireDogEditor::slots_featureTableMenuDelEvent); connect(this->featureLibraryInfoFeatureTableMenuAddAction, &QAction::triggered, this, &FireDogEditor::slots_featureInfoTableMenuAddEvent); connect(this->featureLibraryInfoFeatureTableMenuEditAction, &QAction::triggered, this, &FireDogEditor::slots_featureInfoTableMenuEditEvent); connect(this->featureLibraryInfoFeatureTableMenuDelAction, &QAction::triggered, this, &FireDogEditor::slots_featureInfoTableMenuDelEvent); connect(this->featureLibraryInfoRuleTreeMenuAddAction, &QAction::triggered, this, &FireDogEditor::slots_featureLibraryInfoRuleMenuAddEvent); connect(this->featureLibraryInfoRuleTreeMenuEditAction, &QAction::triggered, this, &FireDogEditor::slots_featureLibraryInfoRuleMenuEditEvent); connect(this->featureLibraryInfoRuleTreeMenuDelAction, &QAction::triggered, this, &FireDogEditor::slots_featureLibraryInfoRuleMenuDelEvent); connect(this->matchingFilesTableDelAction, &QAction::triggered, this, &FireDogEditor::slots_matchingFilesDeleteEvent); connect(ui.pushButtonLibraryInfoSave, &QPushButton::clicked, this, &FireDogEditor::slots_saveBtnClickEvent); connect(ui.pushButtonAddFiles, &QPushButton::clicked, this, &FireDogEditor::slots_matchingAddFilesBtnClickEvent); connect(ui.pushButtonMatch, &QPushButton::clicked, this, &FireDogEditor::slots_matchingBtnClick); connect(ui.pushButtonSearchResult, &QPushButton::clicked, this, &FireDogEditor::slots_searchMatchResultBtnClick); connect(this->fireDogSwitchLanguageUi, SIGNAL(switchSuccess()), this, SLOT(slots_reboot())); //初始化多语言 Gui18nUtil::SetText(ui.menuOperation, "menu-operation"); Gui18nUtil::SetText(ui.actionOpen, "menu-operation-open"); Gui18nUtil::SetText(ui.actionSave, "menu-operation-save"); Gui18nUtil::SetText(ui.actionSaveTo, "menu-operation-saveto"); Gui18nUtil::SetText(ui.menuSetting, "menu-setting"); Gui18nUtil::SetText(ui.actionLanguage, "menu-setting-language"); Gui18nUtil::SetText(ui.menuHelp, "menu-help"); Gui18nUtil::SetText(ui.actionReportIssue, "menu-help-reportissue"); Gui18nUtil::SetText(ui.actionHomePage, "menu-help-github"); Gui18nUtil::SetText(ui.actionAbout, "menu-help-about"); Gui18nUtil::SetText(ui.tabWidgetMain, 0, "main-tab-feature-library"); Gui18nUtil::SetText(ui.tabWidgetMain, 1, "main-tab-matching"); Gui18nUtil::SetText(ui.groupBoxSearch, "feature-search"); Gui18nUtil::SetText(ui.lineEditSearchName, "feature-search-txt-tip"); Gui18nUtil::SetText(ui.pushButtonSearchName, "feature-search-btn"); Gui18nUtil::SetText(ui.groupBoxLibrarays, "feature-librarys"); Gui18nUtil::SetText(ui.groupBoxLibraryInfo, "feature-info"); Gui18nUtil::SetText(ui.labelLibraryInfoName, "feature-info-name"); Gui18nUtil::SetText(ui.labelLibraryInfoAuthor, "feature-info-author"); Gui18nUtil::SetText(ui.labelLibraryInfoDescribe, "feature-info-describe"); Gui18nUtil::SetText(ui.labelLibraryInfoFeatures, "feature-info-features"); Gui18nUtil::SetText(ui.labelLibraryInfoRule, "feature-info-rule"); Gui18nUtil::SetText(ui.pushButtonLibraryInfoSave, "feature-info-btn-add"); Gui18nUtil::SetText(ui.groupBoxInput, "matching-input"); Gui18nUtil::SetText(ui.labelText, "matching-input-text"); Gui18nUtil::SetText(ui.labelHex, "matching-input-hex"); Gui18nUtil::SetText(ui.labelFiles, "matching-input-files"); Gui18nUtil::SetText(ui.pushButtonAddFiles, "matching-input-files-addfiles"); Gui18nUtil::SetText(ui.groupBoxConfig, "matching-config"); Gui18nUtil::SetText(ui.label100MB, "matching-config-ge100mb-slice"); Gui18nUtil::SetText(ui.label1GB, "matching-config-ge1gb-slice"); Gui18nUtil::SetText(ui.labelThreadNum, "matching-config-thread"); Gui18nUtil::SetText(ui.pushButtonMatch, "matching-btn"); Gui18nUtil::SetText(ui.groupBoxResult, "matching-result"); Gui18nUtil::SetText(ui.lineEditSearchResult, "matching-result-search-text"); Gui18nUtil::SetText(ui.pushButtonSearchResult, "matching-result-search-btn"); } void FireDogEditor::slots_openIssue() { QDesktopServices::openUrl(QUrl(QLatin1String("https://github.com/MountCloud/FireDog/issues"))); } void FireDogEditor::slots_openGit() { QDesktopServices::openUrl(QUrl(QLatin1String("https://github.com/MountCloud/FireDog"))); } void FireDogEditor::slots_openFireKylin() { QDesktopServices::openUrl(QUrl(QLatin1String("https://github.com/MountCloud/FireKylin"))); } void FireDogEditor::slots_about() { this->fireDogAbout->exec(); } void FireDogEditor::slots_settingLanguage() { this->fireDogSwitchLanguageUi->exec(); } void FireDogEditor::updateWindowTitle(bool isUpdated) { //设置标题 QString windowTitle = "FireDog Editor "; windowTitle.append(FIREDOG_EDITOR_VERSION); windowTitle.append(" "); windowTitle.append(" "); if (isUpdated) { windowTitle.append("*"); } if (!this->openFilePath.isEmpty()) { windowTitle.append("["); QFile file(openFilePath); QFileInfo fileInfo(file); QString fileName = fileInfo.fileName(); windowTitle.append(fileName); windowTitle.append("]"); } else { QString tempFileTip = Gui18n::GetInstance()->GetConfig("window-title-temporary-file", "Temporary file"); windowTitle.append("["); windowTitle.append(tempFileTip); windowTitle.append("]"); } setWindowTitle(windowTitle); } void FireDogEditor::slots_saveFile() { if (this->featureLibrary == NULL || this->openFilePath.isEmpty()) { QssMessageBox::warn("Please open the file first.", this, "Warn"); return; } //开始保存 QFile file(this->openFilePath); int state = -1; std::string json = this->featureLibrary->toJson(&state); if (state != NO_ERROR) { QssMessageBox::warn("Save fail,Feature library convert to json fail!.", this, "Warn"); return; } QString jsonstr = QString::fromUtf8(QByteArray::fromRawData(json.c_str(), json.size())); if (!file.open(QIODevice::WriteOnly | QIODevice::Text)) { QssMessageBox::warn("File Open fail!.", this, "Warn"); return; } QTextStream in(&file); in << jsonstr; file.flush(); file.close(); updateWindowTitle(false); } void FireDogEditor::slots_saveToFile() { if (this->featureLibrary == NULL || this->featureLibrary->items->size()==0) { QssMessageBox::warn("Please add the feature library first.", this, "Warn"); return; } QString fileName = QFileDialog::getSaveFileName(this, tr("Save To"), "", tr("Feature library Files (*.fdog)")); if (!fileName.isNull()) { if (!fileName.toLower().endsWith(".fdog")) { fileName.append(".fdog"); } //开始保存 QFile file(fileName); if (file.exists()) { QString titleStr = Gui18n::GetInstance()->GetConfig("text-message-box-title-question", "Confirm"); QString messageStr = Gui18n::GetInstance()->GetConfig("file-save-overwrite", "Whether to overwrite this file?"); QMessageBox::StandardButton button = QssMessageBox::question(this, titleStr, messageStr); if (button != QMessageBox::Ok && button != QMessageBox::Yes) { return; } } int state = -1; std::string json = this->featureLibrary->toJson(&state); if (state != NO_ERROR) { QString titleStr = Gui18n::GetInstance()->GetConfig("text-message-box-title-warning", "Warning"); QString messageStr = Gui18n::GetInstance()->GetConfig("file-save-fail-by-convertjson", "Save fail,Feature library convert to json fail!."); QssMessageBox::warn(messageStr, this, titleStr); return; } QString jsonstr = QString::fromUtf8(QByteArray::fromRawData(json.c_str(), json.size())); if (!file.open(QIODevice::WriteOnly | QIODevice::Text)) { QString titleStr = Gui18n::GetInstance()->GetConfig("text-message-box-title-warning", "Warning"); QString messageStr = Gui18n::GetInstance()->GetConfig("file-open-fail", "File Open fail!"); QssMessageBox::warn(messageStr, this, titleStr); return; } QTextStream in(&file); in << jsonstr; file.flush(); file.close(); this->openFilePath = fileName; ui.actionSave->setEnabled(true); updateWindowTitle(false); } } //打开文件 void FireDogEditor::slots_openFile() { //如果已经有打开或者正在编辑的特征库 if (this->featureLibrary != NULL) { QString titleStr = Gui18n::GetInstance()->GetConfig("text-message-box-title-questio", "Confirm"); QString messageStr = Gui18n::GetInstance()->GetConfig("file-open-has-opened", "A file has been opened, do you want to continue?"); QMessageBox::StandardButton botton = QMessageBox::question(this, titleStr, messageStr); if (botton != QMessageBox::Ok && botton != QMessageBox::Yes) { return; } delete this->featureLibrary; this->featureLibrary = NULL; } QString openFileDialogTitle = Gui18n::GetInstance()->GetConfig("file-open-dialog-title", "Open Feature Library File"); QString openFileDialogTip = Gui18n::GetInstance()->GetConfig("file-open-dialog-tip", "Feature Library"); openFileDialogTip.append("(*.fdog)"); QString filePath = QFileDialog::getOpenFileName(this, openFileDialogTitle, "", openFileDialogTip); if (!filePath.isEmpty()) { qDebug(filePath.toStdString().c_str()); this->parseThread->setFilePath(filePath); this->parseThread->start(); openFilePath = filePath; } } //搜索 void FireDogEditor::slots_featureLibraryTableSearch() { QString search = ui.lineEditSearchName->text(); loadFeatureLibraryTable(search); //fireDogFeatureInfoView->exec(); //QssMessageBox::tips("test tip", this, "Tips"); //QssMessageBox::warn("test warn", this, "Warn"); //QssMessageBox::error("test error", this, "Error"); //QssMessageBox::ask("test question", this, "Question"); } void FireDogEditor::slots_parseBinBegin() { this->loadingDialog->show(); } void FireDogEditor::slots_parseBinEnd(firedog::FeatureLibrary* featureLibrary, int state) { this->loadingDialog->loadingEnd(); if (state != 0) { QString titleStr = Gui18n::GetInstance()->GetConfig("text-message-box-title-warning", "Warning"); if (state == PARSE_ERROR_CODE_FILE_READ_FAIL) { QString messageStr = Gui18n::GetInstance()->GetConfig("file-parse-fail-cant-read", "Feature library file read failed."); QssMessageBox::warn(messageStr,this, titleStr); return; } if (state == PARSE_ERROR_CODE_FILE_PARSE_FAIL) { QString messageStr = Gui18n::GetInstance()->GetConfig("file-parse-fail-formate-error", "Feature library file format verification failed."); QssMessageBox::warn(messageStr, this, titleStr); return; } if (state == PARSE_ERROR_CODE_FILE_VERSION_CHECK_FAIL) { QString messageStr = Gui18n::GetInstance()->GetConfig("file-parse-fail-version-error", "Feature library file version verification failed"); QssMessageBox::warn(messageStr, this, titleStr); return; } QString messageStr = Gui18n::GetInstance()->GetConfig("file-parse-fail", "Feature library parse failed."); QssMessageBox::warn(messageStr, this, titleStr); return; } //清空之前的数据 if (this->featureLibrary!=NULL) { delete this->featureLibrary; this->featureLibrary = NULL; if (this->featureLibraryTableModel->rowCount()>0) { this->featureLibraryTableModel->removeRows(0, this->featureLibraryTableModel->rowCount()); } } //如果成功的话 this->featureLibrary = featureLibrary; //清空搜索框 ui.lineEditSearchName->setText(""); //加载数据 loadFeatureLibraryTable(""); //保存按钮和保存到按钮启用 ui.actionSave->setEnabled(true); ui.actionSaveTo->setEnabled(true); updateWindowTitle(false); } void FireDogEditor::loadFeatureLibraryTable(QString search) { if (this->featureLibrary == NULL|| this->featureLibrary->items->size()==0) { return; } QVector tableRows; for (int i = 0; i < featureLibrary->items->size(); i++) { FeatureLibraryItem* item = featureLibrary->items->at(i); //搜索 QString name = item->name.c_str(); QString describe = item->describe.c_str(); QString author = item->author.c_str(); if (!search.isEmpty()) { if (!name.contains(search) && !describe.contains(search) && !author.contains(search)) { continue; } } BigDataTableRow row; row.contents << BigDataTableCol(name) << BigDataTableCol(describe) << BigDataTableCol(author); tableRows.push_back(row); } this->featureLibraryTableModel->handleData(tableRows); } void FireDogEditor::slots_featureTableOpenMenu(QPoint pos) { if (this->isFeatureAdd) { this->featureLibraryTableMenuAddAction->setEnabled(false); } else { this->featureLibraryTableMenuAddAction->setEnabled(true); } int curRow = ui.tableViewLibrary->selectionModel()->currentIndex().row(); //选中行 if (curRow==-1) { featureLibraryTableMenuDelAction->setEnabled(false); } else { featureLibraryTableMenuDelAction->setEnabled(true); } auto index = ui.tableViewLibrary->indexAt(pos); featureLibraryTableMenu->exec(QCursor::pos()); // 菜单出现的位置为当前鼠标的位置 } void FireDogEditor::slots_featureInfoTableOpenMenu(QPoint pos) { int curRow = ui.tableViewLibraryInfoFeatures->selectionModel()->currentIndex().row(); //选中行 if (curRow == -1) { featureLibraryInfoFeatureTableMenuEditAction->setEnabled(false); featureLibraryInfoFeatureTableMenuDelAction->setEnabled(false); } else { featureLibraryInfoFeatureTableMenuEditAction->setEnabled(true); featureLibraryInfoFeatureTableMenuDelAction->setEnabled(true); } auto index = ui.tableViewLibraryInfoFeatures->indexAt(pos); featureLibraryInfoFeatureTableMenu->exec(QCursor::pos()); // 菜单出现的位置为当前鼠标的位置 } void FireDogEditor::slots_matchFilesTableOpenMenu(QPoint pos) { QModelIndexList selectList = ui.tableViewFiles->selectionModel()->selectedRows(); if (selectList.isEmpty()) { matchingFilesTableDelAction->setEnabled(false); } else { matchingFilesTableDelAction->setEnabled(true); } auto index = ui.tableViewFiles->indexAt(pos); matchingFilesTableMenu->exec(QCursor::pos()); // 菜单出现的位置为当前鼠标的位置 } void FireDogEditor::slots_featureInfoRuleTreeOpenMenu(QPoint pos) { QModelIndex currIndex = ui.treeViewLibraryInfoRules->currentIndex();; int curRow = currIndex.row(); //选中行 bool isKey = false; if (curRow != -1) { QStandardItem* item = this->featureLibraryInfoRuleTreeModel->itemFromIndex(currIndex); QString text = item->text(); if (text != "$and" && text != "$or") { isKey = true; } } if (isKey || (this->featureLibraryInfoRuleTreeModel->rowCount() != 0 && curRow == -1)) { this->featureLibraryInfoRuleTreeMenuAddAction->setEnabled(false); } else { this->featureLibraryInfoRuleTreeMenuAddAction->setEnabled(true); } if (curRow == -1) { featureLibraryInfoRuleTreeMenuEditAction->setEnabled(false); featureLibraryInfoRuleTreeMenuDelAction->setEnabled(false); } else { featureLibraryInfoRuleTreeMenuEditAction->setEnabled(true); featureLibraryInfoRuleTreeMenuDelAction->setEnabled(true); } auto index = ui.treeViewLibraryInfoRules->indexAt(pos); featureLibraryInfoRuleTreeMenu->exec(QCursor::pos()); // 菜单出现的位置为当前鼠标的位置 } void FireDogEditor::slots_selectFeatureTableEvent(const QModelIndex& current, const QModelIndex& previous) { clearInfoContent(false); int row = current.row(); BigDataTableRow bigDataTableRow = this->featureLibraryTableModel->getRowData(row); FeatureLibraryItem* item = this->featureLibrary->items->at(row); QString name = item->name.c_str(); QString author = item->author.c_str(); QString describe = item->describe.c_str(); ui.lineEditLibraryInfoName->setText(name); ui.lineEditLibraryInfoAuthor->setText(author); ui.textEditLibraryInfoDescribe->setText(describe); QVector tableRows; if (item->features != NULL) { for (int i = 0; i < item->features->size(); i++) { Feature* feature = item->features->at(i); QString key = feature->key.c_str(); QString type = !feature->hex.empty() ? "hex" : "text"; QString content = !feature->hex.empty() ? feature->hex.c_str() : feature->text.c_str(); BigDataTableRow row; row.contents << BigDataTableCol(key) << BigDataTableCol(type) << BigDataTableCol(content); tableRows.push_back(row); } } this->featureLibraryInfoFeatureTableModel->handleData(tableRows); if (item->rule != NULL) { QStandardItem* ruleItem = ruleToItem(item->rule); this->featureLibraryInfoRuleTreeModel->appendRow(ruleItem); ui.treeViewLibraryInfoRules->expandAll(); } } QStandardItem* FireDogEditor::ruleToItem(mountcloud::Rule* rule) { QStandardItem* ruleitem = new QStandardItem(); if (!rule->id.empty()) { ruleitem->setText(QString(rule->id.c_str())); } else { if (rule->ands != NULL) { ruleitem->setText(QStringLiteral("$and")); for (int i = 0; i < rule->ands->size(); i++) { mountcloud::Rule* andRule = rule->ands->at(i); QStandardItem* andRuleItem = ruleToItem(andRule); ruleitem->appendRow(andRuleItem); } } else if (rule->ors != NULL) { ruleitem->setText(QStringLiteral("$or")); for (int i = 0; i < rule->ors->size(); i++) { mountcloud::Rule* andRule = rule->ors->at(i); QStandardItem* andRuleItem = ruleToItem(andRule); ruleitem->appendRow(andRuleItem); } } else { ruleitem->setText(QStringLiteral("unknown")); } } return ruleitem; } void FireDogEditor::slots_featureTableMenuAddEvent() { clearInfoContent(true); } void FireDogEditor::slots_featureTableMenuDelEvent() { QString titleStr = Gui18n::GetInstance()->GetConfig("text-message-box-title-question", "Confirm"); QString messageStr = Gui18n::GetInstance()->GetConfig("feature-librarys-menu-delete-tip", "Please confirm to delete?"); QMessageBox::StandardButton botton = QMessageBox::question(this, titleStr, messageStr); if (botton != QMessageBox::Ok && botton != QMessageBox::Yes) { return; } clearInfoContent(true); int curRow = ui.tableViewLibrary->selectionModel()->currentIndex().row(); //选中行 FeatureLibraryItem* item = this->featureLibrary->items->at(curRow); delete item; item = NULL; this->featureLibrary->items->erase(this->featureLibrary->items->begin() + curRow); this->featureLibraryTableModel->removeRow(curRow); //标题加星 updateWindowTitle(true); } void FireDogEditor::slots_featureInfoTableMenuAddEvent() { firedog::Feature feature; QStringList existsKeys; int rowCount = this->featureLibraryInfoFeatureTableModel->rowCount(); for (int i = 0; i < rowCount; i++) { BigDataTableRow row = this->featureLibraryInfoFeatureTableModel->getRowData(i); QString key = row.contents[0].content.toString(); existsKeys << key; } bool state = this->fireDogFeatureInfoView->updateFeature(existsKeys,&feature); if (state) { QString key = feature.key.c_str(); QString type = !feature.hex.empty() ? "hex" : "text"; QString content = !feature.hex.empty() ? feature.hex.c_str() : feature.text.c_str(); BigDataTableRow row; row.contents << BigDataTableCol(key) << BigDataTableCol(type) << BigDataTableCol(content); this->featureLibraryInfoFeatureTableModel->addRow(row); } } void FireDogEditor::slots_featureInfoTableMenuEditEvent() { int curRow = ui.tableViewLibraryInfoFeatures->selectionModel()->currentIndex().row(); //选中行 BigDataTableRow row = this->featureLibraryInfoFeatureTableModel->getRowData(curRow); QString key = row.contents[0].content.toString(); QString type = row.contents[1].content.toString(); QString content = row.contents[2].content.toString(); firedog::Feature feature; feature.key = key.toStdString(); if (type == "hex") { feature.hex = content.toStdString(); } else { feature.text = content.toStdString(); } QStringList existsKeys; int rowCount = this->featureLibraryInfoFeatureTableModel->rowCount(); for (int i = 0; i < rowCount; i++) { if (i == curRow) { continue; } BigDataTableRow existrow = this->featureLibraryInfoFeatureTableModel->getRowData(i); QString key = existrow.contents[0].content.toString(); existsKeys << key; } bool isok = this->fireDogFeatureInfoView->updateFeature(existsKeys ,&feature); if (isok) { QString key = feature.key.c_str(); QString type = !feature.hex.empty() ? "hex" : "text"; QString content = !feature.hex.empty() ? feature.hex.c_str() : feature.text.c_str(); BigDataTableRow newrow; newrow.contents << BigDataTableCol(key) << BigDataTableCol(type) << BigDataTableCol(content); this->featureLibraryInfoFeatureTableModel->replaceRow(curRow, newrow); } } void FireDogEditor::slots_featureInfoTableMenuDelEvent() { QString titleStr = Gui18n::GetInstance()->GetConfig("text-message-box-title-question", "Confirm"); QString messageStr = Gui18n::GetInstance()->GetConfig("feature-info-features-menu-del-tep", "Please confirm to delete?"); QMessageBox::StandardButton botton = QMessageBox::question(this, titleStr, messageStr); if (botton != QMessageBox::Ok && botton != QMessageBox::Yes) { return; } int curRow = ui.tableViewLibraryInfoFeatures->selectionModel()->currentIndex().row(); this->featureLibraryInfoFeatureTableModel->removeRow(curRow); } void FireDogEditor::clearInfoContent(bool isFeatureAdd) { this->isFeatureAdd = isFeatureAdd; if (isFeatureAdd) { QString addText = Gui18n::GetInstance()->GetConfig("feature-info-btn-add", "Add"); ui.pushButtonLibraryInfoSave->setText(addText); } else { QString updateText = Gui18n::GetInstance()->GetConfig("feature-info-btn-update", "Update"); ui.pushButtonLibraryInfoSave->setText(updateText); } ui.lineEditLibraryInfoName->setText(""); ui.lineEditLibraryInfoAuthor->setText(""); ui.textEditLibraryInfoDescribe->setText(""); QVector tableRows; this->featureLibraryInfoFeatureTableModel->handleData(tableRows); this->featureLibraryInfoRuleTreeModel->clear(); } void FireDogEditor::slots_featureLibraryInfoRuleMenuAddEvent() { bool ruleIsEmpty = this->featureLibraryInfoRuleTreeModel->rowCount() == 0; bool isRoot = false; if (ruleIsEmpty) { isRoot = true; } QString rule; bool isOk = this->fireDogFeatureRuleInfoView->updateRule(&rule, isRoot); if (isOk) { QStandardItem* ruleitem = new QStandardItem(rule); //如果是空的 if (ruleIsEmpty) { featureLibraryInfoRuleTreeModel->appendRow(ruleitem); } else { QModelIndex currIndex = ui.treeViewLibraryInfoRules->currentIndex();; QStandardItem* selectItem = this->featureLibraryInfoRuleTreeModel->itemFromIndex(currIndex); selectItem->appendRow(ruleitem); } //清空选中 ui.treeViewLibraryInfoRules->selectionModel()->clear(); QModelIndex index = this->featureLibraryInfoRuleTreeModel->indexFromItem(ruleitem); ui.treeViewLibraryInfoRules->selectionModel()->setCurrentIndex(index, QItemSelectionModel::Select | QItemSelectionModel::Rows); } } void FireDogEditor::slots_featureLibraryInfoRuleMenuEditEvent() { QModelIndex currIndex = ui.treeViewLibraryInfoRules->currentIndex();; QStandardItem* selectItem = this->featureLibraryInfoRuleTreeModel->itemFromIndex(currIndex); bool isRoot = selectItem->parent() == NULL; QString rule = selectItem->text(); bool isOk = this->fireDogFeatureRuleInfoView->updateRule(&rule, isRoot); if (isOk) { selectItem->setText(rule); } } void FireDogEditor::slots_featureLibraryInfoRuleMenuDelEvent() { QString titleStr = Gui18n::GetInstance()->GetConfig("text-message-box-title-question", "Confirm"); QString messageStr = Gui18n::GetInstance()->GetConfig("feature-info-rule-menu-del-tip", "Please confirm to delete?"); QMessageBox::StandardButton botton = QMessageBox::question(this, titleStr, messageStr); if (botton != QMessageBox::Ok && botton != QMessageBox::Yes) { return; } QModelIndex currIndex = ui.treeViewLibraryInfoRules->currentIndex();; this->featureLibraryInfoRuleTreeModel->removeRow(currIndex.row(),currIndex.parent()); } void FireDogEditor::slots_saveBtnClickEvent() { firedog::FeatureLibraryItem* item = getInfoViewFeatureItem(); if (item==NULL) { return; } if (this->featureLibrary==NULL) { this->featureLibrary = new FeatureLibrary(); } //搜索 QString name = item->name.c_str(); QString describe = item->describe.c_str(); QString author = item->author.c_str(); BigDataTableRow row; row.contents << BigDataTableCol(name) << BigDataTableCol(describe) << BigDataTableCol(author); //如果是新增 if (this->isFeatureAdd) { this->featureLibrary->items->push_back(item); this->featureLibraryTableModel->addRow(row); } //如果是更新 else { int rowNum = ui.tableViewLibrary->currentIndex().row(); //替换掉旧的特征 firedog::FeatureLibraryItem* olditem = this->featureLibrary->items->at(rowNum); //先赋值 this->featureLibrary->items->at(rowNum) = item; //再删内存 delete olditem; olditem = NULL; this->featureLibraryTableModel->replaceRow(rowNum, row); } //保存到按钮亮起来 ui.actionSaveTo->setEnabled(true); //并且标题加星 updateWindowTitle(true); clearInfoContent(true); } firedog::FeatureLibraryItem* FireDogEditor::getInfoViewFeatureItem() { QString name = ui.lineEditLibraryInfoName->text(); QString auth = ui.lineEditLibraryInfoAuthor->text(); QString describe = ui.textEditLibraryInfoDescribe->toPlainText(); int featureCount = this->featureLibraryInfoFeatureTableModel->rowCount(); int ruleCount = this->featureLibraryInfoRuleTreeModel->rowCount(); QString titleStr = Gui18n::GetInstance()->GetConfig("text-message-box-title-warning", "Warning"); if (name.isEmpty()) { QString messageStr = Gui18n::GetInstance()->GetConfig("feature-info-add-please-enter-name", "Please enter [Name]."); QssMessageBox::warn(messageStr, this, titleStr); return NULL; } if (auth.isEmpty()) { QString messageStr = Gui18n::GetInstance()->GetConfig("feature-info-add-please-enter-auth", "Please enter [Auth]."); QssMessageBox::warn(messageStr, this, titleStr); return NULL; } if (describe.isEmpty()) { QString messageStr = Gui18n::GetInstance()->GetConfig("feature-info-add-please-enter-describe", "Please enter [Describe]."); QssMessageBox::warn(messageStr, this, titleStr); return NULL; } if (featureCount == 0) { QString messageStr = Gui18n::GetInstance()->GetConfig("feature-info-add-please-enter-feature", "Please add [Feature]."); QssMessageBox::warn(messageStr, this, titleStr); return NULL; } if (ruleCount == 0) { QString messageStr = Gui18n::GetInstance()->GetConfig("feature-info-add-please-enter-feature", "Please add [Rule]."); QssMessageBox::warn(messageStr, this, titleStr); return NULL; } //校验规则 QStandardItem* rootItem = featureLibraryInfoRuleTreeModel->item(0); mountcloud::Rule* rule = itemToRule(rootItem); if (rule == NULL) { return NULL; } firedog::FeatureLibraryItem* result = new firedog::FeatureLibraryItem(); result->name = name.toStdString(); result->describe = describe.toStdString(); result->author = auth.toStdString(); //特征 for (int i = 0; i < featureCount; i++) { BigDataTableRow row = this->featureLibraryInfoFeatureTableModel->getRowData(i); QString key = row.contents[0].content.toString(); QString type = row.contents[1].content.toString(); QString content = row.contents[2].content.toString(); Feature* feature = new Feature(); feature->key = key.toStdString(); if (type=="hex") { feature->hex = content.toStdString(); } else { feature->text = content.toStdString(); } result->features->push_back(feature); } //规则 result->rule = rule; return result; } mountcloud::Rule* FireDogEditor::itemToRule(QStandardItem* item) { if (item == NULL) { return NULL; } QString errorMessage; QString text = item->text(); if (text.isEmpty()) { errorMessage = Gui18n::GetInstance()->GetConfig("feature-info-add-rule-error", "This rule is wrong, please fix it and try again."); goto fail; } bool isOperation = text == "$and" || text == "$or"; //根规则只能是逻辑表达式 if (item->parent()==NULL) { if (!isOperation) { errorMessage = Gui18n::GetInstance()->GetConfig("feature-info-add-rule-root-error", "The root rule can only be $and or $or."); goto fail; } } //操作下面必须有节点 if (isOperation) { if (item->rowCount() == 0) { errorMessage = Gui18n::GetInstance()->GetConfig("feature-info-add-rule-childs-error", "The rule childs is empty.");; goto fail; } } mountcloud::Rule* result = new mountcloud::Rule(); if (isOperation) { int childCount = item->rowCount(); for (int i = 0; i < childCount; i++) { QStandardItem* childItem = item->child(i); mountcloud::Rule* childRule = itemToRule(childItem); if (childRule==NULL) { delete result; return NULL; } if(text=="$and"){ if (result->ands == NULL) { result->ands = new std::vector(); } result->ands->push_back(childRule); } else { if (result->ors == NULL) { result->ors = new std::vector(); } result->ors->push_back(childRule); } } } else { result->id = text.toStdString(); } return result; fail: ui.treeViewLibraryInfoRules->selectionModel()->clear(); QModelIndex index = this->featureLibraryInfoRuleTreeModel->indexFromItem(item); ui.treeViewLibraryInfoRules->selectionModel()->setCurrentIndex(index, QItemSelectionModel::Select | QItemSelectionModel::Rows); QString titleStr = Gui18n::GetInstance()->GetConfig("text-message-box-title-warning", "Warning"); QssMessageBox::warn(errorMessage, this, titleStr); return NULL; } void FireDogEditor::slots_matchingAddFilesBtnClickEvent() { QString messageStr = Gui18n::GetInstance()->GetConfig("matching-input-files-select-file", "Select Files"); QStringList files = QFileDialog::getOpenFileNames(this, messageStr,"",""); QStringList nowFiles; int count = this->matchingFilesTableModel->rowCount(); if (count > 0) { for (int i = 0;i < this->matchingFilesTableModel->rowCount(); i++) { BigDataTableRow row = this->matchingFilesTableModel->getRowData(i); nowFiles << row.contents.at(0).content.toString(); } } if (files.size() > 0) { for (int i = 0; i < files.size(); i++) { QString filePath = files.at(i); if (nowFiles.contains(filePath)) { continue; } QFile file(filePath); QString sizeStr = "UNKNOWN"; if (file.exists()) { qint64 size = file.size(); sizeStr = getFileSizeString(size); } BigDataTableRow row; row.contents << BigDataTableCol(filePath) << BigDataTableCol(sizeStr); this->matchingFilesTableModel->addRow(row); } } } QString FireDogEditor::getFileSizeString(const qint64& size) { int integer = 0; //整数位 int decimal = 0; //小数位，保留三位 char unit = 'B'; qint64 standardSize = size; qint64 curSize = size; if (standardSize > 1024) { curSize = standardSize * 1000; curSize /= 1024; integer = curSize / 1000; decimal = curSize % 1000; standardSize /= 1024; unit = 'K'; if (standardSize > 1024) { curSize = standardSize * 1000; curSize /= 1024; integer = curSize / 1000; decimal = curSize % 1000; standardSize /= 1024; unit = 'M'; if (standardSize > 1024) { curSize = standardSize * 1000; curSize /= 1024; integer = curSize / 1000; decimal = curSize % 1000; unit = 'G'; } } } else { integer = size; } QString dec = "0"; if (0 <= decimal && decimal <= 9) { dec = dec + dec + QString::number(decimal); } if (10 <= decimal && decimal <= 99) { dec = "0" + QString::number(decimal); } if (100 <= decimal && decimal <= 999) { dec = QString::number(decimal); } return QString::number(integer) + "." + dec + unit; } void FireDogEditor::slots_matchingFilesDeleteEvent() { QModelIndexList selectList = ui.tableViewFiles->selectionModel()->selectedRows(); for (int i = 0; i < selectList.size(); i++) { QModelIndex index = selectList.at(i); int row = index.row(); matchingFilesTableModel->removeRow(row); } } void FireDogEditor::slots_matchingBtnClick() { if (this->featureLibrary == NULL || this->featureLibrary->items->size() == 0) { QssMessageBox::warn("Please add feature library first.", this, "Warn"); return; } //std::vector parts = PartUtil::getParts(101, 10); QString text = ui.plainTextText->toPlainText(); QString hex = ui.plainTextHex->toPlainText(); text = text.trimmed(); hex = hex.trimmed(); QStringList files; int count = this->matchingFilesTableModel->rowCount(); if (count > 0) { for (int i = 0; i < this->matchingFilesTableModel->rowCount(); i++) { BigDataTableRow row = this->matchingFilesTableModel->getRowData(i); files << row.contents.at(0).content.toString(); } } if (text.isEmpty() && hex.isEmpty() && files.isEmpty()) { QString titleStr = Gui18n::GetInstance()->GetConfig("text-message-box-title-warning", "Warning"); QString messageStr = Gui18n::GetInstance()->GetConfig("matching-please-enter", "Please enter text or hex or add a file."); QssMessageBox::warn(messageStr, this, titleStr); return; } this->loadingDialog->show(); ; int sourceId = 0; int threadNum = ui.spinBoxThreadNum->value(); int ge100MSlice = ui.spinBox100MB->value(); int ge1GSlice = ui.spinBox1GB->value(); ui.pushButtonMatch->setEnabled(false); if (matchButlerThread != NULL) { disconnect(matchButlerThread, &MatchButlerThread::matchEnd, this, &FireDogEditor::slots_matchEnd); delete matchButlerThread; matchButlerThread = NULL; } matchButlerThread = new MatchButlerThread(threadNum, this->featureLibrary);; connect(matchButlerThread, &MatchButlerThread::matchEnd, this, &FireDogEditor::slots_matchEnd); //如果text不是空 if (!text.isEmpty()) { int nowSourceId = sourceId = sourceId + 1; MatchWork work; work.sourceId = nowSourceId; work.workType = WORK_TYPE_TEXT; work.content = text; matchButlerThread->addWork(work); } //如果hex不是空 if (!hex.isEmpty()) { int nowSourceId = sourceId = sourceId + 1; MatchWork work; work.sourceId = nowSourceId; work.workType = WORK_TYPE_HEX; work.content = hex; matchButlerThread->addWork(work); } //如果files不是空 if (!files.isEmpty()) { qint64 fs100M = 1024 * 1024 * 100; qint64 fs1G = 1024 * 1024 * 1024; for (int i = 0; i < files.size(); i++) { QString filePath = files.at(i); int nowSourceId = sourceId = sourceId + 1; QFile file(filePath); if (!file.exists()) { continue; } qint64 fileSize = file.size(); if (fileSize < fs100M) { MatchWork work; work.sourceId = nowSourceId; work.workType = WORK_TYPE_FILE; work.content = filePath; matchButlerThread->addWork(work); } else{ int partSize = 0; if (fileSize < fs1G) { partSize = ge100MSlice; } else { partSize = ge1GSlice; } std::vector parts = PartUtil::getParts(fileSize, partSize); for (int y = 0; y < parts.size(); y++) { Part part = parts.at(y); if (y != 0) { //向前推点 part.start = part.start - 256; } MatchWork work; work.sourceId = nowSourceId; work.workType = WORK_TYPE_PART; work.content = filePath; work.part = part; matchButlerThread->addWork(work); } } } } //开始任务 matchButlerThread->start(); } void FireDogEditor::slots_matchEnd(int worksize, int success, int error, int state) { this->loadingDialog->hide(); ui.pushButtonMatch->setEnabled(true); this->matchResultHits.clear(); if (state == 0) { QVector hits = this->matchButlerThread->getHits(); if (hits.size()>0) { for (int i = 0; i < hits.size(); i++) { HitFeature hit = hits.at(i); this->matchResultHits.append(hit); } } } loadMatchResult(""); } void FireDogEditor::loadMatchResult(QString search) { QVector rows; if (this->matchResultHits.size() > 0) { for (int i = 0; i < this->matchResultHits.size(); i++) { HitFeature hit = this->matchResultHits.at(i); BigDataTableRow row; QString source; if (hit.workType == WORK_TYPE_TEXT) { source = "text"; } else if (hit.workType == WORK_TYPE_HEX) { source = "hex"; } else { source = hit.content; } QString featureName = hit.name; QString featureDescribe = hit.describe; QString featureAuthor = hit.author; if (!search.isEmpty()) { search = search.trimmed(); if (!source.contains(search) && !featureName.contains(search) && !featureDescribe.contains(search) && !featureAuthor.contains(search)) { continue; } } row.contents << BigDataTableCol(source) << BigDataTableCol(featureName) << BigDataTableCol(featureDescribe) << BigDataTableCol(featureAuthor); rows << row; } } this->matchingResultTableModel->handleData(rows); } void FireDogEditor::slots_searchMatchResultBtnClick() { QString search = ui.lineEditSearchResult->text(); loadMatchResult(search); } FireDogEditor::~FireDogEditor() { if (featureLibraryTableModel!=NULL) { delete featureLibraryTableModel; featureLibraryTableModel = NULL; } if (featureLibraryTableMenu != NULL) { delete featureLibraryTableMenu; featureLibraryTableMenu = NULL; } if (featureLibraryTableMenuAddAction != NULL) { delete featureLibraryTableMenuAddAction; featureLibraryTableMenuAddAction = NULL; } if (featureLibraryTableMenuDelAction != NULL) { delete featureLibraryTableMenuDelAction; featureLibraryTableMenuDelAction = NULL; } if (featureLibraryInfoFeatureTableModel != NULL) { delete featureLibraryInfoFeatureTableModel; featureLibraryInfoFeatureTableModel = NULL; } if (loadingDialog != NULL) { delete loadingDialog; loadingDialog = NULL; } if (featureLibrary != NULL) { delete featureLibrary; featureLibrary = NULL; } if (featureLibraryInfoRuleTreeModel != NULL) { delete featureLibraryInfoRuleTreeModel; featureLibraryInfoRuleTreeModel = NULL; } if (featureLibraryInfoFeatureTableMenu != NULL) { delete featureLibraryInfoFeatureTableMenu; featureLibraryInfoFeatureTableMenu = NULL; } if (featureLibraryInfoFeatureTableMenuAddAction != NULL) { delete featureLibraryInfoFeatureTableMenuAddAction; featureLibraryInfoFeatureTableMenuAddAction = NULL; } if (featureLibraryInfoFeatureTableMenuEditAction != NULL) { delete featureLibraryInfoFeatureTableMenuEditAction; featureLibraryInfoFeatureTableMenuEditAction = NULL; } if (featureLibraryInfoFeatureTableMenuDelAction != NULL) { delete featureLibraryInfoFeatureTableMenuDelAction; featureLibraryInfoFeatureTableMenuDelAction = NULL; } if (matchingFilesTableModel != NULL) { delete matchingFilesTableModel; matchingFilesTableModel = NULL; } if (matchingFilesTableMenu != NULL) { delete matchingFilesTableMenu; matchingFilesTableMenu = NULL; } if (matchingFilesTableDelAction != NULL) { delete matchingFilesTableDelAction; matchingFilesTableDelAction = NULL; } if (matchingResultTableModel != NULL) { delete matchingResultTableModel; matchingResultTableModel = NULL; } } void FireDogEditor::resizeEvent(QResizeEvent* event) { int width = this->width(); int height = this->height(); this->loadingDialog->setFixedSize(QSize(width, height)); } void FireDogEditor::slots_reboot() { qApp->exit(EXIT_CODE_REBOOT); }