#include "include\defines.h"
#include "include\boot\dc_header.h"
#ifdef SMALL
#include "crypto_small\sha512_pkcs5_2_small.h"
#else
#include "crypto_fast/sha512_pkcs5_2.h"
#endif

int dc_decrypt_header(dc_header *header, dc_pass *password)
{
	u8        dk[DISKKEY_SIZE];
	int       i, succs = 0;
	xts_key   hdr_key;
	dc_header hcopy;
	
	sha512_pkcs5_2(
		1000, password->pass, password->size, 
		header->salt, PKCS5_SALT_SIZE, dk, PKCS_DERIVE_MAX);

	for (i = 0; i < CF_CIPHERS_NUM; i++)
	{
		xts_set_key(dk, i, &hdr_key);

		xts_decrypt(pv(header), pv(&hcopy), sizeof(dc_header), 0, &hdr_key);

		/* Magic 'DCRP' */
		if (hcopy.sign != DC_VOLUME_SIGN) {
			continue;
		}
		/* copy decrypted part to output */
		memcpy(&header->sign, &hcopy.sign, DC_ENCRYPTEDDATASIZE);
		succs = 1; break;
	}

	/* prevent leaks */
	memset(dk, 0, sizeof(dk));
	memset(&hdr_key, 0, sizeof(xts_key));
	memset(&hcopy, 0, sizeof(dc_header));

	return succs;
}