MastersThesis/PQ_TIIGER_TLS/lib/tls_sockets.cpp

/*
* JS update 29.02.2024
* Uprava vstupnych parametrov pre funkciu rozdielne definovanu pre Linux a Windows
* Odstranene funkcie pre ARDUINO
*/

// Set up and read/write sockets

#include "tls_logger.h"

//int BYTES_READ=0;
//int BYTES_WRITTEN=0;

// open socket
int setclientsock(int port,char *ip,int toms)
{
    int sock = 0; 
    struct sockaddr_in serv_addr; 
    if ((sock = socket(AF_INET, SOCK_STREAM, 0)) < 0) 
        return -1; 
   
    serv_addr.sin_family = AF_INET; 
    serv_addr.sin_port = htons(port); 
       
// Convert IPv4 and IPv6 addresses from text to binary form 
    if(inet_pton(AF_INET, ip, &serv_addr.sin_addr)<=0)  
        return -2; 
   
// Set time-out period    
    struct timeval timeout;
    timeout.tv_sec  = toms/1000;  // after some seconds read() will timeout
    timeout.tv_usec = (toms%1000)*1000;

    // JS uprava casovacu pre Windows funkciu s odlisnym argumentom
    #ifdef __WIN32
        //int nTimeout = 30000;
        setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, (const char*)&timeout.tv_usec, sizeof(timeout.tv_usec));
    #else
        setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &timeout, sizeof(timeout));
    #endif
    
    if (connect(sock, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0) 
        return -3; 
    
    return sock;
}

// get IP address from Hostname
int getIPaddress(char *ip,char *hostname)
{
    hostent * record = gethostbyname(hostname);
    if(record == NULL) return 0;
    in_addr * address = (in_addr * )record->h_addr;
    strcpy(ip,inet_ntoa(* address));
    return 1;
}

// Send Octet
void sendOctad(Socket *client,octad *B)
{
    client->write(B->val,B->len);
//BYTES_WRITTEN+=B->len;

#if VERBOSITY >= IO_WIRE

    char w[4];
    myprintf((char *)"-> ");
    for (int j = 0; j < B->len; j++)
    {
        sprintf(w,"%02x", (unsigned char)B->val[j]);
        myprintf(w);
    }
    myprintf((char *)"\n");

#endif    
}

// Send Octet length
void sendLen(Socket *client,int len)
{
    char buff[2];
    octad B={0, sizeof(buff), buff};
    B.len=2;
    B.val[0]=len&0xff;
    B.val[1]=len/256;
    sendOctad(client,&B);
}

// get expected bytes
int getBytes(Socket *client,char *b,int expected)
{
    int n,more,i=0,len=expected;

//BYTES_READ+=expected;

    unsigned long start=millis();
    while(len>0)
    {
        if (millis()>start+5000)
            return -1;
        more=client->read(&b[i],len);
        if (more==0) continue;
        if (more<0) return -1;
        i+=more;
        len-=more;
    }

#if VERBOSITY >= IO_WIRE

    char w[4];
    myprintf((char *)"<- ");
    for (int j = 0; j < expected; j++)
    {
        sprintf(w,"%02x", (unsigned char)b[j]);
        myprintf(w);
    }
    myprintf((char *)"\n");

#endif
    return 0;
}

// Get 16-bit Integer from stream
int getInt16(Socket *client)
{
    char b[2];
    getBytes(client,b,2);
    return 256*(int)(unsigned char)b[0]+(int)(unsigned char)b[1];
}

// Get expected number of bytes into an octet
int getOctad(Socket *client,octad *B,int expected)
{
    B->len=expected;
    return getBytes(client,B->val,expected);
}