Add record script feature

Signed-off-by: xiaoming <2014500726@smail.xtu.edu.cn>
QQxiaoming · Sep 4, 2024 · e4d1402 · e4d1402
1 parent 42a6bab
commit e4d1402
Show file tree

Hide file tree

Showing 5 changed files with 279 additions and 163 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -7,12 +7,16 @@ en-US:
 - Automatically use the default configuration if the Profile does not exist on Windows
 - Add ToolTip to the connection bar
 - Add system beep support
+- Add record script feature
+- Fix the small probability memory leak problem
 
 zh-CN:
 
 - Windows下Profile如果不存在则自动使用默认配置
 - 增加连接条ToolTip显示
 - 增加系统响铃支持
+- 增加记录脚本功能
+- 修复可能存在的小概率内存泄漏问题
 
 ## [[V0.5.0](https://github.com/QQxiaoming/quardCRT/releases/tag/V0.5.0)] - 2024-08-26
 

diff --git a/docs/changelog.md b/docs/changelog.md
@@ -7,6 +7,8 @@
 - Automatically use the default configuration if the Profile does not exist on Windows
 - Add ToolTip to the connection bar
 - Add system beep support
+- Add record script feature
+- Fix the small probability memory leak problem
 
 ## [[V0.5.0](https://github.com/QQxiaoming/quardCRT/releases/tag/V0.5.0)] - 2024-08-26
 

diff --git a/src/mainwindow.cpp b/src/mainwindow.cpp
@@ -371,6 +371,10 @@ CentralWidget::CentralWidget(QString dir, StartupUIMode mode, QLocale lang, bool
                     if(sessionsWindow->getMainWidget() == widget) {
                         logSessionAction->setChecked(sessionsWindow->isLog());
                         rawLogSessionAction->setChecked(sessionsWindow->isRawLog());
+                        bool isRS = sessionsWindow->isRecordingScript();
+                        startRecordingScriptAction->setEnabled(!isRS);
+                        stopRecordingScriptAction->setEnabled(isRS);
+                        canlcelRecordingScriptAction->setEnabled(isRS);
                         if(hexViewAction->isChecked()) {
                             connect(sessionsWindow,&SessionsWindow::hexDataDup,
                                         hexViewWindow,&HexViewWindow::recvData);
@@ -938,6 +942,9 @@ CentralWidget::CentralWidget(QString dir, StartupUIMode mode, QLocale lang, bool
 #else
     runAction->setEnabled(false);
     cancelAction->setEnabled(false);
+    startRecordingScriptAction->setEnabled(false);
+    stopRecordingScriptAction->setEnabled(false);
+    canlcelRecordingScriptAction->setEnabled(false);
 #endif
     #if defined(Q_OS_MACOS)
     if(mainWindow) {
@@ -1126,9 +1133,6 @@ CentralWidget::CentralWidget(QString dir, StartupUIMode mode, QLocale lang, bool
     });
 
     // TODO:Unimplemented functions are temporarily closed
-    startRecordingScriptAction->setEnabled(false);
-    stopRecordingScriptAction->setEnabled(false);
-    canlcelRecordingScriptAction->setEnabled(false);
     createPublicKeyAction->setEnabled(false);
     publickeyManagerAction->setEnabled(false);
     tileAction->setEnabled(false);
@@ -2195,8 +2199,10 @@ void CentralWidget::menuAndToolBarInit(void) {
     startRecordingScriptAction = new QAction(this);
     scriptMenu->addAction(startRecordingScriptAction);
     stopRecordingScriptAction = new QAction(this);
+    stopRecordingScriptAction->setEnabled(false);
     scriptMenu->addAction(stopRecordingScriptAction);
     canlcelRecordingScriptAction = new QAction(this);
+    canlcelRecordingScriptAction->setEnabled(false);
     scriptMenu->addAction(canlcelRecordingScriptAction);
 
     addBookmarkAction = new QAction(this);
@@ -3369,6 +3375,48 @@ void CentralWidget::menuAndToolBarConnectSignals(void) {
     connect(cancelAction,&QAction::triggered,this,[=](){
         pyRun->cancelScript();
     });
+    connect(startRecordingScriptAction,&QAction::triggered,this,[=](){
+        QWidget *widget = findCurrentFocusWidget();
+        if(widget == nullptr) {
+            return;
+        }
+        SessionsWindow *sessionsWindow = widget->property("session").value<SessionsWindow *>();
+        if(!sessionsWindow->isRecordingScript()) {
+            sessionsWindow->startRecordingScript();
+        }
+        bool isRS = sessionsWindow->isRecordingScript();
+        startRecordingScriptAction->setEnabled(!isRS);
+        stopRecordingScriptAction->setEnabled(isRS);
+        canlcelRecordingScriptAction->setEnabled(isRS);
+    });
+    connect(stopRecordingScriptAction,&QAction::triggered,this,[=](){
+        QWidget *widget = findCurrentFocusWidget();
+        if(widget == nullptr) {
+            return;
+        }
+        SessionsWindow *sessionsWindow = widget->property("session").value<SessionsWindow *>();
+        if(sessionsWindow->isRecordingScript()) {
+            sessionsWindow->stopRecordingScript();
+        }
+        bool isRS = sessionsWindow->isRecordingScript();
+        startRecordingScriptAction->setEnabled(!isRS);
+        stopRecordingScriptAction->setEnabled(isRS);
+        canlcelRecordingScriptAction->setEnabled(isRS);
+    });
+    connect(canlcelRecordingScriptAction,&QAction::triggered,this,[=](){
+        QWidget *widget = findCurrentFocusWidget();
+        if(widget == nullptr) {
+            return;
+        }
+        SessionsWindow *sessionsWindow = widget->property("session").value<SessionsWindow *>();
+        if(sessionsWindow->isRecordingScript()) {
+            sessionsWindow->canlcelRecordingScript();
+        }
+        bool isRS = sessionsWindow->isRecordingScript();
+        startRecordingScriptAction->setEnabled(!isRS);
+        stopRecordingScriptAction->setEnabled(isRS);
+        canlcelRecordingScriptAction->setEnabled(isRS);
+    });
 #endif
     connect(addBookmarkAction, &QAction::triggered, this, [&]() {
         QString path = FileDialog::getExistingDirectory(this,tr("Select a directory"),QDir::homePath());