/**
*  В.Д.Валединский
*  Простейший пример TCP клиента и сервера.
*
*  Файлы
*  cli_tcp.cpp  - пример TCP клиента
*  ser_tcp_select.cpp - пример TCP сервера c использованием select()
*
*  Клиент в цикле получает с клавиатуры текстовую строку и отсылает
*  ее на сервер, читает и выводит на экран ответ сервера.
*  Сервер ждет соединений от клиентов. При установленом
*  соединении получает строку от клиента, переводит ее в верхний
*  регистр и отсылает обратно клиенту. Если клиент посылает строку
*  содержащую слово stop, то сервер закрывает соединение с этим клиентом,
*  а клиент заканчивает свою работу.
*  Клиент и сервер печатают на экран протокол своей работы т.е.
*  разные данные, которые они получают или отсылают.
*
*  Для ясности примеры составлены максимально просто и не анализируют
*  некорректные ответы и запросы клиента и сервера, возможность переполнения
*  буферов ввода вывода, неожиданное 'падение' сервера и т.п.
*
*/


#include "stdafx.h"
#pragma warning(suppress : 4996)
#pragma comment(lib, "Ws2_32.lib")
#define _WINSOCK_DEPRECATED_NO_WARNINGS

#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <stdlib.h>
#include <sys/types.h>
#include <winsock2.h>
#include <ws2tcpip.h>

// Определимся с портом, адресом сервера и другими константами.
// В данном случае берем произвольный порт и адрес обратной связи
// (тестируем на одной машине).
#define  SERVER_PORT     5555
//#define  SERVER_PORT     80
#define  SERVER_NAME    "127.0.0.1"
#define  BUFLEN          512

// Две вспомогательные функции для чтения/записи (см. ниже)
int  writeToServer(int fd);
int  readFromServer(int fd);


int  main(int argc, char ** argv)
{
    // с командной строки задаем адрес
    // по идее можно обратиться куда угодно :)
    const char *serverName;
    serverName = (argc < 2) ? "127.0.0.1" : argv[1];
    printf("connecting to %s\n", serverName);

    int err;
    int sock;
    struct sockaddr_in server_addr;
    struct hostent    *hostinfo;

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

    // Получаем информацию о сервере по его DNS имени
    // или точечной нотации IP адреса.
    hostinfo = gethostbyname(serverName);
    char sss[256];
    printf("host %s\n", inet_ntop(AF_INET, hostinfo->h_addr, sss, 256));
    if (hostinfo == NULL) {
      fprintf(stderr, "Unknown host %s.\n", SERVER_NAME);
      exit(EXIT_FAILURE);
    }
    // можно было бы использовать GetAddrInfo()

    // Заполняем адресную структуру для последующего
    // использования при установлении соединения
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(SERVER_PORT);
    server_addr.sin_addr = *(struct in_addr*) hostinfo->h_addr;
   
    // другой вариант задания IP адреса
    //unsigned int iaddr;
    //inet_pton(AF_INET, serverName, &iaddr);
    //server_addr.sin_addr.s_addr = iaddr;


    // Создаем TCP сокет.
    sock = socket(PF_INET, SOCK_STREAM, 0);
    if (sock<0) {
        perror("Client: socket was not created");
        exit(EXIT_FAILURE);
    }

    // Устанавливаем соединение с сервером
    err = connect(sock, (struct sockaddr*)&server_addr, sizeof(server_addr));
    if (err<0) {
        perror("Client:  connect failure");
        exit(EXIT_FAILURE);
    }
    fprintf(stdout, "Connection is ready\n");

    // Обмениваемся данными
    while (1) {
        if (writeToServer(sock)<0) break;
        if (readFromServer(sock)<0) break;
    }
    fprintf(stdout, "The end\n");

    // Закрываем socket
    closesocket(sock);
    WSACleanup();
    exit(EXIT_SUCCESS);
}



int  writeToServer(int fd)
{
    int   nbytes;
    char  buf[BUFLEN];

    fprintf(stdout, "Send to server > ");
    if (fgets(buf, BUFLEN, stdin) == nullptr) {
        printf("error\n");
    }
    buf[strlen(buf) - 1] = 0;

    nbytes = send(fd, buf, strlen(buf) + 1, 0);
    printf("send %d bytes\n", nbytes);
    //if (nbytes<0) { perror("write"); return -1; } // не работает с win socket
    if (strstr(buf, "stop")) return -1;
    return 0;
}


int  readFromServer(int fd)
{
    int   nbytes;
    char  buf[BUFLEN];

    nbytes = recv(fd, buf, BUFLEN, 0);
    if (nbytes<0) {
        // ошибка чтения
        perror("read");
        return -1;
    }
    else if (nbytes == 0) {
        // нет данных для чтения
        fprintf(stderr, "Client: no message\n");
    }
    else {
        // ответ успешно прочитан
        fprintf(stdout, "Server's replay: %s\n", buf);
    }
    return 0;
}