kunlun/dtest/security_test/test/sha256_test.c

#include "os_types.h"
#include <string.h>
#include "iot_io.h"
#include "mbedtls/sha256.h"
#include "sec_sys.h"

static int sw_sha256 = 0;
//static unsigned char buf[8196];

/*
 * 32-bit integer manipulation macros (big endian)
 */
#ifndef GET_UINT32_BE
#define GET_UINT32_BE(n,b,i)                            \
do {                                                    \
    (n) = ( (uint32_t) (b)[(i)    ] << 24 )             \
        | ( (uint32_t) (b)[(i) + 1] << 16 )             \
        | ( (uint32_t) (b)[(i) + 2] <<  8 )             \
        | ( (uint32_t) (b)[(i) + 3]       );            \
} while( 0 )
#endif

#ifndef PUT_UINT32_BE
#define PUT_UINT32_BE(n,b,i)                            \
do {                                                    \
    (b)[(i)    ] = (unsigned char) ( (n) >> 24 );       \
    (b)[(i) + 1] = (unsigned char) ( (n) >> 16 );       \
    (b)[(i) + 2] = (unsigned char) ( (n) >>  8 );       \
    (b)[(i) + 3] = (unsigned char) ( (n)       );       \
} while( 0 )
#endif
/*
 * FIPS-180-2 test vectors
 */
static const unsigned char sha256_test_buf[3][57] =
{
    { "abc" },
    { "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq" },
    { "" }
};

static const int sha256_test_buflen[3] =
{
    3, 56, 1000
};

static const unsigned char sha256_test_sum[3][32] =
{

    /*
     * SHA-256 test vectors
     */
    { 0xBA, 0x78, 0x16, 0xBF, 0x8F, 0x01, 0xCF, 0xEA,
      0x41, 0x41, 0x40, 0xDE, 0x5D, 0xAE, 0x22, 0x23,
      0xB0, 0x03, 0x61, 0xA3, 0x96, 0x17, 0x7A, 0x9C,
      0xB4, 0x10, 0xFF, 0x61, 0xF2, 0x00, 0x15, 0xAD },
    { 0x24, 0x8D, 0x6A, 0x61, 0xD2, 0x06, 0x38, 0xB8,
      0xE5, 0xC0, 0x26, 0x93, 0x0C, 0x3E, 0x60, 0x39,
      0xA3, 0x3C, 0xE4, 0x59, 0x64, 0xFF, 0x21, 0x67,
      0xF6, 0xEC, 0xED, 0xD4, 0x19, 0xDB, 0x06, 0xC1 },
    { 0xCD, 0xC7, 0x6E, 0x5C, 0x99, 0x14, 0xFB, 0x92,
      0x81, 0xA1, 0xC7, 0xE2, 0x84, 0xD7, 0x3E, 0x67,
      0xF1, 0x80, 0x9A, 0x48, 0xA4, 0x97, 0x20, 0x0E,
      0x04, 0x6D, 0x39, 0xCC, 0xC7, 0x11, 0x2C, 0xD0 }
};

#if 0
static const unsigned char sha256_padding[64] =
{
 0x80, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
};
#endif

/*
 * Checkup routine
 */
int mbedtls_sha256_self_test( int verbose )
{
    int i, ret = 0;
    unsigned char sha256sum[32];
    mbedtls_sha256_context ctx;

    mbedtls_sha256_init( &ctx );

    for( i = 0; i < 2; i++ )
    {
        if( verbose != 0 )
            iot_printf( "  SHA-%d test #%d: ", 256, i + 1 );

        if(sw_sha256){
			mbedtls_sha256_starts( &ctx, 0 );


			mbedtls_sha256_update( &ctx, sha256_test_buf[i],
									 sha256_test_buflen[i] );

			mbedtls_sha256_finish( &ctx, sha256sum );
        }else{
			sec_sys_sha256(sha256_test_buf[i], sha256_test_buflen[i], sha256sum);
        }

        if( memcmp( sha256sum, sha256_test_sum[i], 32 ) != 0 )
        {
            if( verbose != 0 )
                iot_printf( "failed\n" );

            ret = 1;
            goto exit;
        }

        if( verbose != 0 )
            iot_printf( "passed\n" );
    }

    if( verbose != 0 )
        iot_printf( "\n" );

exit:
    mbedtls_sha256_free( &ctx );

    return( ret );
}

static unsigned char test_buf_256[] = "This adapter uses AR7010+AR9280 (and presumably also the UB94 reference design for that chipset), which is found on a few other devices. It is, however, not likely that any alternative adapters will be usable with Sony products.";

int sha256_test( int verbose )
{
    int i = 0;
	int ret = 0;
    unsigned char sha256sum[2][32];
    mbedtls_sha256_context ctx;

    mbedtls_sha256_init( &ctx );


	if( verbose != 0 )
		iot_printf( "  SHA-%d test #%d: ", 256, i + 1 );


	mbedtls_sha256_starts( &ctx, 0 );
	mbedtls_sha256_update( &ctx, test_buf_256, sizeof(test_buf_256) );
	mbedtls_sha256_finish( &ctx, sha256sum[0] );

	sec_sys_sha256(test_buf_256, sizeof(test_buf_256), sha256sum[1]);

	if( memcmp( sha256sum[0], sha256sum[1], 32 ) != 0 )
	{
		if( verbose != 0 )
			iot_printf( "failed\n" );

		ret = 1;
		goto exit;
	}

	if( verbose != 0 )
		iot_printf( "passed\n" );

    if( verbose != 0 )
        iot_printf( "\n" );

exit:
    mbedtls_sha256_free( &ctx );

    return( ret );
}



int sha256_main( )
{
    mbedtls_sha256_self_test(1);

    sha256_test(1);

	return 0;
}