/* ====================================================================
 * Copyright (c) 1998-2008 The OpenSSL Project.  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. All advertising materials mentioning features or use of this
 *    software must display the following acknowledgment:
 *    "This product includes software developed by the OpenSSL Project
 *    for use in the OpenSSL Toolkit. (http://www.openssl.org/)"
 *
 * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
 *    endorse or promote products derived from this software without
 *    prior written permission. For written permission, please contact
 *    openssl-core@openssl.org.
 *
 * 5. Products derived from this software may not be called "OpenSSL"
 *    nor may "OpenSSL" appear in their names without prior written
 *    permission of the OpenSSL Project.
 *
 * 6. Redistributions of any form whatsoever must retain the following
 *    acknowledgment:
 *    "This product includes software developed by the OpenSSL Project
 *    for use in the OpenSSL Toolkit (http://www.openssl.org/)"
 *
 * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``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 OpenSSL PROJECT 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 product includes cryptographic software written by Eric Young
 * (eay@cryptsoft.com).  This product includes software written by Tim
 * Hudson (tjh@cryptsoft.com).
 *
 */

/* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
 * All rights reserved.
 *
 * This package is an SSL implementation written
 * by Eric Young (eay@cryptsoft.com).
 * The implementation was written so as to conform with Netscapes SSL.
 * 
 * This library is free for commercial and non-commercial use as long as
 * the following conditions are aheared to.  The following conditions
 * apply to all code found in this distribution, be it the RC4, RSA,
 * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
 * included with this distribution is covered by the same copyright terms
 * except that the holder is Tim Hudson (tjh@cryptsoft.com).
 * 
 * Copyright remains Eric Young's, and as such any Copyright notices in
 * the code are not to be removed.
 * If this package is used in a product, Eric Young should be given attribution
 * as the author of the parts of the library used.
 * This can be in the form of a textual message at program startup or
 * in documentation (online or textual) provided with the package.
 * 
 * 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 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. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *    "This product includes cryptographic software written by
 *     Eric Young (eay@cryptsoft.com)"
 *    The word 'cryptographic' can be left out if the rouines from the library
 *    being used are not cryptographic related :-).
 * 4. If you include any Windows specific code (or a derivative thereof) from 
 *    the apps directory (application code) you must include an acknowledgement:
 *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
 * 
 * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
 * ANY EXPRESS 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 AUTHOR OR 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.
 * 
 * The licence and distribution terms for any publically available version or
 * derivative of this code cannot be changed.  i.e. this code cannot simply be
 * copied and put under another distribution licence
 * [including the GNU Public Licence.]
 */

#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#include <limits.h>
#include <time.h>
#include <sys/time.h>
#include <string.h>

#ifdef USE_HSM
#include "cr_hsm_code.h"
#include "cr_hsm_alloc.h"	// temp
#endif	// USE_HSM
// openssl
#include <openssl/sha.h>
#include <openssl/x509.h>

#include "cr_generate_id.h"
#include "cr_generate_id_private.h"
#include "cr_alloc.h"

// ビルド時の日時記録
static struct
{
	const u8 *title; const u8 *date; const u8 *time;
}
buildInfo = { "LIBGENID_BUILD_INFO:", __DATE__, __TIME__ };

static u64 generatingCount = 0;	// 現在生成中のID(起動時からの通算)

int cr_print_flag = 0;

// generate_id関数のイニシャライズ
int cr_generate_id_initialize( u8 err_buf[CR_ID_BUF_SIZE] )
{
	int ret_code = CR_GENID_SUCCESS;
	
	// ビルド情報がデッドストリップされないよう参照
	const u8 *dummyPtr = NULL;
	dummyPtr = buildInfo.title;
	dummyPtr = NULL;
	
	// 生成カウンタの初期化
	generatingCount = 0;
	
	// init for error info
	InitErrorInfo();
	
	// OpenSSL のメモリリーク防止のため、オリジナルのアロケータを使用。
	if ( CRYPTO_set_mem_functions( cr_mem_malloc, cr_mem_realloc, cr_mem_free ) != 1 )
	{
		CR_ERR_BUFFER *cr_err_buf = (CR_ERR_BUFFER *)err_buf;
		memset( cr_err_buf, 0, sizeof( CR_ERR_BUFFER ) );
		cr_err_buf->errorCode = CR_GENID_ERROR_SET_MEM_FUNCTIONS;
		GetErrorInfo( cr_err_buf->callStack, &cr_err_buf->recordSize );
#ifdef DEBUG_PRINT
		printf( "error(%d), CALL_STACK : %s\n", (int)cr_err_buf->errorCode, cr_err_buf->callStack );
#endif
		return CR_GENID_FAILED;
	}
	
#ifdef USE_HSM
	ret_code = hsm_initialize();
	if ( ret_code != CR_GENID_SUCCESS )
	{
		CR_ERR_BUFFER *cr_err_buf = (CR_ERR_BUFFER *)err_buf;
		memset( cr_err_buf, 0, sizeof( CR_ERR_BUFFER ) );
		cr_err_buf->errorCode = ret_code;
		GetErrorInfo( cr_err_buf->callStack, &cr_err_buf->recordSize );
#ifdef DEBUG_PRINT
		printf( "error(%d), CALL_STACK : %s\n", (int)cr_err_buf->errorCode, cr_err_buf->callStack );
#endif
		return CR_GENID_FAILED;
	}
#endif
	
	return ret_code;
}	// cr_generate_id_initialize


// generate_id関数のファイナライズ
int cr_generate_id_finalize( u8 err_buf[CR_ID_BUF_SIZE] )
{
	int ret_code = CR_GENID_SUCCESS;
	
	// init for error info
	InitErrorInfo();
	
	// HSM
#ifdef USE_HSM
	ret_code = hsm_finalize();
	if ( ret_code != CR_GENID_SUCCESS )
	{
		CR_ERR_BUFFER *cr_err_buf = (CR_ERR_BUFFER *)err_buf;
		memset( cr_err_buf, 0, sizeof( CR_ERR_BUFFER ) );
		cr_err_buf->errorCode = ret_code;
		GetErrorInfo( cr_err_buf->callStack, &cr_err_buf->recordSize );
#ifdef DEBUG_PRINT
		printf( "error(%d), CALL_STACK : %s\n", (int)cr_err_buf->errorCode, cr_err_buf->callStack );
#endif
	}
#endif // USE_HSM
	
	return ret_code;
}	// cr_generate_id_finalize

// generate_id 関数
int cr_generate_id( u32 device_id[CR_NUM_OF_DEVICEID], u8 id_buf[CR_ID_BUF_SIZE], u8 bonding_option )
{
    int i;
    int ret_code = CR_GENID_SUCCESS;
    CR_ID_BUFFER *cr_id_buf;
    EC_KEY *deviceKeyPair = NULL;

	// エラー発生時に備えて、エラーバッファの初期化とトータルのID生成カウントセット
	InitErrorInfo();
	generatingCount++;
	
#ifdef DEBUG_PRINT
    if( sizeof(CR_ID_BUFFER) != 256 ) {
		printf( "CR_ID_BUFFER size error. %d\n", sizeof(CR_ID_BUFFER) );
    }
    if( sizeof(CR_ERR_BUFFER) != 256 ) {
		printf( "CR_ERR_BUFFER size error. %d\n", sizeof(CR_ERR_BUFFER) );
    }
#endif

    //--------------------------------------------------------------
    // 暗号処理初期化
    //--------------------------------------------------------------
	cr_mem_bufmgr_initialize();

#ifdef MY_CRYPTO_DEBUG
    ERR_load_crypto_strings(); 
#endif /* MY_CRYPTO_DEBUG */

	
	// ダイジェストアルゴリズムを追加する
	OpenSSL_add_all_digests();

    //--------------------------------------------------------------
    // FuseIDバッファに固定データセット
    //--------------------------------------------------------------
    memset(id_buf, 0, CR_ID_BUF_SIZE);

    cr_id_buf = (CR_ID_BUFFER *)id_buf;
    cr_id_buf->magic_number	= CR_GEN_ID_MAGICCODE;	// HSM使用／未使用でマジックコードが変わる。
    cr_id_buf->version		= CR_GEN_ID_VERSION;

	//--------------------------------------------------------------
	// 引数のボンディングオプションをセット
	//--------------------------------------------------------------
	cr_id_buf->bonding_option = bonding_option;

	//--------------------------------------------------------------
	// device_id セット
	//--------------------------------------------------------------
	for(  i = 0 ; i < CR_NUM_OF_DEVICEID ; i++ ) {
		cr_id_buf->device_id[i] = device_id[i];
	}

#ifdef DEBUG_PRINT
  if( cr_print_flag ) {
    printf("device_id:\n");
    printf(" 0x%08x\n",     (unsigned int)device_id[0] );
    printf(" 0x%08x%08x\n", (unsigned int)device_id[2], (unsigned int)device_id[1] );
    printf(" 0x%08x%08x\n", (unsigned int)device_id[4], (unsigned int)device_id[3] );
    printf("\n");
  }
#endif /* DEBUG_PRINT */
	
    //--------------------------------------------------------------
    // タイムスタンプセット
    //--------------------------------------------------------------
	ret_code = GetTimestamp( &cr_id_buf->year,
							 &cr_id_buf->month,
							 &cr_id_buf->mday,
							 &cr_id_buf->hour,
							 &cr_id_buf->min,
							 &cr_id_buf->sec,
							 &cr_id_buf->expiryDate );	// デバイス証明書期限の元データもついでにセットしておく
	if ( ret_code != CR_GENID_SUCCESS ) {
		SetErrorInfo( __FUNCTION__, __LINE__ );
		goto end;
	}
	
    //--------------------------------------------------------------
    // 乱数を生成してセット
    //--------------------------------------------------------------
	ret_code = GenerateRandom( cr_id_buf->random, CR_RANDOM_LENGTH );
	if ( ret_code != CR_GENID_SUCCESS ) {
		SetErrorInfo( __FUNCTION__, __LINE__ );
		goto end;
	}
	DEBUG_PRINT_ARRAY( "rand:", (const char *)cr_id_buf->random, CR_RANDOM_LENGTH );

	//--------------------------------------------------------------
	// 楕円曲線鍵ペアを生成
	//--------------------------------------------------------------
	ret_code = GenarateECCKeyPair( &deviceKeyPair, cr_id_buf->devicePrivKey );
	if ( ret_code != CR_GENID_SUCCESS ) {
		SetErrorInfo( __FUNCTION__, __LINE__ );
		goto end;
	}
	
	//--------------------------------------------------------------
	// 生成した鍵ペアをECDSAで動作確認
	//--------------------------------------------------------------
	ret_code = TestECDSA( deviceKeyPair );
	if ( ret_code != CR_GENID_SUCCESS ) {
		goto end;
	}

	//--------------------------------------------------------------
	// デバイス証明書生成 + 署名の付与 + 証明書期限セット
	//--------------------------------------------------------------
	ret_code = GenerateCTRDeviceCert( deviceKeyPair,
									  cr_id_buf->device_id[0],
									  cr_id_buf->bonding_option,
									  cr_id_buf->deviceCertSign,
									  &cr_id_buf->expiryDate );
	if ( ret_code != CR_GENID_SUCCESS ) {
		SetErrorInfo( __FUNCTION__, __LINE__ );
		goto end;
	}

#if 0
	DEBUG_PRINT_ARRAY( "deviceCertSign:", (const char *)cr_id_buf->deviceCertSign, ECDSA_SIGN_LENGTH );
#endif

	//--------------------------------------------------------------
	// FuseIDバッファ全体のSHA256ハッシュを算出してセット
	//--------------------------------------------------------------
	SHA256(id_buf, CR_ID_BUF_SIZE - SHA256_DIGEST_LENGTH, cr_id_buf->hash);
	DEBUG_PRINT_ARRAY( "SHA256 Digest:", (const char *)cr_id_buf->hash, SHA256_DIGEST_LENGTH );

	//--------------------------------------------------------------
	// FuseID RAWデータ完成
	//--------------------------------------------------------------
	DEBUG_PRINT_ARRAY( "RAW eFuseID:", (const char *)id_buf, CR_ID_BUF_SIZE );
#ifdef DEBUG_OUTPUT_FILE
	DebugFileOutput( device_id[ 0 ], "raw", id_buf, CR_ID_BUF_SIZE );
#endif // DEBUG_OUTPUT_FILE

	//--------------------------------------------------------------
	// FuseIDバッファ全体をAES or RSAで暗号化
	//--------------------------------------------------------------
	ret_code = EncryptID( id_buf, id_buf, bonding_option );
	if( ret_code != CR_GENID_SUCCESS )	{
		SetErrorInfo( __FUNCTION__, __LINE__ );
		goto end;
	}

	DEBUG_PRINT_ARRAY( "ENC eFuseID:", (const char *)id_buf, CR_ID_BUF_SIZE );
#ifdef DEBUG_OUTPUT_FILE
	DebugFileOutput( device_id[ 0 ], "enc", id_buf, CR_ID_BUF_SIZE );
#endif // DEBUG_OUTPUT_FILE

  //--------------------------------------------------------------
  // 終了処理
  //--------------------------------------------------------------
end:

	/* id_buf[]にエラーログを書き込む。 */
	if ( ret_code != CR_GENID_SUCCESS )
	{
		CR_ERR_BUFFER *cr_err_buf = (CR_ERR_BUFFER *)id_buf;
		memset( cr_err_buf, 0, sizeof( CR_ERR_BUFFER ) );
		cr_err_buf->totalCount = generatingCount;
		cr_err_buf->magic_number = 0x01234567;
		cr_err_buf->device_id0 = device_id[0];
		cr_err_buf->errorCode = ret_code;
		GetErrorInfo( cr_err_buf->callStack, &cr_err_buf->recordSize );
		cr_err_buf->bonding_option = bonding_option;
#ifdef DEBUG_PRINT
		printf( "CALL_STACK : %s\n", cr_err_buf->callStack );
#endif
	}

	// リソースの解放
	if ( deviceKeyPair ) EC_KEY_free( deviceKeyPair );

	ERR_remove_state(0);
	EVP_cleanup();
	CRYPTO_cleanup_all_ex_data();

#ifdef MY_CRYPTO_DEBUG
	ERR_free_strings(); 
#endif /* MY_CRYPTO_DEBUG */

#if 0
	if ( cr_print_flag )
	{
		printf( "hsm   alloc counter : %d\n", my_hsm_get_alloc_counter() );
		printf( "hsmbn alloc counter : %d\n", my_bignum_get_alloc_counter() );
		printf( "miya  alloc counter : %d\n", cr_mem_get_counter() );
	}
#endif

	return ret_code; /* success */
}


#ifdef DEBUG_PRINT
void DebugPrintArray( char *pStr, const u8 *pData, int length )
{
	int i;
	if( cr_print_flag ) {
		printf( "%s", pStr );
		for( i = 0 ; i < length; i++ ) {
			if( (i % 16) == 0 ) printf("\n ");
			printf("%02X ", pData[ i ] );
		}
		printf("\n");
	}
}
#endif

void DebugFileOutput( u32 device_id, char *pSuffix, const u8 *pSrc, int length )
{
	if ( cr_print_flag )
	{
		// 証明書の書き込みテスト
		FILE	*fp;
		char	fn[256];
		sprintf( fn, "output/0x%08x.%s", (unsigned int)device_id, pSuffix );
		fp = fopen( fn, "wb" );
		fwrite( pSrc, length, 1, fp );
		fclose( fp );
	}
}