#include <stdio.h>
#include <errno.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>

using namespace std;
#include <iostream>
#include <sstream>

// Определимся с номером порта и другими константами.
#define PORT    5555
#define BUFLEN  4096

// Две вспомогательные функции для чтения/записи (см. ниже)
int   readFromClientHTTP(int fd, char *buf);
void  writeToClientHTTP(int fd, char *buf);


int  main(void)
{
    int     i, err, opt = 1;
    int     sock, new_sock;
    fd_set  active_set, read_set;  
    struct  sockaddr_in  addr;
    struct  sockaddr_in  client;
    char    buf[BUFLEN];
    socklen_t  size;

    printf("http SERVER\n");

    // Инициализация windows sockets
    WSADATA wsaData;
    if ( WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) {
        printf("WSAStartup failed\n");
        return 1;
    }

    // Создаем TCP сокет для приема запросов на соединение
    sock = socket(PF_INET, SOCK_STREAM, 0);
    if (sock<0) {
        perror("Server: cannot create socket");
        exit(EXIT_FAILURE);
    }
    setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (char*)&opt, sizeof(opt));

    // Заполняем адресную структуру и
    // связываем сокет с любым адресом
    addr.sin_family = AF_INET;
    addr.sin_port = htons(PORT);
    addr.sin_addr.s_addr = htonl(INADDR_ANY);
    err = bind(sock, (struct sockaddr*)&addr, sizeof(addr));
    if (err<0) {
        perror("Server: cannot bind socket");
        exit(EXIT_FAILURE);
    }

    // Создаем очередь на 3 входящих запроса соединения
    err = listen(sock, 3);
    if (err<0) {
        perror("Server: listen queue failure");
        exit(EXIT_FAILURE);
    }

    // Подготавливаем множества дескрипторов каналов ввода-вывода.
    // Для простоты не вычисляем максимальное значение дескриптора,
    // а далее будем проверять все дескрипторы вплоть до максимально
    // возможного значения FD_SETSIZE.
    FD_ZERO(&active_set);
    FD_SET(sock, &active_set);

    // Основной бесконечный цикл проверки состояния сокетов
    while (1) {
        // Проверим, не появились ли данные в каком-либо сокете.
        // В нашем варианте ждем до фактического появления данных.
        read_set = active_set;
        if ( select(FD_SETSIZE, &read_set, NULL, NULL, NULL) < 0) {
            perror("Server: select  failure");
            exit(EXIT_FAILURE);
        }
        // fd_set в windows устроена немного иначе - как массив дескрипторов
        // Данные появились. Проверим в каком сокете.
        for (int j = 0; j<read_set.fd_count; j++) {
            i = read_set.fd_array[j];
            if (FD_ISSET(i, &read_set)) {
                if (i == sock) {
                    // пришел запрос на новое соединение
                    size = sizeof(client);
                    new_sock = accept(sock, (struct sockaddr*)&client, &size);
                    if (new_sock<0) {
                        perror("accept");
                        exit(EXIT_FAILURE);
                    }
                    fprintf(stdout, "Server: connect from host %s, port %hu.\n",
                        inet_ntoa(client.sin_addr),
                        ntohs(client.sin_port));
                    FD_SET(new_sock, &active_set);
                    fprintf(stdout, "new socket = %d\n", new_sock);
                }
                else {
                    // пришли данные в уже существующем соединени
                    fprintf(stdout, "socket = %d\n", i);
                    err = readFromClientHTTP(i, buf);
                    if (err<0) {
                        // ошибка или конец данных
                        close(i);
                        fprintf(stdout, "close socket = %d\n", i);
                    } else  {
                        // данные прочитаны нормально
                        writeToClientHTTP(i, buf);
                    }
                    FD_CLR(i, &active_set);
                }
            }
        }
    }
}

// прочитать весь запрос от клиента и положить его в buf
// надеемся что запрос поместится в BUFLEN

int  readFromClientHTTP(int fd, char *buf)
{
    int  nbytes;

    nbytes = recv(fd, buf, BUFLEN, 0);
    fprintf(stdout, "reading %d bytes from socket %d\n", nbytes, fd);
    if (nbytes < 0) {
        // ошибка чтения
        int err = WSAGetLastError();
        printf("Socket read error %d\n", err);
        return -1;
    } else {
        // есть данные
        printf("\n\nServer gets %d bytes:\n", nbytes);
        // поставим 0 в конце прочитанных данных
        buf[min(nbytes, BUFLEN - 1)] = 0;
        printf("%s\n", buf);
        return nbytes;
    }
}



#include <time.h>

void ProcessHTML(stringstream &html, string & request)
{
    /*
    char tbuf[64];   // время, когда обрабатываем этот запрос
    time_t ttt;      // можем потом поставить в заголовок ответа сервера
    time(&ttt);
    ctime_s(tbuf, 256, &ttt);
    printf("%s\n",tbuf);
    */

    //  генерируем выходной html
    // здесь request не влияет на выводимый html, но это нетрудно поменять
 
    html << "<!DOCTYPE html>\r\n";
    html << "<html><head><meta charset = \"cp1251\">\r\n";
    html << "<title>Учебный сервер</title></head>\r\n";
    html << "<body>\r\n";
    html << "<h2>МГУ им М.В.Ломоносова</h2>\r\n";
    html << "<form name=\"myform\" action=\"http://localhost:5555\"  method=\"post\">\r\n";
    //html << "<form name=\"myform\" action=\"http://localhost:5555\"  method=\"get\">\r\n";
    html << "<input name=\"request\" type=\"text\" width=\"20\"> request<br>\r\n";
    html << "<input name=\"comment\" type=\"text\" width=\"20\" value=\"some text\"> comment<br>\r\n";
    //html << "<input name=\"comment\" type=\"text\" width=\"20\" value=\"" << tbuf << "\"> comment<br>\r\n";
    html << "1 <input name=\"rb1\" type=\"radio\" value=\"check1\">\r\n";
    html << "2 <input name=\"rb1\" type=\"radio\" value=\"check2\" checked>\r\n";
    html << "3 <input name=\"rb2\" type=\"radio\" value=\"check3\">\r\n";
    html << "4 <input name=\"rb2\" type=\"radio\" value=\"check4\"><br>\r\n";
    html << "A <input name=\"ch1\" type=\"checkbox\" value=\"checkA\">\r\n";
    html << "B <input name=\"ch2\" type=\"checkbox\" checked>\r\n";
    html << "C <input name=\"ch3\" type=\"checkbox\"><br>\r\n";
    html << "<input name=\"sub\" type=\"submit\" value=\"Send\">\r\n";
    html << "</form>\r\n";
    html << "</body></html>\r\n";
}


//  простейший разбор запроса и отправление ответа

void  writeToClientHTTP(int fd, char *buf)
{
    int  nbytes;
    stringstream head; 
    stringstream html;
    string request = "";   // строка параметров запроса

    if (strstr(buf, "GET")) {  // здесь параметры в первой строке после /
        char *p1 = strchr(buf, '/');
        char *p2 = strchr(p1, ' ');
        *p2 = 0;
        request = p1;
        printf("get request: %s\n", request.c_str());
        *p2 = ' ';
    }
    if (strstr(buf, "POST")) {  // здесь параметры после всего заголовка после пустой строки
        char *p1 = strstr(buf, "\r\n\r\n");
        request = (p1 + 4);
        printf("post request: %s\n", request.c_str());
    }

    // ответ на нераспознанные запросы 
    // не совсем аккуратно, чтобы не загромождать ...
    // отбрасывает непустые GET запросы, в которых нет параметра "request"

    if (request != "/" && request.find("request") == string::npos)  {
        head << "HTTP/1.1 404 Not found\r\n";
        head << "Connection: close\r\n";
        head << "Content-length: 0\r\n";
        head << "\r\n";
        nbytes = send(fd, head.str().c_str(), head.str().length() + 1, 0);
        return;
    }

    // ответ на распознанные запросы
    // формирование HTML
    ProcessHTML(html, request);
    int html_length = html.str().length();

    // формирование HTTP заголовка
    head << "HTTP/1.1 200 OK\r\n";
    head << "Connection: keep-alive\r\n";
    head << "Content-type: text/html\r\n";
    head << "Content-length: " << html_length << "\r\n";
    head << "\r\n";

    // вывод заголовка и html
    nbytes = send(fd, head.str().c_str(), head.str().length() + 1, 0);
    printf("http nb = %d\n", nbytes);
    nbytes = send(fd, html.str().c_str(), html.str().length() + 1, 0);
    printf("html nb = %d\n", nbytes);

    
    // для контроля того, что отправили
    //cout << "Write back\n";
    //cout << head.str();
    //cout << html.str();

    if (nbytes<0) {
        perror("Server: write failure");
    }
}