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



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

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

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


int  main (void)
{
//    int i;
    int err;
    int sock;
    struct sockaddr_in server_addr;
    struct hostent    *hostinfo;

    // Получаем информацию о сервере по его DNS имени
    // или точечной нотации IP адреса.
    hostinfo = gethostbyname(SERVER_NAME);
    if ( hostinfo==NULL ) {
        fprintf (stderr, "Unknown host %s.\n",SERVER_NAME);
        exit (EXIT_FAILURE);
    }

    // Заполняем адресную структуру для последующего
    // использования при установлении соединения
    server_addr.sin_family = PF_INET;
    server_addr.sin_port = htons(SERVER_PORT);
    server_addr.sin_addr = *(struct in_addr*) hostinfo->h_addr;

    // Создаем 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
    close (sock);
    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 = write (fd, buf, strlen(buf)+1);
    if ( nbytes<0 ) { perror("write"); return -1; }
    if (strstr(buf,"stop")) return -1;
    return 0;
}


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

    nbytes = read(fd, buf, BUFLEN);
    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;
}