Diagnostics support

Author: connor-ms

src/rage/script.cpp

@@ -12,9 +12,10 @@
 #define ReadVar(x) stream >> x;
 #define ReadPointer(x) stream >> x; x = x & 0xffffff;
 
-Script::Script(QString path)
+Script::Script(QString path, bool debug)
     : m_funcCount(0)
     , m_script(path)
+    , m_debug(debug)
 {
     if (!m_script.open(QIODevice::ReadOnly))
     {
@@ -95,6 +96,15 @@ void Script::extractData()
 
         int outSize = m_header.getSizeP() + m_header.getSizeV();
 
+        if (m_debug)
+        {
+            QFileInfo info(m_script);
+            QFile out("debug/" + info.fileName() + ".dbg");
+
+            out.open(QIODevice::WriteOnly | QIODevice::Truncate);
+            out.write(m_data);
+        }
+
         if (m_scriptType == ScriptType::TYPE_X360)
         {
             // remove lzx header

src/rage/script.h

@@ -57,7 +57,7 @@ enum ScriptType
 class Script
 {
 public:
-    Script(QString path);
+    Script(QString path, bool debug = false);
 
     ScriptType getScriptType() { return m_scriptType; }
 
@@ -123,6 +123,7 @@ class Script
     QByteArray m_data;
     QFile m_script;
     ScriptType m_scriptType;
+    bool m_debug;
 };
 
 #endif // SCRIPT_H

src/widgets/disassembler.cpp

@@ -12,11 +12,12 @@
 #include "../QHexView/qhexview.h"
 #include "../QHexView/document/buffer/qmemoryrefbuffer.h"
 
-Disassembler::Disassembler(QString file, QWidget *parent)
+Disassembler::Disassembler(QString file, bool debug, QWidget *parent)
     : QMainWindow(parent)
     , m_ui(new Ui::Disassembler)
-    , m_script(file)
+    , m_script(file, debug)
     , m_file(file)
+    , m_debug(debug)
 {
     m_ui->setupUi(this);
 
@@ -136,7 +137,7 @@ void Disassembler::open()
     {
         QApplication::setOverrideCursor(Qt::WaitCursor);
 
-        Disassembler *dsm = new Disassembler(file);
+        Disassembler *dsm = new Disassembler(file, m_debug);
         dsm->show();
 
         close();
@@ -319,18 +320,13 @@ QTableWidget *Disassembler::createStringsTab()
     return stringTable;
 }
 
-#include <QDebug>
-
 void Disassembler::fillDisassembly()
 {
     int invalidCalls = 0;
     bool firstFunc = true;
 
     for (auto op : m_script.getOpcodes())
     {
-        //if (op->getOp() == EOpcodes::OP_NOP)
-        //    continue;
-
         int index = m_disasm->rowCount();
 
         if (op->getOp() == EOpcodes::_SPACER)
@@ -424,7 +420,7 @@ void Disassembler::fillDisassembly()
         else if (op->getOp() >= EOpcodes::OP_JMP && op->getOp() <= EOpcodes::OP_JMPGT)
         {
             int jumpPos = op->getData()[1] + op->getLocation() + 3;
-            //data->setText("@" + m_script.getJumps().at(jumpPos)->getSub());
+
             data->setText("@" + m_script.getJumps().at(jumpPos)->getSub());
             data->setForeground(QColor(255, 0, 0));
         }

src/widgets/disassembler.h

@@ -19,7 +19,7 @@ class Disassembler : public QMainWindow
     Q_OBJECT
 
 public:
-    explicit Disassembler(QString file, QWidget *parent = nullptr);
+    explicit Disassembler(QString file, bool debug, QWidget *parent = nullptr);
     ~Disassembler();
 
 public slots:
@@ -50,6 +50,7 @@ public slots:
     Script m_script;
     QString m_file;
     OpcodeTable *m_disasm;
+    bool m_debug;
 };
 
 #endif // DISASSEMBLER_H

src/widgets/launchscreen.cpp

@@ -21,6 +21,8 @@ LaunchScreen::LaunchScreen(QWidget *parent) :
         m_ui->btnOpenFile->setEnabled(false);
     }
 
+    QDir().mkdir("debug");
+
     connect(m_ui->btnOpenFile, SIGNAL(clicked()), this, SLOT(openFilePressed()));
     connect(m_ui->btnGithub,   SIGNAL(clicked()), this, SLOT(githubPressed()));
     connect(m_ui->btnExit,     SIGNAL(clicked()), this, SLOT(exitPressed()));
@@ -39,7 +41,7 @@ void LaunchScreen::openFilePressed()
     {
         QApplication::setOverrideCursor(Qt::WaitCursor);
 
-        Disassembler *dsm = new Disassembler(file);
+        Disassembler *dsm = new Disassembler(file, m_ui->cbDebug->isChecked());
         dsm->show();
 
         close();

src/widgets/launchscreen.ui

@@ -43,7 +43,7 @@
    <property name="font">
     <font>
      <family>Consolas</family>
-     <pointsize>19</pointsize>
+     <pointsize>18</pointsize>
     </font>
    </property>
    <property name="text">
@@ -92,6 +92,25 @@
     <string>Exit</string>
    </property>
   </widget>
+  <widget class="QCheckBox" name="cbDebug">
+   <property name="geometry">
+    <rect>
+     <x>110</x>
+     <y>310</y>
+     <width>91</width>
+     <height>20</height>
+    </rect>
+   </property>
+   <property name="font">
+    <font>
+     <family>Consolas</family>
+     <pointsize>8</pointsize>
+    </font>
+   </property>
+   <property name="text">
+    <string>Debug mode</string>
+   </property>
+  </widget>
  </widget>
  <resources/>
  <connections/>