Live-Modifiable-Server/get_client.cpp at master · shmundhra/Live-Modifiable-Server · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
#include "livemodifiable.h"

static volatile sig_atomic_t terminate_(0);
static void sig_handler(int signo)
{
    if(signo == SIGINT)
    {
        terminate_ = 1;
    }
}

signed main(int argc, char* argv[])
{
    signal(SIGINT, sig_handler);

    if (argc < 2) {
        RED << "Insufficient CLA"; RESET2;
        exit(EXIT_FAILURE);
    }
    string FileName(argv[1]);

    string IP_ADDRESS(IP_ADDR);
    if (argc >= 3) {
        IP_ADDRESS = argv[2];
    }

    int socket_fd = socket(AF_INET, SOCK_STREAM, 0);
    if (socket_fd < 0) {
        RED << LOG; perror("Error in Creating Connection Socket"); RESET1
        exit(EXIT_FAILURE);
    }

    sockaddr_in serv_addr = {AF_INET, htons(PORT), inet_addr(IP_ADDRESS.c_str()), sizeof(sockaddr_in)};
    if (connect(socket_fd, reinterpret_cast<sockaddr*>(&serv_addr), sizeof(serv_addr)) < 0) {
        RED << LOG; perror("Error in Connecting to TCP Server"); RESET1
        exit(EXIT_FAILURE);
    }
    if (EMULATING) sleep(5);
    GREEN << LOG << "CONNECTED to SERVER..."; RESET2;

    string Command("GET " + FileName);
    if (sendInfo(socket_fd, (char*)Command.c_str()) < 0) {
        RED << LOG; perror("Error in Sending Command to Server"); RESET1
        exit(EXIT_FAILURE);
    }

    int recv_ = 0;
    while(!terminate_)
    {
        WHITE << "WAITING for Packet..."; RESET2;
        PacketType packet_type;
        if ((recv_ = recvType(socket_fd, packet_type)) < 0) {
            RED << LOG; perror("Error in Receiving Packet Type"); RESET1
            exit(EXIT_FAILURE);
        }
        if (recv_ == 0) {
            GREEN << LOG << "SERVER TERMINATED CONNECTION"; RESET2;
            break;
        }

        switch(packet_type)
        {
            case PacketType::ERROR : {
                                        char* error_msg = (char*)calloc(ERRSIZE+1, sizeof(char));
                                        if (recvError(socket_fd, error_msg) < 0)
                                        {
                                            RED << LOG; perror("Error in Receiving Error Packet"); RESET1
                                            exit(EXIT_FAILURE);
                                        }
                                        free(error_msg);
                                        exit(EXIT_FAILURE);
                                     }

            case PacketType::INFO : {
                                        char* info_msg = (char*)calloc(INFOSIZE+1, sizeof(char));
                                        if (recvInfo(socket_fd, info_msg) < 0)
                                        {
                                            RED << LOG; perror("Error in Receiving Info Packet"); RESET1
                                            exit(EXIT_FAILURE);
                                        }
                                        free(info_msg);
                                        break;
                                    }

            case PacketType::DATA : {
                                        /* Receive Produce from Server */
                                        int offset;
                                        char* data = (char*)calloc(DATASIZE+1, sizeof(char));
                                        if ((recv_ = recvData(socket_fd, &offset, data)) < 0)
                                        {
                                            RED << LOG; perror("Error in Receiving Data Packet"); RESET1
                                            exit(EXIT_FAILURE);
                                        }
                                        if (recv_ == 0) {
                                            GREEN << "FILE TRANSFER is COMPLETE"; RESET2;
                                            break;
                                        }
                                        free(data);

                                        /* Consume the Produce Received */
                                        int next_offset = offset + recv_;
                                        // if (next_offset < 1500) next_offset -= next_offset % 33;

                                        /* Send ACK of Consumption */
                                        if (sendAck(socket_fd, next_offset) < 0){
                                            RED << LOG; perror("Error in Sending Ack Packet");
                                            exit(EXIT_FAILURE);
                                        }
                                        break;
                                    }

            default: RED << "Unexpected PacketType"; RESET2;
        }
        if(!recv_) break;
    }
    exit(EXIT_SUCCESS);
}