Socket Programming: UDP Echo Server/Client

UDP Server相對於TCP來講是簡單了許多，但是因為UDP本身並沒有提供可靠的連線，Server本體必須視情況要自己處理這些事情。

TCP Server的所需function:
* Socket()
* Bind()
* Listen()
* Accept() -> 連線建立，可以送/收資料
*連線建立完成以後使用 send(), recv() 就可以寫/收資料

TCP Client 連線所需function:
* Socket();
* Connect(); -> 連線建立，可以送資料
*連線建立完成以後使用 send(), recv() 就可以寫/收資料

注意上面的send/recv必須使用在connected socket的時候，所以其實下面兩個是同樣的意思的：

ssize_t send(sockfd, buf, len, flags);
ssize_t sendto(sockfd, buf, len, flags, NULL, 0);

ssize_t recv(sockfd, buf, len, flags);
ssize_t recvfrom(sockfd, buf, len, flags, NULL, 0);

UDP相較來說就簡單許多
UDP Server:
* Socket()
* Bind() -> 可以接收資料
UDP Client:
* Socket(); -> 可以傳送資料

UDP一整個就是完全甚麼都不用管那....XD 另外在recvfrom的時候，會將對方的socket address也寫入給予的第五個和第六個參數，這樣一來就可以知道這個封包是從哪裡來的。所以通常TCP Server會用fork()寫成Concurrent 的形式，UDP相對來說就比較沒有需要。處理一個封包就知道哪裡來的，要回復的話可以直接回傳。


UDP Echo Server

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

#define MAXLINE 1024

int main(int argc, char** argv){

    int sockfd;
    int len;
    int z;
    char buf[MAXLINE];
    struct sockaddr_in adr_inet;
    struct sockaddr_in adr_clnt;
    len = sizeof(adr_clnt);

    if(argc != 2){
        printf ("server <port>\n");
        exit(1);
    }

    bzero(&adr_inet,sizeof(adr_inet));
    adr_inet.sin_family = AF_INET;
    adr_inet.sin_addr.s_addr = htonl(INADDR_ANY);
    adr_inet.sin_port = htons(atoi(argv[1]));

    len = sizeof(adr_clnt);

    sockfd = socket(AF_INET, SOCK_DGRAM, 0);

    if(sockfd == -1){
        printf("socket error\n");
        exit(1);
    }

    if((z = bind(sockfd, (struct sockaddr*) &adr_inet, sizeof(adr_inet))==-1)){
        printf("bind error\n");
        exit(1);
    }

    while(1){
        bzero(buf,MAXLINE);
        z = recvfrom (sockfd, buf, MAXLINE, 0, (struct sockaddr *)&adr_clnt, &len);
        if(z < 0){
            printf("recvfrom error\n");
            exit(1);
        }
        printf("recieved: %s\n",buf);
        if((sendto(sockfd,buf, MAXLINE, 0, (struct sockaddr*) &adr_clnt, sizeof(adr_clnt)))<0){
            perror("send ping error\n");
        }
    }
    return 0;
}

UDP Echo Client

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

#define MAXLINE 1024

int main(int argc, char** argv){
    int sockfd;
    int z, len; 
    char buf[MAXLINE];
    struct sockaddr_in adr_serv, adr_recv;
    len = sizeof(adr_serv);
    if(argc != 3){
        printf("client <ip> <port>\n");
        exit(1);
    }

    bzero(&adr_serv, sizeof(adr_serv));
    adr_serv.sin_family = AF_INET;
    adr_serv.sin_addr.s_addr = inet_addr(argv[1]);
    adr_serv.sin_port = htons(atoi(argv[2]));

    sockfd = socket(AF_INET, SOCK_DGRAM, 0);

    if(sockfd == -1){
        printf("socket error\n");
        exit(1);
    }

    while(5566){
        bzero(buf,MAXLINE)
        fgets(buf, MAXLINE, stdin)
        z = sendto(sockfd, buf, strlen(buf), 0, (struct sockaddr*) &adr_serv, sizeof(adr_serv));
        if(z < 0){
            perror("sendto error\n");
            exit(1);
        }
        
        if(recvfrom(sockfd, buf, MAXLINE,0,(struct sockaddr*)&adr_recv, &len)<0){
            perror("recv pong error\n");
        }
        printf("%s",buf);
    }
    return 0;
}