web_server/WebServer.cpp at main · NotYaiba/web_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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
#include "WebServer.hpp"

Webserver::Webserver(Connection const &connection)
{
    InitData(connection);

}
Webserver::~Webserver()
{

}

void  Webserver::InitData(Connection const &connection)
{
    servers = connection.getMapServersF();

    _cnx = connection;
    fd_map = _cnx.getFd_map();
    FD_ZERO(&readset);
    FD_ZERO(&writeset);
    FD_ZERO(&readcopy);
    FD_ZERO(&writecopy);
    for (std::map < int, int >::iterator it = fd_map.begin(); it != fd_map.end(); it++)
    {
            FD_SET(it->first, &readset);
            Request req;
            req_map.insert(std::make_pair(it->first,req));
    }
    maxfd = (--fd_map.end())->first ;
    readcopy = readset;
}

void Webserver::RunWebServer()
{
    std::cout << red << "MY SERVER IS LISTING!" << reset << std::endl;
    signal(SIGPIPE, SIG_IGN);
    for (;;)
    {
        readset = readcopy;
        writeset = writecopy;
        int rs = select(maxfd + 1 , &readset, &writeset, NULL, NULL);
        if (rs == -1)
            throw "errooor";
        for (int fd = 0; fd <= maxfd; fd++)
        {
            if (FD_ISSET(fd, &readset) || FD_ISSET(fd, &writeset))
            {
                if (FD_ISSET(fd, &readset))
                {
                    if (fd_map[fd] == 0)
                        NewConnectionRead(fd);
                    else
                        HandleRequest(fd);
                }
                else if (FD_ISSET(fd, &writeset))
                {
                    HandleResponse(fd);
                }
            }
        }
    }
}

void Webserver::NewConnectionRead(int fd)
{
    int new_fd = Socket::AccectSock(fd);
    fcntl(new_fd, F_SETFL,  O_NONBLOCK);
    FD_SET(new_fd, &readcopy);
    fd_map.insert(std::make_pair(new_fd, 1));
    servers[new_fd] = servers[fd];
    Request req;
    req_map.insert(std::make_pair(new_fd ,req));
    fd_map[new_fd] = 1;
    if (new_fd > maxfd)
        maxfd = new_fd;
}

void Webserver::HandleRequest(int fd)
{
    char buf[BUFFER_SIZE];
    int rb = read(fd,buf,BUFFER_SIZE );
    if (rb == -1)
    {
        close(fd);
        FD_CLR(fd ,&readcopy);
    }
    else if (rb > 0)
    {
        Request req;
        std::map<int , Request>::iterator it = req_map.find(fd);
        it->second.fillRequest(buf, rb);

        if ( it->second.isslastRead == true)
        {

            FD_SET(fd, &writecopy);
            FD_CLR(fd, &readcopy);
        }
    }
    else if (rb == 0)
    {
        close(fd);
        FD_CLR(fd ,&readcopy);
    }
}


void Webserver::HandleResponse(int fd)
{
    std::pair <char * , size_t>buffer_new;
    int returnWrite;
    std::map<int, Response >::iterator it =  res_map.find(fd);
    std::map<int, Response >::iterator it2;
    if (it  == res_map.end())
    {
        Response res(servers[fd], req_map[fd]);
        res_map.insert(std::make_pair(fd, res));
        it = res_map.find(fd);
        buffer_new = it->second.getHeader();
    }
    else
        buffer_new = it->second.getBody();
	returnWrite =  write(fd, buffer_new.first, buffer_new.second);
    free(buffer_new.first);
    if (it->second.getIsend())
    {
        res_map.erase(it);
        if (res_map.find(fd)  == res_map.end())
        req_map[fd].InitData();
        FD_CLR(fd, &writecopy);
        close(fd);
    }
}