/* ====================================================================
 * 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"
#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 u64 generatingCount = 0;	// 現在生成中のID(起動時からの通算)

int cr_print_flag = 0;

// generate_id関数のイニシャライズ
int cr_generate_id_initialize( void )
{
	int ret_code = CR_GENID_SUCCESS;

#ifdef USE_HSM
	ret_code = hsm_initialize();
	if ( ret_code != CR_GENID_SUCCESS )
	{
		return 0;
	}
#endif

	// TODO: エラーコードを決める
	ret_code = 1;

	return ( ret_code );
}	// cr_generate_id_initialize


// generate_id関数のファイナライズ
int cr_generate_id_finalize( void )
{
	int	ok = CR_GENID_SUCCESS;
		
	// openssl	TOOD: ここでファイナライズ不要なら、削除
	ERR_remove_state(0);
	EVP_cleanup();
	CRYPTO_cleanup_all_ex_data();
	
	// HSM
#ifdef USE_HSM
	hsm_finalize();
#endif // USE_HSM

	// TODO: エラーコードを決める
	ok = 1;
	
	return ( ok );
}	// cr_generate_id_finalize

/* 
   unsigned long ERR_get_error(void);
   エラーが起こったとき id_buf にエラーログを返すか？
 */


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

	generatingCount++;

	StoreErrorInfo( 100, __FILE__, __LINE__ );
	ret_code = 100;
	goto end;

    if( sizeof(CR_ID_BUFFER) != 256 ) {
      ret_code = CR_GENID_ERROR_ID_BUF_SIZE;
      goto end;
    }

    if( sizeof(CR_ERR_BUFFER) > 256 ) {
      ret_code = CR_GENID_ERROR_ERR_BUF_SIZE;
      goto end;
    }

// ダイジェストアルゴリズムを追加する	TODO: 最終的にはどちらが正しい？
#if 0
	OpenSSL_add_all_digests();
#else
	EVP_add_digest( EVP_ecdsa() );
#endif

    //--------------------------------------------------------------
    // 暗号処理初期化
    //--------------------------------------------------------------

    // メモリリーク防止のため、オリジナルのアロケータを使用。
    CRYPTO_set_mem_functions(cr_mem_malloc, cr_mem_realloc, cr_mem_free);
    cr_mem_bufmgr_initialize();

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

    //--------------------------------------------------------------
    // 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->bondingOption = bondingOption;

	//--------------------------------------------------------------
	// serialNo.セット
	//--------------------------------------------------------------
	for(  i = 0 ; i < CR_NUM_OF_SERIAL ; i++ ) {
		cr_id_buf->serial[i] = serial[i];  /* serial[0] => ec priv key */
	}

#ifdef DEBUG_PRINT
  if( cr_print_flag ) {
    printf("serialNo:\n");
    printf(" 0x%08x\n",     (unsigned int)serial[0] );
    printf(" 0x%08x%08x\n", (unsigned int)serial[2], (unsigned int)serial[1] );
    printf(" 0x%08x%08x\n", (unsigned int)serial[4], (unsigned int)serial[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 ) {
		goto end;
	}

    //--------------------------------------------------------------
    // 乱数を生成してセット
    //--------------------------------------------------------------
	ret_code = GenerateRandom( cr_id_buf->random, CR_RANDOM_LENGTH );
	if ( ret_code != CR_GENID_SUCCESS ) {
		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 ) {
		goto end;
	}

	//--------------------------------------------------------------
	// 生成した鍵ペアをECDSAで動作確認
	//--------------------------------------------------------------
	ret_code = TestECDSA( deviceKeyPair );
	if ( ret_code != CR_GENID_SUCCESS ) {
		goto end;
	}

	//--------------------------------------------------------------
	// デバイス証明書生成 + 署名の付与 + 証明書期限セット
	//--------------------------------------------------------------
	ret_code = GenerateCTRDeviceCert( deviceKeyPair,
									  cr_id_buf->serial[0],
									  cr_id_buf->bondingOption,
									  cr_id_buf->deviceCertSign,
									  &cr_id_buf->expiryDate );
	if ( ret_code != CR_GENID_SUCCESS ) {
		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_EFUSE_ID_OUTPUT_FILE
	DebugFileOutput( serial[ 0 ], "raw", id_buf, CR_ID_BUF_SIZE );
#endif // DEBUG_EFUSE_ID_OUTPUT_FILE

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

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

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

	/* id_buf[]にエラーログを書き込む。 */
	// TODO: 仕様をFIXする必要あり	-> TORIAEZU version
	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->serial0 = serial[0];
		GetErrorInfo( &(cr_err_buf->errorCode),
						cr_err_buf->fileName, 
						&cr_err_buf->lineNumber );
		cr_err_buf->bondingOption = bondingOption;
	}

	// リソースの解放
	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 */

	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 deviceId, char *pSuffix, const u8 *pSrc, int length )
{
	if ( cr_print_flag )
	{
		// 証明書の書き込みテスト
		FILE	*fp;
		char	fn[256];
		sprintf( fn, "output/0x%08x.%s", (unsigned int)deviceId, pSuffix );
		fp = fopen( fn, "wb" );
		fwrite( pSrc, length, 1, fp );
		fclose( fp );
	}
}