#include <stdlib.h>
#include <stdio.h>
#include <stdint.h>
#include <stdbool.h>
#include <errno.h>
#include <unistd.h>
#include <math.h>
#include <fcntl.h>
#include <string.h>
#include "rng.h"

void decrypt_decoded_bytes( unsigned char *encrypted_bytes, size_t byte_count ){
	unsigned char dec_key[256];
	memset( (void*)&dec_key, 0, sizeof(dec_key) );
	size_t recv_key_size = 0;
	for ( ; recv_key_size < 256; recv_key_size++ )
		dec_key[recv_key_size] = (unsigned char)get_random_int( 1, 255 );
	size_t c_byte = 0;
	for ( ; c_byte < byte_count; c_byte++ )
		encrypted_bytes[c_byte] = encrypted_bytes[c_byte] ^ dec_key[byte_count % 255];
	return;
}

void extract_contained_binary(){
	double bin_hex_div = (double)strlen((const char*)BIN_HEX) / 2;
	if ( bin_hex_div == 0.5 ){
		fputs( "Error: binary hex is uneven number of characters\n", stderr );
		return;
	}
	size_t hex_byte_count = (size_t)bin_hex_div;
	unsigned char *decoded_bytes = (unsigned char*)calloc( hex_byte_count + 1, sizeof(unsigned char) );
	if ( decoded_bytes == NULL ){
		fprintf( stderr, "Error: failed to allocate memory for decoding: %s\n", strerror(errno) );
		return;
	}
	const char *bin_hex = (const char*)BIN_HEX;
	size_t byte_pos = 0;
	size_t dec_byte_pos = 0;
	char c_byte[2];
	memset( (void*)&c_byte, 0, sizeof(c_byte) );
	unsigned char decoded_byte = '\0';
	for ( ; dec_byte_pos < hex_byte_count; byte_pos += 2, dec_byte_pos++ ){
		memmove( (void*)&c_byte, (const void*)&bin_hex[byte_pos], 2 );
		decoded_byte = (unsigned char)strtoul( (const char*)&c_byte, NULL, 16 );
		if ( errno != 0 ){
			fprintf( stderr, "Error: failed to decode byte: %s\n", strerror(errno) );
			free( decoded_bytes );
			return;
		}
		decoded_bytes[dec_byte_pos] = decoded_byte;
	}
	decrypt_decoded_bytes( decoded_bytes, hex_byte_count );
	int out_fd = open( "/tmp/test_output_binary", O_RDWR | O_CREAT, 0755 );
	if ( out_fd == -1 ){
		fprintf( stderr, "Error: failed to open /tmp/test_output_binary for writing: %s\n", strerror(errno) );
		free( decoded_bytes );
		return;
	}
	if ( write(out_fd, (const void*)decoded_bytes, hex_byte_count + 1) == -1 ){
		fprintf( stderr, "Error: failed to write /tmp/test_output_binary: %s\n", strerror(errno) );
		free( decoded_bytes );
		close( out_fd );
		return;
	}
	close( out_fd );
	free( decoded_bytes );
	fputs( "Binary extracted, check /tmp for test_output_binary\n", stdout );
	return;
}

int main( int argc, char *const *args ){
	init_rng();
	extract_contained_binary();
	return 0;
}