UDP-QQ聊天程序

 

#include<stdio.h>
#include<stdlib.h>
#include<sys/socket.h>
#include<arpa/inet.h>
#include<netinet/in.h>
#include<string.h>
#include<unistd.h>
#include<pthread.h>

void *put_handler(void *arg)
{
 int fd=(int)arg,flags=0;
 unsigned short port=8000;
 char buf[200]="";
 char chat_ip[16]="";
 struct sockaddr_in chat_addr;

 bzero(&chat_addr,sizeof(chat_addr));

 chat_addr.sin_family = AF_INET;
 chat_addr.sin_port = htons (port);
 //fd=socket(AF_INET,SOCK_DGRAM,0);

 while(1)
 {
  printf("UDP_QQ:");
  bzero(buf,sizeof(buf));
  fgets(buf,sizeof(buf),stdin);
  if(strncmp(buf,"sayto",5) == 0)
  {
   sscanf(buf,"sayto %s",chat_ip);
   inet_pton(AF_INET,chat_ip,&chat_addr.sin_addr);
   //sendto(fd,buf,strlen(buf),0,(struct sockaddr*)&chat_addr,sizeof(chat_addr));
   flags=1;
  }

  if(strstr(buf,"exit") !=0 )
   break;

  if((flags == 1) && !strstr(buf,"sayto"))
  sendto(fd,buf,strlen(buf),0,(struct sockaddr*)&chat_addr,sizeof(chat_addr));
 }
 close(fd);
 exit(1);
}
int main(void)
{
 int server_fd;
 pthread_t tid_put;
 unsigned short port=8000;
 struct sockaddr_in srv_addr;

 bzero(&srv_addr,sizeof(srv_addr));

 srv_addr.sin_family = AF_INET;
 srv_addr.sin_port = htons (port);
 srv_addr.sin_addr.s_addr = htonl(INADDR_ANY);

 server_fd=socket(AF_INET,SOCK_DGRAM,0);
 bind(server_fd,(struct sockaddr*)&srv_addr,sizeof(srv_addr));

 //pthread_create(&tid_put,NULL,put_handler,NULL);
 pthread_create(&tid_put,NULL,put_handler,(void *)server_fd);

 while(1)
 {
  char sendbuf[200]="";
  struct sockaddr_in cli_addr;
  char ip[16]="";
  size_t len=0;
  socklen_t addrlen = sizeof(cli_addr);
  len=recvfrom(server_fd,sendbuf,sizeof(sendbuf),0,(struct sockaddr*)&cli_addr,&addrlen);
  inet_ntop(AF_INET,&cli_addr.sin_addr,ip,addrlen);
  printf("\r[%s]:%s",ip,sendbuf);
  fflush(stdout);
  printf("UDP_QQ:");
  fflush(stdout);
  //sendto(server_fd,sendbuf,len,0,(struct sockaddr*)&cli_addr,sizeof(cli_addr));
 }
 close(server_fd);
}


/******************Select 方法**********************/

#include<stdio.h>
#include<stdlib.h>
#include<sys/socket.h>
#include<arpa/inet.h>
#include<netinet/in.h>
#include<string.h>
#include<unistd.h>
#include<pthread.h>
#include<sys/select.h>
int main(void)
{
 int server_fd,flags=0;
 fd_set rset;
 unsigned short port=8000;
 struct sockaddr_in srv_addr;

 bzero(&srv_addr,sizeof(srv_addr));

 srv_addr.sin_family = AF_INET;
 srv_addr.sin_port = htons (port);
 srv_addr.sin_addr.s_addr = htonl(INADDR_ANY);

 server_fd=socket(AF_INET,SOCK_DGRAM,0);
 bind(server_fd,(struct sockaddr*)&srv_addr,sizeof(srv_addr));

 while(1)
 {
  char sendbuf[200]="";
  struct sockaddr_in cli_addr;
  char ip[16]="";
  char chat_ip[16]="";
  size_t len=0;
  socklen_t addrlen = sizeof(cli_addr);

  FD_ZERO(&rset);
  FD_SET(0,&rset);
  FD_SET(server_fd,&rset);

  if(select(server_fd+1,&rset,NULL,NULL,NULL) > 0)
  {
   if(FD_ISSET(server_fd,&rset))
   {
    len=recvfrom(server_fd,sendbuf,sizeof(sendbuf),0,(struct sockaddr*)&cli_addr,&addrlen);
    inet_ntop(AF_INET,&cli_addr.sin_addr,ip,addrlen);
    printf("\r[%s]:%s",ip,sendbuf);
    fflush(stdout);
    printf("UDP_QQ:");
    fflush(stdout);
   }
   if(FD_ISSET(0,&rset))
   {
    struct sockaddr_in chat_addr;

    chat_addr.sin_family = AF_INET;
    chat_addr.sin_port = htons (port);

    printf("UDP_QQ:");
    fflush(stdout);
    bzero(sendbuf,sizeof(sendbuf));

    fgets(sendbuf,sizeof(sendbuf),stdin);

    if(strncmp(sendbuf,"sayto",5) == 0)
    {
     sscanf(sendbuf,"sayto %s",chat_ip);
     inet_pton(AF_INET,chat_ip,&chat_addr.sin_addr);
     flags=1;
    }

    if(strstr(sendbuf,"exit") !=0 )
     break;

    if((flags == 1) && !strstr(sendbuf,"sayto"))
     sendto(server_fd,sendbuf,strlen(sendbuf),0,(struct sockaddr*)&chat_addr,sizeof(chat_addr));
   }
  }
 }
 close(server_fd);
 return 0;
}