C++ Socket庫數(shù)據(jù)壓縮實現(xiàn)

在C++中，使用Socket庫進行數(shù)據(jù)壓縮可以通過結(jié)合zlib庫來實現(xiàn)。zlib是一個廣泛使用的壓縮庫，提供了豐富的壓縮和解壓縮功能。下面是一個簡單的示例，展示了如何在C++ Socket服務器中使用zlib進行數(shù)據(jù)壓縮。

1. 首先，確保已經(jīng)安裝了zlib庫。在大多數(shù)Linux發(fā)行版中，可以使用包管理器安裝，例如在Ubuntu中輸入以下命令：

sudo apt-get install zlib1g-dev

2. 創(chuàng)建一個C++文件，例如server.cpp，并包含必要的頭文件：

#include <iostream>
#include <string>
#include <vector>
#include <zlib.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>

3. 定義一個函數(shù)來壓縮數(shù)據(jù)：

std::vector<char> compressData(const std::string &data) {
    z_stream zs;
    zs.zalloc = Z_NULL;
    zs.zfree = Z_NULL;
    zs.opaque = Z_NULL;
    zs.avail_in = data.size();
    zs.next_in = reinterpret_cast<Bytef *>(const_cast<char *>(data.c_str()));

    std::vector<char> compressedData;
    const int bufferSize = 1024;
    char buffer[bufferSize];

    do {
        zs.avail_out = bufferSize;
        zs.next_out = reinterpret_cast<Bytef *>(buffer);

        int ret = deflate(&zs, Z_FINISH);
        if (ret != Z_OK && ret != Z_STREAM_END) {
            std::cerr << "Error compressing data" << std::endl;
            break;
        }

        compressedData.insert(compressedData.end(), buffer, buffer + zs.total_out);
    } while (zs.avail_out == 0);

    return compressedData;
}

4. 創(chuàng)建一個簡單的Socket服務器，接收客戶端連接并對數(shù)據(jù)進行壓縮：

int main() {
    int serverSocket = socket(AF_INET, SOCK_STREAM, 0);
    if (serverSocket == -1) {
        std::cerr << "Error creating socket" << std::endl;
        return 1;
    }

    struct sockaddr_in serverAddr;
    serverAddr.sin_family = AF_INET;
    serverAddr.sin_addr.s_addr = INADDR_ANY;
    serverAddr.sin_port = htons(12345);

    if (bind(serverSocket, reinterpret_cast<struct sockaddr*>(&serverAddr), sizeof(serverAddr)) == -1) {
        std::cerr << "Error binding socket" << std::endl;
        return 1;
    }

    if (listen(serverSocket, 5) == -1) {
        std::cerr << "Error listening on socket" << std::endl;
        return 1;
    }

    while (true) {
        struct sockaddr_in clientAddr;
        socklen_t clientAddrLen = sizeof(clientAddr);
        int clientSocket = accept(serverSocket, reinterpret_cast<struct sockaddr*>(&clientAddr), &clientAddrLen);
        if (clientSocket == -1) {
            std::cerr << "Error accepting client connection" << std::endl;
            continue;
        }

        char buffer[1024];
        ssize_t bytesReceived = recv(clientSocket, buffer, sizeof(buffer) - 1, 0);
        if (bytesReceived == -1) {
            std::cerr << "Error receiving data from client" << std::endl;
            close(clientSocket);
            continue;
        }

        buffer[bytesReceived] = '\0';
        std::string data(buffer);

        std::vector<char> compressedData = compressData(data);

        send(clientSocket, reinterpret_cast<const char *>(compressedData.data()), compressedData.size(), 0);
        close(clientSocket);
    }

    close(serverSocket);
    return 0;
}

5. 編譯并運行服務器：

g++ server.cpp -o server -lzlib
./server

現(xiàn)在，服務器已經(jīng)啟動并監(jiān)聽端口12345。當客戶端連接并發(fā)送數(shù)據(jù)時，服務器將使用zlib對數(shù)據(jù)進行壓縮，然后將壓縮后的數(shù)據(jù)發(fā)送回客戶端?？蛻舳诵枰鈮嚎s接收到的數(shù)據(jù)以獲取原始數(shù)據(jù)。