用C语言编写的多线程TCP服务器

4
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和free?
2)在pthread_create被调用后,我的代码是否立即跳转到“free(child);”然后回到while循环的开头?
3)如果同时有5个人连接会发生什么(代码如何运行以生成5个线程?)
1个回答

5
你的代码正确地使用了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时,您没有为该函数提供任何参数,因此每个线程都将在没有创建上下文的情况下运行。这是有意为之吗?

你好,非常感谢您的回复。我为了简单起见删除了参数,因为我将一个结构体传递给interpretMessage函数,我不想用结构体的内容来混淆任何人。我现在明白我需要做什么了,非常感谢! - Eddie

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接