//////////////////////////////////////////////////
// Bakalarska praca                             //
// Meno studenta: Tomas Lukac                   //
// Veduci BP: prof. Ing. Milos Drutarovsky CSc. //
// Skola: KEMT FEI TUKE                         //
// Datum poslednej upravy: 6.4.2020             //
//////////////////////////////////////////////////

#include "kryptografia.h"
#include <wolfssl/user_settings.h>


WOLFSSL_CTX* nastavit_ctx_klient()
{
  WOLFSSL_METHOD* method;
  wolfSSL_Init();
  method = wolfTLSv1_2_client_method();
  WOLFSSL_CTX *ctx = wolfSSL_CTX_new(method);
  if (ctx == NULL) 
  {
    printf("Nepodarilo sa inicializovat WOLFSSL_CTX\n");
    return NULL;
  }
  return ctx;
}
 
WOLFSSL_CTX* nastavit_ctx_server()
{
  WOLFSSL_METHOD* method;
  wolfSSL_Init();
  method = wolfTLSv1_2_server_method();
  WOLFSSL_CTX *ctx = wolfSSL_CTX_new(method);

  if (ctx == NULL) 
  {
    printf("Nepodarilo sa inicializovat WOLFSSL_CTX\n");
    return NULL;
  }
  return ctx;
}

void nastav_sifry(WOLFSSL_CTX* ctx, const char* sifry)
{
  wolfSSL_CTX_set_cipher_list(ctx, sifry);
}

void nastav_funkciu(nastavenia_aplikacie* nastavenia, hashovacia_funkcia funkcia)
{
  nastavenia->funkcia = funkcia;
}

void zobraz_sifru(WOLFSSL* ssl)
{
  printf ("Pouzita sifra: %s\n", wolfSSL_get_cipher(ssl));
}

int generovat_rsa_certifikat(int pocet_bitov, int exponent, int algoritmus, char* krajina, char* mesto, char* domena, char* email)
{
  int uspech = 0;

  //premenne pre ukladanie dat autority
  RsaKey kluc_autorita;
  byte der_autorita_certifikat[VELKOST_BUFFERA];
  int der_autorita_certifikat_velkost;
  byte der_autorita_kluc[VELKOST_BUFFERA];
  int der_autorita_kluc_velkost;
  
  //nacitanie certifikatu autority vo formate DER ulozeneho v subore certifikaty/autorita/autorita_rsa.der
  printf("Nacitavanie certifikatu autority zo suboru.\n");
  FILE* subor = fopen("../certifikaty/autorita/autorita_rsa.der", "rb");
  der_autorita_certifikat_velkost = fread(der_autorita_certifikat, 1, VELKOST_BUFFERA , subor);
  if(der_autorita_certifikat_velkost < 0)
  {
    fprintf(stderr, "Nepodarilo sa nacitat obsah suboru. \n");
    return -1;
  }
  fclose(subor);
  printf("Nacitanie ukoncene. Bolo nacitanych %d bajtov.\n", der_autorita_certifikat_velkost);

  //vytvorenie struktury RsaKey autority zo sukromneho kluca vo formate DER ulozeneho v subore certifikaty/autorita/autorita_kluc_rsa.der
  printf("Nacitavanie sukromneho kluca autority zo suboru.\n");
  subor = fopen("../certifikaty/autorita/autorita_kluc_rsa.der", "rb");
  der_autorita_kluc_velkost = fread(der_autorita_kluc, 1, VELKOST_BUFFERA , subor);
  if(der_autorita_kluc_velkost < 0)
  {
    fprintf(stderr, "Nepodarilo sa nacitat obsah suboru. \n");
    return -1;
  }
  fclose(subor);
  printf("Nacitanie ukoncene. Bolo nacitanych %d bajtov.\n", der_autorita_kluc_velkost);
  
  word32 index = 0; //nastavi odkial zacat citat buffer
  wc_InitRsaKey(&kluc_autorita, 0);
  uspech = wc_RsaPrivateKeyDecode(der_autorita_kluc, &index, &kluc_autorita, der_autorita_kluc_velkost);
  if(uspech != 0)
  {
    fprintf(stderr, "Nepodarilo sa vytvorit RsaKey strukturu z kluca autority\n");
    return -1;
  }
  
  //generator nahodnych cisel
  WC_RNG generator;
  wc_InitRng(&generator);
  
  //premenne pre ukladanie dat vygenerovaneho kluca
  RsaKey kluc;
  byte der_kluc[VELKOST_BUFFERA];
  word32 der_kluc_velkost;
  byte pem_kluc[VELKOST_BUFFERA];
  word32 pem_kluc_velkost;

  //vygenerovanie kluca
  wc_InitRsaKey(&kluc, 0);
  uspech = wc_MakeRsaKey(&kluc, pocet_bitov, exponent, &generator);
  if(uspech != 0)
  {
    fprintf(stderr, "Chyba pri generovani kluca.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  uspech = wc_RsaKeyToDer(&kluc, der_kluc, sizeof(der_kluc));
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri vytvoreni suboru DER so sukromnym klucom.\n");
    return -1;
  }
  der_kluc_velkost = uspech;
  memset(pem_kluc, 0, sizeof(pem_kluc));
  uspech = wc_DerToPem(der_kluc, der_kluc_velkost, pem_kluc, sizeof(pem_kluc), PRIVATEKEY_TYPE);
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri vytvoreni suboru PEM so sukromnym klucom\n");
    return -1;
  }
  pem_kluc_velkost = uspech;

  //zapis vygenerovaneho klucu do suboru
  subor = fopen("../certifikaty/vygenerovany_kluc.key", "wb");
  fwrite(pem_kluc, 1, pem_kluc_velkost, subor);
  fclose(subor);
  
  printf("RSA kluc bol uspesne vygenerovany.\n");
  
  //premenne pre ukladanie dat certifikatu
  Cert certifikat;
  byte der_certifikat[VELKOST_BUFFERA];
  word32 der_certifikat_velkost;
  byte pem_certifikat[VELKOST_BUFFERA];
  word32 pem_certifikat_velkost;

  //vygenerovanie a podpis certifikatu
  wc_InitCert(&certifikat);
  strncpy(certifikat.subject.country, krajina, CTC_NAME_SIZE);
  strncpy(certifikat.subject.state, "-", CTC_NAME_SIZE);
  strncpy(certifikat.subject.locality, mesto, CTC_NAME_SIZE);
  strncpy(certifikat.subject.org, "-", CTC_NAME_SIZE);
  strncpy(certifikat.subject.unit, "-", CTC_NAME_SIZE);
  strncpy(certifikat.subject.commonName, domena, CTC_NAME_SIZE);
  strncpy(certifikat.subject.email, email, CTC_NAME_SIZE);
  certifikat.isCA = 0;
  certifikat.sigType = algoritmus;
  certifikat.daysValid = 1825;
  uspech = wc_SetIssuerBuffer(&certifikat, der_autorita_certifikat, der_autorita_certifikat_velkost);
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri nastaveni autority.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  uspech = wc_MakeCert(&certifikat, der_certifikat, sizeof(der_certifikat), &kluc, NULL, &generator);
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri vytvoreni suboru DER s certifikatom.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  uspech = wc_SignCert(certifikat.bodySz, certifikat.sigType, der_certifikat, sizeof(der_certifikat), &kluc_autorita, NULL, &generator);
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri podpisovani certifikatu.\n");
    return -1;
  }
  der_certifikat_velkost = uspech;
  memset(pem_certifikat, 0, sizeof(pem_certifikat));
  uspech = wc_DerToPem(der_certifikat, der_certifikat_velkost, pem_certifikat, sizeof(pem_certifikat), CERT_TYPE);
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri vytvoreni suboru PEM s certifikatom.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  pem_certifikat_velkost = uspech;

  //zapis vygenerovaneho klucu do suboru
  subor = fopen("../certifikaty/vygenerovany_certifikat.pem", "wb");
  fwrite(pem_certifikat, 1, pem_certifikat_velkost, subor);
  fclose(subor);

  printf("Certifikat bol uspesne vygenerovany\n");
  
  return 0;
}

int generovat_ecc_certifikat(int pocet_bitov, ecc_curve_id kluc_krivka, int algoritmus, char* krajina, char* mesto, char* domena, char* email)
{
  int uspech = 0 ;

  //premenne pre ukladanie dat autority
  ecc_key kluc_autorita;
  byte der_autorita_certifikat[VELKOST_BUFFERA];
  int der_autorita_certifikat_velkost;
  byte der_autorita_kluc[VELKOST_BUFFERA];
  int der_autorita_kluc_velkost;
  
  //nacitanie certifikatu autority vo formate DER ulozeneho v subore certifikaty/autorita/autorita_ecc.der
  printf("Nacitavanie certifikatu autority zo suboru.\n");
  FILE* subor = fopen("../certifikaty/autorita/autorita_ecc.der", "rb");
  der_autorita_certifikat_velkost = fread(der_autorita_certifikat, 1, VELKOST_BUFFERA , subor);
  if(der_autorita_certifikat_velkost < 0)
  {
    fprintf(stderr, "Nepodarilo sa nacitat obsah suboru. \n");
    return -1;
  }
  fclose(subor);
  printf("Nacitanie ukoncene. Bolo nacitanych %d bajtov.\n", der_autorita_certifikat_velkost);

  //vytvorenie struktury ecc_key autority zo sukromneho kluca vo formate DER ulozeneho v subore certifikaty/autorita/autorita_kluc_ecc.der
  printf("Nacitavanie sukromneho kluca autority zo suboru.\n");
  subor = fopen("../certifikaty/autorita/autorita_kluc_ecc.der", "rb");
  der_autorita_kluc_velkost = fread(der_autorita_kluc, 1, VELKOST_BUFFERA , subor);
  if(der_autorita_kluc_velkost < 0)
  {
    fprintf(stderr, "Nepodarilo sa nacitat obsah suboru. \n");
    return -1;
  }
  fclose(subor);
  printf("Nacitanie ukoncene. Bolo nacitanych %d bajtov.\n", der_autorita_kluc_velkost);

  word32 index = 0; //nastavi odkial zacat citat buffer
  wc_ecc_init(&kluc_autorita);
  uspech = wc_EccPrivateKeyDecode(der_autorita_kluc, &index, &kluc_autorita, (word32)der_autorita_kluc_velkost);
  if(uspech != 0)
  {
    fprintf(stderr, "Chyba pri dekodovani kluca autority.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  
  //generator nahodnych cisel
  WC_RNG generator;
  wc_InitRng(&generator);
  
  //premenne pre ukladanie dat vygenerovaneho kluca
  ecc_key kluc;
  byte der_kluc[VELKOST_BUFFERA];
  word32 der_kluc_velkost;
  byte pem_kluc[VELKOST_BUFFERA];
  word32 pem_kluc_velkost;

  //vygenerovanie kluca
  wc_ecc_init(&kluc);
  uspech = wc_ecc_make_key_ex(&generator, pocet_bitov, &kluc, kluc_krivka);
  if(uspech != 0)
  {
    fprintf(stderr, "Chyba pri generovani kluca.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  uspech = wc_EccKeyToDer(&kluc, der_kluc, sizeof(der_kluc));
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri vytvoreni suboru DER so sukromnym klucom.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  der_kluc_velkost = uspech;
  uspech = wc_DerToPem(der_kluc, der_kluc_velkost, pem_kluc, sizeof(pem_kluc), ECC_PRIVATEKEY_TYPE);
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri vytvoreni suboru PEM so sukromnym klucom.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  pem_kluc_velkost = uspech;

  //zapis vygenerovaneho kluca do suboru
  subor = fopen("../certifikaty/vygenerovany_kluc.key", "wb");
  fwrite(pem_kluc, 1, pem_kluc_velkost, subor);
  fclose(subor);
  
  printf("ECC kluc bol uspesne vygenerovany.\n");

  //premenne pre ukladanie dat certifikatu
  Cert certifikat;
  byte der_certifikat[VELKOST_BUFFERA];
  word32 der_certifikat_velkost;
  byte pem_certifikat[VELKOST_BUFFERA];
  word32 pem_certifikat_velkost;
 
  //vygenerovanie a podpis certifikatu
  wc_InitCert(&certifikat);
  strncpy(certifikat.subject.country, krajina, CTC_NAME_SIZE);
  strncpy(certifikat.subject.state, "-", CTC_NAME_SIZE);
  strncpy(certifikat.subject.locality, mesto, CTC_NAME_SIZE);
  strncpy(certifikat.subject.org, "-", CTC_NAME_SIZE);
  strncpy(certifikat.subject.unit, "-", CTC_NAME_SIZE);
  strncpy(certifikat.subject.commonName, domena, CTC_NAME_SIZE);
  strncpy(certifikat.subject.email, email, CTC_NAME_SIZE);
  certifikat.isCA = 0;
  certifikat.sigType = algoritmus;
  certifikat.daysValid = 1825;
  uspech = wc_SetIssuerBuffer(&certifikat, der_autorita_certifikat, der_autorita_certifikat_velkost);
  if(uspech != 0)
  {
    fprintf(stderr, "Nastala chyba pri nastaveni autority.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  uspech = wc_MakeCert(&certifikat, der_certifikat, sizeof(der_certifikat), NULL, &kluc, &generator);
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri vytvoreni suboru DER s certifikatom.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  uspech = wc_SignCert(certifikat.bodySz, certifikat.sigType, der_certifikat, sizeof(der_certifikat), NULL, &kluc_autorita, &generator);
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri podpisovani certifikatu.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  der_certifikat_velkost = uspech;
  memset(pem_certifikat, 0, sizeof(pem_certifikat));
  uspech = wc_DerToPem(der_certifikat, der_certifikat_velkost, pem_certifikat, sizeof(pem_certifikat), CERT_TYPE);
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri vytvoreni suboru PEM s certifikatom.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  pem_certifikat_velkost = uspech;

  //zapis vygenerovaneho klucu do suboru
  subor = fopen("../certifikaty/vygenerovany_certifikat.pem", "wb");
  fwrite(pem_certifikat, 1, pem_certifikat_velkost, subor);
  fclose(subor);

  return 0; 
}

int vygenerovat_rsa_ziadost(int pocet_bitov, int exponent, int algoritmus, char* krajina, char* mesto, char* domena, char* email)
{
  int uspech = 0;

  //generator nahodnych cisel
  WC_RNG generator;
  wc_InitRng(&generator);

  //premenne pre ukladanie dat vygenerovaneho kluca
  RsaKey kluc;
  byte der_kluc[VELKOST_BUFFERA];
  word32 der_kluc_velkost;
  byte pem_kluc[VELKOST_BUFFERA];
  word32 pem_kluc_velkost;

  //vygenerovanie kluca
  wc_InitRsaKey(&kluc, 0);
  uspech = wc_MakeRsaKey(&kluc, pocet_bitov, exponent, &generator);
  if(uspech != 0)
  {
    fprintf(stderr, "Chyba pri generovani kluca.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  uspech = wc_RsaKeyToDer(&kluc, der_kluc, sizeof(der_kluc));
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri vytvoreni suboru DER so sukromnym klucom.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  der_kluc_velkost = uspech;
  memset(pem_kluc, 0, sizeof(pem_kluc));
  pem_kluc_velkost = wc_DerToPem(der_kluc, der_kluc_velkost, pem_kluc, sizeof(pem_kluc), PRIVATEKEY_TYPE);
  if(pem_kluc_velkost < 0)
  {
    fprintf(stderr, "Nastala chyba pri vytvoreni suboru PEM so sukromnym klucom\n");
    return -1;
  }

  //premenne pre ukladanie dat ziadosti
  Cert ziadost;
  byte der_ziadost[VELKOST_BUFFERA];
  word32 der_ziadost_velkost;
  byte pem_ziadost[VELKOST_BUFFERA];
  word32 pem_ziadost_velkost;
 
  //vygenerovanie a podpis ziadosti
  wc_InitCert(&ziadost);
  strncpy(ziadost.subject.country, krajina, CTC_NAME_SIZE);
  strncpy(ziadost.subject.state, "-", CTC_NAME_SIZE);
  strncpy(ziadost.subject.locality, mesto, CTC_NAME_SIZE);
  strncpy(ziadost.subject.org, "-", CTC_NAME_SIZE);
  strncpy(ziadost.subject.unit, "-", CTC_NAME_SIZE);
  strncpy(ziadost.subject.commonName, domena, CTC_NAME_SIZE);
  strncpy(ziadost.subject.email, email, CTC_NAME_SIZE);
  ziadost.sigType = algoritmus;
  uspech = wc_MakeCertReq(&ziadost, der_ziadost, sizeof(der_ziadost), &kluc, NULL);
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri generovani ziadosti.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  uspech = wc_SignCert(ziadost.bodySz, ziadost.sigType, der_ziadost, sizeof(der_ziadost), &kluc, NULL, &generator);
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri podpisovani ziadosti.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  der_ziadost_velkost = uspech;
  memset(pem_ziadost, 0, sizeof(pem_ziadost));
  uspech = wc_DerToPem(der_ziadost, der_ziadost_velkost, pem_ziadost, sizeof(pem_ziadost), CERTREQ_TYPE);
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri vytvoreni suboru PEM so ziadostou.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
}

int vygenerovat_ecc_ziadost(int pocet_bitov, ecc_curve_id kluc_krivka, int algoritmus, char* krajina, char* mesto, char* domena, char* email)
{
  int uspech = 0;

  //generator nahodnych cisel
  WC_RNG generator;
  wc_InitRng(&generator);

  //premenne pre ukladanie dat vygenerovaneho kluca
  ecc_key kluc;
  byte der_kluc[VELKOST_BUFFERA];
  word32 der_kluc_velkost;
  byte pem_kluc[VELKOST_BUFFERA];
  word32 pem_kluc_velkost;

  //vygenerovanie kluca
  wc_ecc_init(&kluc);
  uspech = wc_ecc_make_key_ex(&generator, pocet_bitov, &kluc, kluc_krivka);
  if(uspech != 0)
  {
    fprintf(stderr, "Chyba pri generovani kluca.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  uspech = wc_EccKeyToDer(&kluc, der_kluc, sizeof(der_kluc));
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri vytvoreni suboru DER so sukromnym klucom.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  der_kluc_velkost = uspech;
  memset(pem_kluc, 0, sizeof(pem_kluc));
  pem_kluc_velkost = wc_DerToPem(der_kluc, der_kluc_velkost, pem_kluc, sizeof(pem_kluc), ECC_PRIVATEKEY_TYPE);
  if(pem_kluc_velkost < 0)
  {
    fprintf(stderr, "Nastala chyba pri vytvoreni suboru PEM so sukromnym klucom\n");
    return -1;
  }

  //premenne pre ukladanie dat ziadosti
  Cert ziadost;
  byte der_ziadost[VELKOST_BUFFERA];
  word32 der_ziadost_velkost;
  byte pem_ziadost[VELKOST_BUFFERA];
  word32 pem_ziadost_velkost;
 
  //vygenerovanie a podpis ziadosti
  wc_InitCert(&ziadost);
  strncpy(ziadost.subject.country, krajina, CTC_NAME_SIZE);
  strncpy(ziadost.subject.state, "-", CTC_NAME_SIZE);
  strncpy(ziadost.subject.locality, mesto, CTC_NAME_SIZE);
  strncpy(ziadost.subject.org, "-", CTC_NAME_SIZE);
  strncpy(ziadost.subject.unit, "-", CTC_NAME_SIZE);
  strncpy(ziadost.subject.commonName, domena, CTC_NAME_SIZE);
  strncpy(ziadost.subject.email, email, CTC_NAME_SIZE);
  ziadost.sigType = algoritmus;
  uspech = wc_MakeCertReq(&ziadost, der_ziadost, sizeof(der_ziadost), NULL, &kluc);
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri generovani ziadosti.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  uspech = wc_SignCert(ziadost.bodySz, ziadost.sigType, der_ziadost, sizeof(der_ziadost), NULL, &kluc, &generator);
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri podpisovani ziadosti.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
  der_ziadost_velkost = uspech;
  memset(pem_ziadost, 0, sizeof(pem_ziadost));
  uspech = wc_DerToPem(der_ziadost, der_ziadost_velkost, pem_ziadost, sizeof(pem_ziadost), CERTREQ_TYPE);
  if(uspech < 0)
  {
    fprintf(stderr, "Nastala chyba pri vytvoreni suboru PEM so ziadostou.\nCislo chyby: %d\nDovod chyby: %s\n", uspech, wc_GetErrorString(uspech));
    return -1;
  }
}

int nacitat_certifikaty(WOLFSSL_CTX* ctx, const char* subor_certifikat, const char* subor_kluc)
{
  int uspech = 0;
  
  //nacitanie certifikatu do WOLFSSL_CTX struktury, cesta ku suboru sa predava z argumentu funkcie
  uspech = wolfSSL_CTX_use_certificate_file(ctx, subor_certifikat, SSL_FILETYPE_PEM);
  if(uspech != SSL_SUCCESS)
  {
    printf("Chyba pri nacitani certifikatu\n");
    return -1;
  }
  
  //nacitanie sukromneho kluca do WOLFSSL_CTX struktury, cesta ku suboru sa predava z argumentu funkcie
  uspech = wolfSSL_CTX_use_PrivateKey_file(ctx, subor_kluc, SSL_FILETYPE_PEM);
  if(uspech != SSL_SUCCESS)
  {
    printf("Chyba pri nacitani kluca\n");
    return -1;
  }
  
  printf("Certifikat a kluc boli uspesne nacitane\n");
  
  //kontrola sukromneho kluca
  if(!wolfSSL_CTX_check_private_key(ctx))
  {
    fprintf(stderr, "Sukromny kluc sa nezhoduje s certifikatom\n");
    return -1;
  } 
  return 0;
}

void zobraz_certifikat(WOLFSSL* ssl)
{
  printf("Informacie o certifikate:\n");
  WOLFSSL_X509 *certifikat = wolfSSL_get_peer_certificate(ssl);
  char* pole;
  if (!certifikat)
  {
    printf("Nebolo mozne ziskat ziadny certifikat\n");
  }
  if ((pole = wolfSSL_X509_NAME_oneline(wolfSSL_X509_get_subject_name(certifikat), 0, 0))) 
  {
    printf("Subjekt: %s\n", pole);
    //wolfSSL_free(line);
  }
  if ((pole = wolfSSL_X509_NAME_oneline(wolfSSL_X509_get_issuer_name(certifikat), 0, 0))) 
  {
    printf("Certifikacna autorita: %s\n", pole);
    //wolfSSL_free(line);
  }
  X509_free(certifikat);
}

byte* generovat_kontrolny_sucet_suboru(hashovacia_funkcia funkcia, char* cesta, int* velkost)
{
  int pocet_bajtov;
  FILE* subor = fopen(cesta, "rb");
  unsigned char data[1024];
  byte* vysledok;
  if(funkcia == funkcia_SHA)
  {
    *velkost = SHA_DIGEST_SIZE;
    vysledok = calloc(SHA_DIGEST_SIZE, sizeof(byte));
    Sha sha;
    wc_InitSha(&sha);
    while ((pocet_bajtov = fread (data, 1, 1024, subor)) != 0)
    wc_ShaUpdate(&sha, data, pocet_bajtov);  
    wc_ShaFinal(&sha, vysledok);
  }
  else if(funkcia == funkcia_SHA224)
  {
    *velkost = SHA224_DIGEST_SIZE;
    vysledok = calloc(SHA224_DIGEST_SIZE, sizeof(byte));
    Sha224 sha;
    wc_InitSha224(&sha);
    while ((pocet_bajtov = fread (data, 1, 1024, subor)) != 0)
    wc_Sha224Update(&sha, data, pocet_bajtov);  
    wc_Sha224Final(&sha, vysledok);
  }
  else if(funkcia == funkcia_SHA256)
  {
    *velkost = SHA256_DIGEST_SIZE;
    byte* vysledok = calloc(SHA256_DIGEST_SIZE, sizeof(byte));
    Sha256 sha;
    wc_InitSha256(&sha);
    while ((pocet_bajtov = fread (data, 1, 1024, subor)) != 0)
    wc_Sha256Update(&sha, data, pocet_bajtov);  
    wc_Sha256Final(&sha, vysledok);
  }
  else if(funkcia == funkcia_SHA384)
  {
    *velkost = SHA384_DIGEST_SIZE;
    vysledok = calloc(SHA384_DIGEST_SIZE, sizeof(byte));
    Sha384 sha;
    wc_InitSha384(&sha);
    while ((pocet_bajtov = fread (data, 1, 1024, subor)) != 0)
    wc_Sha384Update(&sha, data, pocet_bajtov);  
    wc_Sha384Final(&sha, vysledok);
  }
  else if(funkcia == funkcia_SHA512)
  {
    *velkost = WC_SHA512_DIGEST_SIZE;
    vysledok = calloc(WC_SHA512_DIGEST_SIZE, sizeof(byte));
    wc_Sha512 sha;
    wc_InitSha512(&sha);
    while ((pocet_bajtov = fread (data, 1, 1024, subor)) != 0)
    wc_Sha512Update(&sha, data, pocet_bajtov);  
    wc_Sha512Final(&sha, vysledok);
  }
  else if(funkcia == funkcia_BLAKE2B)
  {
    *velkost = 64;
    vysledok = calloc(64, sizeof(byte));
    Blake2b b2b;
    wc_InitBlake2b(&b2b, 64);
    while ((pocet_bajtov = fread (data, 1, 1024, subor)) != 0)
    wc_Blake2bUpdate(&b2b, data, pocet_bajtov);  
    wc_Blake2bFinal(&b2b, vysledok, 64);
  }
  else if(funkcia == funkcia_RIPEMD160)
  {
    *velkost = RIPEMD_DIGEST_SIZE;
    vysledok = calloc(RIPEMD_DIGEST_SIZE, sizeof(byte));
    RipeMd ripemd;
    wc_InitRipeMd(&ripemd);
    while ((pocet_bajtov = fread (data, 1, 1024, subor)) != 0)
    wc_RipeMdUpdate(&ripemd, data, pocet_bajtov);  
    wc_RipeMdFinal(&ripemd, vysledok);
  }
  else
  {
    fprintf(stderr, "Nespravny nazov hashovacej funkcie.\n");
    exit(0);
  }
  return vysledok;
}

void ukoncit_spojenie(WOLFSSL *ssl, WOLFSSL_CTX *ctx)
{
  printf("Ukoncujem program.\n");
  wolfSSL_shutdown(ssl);
  wolfSSL_free(ssl);
  printf("Spojenie ukoncene.\n");
  wolfSSL_CTX_free(ctx);
  wolfSSL_Cleanup();
  printf("Program bol ukonceny.\n");
}