#include #include #include #include #include #include #include #include #include #include #define LISTENQ 1000 //最大連接數 #define MAXLINE 512 //發送消息的最長字節 #define MAXMEM 1000 #define NAMELEN 20 //名字長度 struct socketcfd { char name[20]; int fd; }; int listenfd; //分別記錄服務器端的套接字與連接的多個客戶端的套接字 struct socketcfd connfd[MAXMEM]; void rcv_snd(void *arg) //服務器接受并轉發消息函數 { char* ask="請輸入你的名字："; char* ask1="使用說明：\n1、群聊可以直接發送消息，\n2、私聊請先輸入@+人名+空格+內容，\n3、退出直接輸入“bye”\n-------------------------------------------"; char* ch="此用戶不存在"; char* ch1="群聊"; char* ch2="私聊"; char buff[MAXLINE]; //用戶姓名 char buff1[MAXLINE]; //聊天內容（包括私聊和群聊，然后對它進行分析） char buff2[MAXLINE]; //發送消息的時間 char buff3[MAXLINE]; //私聊時接受用戶的姓名 char buff4[MAXLINE]; //私聊時發送的內容 time_t ticks; int i=0; int retval; int len; int k=0,j=0,m=0; int p; int n=(int *)arg; //獲取此進程對應的套接字用戶的名字 write(connfd[n].fd,ask,strlen(ask)); len=read(connfd[n].fd,connfd[n].name,NAMELEN); if(len>0){ connfd[n].name[len]=0; } printf("用戶姓名：%s\n",connfd[n].name); //把當前用戶的加入告知所有用戶 strcpy(buff,connfd[n].name); strcat(buff,"\t加入聊天"); for(i=0;i0) { buff1[len]=0; //當前用戶的輸入信息為“bye”時，當前用戶退出 if(strcmp("bye",buff1)==0){ printf("%s用戶已退出\n",connfd[n].name); close(connfd[n].fd); connfd[n].fd=-1; pthread_exit(&retval); } //通過分析發送的信息判斷是群聊還是私聊 if(buff1[0]=='@'){ j=0;m=0; for(k=1;kp){ buff4[m]=buff1[i]; m++; } else if(iMAXMEM){ write(connfd[n].fd,ch,strlen(ch)); } } }else{ ticks=time(NULL); sprintf(buff2,"%.24s\r\n",ctime(&ticks)); strcpy(buff,buff2); strcat(buff,ch1); strcat(buff,"\t"); strcat(buff,connfd[n].name); strcat(buff,"\n"); strcat(buff,buff1); for(i=0;i

#include #include #include #include #include #include #include #include #include #include #define MAXLINE 512 #define NAMELEN 20 #define PORT 8080 int sockfd; //發送消息的函數snd void snd() { char name[NAMELEN]; char buff[MAXLINE]; gets(name); write(sockfd,name,strlen(name)); while(1) { gets(buff); write(sockfd,buff,strlen(buff)); if(strcmp("bye",buff)==0) { exit(0); } } } int main(int argc,char **argv) { pthread_t thread; struct sockaddr_in servaddr; if(argc != 2) { printf("usage: echo ip\n"); return -1; } //調用socket函數創建客戶端的套接字 sockfd=socket(AF_INET,SOCK_STREAM,0); if(sockfd<0) { printf("Socket create failed\n"); return -1; } //初始化服務器端的地址 servaddr.sin_family=AF_INET; servaddr.sin_port=htons(PORT); if(inet_aton(argv[1],&servaddr.sin_addr)<0) { printf("inet_aton error.\n"); return -1; } //調用connect函數實現與服務器端建立連接 printf("Connecting...\n"); if(connect(sockfd,(struct sockaddr*)&servaddr,sizeof(servaddr))<0) { printf("Connect server failed.\n"); return -1; } //從此處開始程序分做兩個線程 //創建發送消息的線程，調用了發送消息的函數snd pthread_create(&thread,NULL,(void*)(&snd),NULL); //從此處開始向下為接收消息的線程 char buff[MAXLINE]; int len; while(1) { if((len=read(sockfd,buff,MAXLINE))>0) { buff[len]=0; printf("\n%s\n\n",buff); } } return 0; }