尝试保存与成员向量的连接 - 一次只保存一个

时间:2014-03-04 04:54:06

标签: c++ multithreading sockets ubuntu multiprocessing

我正在尝试编写一个简单的服务器应用程序,它可以从多个客户端获取连接,并将这些连接保存到向量中以供另一个对象读取。

我有一个类NetworkConnection,它是另一个类Manager的成员。这是我的主要方法:

int main(int argc, const char* argv[])
{
    RoutingManager *manager = new RoutingManager();
    manager->ParseInputFile("topo.txt", 10, 3, " ");

    manager->myConnection = new NetworkConnection("localhost", "7777");
    manager->myConnection->SetSocketHints();
    manager->myConnection->PopulateAddressInfo();
    manager->myConnection->BindSocket();
    manager->myConnection->ListenForConnections();

    return 0;
}

我遗漏了很多细节,因为它们非常类似于你可以在任何地方找到的常规套接字编程指南......主要问题是在此调用期间:

int
NetworkConnection::ListenForConnections()
{
    char s[INET6_ADDRSTRLEN];

    if (listen(socketFd, 15) == -1) {
        perror("listen");
        exit(1);
    }
/*
    sigAction.sa_handler = sigchld_handler; // reap all dead processes    
    sigemptyset(&sigAction.sa_mask);
    sigAction.sa_flags = SA_RESTART;
*/
    if (sigaction(SIGCHLD, &sigAction, NULL) == -1) {
        perror("sigaction");
        exit(1);
    }

    printf("server: waiting for connections...\n");

    while(1) {  // main accept() loop
        sin_size = sizeof theirAddress;
        int new_fd = accept(socketFd, (struct sockaddr *)&theirAddress, &sin_size);
        if (new_fd == -1) {
            perror("accept");
            continue;
        }

        inet_ntop(theirAddress.ss_family,
            get_in_addr((struct sockaddr *)&theirAddress),
            s, sizeof s);
        printf("server: got connection from %s\n", s);

        if (!fork()) { // this is the child process
            close(socketFd); // child doesn't need the listener

            NodeConnection nc = NodeConnection();

            char ipstr[INET6_ADDRSTRLEN];
            getpeername(new_fd, (struct sockaddr*)&theirAddress, &sin_size);
            struct sockaddr_in *s = (struct sockaddr_in *)&theirAddress;
            int port = ntohs(s->sin_port);
            inet_ntop(AF_INET, &s->sin_addr, nc.ipstr, sizeof ipstr);

            nc.fd = new_fd;
            nc.theirAddress = theirAddress;
            nc.sin_size = sin_size;
            nc.port = port;

            newConnections.push_back(nc);
            for(vector<NodeConnection>::iterator it = newConnections.begin(); it != newConnections.end(); ++it)
            {                
                cout << "Socket ID: " << it->fd << endl;
                cout << "Their Address: " << it->ipstr << endl;
                cout << "Their Port: " << it->port << endl;
                cout << "My Size: " << newConnections.size() << endl;
                cout << "**************\n";
                //cout << "Their Port: " << (int)ntohs(it->theirAddress.sin_port) << endl;
            }
            exit(0);
        }
       // close(new_fd);  // parent doesn't need this
    }
}

基本上,每当我获得新连接时,我都希望将该连接的详细信息存储到向量中。我决定通过向NodeConncetion结构添加细节来实现这一目的:

#include <sys/socket.h>
#include <arpa/inet.h>
struct NodeConnection
{
    int fd;
    socklen_t sin_size;
    struct sockaddr_storage theirAddress; 
    char ipstr[INET6_ADDRSTRLEN];
    int port;
};

这样,我的Manager对象可以轮询这些连接以获取数据。

我遇到的问题是,一旦客户端连接,它就会打印出细节。当另一个连接时,它也打印出那个的细节;但是,它没有像我预期的那样打印第一个连接,因为我做了.push_back(nc)来保存新的NodeConnection。

示例输出:

Hints set.
Binding Socket... Socket Bound!
server: waiting for connections...
server: got connection from 127.0.0.1
Socket ID: 4
Their Address: 127.0.0.1
Their Port: 33744
My Size: 1
**************
server: got connection from 127.0.0.1
Socket ID: 5
Their Address: 127.0.0.1
Their Port: 33745
My Size: 1
**************

我可能无法完全理解如何保存连接数据..所以如果我的方法不对,请告诉我。

1 个答案:

答案 0 :(得分:2)

在调用fork()之后,您正在子进程中将连接详细信息推送到向量。

fork()创建一个全新的进程,它不与父进程共享任何内存。孩子开始父母的相同副本,但是对孩子中的变量所做的更改在父母中是不可见的,反之亦然。

所以你的序列是:

  1. 父接受连接并创建子
  2. Child将连接数据添加到其向量(为空)
  3. 儿童打印矢量(现在包含一个条目)
  4. 孩子退出(它的副本现在已经消失了)
  5. 重复步骤1中的下一个连接。
  6. 如果要保留此连接数据,则需要将其添加到父进程中的向量。

    我注意到你标记了你的问题'多线程' - 这可能解释了混乱,因为你的程序不是多线程的,它是多进程的。同一进程中的多个线程共享内存。我并不建议您更改为使用线程 - 这会带来您可能不需要的一整套其他问题。