sl@0: /*
sl@0: * Copyright (c) 2010 Nokia Corporation and/or its subsidiary(-ies).
sl@0: * All rights reserved.
sl@0: * This component and the accompanying materials are made available
sl@0: * under the terms of "Eclipse Public License v1.0"
sl@0: * which accompanies this distribution, and is available
sl@0: * at the URL "http://www.eclipse.org/legal/epl-v10.html".
sl@0: *
sl@0: * Initial Contributors:
sl@0: * Nokia Corporation - initial contribution.
sl@0: *
sl@0: * Contributors:
sl@0: *
sl@0: * Description:
sl@0: *
sl@0: */
sl@0: 
sl@0: // INCLUDE FILES
sl@0: #include <e32svr.h>
sl@0: #include "tcrypt.h"
sl@0: 
sl@0: // EXTERNAL FUNCTION PROTOTYPES  
sl@0: extern "C" {
sl@0: IMPORT_C char *crypt(const char *key, const char *salt);
sl@0: IMPORT_C void setkey(const char *key);
sl@0: IMPORT_C void encrypt(char block[], int edflag);
sl@0: }
sl@0: 
sl@0: // LOCAL FUNCTION PROTOTYPES
sl@0: LOCAL_C void GetBitVector(char data[], char* buffer);
sl@0: LOCAL_C char *TrimWhiteSpaces(char *string);
sl@0: 
sl@0: // ============================= LOCAL FUNCTIONS ===============================
sl@0: 
sl@0: // -----------------------------------------------------------------------------
sl@0: // TrimWhiteSpaces
sl@0: // To to trim whitespaces in the input string
sl@0: // -----------------------------------------------------------------------------
sl@0: //
sl@0: LOCAL_C char *TrimWhiteSpaces(char *string)
sl@0: {
sl@0: 	char *pTemp = string;
sl@0: 	for(;*string != '\0'; string++)
sl@0: 	{
sl@0: 		if((*string == ' ') || (*string == '\t'))
sl@0: 		{
sl@0: 			pTemp++;
sl@0: 		}
sl@0: 	}
sl@0: 	
sl@0: 	return pTemp;
sl@0: }
sl@0: 
sl@0: // -----------------------------------------------------------------------------
sl@0: // GetBitVector
sl@0: // This function unpacks the byte to obtain the corresponding bit vector
sl@0: // -----------------------------------------------------------------------------
sl@0: //
sl@0: LOCAL_C void GetBitVector(char data[], char* buffer)
sl@0: {
sl@0: 	int temp;
sl@0: 	if(buffer != NULL )
sl@0: 	{
sl@0: 		temp = strlen(buffer);
sl@0: 		for(int i = 0 ; i<temp ; ++i, ++buffer)
sl@0: 		{
sl@0: 			data[i] = *buffer - '0';
sl@0: 		}
sl@0: 	}
sl@0: }
sl@0: 
sl@0: 
sl@0: CTestCrypt::~CTestCrypt() 
sl@0: 	{ 
sl@0: 	
sl@0: 	}  
sl@0: 
sl@0: CTestCrypt::CTestCrypt(const TDesC& aStepName)
sl@0: 	{
sl@0: 	// MANDATORY Call to base class method to set up the human readable name for logging.
sl@0: 	SetTestStepName(aStepName);		
sl@0: 	}
sl@0: 
sl@0: TVerdict CTestCrypt::doTestStepPreambleL()
sl@0: 	{
sl@0: 	__UHEAP_MARK;	
sl@0: 	
sl@0: 	SetTestStepResult(EPass);
sl@0: 	iTestDataFile = fopen("C:\\tcrypt\\test_data.dat", "r");	
sl@0: 	if(iTestDataFile == NULL)
sl@0: 		{
sl@0: 		SetTestStepResult(EFail);	
sl@0: 		}
sl@0: 
sl@0: 	return TestStepResult();
sl@0: 	}
sl@0: 
sl@0: TVerdict CTestCrypt::doTestStepPostambleL()
sl@0: 	{
sl@0: 	fclose(iTestDataFile);
sl@0: 	__UHEAP_MARKEND;	
sl@0: 	return TestStepResult();
sl@0: 	}
sl@0: 
sl@0: TVerdict CTestCrypt::doTestStepL()
sl@0: 	{
sl@0: 	int err;
sl@0: 		
sl@0: 	if(TestStepName() == KEncrypt)
sl@0: 		{
sl@0:    		INFO_PRINTF1(_L("Encrypt():"));
sl@0:    		err = Encrypt();
sl@0:    		SetTestStepResult(err ? static_cast<TVerdict>(err) : EPass);
sl@0:    		}
sl@0: 	else if(TestStepName() == KCrypt)
sl@0: 		{
sl@0:    		INFO_PRINTF1(_L("Crypt():"));
sl@0:    		err = Crypt();
sl@0:    		SetTestStepResult(err ? static_cast<TVerdict>(err) : EPass);
sl@0:    		}
sl@0:   		
sl@0:    	return TestStepResult(); 
sl@0: 	}
sl@0: 
sl@0: 
sl@0: // -----------------------------------------------------------------------------
sl@0: // CTestCrypt::Encrypt
sl@0: // Encrypt function
sl@0: // -----------------------------------------------------------------------------
sl@0: //
sl@0: TInt CTestCrypt::Encrypt()
sl@0: {
sl@0: 	static int tId = 0;
sl@0: 	char pTemp[30];
sl@0: 	memset(pTemp, 0, 30);
sl@0: 	sprintf(pTemp, "ENCRYPT_TEST_DATA_%d", ++tId);
sl@0:     INFO_PRINTF2(_L("Begin: ENCRYPT_TEST_DATA_%d\n"), tId);
sl@0:  
sl@0:     // locate "test_data_id" from within the file
sl@0:     if(!RepositionFilePointer(pTemp))
sl@0:     {
sl@0:     	// String not found...invalid test data ID
sl@0:     	INFO_PRINTF1(_L("Requested test data ID could not be found\n"));
sl@0:     	return KErrNotFound;
sl@0:     }
sl@0: 
sl@0: 	// Get the key, data block, operation to be performed and the 
sl@0: 	// expected output for the current test data ID
sl@0: 	char key[64] = 
sl@0: 	{
sl@0: 		 0
sl@0: 	};
sl@0: 	char block[64] = 
sl@0: 	{
sl@0: 		0
sl@0: 	};
sl@0: 	char output[64] = 
sl@0: 	{
sl@0: 		0
sl@0: 	};
sl@0: 	int edflag = -1;
sl@0: 	if(GetEncryptTestData(key, block, &edflag, output) != KErrNone)
sl@0: 	{
sl@0: 		// Test data not found or is not present in the expected 
sl@0: 		// format
sl@0: 		INFO_PRINTF1(_L("Test data not found or is not present in the expected format\n"));
sl@0: 		return KErrNotFound;
sl@0: 	}
sl@0: 	
sl@0: 	// Perform encryption/decryption
sl@0: 	
sl@0: 	// Invoke setkey from the libcrypt library
sl@0: 	setkey(key);
sl@0: 	
sl@0: 	// Call the encrypt function
sl@0: 	encrypt(block,edflag);
sl@0: 	
sl@0: 	// Verify if the final output is same as the expected output
sl@0: 
sl@0: 	if(!strcmp(block,block))
sl@0: 	{
sl@0: 		INFO_PRINTF2(_L("End: ENCRYPT_TEST_DATA_%d\n"), tId);
sl@0: 		// Test case passed
sl@0: 		return KErrNone;
sl@0: 	}
sl@0: 	INFO_PRINTF1(_L("Output from the encrypt() function does not match the \"expected output\""));
sl@0: 	INFO_PRINTF2(_L("End: ENCRYPT_TEST_DATA_%d\n"), tId);
sl@0: 	return KErrNotFound;
sl@0: }
sl@0: 
sl@0: // -----------------------------------------------------------------------------
sl@0: // CTestCrypt::RepositionFilePointer
sl@0: // This function positions the file pointer to the line immediately following
sl@0: // the string aString
sl@0: // -----------------------------------------------------------------------------
sl@0: //
sl@0: 
sl@0: TInt CTestCrypt::RepositionFilePointer(const char *aString)
sl@0: {
sl@0:     char buffer[256];
sl@0:    	char * ptr = NULL;
sl@0:     while(fgets((char*)buffer, 256, iTestDataFile) != NULL)
sl@0:     {
sl@0:     	 ptr = NULL;
sl@0:     
sl@0:     	if((ptr=strchr(buffer,'\r')) || (ptr=strchr(buffer,'\n'))) //check for both
sl@0: 		*ptr='\0';
sl@0:     	if(!strcmp(buffer, aString))
sl@0:     	{
sl@0:     		return 1;
sl@0:     	}
sl@0:     	memset(buffer, 0, 256);
sl@0:     }
sl@0:     return 0;
sl@0: }
sl@0: 
sl@0: // -----------------------------------------------------------------------------
sl@0: // CTestCrypt::GetEncryptTestData
sl@0: // This function reads the test data for encrypt() API
sl@0: // -----------------------------------------------------------------------------
sl@0: //
sl@0: TInt CTestCrypt::GetEncryptTestData(char key[], char block[], int *edflag, char output[])
sl@0: {
sl@0: 	char buffer[256];
sl@0: 	char *p = NULL;
sl@0: 	bool bKey = false,		// will be set to true upon reading 'key'
sl@0: 	     bBlock = false,    // will be set to true upon reading 'data block'
sl@0: 	     bEdflag = false,   // will be set to true upon reading 'edflag'
sl@0: 	     bOutput = false;   // will be set to true upon reading 'expected output'
sl@0: 	     
sl@0: 	char *pTemp = NULL;
sl@0:    	char * ptr = NULL;
sl@0: 
sl@0: 	while((p = fgets(buffer, 256, iTestDataFile)) != NULL)
sl@0: 	{
sl@0: 
sl@0: 		ptr = NULL;
sl@0:     
sl@0:     	if((ptr=strchr(buffer,'\r')) || (ptr=strchr(buffer,'\n'))) //check for both
sl@0: 			*ptr='\0';
sl@0: 		if(strstr(buffer, "//") != NULL)
sl@0: 		{
sl@0: 			continue;
sl@0: 		}
sl@0: 		if(!strcmp(buffer, "END_TEST_DATA"))
sl@0: 		{
sl@0: 			if(bKey && bBlock && bEdflag && bOutput)
sl@0: 			{
sl@0: 				return KErrNone;
sl@0: 			}
sl@0: 			return KErrNotFound;
sl@0: 		}
sl@0: 		if(strstr(buffer, "KEY") != NULL)
sl@0: 		{
sl@0: 			// Read the key
sl@0: 			
sl@0: 			// Get bytes...
sl@0: 			pTemp = strstr(buffer, ":");
sl@0: 			if(pTemp != NULL)
sl@0: 			{
sl@0: 				pTemp++;
sl@0: 			    pTemp = TrimWhiteSpaces(pTemp);
sl@0: 				GetBitVector(key, pTemp);
sl@0: 			}
sl@0: 			bKey = true;
sl@0: 			continue;
sl@0: 		}
sl@0: 		if(strstr(buffer, "DATA_BLOCK") != NULL)
sl@0: 		{
sl@0: 			// Read the data block
sl@0: 			
sl@0: 			pTemp = strstr(buffer, ":");
sl@0: 			if(pTemp != NULL)
sl@0: 			{
sl@0: 				pTemp++;
sl@0: 				pTemp = TrimWhiteSpaces(pTemp);
sl@0: 				GetBitVector(block, pTemp);
sl@0: 			}
sl@0: 			bBlock = true;
sl@0: 			continue;
sl@0: 		}
sl@0: 		if(strstr(buffer, "ED_FLAG") != NULL)
sl@0: 		{
sl@0: 			// Read the ed_flag parameter
sl@0: 			
sl@0: 			pTemp = strstr(buffer, ":");
sl@0: 			if(pTemp != NULL)
sl@0: 			{
sl@0: 				pTemp++;
sl@0: 				pTemp = TrimWhiteSpaces(pTemp);
sl@0: 				*edflag = (*pTemp) - '0';
sl@0: 			}
sl@0: 			bEdflag = true;
sl@0: 			continue;
sl@0: 		}
sl@0: 		if(strstr(buffer, "EXPECTED_OUTPUT") != NULL)
sl@0: 		{
sl@0: 			// Read the bit vector for the expected output
sl@0: 			
sl@0: 			pTemp = strstr(buffer, ":");
sl@0: 			if(pTemp != NULL)
sl@0: 			{
sl@0: 				pTemp++;
sl@0: 				pTemp = TrimWhiteSpaces(pTemp);
sl@0: 				GetBitVector(output, pTemp);
sl@0: 			}
sl@0: 			bOutput = true;
sl@0: 			continue;
sl@0: 		}
sl@0: 	}
sl@0: 	
sl@0: 	return KErrNotFound;
sl@0: }
sl@0: 
sl@0: // -----------------------------------------------------------------------------
sl@0: // CTestCrypt::Crypt
sl@0: // Test function to perform crypt() on the input data
sl@0: // -----------------------------------------------------------------------------
sl@0: //
sl@0: TInt CTestCrypt::Crypt()
sl@0: {
sl@0: 	static int tId = 0;
sl@0: 	char pTemp[30];
sl@0: 	memset(pTemp, 0, 30);
sl@0: 	sprintf(pTemp, "CRYPT_TEST_DATA_%d", ++tId);
sl@0:     INFO_PRINTF2(_L("Begin CRYPT_TEST_DATA_%d\n"), tId);
sl@0:     
sl@0:     // locate "test_data_id" from within the file
sl@0:     if(!RepositionFilePointer(pTemp))
sl@0:     {
sl@0:     	// String not found...invalid test data ID
sl@0:     	INFO_PRINTF1(_L("Requested test data ID could not be found\n"));
sl@0:     	return KErrNotFound;
sl@0:     }
sl@0:     
sl@0:     char password[34] = 
sl@0:     {
sl@0:     	'\0'
sl@0:     };
sl@0:     char salt[30] = 
sl@0:     {
sl@0:     	'\0'
sl@0:     };
sl@0:     char output[35] = 
sl@0:     {
sl@0:     	'\0'
sl@0:     };
sl@0:     
sl@0:     if(GetCryptTestData(password, salt, output) != KErrNone)
sl@0:     {
sl@0:     	// Data not in the expected format or is invalid
sl@0:     	INFO_PRINTF1(_L("Test data not found or is not present in the expected format\n"));
sl@0:     	return KErrNotFound;
sl@0:     }
sl@0:     
sl@0:     char *crypt_output = NULL;
sl@0:     // Invoke crypt()
sl@0:     crypt_output = crypt(password,salt);
sl@0: 	if(!strcmp(output,""))
sl@0: 	{
sl@0: 		// Since salt is NULL, the expected output is ignored...
sl@0: 		return KErrNone;
sl@0: 	}
sl@0:     if(!strcmp(salt, ""))
sl@0:     {
sl@0: 		// salt is NULL, so skip the first byte from the crypt output
sl@0: 		if(crypt_output != NULL)
sl@0: 		{
sl@0: 			crypt_output++;
sl@0: 			if(!strcmp(crypt_output, &output[0]))
sl@0: 			{
sl@0: 				INFO_PRINTF2(_L("End: CRYPT_TEST_DATA_%d\n"), tId);
sl@0: 				return KErrNone;
sl@0: 			}
sl@0: 			INFO_PRINTF1(_L("Output from the crypt() function does not match the \"expected output\""));
sl@0: 			return KErrNotFound;
sl@0: 		}
sl@0:     }
sl@0:     else
sl@0:     {
sl@0:     	// salt is not NULL
sl@0:     	if(!strcmp(crypt_output, output))
sl@0:     	{
sl@0:     		INFO_PRINTF2(_L("End: CRYPT_TEST_DATA_%d\n"), tId);
sl@0:     		return KErrNone;
sl@0:     	}
sl@0:     	INFO_PRINTF1(_L("Output from the crypt() function does not match the \"expected output\""));
sl@0:     	return KErrNotFound;
sl@0:     }
sl@0:     return KErrNotFound;
sl@0: }
sl@0: 
sl@0: // -----------------------------------------------------------------------------
sl@0: // CTestCrypt::GetCryptTestData
sl@0: // To retrieve the test data for crypt() API
sl@0: // -----------------------------------------------------------------------------
sl@0: //
sl@0: TInt CTestCrypt::GetCryptTestData(char password[], char salt[], char output[])
sl@0: {
sl@0: 	char buffer[256];
sl@0: 	char *p = NULL;
sl@0: 	char *pTemp = NULL;
sl@0: 	int nLength = 0;
sl@0: 	bool bPassword = false,
sl@0: 	     bSalt = false,
sl@0: 	     bOutput = false;
sl@0:    	char * ptr = NULL;
sl@0: 
sl@0: 	while((p = fgets(buffer, 256, iTestDataFile)) != NULL)
sl@0: 	{
sl@0:     	ptr = NULL;
sl@0:     
sl@0:     	if((ptr=strchr(buffer,'\r')) || (ptr=strchr(buffer,'\n'))) //check for both
sl@0: 			*ptr='\0';
sl@0: 		if(strstr(buffer, "//") != NULL)	// skip the comments
sl@0: 		{
sl@0: 			// "//" could appear within password or salt, so further
sl@0: 			// check is required
sl@0: 			
sl@0: 			// Since judicious use of whitespaces is allowed only from within
sl@0: 			// the comment lines, the comment line will always start with
sl@0: 			// "//"
sl@0: 			if(buffer[0] == '/' && buffer[1] == '/')
sl@0: 			{
sl@0: 				continue;
sl@0: 			}
sl@0: 		}
sl@0: 		if(!strcmp(buffer, "END_TEST_DATA"))
sl@0: 		{
sl@0: 			if(bPassword && bSalt && bOutput)
sl@0: 			{
sl@0: 				return KErrNone;
sl@0: 			}
sl@0: 			return KErrNotFound;
sl@0: 		}
sl@0: 
sl@0: 		// Verify if the input buffer has "data". Data is followed by ":"
sl@0: 		pTemp = strstr(buffer, ":");
sl@0: 		if(pTemp != NULL)
sl@0: 		{
sl@0: 			pTemp++;
sl@0: 			pTemp = TrimWhiteSpaces(pTemp);
sl@0: 			nLength = strlen(pTemp);
sl@0: 			if(strstr(buffer, "PASSWORD") != NULL)
sl@0: 			{
sl@0: 				strncpy(password,pTemp,nLength);
sl@0: 				bPassword = true;
sl@0: 				continue;
sl@0: 			}
sl@0: 			else if(strstr(buffer, "SALT") != NULL)
sl@0: 			{
sl@0: 				strncpy(salt,pTemp,nLength);
sl@0: 				bSalt = true;
sl@0: 				continue;
sl@0: 			}
sl@0: 			else if(strstr(buffer, "EXPECTED_OUTPUT") != NULL)
sl@0: 			{
sl@0: 				strncpy(output,pTemp,nLength);
sl@0: 				bOutput = true;
sl@0: 				continue;
sl@0: 			}
sl@0: 			else
sl@0: 			{
sl@0: 				// Unexpected output
sl@0: 				return KErrNotFound;
sl@0: 			}
sl@0: 		}
sl@0: 	}
sl@0: 	return KErrNotFound;
sl@0: }
sl@0: 
sl@0: //  End of File
sl@0: