bool encriptador_AES::inicializar_AES(string _contraseña, int _rondas)
{
	u_char key[32], iv[32];
	u_int salto[2] = { MINIMO_SALTO, MAXIMO_SALTO };
	if (EVP_BytesToKey(EVP_aes_256_cbc(), EVP_sha1(), (u_char*)&salto, (u_char*)_contraseña.c_str(), _contraseña.length(), _rondas, key, iv) != 32)
		return false;
	EVP_CIPHER_CTX_init(&en);
	EVP_EncryptInit_ex(&en, EVP_aes_256_cbc(), NULL, key, iv);
	EVP_CIPHER_CTX_init(&de);
	EVP_DecryptInit_ex(&de, EVP_aes_256_cbc(), NULL, key, iv);
	return true;
}
 
bytes encriptador_AES::encriptar_AES(u_char *_entrada, int _tamaño)
{
	int c_len = _tamaño + AES_BLOCK_SIZE, f_len = 0;
	u_char *salida = (u_char*)malloc(c_len);
	EVP_EncryptInit_ex(&en, NULL, NULL, NULL, NULL);
	EVP_EncryptUpdate(&en, salida, &c_len, _entrada, _tamaño);
	EVP_EncryptFinal_ex(&en, salida + c_len, &f_len);
	return bytes(salida, c_len + f_len);
}
 
string encriptador_AES::desencriptar_AES(u_char *_entrada, int _tamaño)
{
	int p_len = _tamaño, f_len = 0;
	u_char *salida = (u_char*)malloc(p_len);
	EVP_DecryptInit_ex(&de, NULL, NULL, NULL, NULL);
	EVP_DecryptUpdate(&de, salida, &p_len, _entrada, _tamaño);
	EVP_DecryptFinal_ex(&de, salida + p_len, &f_len);
	string ret;
	for (int i = 0; i < p_len + f_len; i++) ret += salida[i];
	delete salida;
	return ret;
}

almacen::almacen(string _directorio)
{
	directorio = _directorio;
	cifrador.inicializar_AES(CLAVE_AES, 5);
}
 
void almacen::escribir_cifrado(bytes _datos)
{
	ofstream archivo;
	archivo.open(directorio, ofstream::app | ofstream::binary);
	_datos = cifrador.encriptar_AES(get<CONTENIDO>(_datos), get<TAMAÑO>(_datos));
	archivo.write((char*)get<CONTENIDO>(_datos), get<TAMAÑO>(_datos));
	archivo.close();
}
 
string almacen::leer_descifrado()
{
	//probar que vaya parte a parte mejor, en vez de entera.
	ifstream archivo;
	archivo.open(directorio, ifstream::in | ifstream::binary);
	archivo.seekg(0, ifstream::end);
	streampos tamaño = archivo.tellg();
	char *contenido = new char[tamaño]();
	archivo.seekg(0, ifstream::beg);
	archivo.read(contenido, tamaño);
	archivo.close();
	string ret = cifrador.desencriptar_AES((u_char*)contenido, tamaño);
	delete contenido;
	return ret;
}

_archivo.escribir_cifrado(bytes((u_char*)contenido.c_str(), contenido.length()));

int main(){
almacen o("C:\\Users\\username\\Desktop\\PRUEBA.txt");
	cout << o.leer_descifrado();
}