#include <stdio.h>
#include <openssl/dh.h>

int main (int argc, const char * argv[]) {

	DH *dh1,*dh2;
	int i;
	unsigned char *key1, *key2;
	
	dh1=DH_generate_parameters(1024,2,NULL,NULL);

	DHparams_print_fp(stdout, dh1);	
	
	dh2=DH_new();	
	*dh2=*dh1;
	
	if (!DH_generate_key(dh1))
		printf("Erro a gerar chave privada 1\n");
	
	
	if (!DH_generate_key(dh2))
		printf("Erro a gerar chave privada 2\n");
	
	key1 = (unsigned char *)malloc(DH_size(dh1));
	key2 = (unsigned char *)malloc(DH_size(dh2));	

	if (! DH_compute_key(key1,dh2->pub_key,dh1))
		printf("Erro a gerar chave secreta 1\n");
	
	if (! DH_compute_key(key2,dh1->pub_key,dh2))
		printf("Erro a gerar chave secreta 2\n");

	for(i=0; i<DH_size(dh1); i++) {
		if (key1[i]!=key2[i])
			printf("ERROR!!!\n");
	}

    return 0;
}