zip/test/test.c

#include <zip.h>

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

#define ZIPNAME "test.zip"
#define TESTDATA1 "Some test data 1...\0"
#define CRC32DATA1 2220805626
#define TESTDATA2 "Some test data 2...\0"
#define CRC32DATA2 2532008468

static int total_entries = 0;

static void test_write(void) {
    struct zip_t *zip = zip_open(ZIPNAME, ZIP_DEFAULT_COMPRESSION_LEVEL, 'w');
    assert(zip != NULL);

    assert(0 == zip_entry_open(zip, "test/test-1.txt"));
    assert(0 == zip_entry_write(zip, TESTDATA1, strlen(TESTDATA1)));
    assert(0 == strcmp(zip_entry_name(zip), "test/test-1.txt"));
    assert(total_entries == zip_entry_index(zip));
    assert(strlen(TESTDATA1) == zip_entry_size(zip));
    assert(CRC32DATA1 == zip_entry_crc32(zip));
    ++total_entries;
    assert(0 == zip_entry_close(zip));

    zip_close(zip);
}

static void test_append(void) {
    struct zip_t *zip = zip_open(ZIPNAME, ZIP_DEFAULT_COMPRESSION_LEVEL, 'a');
    assert(zip != NULL);

    assert(0 == zip_entry_open(zip, "test\\test-2.txt"));
    assert(0 == strcmp(zip_entry_name(zip), "test/test-2.txt"));
    assert(total_entries == zip_entry_index(zip));
    assert(0 == zip_entry_write(zip, TESTDATA2, strlen(TESTDATA2)));
    assert(strlen(TESTDATA2) == zip_entry_size(zip));
    assert(CRC32DATA2 == zip_entry_crc32(zip));

    ++total_entries;
    assert(0 == zip_entry_close(zip));

    assert(0 == zip_entry_open(zip, "test\\empty/"));
    assert(0 == strcmp(zip_entry_name(zip), "test/empty/"));
    assert(0 == zip_entry_size(zip));
    assert(0 == zip_entry_crc32(zip));

    assert(total_entries == zip_entry_index(zip));
    ++total_entries;
    assert(0 == zip_entry_close(zip));

    assert(0 == zip_entry_open(zip, "empty/"));
    assert(0 == strcmp(zip_entry_name(zip), "empty/"));
    assert(0 == zip_entry_size(zip));
    assert(0 == zip_entry_crc32(zip));

    assert(total_entries == zip_entry_index(zip));
    ++total_entries;
    assert(0 == zip_entry_close(zip));

    zip_close(zip);
}

static void test_read(void) {
    char *buf = NULL;
    size_t bufsize;
    struct zip_t *zip = zip_open(ZIPNAME, 0, 'r');
    assert(zip != NULL);

    assert(0 == zip_entry_open(zip, "test\\test-1.txt"));
    assert(strlen(TESTDATA1) == zip_entry_size(zip));
    assert(CRC32DATA1 == zip_entry_crc32(zip));
    assert(0 == zip_entry_read(zip, (void **)&buf, &bufsize));
    assert(bufsize == strlen(TESTDATA1));
    assert(0 == strncmp(buf, TESTDATA1, bufsize));
    assert(0 == zip_entry_close(zip));
    free(buf);
    buf = NULL;
    bufsize = 0;

    assert(0 == zip_entry_open(zip, "test/test-2.txt"));
    assert(strlen(TESTDATA2) == zip_entry_size(zip));
    assert(CRC32DATA2 == zip_entry_crc32(zip));    
    assert(0 == zip_entry_read(zip, (void **)&buf, &bufsize));
    assert(bufsize == strlen(TESTDATA2));
    assert(0 == strncmp(buf, TESTDATA2, bufsize));
    assert(0 == zip_entry_close(zip));
    free(buf);
    buf = NULL;
    bufsize = 0;

    assert(0 == zip_entry_open(zip, "test\\empty/"));
    assert(0 == strcmp(zip_entry_name(zip), "test/empty/"));
    assert(0 == zip_entry_size(zip));
    assert(0 == zip_entry_crc32(zip));
    assert(0 == zip_entry_close(zip));

    zip_close(zip);
}

struct buffer_t {
    char *data;
    size_t size;
};

static size_t on_extract(void *arg, unsigned long long offset, const void *data,
                         size_t size) {
    struct buffer_t *buf = (struct buffer_t *)arg;
    buf->data = realloc(buf->data, buf->size + size + 1);
    assert(NULL != buf->data);

    memcpy(&(buf->data[buf->size]), data, size);
    buf->size += size;
    buf->data[buf->size] = 0;

    return size;
}

static void test_extract(void) {
    struct buffer_t buf = {0};

    struct zip_t *zip = zip_open(ZIPNAME, 0, 'r');
    assert(zip != NULL);

    assert(0 == zip_entry_open(zip, "test/test-1.txt"));
    assert(0 == zip_entry_extract(zip, on_extract, &buf));

    assert(buf.size == strlen(TESTDATA1));
    assert(0 == strncmp(buf.data, TESTDATA1, buf.size));
    assert(0 == zip_entry_close(zip));
    free(buf.data);
    buf.data = NULL;
    buf.size = 0;

    zip_close(zip);
}

static void test_total_entries(void) {
    struct zip_t *zip = zip_open(ZIPNAME, 0, 'r');
    assert(zip != NULL);

    int n = zip_total_entries(zip);
    zip_close(zip);

    assert(n == total_entries);
}

static void test_entry_name(void) {
    struct zip_t *zip = zip_open(ZIPNAME, 0, 'r');
    assert(zip != NULL);

    assert(zip_entry_name(zip) == NULL);

    assert(0 == zip_entry_open(zip, "test\\test-1.txt"));
    assert(NULL != zip_entry_name(zip));
    assert(0 == strcmp(zip_entry_name(zip), "test/test-1.txt"));
    assert(0 == zip_entry_close(zip));

    assert(0 == zip_entry_open(zip, "test/test-2.txt"));
    assert(NULL != zip_entry_name(zip));
    assert(0 == strcmp(zip_entry_name(zip), "test/test-2.txt"));
    assert(0 == zip_entry_close(zip));

    zip_close(zip);
}

static void test_entry_index(void) {
    struct zip_t *zip = zip_open(ZIPNAME, 0, 'r');
    assert(zip != NULL);

    assert(0 == zip_entry_open(zip, "test\\test-1.txt"));
    assert(0 == zip_entry_index(zip));
    assert(0 == zip_entry_close(zip));

    assert(0 == zip_entry_open(zip, "test/test-2.txt"));
    assert(1 == zip_entry_index(zip));
    assert(0 == zip_entry_close(zip));

    zip_close(zip);
}

static void test_entry_openbyindex(void) {
    struct zip_t *zip = zip_open(ZIPNAME, 0, 'r');
    assert(zip != NULL);

    assert(0 == zip_entry_openbyindex(zip, 1));
    assert(1 == zip_entry_index(zip));

    assert(0 == strcmp(zip_entry_name(zip), "test/test-2.txt"));
    assert(0 == zip_entry_close(zip));

    assert(0 == zip_entry_openbyindex(zip, 0));
    assert(0 == zip_entry_index(zip));
    assert(0 == strcmp(zip_entry_name(zip), "test/test-1.txt"));
    assert(0 == zip_entry_close(zip));

    zip_close(zip);
}

static void test_list_entries(void) {
    struct zip_t *zip = zip_open(ZIPNAME, 0, 'r');
    assert(zip != NULL);

    int i = 0, n = zip_total_entries(zip);
    for (; i < n; ++i) {
        assert(0 == zip_entry_openbyindex(zip, i));
        fprintf(stdout, "[%d]: %s", i, zip_entry_name(zip));
        if (zip_entry_isdir(zip)) {
            fprintf(stdout, " (DIR)");
        }
        fprintf(stdout, "\n");
        assert(0 == zip_entry_close(zip));
    }

    zip_close(zip);
}

int main(int argc, char *argv[]) {
    test_write();
    test_append();
    test_read();
    test_extract();
    test_total_entries();
    test_entry_name();
    test_entry_index();
    test_entry_openbyindex();
    test_list_entries();

    return remove(ZIPNAME);
}