z_stream 怎么用

z_stream 是一个结构体，用于 zlib 压缩库中的数据压缩和解压缩操作。下面是一个简单的示例代码，展示了如何使用 z_stream 进行数据压缩和解压缩：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <zlib.h>

#define CHUNK_SIZE 1024

int compress_data(char *input, int input_len, char *output, int output_len) {
    z_stream strm;
    int ret;
    unsigned have;
    unsigned char in[CHUNK_SIZE];
    unsigned char out[CHUNK_SIZE];

    // 初始化 z_stream 结构体
    memset(&strm, 0, sizeof(strm));
    strm.zalloc = Z_NULL;
    strm.zfree = Z_NULL;
    strm.opaque = Z_NULL;

    // 初始化压缩器
    ret = deflateInit(&strm, Z_DEFAULT_COMPRESSION);
    if (ret != Z_OK) {
        return ret;
    }

    // 开始压缩数据
    strm.next_in = (unsigned char*)input;
    strm.avail_in = input_len;
    do {
        strm.avail_out = CHUNK_SIZE;
        strm.next_out = out;
        ret = deflate(&strm, Z_FINISH);
        have = CHUNK_SIZE - strm.avail_out;
        memcpy(output, out, have);
        output += have;
        output_len -= have;
    } while (strm.avail_out == 0);

    // 结束压缩
    deflateEnd(&strm);

    return output_len;
}

int decompress_data(char *input, int input_len, char *output, int output_len) {
    z_stream strm;
    int ret;
    unsigned have;
    unsigned char in[CHUNK_SIZE];
    unsigned char out[CHUNK_SIZE];

    // 初始化 z_stream 结构体
    memset(&strm, 0, sizeof(strm));
    strm.zalloc = Z_NULL;
    strm.zfree = Z_NULL;
    strm.opaque = Z_NULL;

    // 初始化解压器
    ret = inflateInit(&strm);
    if (ret != Z_OK) {
        return ret;
    }

    // 开始解压数据
    strm.next_in = (unsigned char*)input;
    strm.avail_in = input_len;
    do {
        strm.avail_out = CHUNK_SIZE;
        strm.next_out = out;
        ret = inflate(&strm, Z_NO_FLUSH);
        switch (ret) {
            case Z_NEED_DICT:
                ret = Z_DATA_ERROR;
            case Z_DATA_ERROR:
            case Z_MEM_ERROR:
                inflateEnd(&strm);
                return ret;
        }
        have = CHUNK_SIZE - strm.avail_out;
        memcpy(output, out, have);
        output += have;
        output_len -= have;
    } while (strm.avail_out == 0);

    // 结束解压
    inflateEnd(&strm);

    return output_len;
}

int main() {
    char input[] = "Hello, world!";
    char compressed[1024] = {0};
    char decompressed[1024] = {0};
    int compressed_len, decompressed_len;

    compressed_len = compress_data(input, strlen(input), compressed, sizeof(compressed));
    printf("Compressed data: %s\n", compressed);

    decompressed_len = decompress_data(compressed, compressed_len, decompressed, sizeof(decompressed));
    printf("Decompressed data: %s\n", decompressed);

    return 0;
}

这个示例代码中，我们首先定义了一个 CHUNK_SIZE 常量，代表每次压缩或解压数据的大小。接着，在 compress_data 函数中，我们初始化了一个 z_stream 结构体，然后使用 deflateInit 函数初始化压缩器。接着，我们将要压缩的数据传入栈中的 in 数组，然后不断调用 deflate 函数进行压缩，直到输入数据全部被压缩完成。最后，我们使用 deflateEnd 函数结束压缩。

在 decompress_data 函数中，我们同样初始化了一个 z_stream 结构体，然后使用 inflateInit 函数初始化解压器。接着，我们将要解压的数据传入栈中的 in 数组，然后不断调用 inflate 函数进行解压，直到输入数据全部被解压完成。最后，我们使用 inflateEnd 函数结束解压。

在 main 函数中，我们定义了一个字符串 input，将其压缩后，再将其解压。最后，我们输出了压缩前后的数据，以验证数据压缩和解压的正确性。