C语言实现网络聊天室 socket的简单应用

前言：环境是Linux ，使用了 socket和pthread，主要分为服务器端和客户端两部分，服务器端监听端口发来的请求，收到后向客户端发送一个消息,客户机负责发送消息并打印收到的消息.

编程模型服务端s客户端c创建套接字(socket)创建套接字(socket)准备地址(自己的)准备地址(服务器的,为连接准备)绑定地址与socket设置监听socket等待连接连接服务器接受消息发送消息发送消息接收消息关闭套接字关闭

客户端

#include<stdio.h>#include<stdlib.h>#include<stdbool.h>#include<string.h>#include<unistd.h>#include<sys/types.h>#include<sys/socket.h>#include<netinet/in.h>#include<arpa/inet.h>#include <pthread.h>#include <signal.h>int sockfd;void sigint(int signum){close(sockfd);puts("退出聊天！");exit(0);}void* start(void* p){int fd =*(int*)p;while(1){char buf[1024] = {};if (0 >= recv(fd,buf,sizeof(buf),0) ){return NULL;}printf("\r%s\n",buf);}}int main(){system("clear");signal(SIGINT,sigint);signal(SIGQUIT,sigint);//创建socket对象int sockfd = socket(AF_INET,SOCK_STREAM,0);if(0>sockfd){perror("socket");return -1;}//准备通信地址struct sockaddr_in addr ={};addr.sin_family = AF_INET;addr.sin_port = htons(1703);//端口号addr.sin_addr.s_addr = inet_addr("192.168.1.118");//本地ip//连接if(0>connect(sockfd,(struct sockaddr*)&addr,sizeof(addr))){perror("connect");return -1;}pthread_t pid;pthread_create(&pid,0,start,&sockfd);//发送数据while(1){char buf[1024] ={};char srt[255]={};//printf(">>>");gets(srt);sprintf(buf,"\33[;44m%s\33[0m",srt);send(sockfd,buf,strlen(buf)+1,0);}}

服务器端

#include <stdio.h>#include <string.h>#include <stdbool.h>#include <stdlib.h>#include <unistd.h>#include <sys/types.h>#include <sys/socket.h>#include <netinet/in.h>#include <arpa/inet.h>#include <pthread.h>#include <signal.h>typedef struct sockaddr SockAddr;int fds[20];char ip[20][20];int size = 20;void sigint(int signum){for(int i=0;i<size;i++){close(fds[i]);}puts("客户端已关闭！");exit(0);}void* start(void* p){//向所有聊天室成员提示新用户来了int fd =*(int*)p;char buf3[1024] = {};int index = 0;for(int i=0;i<size;i++){if(fds[i] == fd){index = i;break;}}sprintf(buf3,"\33[;45m%s %s\33[0m ","欢迎",ip[index]);for (int i=0;i<size;i++){if (fds[i]!=0 && fds[i]!=fd){printf("send to %d\n",fds[i]);send(fds[i],buf3,strlen(buf3)+1,0);}}// 收发数据while(1){char buf1[1024] = {};char buf2[1024] = {};//接受信息小于零，则用户退出if (0 >= recv(fd,buf1,sizeof(buf1),0)){//退出后将位置空出来for (int i=0;i<size;i++){if (fd == fds[i]){fds[i] = 0;break;}}//提示用户退出sprintf(buf2,"%s \33[;31m%s\33[0m ",ip[index],"已经退出！");puts(buf2);int i;for ( i=0;i<size;i++){if (fds[i]!=0 && fds[i]!=fd)send(fds[i],buf2,strlen(buf2)+1,0);}send(fds[i],buf2,strlen(buf2)+1,0);//结束线程pthread_exit((void*)i);}//向所有用户发送信息printf("%s:%s\n",ip[index],buf1);sprintf(buf2,"\33[;42m%s\33[0m %s:%s","recv",ip[index],buf1);for (int i=0;i<size;i++){if (fds[i]!=0 && fds[i]!=fd){printf("send to %d\n",fds[i]);send(fds[i],buf2,strlen(buf2)+1,0);}}}}int main(){signal(SIGINT,sigint);signal(SIGQUIT,sigint);// 创建socket对象int sockfd = socket(AF_INET,SOCK_STREAM,0);if(0 > sockfd){perror("socket");return -1;}// 准备通信地址struct sockaddr_in addr = {};addr.sin_family = AF_INET;addr.sin_port = htons(1710);addr.sin_addr.s_addr = inet_addr("192.168.1.117");// 绑定socket对象与通信地址socklen_t len = sizeof(addr);if(0 > bind(sockfd,(SockAddr*)&addr,len)){perror("bind");return -1;}// 设置监听socket对象listen(sockfd,size);// 等待连接while(1){struct sockaddr_in from_addr;int fd = accept(sockfd,(SockAddr*)&from_addr,&len);if(0 > fd){printf("客户端连接出错...\n");continue;}int i;for (i=0;i<size;i++){if (fds[i] == 0){//记录客户端的socketfds[i] = fd;//有客户端连接之后，启动线程给此客户服务pthread_t pid;pthread_create(&pid,NULL,start,&fd);strcpy(ip[i],inet_ntoa(from_addr.sin_addr));printf("客户端：%s连接成功!\n",ip[i]);break;}}if (size == i){//发送给客户端说聊天室满了char* str = "对不起，聊天室已经满了!";send(fd,str,strlen(str),0); close(fd);}}}