src/parsec/disk-image/parsec/parsec-benchmark/pkgs/libs/ssl/src/fips/fips_test_suite.c - public/gem5-resources - Git at Google

 /* ====================================================================
  * Copyright (c) 2003 The OpenSSL Project.  All rights reserved.
  *
  *
  * This command is intended as a test driver for the FIPS-140 testing
  * lab performing FIPS-140 validation.  It demonstrates the use of the
  * OpenSSL library ito perform a variety of common cryptographic
  * functions.  A power-up self test is demonstrated by deliberately
  * pointing to an invalid executable hash
  *
  * Contributed by Steve Marquess.
  *
  */
 #include <stdio.h>
 #include <assert.h>
 #include <ctype.h>
 #include <string.h>
 #include <stdlib.h>
 #include <openssl/aes.h>
 #include <openssl/des.h>
 #include <openssl/rsa.h>
 #include <openssl/dsa.h>
 #include <openssl/dh.h>
 #include <openssl/hmac.h>
 #include <openssl/err.h>

 #include <openssl/bn.h>
 #include <openssl/rand.h>
 #include <openssl/sha.h>


 #ifndef OPENSSL_FIPS
 int main(int argc, char *argv[])
     {
     printf("No FIPS support\n");
     return(0);
     }
 #else

 #include <openssl/fips.h>
 #include "fips_utl.h"

 /* AES: encrypt and decrypt known plaintext, verify result matches original plaintext
 */
 static int FIPS_aes_test(void)
 	{
 	int ret = 0;
 	unsigned char pltmp[16];
 	unsigned char citmp[16];
 	unsigned char key[16] = { 1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16};
 	unsigned char plaintext[16] = "etaonrishdlcu";
 	EVP_CIPHER_CTX ctx;
 	EVP_CIPHER_CTX_init(&ctx);
 	if (EVP_CipherInit_ex(&ctx, EVP_aes_128_ecb(),NULL, key, NULL, 1) <= 0)
 		goto err;
 	EVP_Cipher(&ctx, citmp, plaintext, 16);
 	if (EVP_CipherInit_ex(&ctx, EVP_aes_128_ecb(),NULL, key, NULL, 0) <= 0)
 		goto err;
 	EVP_Cipher(&ctx, pltmp, citmp, 16);
 	if (memcmp(pltmp, plaintext, 16))
 		goto err;
 	ret = 1;
 	err:
 	EVP_CIPHER_CTX_cleanup(&ctx);
 	return ret;
 	}

 static int FIPS_des3_test(void)
 	{
 	int ret = 0;
 	unsigned char pltmp[8];
 	unsigned char citmp[8];
     	unsigned char key[] = { 1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,
 		              19,20,21,22,23,24};
     	unsigned char plaintext[] = { 'e', 't', 'a', 'o', 'n', 'r', 'i', 's' };
 	EVP_CIPHER_CTX ctx;
 	EVP_CIPHER_CTX_init(&ctx);
 	if (EVP_CipherInit_ex(&ctx, EVP_des_ede3_ecb(),NULL, key, NULL, 1) <= 0)
 		goto err;
 	EVP_Cipher(&ctx, citmp, plaintext, 8);
 	if (EVP_CipherInit_ex(&ctx, EVP_des_ede3_ecb(),NULL, key, NULL, 0) <= 0)
 		goto err;
 	EVP_Cipher(&ctx, pltmp, citmp, 8);
 	if (memcmp(pltmp, plaintext, 8))
 		goto err;
 	ret = 1;
 	err:
 	EVP_CIPHER_CTX_cleanup(&ctx);
 	return ret;
 	}

 /*
  * DSA: generate keys and sign, verify input plaintext.
  */
 static int FIPS_dsa_test(int bad)
     {
     DSA *dsa = NULL;
     EVP_PKEY pk;
     unsigned char dgst[] = "etaonrishdlc";
     unsigned char buf[60];
     unsigned int slen;
     int r = 0;
     EVP_MD_CTX mctx;

     ERR_clear_error();
     EVP_MD_CTX_init(&mctx);
     dsa = FIPS_dsa_new();
     if (!dsa)
 	goto end;
     if (!DSA_generate_parameters_ex(dsa, 1024,NULL,0,NULL,NULL,NULL))
 	goto end;
     if (!DSA_generate_key(dsa))
 	goto end;
     if (bad)
 	    BN_add_word(dsa->pub_key, 1);

     pk.type = EVP_PKEY_DSA;
     pk.pkey.dsa = dsa;

     if (!EVP_SignInit_ex(&mctx, EVP_dss1(), NULL))
 	goto end;
     if (!EVP_SignUpdate(&mctx, dgst, sizeof(dgst) - 1))
 	goto end;
     if (!EVP_SignFinal(&mctx, buf, &slen, &pk))
 	goto end;

     if (!EVP_VerifyInit_ex(&mctx, EVP_dss1(), NULL))
 	goto end;
     if (!EVP_VerifyUpdate(&mctx, dgst, sizeof(dgst) - 1))
 	goto end;
     r = EVP_VerifyFinal(&mctx, buf, slen, &pk);
     end:
     EVP_MD_CTX_cleanup(&mctx);
     if (dsa)
   	  FIPS_dsa_free(dsa);
     if (r != 1)
 	return 0;
     return 1;
     }

 /*
  * RSA: generate keys and sign, verify input plaintext.
  */
 static int FIPS_rsa_test(int bad)
     {
     RSA *key;
     unsigned char input_ptext[] = "etaonrishdlc";
     unsigned char buf[256];
     unsigned int slen;
     BIGNUM *bn;
     EVP_MD_CTX mctx;
     EVP_PKEY pk;
     int r = 0;

     ERR_clear_error();
     EVP_MD_CTX_init(&mctx);
     key = FIPS_rsa_new();
     bn = BN_new();
     if (!key || !bn)
 	return 0;
     BN_set_word(bn, 65537);
     if (!RSA_generate_key_ex(key, 1024,bn,NULL))
 	return 0;
     BN_free(bn);
     if (bad)
 	    BN_add_word(key->n, 1);

     pk.type = EVP_PKEY_RSA;
     pk.pkey.rsa = key;

     if (!EVP_SignInit_ex(&mctx, EVP_sha1(), NULL))
 	goto end;
     if (!EVP_SignUpdate(&mctx, input_ptext, sizeof(input_ptext) - 1))
 	goto end;
     if (!EVP_SignFinal(&mctx, buf, &slen, &pk))
 	goto end;

     if (!EVP_VerifyInit_ex(&mctx, EVP_sha1(), NULL))
 	goto end;
     if (!EVP_VerifyUpdate(&mctx, input_ptext, sizeof(input_ptext) - 1))
 	goto end;
     r = EVP_VerifyFinal(&mctx, buf, slen, &pk);
     end:
     EVP_MD_CTX_cleanup(&mctx);
     if (key)
   	  FIPS_rsa_free(key);
     if (r != 1)
 	return 0;
     return 1;
     }

 /* SHA1: generate hash of known digest value and compare to known
    precomputed correct hash
 */
 static int FIPS_sha1_test()
     {
     unsigned char digest[SHA_DIGEST_LENGTH] =
         { 0x11, 0xf1, 0x9a, 0x3a, 0xec, 0x1a, 0x1e, 0x8e, 0x65, 0xd4, 0x9a, 0x38, 0x0c, 0x8b, 0x1e, 0x2c, 0xe8, 0xb3, 0xc5, 0x18 };
     unsigned char str[] = "etaonrishd";

     unsigned char md[SHA_DIGEST_LENGTH];

     ERR_clear_error();
     if (!EVP_Digest(str,sizeof(str) - 1,md, NULL, EVP_sha1(), NULL)) return 0;
     if (memcmp(md,digest,sizeof(md)))
         return 0;
     return 1;
     }

 /* SHA256: generate hash of known digest value and compare to known
    precomputed correct hash
 */
 static int FIPS_sha256_test()
     {
     unsigned char digest[SHA256_DIGEST_LENGTH] =
 	{0xf5, 0x53, 0xcd, 0xb8, 0xcf, 0x1, 0xee, 0x17, 0x9b, 0x93, 0xc9, 0x68, 0xc0, 0xea, 0x40, 0x91,
 	 0x6, 0xec, 0x8e, 0x11, 0x96, 0xc8, 0x5d, 0x1c, 0xaf, 0x64, 0x22, 0xe6, 0x50, 0x4f, 0x47, 0x57};
     unsigned char str[] = "etaonrishd";

     unsigned char md[SHA256_DIGEST_LENGTH];

     ERR_clear_error();
     if (!EVP_Digest(str,sizeof(str) - 1,md, NULL, EVP_sha256(), NULL)) return 0;
     if (memcmp(md,digest,sizeof(md)))
         return 0;
     return 1;
     }

 /* SHA512: generate hash of known digest value and compare to known
    precomputed correct hash
 */
 static int FIPS_sha512_test()
     {
     unsigned char digest[SHA512_DIGEST_LENGTH] =
 	{0x99, 0xc9, 0xe9, 0x5b, 0x88, 0xd4, 0x78, 0x88, 0xdf, 0x88, 0x5f, 0x94, 0x71, 0x64, 0x28, 0xca,
 	 0x16, 0x1f, 0x3d, 0xf4, 0x1f, 0xf3, 0x0f, 0xc5, 0x03, 0x99, 0xb2, 0xd0, 0xe7, 0x0b, 0x94, 0x4a,
 	 0x45, 0xd2, 0x6c, 0x4f, 0x20, 0x06, 0xef, 0x71, 0xa9, 0x25, 0x7f, 0x24, 0xb1, 0xd9, 0x40, 0x22,
 	 0x49, 0x54, 0x10, 0xc2, 0x22, 0x9d, 0x27, 0xfe, 0xbd, 0xd6, 0xd6, 0xeb, 0x2d, 0x42, 0x1d, 0xa3};
     unsigned char str[] = "etaonrishd";

     unsigned char md[SHA512_DIGEST_LENGTH];

     ERR_clear_error();
     if (!EVP_Digest(str,sizeof(str) - 1,md, NULL, EVP_sha512(), NULL)) return 0;
     if (memcmp(md,digest,sizeof(md)))
         return 0;
     return 1;
     }

 /* HMAC-SHA1: generate hash of known digest value and compare to known
    precomputed correct hash
 */
 static int FIPS_hmac_sha1_test()
     {
     unsigned char key[] = "etaonrishd";
     unsigned char iv[] = "Sample text";
     unsigned char kaval[EVP_MAX_MD_SIZE] =
 	{0x73, 0xf7, 0xa0, 0x48, 0xf8, 0x94, 0xed, 0xdd, 0x0a, 0xea, 0xea, 0x56, 0x1b, 0x61, 0x2e, 0x70,
 	 0xb2, 0xfb, 0xec, 0xc6};

     unsigned char out[EVP_MAX_MD_SIZE];
     unsigned int outlen;

     ERR_clear_error();
     if (!HMAC(EVP_sha1(),key,sizeof(key)-1,iv,sizeof(iv)-1,out,&outlen)) return 0;
     if (memcmp(out,kaval,outlen))
         return 0;
     return 1;
     }

 /* HMAC-SHA224: generate hash of known digest value and compare to known
    precomputed correct hash
 */
 static int FIPS_hmac_sha224_test()
     {
     unsigned char key[] = "etaonrishd";
     unsigned char iv[] = "Sample text";
     unsigned char kaval[EVP_MAX_MD_SIZE] =
 	{0x75, 0x58, 0xd5, 0xbd, 0x55, 0x6d, 0x87, 0x0f, 0x75, 0xff, 0xbe, 0x1c, 0xb2, 0xf0, 0x20, 0x35,
 	 0xe5, 0x62, 0x49, 0xb6, 0x94, 0xb9, 0xfc, 0x65, 0x34, 0x33, 0x3a, 0x19};

     unsigned char out[EVP_MAX_MD_SIZE];
     unsigned int outlen;

     ERR_clear_error();
     if (!HMAC(EVP_sha224(),key,sizeof(key)-1,iv,sizeof(iv)-1,out,&outlen)) return 0;
     if (memcmp(out,kaval,outlen))
         return 0;
     return 1;
     }

 /* HMAC-SHA256: generate hash of known digest value and compare to known
    precomputed correct hash
 */
 static int FIPS_hmac_sha256_test()
     {
     unsigned char key[] = "etaonrishd";
     unsigned char iv[] = "Sample text";
     unsigned char kaval[EVP_MAX_MD_SIZE] =
 	{0xe9, 0x17, 0xc1, 0x7b, 0x4c, 0x6b, 0x77, 0xda, 0xd2, 0x30, 0x36, 0x02, 0xf5, 0x72, 0x33, 0x87,
 	 0x9f, 0xc6, 0x6e, 0x7b, 0x7e, 0xa8, 0xea, 0xaa, 0x9f, 0xba, 0xee, 0x51, 0xff, 0xda, 0x24, 0xf4};

     unsigned char out[EVP_MAX_MD_SIZE];
     unsigned int outlen;

     ERR_clear_error();
     if (!HMAC(EVP_sha256(),key,sizeof(key)-1,iv,sizeof(iv)-1,out,&outlen)) return 0;
     if (memcmp(out,kaval,outlen))
         return 0;
     return 1;
     }

 /* HMAC-SHA384: generate hash of known digest value and compare to known
    precomputed correct hash
 */
 static int FIPS_hmac_sha384_test()
     {
     unsigned char key[] = "etaonrishd";
     unsigned char iv[] = "Sample text";
     unsigned char kaval[EVP_MAX_MD_SIZE] =
 	{0xb2, 0x9d, 0x40, 0x58, 0x32, 0xc4, 0xe3, 0x31, 0xb6, 0x63, 0x08, 0x26, 0x99, 0xef, 0x3b, 0x10,
 	 0xe2, 0xdf, 0xf8, 0xff, 0xc6, 0xe1, 0x03, 0x29, 0x81, 0x2a, 0x1b, 0xac, 0xb0, 0x07, 0x39, 0x08,
 	 0xf3, 0x91, 0x35, 0x11, 0x76, 0xd6, 0x4c, 0x20, 0xfb, 0x4d, 0xc3, 0xf3, 0xb8, 0x9b, 0x88, 0x1c};

     unsigned char out[EVP_MAX_MD_SIZE];
     unsigned int outlen;

     ERR_clear_error();
     if (!HMAC(EVP_sha384(),key,sizeof(key)-1,iv,sizeof(iv)-1,out,&outlen)) return 0;
     if (memcmp(out,kaval,outlen))
         return 0;
     return 1;
     }

 /* HMAC-SHA512: generate hash of known digest value and compare to known
    precomputed correct hash
 */
 static int FIPS_hmac_sha512_test()
     {
     unsigned char key[] = "etaonrishd";
     unsigned char iv[] = "Sample text";
     unsigned char kaval[EVP_MAX_MD_SIZE] =
 	{0xcd, 0x3e, 0xb9, 0x51, 0xb8, 0xbc, 0x7f, 0x9a, 0x23, 0xaf, 0xf3, 0x77, 0x59, 0x85, 0xa9, 0xe6,
 	 0xf7, 0xd1, 0x51, 0x96, 0x17, 0xe0, 0x92, 0xd8, 0xa6, 0x3b, 0xc1, 0xad, 0x7e, 0x24, 0xca, 0xb1,
 	 0xd7, 0x79, 0x0a, 0xa5, 0xea, 0x2c, 0x02, 0x58, 0x0b, 0xa6, 0x52, 0x6b, 0x61, 0x7f, 0xeb, 0x9c,
 	 0x47, 0x86, 0x5d, 0x74, 0x2b, 0x88, 0xdf, 0xee, 0x46, 0x69, 0x96, 0x3d, 0xa6, 0xd9, 0x2a, 0x53};

     unsigned char out[EVP_MAX_MD_SIZE];
     unsigned int outlen;

     ERR_clear_error();
     if (!HMAC(EVP_sha512(),key,sizeof(key)-1,iv,sizeof(iv)-1,out,&outlen)) return 0;
     if (memcmp(out,kaval,outlen))
         return 0;
     return 1;
     }


 /* DH: generate shared parameters
 */
 static int dh_test()
     {
     DH *dh;
     ERR_clear_error();
     dh = FIPS_dh_new();
     if (!dh)
 	return 0;
     if (!DH_generate_parameters_ex(dh, 1024, 2, NULL))
 	return 0;
     FIPS_dh_free(dh);
     return 1;
     }

 /* Zeroize
 */
 static int Zeroize()
     {
     RSA *key;
     BIGNUM *bn;
     unsigned char userkey[16] =
 	{ 0x48, 0x50, 0xf0, 0xa3, 0x3a, 0xed, 0xd3, 0xaf, 0x6e, 0x47, 0x7f, 0x83, 0x02, 0xb1, 0x09, 0x68 };
     int i, n;

     key = FIPS_rsa_new();
     bn = BN_new();
     if (!key || !bn)
 	return 0;
     BN_set_word(bn, 65537);
     if (!RSA_generate_key_ex(key, 1024,bn,NULL))
 	return 0;
     BN_free(bn);

     n = BN_num_bytes(key->d);
     printf(" Generated %d byte RSA private key\n", n);
     printf("\tBN key before overwriting:\n");
     do_bn_print(stdout, key->d);
     BN_rand(key->d,n*8,-1,0);
     printf("\tBN key after overwriting:\n");
     do_bn_print(stdout, key->d);

     printf("\tchar buffer key before overwriting: \n\t\t");
     for(i = 0; i < sizeof(userkey); i++) printf("%02x", userkey[i]);
         printf("\n");
     RAND_bytes(userkey, sizeof userkey);
     printf("\tchar buffer key after overwriting: \n\t\t");
     for(i = 0; i < sizeof(userkey); i++) printf("%02x", userkey[i]);
         printf("\n");

     return 1;
     }

 static int Error;
 const char * Fail(const char *msg)
     {
     do_print_errors();
     Error++;
     return msg;
     }

 int main(int argc,char **argv)
     {

     int do_corrupt_rsa_keygen = 0, do_corrupt_dsa_keygen = 0;
     int bad_rsa = 0, bad_dsa = 0;
     int do_rng_stick = 0;
     int no_exit = 0;

     printf("\tFIPS-mode test application\n\n");

     /* Load entropy from external file, if any */
     RAND_load_file(".rnd", 1024);

     if (argv[1]) {
         /* Corrupted KAT tests */
         if (!strcmp(argv[1], "aes")) {
             FIPS_corrupt_aes();
             printf("AES encryption/decryption with corrupted KAT...\n");
         } else if (!strcmp(argv[1], "des")) {
             FIPS_corrupt_des();
             printf("DES3-ECB encryption/decryption with corrupted KAT...\n");
         } else if (!strcmp(argv[1], "dsa")) {
             FIPS_corrupt_dsa();
             printf("DSA key generation and signature validation with corrupted KAT...\n");
         } else if (!strcmp(argv[1], "rsa")) {
             FIPS_corrupt_rsa();
             printf("RSA key generation and signature validation with corrupted KAT...\n");
         } else if (!strcmp(argv[1], "rsakey")) {
             printf("RSA key generation and signature validation with corrupted key...\n");
 	    bad_rsa = 1;
 	    no_exit = 1;
         } else if (!strcmp(argv[1], "rsakeygen")) {
 	    do_corrupt_rsa_keygen = 1;
 	    no_exit = 1;
             printf("RSA key generation and signature validation with corrupted keygen...\n");
         } else if (!strcmp(argv[1], "dsakey")) {
             printf("DSA key generation and signature validation with corrupted key...\n");
 	    bad_dsa = 1;
 	    no_exit = 1;
         } else if (!strcmp(argv[1], "dsakeygen")) {
 	    do_corrupt_dsa_keygen = 1;
 	    no_exit = 1;
             printf("DSA key generation and signature validation with corrupted keygen...\n");
         } else if (!strcmp(argv[1], "sha1")) {
             FIPS_corrupt_sha1();
             printf("SHA-1 hash with corrupted KAT...\n");
 	} else if (!strcmp(argv[1], "rng")) {
 	    FIPS_corrupt_rng();
 	} else if (!strcmp(argv[1], "rngstick")) {
 	    do_rng_stick = 1;
 	    no_exit = 1;
 	    printf("RNG test with stuck continuous test...\n");
         } else {
             printf("Bad argument \"%s\"\n", argv[1]);
             exit(1);
         }
 	if (!no_exit) {
         	if (!FIPS_mode_set(1)) {
  		    do_print_errors();
         	    printf("Power-up self test failed\n");
 		    exit(1);
 		}
         	printf("Power-up self test successful\n");
         	exit(0);
 	}
     }

     /* Non-Approved cryptographic operation
     */
     printf("1. Non-Approved cryptographic operation test...\n");
     printf("\ta. Included algorithm (D-H)...");
     printf( dh_test() ? "successful\n" :  Fail("FAILED!\n") );

     /* Power-up self test
     */
     ERR_clear_error();
     printf("2. Automatic power-up self test...");
     if (!FIPS_mode_set(1))
 	{
 	do_print_errors();
         printf(Fail("FAILED!\n"));
 	exit(1);
 	}
     printf("successful\n");
     if (do_corrupt_dsa_keygen)
             FIPS_corrupt_dsa_keygen();
     if (do_corrupt_rsa_keygen)
             FIPS_corrupt_rsa_keygen();
     if (do_rng_stick)
             FIPS_rng_stick();

     /* AES encryption/decryption
     */
     printf("3. AES encryption/decryption...");
     printf( FIPS_aes_test() ? "successful\n" :  Fail("FAILED!\n") );

     /* RSA key generation and encryption/decryption
     */
     printf("4. RSA key generation and encryption/decryption...");
     printf( FIPS_rsa_test(bad_rsa) ? "successful\n" :  Fail("FAILED!\n") );

     /* DES-CBC encryption/decryption
     */
     printf("5. DES-ECB encryption/decryption...");
     printf( FIPS_des3_test() ? "successful\n" :  Fail("FAILED!\n") );

     /* DSA key generation and signature validation
     */
     printf("6. DSA key generation and signature validation...");
     printf( FIPS_dsa_test(bad_dsa) ? "successful\n" :  Fail("FAILED!\n") );

     /* SHA-1 hash
     */
     printf("7a. SHA-1 hash...");
     printf( FIPS_sha1_test() ? "successful\n" :  Fail("FAILED!\n") );

     /* SHA-256 hash
     */
     printf("7b. SHA-256 hash...");
     printf( FIPS_sha256_test() ? "successful\n" :  Fail("FAILED!\n") );

     /* SHA-512 hash
     */
     printf("7c. SHA-512 hash...");
     printf( FIPS_sha512_test() ? "successful\n" :  Fail("FAILED!\n") );

     /* HMAC-SHA-1 hash
     */
     printf("7d. HMAC-SHA-1 hash...");
     printf( FIPS_hmac_sha1_test() ? "successful\n" :  Fail("FAILED!\n") );

     /* HMAC-SHA-224 hash
     */
     printf("7e. HMAC-SHA-224 hash...");
     printf( FIPS_hmac_sha224_test() ? "successful\n" :  Fail("FAILED!\n") );

     /* HMAC-SHA-256 hash
     */
     printf("7f. HMAC-SHA-256 hash...");
     printf( FIPS_hmac_sha256_test() ? "successful\n" :  Fail("FAILED!\n") );

     /* HMAC-SHA-384 hash
     */
     printf("7g. HMAC-SHA-384 hash...");
     printf( FIPS_hmac_sha384_test() ? "successful\n" :  Fail("FAILED!\n") );

     /* HMAC-SHA-512 hash
     */
     printf("7h. HMAC-SHA-512 hash...");
     printf( FIPS_hmac_sha512_test() ? "successful\n" :  Fail("FAILED!\n") );

     /* Non-Approved cryptographic operation
     */
     printf("8. Non-Approved cryptographic operation test...\n");
     printf("\ta. Included algorithm (D-H)...");
     printf( dh_test() ? "successful as expected\n"
 	    : Fail("failed INCORRECTLY!\n") );

     /* Zeroization
     */
     printf("9. Zero-ization...\n");
     printf( Zeroize() ? "\tsuccessful as expected\n"
 	    : Fail("\tfailed INCORRECTLY!\n") );

     printf("\nAll tests completed with %d errors\n", Error);
     return Error ? 1 : 0;
     }

 #endif
	/* ====================================================================
	* Copyright (c) 2003 The OpenSSL Project. All rights reserved.
	*
	*
	* This command is intended as a test driver for the FIPS-140 testing
	* lab performing FIPS-140 validation. It demonstrates the use of the
	* OpenSSL library ito perform a variety of common cryptographic
	* functions. A power-up self test is demonstrated by deliberately
	* pointing to an invalid executable hash
	*
	* Contributed by Steve Marquess.
	*
	*/
	#include <stdio.h>
	#include <assert.h>
	#include <ctype.h>
	#include <string.h>
	#include <stdlib.h>
	#include <openssl/aes.h>
	#include <openssl/des.h>
	#include <openssl/rsa.h>
	#include <openssl/dsa.h>
	#include <openssl/dh.h>
	#include <openssl/hmac.h>
	#include <openssl/err.h>

	#include <openssl/bn.h>
	#include <openssl/rand.h>
	#include <openssl/sha.h>


	#ifndef OPENSSL_FIPS
	int main(int argc, char *argv[])
	{
	printf("No FIPS support\n");
	return(0);
	}
	#else

	#include <openssl/fips.h>
	#include "fips_utl.h"

	/* AES: encrypt and decrypt known plaintext, verify result matches original plaintext
	*/
	static int FIPS_aes_test(void)
	{
	int ret = 0;
	unsigned char pltmp[16];
	unsigned char citmp[16];
	unsigned char key[16] = { 1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16};
	unsigned char plaintext[16] = "etaonrishdlcu";
	EVP_CIPHER_CTX ctx;
	EVP_CIPHER_CTX_init(&ctx);
	if (EVP_CipherInit_ex(&ctx, EVP_aes_128_ecb(),NULL, key, NULL, 1) <= 0)
	goto err;
	EVP_Cipher(&ctx, citmp, plaintext, 16);
	if (EVP_CipherInit_ex(&ctx, EVP_aes_128_ecb(),NULL, key, NULL, 0) <= 0)
	goto err;
	EVP_Cipher(&ctx, pltmp, citmp, 16);
	if (memcmp(pltmp, plaintext, 16))
	goto err;
	ret = 1;
	err:
	EVP_CIPHER_CTX_cleanup(&ctx);
	return ret;
	}

	static int FIPS_des3_test(void)
	{
	int ret = 0;
	unsigned char pltmp[8];
	unsigned char citmp[8];
	unsigned char key[] = { 1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,
	19,20,21,22,23,24};
	unsigned char plaintext[] = { 'e', 't', 'a', 'o', 'n', 'r', 'i', 's' };
	EVP_CIPHER_CTX ctx;
	EVP_CIPHER_CTX_init(&ctx);
	if (EVP_CipherInit_ex(&ctx, EVP_des_ede3_ecb(),NULL, key, NULL, 1) <= 0)
	goto err;
	EVP_Cipher(&ctx, citmp, plaintext, 8);
	if (EVP_CipherInit_ex(&ctx, EVP_des_ede3_ecb(),NULL, key, NULL, 0) <= 0)
	goto err;
	EVP_Cipher(&ctx, pltmp, citmp, 8);
	if (memcmp(pltmp, plaintext, 8))
	goto err;
	ret = 1;
	err:
	EVP_CIPHER_CTX_cleanup(&ctx);
	return ret;
	}

	/*
	* DSA: generate keys and sign, verify input plaintext.
	*/
	static int FIPS_dsa_test(int bad)
	{
	DSA *dsa = NULL;
	EVP_PKEY pk;
	unsigned char dgst[] = "etaonrishdlc";
	unsigned char buf[60];
	unsigned int slen;
	int r = 0;
	EVP_MD_CTX mctx;

	ERR_clear_error();
	EVP_MD_CTX_init(&mctx);
	dsa = FIPS_dsa_new();
	if (!dsa)
	goto end;
	if (!DSA_generate_parameters_ex(dsa, 1024,NULL,0,NULL,NULL,NULL))
	goto end;
	if (!DSA_generate_key(dsa))
	goto end;
	if (bad)
	BN_add_word(dsa->pub_key, 1);

	pk.type = EVP_PKEY_DSA;
	pk.pkey.dsa = dsa;

	if (!EVP_SignInit_ex(&mctx, EVP_dss1(), NULL))
	goto end;
	if (!EVP_SignUpdate(&mctx, dgst, sizeof(dgst) - 1))
	goto end;
	if (!EVP_SignFinal(&mctx, buf, &slen, &pk))
	goto end;

	if (!EVP_VerifyInit_ex(&mctx, EVP_dss1(), NULL))
	goto end;
	if (!EVP_VerifyUpdate(&mctx, dgst, sizeof(dgst) - 1))
	goto end;
	r = EVP_VerifyFinal(&mctx, buf, slen, &pk);
	end:
	EVP_MD_CTX_cleanup(&mctx);
	if (dsa)
	FIPS_dsa_free(dsa);
	if (r != 1)
	return 0;
	return 1;
	}

	/*
	* RSA: generate keys and sign, verify input plaintext.
	*/
	static int FIPS_rsa_test(int bad)
	{
	RSA *key;
	unsigned char input_ptext[] = "etaonrishdlc";
	unsigned char buf[256];
	unsigned int slen;
	BIGNUM *bn;
	EVP_MD_CTX mctx;
	EVP_PKEY pk;
	int r = 0;

	ERR_clear_error();
	EVP_MD_CTX_init(&mctx);
	key = FIPS_rsa_new();
	bn = BN_new();
	if (!key \|\| !bn)
	return 0;
	BN_set_word(bn, 65537);
	if (!RSA_generate_key_ex(key, 1024,bn,NULL))
	return 0;
	BN_free(bn);
	if (bad)
	BN_add_word(key->n, 1);

	pk.type = EVP_PKEY_RSA;
	pk.pkey.rsa = key;

	if (!EVP_SignInit_ex(&mctx, EVP_sha1(), NULL))
	goto end;
	if (!EVP_SignUpdate(&mctx, input_ptext, sizeof(input_ptext) - 1))
	goto end;
	if (!EVP_SignFinal(&mctx, buf, &slen, &pk))
	goto end;

	if (!EVP_VerifyInit_ex(&mctx, EVP_sha1(), NULL))
	goto end;
	if (!EVP_VerifyUpdate(&mctx, input_ptext, sizeof(input_ptext) - 1))
	goto end;
	r = EVP_VerifyFinal(&mctx, buf, slen, &pk);
	end:
	EVP_MD_CTX_cleanup(&mctx);
	if (key)
	FIPS_rsa_free(key);
	if (r != 1)
	return 0;
	return 1;
	}

	/* SHA1: generate hash of known digest value and compare to known
	precomputed correct hash
	*/
	static int FIPS_sha1_test()
	{
	unsigned char digest[SHA_DIGEST_LENGTH] =
	{ 0x11, 0xf1, 0x9a, 0x3a, 0xec, 0x1a, 0x1e, 0x8e, 0x65, 0xd4, 0x9a, 0x38, 0x0c, 0x8b, 0x1e, 0x2c, 0xe8, 0xb3, 0xc5, 0x18 };
	unsigned char str[] = "etaonrishd";

	unsigned char md[SHA_DIGEST_LENGTH];

	ERR_clear_error();
	if (!EVP_Digest(str,sizeof(str) - 1,md, NULL, EVP_sha1(), NULL)) return 0;
	if (memcmp(md,digest,sizeof(md)))
	return 0;
	return 1;
	}

	/* SHA256: generate hash of known digest value and compare to known
	precomputed correct hash
	*/
	static int FIPS_sha256_test()
	{
	unsigned char digest[SHA256_DIGEST_LENGTH] =
	{0xf5, 0x53, 0xcd, 0xb8, 0xcf, 0x1, 0xee, 0x17, 0x9b, 0x93, 0xc9, 0x68, 0xc0, 0xea, 0x40, 0x91,
	0x6, 0xec, 0x8e, 0x11, 0x96, 0xc8, 0x5d, 0x1c, 0xaf, 0x64, 0x22, 0xe6, 0x50, 0x4f, 0x47, 0x57};
	unsigned char str[] = "etaonrishd";

	unsigned char md[SHA256_DIGEST_LENGTH];

	ERR_clear_error();
	if (!EVP_Digest(str,sizeof(str) - 1,md, NULL, EVP_sha256(), NULL)) return 0;
	if (memcmp(md,digest,sizeof(md)))
	return 0;
	return 1;
	}

	/* SHA512: generate hash of known digest value and compare to known
	precomputed correct hash
	*/
	static int FIPS_sha512_test()
	{
	unsigned char digest[SHA512_DIGEST_LENGTH] =
	{0x99, 0xc9, 0xe9, 0x5b, 0x88, 0xd4, 0x78, 0x88, 0xdf, 0x88, 0x5f, 0x94, 0x71, 0x64, 0x28, 0xca,
	0x16, 0x1f, 0x3d, 0xf4, 0x1f, 0xf3, 0x0f, 0xc5, 0x03, 0x99, 0xb2, 0xd0, 0xe7, 0x0b, 0x94, 0x4a,
	0x45, 0xd2, 0x6c, 0x4f, 0x20, 0x06, 0xef, 0x71, 0xa9, 0x25, 0x7f, 0x24, 0xb1, 0xd9, 0x40, 0x22,
	0x49, 0x54, 0x10, 0xc2, 0x22, 0x9d, 0x27, 0xfe, 0xbd, 0xd6, 0xd6, 0xeb, 0x2d, 0x42, 0x1d, 0xa3};
	unsigned char str[] = "etaonrishd";

	unsigned char md[SHA512_DIGEST_LENGTH];

	ERR_clear_error();
	if (!EVP_Digest(str,sizeof(str) - 1,md, NULL, EVP_sha512(), NULL)) return 0;
	if (memcmp(md,digest,sizeof(md)))
	return 0;
	return 1;
	}

	/* HMAC-SHA1: generate hash of known digest value and compare to known
	precomputed correct hash
	*/
	static int FIPS_hmac_sha1_test()
	{
	unsigned char key[] = "etaonrishd";
	unsigned char iv[] = "Sample text";
	unsigned char kaval[EVP_MAX_MD_SIZE] =
	{0x73, 0xf7, 0xa0, 0x48, 0xf8, 0x94, 0xed, 0xdd, 0x0a, 0xea, 0xea, 0x56, 0x1b, 0x61, 0x2e, 0x70,
	0xb2, 0xfb, 0xec, 0xc6};

	unsigned char out[EVP_MAX_MD_SIZE];
	unsigned int outlen;

	ERR_clear_error();
	if (!HMAC(EVP_sha1(),key,sizeof(key)-1,iv,sizeof(iv)-1,out,&outlen)) return 0;
	if (memcmp(out,kaval,outlen))
	return 0;
	return 1;
	}

	/* HMAC-SHA224: generate hash of known digest value and compare to known
	precomputed correct hash
	*/
	static int FIPS_hmac_sha224_test()
	{
	unsigned char key[] = "etaonrishd";
	unsigned char iv[] = "Sample text";
	unsigned char kaval[EVP_MAX_MD_SIZE] =
	{0x75, 0x58, 0xd5, 0xbd, 0x55, 0x6d, 0x87, 0x0f, 0x75, 0xff, 0xbe, 0x1c, 0xb2, 0xf0, 0x20, 0x35,
	0xe5, 0x62, 0x49, 0xb6, 0x94, 0xb9, 0xfc, 0x65, 0x34, 0x33, 0x3a, 0x19};

	unsigned char out[EVP_MAX_MD_SIZE];
	unsigned int outlen;

	ERR_clear_error();
	if (!HMAC(EVP_sha224(),key,sizeof(key)-1,iv,sizeof(iv)-1,out,&outlen)) return 0;
	if (memcmp(out,kaval,outlen))
	return 0;
	return 1;
	}

	/* HMAC-SHA256: generate hash of known digest value and compare to known
	precomputed correct hash
	*/
	static int FIPS_hmac_sha256_test()
	{
	unsigned char key[] = "etaonrishd";
	unsigned char iv[] = "Sample text";
	unsigned char kaval[EVP_MAX_MD_SIZE] =
	{0xe9, 0x17, 0xc1, 0x7b, 0x4c, 0x6b, 0x77, 0xda, 0xd2, 0x30, 0x36, 0x02, 0xf5, 0x72, 0x33, 0x87,
	0x9f, 0xc6, 0x6e, 0x7b, 0x7e, 0xa8, 0xea, 0xaa, 0x9f, 0xba, 0xee, 0x51, 0xff, 0xda, 0x24, 0xf4};

	unsigned char out[EVP_MAX_MD_SIZE];
	unsigned int outlen;

	ERR_clear_error();
	if (!HMAC(EVP_sha256(),key,sizeof(key)-1,iv,sizeof(iv)-1,out,&outlen)) return 0;
	if (memcmp(out,kaval,outlen))
	return 0;
	return 1;
	}

	/* HMAC-SHA384: generate hash of known digest value and compare to known
	precomputed correct hash
	*/
	static int FIPS_hmac_sha384_test()
	{
	unsigned char key[] = "etaonrishd";
	unsigned char iv[] = "Sample text";
	unsigned char kaval[EVP_MAX_MD_SIZE] =
	{0xb2, 0x9d, 0x40, 0x58, 0x32, 0xc4, 0xe3, 0x31, 0xb6, 0x63, 0x08, 0x26, 0x99, 0xef, 0x3b, 0x10,
	0xe2, 0xdf, 0xf8, 0xff, 0xc6, 0xe1, 0x03, 0x29, 0x81, 0x2a, 0x1b, 0xac, 0xb0, 0x07, 0x39, 0x08,
	0xf3, 0x91, 0x35, 0x11, 0x76, 0xd6, 0x4c, 0x20, 0xfb, 0x4d, 0xc3, 0xf3, 0xb8, 0x9b, 0x88, 0x1c};

	unsigned char out[EVP_MAX_MD_SIZE];
	unsigned int outlen;

	ERR_clear_error();
	if (!HMAC(EVP_sha384(),key,sizeof(key)-1,iv,sizeof(iv)-1,out,&outlen)) return 0;
	if (memcmp(out,kaval,outlen))
	return 0;
	return 1;
	}

	/* HMAC-SHA512: generate hash of known digest value and compare to known
	precomputed correct hash
	*/
	static int FIPS_hmac_sha512_test()
	{
	unsigned char key[] = "etaonrishd";
	unsigned char iv[] = "Sample text";
	unsigned char kaval[EVP_MAX_MD_SIZE] =
	{0xcd, 0x3e, 0xb9, 0x51, 0xb8, 0xbc, 0x7f, 0x9a, 0x23, 0xaf, 0xf3, 0x77, 0x59, 0x85, 0xa9, 0xe6,
	0xf7, 0xd1, 0x51, 0x96, 0x17, 0xe0, 0x92, 0xd8, 0xa6, 0x3b, 0xc1, 0xad, 0x7e, 0x24, 0xca, 0xb1,
	0xd7, 0x79, 0x0a, 0xa5, 0xea, 0x2c, 0x02, 0x58, 0x0b, 0xa6, 0x52, 0x6b, 0x61, 0x7f, 0xeb, 0x9c,
	0x47, 0x86, 0x5d, 0x74, 0x2b, 0x88, 0xdf, 0xee, 0x46, 0x69, 0x96, 0x3d, 0xa6, 0xd9, 0x2a, 0x53};

	unsigned char out[EVP_MAX_MD_SIZE];
	unsigned int outlen;

	ERR_clear_error();
	if (!HMAC(EVP_sha512(),key,sizeof(key)-1,iv,sizeof(iv)-1,out,&outlen)) return 0;
	if (memcmp(out,kaval,outlen))
	return 0;
	return 1;
	}


	/* DH: generate shared parameters
	*/
	static int dh_test()
	{
	DH *dh;
	ERR_clear_error();
	dh = FIPS_dh_new();
	if (!dh)
	return 0;
	if (!DH_generate_parameters_ex(dh, 1024, 2, NULL))
	return 0;
	FIPS_dh_free(dh);
	return 1;
	}

	/* Zeroize
	*/
	static int Zeroize()
	{
	RSA *key;
	BIGNUM *bn;
	unsigned char userkey[16] =
	{ 0x48, 0x50, 0xf0, 0xa3, 0x3a, 0xed, 0xd3, 0xaf, 0x6e, 0x47, 0x7f, 0x83, 0x02, 0xb1, 0x09, 0x68 };
	int i, n;

	key = FIPS_rsa_new();
	bn = BN_new();
	if (!key \|\| !bn)
	return 0;
	BN_set_word(bn, 65537);
	if (!RSA_generate_key_ex(key, 1024,bn,NULL))
	return 0;
	BN_free(bn);

	n = BN_num_bytes(key->d);
	printf(" Generated %d byte RSA private key\n", n);
	printf("\tBN key before overwriting:\n");
	do_bn_print(stdout, key->d);
	BN_rand(key->d,n*8,-1,0);
	printf("\tBN key after overwriting:\n");
	do_bn_print(stdout, key->d);

	printf("\tchar buffer key before overwriting: \n\t\t");
	for(i = 0; i < sizeof(userkey); i++) printf("%02x", userkey[i]);
	printf("\n");
	RAND_bytes(userkey, sizeof userkey);
	printf("\tchar buffer key after overwriting: \n\t\t");
	for(i = 0; i < sizeof(userkey); i++) printf("%02x", userkey[i]);
	printf("\n");

	return 1;
	}

	static int Error;
	const char * Fail(const char *msg)
	{
	do_print_errors();
	Error++;
	return msg;
	}

	int main(int argc,char **argv)
	{

	int do_corrupt_rsa_keygen = 0, do_corrupt_dsa_keygen = 0;
	int bad_rsa = 0, bad_dsa = 0;
	int do_rng_stick = 0;
	int no_exit = 0;

	printf("\tFIPS-mode test application\n\n");

	/* Load entropy from external file, if any */
	RAND_load_file(".rnd", 1024);

	if (argv[1]) {
	/* Corrupted KAT tests */
	if (!strcmp(argv[1], "aes")) {
	FIPS_corrupt_aes();
	printf("AES encryption/decryption with corrupted KAT...\n");
	} else if (!strcmp(argv[1], "des")) {
	FIPS_corrupt_des();
	printf("DES3-ECB encryption/decryption with corrupted KAT...\n");
	} else if (!strcmp(argv[1], "dsa")) {
	FIPS_corrupt_dsa();
	printf("DSA key generation and signature validation with corrupted KAT...\n");
	} else if (!strcmp(argv[1], "rsa")) {
	FIPS_corrupt_rsa();
	printf("RSA key generation and signature validation with corrupted KAT...\n");
	} else if (!strcmp(argv[1], "rsakey")) {
	printf("RSA key generation and signature validation with corrupted key...\n");
	bad_rsa = 1;
	no_exit = 1;
	} else if (!strcmp(argv[1], "rsakeygen")) {
	do_corrupt_rsa_keygen = 1;
	no_exit = 1;
	printf("RSA key generation and signature validation with corrupted keygen...\n");
	} else if (!strcmp(argv[1], "dsakey")) {
	printf("DSA key generation and signature validation with corrupted key...\n");
	bad_dsa = 1;
	no_exit = 1;
	} else if (!strcmp(argv[1], "dsakeygen")) {
	do_corrupt_dsa_keygen = 1;
	no_exit = 1;
	printf("DSA key generation and signature validation with corrupted keygen...\n");
	} else if (!strcmp(argv[1], "sha1")) {
	FIPS_corrupt_sha1();
	printf("SHA-1 hash with corrupted KAT...\n");
	} else if (!strcmp(argv[1], "rng")) {
	FIPS_corrupt_rng();
	} else if (!strcmp(argv[1], "rngstick")) {
	do_rng_stick = 1;
	no_exit = 1;
	printf("RNG test with stuck continuous test...\n");
	} else {
	printf("Bad argument \"%s\"\n", argv[1]);
	exit(1);
	}
	if (!no_exit) {
	if (!FIPS_mode_set(1)) {
	do_print_errors();
	printf("Power-up self test failed\n");
	exit(1);
	}
	printf("Power-up self test successful\n");
	exit(0);
	}
	}

	/* Non-Approved cryptographic operation
	*/
	printf("1. Non-Approved cryptographic operation test...\n");
	printf("\ta. Included algorithm (D-H)...");
	printf( dh_test() ? "successful\n" : Fail("FAILED!\n") );

	/* Power-up self test
	*/
	ERR_clear_error();
	printf("2. Automatic power-up self test...");
	if (!FIPS_mode_set(1))
	{
	do_print_errors();
	printf(Fail("FAILED!\n"));
	exit(1);
	}
	printf("successful\n");
	if (do_corrupt_dsa_keygen)
	FIPS_corrupt_dsa_keygen();
	if (do_corrupt_rsa_keygen)
	FIPS_corrupt_rsa_keygen();
	if (do_rng_stick)
	FIPS_rng_stick();

	/* AES encryption/decryption
	*/
	printf("3. AES encryption/decryption...");
	printf( FIPS_aes_test() ? "successful\n" : Fail("FAILED!\n") );

	/* RSA key generation and encryption/decryption
	*/
	printf("4. RSA key generation and encryption/decryption...");
	printf( FIPS_rsa_test(bad_rsa) ? "successful\n" : Fail("FAILED!\n") );

	/* DES-CBC encryption/decryption
	*/
	printf("5. DES-ECB encryption/decryption...");
	printf( FIPS_des3_test() ? "successful\n" : Fail("FAILED!\n") );

	/* DSA key generation and signature validation
	*/
	printf("6. DSA key generation and signature validation...");
	printf( FIPS_dsa_test(bad_dsa) ? "successful\n" : Fail("FAILED!\n") );

	/* SHA-1 hash
	*/
	printf("7a. SHA-1 hash...");
	printf( FIPS_sha1_test() ? "successful\n" : Fail("FAILED!\n") );

	/* SHA-256 hash
	*/
	printf("7b. SHA-256 hash...");
	printf( FIPS_sha256_test() ? "successful\n" : Fail("FAILED!\n") );

	/* SHA-512 hash
	*/
	printf("7c. SHA-512 hash...");
	printf( FIPS_sha512_test() ? "successful\n" : Fail("FAILED!\n") );

	/* HMAC-SHA-1 hash
	*/
	printf("7d. HMAC-SHA-1 hash...");
	printf( FIPS_hmac_sha1_test() ? "successful\n" : Fail("FAILED!\n") );

	/* HMAC-SHA-224 hash
	*/
	printf("7e. HMAC-SHA-224 hash...");
	printf( FIPS_hmac_sha224_test() ? "successful\n" : Fail("FAILED!\n") );

	/* HMAC-SHA-256 hash
	*/
	printf("7f. HMAC-SHA-256 hash...");
	printf( FIPS_hmac_sha256_test() ? "successful\n" : Fail("FAILED!\n") );

	/* HMAC-SHA-384 hash
	*/
	printf("7g. HMAC-SHA-384 hash...");
	printf( FIPS_hmac_sha384_test() ? "successful\n" : Fail("FAILED!\n") );

	/* HMAC-SHA-512 hash
	*/
	printf("7h. HMAC-SHA-512 hash...");
	printf( FIPS_hmac_sha512_test() ? "successful\n" : Fail("FAILED!\n") );

	/* Non-Approved cryptographic operation
	*/
	printf("8. Non-Approved cryptographic operation test...\n");
	printf("\ta. Included algorithm (D-H)...");
	printf( dh_test() ? "successful as expected\n"
	: Fail("failed INCORRECTLY!\n") );

	/* Zeroization
	*/
	printf("9. Zero-ization...\n");
	printf( Zeroize() ? "\tsuccessful as expected\n"
	: Fail("\tfailed INCORRECTLY!\n") );

	printf("\nAll tests completed with %d errors\n", Error);
	return Error ? 1 : 0;
	}

	#endif