Web Сервер

#include <iostream>
#include <WinSock2.h>
#include <WS2tcpip.h> //заголовочный файл для работы с IP-адресам
#include <sstream>
#include <string>
 
#pragma comment(lib, "Ws2_32.lib")
using namespace std;
 
char html[] =
"HTTP/1.0 200 OK\r\n"
"Connection: close\r\n"
"Content - Type: text / plain; charset = utf - 8\r\n"
"Content - Language: ru\r\n"
"Content - Length: 5\r\n"
"\r\n";
long lSize;
char *buffer;
 
void sendRespond(SOCKET client_socket)
{
    char *tmp = new char[lSize + strlen(html)];
 
    int intlt = 0;
    for (; intlt < strlen(html); intlt++)
        tmp[intlt] = html[intlt];
    for (int i = 0; intlt < strlen(html) + lSize; intlt++, i++)
        tmp[intlt] = buffer[i];
 
    send(client_socket, tmp, intlt, 0);
}
 
bool loadFile(char * file)
{
    
    if (file[2] == 0)
    file = ".\\index.html";
    FILE *File = fopen(file, "rb");
    if (File == NULL)
        return false;
 
    fseek(File, 0, SEEK_END);
    lSize = ftell(File);
    rewind(File);
 
    buffer = new char[lSize];
    fread(buffer, sizeof(char), lSize, File);
    fclose(File);
    return true;
}
 
 
int handleClient(SOCKET client_socket)
{
    char cCur = 0, cLast = 0, cBuf = 0;
    bool headersEnded = false;
    char reqHdr[1024] = { 0 };
    int reqHdrLen = 0;
 
    // Ожидание, пока мы не получили конец заголовков HTTP запроса
    // Прежде чем начать отправку ответа
    do {
        int count = recv(client_socket, &cBuf, 1, 0);
        if (count == SOCKET_ERROR) {
            cout << "Could not recv from client";
            return 1;
        }
 
        if (cBuf != '\r' && cBuf != '\b') {
            cLast = cCur;
            cCur = cBuf;
            if (reqHdrLen < 1024)
                reqHdr[reqHdrLen++] = cBuf;
        }
 
        if (cCur == '\n' && cLast == '\n')
            headersEnded = true;
    } while (!headersEnded); 
    int verbose = 1;
    if (verbose)
    cout << reqHdr; //Выводим запросы
 
    char get[1024] = {0};
    int cursor = 0, target = 0;
    while (reqHdr[cursor++] != ' '){}
    get[target++] = '.';
    get[target++] = '\\';
    cursor++;
    
    while (reqHdr[cursor] != ' ')
        get[target++] = reqHdr[cursor++];
 
    if (strstr(get, ".\\error"))
    {
        char* ptr = "HTTP/1.1 503 Service Temporarily Unavailable\r\nConnection: close\r\n\r\nSERVER PUT DOWN";
        send(client_socket, ptr, strlen(ptr), 0);
 
        return 0;
    }
 
    ;
    if (loadFile(get))
        sendRespond(client_socket);
    else
    {
        char* ptr = "HTTP/1.1 404 File Not Found\r\nConnection: close\r\n\r\nNETU STRANICY";
        send(client_socket, ptr, strlen(ptr), 0);
    }
    return 0;
}
 
 
int main(int argc, char argv[])
{
    
//  char test_page[6592];
    // служебная структура для хранение информации
    // о реализации Windows Sockets
    WSADATA wsaData;  
 
    // старт использования библиотеки сокетов процессом
    // (подгружается Ws2_32.dll)
    int result = WSAStartup(MAKEWORD(2, 2), &wsaData);
 
    // Если произошла ошибка подгрузки библиотеки
    if (result != 0) {
        cerr << "WSAStartup failed: " << result << "\n";
        return result;
    }
    
    // Шаблон для инициализации структуры адреса
    struct addrinfo* addr = NULL; // структура, хранящая информацию об IP-адресе  слущающего сокета
    struct addrinfo hints;
    ZeroMemory(&hints, sizeof(hints)); // убедимся, что структура пуста
    hints.ai_flags = AI_PASSIVE; //установка на прием входящих сообщений
    hints.ai_family = AF_UNSPEC;//неважно какой Ipv
    hints.ai_socktype = SOCK_STREAM;//задаем потоковый тип сокета
    hints.ai_protocol = IPPROTO_TCP;//используем протокол TCP
 
    //Инициализируем структуру, храняшую адррес соката - addr
    //HTTP сервер весит на порту, взятый из файла настройки(port.txt);
    result = getaddrinfo("127.0.0.1", "80", &hints, &addr);
    //Если инициализация не удалась, выводим сообщение
    // и завершаем программу
    if (result != 0) {
        cerr << "getaddrinfo failed: " << result << "\n";
        WSACleanup();
        return 1;
    }
    //Создание сокета
    int s = socket(addr->ai_family, addr->ai_socktype, addr->ai_protocol);
    //Если создание не удалалось, выводим сообщение, освобождаем память
    //выгружаем dll-библиотеку, закрываем программу
    if (s == INVALID_SOCKET)
    {
        cerr << "Error at socket: " << WSAGetLastError() << "\n";
        freeaddrinfo(addr);
        WSACleanup();
        return 1;
    }
 
    //Привязка сокета к адресу
    result = bind(s, addr->ai_addr, (int)addr->ai_addrlen);
 
    //Если привязка не удалась, выводим сообщение
    //освобождаем память, закрываем сокет
    //выгружаем dll-библиотеку и закрываем программу
 
    if (result == SOCKET_ERROR){
        cerr << "Bind failed with error: " << WSAGetLastError << "\n";
        freeaddrinfo(addr);
        closesocket(s);
        WSACleanup();
        return 1;
    }
 
    // Инициализируем слушающий сокет
    //Если инициализация не удалась, выводим сообщение,
    //закрываем сокет, выгружаем dll-библиотеку
    //закрываем программу
    if (listen(s, 5) == SOCKET_ERROR) {
        cerr << "Listen failed with error: " << WSAGetLastError() << "\n";
        closesocket(s);
        WSACleanup();
        return 1;
    }
    //cout << "Listening on port: " << port;
    //Принимаем входящие сообщения
    //Если произошла ошибка соединения, выводим сообщение,
    //закрываем сокет, выгружаем dll-библиотеку
    //закрываем программу
    int client_socket = INVALID_SOCKET;
    while (client_socket = accept(s, NULL, NULL))
    {
        if (client_socket == INVALID_SOCKET){
            cerr << "Accept failed: " << WSAGetLastError() << "\n";
            closesocket(s);
            WSACleanup();
            return 1;
        }
        if (client_socket == SOCKET_ERROR) {
            //ошбика получения данных
            cerr << "Recv failed: " << result << "\n";
            closesocket(client_socket);
        }
        else if (client_socket == 0) {
            cerr << "\nConnection closed...\n";
        }
        else if (client_socket > 0) {
            handleClient(client_socket);
            if (client_socket == SOCKET_ERROR) {
                // произошла ошибка при отправле данных
                cerr << "send failed: " << WSAGetLastError() << "\n";
            }
        }
        // Закрываем соединение к клиентом
        closesocket(client_socket);
    }
        closesocket(s);
        freeaddrinfo(addr);
        WSACleanup();
        return 0;
}