linux下c语言文件压缩

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

#define CHUNK 16384

int compress_file(const char *source, const char *destination) {
    FILE *infile = fopen(source, "rb");
    FILE *outfile = fopen(destination, "wb");
    if (!infile || !outfile) {
        perror("File opening failed");
        return -1;
    }

    unsigned char in[CHUNK];
    unsigned char out[CHUNK];
    z_stream strm = {0};
    strm.zalloc = Z_NULL;
    strm.zfree = Z_NULL;
    strm.opaque = Z_NULL;

    if (deflateInit(&strm, Z_BEST_COMPRESSION) != Z_OK) {
        fprintf(stderr, "deflateInit error\n");
        return -1;
    }

    int ret;
    do {
        strm.avail_in = fread(in, 1, CHUNK, infile);
        if (ferror(infile)) {
            inflateEnd(&strm);
            return -1;
        }
        if (strm.avail_in == 0)
            break;
        strm.next_in = in;

        do {
            strm.avail_out = CHUNK;
            strm.next_out = out;
            ret = deflate(&strm, Z_NO_FLUSH);
            if (ret == Z_STREAM_ERROR) {
                inflateEnd(&strm);
                return -1;
            }
            switch (ret) {
                case Z_NEED_DICT:
                case Z_DATA_ERROR:
                case Z_MEM_ERROR:
                    inflateEnd(&strm);
                    return -1;
            }
            unsigned have = CHUNK - strm.avail_out;
            if (fwrite(out, 1, have, outfile) != have || ferror(outfile)) {
                inflateEnd(&strm);
                return -1;
            }
        } while (strm.avail_out == 0);
    } while (ret != Z_STREAM_END);

    deflateEnd(&strm);
    fclose(infile);
    fclose(outfile);

    if (ret != Z_STREAM_END) {
        return -1;
    }
    return 0;
}

int main(int argc, char *argv[]) {
    if (argc != 3) {
        fprintf(stderr, "Usage: %s <source> <destination>\n", argv[0]);
        return 1;
    }
    if (compress_file(argv[1], argv[2]) != 0) {
        fprintf(stderr, "Compression failed\n");
        return 1;
    }
    printf("Compression successful\n");
    return 0;
}

gcc -o compress_file compress_file.c -lz
./compress_file source_file.txt compressed_file.gz