0

我有一个如下的客户端程序，我需要使它成为多线程的，即每个连接一个线程。但是变量 sockfd 对于一个线程来说是全局的。我知道要做到这一点，我需要使用 pthread_key_t、pthread_key_create...等。但是，我很困惑如何使用它。如果有任何帮助，我将不胜感激。

int sockfd;
pthread_key_t key_to_sockfd;

void error(const char *msg)
{
   perror(msg);
   exit(0);
}


void set_connection(char *argv[])
{

   int portno;
   struct sockaddr_in serv_addr;
   struct hostent *server;

   char buffer[256];
   portno = atoi(argv[2]);
   sockfd = socket(AF_INET, SOCK_STREAM, 0);
   if (sockfd < 0) 
       error("ERROR opening socket");
   server = gethostbyname(argv[1]);
   if (server == NULL) {
       fprintf(stderr,"ERROR, no such host\n");
       exit(0);
 }
bzero((char *) &serv_addr, sizeof(serv_addr));
serv_addr.sin_family = AF_INET;
bcopy((char *)server->h_addr, 
(char *)&serv_addr.sin_addr.s_addr,server->h_length);
serv_addr.sin_port = htons(portno);
if (connect(sockfd,(struct sockaddr *) &serv_addr,sizeof(serv_addr)) < 0) 
    error("ERROR connecting");
return;
}

void send_message()
{
    char buffer[256];
    int i=0,n;
    do{
      printf("Please enter the message: ");
      bzero(buffer,256);
      fgets(buffer,255,stdin);
       n = write(sockfd,buffer,strlen(buffer));
      if (n < 0) 
        error("ERROR writing to socket");
    i++;
 }while(i<3);
bzero(buffer,256);
n = read(sockfd,buffer,255);
if (n < 0) 
     error("ERROR reading from socket");
printf("%s\n",buffer);
return;
}

void disconnect()
{
   close(sockfd);
   return;
}

void client_thrd(char *argv[])
{
  set_connection(argv);
  send_message();
  disconnect();
}
int main(int argc, char *argv[])
{

pthread_t thid[2];  
int i;  
void *status;
if (argc < 3) {
   fprintf(stderr,"usage %s hostname port\n", argv[0]);
   exit(0);
}
for(i=0;i<1;i++)
    pthread_create(&thid[i],NULL,(void*)&client_thrd,(void*)argv);
for(i=0;i<1;i++)
    pthread_join(thid[i],&status);


return 0;
}

我已经为一个线程执行了程序并且工作正常。但是当我将线程数增加到一个以上时，它当然不起作用，因为全局变量 sockfd。这是一个更大的测试程序。

谢谢并恭祝安康

4

3 回答 3

3

这是非常容易使用。您pthread_key_create(&key_to_sockfd, NULL)在程序开始时调用，然后在每个线程中您将看到 NULL 的初始值。pthread_setspecific(key_to_sockfd, <pointer to a struct which contains sockfd>)在每个线程中使用该函数。

从那时起，每个线程将看到一个指向包含您的 sockfd 的结构的不同指针。当你的线程完成后，你删除结构并使用pthread_setspecific(key_to_sockfd, NULL)

线程完成后，您调用pthread_key_delete(key_to_sockfd)以删除存储。您还可以通过提供回调函数来自动清理，pthread_key_create以便在线程完成时释放内存。

于 2011-09-14T06:03:02.897 回答
2

据我所知，您不需要全局变量来实现您想要做的事情。创建一个“状态”数据结构并在启动之前为每个线程初始化它。线程接口中的void*参数就是为此而制定的。

于 2011-09-14T06:06:16.437 回答
0

如果我查看您的代码，您会用每个线程覆盖您启动套接字描述符 sockfd。所以每个线程都会建立一个新的连接并获得一个新的 sockfd。使用此代码，您很有可能一个线程正在关闭另一个线程的连接。如果你想为每个线程使用它自己的套接字描述符，为什么不使用指针在同一个线程中共享同一个套接字描述符？

void client_thrd(char *argv[])
{
  int sockfd;
  set_connection(&sockfd, argv);
  send_message(&sockfd);
  disconnect(&sockfd);
}
于 2011-09-14T06:07:42.007 回答