src/lib-dcrypt/test-stream.c

	test-stream.c revision 316cbe323513a0f20d1cf519fe9405e231d633e2
#include "lib.h"
#include "buffer.h"
#include "str.h"
#include "dcrypt.h"
#include "ostream.h"
#include "ostream-encrypt.h"
#include "istream.h"
#include "istream-decrypt.h"
#include "istream-hash.h"
#include "iostream-temp.h"
#include "randgen.h"
#include "hash-method.h"
#include "test-common.h"
#include "hex-binary.h"
#include <fcntl.h>
#include <sys/stat.h>
#include <stdio.h>

static const char key_v1_priv[] = "-----BEGIN PRIVATE KEY-----\n" \
"MIGpAgEAMBAGByqGSM49AgEGBSuBBAAjBIGRMIGOAgEBBEGz2V2VMi/5s+Z+GJh7\n" \
"4WfqZjZUpqqm+NJWojm6BbrZMY+9ZComlTGVcUZ007acFxV93oMmrfmtRUb5ynrb\n" \
"MRFskKFGA0QAAwHrAJc8TvyPzspOoz6UH1C1YRmaUVm8tsLu2d0dYtZeOKJUl52J\n" \
"4o8MKIg+ce4q0mTNFrhj+glKj29ppWti6JGAQA==\n" \
"-----END PRIVATE KEY-----";

static const char key_v1_pub[] = "-----BEGIN PUBLIC KEY-----\n" \
"MFgwEAYHKoZIzj0CAQYFK4EEACMDRAADAesAlzxO/I/Oyk6jPpQfULVhGZpRWby2\n" \
"wu7Z3R1i1l44olSXnYnijwwoiD5x7irSZM0WuGP6CUqPb2mla2LokYBA\n" \
"-----END PUBLIC KEY-----";

static const char key_v2_priv[] = "-----BEGIN PRIVATE KEY-----\n" \
"MIGHAgEAMBMGByqGSM49AgEGCCqGSM49AwEHBG0wawIBAQQgtuQJA+uboZWVwgHc\n" \
"DciyVdrovAPwlMqshDK3s78IDDuhRANCAAQm0VEdzLB9PtD0HA8JK1zifWnj8M00\n" \
"FQzedfp9SQsWyA8dzs5/NFR5MTe6Xbh/ndKEs1zZH3vZ4FlNrilZc0st\n" \
"-----END PRIVATE KEY-----";

static const char key_v2_pub[] = "-----BEGIN PUBLIC KEY-----\n" \
"MFkwEwYHKoZIzj0CAQYIKoZIzj0DAQcDQgAEJtFRHcywfT7Q9BwPCStc4n1p4/DN\n" \
"NBUM3nX6fUkLFsgPHc7OfzRUeTE3ul24f53ShLNc2R972eBZTa4pWXNLLQ==\n" \
"-----END PUBLIC KEY-----";

static const char test_sample_v1_hash[] = "1d7cc2cc1f1983f76241cc42389911e88590ad58cf9d54cafeb5b198d3723dd1";
static const char test_sample_v2_hash[] = "2e31218656dd34db65b321688bf418dee4ee785e99eb9c21e0d29b4af27a863e";

static struct dcrypt_keypair test_v1_kp;
static struct dcrypt_keypair test_v2_kp;

static
void test_static_v1_input(void)
{
    ssize_t siz;
    const struct hash_method *hash = hash_method_lookup("sha256");
    unsigned char hash_ctx[hash->context_size];
    unsigned char hash_dgst[hash->digest_size];
    hash->init(hash_ctx);

    test_begin("test_static_v1_input");

    struct istream *is_1 = i_stream_create_file("sample-v1.bin", IO_BLOCK_SIZE);
    struct istream *is_2 = i_stream_create_decrypt(is_1, test_v1_kp.priv);
    i_stream_unref(&is_1);
    struct istream *is_3 = i_stream_create_hash(is_2, hash, hash_ctx);
    i_stream_unref(&is_2);

    while((siz = i_stream_read(is_3))>0) { i_stream_skip(is_3, siz); }

    if (is_3->stream_errno != 0)
        i_debug("error: %s", i_stream_get_error(is_3));

    test_assert(is_3->stream_errno == 0);

    i_stream_unref(&is_3);

    hash->result(hash_ctx, hash_dgst);

    test_assert(strcmp(test_sample_v1_hash, binary_to_hex(hash_dgst, sizeof(hash_dgst))) == 0);

    test_end();
}

static
void test_static_v2_input(void)
{
    test_begin("test_static_v2_input");
    ssize_t amt;
    const struct hash_method *hash = hash_method_lookup("sha256");
    unsigned char hash_ctx[hash->context_size];
    unsigned char hash_dgst[hash->digest_size];
    hash->init(hash_ctx);

    struct istream *is_1 = i_stream_create_file("sample-v2.bin", IO_BLOCK_SIZE);
    struct istream *is_2 = i_stream_create_decrypt(is_1, test_v2_kp.priv);
    i_stream_unref(&is_1);
    struct istream *is_3 = i_stream_create_hash(is_2, hash, hash_ctx);
    i_stream_unref(&is_2);

    while((amt = i_stream_read(is_3))>0) { i_stream_skip(is_3, amt); }

        if (is_3->stream_errno != 0)
                i_debug("error: %s", i_stream_get_error(is_3));

    test_assert(is_3->stream_errno == 0);

    i_stream_unref(&is_3);

    hash->result(hash_ctx, hash_dgst);

    test_assert(strcmp(test_sample_v2_hash, binary_to_hex(hash_dgst, sizeof(hash_dgst))) == 0);

    test_end();

/** this code is left here to show how the sample file is created
    struct istream *is = i_stream_create_file("../lib-fts/udhr_fra.txt", 8192);
    struct istream *is_2 = i_stream_create_hash(is, hash, hash_ctx);
    int fd = open("sample-v2.bin", O_CREAT|O_TRUNC|O_WRONLY, S_IRWXU);
    struct ostream *os = o_stream_create_fd_file(fd, 0, TRUE);
    struct ostream *os_2 = o_stream_create_encrypt(os, "aes-256-gcm-sha256", test_v2_kp.pub, IO_STREAM_ENC_INTEGRITY_AEAD);
    const unsigned char *ptr;
    size_t siz;

    while(i_stream_read_data(is_2, &ptr, &siz, 0)>0) {
        o_stream_nsend(os_2, ptr, siz);
        i_stream_skip(is_2, siz);
    }

    i_assert(o_stream_nfinish(os_2)==0);

    o_stream_close(os_2);
    i_stream_close(is_2);

    hash->result(hash_ctx, hash_dgst);
    printf("%s\n", binary_to_hex(hash_dgst, sizeof(hash_dgst)));
*/
}

static
void test_write_read_v1(void)
{
    test_begin("test_write_read_v1");
    unsigned char payload[IO_BLOCK_SIZE];
    const unsigned char *ptr;
    size_t pos = 0, siz;
    random_fill_weak(payload, IO_BLOCK_SIZE);

    struct ostream *os = iostream_temp_create("/tmp", 0);
    struct ostream *os_2 = o_stream_create_encrypt(os, "<unused>", test_v2_kp.pub, IO_STREAM_ENC_VERSION_1);
    o_stream_nsend(os_2, payload, sizeof(payload));

        if (os_2->stream_errno != 0)
                i_debug("error: %s", o_stream_get_error(os_2));

    test_assert(os_2->stream_errno == 0);
    test_assert(o_stream_nfinish(os_2) == 0);
    test_assert(os_2->stream_errno == 0);

    o_stream_unref(&os_2);

    struct istream *is = iostream_temp_finish(&os, IO_BLOCK_SIZE);
    struct istream *is_2 = i_stream_create_decrypt(is, test_v2_kp.priv);
    i_stream_unref(&is);

    while(i_stream_read_data(is_2, &ptr, &siz, 0)>0) {
        test_assert_idx(pos + siz <= sizeof(payload), pos);
        if (pos + siz > sizeof(payload)) break;
        test_assert_idx(memcmp(ptr, payload + pos, siz) == 0, pos);
        i_stream_skip(is_2, siz);
    }

    test_assert(is_2->stream_errno == 0);

    i_stream_unref(&is_2);

    test_end();
}

static
void test_write_read_v2(void)
{
    test_begin("test_write_read_v2");
    unsigned char payload[IO_BLOCK_SIZE];
    const unsigned char *ptr;
    size_t pos = 0, siz;
    random_fill_weak(payload, IO_BLOCK_SIZE);

    struct ostream *os = iostream_temp_create("/tmp", 0);
    struct ostream *os_2 = o_stream_create_encrypt(os, "aes-256-gcm-sha256", test_v1_kp.pub, IO_STREAM_ENC_INTEGRITY_AEAD);
    o_stream_nsend(os_2, payload, IO_BLOCK_SIZE);
    test_assert(o_stream_nfinish(os_2) == 0);
        if (os_2->stream_errno != 0)
                i_debug("error: %s", o_stream_get_error(os_2));

    o_stream_unref(&os_2);

    struct istream *is = iostream_temp_finish(&os, IO_BLOCK_SIZE);
    struct istream *is_2 = i_stream_create_decrypt(is, test_v1_kp.priv);
    i_stream_unref(&is);

    while(i_stream_read_data(is_2, &ptr, &siz, 0)>0) {
        test_assert_idx(pos + siz <= sizeof(payload), pos);
        if (pos + siz > sizeof(payload)) break;
        test_assert_idx(memcmp(ptr, payload + pos, siz) == 0, pos);
        i_stream_skip(is_2, siz);
    }

    test_assert(is_2->stream_errno == 0);
        if (is_2->stream_errno != 0)
                i_debug("error: %s", i_stream_get_error(is_2));

    i_stream_unref(&is_2);

    test_end();
}

int main(void) {
    dcrypt_initialize("openssl", NULL);
    random_init();

    dcrypt_key_load_private(&test_v1_kp.priv, DCRYPT_FORMAT_PEM, key_v1_priv, NULL, NULL, NULL);
    dcrypt_key_load_public(&test_v1_kp.pub, DCRYPT_FORMAT_PEM, key_v1_pub, NULL);
    dcrypt_key_load_private(&test_v2_kp.priv, DCRYPT_FORMAT_PEM, key_v2_priv, NULL, NULL, NULL);
    dcrypt_key_load_public(&test_v2_kp.pub, DCRYPT_FORMAT_PEM, key_v2_pub, NULL);

    static void (*test_functions[])(void) = {
        test_static_v1_input,
        test_static_v2_input,
        test_write_read_v1,
        test_write_read_v2,
        NULL
    };

    return test_run(test_functions);
}