Commit e6ed2b91 authored by Andy Polyakov's avatar Andy Polyakov
Browse files

Add test/bio_enc_test.c.



RT#4628

Reviewed-by: default avatarRich Salz <rsalz@openssl.org>
parent c1a7dcbe
Loading
Loading
Loading
Loading

test/bio_enc_test.c

0 → 100644
+140 −0
Original line number Diff line number Diff line
/*
 * Copyright 2016 The OpenSSL Project Authors. All Rights Reserved.
 *
 * Licensed under the OpenSSL license (the "License").  You may not use
 * this file except in compliance with the License.  You can obtain a copy
 * in the file LICENSE in the source distribution or at
 * https://www.openssl.org/source/license.html
 */
#include <stdio.h>
#include <string.h>
#include <openssl/evp.h>
#include <openssl/bio.h>

int main()
{
    BIO *b;
    static const unsigned char key[16] = { 0 };
    static unsigned char inp[1024] = { 0 };
    unsigned char out[1024], ref[1024];
    int i, lref, len;

    b = BIO_new(BIO_f_cipher());
    if (!BIO_set_cipher(b, EVP_camellia_128_ctr(), key, NULL, 0))
        return -1;
    BIO_push(b, BIO_new_mem_buf(inp, sizeof(inp)));
    lref = BIO_read(b, inp, sizeof(inp));
    BIO_free_all(b);

    /*
     * Exercise CBC cipher
     */

    /* reference output for single-chunk operation */
    b = BIO_new(BIO_f_cipher());
    if (!BIO_set_cipher(b, EVP_aes_128_cbc(), key, NULL, 0))
        return -1;
    BIO_push(b, BIO_new_mem_buf(inp, sizeof(inp)));
    lref = BIO_read(b, ref, sizeof(ref));
    BIO_free_all(b);

    /* perform split operations and compare to reference */
    for (i = 1; i < lref; i++) {
        b = BIO_new(BIO_f_cipher());
        if (!BIO_set_cipher(b, EVP_aes_128_cbc(), key, NULL, 0))
             return -1;
        BIO_push(b, BIO_new_mem_buf(inp, sizeof(inp)));
        memset(out, 0, sizeof(out));
        out[i] = ~ref[i];
        len = BIO_read(b, out, i);
        /* check for overstep */
        if (out[i] != (unsigned char)~ref[i]) {
            fprintf(stderr, "CBC output overstep@%d\n", i);
            return 1;
        }
        len += BIO_read(b, out + len, sizeof(out) - len);
        BIO_free_all(b);

        if (len != lref || memcmp(out, ref, len)) {
            fprintf(stderr, "CBC output mismatch@%d\n", i);
            return 2;
        }
    }

    /* perform small-chunk operations and compare to reference */
    for (i = 1; i < lref / 2; i++) {
        int delta;

        b = BIO_new(BIO_f_cipher());
        if (!BIO_set_cipher(b, EVP_aes_128_cbc(), key, NULL, 0))
             return -1;
        BIO_push(b, BIO_new_mem_buf(inp, sizeof(inp)));
        memset(out, 0, sizeof(out));
        for (len = 0; (delta = BIO_read(b, out + len, i)); ) {
            len += delta;
        }
        BIO_free_all(b);

        if (len != lref || memcmp(out, ref, len)) {
            fprintf(stderr, "CBC output mismatch@%d\n", i);
            return 3;
        }
    }

    /*
     * Exercise CTR cipher
     */

    /* reference output for single-chunk operation */
    b = BIO_new(BIO_f_cipher());
    if (!BIO_set_cipher(b, EVP_aes_128_cbc(), key, NULL, 0))
         return -1;
    BIO_push(b, BIO_new_mem_buf(inp, sizeof(inp)));
    lref = BIO_read(b, ref, sizeof(ref));
    BIO_free_all(b);

    /* perform split operations and compare to reference */
    for (i = 1; i < lref; i++) {
        b = BIO_new(BIO_f_cipher());
        if (!BIO_set_cipher(b, EVP_aes_128_cbc(), key, NULL, 0))
             return -1;
        BIO_push(b, BIO_new_mem_buf(inp, sizeof(inp)));
        memset(out, 0, sizeof(out));
        out[i] = ~ref[i];
        len = BIO_read(b, out, i);
        /* check for overstep */
        if (out[i] != (unsigned char)~ref[i]) {
            fprintf(stderr, "CTR output overstep@%d\n", i);
            return 4;
        }
        len += BIO_read(b, out + len, sizeof(out) - len);
        BIO_free_all(b);

        if (len != lref || memcmp(out, ref, len)) {
            fprintf(stderr, "CTR output mismatch@%d\n", i);
            return 5;
        }
    }

    /* perform small-chunk operations and compare to reference */
    for (i = 1; i < lref / 2; i++) {
        int delta;

        b = BIO_new(BIO_f_cipher());
        if (!BIO_set_cipher(b, EVP_aes_128_cbc(), key, NULL, 0))
             return -1;
        BIO_push(b, BIO_new_mem_buf(inp, sizeof(inp)));
        memset(out, 0, sizeof(out));
        for (len = 0; (delta = BIO_read(b, out + len, i)); ) {
            len += delta;
        }
        BIO_free_all(b);

        if (len != lref || memcmp(out, ref, len)) {
            fprintf(stderr, "CTR output mismatch@%d\n", i);
            return 6;
        }
    }

    return 0;
}
+5 −1
Original line number Diff line number Diff line
@@ -16,7 +16,7 @@ IF[{- !$disabled{tests} -}]
          packettest asynctest secmemtest srptest memleaktest \
          dtlsv1listentest ct_test threadstest afalgtest d2i_test \
          ssl_test_ctx_test ssl_test x509aux cipherlist_test asynciotest \
          bioprinttest sslapitest dtlstest
          bioprinttest sslapitest dtlstest bio_enc_test

  SOURCE[aborttest]=aborttest.c
  INCLUDE[aborttest]=../include
@@ -270,6 +270,10 @@ IF[{- !$disabled{tests} -}]
  SOURCE[dtlstest]=dtlstest.c ssltestlib.c testutil.c
  INCLUDE[dtlstest]=../include .
  DEPEND[dtlstest]=../libcrypto ../libssl

  SOURCE[bio_enc_test]=bio_enc_test.c
  INCLUDE[bio_enc_test]=../include
  DEPEND[bio_enc_test]=../libcrypto
ENDIF

{-
+12 −0
Original line number Diff line number Diff line
#! /usr/bin/env perl
# Copyright 2015-2016 The OpenSSL Project Authors. All Rights Reserved.
#
# Licensed under the OpenSSL license (the "License").  You may not use
# this file except in compliance with the License.  You can obtain a copy
# in the file LICENSE in the source distribution or at
# https://www.openssl.org/source/license.html


use OpenSSL::Test::Simple;

simple_test("test_bio_enc", "bio_enc_test", "bio_enc");