C中的多线程TCP服务器

时间:2011-02-22 21:23:53

标签: c multithreading tcp client

int sock, connected, bytes_received, true = 1;
struct sockaddr_in server_addr, client_addr;
int sin_size;

if ((sock = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
    perror("Socket");
    exit(1);
}

if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &true, sizeof (int)) == -1) {
    perror("Setsockopt");
    exit(1);
}

server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(atoi(argv[1]));
server_addr.sin_addr.s_addr = INADDR_ANY;
bzero(&(server_addr.sin_zero), 8);

if (bind(sock, (struct sockaddr *) &server_addr, sizeof (struct sockaddr))
        == -1) {
    perror("Unable to bind");
    exit(1);
}

if (listen(sock, 5) == -1) {
    perror("Listen");
    exit(1);
}

printf("\nTCPServer Waiting for client on port 5003");
fflush(stdout); 

while (1) 
{
    pthread_t *child = (pthread_t *)malloc( sizeof(pthread_t) );

    sin_size = sizeof (struct sockaddr_in);
    connected = accept(sock, (struct sockaddr *) &client_addr, &sin_size);
    printf("\n I got a connection from (%s , %d)\n", inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));

    pthread_create(child, NULL, interpretMessage, NULL);
    free(child);
}

我的TCP服务器侦听端口5003.当建立连接时,我想生成一个线程并运行函数interpretMessage。我对此有几个问题......

1)我是否正在调用malloc并在正确的位置释放?

2)调用pthread_create后,我的代码会立即跳转到“free(child);”然后回到while循环的开头?

3)如果5个人在同一时间连接会发生什么(代码如何运行以产生5个线程?)

1 个答案:

答案 0 :(得分:6)

您的代码正确使用mallocfree,但您这样做的方式会使实现变得更加复杂。 pthread_create函数的工作原理是将新线程的ID写入其第一个参数指向的内存中。在您的情况下,您已动态分配此缓冲区,然后立即释放它。因此,内存范围限定为while循环的一次迭代。如果这是你想要的,你可能最好只是将pthread_t堆栈分配并将指针传递给pthread_create

while (1) 
{
    pthread_t child;

    /* ... */

    pthread_create(&child, NULL, interpretMessage, NULL);
}

现在child本地作用于循环,内存管理将自动处理,无需调用mallocfree

关于控件是否继续free (child)然后在调用pthread_create后返回到循环顶部的第二个问题,是的,这是正确的。将创建一个运行interpretMessage的第二个线程,因此如果原始进程被延迟可能会有一些延迟,但是控制确实从此时开始。

对于您的上一个问题,如果五个人都在同一时间连接,那么在接下来的五次调用accept时,该函数将为下一个传入连接提供单个套接字。也就是说,操作系统会自动将传入的连接排队到某个顺序,并且在循环的每次迭代中,您的代码都会注意到有连接,为它获取套接字,然后产生一个线程来处理消息。

我在你的代码中注意到的一件事 - 当你产生一个调用interpretMessage的线程时,你没有为函数提供任何参数,所以每个线程都会在没有任何上下文的情况下运行它创建了。这是故意的吗?