Чат на сокетах - C++ - Ответ 9192290

// сервер
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
 
#define PORT 9734
#define BUFSIZE 1024
 
 
void send_to_all(int j, int i, int sockfd, int nbytes_recvd, char *recv_buf, fd_set *master)
{
    if(FD_ISSET(j, master))
    {
        if(j != sockfd && j != i)
        {
            if(send(j, recv_buf, nbytes_recvd, 0) == -1)
                perror("send error");
        }
    }
}
 
void send_recv(int i, fd_set *master, int sockfd, int fdmax)
{
    int nbytes_recvd, j;
    char recv_buf[BUFSIZE], buf[BUFSIZE];
 
    if((nbytes_recvd = recv(i, recv_buf, BUFSIZE, 0)) <= 0)
    {
        if (nbytes_recvd == 0)
            printf("socket %d hung up\n", i);
        else
            perror("recv error");
        close(i);
        FD_CLR(i, master);
    }
    else
    {
        for(j = 0; j <= fdmax; j++)
        {
            send_to_all(j, i, sockfd, nbytes_recvd, recv_buf, master);
        }
 
    }
 
}
 
void connection_accept(fd_set *master, int *fdmax, int sockfd, struct sockaddr_in *client_addr)
{
    socklen_t addrlen;
    int newsockfd;
 
    addrlen = sizeof(struct sockaddr_in);
    if((newsockfd = accept(sockfd, (struct sockaddr *)client_addr, &addrlen)) == -1)
    {
        perror("accept error");
        exit(1);
    }
    else
    {
        FD_SET(newsockfd, master);
        if(newsockfd > *fdmax)
            *fdmax = newsockfd;
        printf("new connection from %s on port %d \n",inet_ntoa(client_addr->sin_addr), ntohs(client_addr->sin_port));
    }
}
 
void connect_request(int *sockfd, struct sockaddr_in *my_addr)
{
    int yes = 1;
 
    if((*sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
    {
        perror("Socket");
        exit(1);
    }
 
    my_addr->sin_family = AF_INET;
    my_addr->sin_port = htons(PORT);
    my_addr->sin_addr.s_addr = INADDR_ANY;
    memset(my_addr->sin_zero, '\0', sizeof my_addr->sin_zero);
 
    if(setsockopt(*sockfd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int)) == -1)
    {
        perror("setsockopt");
        exit(1);
    }
 
    if(bind(*sockfd, (struct sockaddr *)my_addr, sizeof(struct sockaddr)) == -1)
    {
        perror("bind error");
        exit(1);
    }
 
    if(listen(*sockfd, 10) == -1)
    {
        perror("Lisen error");
        exit(1);
    }
    printf("\nTCPServer Waiting for client on port 4950\n");
    fflush(stdout);
 
}
 
int main()
{
    fd_set master;
    fd_set read_fds;
    int fdmax, i;
    int sockfd = 0;
    struct sockaddr_in my_addr, client_addr;
 
    FD_ZERO(&master);
    FD_ZERO(&read_fds);
    connect_request(&sockfd, &my_addr);
    FD_SET(sockfd, &master);
 
    fdmax = sockfd;
    while(1)
    {
        read_fds = master;
        if(select(fdmax+1, &read_fds, NULL, NULL, NULL) == -1)
        {
            perror("select");
            exit(4);
        }
        for (i = 0; i <= fdmax; i++){
 
            if(FD_ISSET(i, &read_fds)){
                if(i == sockfd)
                    //new connection
                    connection_accept(&master, &fdmax, sockfd, &client_addr);
                else
                    //send_recv
                    send_recv(i, &master, sockfd, fdmax);
            }
        }
    }
    return 0;
}

//клиент
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
 
 int main()
{
    char mes[2048];
    char buf[sizeof(mes)];
    char *IP, *point_str_htons;
    char str[20];
    int port,ch;
    int nomer;
    int read = 0;
    int sockfd;
 
    struct sockaddr_in address;
    struct sockaddr_in client;
 
    /* 1 створення TCP-сокета */
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
 
    if(sockfd < 0)
    {
        printf("socket");
        exit(1);
    }
    /* 2 Вказуємо параметри сервера */
    address.sin_family = AF_INET;
    address.sin_port = htons(9734);
    address.sin_addr.s_addr=inet_addr("192.168.....");
   if(connect(sockfd, (struct sockaddr *)&address, sizeof(address)) < 0)
    {
        perror("Connection");
        exit(2);
    }
 
while(1)
    {
        printf("Введіть повідомлення  ");
        if (!strcmp(gets(mes), "exit"))
        {
            close(sockfd);
            return 0;
        }
        printf("відправка повідомлення на сервер.\n ");
 
        /* 3 відправка повідомлення на сервер*/
        send(sockfd, mes, sizeof(mes), 0); /*Відправка повідомлення на сервер */
 
        printf("очікування повідомлення.\n");
       read = recv(sockfd, buf, sizeof(mes),1);
       printf("В= %d\n%s\n", read, buf); // Прийом повідомлення від сервера
 
    }
return 0;
}