2013-10-13 34 views
6

我有一個簡單的服務器和客戶端C代碼來爲多客戶端使用線程(pthread庫)做一個聊天室。我遇到的問題是我不能想辦法讓服務器將客戶端通過套接字發送到所有其他客戶端的每條消息寫入。我在這裏看過其他類似的帖子,很無奈。請幫助我,我需要爲學校做這個。我會馬上發送這兩個代碼。在Linux中的C/Socket編程中的聊天室

Server.c:

#include<stdio.h> 
#include<string.h> //strlen 
#include<stdlib.h> //strlen 
#include<sys/socket.h> 
#include<arpa/inet.h> //inet_addr 
#include<unistd.h> //write 

#include<pthread.h> //for threading , link with lpthread 

void *connection_handler(void *); 


int main(int argc , char *argv[]) 
{ 
    int socket_desc , new_socket , c , *new_sock; 
    struct sockaddr_in server , client; 
    char *message; 

    //Create socket 
    socket_desc = socket(AF_INET , SOCK_STREAM , 0); 
    if (socket_desc == -1) 
    { 
     printf("Could not create socket"); 
    } 

    //Prepare the sockaddr_in structure 
    server.sin_family = AF_INET; 
    server.sin_addr.s_addr = INADDR_ANY; 
    server.sin_port = htons(8888); 

    //Bind 
    if(bind(socket_desc,(struct sockaddr *)&server , sizeof(server)) < 0) 
    { 
     puts("bind failed"); 
     return 1; 
    } 
    puts("bind done"); 

    //Listen 
    listen(socket_desc , 3); 

    //Accept and incoming connection 
    puts("Waiting for incoming connections..."); 
    c = sizeof(struct sockaddr_in); 
    while((new_socket = accept(socket_desc, (struct sockaddr *)&client, (socklen_t*)&c))) 
    { 
     puts("Connection accepted"); 

     pthread_t sniffer_thread; 
     new_sock = malloc(1); 
     *new_sock = new_socket; 

     if(pthread_create(&sniffer_thread , NULL , connection_handler , (void*) new_sock) < 0) 
     { 
      perror("could not create thread"); 
      return 1; 
     } 

     //Now join the thread , so that we dont terminate before the thread 
     //pthread_join(sniffer_thread , NULL); 
     puts("Handler assigned"); 
    } 

    if (new_socket<0) 
    { 
     perror("accept failed"); 
     return 1; 
    } 

    return 0; 
} 

/* 
* This will handle connection for each client 
* */ 
void *connection_handler(void *socket_desc) 
{ 
    //Get the socket descriptor 
    int sock = *(int*)socket_desc; 
    int read_size; 
    char *message , client_message[2000]; 


    //Receive a message from client 
    while((read_size = recv(sock , client_message , 2000 , 0)) > 0) 
    { 
     //Send the message back to client 
     write(sock , client_message , strlen(client_message)); 
    } 

    if(read_size == 0) 
    { 
     puts("Client disconnected"); 
     fflush(stdout); 
    } 
    else if(read_size == -1) 
    { 
     perror("recv failed"); 
    } 

    //Free the socket pointer 
    free(socket_desc); 

    return 0; 
} 

Client.c

#include<stdio.h> //printf 
#include<string.h> //strlen 
#include<sys/socket.h> //socket 
#include<arpa/inet.h> //inet_addr 

int main(int argc , char *argv[]) 
{ 
    int sock; 
    struct sockaddr_in server; 
    char message[1000] , server_reply[2000]; 

    //Create socket 
    sock = socket(AF_INET , SOCK_STREAM , 0); 
    if (sock == -1) 
    { 
     printf("Could not create socket"); 
    } 
    puts("Socket created"); 

    server.sin_addr.s_addr = inet_addr("127.0.0.1"); 
    server.sin_family = AF_INET; 
    server.sin_port = htons(8888); 

    //Connect to remote server 
    if (connect(sock , (struct sockaddr *)&server , sizeof(server)) < 0) 
    { 
     perror("connect failed. Error"); 
     return 1; 
    } 

    puts("Connected\n"); 
    puts("Bienvenido al Chatroom, puedes empezar a escribir en la sala!"); 

    //keep communicating with server 
    while(1) 
    { 

     printf("Enter message: "); 
    fgets(message, sizeof(message),stdin); 
     //scanf("%s" , message); 

     //Send some data 
     if(send(sock , message , strlen(message) , 0) < 0) 
     { 
      puts("Send failed"); 
      return 1; 
     } 

     //Receive a reply from the server 
     if(recv(sock , server_reply , 2000 , 0) < 0) 
     { 
      puts("recv failed"); 
      break; 
     } 

    printf("Server Reply: %s\n", server_reply); 
    server_reply[0]='\0'; 
    } 

    close(sock); 
    return 0; 
} 

這些程序很簡單,客戶端發送什麼用戶寫在控制檯和服務器發送相同的消息回來了。我只需要服務器向連接的每個線程(客戶端)發送相同的消息(不僅是發送原始消息的那個人)。

我知道這是很長的人關心,但是如果可以的話,我會很高興得到一些幫助:)

+0

是否需要使用此線程?一個更簡單的設計將是單線程的,並使用select()或poll()來複用I/O操作。然後寫入所有客戶的套接字可以通過一個簡單的for循環完成。 –

+0

哦謝謝,這可能會更容易。不過,我不知道如何使用select()或poll()哈哈。我使用了線程,因爲最近我學會了如何使用它們,並且它們似乎是我的問題的很好的解決方案。任何機會,你可以給我一些關於如何使用這些功能的細節? :) –

+0

線程是強大的藥物 - 他們可以咬你,除非你非常小心。除非沒有其他方式來完成任務,否則我儘量不要使用它們。至於select(),請參閱以下鏈接:http://www.lowtek.com/sockets/select.html http://www.binarytides.com/multiple-socket-connections-fdset-select-linux/ –

回答

5

你需要的所有客戶端的全局表,由一個互斥保護。當新客戶端連接時,將它們添加到全局客戶端表中。當客戶端斷開時,將它們從全局表中刪除。

當客戶端發送消息時,獲取全局表上的鎖並遍歷它。將消息發送給所有客戶端(除了發送消息的客戶端以外,如果不想將消息回送到其源)。

這是真正的服務器如何做到這一點的過分簡化。但它應該足以讓你開始。