275 lines
8.2 KiB
C
275 lines
8.2 KiB
C
//////////////////////////////////////////////////
|
|
// Bakalarska praca //
|
|
// Meno studenta: Tomas Lukac //
|
|
// Veduci BP: prof. Ing. Milos Drutarovsky CSc. //
|
|
// Skola: KEMT FEI TUKE //
|
|
// Datum poslednej upravy: 5.6.2020 //
|
|
//////////////////////////////////////////////////
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <wolfssl/ssl.h>
|
|
#include <wolfssl/certs_test.h>
|
|
#include <wolfssl/wolfcrypt/types.h>
|
|
|
|
#ifdef _WIN32
|
|
#include <Windows.h>
|
|
#include <io.h>
|
|
#include <fcntl.h>
|
|
#define O_NOCTTY 0x8000
|
|
#define O_NDELAY 0x4000
|
|
#else
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
#include "../../kniznica/kryptografia.h"
|
|
#include "../../kniznica/komunikacia.h"
|
|
|
|
#define RSA_VELKOST 2048
|
|
#define ECC_VELKOST 32
|
|
#define RSA_EXPONENT 65537
|
|
|
|
//cesty ku suborom
|
|
#define RSA_KLUC "../../certifikaty/server/server_rsa.key"
|
|
#define RSA_CERTIFIKAT "../../certifikaty/server/server_rsa.pem"
|
|
#define ECC_KLUC "../../certifikaty/server/server_ecc.key"
|
|
#define ECC_CERTIFIKAT "../../certifikaty/server/server_ecc.pem"
|
|
#define VYGENEROVANY_KLUC "../../certifikaty/server/vygenerovany_kluc.key"
|
|
#define VYGENEROVANY_CERTIFIKAT "../../certifikaty/server/vygenerovany_certifikat.pem"
|
|
#define KOMUNIKACNY_SUBOR "server.txt"
|
|
|
|
//subor do ktoreho sa zapisuje komunikacia
|
|
int k_subor;
|
|
|
|
int cislo_rozhrania = 0;
|
|
|
|
//velkost dat ktore sa prave nachadzaju v seriovom bufferi
|
|
//a cakaju na nacitanie do buffera pre desifrovanie
|
|
int aktualne_data = 0;
|
|
|
|
//mnozstvo dat ktore boli nacitane do buffera pre desifrovanie
|
|
//pri zavolani funkcie rs232_citanie()
|
|
int nacitane_data = 0;
|
|
|
|
int rs232_citanie(WOLFSSL *ssl, char *buf, int sz, void *ctx)
|
|
{
|
|
//ziskanie dat zo serioveho rozhrania
|
|
if(aktualne_data == 0)
|
|
{
|
|
aktualne_data = rs232_prijat_spravu(cislo_rozhrania, KOMUNIKACNY_SUBOR);
|
|
if(aktualne_data <= 0)
|
|
{
|
|
fprintf(stderr, "Nastala chyba pri prijati spravy\n");
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
//nacitanie dat zo suboru do buffera pre desifrovanie
|
|
nacitane_data = 0;
|
|
while(nacitane_data <= 0)
|
|
nacitane_data = read(k_subor, buf, sz);
|
|
aktualne_data -= nacitane_data;
|
|
return nacitane_data;
|
|
}
|
|
|
|
int rs232_zapis(WOLFSSL *ssl, char *buf, int sz, void *ctx)
|
|
{
|
|
int uspech = 0;
|
|
int odoslane_data = 0;
|
|
|
|
//poslanie dat na seriove rozhranie
|
|
odoslane_data = rs232_odoslat_spravu(cislo_rozhrania, buf, sz);
|
|
if(odoslane_data <= 0)
|
|
{
|
|
fprintf(stderr, "Nastala chyba pri posielani dat\n");
|
|
return -1;
|
|
}
|
|
|
|
return odoslane_data;
|
|
}
|
|
|
|
int main(int argc, char const *argv[])
|
|
{
|
|
//nastavenie rezimu rezhrania
|
|
//8 -> pocet bitov, N -> no parity, 1 -> jeden stopbit
|
|
char rezim[]={'8','N','1', 0};
|
|
|
|
//nastavenie rychlosti rozhrania v baudoch
|
|
int rychlost = 128000;
|
|
|
|
WOLFSSL *ssl;
|
|
WOLFSSL_CTX *ctx = NULL;
|
|
int zadane_rozhranie = 0;
|
|
int generovanie_certifikatu = 0;
|
|
int nacitanie_zo_suboru = 0;
|
|
nastavenia_aplikacie nastavenia;
|
|
|
|
k_subor = open(KOMUNIKACNY_SUBOR, O_RDWR | O_NOCTTY | O_NDELAY);
|
|
|
|
|
|
int uspech;
|
|
if((ctx = nastavit_ctx_server()) == NULL)
|
|
{
|
|
return -1;
|
|
}
|
|
else
|
|
{
|
|
for(int i = 0; i < argc; i++)
|
|
{
|
|
if( (!strcmp(argv[i], "-port")) )
|
|
{
|
|
zadane_rozhranie = 1;
|
|
if((argv[i+1] == NULL))
|
|
{
|
|
printf("Nezadali ste cislo serioveho rozhrania\n");
|
|
return -1;
|
|
}
|
|
else
|
|
{
|
|
cislo_rozhrania = atoi(argv[i+1]);
|
|
if(rs232_otvorit_rozhranie(cislo_rozhrania, rychlost, rezim, 1))
|
|
{
|
|
return -1;
|
|
}
|
|
}
|
|
}
|
|
else if( (!strcmp(argv[i], "-n")) )
|
|
{
|
|
nacitanie_zo_suboru = 1;
|
|
if(generovanie_certifikatu)
|
|
{
|
|
fprintf(stderr, "Nie je mozne zvolit obidve metody nacitania certifikatov naraz\n");
|
|
return -1;
|
|
}
|
|
if((argv[i+1] == NULL) || (i == argc-1))
|
|
{
|
|
printf("Nezadali ste typ certifikatu ktory chcete nacitat zo suboru\n");
|
|
return -1;
|
|
}
|
|
else if(!strcmp(argv[i+1], "rsa"))
|
|
{
|
|
wolfSSL_CTX_load_verify_locations(ctx, "../../certifikaty/autorita/autorita_rsa.pem", NULL);
|
|
if(nacitat_certifikaty(ctx, RSA_CERTIFIKAT, RSA_KLUC) == -1) return -1;
|
|
wolfSSL_CTX_set_verify(ctx, SSL_VERIFY_PEER | SSL_VERIFY_FAIL_IF_NO_PEER_CERT, 0);
|
|
|
|
}
|
|
else if(!strcmp(argv[i+1], "ecc"))
|
|
{
|
|
wolfSSL_CTX_load_verify_locations(ctx, "../../certifikaty/autorita/autorita_ecc.pem", NULL);
|
|
if(nacitat_certifikaty(ctx, ECC_CERTIFIKAT, ECC_KLUC) == -1) return -1;
|
|
wolfSSL_CTX_set_verify(ctx, SSL_VERIFY_PEER | SSL_VERIFY_FAIL_IF_NO_PEER_CERT, 0);
|
|
|
|
}
|
|
else
|
|
{
|
|
printf("Zadali ste nespravny typ certifikatu\n");
|
|
return -1;
|
|
}
|
|
}
|
|
else if( (!strcmp(argv[i], "-g")) )
|
|
{
|
|
generovanie_certifikatu = 1;
|
|
if(nacitanie_zo_suboru)
|
|
{
|
|
fprintf(stderr, "Nie je mozne zvolit obidve metody nacitania certifikatov naraz\n");
|
|
return -1;
|
|
}
|
|
if((argv[i+1] == NULL) || (i == argc-1))
|
|
{
|
|
printf("Nezadali ste typ certifikatu ktory chcete vygenerovat\n");
|
|
return -1;
|
|
}
|
|
else if(!strcmp(argv[i+1], "rsa"))
|
|
{
|
|
wolfSSL_CTX_load_verify_locations(ctx, "../../certifikaty/autorita/autorita_rsa.pem", NULL);
|
|
if(generovat_rsa_certifikat(RSA_VELKOST, RSA_EXPONENT, CTC_SHA256wRSA, "SR", "Kosice", "local.dev", "server@server.sk") == -1) return -1;
|
|
if(nacitat_certifikaty(ctx, VYGENEROVANY_CERTIFIKAT, VYGENEROVANY_KLUC) == -1) return -1;
|
|
|
|
}
|
|
else if(!strcmp(argv[i+1], "ecc"))
|
|
{
|
|
wolfSSL_CTX_load_verify_locations(ctx, "../../certifikaty/autorita/autorita_ecc.pem", NULL);
|
|
if(generovat_ecc_certifikat(ECC_VELKOST, ECC_SECP256R1, CTC_SHAwECDSA, "SR", "Kosice", "local.dev", "server@server.sk") == -1) return -1;
|
|
if(nacitat_certifikaty(ctx, VYGENEROVANY_CERTIFIKAT, VYGENEROVANY_KLUC) == -1) return -1;
|
|
}
|
|
else
|
|
{
|
|
printf("Zadali ste nespravny typ certifikatu\n");
|
|
return -1;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if(!zadane_rozhranie)
|
|
{
|
|
fprintf(stderr, "Nebol urcene seriove rozhranie\n");
|
|
return -1;
|
|
}
|
|
else if(!generovanie_certifikatu && !nacitanie_zo_suboru)
|
|
{
|
|
fprintf(stderr, "Nebola zvolena metoda nacitania certifikatov\n");
|
|
printf("Zadajde prepinac -g (generovanie), alebo -n (nacitanie_zo_suboru) s parametrom rsa alebo ecc\n");
|
|
return -1;
|
|
}
|
|
|
|
//nastav_sifry(ctx, "TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256");
|
|
|
|
//nastavenie vlastnych I/O funkcii
|
|
wolfSSL_SetIOSend(ctx, rs232_zapis);
|
|
wolfSSL_SetIORecv(ctx, rs232_citanie);
|
|
|
|
//pokus o vytvorenie novej wolfSSL relacie
|
|
if ((ssl = wolfSSL_new(ctx)) == NULL)
|
|
{
|
|
printf("Nepodarilo sa vytvorit ssl relaciu\n");
|
|
wolfSSL_CTX_free(ctx);
|
|
return -1;
|
|
}
|
|
|
|
//priradenie file descriptora suboru ako I/O pre TLS spojenie
|
|
wolfSSL_set_fd(ssl, k_subor);
|
|
wolfSSL_set_using_nonblock(ssl, k_subor);
|
|
|
|
//cakanie na inicializaciu TLS handshaku klientom
|
|
uspech = wolfSSL_accept(ssl);
|
|
if(uspech != SSL_SUCCESS)
|
|
{
|
|
char* popis_chyby = calloc(100, sizeof(char));
|
|
int chyba = wolfSSL_get_error(ssl, 0);
|
|
wolfSSL_ERR_error_string(chyba, popis_chyby);
|
|
fprintf(stderr, "Nastala chyba v spojeni.\nCislo chyby: %d\nDovod chyby: %s\n", chyba, popis_chyby);
|
|
printf("Skontrolujte certifikaty.\n");
|
|
fclose(fopen(KOMUNIKACNY_SUBOR, "wb"));
|
|
free(popis_chyby);
|
|
return -1;
|
|
}
|
|
zobraz_sifru(ssl);
|
|
zobraz_certifikat(ssl);
|
|
|
|
//nastavenie funkcie pre vypocet kontrolneho suctu
|
|
//nastavime typ "crc" alebo "hash" a nazov funkcie, nazov druhej funkcie nastavime na NULL
|
|
nastav_funkciu(&nastavenia, "crc", (hashovacia_funkcia)NULL, funkcia_CRC32);
|
|
|
|
//vypocet casu prenosu suboru
|
|
clock_t cas;
|
|
cas = clock();
|
|
if(prijat_subor(ssl, ctx, &nastavenia) == -1)
|
|
{
|
|
fclose(fopen(KOMUNIKACNY_SUBOR, "wb"));
|
|
return -1;
|
|
}
|
|
|
|
cas = clock() - cas;
|
|
printf("Cas prenosu suboru: %f sekund\n", (double)cas/CLOCKS_PER_SEC);
|
|
|
|
//ukoncenie spojenia, vymazanie komunikacneho suboru
|
|
//a vycistenie serioveho buffera
|
|
ukoncit_spojenie(ssl, ctx);
|
|
rs232_zatvorit_rozhranie(cislo_rozhrania);
|
|
close(k_subor);
|
|
fclose(fopen(KOMUNIKACNY_SUBOR, "wb"));
|
|
return 0;
|
|
}
|