学习笔记：UDP用户数据报协议(1)(2)

4、UDP服务器的设计与实现

/*********************************************************************************  
  *Author     :  wph  
  *Version    :  1.0  
  *Date       :  2014/03/01    
  *Description:  udp server   
  *Others     :   
  *History    :     
**********************************************************************************/   
#include<stdio.h>     
#include<string.h>     
#include<unistd.h>     
#include<sys/types.h>     
#include<sys/socket.h>     
#include<stdlib.h>     
#include<netinet/in.h>     
#include<arpa/inet.h>   
#include<event2/event.h>   
   
#include "errocode.h"   
#include "basetype.h"   
   
#define INVALID_FD      -1   
#define PORT            1234     
#define MAXDATASIZE     512    
   
STATIC INT g_iudpFd = INVALID_FD;   
   
VOID udp_callback(evutil_socket_t fd, short what, void *arg)   
{   
    struct sockaddr_in client;     
    socklen_t addrlen;     
    int num;     
    char buf[MAXDATASIZE];    
       
    memset(buf, 0, MAXDATASIZE);   
    num = recvfrom(fd, buf, MAXDATASIZE, 0, (struct sockaddr*)&client, &addrlen);                                        
    if (num < 0)     
    {     
        perror("recvfrom() error\n");     
        exit(1);     
    }    
       
    printf("You got a message (%s) from client.\nIt's ip is%s, port is %d.\n",    
            buf, inet_ntoa(client.sin_addr), htons(client.sin_port));    
       
    sendto(fd, buf, num, 0, (struct sockaddr *)&client, addrlen);     
}   
   
ULONG udp_init(VOID)   
{   
    int sockfd;     
    struct sockaddr_in server;     
       
    /* The caller has already set up fd1, fd2 somehow, and make them  
       nonblocking. */   
    sockfd = socket(AF_INET, SOCK_DGRAM, 0);   
    if(INVALID_FD == sockfd)      
    {     
        perror("Creatingsocket failed.");     
        exit(1);     
    }    
       
    bzero(&server, sizeof(server));     
    server.sin_family = AF_INET;     
    server.sin_port= htons(PORT);     
    server.sin_addr.s_addr= htonl(INADDR_ANY);     
    if(-1 == bind(sockfd, (struct sockaddr *)&server, sizeof(server)))     
    {     
        perror("Bind()error.");     
        exit(1);     
    }   
       
    g_iudpFd = sockfd;   
       
    return EROOR_SUCCESS;   
}   
   
VOID udp_fini(VOID)   
{   
    int sockfd = g_iudpFd;     
   
    if (INVALID_FD != sockfd)   
    {      
        close(sockfd);   
    }   
}   
VOID main_loop(VOID)   
{   
    INT ifd = g_iudpFd;   
        struct event *ev1;   
    struct timeval five_seconds = {5,0};   
    struct event_base *base = event_base_new();   
   
    /* upd 服务器采用libevet进行处理 */   
    ev1 = event_new(base, ifd, EV_TIMEOUT|EV_READ|EV_PERSIST, udp_callback, NULL);   
   
    event_add(ev1, &five_seconds);   
    event_base_dispatch(base);   
       
    return ;   
}   
INT main()   
{   
    if(EROOR_SUCCESS != udp_init())   
    {   
        return -1;   
    }   
       
    main_loop();   
   
    udp_fini();   
       
    return 0;   
}   

/*********************************************************************************  
  *Copyright(C),2010-2011,  
  *Author     :  wph  
  *Version    :  1.0  
  *Date       :  2014/03/01    
  *Description:  udp client   
  *Others     :   
  *History    :     
**********************************************************************************/   
#include <stdio.h>     
#include <stdlib.h>     
#include <unistd.h>     
#include <string.h>     
#include <sys/types.h>     
#include <sys/socket.h>     
#include <netinet/in.h>     
#include <netdb.h>     
   
#include "errocode.h"   
#include "basetype.h"   
   
#define INVALID_FD      -1   
#define PORT            1234     
#define MAXDATASIZE     512    
    
INT main(INT argc, CHAR *argv[])     
{     
    INT  isockfd = INVALID_FD;   
    UINT uinum = 0;     
    char buf[MAXDATASIZE];   
    socklen_t  addrlen;     
    struct hostent *he;     
    struct sockaddr_in server;   
    struct sockaddr_in peer;     
   
    if (3 != argc)     
    {     
        printf("Usage: %s <IP Address><message>\n", argv[0]);     
        exit(1);     
    }     
   
    if (NULL == (he=gethostbyname(argv[1])))     
    {     
        printf("gethostbyname()error\n");     
        exit(1);     
    }     
   
    if (INVALID_FD == (isockfd = socket(AF_INET, SOCK_DGRAM,0)))     
    {     
        printf("socket() error\n");     
        exit(1);     
    }     
   
    bzero(&server,sizeof(server));     
    server.sin_family = AF_INET;     
    server.sin_port = htons(PORT);     
    server.sin_addr= *((struct in_addr *)he->h_addr);     
    sendto(isockfd, argv[2],strlen(argv[2]),0,(struct sockaddr *)&server,sizeof(server));     
    addrlen=sizeof(server);     
    while (1)     
    {     
        if((uinum=recvfrom(isockfd, buf, MAXDATASIZE, 0, (struct sockaddr *)&peer, &addrlen))== -1)     
        {     
            printf("recvfrom() error\n");     
            exit(1);     
        }     
        if (addrlen != sizeof(server) || memcmp((const void *)&server, (const void *)&peer, addrlen) != 0)     
        {     
            printf("Receive message from otherserver.\n");     
            continue;     
        }     
   
        buf[uinum]='\0';     
        printf("Server Message:%s\n", buf);     
        break;     
   }     
   
 close(isockfd);     
}     

演示：

[root@localhost server]# ./udpserver &

[1] 1389

[root@localhost server]# ./udpclient 127.0.0.1 "i like you"

You got a message (i like you) from client.

It's ip is127.0.0.1, port is 34340.

Server Message:i like you

[root@localhost server]# netstat -aun

Active Internet connections (servers and established)

---

---