From 526bd89873659affbe8b8d4de299b3b66a1fa8a0 Mon Sep 17 00:00:00 2001
From: wh201906 <wh201906@yandex.com>
Date: Wed, 29 Apr 2020 17:50:52 +0800
Subject: [PATCH] Add running state

---
 ui/mainwindow.cpp | 46 ++++++++++++++++++++++++++++++++++++++++++++++
 ui/mainwindow.h   |  1 +
 ui/mainwindow.ui  |  2 +-
 3 files changed, 48 insertions(+), 1 deletion(-)

diff --git a/ui/mainwindow.cpp b/ui/mainwindow.cpp
index 3fdf3f2..f631bcb 100644
--- a/ui/mainwindow.cpp
+++ b/ui/mainwindow.cpp
@@ -36,6 +36,7 @@ void MainWindow::initUI() // will be called by main.app
     ui->retranslateUi(this);
     uiInit();
     signalInit();
+    setState(false);
 }
 
 // ******************** basic functions ********************
@@ -78,6 +79,7 @@ void MainWindow::on_PM3_connectButton_clicked()
 void MainWindow::onPM3StateChanged(bool st, QString info)
 {
     pm3state = st;
+    setState(st);
     if(st == true)
     {
         setStatusBar(PM3VersionBar, info);
@@ -92,6 +94,7 @@ void MainWindow::onPM3StateChanged(bool st, QString info)
 void MainWindow::on_PM3_disconnectButton_clicked()
 {
     pm3state = false;
+    setState(false);
     emit killPM3();
     emit setSerialListener("", false);
     setStatusBar(connectStatusBar, tr("Not Connected"));
@@ -344,12 +347,16 @@ void MainWindow::on_MF_Attack_infoButton_clicked()
 
 void MainWindow::on_MF_Attack_chkButton_clicked()
 {
+    setState(false);
     mifare->chk();
+    setState(true);
 }
 
 void MainWindow::on_MF_Attack_nestedButton_clicked()
 {
+    setState(false);
     mifare->nested();
+    setState(true);
 }
 
 void MainWindow::on_MF_Attack_hardnestedButton_clicked()
@@ -359,22 +366,30 @@ void MainWindow::on_MF_Attack_hardnestedButton_clicked()
 
 void MainWindow::on_MF_RW_readAllButton_clicked()
 {
+    setState(false);
     mifare->readAll();
+    setState(true);
 }
 
 void MainWindow::on_MF_RW_readBlockButton_clicked()
 {
+    setState(false);
     mifare->read();
+    setState(true);
 }
 
 void MainWindow::on_MF_RW_writeBlockButton_clicked()
 {
+    setState(false);
     mifare->write();
+    setState(true);
 }
 
 void MainWindow::on_MF_RW_writeAllButton_clicked()
 {
+    setState(false);
     mifare->writeAll();
+    setState(true);
 }
 
 void MainWindow::on_MF_RW_dumpButton_clicked()
@@ -389,22 +404,30 @@ void MainWindow::on_MF_RW_restoreButton_clicked()
 
 void MainWindow::on_MF_UID_readAllButton_clicked()
 {
+    setState(false);
     mifare->readAllC();
+    setState(true);
 }
 
 void MainWindow::on_MF_UID_readBlockButton_clicked()
 {
+    setState(false);
     mifare->readC();
+    setState(true);
 }
 
 void MainWindow::on_MF_UID_writeAllButton_clicked()
 {
+    setState(false);
     mifare->writeAllC();
+    setState(true);
 }
 
 void MainWindow::on_MF_UID_writeBlockButton_clicked()
 {
+    setState(false);
     mifare->writeC();
+    setState(true);
 }
 
 void MainWindow::on_MF_UID_wipeButton_clicked()
@@ -435,7 +458,9 @@ void MainWindow::on_MF_UID_aboutUIDButton_clicked()
 
 void MainWindow::on_MF_UID_setParaButton_clicked()
 {
+    setState(false);
     mifare->setParameterC();
+    setState(true);
 }
 
 void MainWindow::on_MF_UID_lockButton_clicked()
@@ -445,7 +470,9 @@ void MainWindow::on_MF_UID_lockButton_clicked()
 
 void MainWindow::on_MF_Sniff_sniffButton_clicked()
 {
+    setState(false);
     mifare->sniff();
+    setState(true);
 }
 
 void MainWindow::on_MF_Sniff_listButton_clicked()
@@ -599,4 +626,23 @@ bool MainWindow::eventFilter(QObject *watched, QEvent *event) // drag support
     return QMainWindow::eventFilter(watched, event);
 }
 
+void MainWindow::setState(bool st)
+{
+    if(!st && pm3state)
+    {
+        setStatusBar(programStatusBar, tr("Running"));
+    }
+    else
+    {
+        setStatusBar(programStatusBar, tr("Idle"));
+    }
+    ui->MF_attackGroupBox->setEnabled(st);
+    ui->MF_normalGroupBox->setEnabled(st);
+    ui->MF_UIDGroupBox->setEnabled(st);
+    ui->MF_simGroupBox->setEnabled(st);
+    ui->MF_sniffGroupBox->setEnabled(st);
+    ui->Raw_CMDEdit->setEnabled(st);
+    ui->Raw_sendCMDButton->setEnabled(st);
+}
+
 // ***********************************************
diff --git a/ui/mainwindow.h b/ui/mainwindow.h
index 70dd0a1..1d892ac 100644
--- a/ui/mainwindow.h
+++ b/ui/mainwindow.h
@@ -137,6 +137,7 @@ private:
     void signalInit();
     void MF_widgetReset();
     void setTableItem(QTableWidget *widget, int row, int column, const QString &text);
+    void setState(bool st);
 signals:
     void connectPM3(const QString path, const QString port);
     void killPM3();
diff --git a/ui/mainwindow.ui b/ui/mainwindow.ui
index 9198235..356256a 100644
--- a/ui/mainwindow.ui
+++ b/ui/mainwindow.ui
@@ -618,7 +618,7 @@
               </spacer>
              </item>
              <item>
-              <widget class="QGroupBox" name="MF_NormalGroupBox">
+              <widget class="QGroupBox" name="MF_normalGroupBox">
                <property name="title">
                 <string>Normal(Require Password)</string>
                </property>