www.pudn.com > sha1.zip > shatest.cpp

/* 
 *	shatest.cpp 
 * 
 *	Copyright (C) 1998 
 *	Paul E. Jones  
 *	All Rights Reserved 
 * 
 ***************************************************************************** 
 *	$Id: shatest.cpp,v 1.4 2004/03/27 18:02:38 paulej Exp $ 
 ***************************************************************************** 
 * 
 *	Description: 
 * 		This file will exercise the SHA1 class and perform the three 
 * 		tests documented in FIPS PUB 180-1. 
 * 
 *	Portability Issues: 
 *		None. 
 * 
 */ 
 
#include  
#include "sha1.h" 
 
using namespace std; 
 
/* 
 *	Define patterns for testing 
 */ 
#define TESTA	"abc" 
#define TESTB	"abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq" 
 
/* 
 *	Function prototype 
 */ 
void DisplayMessageDigest(unsigned *message_digest); 
 
/*	 
 *	main 
 * 
 *	Description: 
 *		This is the entry point for the program 
 * 
 *	Parameters: 
 *		None. 
 * 
 *	Returns: 
 *		Nothing. 
 * 
 *	Comments: 
 * 
 */ 
int main() 
{ 
	SHA1		sha; 
	unsigned	message_digest[5]; 
 
	/* 
	 *	Perform test A 
	 */ 
	cout << endl << "Test A: 'abc'" << endl; 
 
	sha.Reset(); 
	sha << TESTA; 
 
	if (!sha.Result(message_digest)) 
	{ 
		cerr << "ERROR-- could not compute message digest" << endl; 
	} 
	else 
	{ 
		DisplayMessageDigest(message_digest); 
		cout << "Should match:" << endl; 
		cout << '\t' << "A9993E36 4706816A BA3E2571 7850C26C 9CD0D89D" << endl; 
	} 
 
	/* 
	 *	Perform test B 
	 */ 
	cout << endl << "Test B: " << TESTB << endl; 
 
	sha.Reset(); 
	sha << TESTB; 
 
	if (!sha.Result(message_digest)) 
	{ 
		cerr << "ERROR-- could not compute message digest" << endl; 
	} 
	else 
	{ 
		DisplayMessageDigest(message_digest); 
		cout << "Should match:" << endl; 
		cout << '\t' << "84983E44 1C3BD26E BAAE4AA1 F95129E5 E54670F1" << endl; 
	} 
 
	/* 
	 *	Perform test C 
	 */ 
	cout << endl << "Test C: One million 'a' characters" << endl; 
 
	sha.Reset(); 
	for(int i = 1; i <= 1000000; i++) sha.Input('a'); 
 
	if (!sha.Result(message_digest)) 
	{ 
		cerr << "ERROR-- could not compute message digest" << endl; 
	} 
	else 
	{ 
		DisplayMessageDigest(message_digest); 
		cout << "Should match:" << endl; 
		cout << '\t' << "34AA973C D4C4DAA4 F61EEB2B DBAD2731 6534016F" << endl; 
	} 
 
	return 0; 
} 
 
/*	 
 *	DisplayMessageDigest 
 * 
 *	Description: 
 *		Display Message Digest array 
 * 
 *	Parameters: 
 *		None. 
 * 
 *	Returns: 
 *		Nothing. 
 * 
 *	Comments: 
 * 
 */ 
void DisplayMessageDigest(unsigned *message_digest) 
{ 
	ios::fmtflags	flags; 
 
	cout << '\t'; 
 
	flags = cout.setf(ios::hex|ios::uppercase,ios::basefield); 
	cout.setf(ios::uppercase); 
 
	for(int i = 0; i < 5 ; i++) 
	{ 
		cout << message_digest[i] << ' '; 
	} 
 
	cout << endl; 
 
	cout.setf(flags); 
}