群聊模拟

客户端

/*************************************************************************> File Name: client.c> Author: 联合作战实验室> Email: [e-mail] > Created Time: Sat 25 Mar 2017 05:18:10 PM CST ************************************************************************/#include<stdio.h>#include<stdlib.h>#include<signal.h>#include<string.h>#include<unistd.h>#include<arpa/inet.h>#include<sys/socket.h>#include<pthread.h>#include<netinet/in.h>int sockfd; // socket descriptorvoid init_client(int sip, int port);// init clientvoid start_client();// send msg to servervoid destroy_client(int signo);// close clientvoid* rcvmsg(void* p);// recv msg form serverint main(int argc, const char *argv[]){int port, sip;if(argc < 3){printf("usage: %s serverip port.\n", argv[0]);return -1;}port = htons(atoi(argv[2]));sip = inet_addr(argv[1]);signal(SIGINT, destroy_client); // close clientinit_client(sip, port);start_client();return 0;}void init_client(int sip, int port){printf("connect server......\n");sleep(1);// 1.create socketsockfd = socket(AF_INET, SOCK_STREAM, 0);if(-1 == sockfd) {perror("sockfd"), exit(-1);}// 2.create communication objectstruct sockaddr_in serveraddr;serveraddr.sin_family = AF_INET;serveraddr.sin_port = port;serveraddr.sin_addr.s_addr = sip;//3.connect serverint res = connect(sockfd, (struct sockaddr*)&serveraddr, sizeof(serveraddr));if(-1 == res) {perror("connect client error: "), close(sockfd), exit(-1);}printf("connect server success.\n");}void start_client(void){char name[100] = {};char buf[100] = {};char msg[200] = {};pthread_t tid;// 1.send msg to server.printf("Please input you net-name: ");scanf("%s", name);strcat(name, ":");// 2. create new thread to accept msg form server.pthread_create(&tid, 0, rcvmsg, 0); // recv msg form server// send msg to server.while(1) {printf("Please input you msg: "); // send msg.scanf("%s", buf);fflush(stdin);sprintf(msg, "%s%s", name, buf);send(sockfd, msg, strlen(msg), 0);memset(msg, 0, sizeof(msg));memset(buf, 0, sizeof(buf));}}void* rcvmsg(void* p){char buf[200] = {};while(1) {recv(sockfd, buf, sizeof(buf), 0);printf("\n%s\n", buf);memset(buf, 0, sizeof(buf));}}void destroy_client(int signo){printf("Closing client......\n");close(sockfd);sleep(1);exit(0);}

服务器

/*************************************************************************> File Name: server.c> Author: 联合作战实验室> Email: [e-mail] > Created Time: Sat 25 Mar 2017 06:01:20 PM CST ************************************************************************/#include <sys/types.h>#include <sys/socket.h>#include <stdio.h>#include <stdlib.h>#include <unistd.h>#include <netinet/in.h>#include <arpa/inet.h>#include <pthread.h>#include <signal.h>#include <string.h>#include "list.h"int sockfd;//socket descriptorsocklist *slist = NULL;// all client socket fd list;static void init_server(int sip, int port);// init serverstatic void start_server(void);// start serverstatic void destroy_server(int signo);// close serverstatic void* rcvMsg(void* p);// recv msg form clientstatic void sendMsgToAll(void* msg);// send msg to all on line clientint main(int argc, const char* argv[]){int sip, port;if(argc < 3) {printf("usage: %s serverip port\n", argv[0]);return -1;}sip = inet_addr(argv[1]);port = htons(atoi(argv[2]));signal(SIGINT, destroy_server);init_server(sip, port);start_server();return 0;}static void init_server(int sip, int port){printf("Initializing server...\n");sleep(1);slist = create_empty_linklist();//1. create socketsockfd = socket(AF_INET, SOCK_STREAM, 0);if(-1 == sockfd){perror("socket server: "),exit(-1);}//2. communication object.struct sockaddr_in serveraddr = {};serveraddr.sin_family = AF_INET;serveraddr.sin_port = port;serveraddr.sin_addr.s_addr = sip;//3. bind socket and socket addressint res = bind(sockfd, (struct sockaddr*)&serveraddr, sizeof(serveraddr));if(-1 == res){perror("bind server: "),exit(-1);}printf("bind success.\n");//4. wait for client connect.res = listen(sockfd, 100);if(-1 == res){perror("listen server: "),exit(-1);}printf("Server is initialized.\n");}static void start_server(void){printf("Starting server...\n");sleep(1);printf("Server is ready.\n");while(1){//1.准备一个新地址用来接受客户端的地址struct sockaddr_in acvAddr;socklen_t len = sizeof(acvAddr);//2.响应客户端的请求，然后开辟新线程去响应客户端,为了腾出主线程继续响应新的客户端int resfd = accept(sockfd,(struct sockaddr*)&acvAddr,&len);if(-1 == resfd){perror("accept server"),exit(-1);}//3.开辟新线程去处理客户端的消息pthread_t tid;pthread_create(&tid, 0, rcvMsg, &resfd);}}static void destroy_server(int signo){printf("Server closing...\n");close(sockfd);sleep(1);exit(0);}static void* rcvMsg(void* p){char buf[100] = {};char temp[50] = {};char name[30] = {};char *bp = buf;int len = 0;int login_flag = 1;int tempFd = *(int*)p;while(1){//不停地接受来自客户端的消息memset(buf, 0, sizeof(buf));int ret = recv(tempFd, buf, sizeof(buf), 0);if(login_flag) {// 取出冒号前面的网名while(*bp++ != ':') {len++;}strncpy(temp, buf, len);temp[len] = '\0';strcpy(name, temp);// 通知大家有人上线strcat(temp , " login.\n");sendMsgToAll(temp);// 首次登入计入在线链表insert_tail_linklist(slist, tempFd);login_flag = 0;bp = NULL;}if(ret < 0) {perror("recv server");} else if(0 == ret) {printf("client %d exit.\n", tempFd);strcat(name, " log out.\n");remove_data_linklist(slist, tempFd);sendMsgToAll(name);break;}//把接受到的消息发送给所有客户端sendMsgToAll(buf);}pthread_exit(NULL);}static void sendMsgToAll(void* msg){char* datap = (char*)msg;socklist* temp = slist->next;while(temp) {send(temp->data, datap, strlen(datap), 0);temp = temp->next;}}