多线程、全双工、UDP、对话源代码

多线程、全双工、UDP、对话源代码

服务端： /*头文件*/ #include <netinet/in.h> #include <sys/socket.h> #include <stdlib.h> #include <string.h> #include <sys/types.h> #include <stdio.h> #include <arpa/inet.h> #include <unistd.h> #include <pthread.h> /*宏定义*/ #define BUFFER_SIZE 1024 #define PORT 4001 /*函数声明*/ void *function(void *arg); /*结构体定义*/ struct ARG { int sockfd; int len; }; struct sockaddr_in client; //定义全局结构体变量 /*主函数*/ main() { /*定义变量*/ pthread_t tid; struct ARG *arg; int sockfd,len; char se_mes[BUFFER_SIZE]; struct sockaddr_in server; /*调用套接字函数*/ sockfd=socket(AF_INET,SOCK_DGRAM,0); if(sockfd==-1) { perror("socket() error"); exit(0); } //printf("Socket success!/n"); /*初始化server套接字地址结构，并对地址结构中的成员变量赋值*/ bzero(&server,sizeof(server)); server.sin_family=AF_INET; server.sin_port=htons(PORT); server.sin_addr.s_addr=htonl(INADDR_ANY); if(bind(sockfd,(struct sockaddr *)&server,sizeof(server))==-1) { perror("bind() error"); exit(0); } //printf("Bind success!/n"); printf("等待客户端连接....../n"); len=sizeof(client); arg=(struct ARG *)malloc(sizeof(struct ARG)); //给结构体分配内存空间 /*给结构体成员变量赋值*/ arg->sockfd=sockfd; arg->len=len; /*产生子线程*/ if(pthread_create(&tid,NULL,function,(void *)arg)) { perror("pthread_create() error"); exit(0); } /*发送消息给客户端*/ while(1) { bzero(se_mes,sizeof(se_mes)); gets(se_mes); sendto(sockfd,se_mes,BUFFER_SIZE,0,(struct sockaddr *)&client,sizeof(client)); } /*关闭套接字*/ close(sockfd); } /*自定义函数*/ void * function(void *arg) { /*定义变量*/ int sockfd,recvbytes,len; char re_mes[BUFFER_SIZE]; struct ARG *info; info=(struct ARG *)arg; sockfd=info->sockfd; len=info->len; /*接收客户端发来的消息*/ bzero(re_mes,sizeof(re_mes)); recvbytes=recvfrom(sockfd,re_mes,BUFFER_SIZE,0,(struct sockaddr *)&client,&len); if(recvbytes==-1) { perror("recvfrom() error"); exit(0); } printf("You got a message from client./nIt's ip is %s,port is %d./n",inet_ntoa(client.sin_addr),ntohs(client.sin_port)); //显示客户端信息 printf("client: %s/n",re_mes); sendto(sockfd,"欢迎你",9,0,(struct sockaddr *)&client,len); while(1) { bzero(re_mes,sizeof(re_mes)); recvbytes=recvfrom(sockfd,re_mes,BUFFER_SIZE,0,(struct sockaddr *)&client,&len); if(recvbytes==-1) { perror("recvfrom() error"); exit(0); } printf("客户端消息: %s/n",re_mes); if(!strcmp(re_mes,"bye")) { sendto(sockfd,"byebye",6,0,(struct sockaddr *)&client,len); break; } if(!strcmp(re_mes,"byebye")) { break; } } free(arg); //释放内存空间 close(sockfd); exit(0); } 客户端： /*头文件*/ #include <netinet/in.h> #include <netdb.h> #include <sys/socket.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <sys/types.h> #include <unistd.h> #include <pthread.h> /*宏定义*/ #define BUFFER_SIZE 1024 #define PORT 4001 /*结构体定义*/ struct ARG { int sockfd; }; /*函数声明*/ void *function(void *arg); /*主函数*/ main(int argc,char *argv[]) { /*定义变量*/ pthread_t tid; struct ARG *arg; int sockfd,len; char se_mes[BUFFER_SIZE]; struct sockaddr_in server; struct hostent *he; /*判断命令行参数*/ if(argc!=2) { printf("Usage: %s <IP Address>/n",argv[0]); exit(0); } he=gethostbyname(argv[1]); if(he==NULL) { printf("gethostbyname() error"); exit(0); } /*调用套接字函数*/ sockfd=socket(AF_INET,SOCK_DGRAM,0); if(sockfd==-1) { perror("socket() error"); exit(0); } /*初始化server的地址结构，并为地址结构的成员赋值*/ bzero(&server,sizeof(server)); server.sin_family=AF_INET; server.sin_port=htons(PORT); server.sin_addr=*((struct in_addr *)he->h_addr); sendto(sockfd,"我来了",9,0,(struct sockaddr *)&server,sizeof(server)); arg=(struct ARG *)malloc(sizeof(struct ARG)); //给结构体分配内存空间 arg->sockfd=sockfd; /*给结构体成员变量赋值*/ /*产生子线程*/ if(pthread_create(&tid,NULL,function,(void *)arg)) { perror("pthread_create() error"); exit(0); } /*发送消息给服务器*/ while(1) { bzero(se_mes,sizeof(se_mes)); gets(se_mes); sendto(sockfd,se_mes,BUFFER_SIZE,0,(struct sockaddr *)&server,sizeof(server)); } /*关闭套接字*/ close(sockfd); } /*自定义函数*/ void * function(void *arg) { int sockfd,recvbytes,len; struct sockaddr_in peer; char re_mes[BUFFER_SIZE]; struct ARG *info; info=(struct ARG *)arg; sockfd=info->sockfd; len=sizeof(peer); /*接收服务器发来的消息*/ while(1) { bzero(re_mes,sizeof(re_mes)); recvbytes=recvfrom(sockfd,re_mes,BUFFER_SIZE,0,(struct sockaddr *)&peer,&len); if(recvbytes==-1) { perror("recv() error"); exit(0); } printf("服务器消息: %s/n",re_mes); if(!strcmp(re_mes,"byebye")) { break; } if(!strcmp(re_mes,"bye")) { sendto(sockfd,"byebye",6,0,(struct sockaddr *)&peer,sizeof(peer)); break; } } free(arg); //释放内存空间 close(sockfd); exit(0); }