Linux聊天服务器

服务器端

#include <stdio.h>
#include <ctype.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <strings.h>
#include <string.h>
#include <arpa/inet.h>
#include <errno.h>
#include <sys/wait.h>
#include <pthread.h>
#define PORT 8888

#define Max 10   //最大连接数,也就是可以参与群聊的最大人数
 
#define MAXSIZE 1024

//转发函数的声明
int SendToClient(int fd,char* buf,int Size);

/*定义全局变量*/
int fdt[Max]={0};    //用来存套接字(文件描述符)的数组
char mes[MAXSIZE];   //接收缓冲区
/**/

//子线程函数
void *pthread_service(void* sfd)
{
    int fd=*(int *)sfd;
    while(1)
    {
      
        int i;
        
        int numbytes=recv(fd,mes,MAXSIZE,0);

    if(numbytes<=0)
    {
            for(i=0;i<Max;i++){
                if(fd==fdt[i]){
                    fdt[i]=0;               
                }
            }
        printf("客户端 %d 已退出\n",fd);
        break;
    }

        printf("%s",mes);
           SendToClient(fd,mes,numbytes);
        bzero(mes,MAXSIZE);
    }

    close(fd);
    pthread_exit(0);
}

/*转发函数*/ 
int SendToClient(int fd,char* buf,int Size)
{
    int i;
    int e;
    
    for(i=0;i<Max;i++)
    {     
        //给在线的客户端转发消息
        if((fdt[i]!=0)&&(fdt[i]!=fd)) 
        send(fdt[i],buf,Size,0); 
    }
    bzero(buf,sizeof(buf));
    return 0;
}

int  main()  
{ 

    int listenfd, connectfd;    
    struct sockaddr_in server; 
    struct sockaddr_in client;      
    int sin_size; 
    sin_size=sizeof(struct sockaddr_in); 
    int number=0;

    if ((listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) 
    {   
        perror("创建套接字失败\n");
        exit(1);
    }
    
    bzero(&server,sizeof(server));
      
    server.sin_family=AF_INET; 
    server.sin_port=htons(PORT); 
    server.sin_addr.s_addr = htonl (INADDR_ANY); 

    if (bind(listenfd, (struct sockaddr *)&server, sizeof(struct sockaddr)) == -1) { 
    perror("绑定IP失败\n");
    exit(1); 
    }   

    if(listen(listenfd,1) == -1){  
    perror("监听失败\n"); 
    exit(1); 
    } 
    printf("等待用户连接....\n");

    while(1)
    {

        if ((connectfd = accept(listenfd,(struct sockaddr *)&client,&sin_size))==-1) {
        perror("接收消息失败\n"); 
        exit(1); 
        }

        if(number>=Max){
            printf("不允许更多用户连接\n");
            close(connectfd);
        }

        int i;
        for(i=0;i<Max;i++)
        {
            if(fdt[i]==0)
            {
                fdt[i]=connectfd;
                break;
            }
         }
         
        pthread_t tid;
        pthread_create(&tid,NULL,(void*)pthread_service,&connectfd);
        pthread_detach(tid);
        number=number+1;
    }
    
    close(listenfd);  
    return 0;          
    }
客户端

#include <stdio.h> 
#include <stdlib.h> 
#include <unistd.h>
#include <strings.h>
#include <sys/types.h> 
#include <sys/socket.h> 
#include <netinet/in.h> 
#include <netdb.h>     
#include<string.h>
#include <pthread.h>
#define PORT 8888   
#define MAXSIZE 4096

    char sendbuf[MAXSIZE];   
    char recvbuf[MAXSIZE];
   
    char name[100];
    

    int fd;    //client端只有一个套接字

     //用来接收消息的子线程函数
    void  *pthread_recv(void* ptr)
{
    while(1)
    {
    if ((recv(fd,recvbuf,MAXSIZE,0)) == -1){ 
        printf("接收失败\n"); 
        exit(1); 
       } 
    printf("%s",recvbuf);
    memset(recvbuf,0,sizeof(recvbuf));
    }
}

    int main(int argc, char *argv[]) 
    { 
    int  numbytes;   
    char buf[MAXSIZE];   
    struct hostent *he;       
    struct sockaddr_in saddr;  

    if (argc !=2) {         printf("用户: %s <IP 地址未输入>\n",argv[0]); 
    exit(1); 
    } 

    if ((he=gethostbyname(argv[1]))==NULL){  
    printf("服务器IP获取失败\n"); 
    exit(1); 
    } 

    if ((fd=socket(AF_INET, SOCK_STREAM, 0))==-1){ 
    printf("绑定套接字失败\n"); 
    exit(1); 
    } 

    bzero(&saddr,sizeof(saddr));

    saddr.sin_family = AF_INET; 
    saddr.sin_port = htons(PORT); 
    saddr.sin_addr = *((struct in_addr *)he->h_addr);  
    
    if(connect(fd, (struct sockaddr *)&saddr,sizeof(struct sockaddr))==-1){  
    printf("连接服务器失败\n"); 
    exit(1); 
    } 

    printf("连接聊天室成功\n");
    char str[]="已进入KK聊天室\n";
    printf("请输入用户名：");
    
   //读入一行值，直到遇到回车
    fgets(name,sizeof(name),stdin);
  
    send(fd,name,(strlen(name)-1),0);
    send(fd,str,(strlen(str)),0);

    //创建子线程
    pthread_t tid;
    pthread_create(&tid,NULL,pthread_recv,NULL);
    pthread_detach(tid); 

    //客户端的输入
    while(1)
    {
        memset(sendbuf,0,sizeof(sendbuf));
        
        fgets(sendbuf,sizeof(sendbuf),stdin);
        
        if(strcmp(sendbuf,"exit\n")==0){
            memset(sendbuf,0,sizeof(sendbuf));
            printf("您已退出群聊\n");
            send(fd,sendbuf,(strlen(sendbuf)),0);
            break;
        }
        
        send(fd,name,(strlen(name)-1),0);
        
        send(fd,":",1,0);

        send(fd,sendbuf,(strlen(sendbuf)),0);

    } 
    close(fd);  
    return 0;
 }