udp回射服務器

UDP API

#include <sys/socket.h>
ssize_t recvfrom(int sockfd,void *buff,size_t nbytes,int flags,struct sockaddr *from,socklen_t *addrlen);
ssize_t sendto(int sockfd,const void *buff,size_t nbytes,int flags,const struct sockaddr *to,socklen_t *addrlen);

recvfrom 的最後兩個參數代表數據包發送者的地址結構和該地址結構的字節數。
sendto 的最後兩個參數代表數據包接收者的地址結構和地址結構的字節數。

udp 回射服務器第一版

server端

文件名:server.cpp
作用：udp 服務器的第一版

#include <sys/socket.h>
#include <sys/socket.h>
#include <strings.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <unistd.h>
#include <err.h>
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <unistd.h>
#include<signal.h>
#include<wait.h>

#define MAXLINE 4096
#define SERV_PORT 9877

void dg_echo(int sockfd,struct sockaddr *pcliaddr,socklen_t clilen)
{
    int n;
    socklen_t len;
    char mesg[MAXLINE];
    for(;;)
    {
        len = clilen;
        n = recvfrom(sockfd,mesg,MAXLINE,0,pcliaddr,&len);
        sendto(sockfd,mesg,n,0,pcliaddr,len);
    }
}



int main(int argc,char **argv)
{
    int sockfd;
    struct sockaddr_in servaddr,cliaddr;
    sockfd = socket(AF_INET,SOCK_DGRAM,0);
    bzero(&servaddr,sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servaddr.sin_port = htons(SERV_PORT);
    bind(sockfd,(struct sockaddr *) &servaddr,sizeof(servaddr));
    dg_echo(sockfd,(struct sockaddr *)&cliaddr,sizeof(cliaddr));

}

client端

文件名:client.cpp
作用：udp 客戶端的第一版

#include <sys/socket.h>
#include <string.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <unistd.h>
#include <err.h>
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <unistd.h>
#include<signal.h>
#include<wait.h>
#include<arpa/inet.h>


#define MAXLINE 4096
#define SERV_PORT 9877

void dg_cli(FILE *fp,int sockfd,const struct sockaddr* pservaddr,socklen_t servlen)
{
    int n;
    char sendline[MAXLINE],recvline[MAXLINE+1];
    while(fgets(sendline,MAXLINE,fp)!=NULL)
    {
        sendto(sockfd,sendline,strlen(sendline),0,pservaddr,servlen);
        n = recvfrom(sockfd,recvline,MAXLINE,0,NULL,NULL);
        recvline[n] = 0;
        fputs(recvline,stdout);
    }
}


int main(int argc,char **argv)
{
    int sockfd;
    struct sockaddr_in servaddr;
    if(argc != 2 )
    {
        printf("usage:udpcli <ipaddr>");
        exit(1);
    }
    bzero(&servaddr,sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(SERV_PORT);
    inet_pton(AF_INET,argv[1],&servaddr.sin_addr);
    sockfd = socket(AF_INET,SOCK_DGRAM,0);
    dg_cli(stdin,sockfd,(struct sockaddr *)&servaddr,sizeof(servaddr));
    exit(0);   
}

第一版中所存在的問題

在第一版中，任何知道客戶臨時端口號的其他進程都可以向客戶端發送數據包，然而客戶端指向受到服務器端的數據包。這就意味着，我們需要對數據包做一層過濾，只保留服務器端發送的數據包，忽略其他數據報。

文件名:client02.cpp
作用：udp 客戶端的第二版

#include <sys/socket.h>
#include <string.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <unistd.h>
#include <err.h>
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <unistd.h>
#include<signal.h>
#include<wait.h>
#include<arpa/inet.h>


#define MAXLINE 4096
#define SERV_PORT 9877

void dg_cli(FILE *fp,int sockfd,const struct sockaddr* pservaddr,socklen_t servlen)
{
    int n;
    socklen_t len;
    struct sockaddr *preply_addr;
    malloc(servlen);
    char sendline[MAXLINE],recvline[MAXLINE+1];
    while(fgets(sendline,MAXLINE,fp)!=NULL)
    {
        sendto(sockfd,sendline,strlen(sendline),0,pservaddr,servlen);
        len = servlen;
        n = recvfrom(sockfd,recvline,MAXLINE,0,preply_addr,&len);
        if(len != servlen || memcmp(pservaddr,preply_addr,len)!= 0){
            printf("reply not from server,ignored)\n");
            continue;
        }
        recvline[n] = 0;
        fputs(recvline,stdout);
    }
}


int main(int argc,char **argv)
{
    int sockfd;
    struct sockaddr_in servaddr;
    if(argc != 2 )
    {
        printf("usage:udpcli <ipaddr>");
        exit(1);
    }
    bzero(&servaddr,sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(SERV_PORT);
    inet_pton(AF_INET,argv[1],&servaddr.sin_addr);
    sockfd = socket(AF_INET,SOCK_DGRAM,0);
    dg_cli(stdin,sockfd,(struct sockaddr *)&servaddr,sizeof(servaddr));
    exit(0);   
}