update

Icingworld · Icingworld · commit 50e8b5d283d1 · 2023-06-25T11:12:44.000+08:00
diff --git a/Steganalysis.pro b/Steganalysis.pro
@@ -8,6 +8,10 @@ CONFIG += c++11
 # In order to do so, uncomment the following line.
 #DEFINES += QT_DISABLE_DEPRECATED_BEFORE=0x060000    # disables all the APIs deprecated before Qt 6.0.0
 
+win32-msvc* {
+    QMAKE_CXXFLAGS += /source-charset:utf-8 /execution-charset:utf-8
+}
+
 SOURCES += \
     lsb.cpp \
     main.cpp \
@@ -36,4 +40,5 @@ RESOURCES += \
 DISTFILES += \
     logo.rc
 
-RC_FILE += logo.rc
+RC_FILE += \
+    logo.rc
diff --git a/lsb.cpp b/lsb.cpp
@@ -1,4 +1,4 @@
-﻿#include "lsb.h"
+#include "lsb.h"
 
 LSB::LSB()
 {
@@ -15,30 +15,41 @@ void LSB::LoadImg(const QString & path)
     img.load(path);
     width = img.width();
     height = img.height();
-    max = (width * height * 3 - 12) / 8;
+    max = (width * height * 3 - 12) / 4;
 }
 
 void LSB::WriteImg(const QString & path)
 {
     img.save(path);
 }
 
-void LSB::CoverPixel()
+void LSB::CoverPixel(int TYPE)
 {
     int len = coded.length();
     int index = 0;
+    int red, green, blue, alpha;
+    QRgb pixelValue, newPixelValue;
+    QVector <QString> list;
+    QString value;
     for (int i=0;i<width;i++) {
         for (int j=0;j<height;j++) {
-            QRgb pixelValue = img.pixel(i, j);
-            int red = qRed(pixelValue);
-            int green = qGreen(pixelValue);
-            int blue = qBlue(pixelValue);
-            QVector <QString> list;
+            if (index <= len)
+            {
+            pixelValue = img.pixel(i, j);
+            red = qRed(pixelValue);
+            green = qGreen(pixelValue);
+            blue = qBlue(pixelValue);
+            if (TYPE == PNG)
+            {
+                alpha = qAlpha(pixelValue);
+            }
+            list.clear();
+            list.shrink_to_fit();
             list.append(QString::number(red, 2));
             list.append(QString::number(green, 2));
             list.append(QString::number(blue, 2));
             for (int t=0;t<3;t++) {
-                QString value = list[t];
+                value = list[t];
                 if (index >= len)
                 {
                     value[value.length()-1] = '0';
@@ -52,8 +63,14 @@ void LSB::CoverPixel()
             red = list[0].toInt(&n, 2);
             green = list[1].toInt(&n, 2);
             blue = list[2].toInt(&n, 2);
-            QRgb newPixelValue = qRgb(red, green, blue);
+            if (TYPE == PNG)
+            {
+                newPixelValue = qRgba(red, green, blue, alpha);
+            } else {
+                newPixelValue = qRgb(red, green, blue);
+            }
             img.setPixel(i, j, newPixelValue);
+            }
         }
     }
 }
@@ -63,20 +80,23 @@ QString LSB::SplitPixel()
     QString head = "";
     int len = 0;
     bool flag = false;
-    QString decoded = "";
+    QString decoded = "", value;
+    QRgb pixelValue;
     int index = 0;
+    QVector <QString> list;
     for (int i=0;i<width;i++) {
         for (int j=0;j<height;j++) {
-            QRgb pixelValue = img.pixel(i, j);
+            pixelValue = img.pixel(i, j);
             int red = qRed(pixelValue);
             int green = qGreen(pixelValue);
             int blue = qBlue(pixelValue);
-            QVector <QString> list;
+            list.clear();
+            list.shrink_to_fit();
             list.append(QString::number(red, 2));
             list.append(QString::number(green, 2));
             list.append(QString::number(blue, 2));
             for (int t=0;t<3;t++) {
-                QString value = list[t];
+                value = list[t];
                 if (index < 12)
                 {
                     head += value[value.length() - 1];
@@ -100,6 +120,7 @@ QString LSB::SplitPixel()
     return "";
 }
 
+
 void LSB::SetText(const QString & string)
 {
     raw = string;
diff --git a/lsb.h b/lsb.h
@@ -1,4 +1,4 @@
-﻿#ifndef LSB_H
+#ifndef LSB_H
 #define LSB_H
 
 #include <iostream>
@@ -8,6 +8,8 @@
 #include <QVector>
 #include <QDebug>
 
+enum TYPE {PNG, DEFAULT};
+
 class LSB
 {
 private:
@@ -31,7 +33,7 @@ class LSB
     int GetWidth();
     QImage GetImg();
     QString GetDecoded();
-    void CoverPixel();
+    void CoverPixel(int TYPE=DEFAULT);
     QString SplitPixel();
     void Encode();
     void Decode();
diff --git a/mainwindow.cpp b/mainwindow.cpp
@@ -62,27 +62,27 @@ void MainWindow::Warning(const QString & string)
     messageBox.setWindowTitle("Warning");
     messageBox.setText(string);
     messageBox.setIcon(QMessageBox::Information);
-    messageBox.addButton("confirm", QMessageBox::AcceptRole);
+    messageBox.addButton("确认", QMessageBox::AcceptRole);
     messageBox.exec();
 }
 
 
 void MainWindow::on_choose_clicked()
 {
-    Path = QFileDialog::getOpenFileName(this, "Choose a image file", "/", "(*.png *.jpg *.jpeg *.bmp)");
+    Path = QFileDialog::getOpenFileName(this, QString("选择一张图片"), "/", "(*.png *.bmp)");
     if (Path.isEmpty())
     {
         qDebug() << "Error";
     }
     else
     {
+        Reset();
         QFileInfo fileInfo(Path);
         path = fileInfo.path();
         fileName = fileInfo.fileName();
         suffix = fileInfo.completeSuffix();
         ui->rawfile->setText(fileName);
         ui->button->setEnabled(true);
-
         lsb.LoadImg(Path);
         max = QString::number(lsb.GetMax());
         if (mode == ENC)
@@ -98,7 +98,8 @@ void MainWindow::on_choose_clicked()
 
 void MainWindow::on_save_clicked()
 {
-    lsb.WriteImg(path + "/" + ui->name->text());
+    QString new_path = path + "/" + ui->name->text();
+    lsb.WriteImg(new_path);
 }
 
 
@@ -110,15 +111,20 @@ void MainWindow::on_button_clicked()
         QString text = ui->text->toPlainText();
         if (text == "")
         {
-            Warning("At least one word");
+            Warning("至少一个字符");
         } else {
             if (text.length() >= max.toInt())
             {
-                Warning("Words over the threshold");
+                Warning("字符过多");
             }
             lsb.SetText(text);
             lsb.Encode();
-            lsb.CoverPixel();
+            if (suffix == "png")
+            {
+                lsb.CoverPixel(PNG);
+            } else {
+                lsb.CoverPixel(DEFAULT);
+            }
             ImgShow(ui->encrypt, lsb.GetImg());
             ui->name->setText(fileName.remove(fileName.length() - suffix.length() - 1, suffix.length() + 1) + "_encrypt." + suffix);
             ui->name->setReadOnly(true);
@@ -135,7 +141,6 @@ void MainWindow::on_button_clicked()
         break;
     }
     }
-
 }
 
 
@@ -145,8 +150,8 @@ void MainWindow::on_choose1_stateChanged(int arg1)
     {
         mode = ENC;
         on_clear_clicked();
-        ui->num->setText("text to be encrypted");
-        ui->button->setText("encrypt");
+        ui->num->setText("待加密密文");
+        ui->button->setText("加密");
         ui->choose2->setChecked(false);
         Reset();
     }
@@ -159,8 +164,8 @@ void MainWindow::on_choose2_stateChanged(int arg1)
     {
         mode = DEC;
         on_clear_clicked();
-        ui->num->setText("text decrypted");
-        ui->button->setText("decrypt");
+        ui->num->setText("解密明文");
+        ui->button->setText("解密");
         ui->choose1->setChecked(false);
         Reset();
     }
diff --git a/mainwindow.ui b/mainwindow.ui
@@ -57,12 +57,12 @@
      <rect>
       <x>180</x>
       <y>20</y>
-      <width>72</width>
-      <height>15</height>
+      <width>51</width>
+      <height>16</height>
      </rect>
     </property>
     <property name="text">
-     <string>raw</string>
+     <string>原图片</string>
     </property>
    </widget>
    <widget class="QTextEdit" name="text">
@@ -120,7 +120,7 @@
      </rect>
     </property>
     <property name="text">
-     <string>save</string>
+     <string>保存</string>
     </property>
    </widget>
    <widget class="QPushButton" name="button">
@@ -133,7 +133,7 @@
      </rect>
     </property>
     <property name="text">
-     <string>encrypt</string>
+     <string>加密</string>
     </property>
    </widget>
    <widget class="QLineEdit" name="name">
@@ -159,7 +159,7 @@
      </rect>
     </property>
     <property name="text">
-     <string>encrypt</string>
+     <string>加密</string>
     </property>
    </widget>
    <widget class="QCheckBox" name="choose2">
@@ -172,7 +172,7 @@
      </rect>
     </property>
     <property name="text">
-     <string>decrypt</string>
+     <string>解密</string>
     </property>
    </widget>
    <widget class="QPushButton" name="clear">
@@ -185,7 +185,7 @@
      </rect>
     </property>
     <property name="text">
-     <string>clr</string>
+     <string>清除</string>
     </property>
    </widget>
    <widget class="QPushButton" name="copy">
@@ -198,7 +198,7 @@
      </rect>
     </property>
     <property name="text">
-     <string>copy</string>
+     <string>复制</string>
     </property>
    </widget>
    <widget class="QLabel" name="num">
@@ -211,7 +211,7 @@
      </rect>
     </property>
     <property name="text">
-     <string>text to be encrypted</string>
+     <string>待加密密文</string>
     </property>
    </widget>
   </widget>

Original file line number	Diff line number	Diff line change
`@@ -62,27 +62,27 @@ void MainWindow::Warning(const QString & string)`
`62`	`62`	`messageBox.setWindowTitle("Warning");`
`63`	`63`	`messageBox.setText(string);`
`64`	`64`	`messageBox.setIcon(QMessageBox::Information);`
`65`		`- messageBox.addButton("confirm", QMessageBox::AcceptRole);`
	`65`	`+ messageBox.addButton("确认", QMessageBox::AcceptRole);`
`66`	`66`	`messageBox.exec();`
`67`	`67`	`}`
`68`	`68`
`69`	`69`
`70`	`70`	`void MainWindow::on_choose_clicked()`
`71`	`71`	`{`
`72`		`- Path = QFileDialog::getOpenFileName(this, "Choose a image file", "/", "(.png .jpg .jpeg .bmp)");`
	`72`	`+ Path = QFileDialog::getOpenFileName(this, QString("选择一张图片"), "/", "(.png .bmp)");`
`73`	`73`	`if (Path.isEmpty())`
`74`	`74`	`{`
`75`	`75`	`qDebug() << "Error";`
`76`	`76`	`}`
`77`	`77`	`else`
`78`	`78`	`{`
	`79`	`+ Reset();`
`79`	`80`	`QFileInfo fileInfo(Path);`
`80`	`81`	`path = fileInfo.path();`
`81`	`82`	`fileName = fileInfo.fileName();`
`82`	`83`	`suffix = fileInfo.completeSuffix();`
`83`	`84`	`ui->rawfile->setText(fileName);`
`84`	`85`	`ui->button->setEnabled(true);`
`85`		`-`
`86`	`86`	`lsb.LoadImg(Path);`
`87`	`87`	`max = QString::number(lsb.GetMax());`
`88`	`88`	`if (mode == ENC)`
`@@ -98,7 +98,8 @@ void MainWindow::on_choose_clicked()`
`98`	`98`
`99`	`99`	`void MainWindow::on_save_clicked()`
`100`	`100`	`{`
`101`		`- lsb.WriteImg(path + "/" + ui->name->text());`
	`101`	`+ QString new_path = path + "/" + ui->name->text();`
	`102`	`+ lsb.WriteImg(new_path);`
`102`	`103`	`}`
`103`	`104`
`104`	`105`
`@@ -110,15 +111,20 @@ void MainWindow::on_button_clicked()`
`110`	`111`	`QString text = ui->text->toPlainText();`
`111`	`112`	`if (text == "")`
`112`	`113`	`{`
`113`		`- Warning("At least one word");`
	`114`	`+ Warning("至少一个字符");`
`114`	`115`	`} else {`
`115`	`116`	`if (text.length() >= max.toInt())`
`116`	`117`	`{`
`117`		`- Warning("Words over the threshold");`
	`118`	`+ Warning("字符过多");`
`118`	`119`	`}`
`119`	`120`	`lsb.SetText(text);`
`120`	`121`	`lsb.Encode();`
`121`		`- lsb.CoverPixel();`
	`122`	`+ if (suffix == "png")`
	`123`	`+ {`
	`124`	`+ lsb.CoverPixel(PNG);`
	`125`	`+ } else {`
	`126`	`+ lsb.CoverPixel(DEFAULT);`
	`127`	`+ }`
`122`	`128`	`ImgShow(ui->encrypt, lsb.GetImg());`
`123`	`129`	`ui->name->setText(fileName.remove(fileName.length() - suffix.length() - 1, suffix.length() + 1) + "_encrypt." + suffix);`
`124`	`130`	`ui->name->setReadOnly(true);`
`@@ -135,7 +141,6 @@ void MainWindow::on_button_clicked()`
`135`	`141`	`break;`
`136`	`142`	`}`
`137`	`143`	`}`
`138`		`-`
`139`	`144`	`}`
`140`	`145`
`141`	`146`
`@@ -145,8 +150,8 @@ void MainWindow::on_choose1_stateChanged(int arg1)`
`145`	`150`	`{`
`146`	`151`	`mode = ENC;`
`147`	`152`	`on_clear_clicked();`
`148`		`- ui->num->setText("text to be encrypted");`
`149`		`- ui->button->setText("encrypt");`
	`153`	`+ ui->num->setText("待加密密文");`
	`154`	`+ ui->button->setText("加密");`
`150`	`155`	`ui->choose2->setChecked(false);`
`151`	`156`	`Reset();`
`152`	`157`	`}`
`@@ -159,8 +164,8 @@ void MainWindow::on_choose2_stateChanged(int arg1)`
`159`	`164`	`{`
`160`	`165`	`mode = DEC;`
`161`	`166`	`on_clear_clicked();`
`162`		`- ui->num->setText("text decrypted");`
`163`		`- ui->button->setText("decrypt");`
	`167`	`+ ui->num->setText("解密明文");`
	`168`	`+ ui->button->setText("解密");`
`164`	`169`	`ui->choose1->setChecked(false);`
`165`	`170`	`Reset();`
`166`	`171`	`}`