Чат на C++ под Linux - C++ - Ответ 5836397

#include <sys/socket.h>
#include <sys/types.h>
#include <sys/un.h>
#include <iostream>
#include <vector>
#include <fcntl.h>
#include <algorithm>
#include <set>
 
#define PORT 5000 
using namespace std;
 
class Client{
 
public:
        vector<char> name;
    int sock;
    struct sockaddr_un address;
    void show_vector( vector<char>&a)
    {
        for (vector<char>::iterator it = a.begin() ; it!=a.end() ; ++it)
            cout<<*it;
    }
    Client(vector<char> name){
        cout<<"Hey ";
        show_vector(name);
        cout<<",what's up?"<<endl;
    }
    ~Client(){
        cout<<"Bye"<<endl;
    }
    
    int create_socket(){
            sock = socket(AF_UNIX, SOCK_STREAM,0);
        address.sun_family = AF_UNIX;
        strcpy(address.sun_path, "server_socket");
        int len = sizeof(address);
        int result = connect(sock, (struct sockaddr *)&address, len);
        if(result==-1){
            cout<<("oops: client");
            return 0;
        }
    }
    
    int send_message(char *message, int size){
        send(sock, message, size, 0);
        return 0;
    }
 
    int send_message(vector<char> message, int size){
        char* str=&message[0];
        cout<<"sending="<<str<<endl;
        cout<<"size="<<message.size()<<endl;
        send(sock, str, size, 0);
        cout<<"send="<<str<<endl;
        return 0;
    }
 
    char* recv_message(){   
        char buf[1024];
        recv(sock, buf, sizeof(buf), 0);
        return buf;
    }
    
    int close_socket(){
        return close(sock);
    }
    
};
class Server{
 
public:
    int  server_sock;
    struct sockaddr_un server_address;
    set<int> clients;
    Server(){
        unlink("server_socket");
        cout<<"Running"<<endl;
    }
    ~Server(){
        unlink("server_socket");
        cout<<"Stopping"<<endl;
    }
    
    int create_socket(){
        server_sock=socket(AF_UNIX, SOCK_STREAM,0);
        fcntl(server_sock, F_SETFL, O_NONBLOCK);
        server_address.sun_family = AF_UNIX;
        strcpy(server_address.sun_path,"server_socket");
        int server_len=sizeof(server_address);
        bind(server_sock, (struct sockaddr *)&server_address, server_len);
        listen(server_sock,2);
        return 0;
    }
    
    /*int get_client(){
        cout<<"new connection "<<sock<<endl;
        return 0;
 
    }*/
    
    int show_all(){
 
        return 0;
    }
    
    int get_connect(){
        char buf[50]={0};   
        fd_set readset;
        fd_set writeset;
        FD_ZERO(&readset);
        FD_ZERO(&writeset);
        FD_SET(server_sock,&readset);
        for(set<int>::iterator it = clients.begin(); it != clients.end(); it++)
        {
                FD_SET(*it, &readset);
            FD_SET(*it, &writeset);
        }   
        timeval timeout;
            timeout.tv_sec = 30;
            timeout.tv_usec = 0;
        int mx = max(server_sock, *max_element(clients.begin(), clients.end()));
        if(select(mx+1, &readset, &writeset, NULL, &timeout) <= 0)
            {
                cout<<"select error"<<endl;
            return 0;
        }
        if(FD_ISSET(server_sock, &readset))
            {
                 // Поступил новый запрос на соединение, используем accept
            int sock = accept(server_sock, NULL, NULL);
            
                    if(sock < 0)
            {
                        cout<<"error accept"<<endl;
                        return 0;
            }
                    fcntl(sock, F_SETFL, O_NONBLOCK);
            clients.insert(sock);
 
            //имя клиента
            int bytes_read = recv(sock, buf, 10, 0);
            cout<<"recv name="<<buf<<" - "<<sock<<endl;
            buf[1]=0;
            }
        for(set<int>::iterator it = clients.begin(); it != clients.end(); it++)
            {
                    if(FD_ISSET(*it, &readset))
            {
                int to;
                char str[2]={0};
 
                //получили адрес сокета куда отправлено сообщение
                int bytes_read = recv(*it, str, 2, 0);
                cout<<"bytes_read="<<bytes_read<<endl;
                cout<<"recv address="<<str<<endl;
                to=atoi(str);
                cout<<"to="<<to<<endl;
 
                // Поступили данные от клиента, читаем их
                buf[0]=0;
                bytes_read = recv(*it, buf, 50, 0);// Проблема! Получаем что соединение закрыто
                cout<<"bytes_read="<<bytes_read<<endl;
                cout<<"message="<<buf<<endl;
                cout<<"from soket="<<*it<<endl;
 
                if(bytes_read <= 0)
                {
                    // Соединение разорвано, удаляем сокет из множества
                    close(*it);
                    clients.erase(*it);
                    continue;
                }
 
                // Отправляем данные клиенту
                send(to, buf, 1024, 0);
            }   
            }
    }
    int close_socket(){
        return close(server_sock);
    }
};

#include "soket.h"
vector<char> get_message(){
               char s;
               vector <char> name;
               while ((cin >> s) && (s!= '0') ) name.push_back(s);//Проблема! если писать s!= '\n' то не воспринимает
               return name;
        }
 int main(){
    cout<<"Put your name please>>";
    vector<char> name=get_message(); 
    Client client1(name);
    client1.create_socket();
    client1.send_message(name,10);//посылаем имя клиента
    char str[2];    
    cout<<"client socket="<<client1.sock<<endl;
    while(1){
        char buf[50];
        int user;
        cout<<"to whom?>>";
        cin>>user;
        str[0] = user + '0';
        str[1] = 0;
        cout<<"to user="<<str<<endl;
        client1.send_message(str,2);//посылаем номер сокета клиента
        cout<<">>";
        vector<char> message=get_message();
        client1.send_message(message,50);//посылка сообщения
        cout<<"sended\n";       
        cout<<client1.recv_message();
    }
    cout<<"crap\n";
    client1.close_socket();
    return 0;
}

#include "soket.h"
int main(){
    Server server;
    server.create_socket();
    while(1){
        server.get_connect();
}
    server.close_socket();
}