/* Copyright (c) 2013 Sven Weidauer
 * Code is under the MIT license, see the LICENSE file.
 */

#include "Common.h"
#include "Buffer.h"

#include <CoreFoundation/CoreFoundation.h>
#include <CommonCrypto/CommonCrypto.h>


void ReadHeader( struct buffer *buf, struct header *header )
{
	Read( buf, 4, header->magic );
	header->payload_size = Read32( buf );
	Read( buf, 8, header->header_md5 );
	Read( buf, 7, header->etl );
	Read( buf, 1, &header->unused1 );
	header->password_len = Read16( buf );
	header->padding_len = Read16( buf );
	Read( buf, 4, header->unused2 );
	Read( buf, 16, header->plaintext_md5 );
}

int main( int argc, const char *argv[] ) 
{
	if (argc != 3) {
		fprintf( stderr, "Call %s <input> <output>\n", argv[0] );
		return 1;
	}
	
	struct buffer buf = { 0 };
	
	if (!OpenFile( &buf, argv[1] )) {
		fprintf( stderr, "Cannot open file %s\n", argv[1] );
		return 1;
	}
	
	printf( "Openend file, size: %zu\n", buf.length );
	
	struct header header = { 0 };
	ReadHeader( &buf, &header );

	if (memcmp( header.magic, "CFG1", 4 ) != 0) {
		fprintf( stderr, "Error: invalid magic\n" );
		return 1;
	}
	
	char header_data[HeaderSize];
	memcpy( header_data, buf.start, sizeof header_data );
	memset( &header_data[8], 0, 8 );
	
	unsigned char header_md[CC_MD5_DIGEST_LENGTH];
	CC_MD5( header_data, sizeof header_data, header_md );
	if (memcmp( header.header_md5, header_md, sizeof header.header_md5 ) != 0) {
		printf( "Warning: Header MD5 doesn't match\n" );
	}
	
	printf( "Payload size: %d\n", header.payload_size );
	printf( "Padding: %d\n", header.padding_len );
	printf( "Password len: %d\n", header.password_len );
	printf( "ETL: %s\n", header.etl );
	
	if (buf.length - header.payload_size != HeaderSize) {
		fprintf( stderr, "Error: Invalid file size\n" );
		return 1;
	}
	
	size_t plain_length = header.payload_size;
	void *plain = malloc( plain_length );
	if (!plain) {
		fprintf( stderr, "Error: Not enough memory for decryption\n" );
		return 1;
	}
	
	CCCryptorRef cryptor;
	size_t result_length = 0;
	if (CCCrypt( kCCDecrypt, kCCAlgorithmAES, kCCOptionECBMode, aes_key, sizeof aes_key, NULL, buf.next, plain_length, plain, plain_length, &result_length ) != kCCSuccess) {
		fprintf( stderr, "Error: Cannot decrypt\n" );
		return 1;
	}
	
	result_length -= header.padding_len;
	
	int fdout = open( argv[2], O_WRONLY | O_CREAT | O_TRUNC, 0600 );
	if (fdout < 0) {
		fprintf( stderr, "Error: cannot write output file\n" );
		return 1;
	}
	if (write( fdout, plain, result_length  ) != result_length) {
		fprintf( stderr, "Error: cannot write output file\n" );
		return 1;
	}
	close( fdout );
	
	unsigned char md[CC_MD5_DIGEST_LENGTH];
	CC_MD5( plain, plain_length, md );

	if (memcmp( md, header.plaintext_md5, sizeof md ) != 0) {
		printf( "Warning: payload md5 doesn't match\n" );
	}
}