基于Linux用C语言编写的网络聊天室项目（TCP、多线程传输）

server.c       服务端

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<unistd.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<pthread.h>
#include<arpa/inet.h>

#define MAX_BUF 50
#define NAME   20
#define REMIND  50

typedef struct Users
{
  char name[NAME];
  int use_sock;
  pthread_t tid;
  struct sockaddr_in use_addr;
}Users;

Users user[MAX_BUF];

size_t user_count = 0;

void* run(void* arg)
{
  int use_sock = *(int *)arg;
  int index = 0;

  char buf[4096] = {};
  size_t buf_size = sizeof(buf);

  for(int i=0;i<MAX_BUF;i++)
  {
    if(use_sock==user[i].use_sock)
    {
      index = i;
      break;
    }
  }
  recv(use_sock,buf,buf_size,0);
  strcpy(user[index].name,buf);
  sprintf(buf,"欢迎%s来到聊天室^.^",user[index].name);
  puts(buf);
  for(int i=0;i<MAX_BUF;i++)
  {
    if(0!=user[i].use_sock&&use_sock!=user[i].use_sock)
    {
      send(user[i].use_sock,buf,strlen(buf)+1,0);
    }
  }

  for(;;)
  {
    size_t rec_size = recv(use_sock,buf,buf_size,0);
    if(0==rec_size||0==strcmp("quit",buf))
    {
      user[index].use_sock=0;
      close(use_sock);
      user_count--;
      sprintf(buf,"%s退出聊天室...",user[index].name);
      puts(buf);
      for(int i=0;i<MAX_BUF;i++)
      {
        if(0!=user[i].use_sock&&use_sock!=user[i].use_sock)
        {
          send(user[i].use_sock,buf,strlen(buf)+1,0);
        }
      }
      break;
    }

    for(int i=0;i<MAX_BUF;i++)
    {
      if(0!=user[i].use_sock&&use_sock!=user[i].use_sock)
      {
        send(user[i].use_sock,buf,strlen(buf)+1,0);
      }
    }
  }
}

int main()
{
  printf("服务器连接中....\n");
  int sockfd = socket(AF_INET,SOCK_STREAM,0);

  if(0>sockfd)
  {
    perror("socket");
    return EXIT_FAILURE;
  }

  struct sockaddr_in addr = {},temp = {};
  addr.sin_family = AF_INET;
  addr.sin_port = htons(8888);
  addr.sin_addr.s_addr = INADDR_ANY;
  socklen_t addrlen = sizeof(addr);

  if(bind(sockfd,(struct sockaddr*)&addr,addrlen))
  {
    perror("bind");
    return EXIT_FAILURE;
  }

  if(listen(sockfd,5))
  {
    perror("listen");
    return EXIT_FAILURE;
  }

  size_t index = 0;
  for(;;)
  {
    if(0==user[index].use_sock)  
    {
      user[index].use_sock = accept(sockfd,(struct sockaddr*)&temp,&addrlen);
      if(0>user[index].use_sock)
      {
        perror("accept");
        break;
      }
      pthread_t tid;
      user[index].tid = tid;
      pthread_create(&user[index].tid,NULL,run,&user[index].use_sock);
      user_count++;
    }
    else
    {
      if(user_count>=50)
      {
        printf("聊天室已满,请稍等!!!\n");
        sleep(10);
      }
      index = (index+1)%50;
    }
  }
}

client.c          客户端

#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <pthread.h>

#define NAME  	20
#define BUF_SIZE 4096

void* start_recv(void* arg)
{
	int sockfd = *(int*)arg;
	char buf[BUF_SIZE] = {};
	size_t buf_size = sizeof(buf);
	for(;;)
	{
		size_t rec_size = recv(sockfd,buf,buf_size,0);
		if(0==rec_size)
		{
			printf("服务器已断开!!\n");
			return NULL;
		}
		printf("%s\n",buf);
	}
}

int main()
{
	//创建套接字
	int sockfd = socket(AF_INET,SOCK_STREAM,0);
	if(0 > sockfd)
	{
		perror("socket");
		return EXIT_FAILURE;
	}
	//准备地址
	struct sockaddr_in addr = {};
	addr.sin_family = AF_INET;
	addr.sin_port = htons(8890);
	addr.sin_addr.s_addr = inet_addr("112.125.125.79");
	socklen_t len = sizeof(addr);

	//连接
	if(connect(sockfd,(struct sockaddr*)&addr,len))
	{
		perror("connect");
		return EXIT_FAILURE;
	}

	puts("***欢迎来到阿强聊天室***");
	printf("请输入你的聊天名：");
	char name[NAME] = {};
	gets(name);
	//发送姓名
	send(sockfd,name,strlen(name)+1,0);
	pthread_t pid;
	pthread_create(&pid,NULL,start_recv,&sockfd);
	
	char buf[BUF_SIZE] = {};
	char buf2[BUF_SIZE] = {};
	for(;;)
	{
		printf(">>>");
		gets(buf);
		sprintf(buf2,"%s：%s",name,buf);
		size_t rec_size = send(sockfd,buf2,strlen(buf2)+1,0);
		if(0==rec_size||0 == strcmp("quit",buf))
		{
			printf("***退出聊天室***\n");
			close(sockfd);
			return 0;
		}
	}
}