[RCOM] Projecto

#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <signal.h>
#include <termios.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/ioctl.h>

// Variáveis de configuração

#define BAUDRATE B38400
#define MODEMDEVICE "/dev/ttyS"
#define RECEIVER 0
#define TRANSMITTER 1
#define alarm_timeout 3
#define numTransmissions 2
#define filename "pinguim.gif"
#define packetsize 100

// Variáveis de mensagem

#define F 0x7e
#define A 0x03
#define S 0x03
#define U 0x07

#define I0 0x00
#define I1 0x40

#define RR0 0x05
#define RR1 0x85

#define REJ0 0x01
#define REJ1 0x81

#define DISC 0x0B


// Variáveis globais

int time_flag, counter = 0;
struct termios oldtio;

typedef struct
{
    long int filesize;
    char *filepath;
    char *filedata;
} imageFile;

imageFile ficheiro_de_imagem;

typedef struct
{
    int dataSeq, packetSize;
    char tramaSeq;
} dataConnection;

dataConnection dataHandler;

// Funções

void atende()
{
    time_flag = 1;
    counter++;
}

void printBar (char symbol)
{
    int i;
    struct winsize w;

    ioctl(STDOUT_FILENO, TIOCGWINSZ, &w);

    for (i = 0; i < w.ws_col; i++)
        printf ("%c", symbol);
}

void printm (char *string, int adjust)
{
    int i;
    struct winsize w;

    ioctl(STDOUT_FILENO, TIOCGWINSZ, &w);

    for (i = 0; i < (w.ws_col / 2) - (strlen(string) / 2) + adjust ; i++)
        printf (" ");

    printf ("%s", string);
}

void create_information (char *buffer, int *buffer_size, char c, char *data, int data_size, char BCC2)
{
    int i;

    *buffer_size = data_size + 6;

    printm ("Data:", 0);
    printf ("\n");
    for (i = 0; i < data_size; i++)
        printf ("%x |", data[i]);
    printf ("\n");

    buffer[0] = F;
    buffer[1] = A;
    buffer[2] = c;
    buffer[3] = buffer[1] ^ buffer[2];

    for (i = 0; i < data_size; i++)
        buffer[i + 4] = data[i];

    if (BCC2 != NULL)
    {
        buffer[data_size + 4] = BCC2;
        buffer[data_size + 5] = F;
    }
    else
    {
        buffer[data_size + 4] = F;
        *buffer_size--;
    }

    printm ("Information:", 0);
    printf ("\n");
    for (i = 0; i < *buffer_size; i++)
        printf ("%x |", buffer[i]);
    printf ("\n");
}

unsigned char create_control_packet (char *buffer, int *buffer_size , int c, char *file_name, int filesize)
{
    int i;
    unsigned char BCC2;

    *buffer_size = strlen(file_name) + 6;

    buffer[0] = (char) c;

    buffer[1] = (char) 0;
    buffer[2] = (char) sizeof ((char) filesize);
    buffer[3] = (char) filesize;

    buffer[4] = (char) 1;
    buffer[5] = (char) sizeof (file_name);
    for (i = 0; i < strlen(file_name); i++)
        buffer[i + 6] = (char) file_name[i];

    BCC2 = buffer[0];
    for (i = 1; i < strlen(file_name) + 6; i++)
        BCC2 ^= buffer[i];

    return BCC2;
}

unsigned char create_data_packet (char* buffer, int* buffer_size, int n, char *file_name, int filesize)
{
    char BCC2;
    int i, L2 = (int) (packetsize * 8 / 256), L1 = (int) (packetsize * 8) - L2 * 256;

    *buffer_size = strlen(file_name) + 4;

    buffer[0] = (char) 1;
    buffer[1] = (char) n;

    buffer[2] = (char) L2;
    buffer[3]= (char) L1;

    for(i = 0; i < strlen(file_name); i++)
        buffer[4+i]=file_name[i];

    BCC2 = buffer[0];
    for (i = 1; i < strlen(file_name) + 4; i++)
        BCC2 ^= buffer[i];

    return BCC2;
}

int menu ()
{
    int op = 0;

    system ("clear");
    printBar ('*');
    printm ("Programa iniciado!", 0);
    printf ("\n\n\n");
    printm ("Menu (escolher função do programa):", 0);
    printf ("\n");
    printm ("1. Emissor (llwrite).", 0);
    printf ("\n");
    printm ("2. Receptor (llread).", 0);
    printf ("\n");
    printm ("3. Sair.", 0);
    printf ("\n\n");
    printm ("Opcao: ", 0);
    scanf ("%d", &op);

    return op;
}

void sair ()
{
    system ("clear");

    printBar ('*');
    printm ("Programa a encerrar...", 0);
    printf ("\n");
    printBar ('*');
    sleep (1);
}

char read_control_packet (int fd, char BCC2)
{
    char bit_received, auxFile[30], received_message[255];
    int state=0, i, res, count = 0;

    while (state < 6)
    {
        res = read (fd, &bit_received, 1);
        if(res == -1)
            return NULL;

        switch (state)
        {
                case 0:
                    if (bit_received == 0)
                    {
                        received_message[count] = bit_received;
                        state++;
                        count++;
                    }
                    else
                        return NULL;
                    break;

                case 1:
                    received_message[count] = bit_received;
                    state++;
                    count++;
                    break;

                case 2:
                    auxFile[0] = bit_received;
                    received_message[count++] = bit_received;
                    for(i = 1; i < received_message[state--]; i++, count++)
                    {
                        res = read (fd, &bit_received, 1);
                        if(res == -1)
                            return NULL;

                        auxFile[i] = bit_received;
                        received_message[count] = bit_received;
                    }

                    ficheiro_de_imagem.filesize = (int) auxFile; // Isto tem de ser no llread() caso a mensagem esteja correcta.

                    state++;
                    count++;
                    break;

                case 3:
                    if (bit_received == 1)
                    {
                        received_message[count] = bit_received;
                        state++;
                        count++;
                    }
                    else
                        return NULL;
                    break;

                case 4:
                    received_message[count] = bit_received;
                    state++;
                    count++;
                    break;

                case 5:
                    auxFile[0] = bit_received;
                    received_message[count++] = bit_received;
                    for(i = 1; i < received_message[state--]; i++, count++)
                    {
                        res = read (fd, &bit_received, 1);
                        if(res == -1)
                            return NULL;

                        auxFile[i] = bit_received;
                        received_message[count] = bit_received;
                    }

                    // Isto tem de ser no llread() caso a mensagem esteja correcta.
                    ficheiro_de_imagem.filepath = (long int) malloc (received_message[state--]);
                    for (i = 0; i < received_message[state--]; i++)
                        ficheiro_de_imagem.filepath[i] = auxFile[i];

                    state++;
                    count++;
                    break;

                default:
                    break;
        }
    }

    for (i = 0; i < count; i++)
        BCC2 ^= received_message[i];

    return BCC2;
}

char read_data_packet (int fd, char BCC2, char *data)
{
    int i, count = 0, state = 0, res;
    char received_message[255], bit_received, auxFile[255];

    while (state < 4)
    {
        res = read (fd, &bit_received, 1);
        if(res == -1)
            return NULL;

        switch (state)
        {
            case 0:
                received_message[count] = bit_received;
                dataHandler.dataSeq = (int) bit_received;
                state++;
                count++;
                break;

            case 1:
                received_message[count] = bit_received;
                state++;
                count++;
                break;

            case 2:
                received_message[count] = bit_received;
                dataHandler.packetSize = (int) (256 * (int) received_message[count - 1] + (int) received_message[count]);
                state++;
                count++;
                break;

            case 3:
                auxFile[0] = bit_received;
                received_message[count++] = bit_received;
                for(i = 1; i < dataHandler.packetSize; i++, count++)
                {
                    res = read (fd, &bit_received, 1);
                    if(res == -1)
                        return NULL;

                    auxFile[i] = bit_received;
                    received_message[count] = bit_received;
                }

                /*ficheiro_de_imagem->filedata = (char *) malloc (sizeof(char) * dataHandler->packetSize);
                for (i = 0; i < received_message[state--]; i++)
                    ficheiro_de_imagem->filedata[i] = auxFile[i];*/

                state++;
                count++;
                break;
        }
    }
    for (i = 0; i < count; i++)
        BCC2 ^= received_message[i];

    data = (char *) malloc (sizeof(char) * dataHandler.packetSize);
    for(i = 0; i < dataHandler.packetSize; i++)
        data[i] = auxFile[i];

    return BCC2;
}

void main (int argc, char** argv)
{
    int op = 0, fd, filesize;
    char *buffer = NULL;

    while (op != 3)
    {
        op = menu();
        switch (op)
        {
            case 1:
                fd = llopen (0, TRANSMITTER);
                if (fd != -1)
                {
                    //filesize = myWrite (buffer);
                    llwrite(fd, buffer, filesize);
                }
                break;
            case 2:
                fd = llopen (0, RECEIVER);
                llread(fd, buffer);
                //saveToFile(buffer);
                break;
            case 3:
                llclose (fd);
                sair();
                break;
            default:
                break;
        }
    }
}

int llopen (int porta, int flag)
{
    /* -----    Variáveis  ----- */

    unsigned char send_message[5], receive_message[5], bit_received;
    char buffer[1], gate[10] = MODEMDEVICE;
    int vtime, fd, res[2], return_value, i;
    unsigned int msg_sent = 0, state;
    struct termios newtio;

    /* -----    Código     ----- */

    sprintf (buffer, "%d", porta);
    strcat (gate, buffer);

    system ("clear");

    printBar ('*');
    printm ("Função \"llopen(", -7);
    printf ("%d, %d)\" iniciada!\n", porta, flag);

    if (flag == TRANSMITTER)
    {
        send_message[0] = F;
        send_message[1] = A;
        send_message[2] = S;
        send_message[3] = send_message[1] ^ send_message[2];
        send_message[4] = F;

        vtime = 3;

        fd = open(gate, O_RDWR | O_NOCTTY | O_NONBLOCK);
    }
    else if (flag == RECEIVER)
    {
        send_message[0] = F;
        send_message[1] = A;
        send_message[2] = U;
        send_message[3] = send_message[1] ^ send_message[2];
        send_message[4] = F;

        vtime = 0;

        fd = open(gate, O_RDWR | O_NOCTTY);
    }
    else
        return -1;

    if (fd < 0)
            return -1;

    // Teste ao apontador para o Pipe.

    if (tcgetattr (fd, &oldtio) == -1)
    {
        perror ("tcgetattr");
        return -1;
    }

    // Configuração da ligação.

    bzero(&newtio, sizeof(newtio));
    newtio.c_cflag = BAUDRATE | CS8 | CLOCAL | CREAD;
    newtio.c_iflag = IGNPAR;
    newtio.c_oflag = 0;

    /* set input mode (non-canonical, no echo,...) */
    newtio.c_lflag = 0;

    newtio.c_cc[VTIME]    = vtime;   /* inter-character timer unused */
    newtio.c_cc[VMIN]     = 1;   /* blocking read until 1 chars received */

    /* VTIME e VMIN devem ser alterados de forma a proteger com um temporizador a leitura do(s) próximo(s) caracter(es) */

    tcflush(fd, TCIOFLUSH);
    if (tcsetattr (fd, TCSANOW, &newtio) == -1)
    {
        perror ("tcsetattr");
        return -1;
    }

    (void) signal(SIGALRM, atende);

    if (flag == TRANSMITTER)
    {
        for (time_flag = 1, state = 0, counter = 0; counter < numTransmissions && msg_sent != 1;)
        {
            if (time_flag)
            {
                alarm (alarm_timeout);
                time_flag = 0;

                printm ("A enviar SETUP: tentativa #", -1);
                printf ("%d\n", counter + 1);

                res[0] = write (fd, send_message, 5);

                while (state < 5)
                {
                    res[1] = read (fd, &bit_received, 1);
                    if(res[1] == -1)
                        break;

                    switch (state)
                    {
                        case 0:
                          if (bit_received == F)
                          {
                            receive_message[state]=bit_received;
                            state++;
                          }
                          break;

                        case 1:
                          if (bit_received == A)
                          {
                            receive_message[state]=bit_received;
                            state++;
                          }
                          else if (bit_received != A && bit_received !=F)
                            state--;
                          break;

                        case 2:
                          if (bit_received == U)
                          {
                            receive_message[state]=bit_received;
                            state++;
                          }
                          else if (bit_received == F)
                            state--;
                          else
                            state = 0;
                          break;

                        case 3:
                          if (bit_received == (A^U))
                          {
                            receive_message[state]=bit_received;
                            state++;
                          }
                          else
                            state = 0;
                          break;

                        case 4:
                          if (bit_received == F)
                          {
                            receive_message[state]=bit_received;
                            state++;
                            msg_sent = 1;
                          }
                          else
                            state = 0;
                          break;

                        default:
                            break;
                    }
                }
            }
        }

        if (msg_sent == 0)
        {
            printm ("Mensagem não enviada (não recebeu UA).", +3);
            printf ("\n");
            return_value = -1;
        }
        else
        {
            printm ("Mensagem enviada (recebeu UA).", 0);
            printf ("\n");
            printm ("Mensagem UA:", 0);
            printf ("\n");
            printm (" ", -5);
            for(i = 0; i < state; i++)
                printf("%x", receive_message[i]);
            printf ("\n");
            return_value = fd;
        }
    }
    else if (flag == RECEIVER)
    {
        state = 0;
        while (state < 5)
        {
            res[1] = read (fd, &bit_received, 1);

            switch (state)
            {
                    case 0:
                            if (bit_received == F)
                            {
                                    receive_message[state] = bit_received;
                                    state++;
                            }
                            break;
                    case 1:
                            if (bit_received == A)
                            {
                                    receive_message[state] = bit_received;
                                    state++;
                            }
                            else if (bit_received != F && bit_received != A)
                                    state--;
                            break;
                    case 2:
                            if (bit_received == S)
                            {
                                    receive_message[state] = bit_received;
                                    state++;
                            }
                            else if (bit_received == F)
                                    state--;
                            else
                                    state = 0;
                            break;
                    case 3:
                            if (bit_received == (A ^ S))
                            {
                                    receive_message[state] = bit_received;
                                    state++;
                            }
                            else if (bit_received == F)
                                    state = 1;
                            else
                                    state = 0;
                            break;
                    case 4:
                            if (bit_received == F)
                            {
                                    receive_message[state] = bit_received;
                                    state++;
                                    msg_sent = 1;
                            }
                            else
                                    state = 0;
                            break;
                    default:
                        break;
            }
        }

        if (msg_sent == 0)
        {
            printm ("Mensagem não enviada (não recebeu SETUP).", +3);
            printf ("\n");
            return_value = -1;
        }
        else
        {
            printm ("Mensagem enviada (recebeu SETUP).", 0);
            printf ("\n");
            printm ("Mensagem SETUP:", 0);
            printf ("\n");
            printm (" ", -5);
            for(i = 0; i < state; i++)
                printf("%x", receive_message[i]);
            printf ("\n");

            res[0] = write(fd, send_message, 5);
            /*if (!res[0])
                printf("Mensagem (UA) não enviada.\n");
            else
                printf("%d bytes (UA) enviados.\n", res[0]);*/

            return_value = fd;
        }
    }

    printf ("\n\n");
    printm ("Função \"llopen(", -7);
    printf ("%d, %d)\" a fechar!\n", porta, flag);
    printm ("Clique numa tecla para voltar ao menu.", 0);
    printf ("\n");
    printBar ('*');
    getchar();
    getchar();

    return return_value;
}

int llwrite (int fd, char * buffer, int length)
{
    char *cntrl_message, *data_message, *message, *message_received, bit_received;
    unsigned char BCC2;
    int file_found = 1, i, cntrl_size, message_size, msg_error = 0, state, res, buffer_size;
    FILE *pImg;

    while (file_found)
    {
        system ("clear");
        printBar ('*');
        printm ("Nome do ficheiro a usar: pinguim.gif", 0);
        printf ("\n");
        printm ("(certifique-se que o ficheiro está na pasta do programa)", 0);
        printf ("\n\n\n");

        pImg = fopen (filename, "r");
        if (pImg == NULL)
        {
            printm ("Ficheiro não encontrado.", 0);
            printf ("\n");
            printm ("(certifique-se que o ficheiro está na pasta do programa)", 0);
            printf ("\n");
            printm ("Clique <Enter> para voltar a tentar.", 0);
            getchar ();
        }
        else
        {
            file_found = 0;
            printm ("Ficheiro encontrado.", 0);
            printf ("\n");
            printm ("Clique <Enter> para continuar.", 0);
            getchar ();
        }
    }

    // Leitura do conteúdo da imagem para variável "buffer"
    fseek (pImg, 0, SEEK_END);
    length = ftell (pImg);
    fseek (pImg, 0, SEEK_SET);

    buffer = (char *) malloc (sizeof(char) * length);

    for (i = 0; i < length; i++)
        buffer[i] = getc(pImg);

    fclose (pImg);
    // Leitura do conteúdo da imagem para variável "buffer"

    // Primeira mensagem a enviar: Pacote de controlo (start)
    cntrl_message = (char *) malloc (sizeof(char) * (strlen(filename) + 6));
    BCC2 = create_control_packet (cntrl_message, &cntrl_size, 2, filename, length);

    message = (char *) malloc (sizeof(char) * (cntrl_size + 6));
    create_information (message, &message_size, I0, cntrl_message, cntrl_size, BCC2);

    // Primeira mensagem a enviar: Pacote de controlo (start)

    // Leitura de verificação
    for (time_flag = 1, state = 0, counter = 0; counter < numTransmissions;)
    {
        if (time_flag)
        {
            alarm (alarm_timeout);
            time_flag = 0;

            write (fd, message, message_size); // envio do pacote de controlo

            while (state < 6 || denied != 1)
            {
                res = read (fd, &bit_received, 1);
                if(res == -1)
                    break;

                switch (state)
                {
                    case 0:
                        if (bit_received == F)
                            state++;
                        else
                            denied = 1;

                        break;

                    case 1:
                        if (bit_received == A)
                            state++;
                        else
                            denied = 1;

                        break;

                    case 2:
                        auxCntrl = bit_received;
                        state++;
                        break;

                    case 3:
                        if (bit_received == auxCntrl ^ A)
                            state++;
                        else
                            denied = 1;

                        break;

                    case 4:
                        if (bit_received == F)
                            state++;
                        else
                            denied = 1;

                        break;

                    case 5: // verifcar RRs, REJs e DISC
                        if (auxCntrl == RR1 && dataHandler.tramaSeq == I0)
                        {
                            dataHandler.tramaSeq = I1;
                            // create_new (message);
                        }

                }
            }
        }
    }
    // Leitura de verificação

    return length;
}

/* ----- llread -----
argumentos
    fd: identificador da ligação de dados
    buffer: array de caracteres recebidos
retorno
    comprimento do array (número de caracteres lidos)
    valor negativo em caso de erro
*/
int llread (int fd, char *buffer)
{
    char bit_received, *data, BCC2, auxCntrl;
    int msg_wanted = 0, state = 0, res, final = 0;

    dataHandler.tramaSeq = I0;

    /* while (auxCntrl != DISC) */
    msg[0] = F;
    while (bit_received != F)
    {
        read (fd, &bit_received, 1);
        msg[i++] = bit_received;
    }

    while (state < 8)
    {
        /*res = read (fd, &bit_received, 1);
        if(res == -1)
            return NULL;*/

        switch (state)
        {
                case 0:
                    if (bit_received == F)
                        state++;

                    break;

                case 1:
                    if (bit_received == A)
                        state++;
                    else
                        state = 0;

                    break;

                case 2:
                    if (bit_received == dataHandler.tramaSeq)
                    {
                        msg_wanted = 1;
                        state++;
                    }

                    break;

                case 3:
                    if (bit_received == A ^ dataHandler.tramaSeq)
                        state++;
                    else
                        state = 0;

                    break;

                case 4:
                    if (bit_received == 1)
                    {
                        BCC2 = read_data_packet (fd, bit_received, data);
                        state++;
                    }
                    else if (bit_received == 2)
                    {
                        BCC2 = read_control_packet (fd, bit_received);
                        state++;
                    }
                    else if (bit_received == 3)
                    {
                        BCC2 = read_control_packet (fd, bit_received);
                        state++;
                        final = 1;
                    }
                    else
                        state = 0;

                    break;

                case 5:
                    if (bit_received == BCC2)
                        state++;
                    else
                        state = 0;

                    break;

                case 6:
                    if (bit_received == F)
                        state++;
                    else
                        state = 0;

                    break;

                case 7:
                    if (msg_wanted && final == 0)
                    {
                        if (dataHandler.tramaSeq == I0)
                        {
                            dataHandler.tramaSeq = I1;
                            auxCntrl = RR1;
                        }
                        else if (dataHandler.tramaSeq == I1)
                        {
                            dataHandler.tramaSeq = I0;
                            auxCntrl = RR0;
                        }
                        // concatenar data ou guardar filepath e filesize
                    }
                    else if (msg_wanted && final == 1)
                    {
                        auxCntrl = DISC;
                    }
                    else if (!msg_wanted)
                    {
                        if (dataHandler.tramaSeq == I0)
                            auxCntrl = REJ0;
                        else if (dataHandler.tramaSeq == I1)
                            auxCntrl = REJ1;
                    }

                    buffer = (char *) malloc (sizeof(char) * 5);
                    create_information (buffer, 5, auxCntrl, NULL, 0, NULL);
                    write (fd, buffer, 5);

                default:
                    break;

        }
    }
    /* --- */

    // Criação do ficheiro com data
}


int llclose (int fd)
{
    if (tcgetattr (fd, &oldtio) == -1)
    {
        perror ("tcgetattr");
        return -1;
    }

    close (fd);

    return 0;
}