允许设定最小压缩大小

CMakeLists.txt

@@ -1,8 +1,28 @@
 cmake_minimum_required(VERSION 3.16)
 project(Transmission)
 
+if(NOT DEFINED TARGET_SYSTEM_NAME)
+    set(TARGET_SYSTEM_NAME ${CMAKE_SYSTEM_NAME})
+    message("使用默认目标 : ${TARGET_SYSTEM_NAME}")
+endif ()
+if (${TARGET_SYSTEM_NAME} STREQUAL "Linux")
+    add_definitions(-DBUILD_LINUX=1)
+    message("以Linux为构建目标")
+elseif (${TARGET_SYSTEM_NAME} STREQUAL "Windows")
+    add_definitions(-DBUILD_WINDOWS=1)
+    message("以Windows为构建目标")
+elseif(${TARGET_SYSTEM_NAME} STREQUAL "Android")
+    add_definitions(-DBUILD_ANDROID=1)
+    message("以Android为构建目标")
+else ()
+    message(FATAL_ERROR "未知的目标操作系统")
+endif ()
+
 add_subdirectory(src)
-add_subdirectory(test)
+if (IS_TEST)
+    add_subdirectory(test)
+    message("包含Gtest目标")
+endif ()
 
 add_executable(exec main.cpp
 )

src/CMakeLists.txt

@@ -21,4 +21,18 @@ add_library(Transmission
 )
 
 target_include_directories(Transmission PUBLIC ${CMAKE_CURRENT_BINARY_DIR})
-target_link_libraries(Transmission protobuf proto lzma)
+if(NOT DEFINED PROTOBUF_LIB_PATH)
+    set(PROTOBUF_LIB_PATH "protobuf")
+else ()
+    message("手动指定protobuf库为${PROTOBUF_LIB_PATH}")
+endif ()
+if(DEFINED PROTOBUF_INCLUDE_PATH)
+    message("手动指定protobuf头文件路径为${PROTOBUF_INCLUDE_PATH}")
+    target_include_directories(proto PUBLIC ${PROTOBUF_INCLUDE_PATH})
+endif ()
+if(NOT DEFINED LZMA_LIB_PATH)
+    set(LZMA_LIB_PATH "lzma")
+else ()
+    message("手动指定LZMA库为${LZMA_LIB_PATH}")
+endif ()
+target_link_libraries(Transmission ${PROTOBUF_LIB_PATH} proto ${LZMA_LIB_PATH})

src/Transmit.cpp

@@ -13,12 +13,13 @@
 namespace Transmission {
 
     const int32_t Transmit::DATA_STOP = 0x20030507;
+    int Transmit::compressSize = 64;
 
     Transmit::Transmit(SOCKET fd, std::string ip) : ip(std::move(ip)), fd(fd) {
 
     }
 
-    void Transmit::dataArrives(unsigned char *data, size_t size) {
+    void Transmit::dataArrives(const unsigned char *data, size_t size) {
         std::unique_lock<std::mutex> lock(dataMutex);
         charBuff.put(data, 0, size);
         if (this->packSize < 0) {
@@ -51,7 +52,7 @@ namespace Transmission {
         if (!dataPackets.ParseFromArray(buff.get(), this->packSize))
             throw PrepareDataException("反序列化出错");
         this->packSize = -1;
-        switch (dataPackets.type()) {
+        switch (dataPackets.algorithm()) {
             //无压缩
             case CompressAlgorithm::NOT: {
                 auto pack = copyMem(dataPackets.data());
@@ -118,13 +119,13 @@ namespace Transmission {
         return ptr;
     }
 
-    void Transmit::sendData(unsigned char *data, size_t size, int type) {
+    void Transmit::sendData(const unsigned char *data, size_t size, int type) {
         DataPackets dataPackets;
         dataPackets.set_type(type);
-        std::string byteData(reinterpret_cast<char *>(data), size);
+        std::string byteData(reinterpret_cast<const char *>(data), size);
         dataPackets.set_data(byteData);
-        if (size < 64) {
-            //小于64字节的数据包不压缩
+        if (size < compressSize) {
+            //小于compressSize的数据包不压缩
             pushData(dataPackets);
         } else {
             //大于64字节的数据使用lzma压缩后发送
@@ -132,10 +133,10 @@ namespace Transmission {
         }
     }
 
-    void Transmit::pushData(unsigned char *data, int32_t size) const {
-        ::send(this->fd, &size, sizeof(int32_t), 0);
-        ::send(this->fd, data, size, 0);
-        ::send(this->fd, &DATA_STOP, sizeof(DATA_STOP), 0);
+    void Transmit::pushData(const unsigned char *data, int32_t size) const {
+        ::send(this->fd, (const char *) &size, sizeof(int32_t), 0);
+        ::send(this->fd, (const char *) data, size, 0);
+        ::send(this->fd, (const char *) &DATA_STOP, sizeof(DATA_STOP), 0);
     }
 
     void Transmit::pushData(DataPackets &data) {
@@ -174,7 +175,7 @@ namespace Transmission {
                 buff.put(ptr.get(), 0, (int64_t) lastSize);
                 break;
             }
-            if(ret != LZMA_OK)
+            if (ret != LZMA_OK)
                 throw PrepareDataException("LZMA压缩失败");
             lastSize = stream.total_out - lastSize;
             buff.put(ptr.get(), 0, (int64_t) lastSize);

src/Transmit.h

@@ -8,7 +8,22 @@
 #define TRANSMISSION_TRANSMIT_H
 
 #include <string>
+
+#ifdef BUILD_LINUX
+
 #include <pcap/socket.h>
+
+#endif
+#ifdef BUILD_WINDOWS
+#include <winsock.h>
+#include <io.h>
+#include <ws2tcpip.h>
+#endif
+#ifdef BUILD_ANDROID
+#include <sys/socket.h>
+#define SOCKET int
+#endif
+
 #include <memory>
 #include "CharBuff.h"
 #include "transmission.pb.h"
@@ -33,7 +48,7 @@ namespace Transmission {
         void pushLzmaData(DataPackets &data);
 
         /// 实际发送
-        virtual void pushData(unsigned char *data, int32_t size) const;
+        virtual void pushData(const unsigned char *data, int32_t size) const;
 
     protected:
 
@@ -43,15 +58,17 @@ namespace Transmission {
         virtual std::shared_ptr<unsigned char> unLzma(const std::string &str, size_t &size);
 
     public:
+        //最小压缩大小
+        static int compressSize;
         static const int32_t DATA_STOP;
 
         explicit Transmit(SOCKET fd, std::string ip);
 
         /// 收到来自网络的数据
-        virtual void dataArrives(unsigned char *data, size_t size);
+        virtual void dataArrives(const unsigned char *data, size_t size);
 
         /// 发送数据
-        virtual void sendData(unsigned char *data, size_t size, int type = 0);
+        virtual void sendData(const unsigned char *data, size_t size, int type);
 
         /// 数据包就绪
         /// @param type 数据包类型

test/test_main.cpp

@@ -9,6 +9,15 @@
 static bool isExec = false;
 static const char *shortStr = "Client Hello";
 static const char *LongStr = "Client HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient Hello";
+static const char *LongLongStr = "Client HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient Hello"
+                                 "Client HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient Hello"
+                                 "Client HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient Hello"
+                                 "Client HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient Hello"
+                                 "Client HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient Hello"
+                                 "Client HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient Hello"
+                                 "Client HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient Hello"
+                                 "Client HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient Hello"
+                                 "Client HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient HelloClient Hello";
 
 class Transmit : public Transmission::Transmit {
 private:
@@ -23,12 +32,14 @@ public:
             ASSERT_TRUE(strcmp((char *) data.get(), shortStr) == 0);
         else if (type == 2)
             ASSERT_TRUE(strcmp((char *) data.get(), LongStr) == 0);
-        else
+        else if (type == 3) {
+            ASSERT_TRUE(strcmp((char *) data.get(), LongLongStr) == 0);
+        } else
             ASSERT_TRUE(false);
         isExec = true;
     }
 
-    void pushData(unsigned char *data, int32_t size) const override {
+    void pushData(const unsigned char *data, int32_t size) const override {
         ::write(this->fd, &size, sizeof(int32_t));
         ::write(this->fd, data, size);
         ::write(this->fd, &DATA_STOP, sizeof(DATA_STOP));
@@ -62,4 +73,8 @@ TEST(Transmit测试, 1) {
     transmitWrite.sendData((unsigned char *) LongStr, strlen(LongStr) + 1, 2);
     transmitRead.read();
     ASSERT_TRUE(isExec);
+    isExec = false;
+    transmitWrite.sendData((unsigned char *) LongLongStr, strlen(LongLongStr) + 1, 3);
+    transmitRead.read();
+    ASSERT_TRUE(isExec);
 }