/* ====================================================================
 * The Apache Software License, Version 1.1
 *
 * Copyright (c) 2000-2003 The Apache Software Foundation.  All rights
 * reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 *
 * 3. The end-user documentation included with the redistribution,
 *    if any, must include the following acknowledgment:
 *       "This product includes software developed by the
 *        Apache Software Foundation (http://www.apache.org/)."
 *    Alternately, this acknowledgment may appear in the software itself,
 *    if and wherever such third-party acknowledgments normally appear.
 *
 * 4. The names "Apache" and "Apache Software Foundation" must
 *    not be used to endorse or promote products derived from this
 *    software without prior written permission. For written
 *    permission, please contact apache@apache.org.
 *
 * 5. Products derived from this software may not be called "Apache",
 *    nor may "Apache" appear in their name, without prior written
 *    permission of the Apache Software Foundation.
 *
 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED.  IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR
 * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 * ====================================================================
 *
 * This software consists of voluntary contributions made by many
 * individuals on behalf of the Apache Software Foundation.  For more
 * information on the Apache Software Foundation, please see
 * <http://www.apache.org/>.
 */

#include "apr_general.h"
#include "apr_random.h"
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include "test_apr.h"

static void hexdump(const unsigned char *b,int n)
    {
    int i;

    for(i=0 ; i < n ; ++i)
	{
#if 0
	if((i&0xf) == 0)
	    printf("%04x",i);
	printf(" %02x",b[i]);
	if((i&0xf) == 0xf)
	    printf("\n");
#else
	printf("0x%02x,",b[i]);
	if((i&7) == 7)
	    printf("\n");
#endif
	}
    printf("\n");
    }

static apr_random_t *r;

typedef apr_status_t rnd_fn(apr_random_t *r,void *b,apr_size_t n);

static void rand_run_kat(CuTest *tc,rnd_fn *f,apr_random_t *r,
			 const unsigned char expected[128])
    {
    unsigned char c[128];
    apr_status_t rv;

    rv=f(r,c,128);
    CuAssertIntEquals(tc,0,rv);
    if(rv)
	return;
    if(memcmp(c,expected,128))
	{
	hexdump(c,128);
	hexdump(expected,128);
	CuFail(tc,"Randomness mismatch");
	}
    }

static void rand_add_zeroes(apr_random_t *r)
    {
    static unsigned char c[2048];

    apr_random_add_entropy(r,c,sizeof c);
    }

static void rand_run_seed_short(CuTest *tc,rnd_fn *f,apr_random_t *r,
				int count)
    {
    int i;
    apr_status_t rv;
    char c[1];

    for(i=0 ; i < count ; ++i)
	rand_add_zeroes(r);
    rv=f(r,c,1);
    CuAssertIntEquals(tc,rv,APR_ENOTENOUGHENTROPY);
    }

static void rand_seed_short(CuTest *tc)
    {
    r=apr_random_standard_new(p);
    rand_run_seed_short(tc,apr_random_insecure_bytes,r,32);
    }

static void rand_kat(CuTest *tc)
    {
    unsigned char expected[128]=
	{ 0x82,0x04,0xad,0xd2,0x0b,0xd5,0xac,0xda,
	  0x3d,0x85,0x58,0x38,0x54,0x6b,0x69,0x45,
	  0x37,0x4c,0xc7,0xd7,0x87,0xeb,0xbf,0xd9,
	  0xb1,0xb8,0xb8,0x2d,0x9b,0x33,0x6e,0x97,
	  0x04,0x1d,0x4c,0xb0,0xd1,0xdf,0x3d,0xac,
	  0xd2,0xaa,0xfa,0xcd,0x96,0xb7,0xcf,0xb1,
	  0x8e,0x3d,0xb3,0xe5,0x37,0xa9,0x95,0xb4,
	  0xaa,0x3d,0x11,0x1a,0x08,0x20,0x21,0x9f,
	  0xdb,0x08,0x3a,0xb9,0x57,0x9f,0xf2,0x1f,
	  0x27,0xdc,0xb6,0xc0,0x85,0x08,0x05,0xbb,
	  0x13,0xbe,0xb1,0xe9,0x63,0x2a,0xe2,0xa4,
	  0x23,0x15,0x2a,0x10,0xbf,0xdf,0x09,0xb3,
	  0xc7,0xfb,0x2d,0x87,0x48,0x19,0xfb,0xc0,
	  0x15,0x8c,0xcb,0xc6,0xbd,0x89,0x38,0x69,
	  0xa3,0xae,0xa3,0x21,0x58,0x50,0xe7,0xc4,
	  0x87,0xec,0x2e,0xb1,0x2d,0x6a,0xbd,0x46 };

    rand_add_zeroes(r);
    rand_run_kat(tc,apr_random_insecure_bytes,r,expected);
    }    

static void rand_seed_short2(CuTest *tc)
    {
    rand_run_seed_short(tc,apr_random_secure_bytes,r,320);
    }

static void rand_kat2(CuTest *tc)
    {
    unsigned char expected[128]=
	{ 0x38,0x8f,0x01,0x29,0x5a,0x5c,0x1f,0xa8,
	  0x00,0xde,0x16,0x4c,0xe5,0xf7,0x1f,0x58,
	  0xc0,0x67,0xe2,0x98,0x3d,0xde,0x4a,0x75,
	  0x61,0x3f,0x23,0xd8,0x45,0x7a,0x10,0x60,
	  0x59,0x9b,0xd6,0xaf,0xcb,0x0a,0x2e,0x34,
	  0x9c,0x39,0x5b,0xd0,0xbc,0x9a,0xf0,0x7b,
	  0x7f,0x40,0x8b,0x33,0xc0,0x0e,0x2a,0x56,
	  0xfc,0xe5,0xab,0xde,0x7b,0x13,0xf5,0xec,
	  0x15,0x68,0xb8,0x09,0xbc,0x2c,0x15,0xf0,
	  0x7b,0xef,0x2a,0x97,0x19,0xa8,0x69,0x51,
	  0xdf,0xb0,0x5f,0x1a,0x4e,0xdf,0x42,0x02,
	  0x71,0x36,0xa7,0x25,0x64,0x85,0xe2,0x72,
	  0xc7,0x87,0x4d,0x7d,0x15,0xbb,0x15,0xd1,
	  0xb1,0x62,0x0b,0x25,0xd9,0xd3,0xd9,0x5a,
	  0xe3,0x47,0x1e,0xae,0x67,0xb4,0x19,0x9e,
	  0xed,0xd2,0xde,0xce,0x18,0x70,0x57,0x12 };

    rand_add_zeroes(r);
    rand_run_kat(tc,apr_random_secure_bytes,r,expected);
    }    

static void rand_barrier(CuTest *tc)
    {
    apr_random_barrier(r);
    rand_run_seed_short(tc,apr_random_secure_bytes,r,320);
    }

static void rand_kat3(CuTest *tc)
    {
    unsigned char expected[128]=
	{ 0xe8,0xe7,0xc9,0x45,0xe2,0x2a,0x54,0xb2,
	  0xdd,0xe0,0xf9,0xbc,0x3d,0xf9,0xce,0x3c,
	  0x4c,0xbd,0xc9,0xe2,0x20,0x4a,0x35,0x1c,
	  0x04,0x52,0x7f,0xb8,0x0f,0x60,0x89,0x63,
	  0x8a,0xbe,0x0a,0x44,0xac,0x5d,0xd8,0xeb,
	  0x24,0x7d,0xd1,0xda,0x4d,0x86,0x9b,0x94,
	  0x26,0x56,0x4a,0x5e,0x30,0xea,0xd4,0xa9,
	  0x9a,0xdf,0xdd,0xb6,0xb1,0x15,0xe0,0xfa,
	  0x28,0xa4,0xd6,0x95,0xa4,0xf1,0xd8,0x6e,
	  0xeb,0x8c,0xa4,0xac,0x34,0xfe,0x06,0x92,
	  0xc5,0x09,0x99,0x86,0xdc,0x5a,0x3c,0x92,
	  0xc8,0x3e,0x52,0x00,0x4d,0x01,0x43,0x6f,
	  0x69,0xcf,0xe2,0x60,0x9c,0x23,0xb3,0xa5,
	  0x5f,0x51,0x47,0x8c,0x07,0xde,0x60,0xc6,
	  0x04,0xbf,0x32,0xd6,0xdc,0xb7,0x31,0x01,
	  0x29,0x51,0x51,0xb3,0x19,0x6e,0xe4,0xf8 };

    rand_run_kat(tc,apr_random_insecure_bytes,r,expected);
    }    

static void rand_kat4(CuTest *tc)
    {
    unsigned char expected[128]=
	{ 0x7d,0x0e,0xc4,0x4e,0x3e,0xac,0x86,0x50,
	  0x37,0x95,0x7a,0x98,0x23,0x26,0xa7,0xbf,
	  0x60,0xfb,0xa3,0x70,0x90,0xc3,0x58,0xc6,
	  0xbd,0xd9,0x5e,0xa6,0x77,0x62,0x7a,0x5c,
	  0x96,0x83,0x7f,0x80,0x3d,0xf4,0x9c,0xcc,
	  0x9b,0x0c,0x8c,0xe1,0x72,0xa8,0xfb,0xc9,
	  0xc5,0x43,0x91,0xdc,0x9d,0x92,0xc2,0xce,
	  0x1c,0x5e,0x36,0xc7,0x87,0xb1,0xb4,0xa3,
	  0xc8,0x69,0x76,0xfc,0x35,0x75,0xcb,0x08,
	  0x2f,0xe3,0x98,0x76,0x37,0x80,0x04,0x5c,
	  0xb8,0xb0,0x7f,0xb2,0xda,0xe3,0xa3,0xba,
	  0xed,0xff,0xf5,0x9d,0x3b,0x7b,0xf3,0x32,
	  0x6c,0x50,0xa5,0x3e,0xcc,0xe1,0x84,0x9c,
	  0x17,0x9e,0x80,0x64,0x09,0xbb,0x62,0xf1,
	  0x95,0xf5,0x2c,0xc6,0x9f,0x6a,0xee,0x6d,
	  0x17,0x35,0x5f,0x35,0x8d,0x55,0x0c,0x07 };

    rand_add_zeroes(r);
    rand_run_kat(tc,apr_random_secure_bytes,r,expected);
    }    

CuSuite *testrand2(void)
    {
    CuSuite *suite = CuSuiteNew("Random2");

    SUITE_ADD_TEST(suite, rand_seed_short);
    SUITE_ADD_TEST(suite, rand_kat);
    SUITE_ADD_TEST(suite, rand_seed_short2);
    SUITE_ADD_TEST(suite, rand_kat2);
    SUITE_ADD_TEST(suite, rand_barrier);
    SUITE_ADD_TEST(suite, rand_kat3);
    SUITE_ADD_TEST(suite, rand_kat4);

    return suite;
    }