下用c語(yǔ)言編寫(xiě)的聊天室程序服務(wù)器和客戶(hù)端

1、/* * server.c * * Created on: 2012-6-15 * Author: root */#include #include #include #include #include #include #include #include #include#define PORT 7999#define MAX_NUM 3 /client連接最大個(gè)數(shù)#define MAX_CLIENT 15#define MAX_SIZE 1024pthread_rwlock_t idx_lock, wait_lock;/client 信息typedef struct _client int

2、 sockfd;char name20;pthread_t pid;int flg; c_client;c_client clientMAX_CLIENT;/定義client;/等待的clientstruct _client_ int sockfd;char name20;pthread_t pid;struct _client_ *next;typedef struct _client_ c_client_c;c_client_c *head = NULL;c_client_c *temp_c1 = NULL, *temp_c2 = NULL;/等待的/初始化client信息void ini

3、t_client() int i = 0;for (i = 0; i MAX_CLIENT; i+) clienti.sockfd = -1;memset(, 0, 20);clienti.pid = -1;clienti.flg = -1;/查找結(jié)構(gòu)體數(shù)組中sockfd為-1的下標(biāo)值int find_fd(c_client *client) int i = 0;while (i sockfd & i MAX_NUM) if (strcmp(, p) = 0)return 0;i+;return 1;/SHOWvoid showuser(c_cl

4、ient *p_client) int i = 0;char buf1024 = 0 ;strcpy(buf, 200rn);for (i = 0; i sockfd, buf, strlen(buf), 0);/ALLvoid sendto_all(c_client *p_client, char *buf) int i = 0;char sendbuf1024 = 0 ;sprintf(sendbuf, AFROMrn%srn%s, p_client-name, buf + 5);for (i = 0; i MAX_NUM; i+) if (clienti.sockfd != -1 & c

5、lienti.flg != -1)if(send(clienti.sockfd, sendbuf, strlen(sendbuf), 0) = 0)printf(send errrrrrn);exit(1);int findname(char *name) int i = 0;for (i = 0; i sockfd, sendbuf, strlen(sendbuf), 0); else sprintf(sendbuf, FROMrn%srn%s, p_client-name, buf + 4 + strlen(name) + 2);if(send(sock, sendbuf, strlen(

6、sendbuf), 0)name);close(p_client-sockfd);p_client-sockfd = -1;p_client-pid = -1;p_client-flg = -1;sprintf(buf,NOTICE1rn%s退出聊天室rnrn,p_client-name);memset(p_client-name,0,20);for(i=0;inext != NULL)memset(buf,0,1024);pthread_rwlock_rdlock(&idx_lock);idx = find_fd(client);pthread_rwlock_unlock(&idx_lock

7、);clientidx.sockfd = head-next-sockfd;pthread_rwlock_wrlock(&wait_lock);temp = head-next;head-next = head-next-next;free(temp);pthread_rwlock_unlock(&wait_lock);sprintf(buf,NOTICErn您已被喚醒,請(qǐng)繼續(xù)操作rnrn);send(clientidx.sockfd,buf,strlen(buf),0);if (pthread_create(&clientidx.pid, NULL, (void *)pthread_fun,

8、(void *) &clientidx) != 0) perror(pthread_create);exit(1);pthread_detach(clientidx.pid);void pthread_fun(void* cclient) c_client *p_client = (c_client *) cclient;char bufMAX_SIZE = 0 ;char sendbuf1024 = 0 ;int i, n;char *p;sprintf(sendbuf, %s, NOTICErn通訊通道開(kāi)啟rnrn);if (send(p_client-sockfd, sendbuf, s

9、trlen(sendbuf), 0) sockfd, buf, sizeof(buf) - 1, MSG_NOSIGNAL);if (n sockfd);p_client-sockfd = -1;break;if (logform(buf) if (cmpname(buf, p_client) = 0) send(p_client-sockfd, ERRORrn用戶(hù)名重復(fù)rnrn, 26, 0);continue; else p_client-flg = 1;p = strtok(buf + 7, rnrn);strcpy(p_client-name, p);sprintf(sendbuf,

10、100rn%srnrn, p_client-name);send(p_client-sockfd, sendbuf, sizeof(sendbuf), 0);printf(%s進(jìn)入聊天室n, p_client-name);for (i = 0; i sockfd & clienti.flg != -1)send(clienti.sockfd, sendbuf, sizeof(sendbuf), 0);memset(sendbuf, 0, 1024);while (1) memset(buf, 0, MAX_SIZE);if (n = recv(p_client-sockfd, buf, MAX

11、_SIZE, 0) sockfd, ERRORrn信息不符合協(xié)議要求rnrn,38, 0); else send(p_client-sockfd, ERRORrn未登錄,請(qǐng)您登錄再進(jìn)行其他操作rnrn, 56, 0);pthread_exit(NULL);int main() int ser_sockfd, clt_sockfd;struct sockaddr_in addr;int idx;char buf1024 = 0 ;/pthread_rwlock_t idx_lock,wait_lock;pthread_rwlock_init(&idx_lock, NULL);pthread_rw

12、lock_init(&wait_lock, NULL);init_client();/創(chuàng)建服務(wù)器sockfdif (ser_sockfd = socket(AF_INET, SOCK_STREAM, 0) = -1) perror(socket);exit(1);/設(shè)置服務(wù)器網(wǎng)絡(luò)地址bzero(&addr, sizeof(addr);addr.sin_family = AF_INET;addr.sin_port = htons(PORT);addr.sin_addr.s_addr = htonl(INADDR_ANY);/設(shè)置端口可重用int opt = 1;setsockopt(ser_so

13、ckfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt);/將套接字綁定到服務(wù)器的網(wǎng)絡(luò)地址上if (bind(ser_sockfd, (struct sockaddr*)&addr, sizeof(addr) = -1) perror(bind);exit(1);printf(bind successn);/監(jiān)聽(tīng)連接請(qǐng)求-監(jiān)聽(tīng)隊(duì)列長(zhǎng)度為10if (listen(ser_sockfd, 10) = -1) perror(listen);exit(1);printf(listen successn);while (1) if (clt_sockfd = a

14、ccept(ser_sockfd, NULL, NULL) = -1) perror(accept);exit(1);pthread_rwlock_rdlock(&idx_lock);idx = find_fd(client);/printf(idx=%dn,idx);pthread_rwlock_unlock(&idx_lock);if (idx != -1) /連接末滿(mǎn)clientidx.sockfd = clt_sockfd;if (pthread_create(&clientidx.pid, NULL,(void *) pthread_fun,(void *)&clientidx) !

15、= 0) perror(pthread_create);exit(1);pthread_detach(clientidx.pid); else /連接已滿(mǎn),等待temp_c1 = (c_client_c *) malloc(sizeof(c_client_c);temp_c1-sockfd = clt_sockfd;temp_c1-next = NULL;pthread_rwlock_wrlock(&wait_lock);if (head = NULL) head = (c_client_c *) malloc(sizeof(c_client_c);head-next = temp_c1; e

16、lse for (temp_c2 = head; temp_c2-next != NULL; temp_c2= temp_c2-next);temp_c2-next = temp_c1;pthread_rwlock_unlock(&wait_lock);memset(buf, 0, 1024);sprintf(buf, NOTICErn服務(wù)器已滿(mǎn),請(qǐng)等候rnrn);/客戶(hù)端接受 則等待if (send(temp_c1-sockfd, buf, strlen(buf), 0) = 0) printf(sendr errn);/* * client.c * * Created on: 2012-6

17、-18 * Author: root */#include #include #include #include #include #include #include #include #include #include #include #define MAX_SIZE 1024#define PORT 7999static int FLAGE = -1;char name20 = 0;void fun_show(int sockfd)char sendbuf256 = 0;sprintf(sendbuf,SHOWrnrn);if(send(sockfd,sendbuf,strlen(sen

18、dbuf),0)=0)printf(send errn);close(sockfd);exit(1);void fun_all(int sockfd)char sendbufMAX_SIZE = 0;sprintf(sendbuf,ALLrn,5);printf(輸入發(fā)送的內(nèi)容:n);scanf(%s,sendbuf+5);sprintf(sendbuf+strlen(sendbuf),rnrn);if(send(sockfd,sendbuf,strlen(sendbuf),0) = 0)printf(send errn);close(sockfd);exit(1);void fun_one(

19、int sockfd)char sendbufMAX_SIZE = 0;char name320 = 0;printf(輸入聊天對(duì)象:);scanf(%s,name3);sprintf(sendbuf,TOrn%srn,name3);printf(輸入聊天內(nèi)容:n);scanf(%s,sendbuf+strlen(sendbuf);sprintf(sendbuf+strlen(sendbuf),rnrn);if(send(sockfd,sendbuf,strlen(sendbuf),0) = 0)printf(send errn);close(sockfd);exit(1);void fun_

20、quit(int sockfd)char sendbuf256 = QUITrnrn;if(send(sockfd,sendbuf,strlen(sendbuf),0) = 0)printf(send errn);close(sockfd);exit(1);void *pthread_fun(int *sock)int sockfd = *sock;char recvbuf1024 = 0;int n = 0;char *p,*q;char name220 = 0;while(1)memset(recvbuf,0,1024);n = recv(sockfd,recvbuf,sizeof(rec

21、vbuf),0);if(n=0)printf(recv failedn);exit(1);if(!strncmp(recvbuf,NOTICErn通訊通道開(kāi)啟rnrn,30)printf(通訊通道開(kāi)啟n);FLAGE=1;if (!strncmp(recvbuf, 100rn, 5) char *p = strtok(recvbuf + 5, rnrn);/100rn：3個(gè)字符后取出rnrn前的名字strcpy(name2, p);printf(NOTICE%s進(jìn)入聊天室n, name2);FLAGE = 4;if (!strncmp(recvbuf, ERRORrn用戶(hù)名重復(fù)rnrn, 26

22、) printf(用戶(hù)名重復(fù)n);FLAGE = 3;if (!strncmp(recvbuf, 200rn, 5) p = strtok(recvbuf+5,rnrn);while(p != NULL)printf(%sn,p);p = strtok(NULL,rnrn);printf(please input con:n);if(!strncmp(recvbuf,AFROMrn,7)/printf(recvbuf=%sn,recvbuf);p = strtok(recvbuf+6,rn);q = strtok(NULL,rnrn);printf(%s)群聊:%sn,p, q);printf

23、(please input con:n);if(!strncmp(recvbuf,FROMrn,6)p = strtok(recvbuf+6,rn);q = strtok(NULL,rnrn);printf(%s)私聊(%s):%sn,p,name,q);printf(please input con:n);if(!strncmp(recvbuf,ERROR2rn,8)p = strtok(recvbuf + 8, rn);printf(%sn, p);if(!strncmp(recvbuf,NOTICE1rn,9)p = strtok(recvbuf + 9, rn);printf(用戶(hù)%s

24、n, p);if(!strncmp(recvbuf,NOTICErn服務(wù)器已滿(mǎn),請(qǐng)等候rnrn,37)printf(服務(wù)器已滿(mǎn),請(qǐng)等候n);FLAGE = 0;if(!strncmp(recvbuf,NOTICErn您已被喚醒,請(qǐng)繼續(xù)操作rnrn,45)printf(你已經(jīng)被喚醒,請(qǐng)繼續(xù)操作n);FLAGE = 2;int main(int argc,char *argv)if(argc != 2)printf(input server ip:n);exit(1);pthread_t pid;int sockfd;struct sockaddr_in addr;char recvbuf1024

25、 = 0;char sendbuf1024 = 0;int k=0;int n;char str6 = 0;char *p,*q;if(sockfd = socket(AF_INET,SOCK_STREAM,0) = -1)perror(socket);exit(1);bzero(&addr,sizeof(addr);addr.sin_family = AF_INET;addr.sin_port = htons(PORT);if(inet_pton(AF_INET,argv1,(void *)&addr.sin_addr) 0)usleep(100);/讓子進(jìn)程先運(yùn)行while(FLAGE=0)printf(服務(wù)器已滿(mǎn),wait.n);sleep(1);tcflush(0,TCIFLUSH);while(FLAGE=2) /當(dāng)FLAGE2時(shí)正好父進(jìn)程運(yùn)行，則等