网络编程—简单的udp套接字编程(5)

小咪咪 2022-05-22 05:58 208阅读 0赞

![这里写图片描述][70]

从上面图片可以看出，在udp编程模型中，服务器端不需要设置监听的套接字，客户端中也不需要建立连接的过程。  
客户端的模型中，也可以在创建socket之后调用connect函数，不过此时udp的connect作用和tcp中的connect不一样，udp中的connect只是将sockaddr信息设置到socket中去，后面调用sendto和recvfrom的函数可以改为tcp中的send和recv。  
详情参见下面代码：

//客户端代码
    #include <stdlib.h>
    #include <unistd.h>
    #include <resolv.h>
    #include <sys/socket.h>
    #include <sys/time.h>
    #include <time.h>
    #include <string.h>
    #include <signal.h>
    
    const int MAXBUF=1000;
    
    void sig_handler(int signo)
    {
            if(signo == SIGINT)
            {
                    printf("server close\n");
                    exit(1);
            }
    }
    
    int main()
    {
            // 第一个参数选用AF_INET表示选用ipv4,第二个参数选用SOCK_DGRAM(udp)
            int socketfd = socket(AF_INET, SOCK_DGRAM, 0);
            struct sockaddr_in dst;
            dst.sin_family = AF_INET;
            dst.sin_port = htons(10567);
            dst.sin_addr.s_addr = htonl(0x7f000001);
            int ret = 0;
            // 将sockaddr信息设置到socketfd中
            ret = connect(socketfd, (struct sockaddr*)&dst, sizeof(sockaddr_in));
            if(ret != 0)
            {
                    printf("connect failed\n");
                    close(socketfd);
                    socketfd = -1;
            }
            char buf[MAXBUF] = {
       0};
            int i = 0;
    
            {
                    memset(buf, 0, sizeof(buf));
                    sprintf(buf, "i = %d, socket = %d, time = %lu", i, socketfd, clock());
                    printf("send buffer = [%s]\n", buf);
                    send(socketfd,  buf, strlen(buf), 0);
                    memset(buf, 0, sizeof(buf));
                    ret = recv(socketfd, buf, sizeof(buf), 0);
                    if(ret == 0)
                    {
                            printf("error find\n");
                            //break;
                    }
                    else
                    {
                            printf("recv len = %d, i = %d, socket = %d,\n", ret, i, socketfd);
                            printf("recv buffer = [%s]\n", buf);
                    }
            }
            close(socketfd);
            socketfd = -1;
            return 0;
    }

服务器端代码：

#include <sys/socket.h>
    #include <string.h>
    #include <stdio.h>
    #include <stdlib.h>
    #include <sys/time.h>
    #include <time.h>
    #include <unistd.h>
    #include <arpa/inet.h>
    #include <netdb.h>
    #include <fcntl.h>
    #include <errno.h>
    #include <signal.h>
    #include <sys/types.h>
    #include <sys/wait.h>
    
    const unsigned int MAXBUF = 1000;
    int listenFd = -1;
    
    // 信号处理函数
    void sig_handler(int signo)
    {
        // ctrl+c信号
            if(signo == SIGINT)
            {
                    printf("server close\n");
                    close(listenFd);
                    listenFd = -1;
                    exit(1);
            }
                //子进程结束信号
            if(signo == SIGCHLD)
            {
                    printf("child process exit\n");
                    wait(0);
            }
    }
    
    // 输出sockaddr信息
    void out_addr(struct sockaddr_in *addr)
    {
            if(addr == NULL)
            {
                    printf("invalid param addr\n");
                    return;
            }
            switch(addr->sin_family)
            {
                    case AF_INET:
                            printf("AF_INET\n");
                            break;
                    case AF_INET6:
                            printf("AF_INET6\n");
                            break;
                    default:
                            printf("unknown\n");
                            break;
            }
            char ip[20] = {
       0};
            memset(ip, 0, sizeof(ip));
            //将ip地址转换成主机点分十进制
            inet_ntop(addr->sin_family, &addr->sin_addr, ip, sizeof(ip));
            printf("port = %d, ip = %s\n", ntohs(addr->sin_port), ip);
    
    }
    
    void do_service(int sockfd)
    {
            struct sockaddr_in clientAddr;
            socklen_t len = sizeof(clientAddr);
            char buffer[MAXBUF] = {
       0};
            memset(buffer, 0, sizeof(buffer));
            if(recvfrom(sockfd, buffer, sizeof(buffer), 0, (struct sockaddr*)&clientAddr, &len) < 0)
            {
                    printf("recv failed\n");
                    return;
            }
            else
            {
                    out_addr(&clientAddr);
                    printf("client send = [%s]\n", buffer);
                    memset(buffer, 0, sizeof(buffer));
                    long int t = time(0);
                    char *ptr = ctime(&t);
                    size_t size = strlen(ptr) * sizeof(char);
                    if(sendto(sockfd, ptr, size, 0, (struct sockaddr*)&clientAddr, len) < 0)
                    {
                            printf("send failed [%s]\n", buffer);
                    }
                    else
                    {
    
                    }
            }
    }
    
    
    
    int main()
    {
            if(signal(SIGINT, sig_handler) == SIG_ERR)
            {
                    perror("signal error\n");
                    exit(1);
            }
            listenFd = socket(AF_INET, SOCK_DGRAM, 0);
            if(listenFd <= 0)
            {
                    printf("create tcp socket failed\n");
                    return -1;
            }
    
            int ret = 0;
            int opt = 1;
               // 设置地址重用
            if((ret = setsockopt(listenFd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt))) < 0)
            {
                    perror("setsockopt error\n");
                    exit(1);
            }
    
            struct sockaddr_in local;
            local.sin_family = AF_INET;
            local.sin_port = htons(10567);
            local.sin_addr.s_addr = INADDR_ANY;// htonl(0x7f000001);
            ret = bind(listenFd, (struct sockaddr*)&local, sizeof(sockaddr));
            if(ret != 0)
            {
                    printf("bind ret = %d\n", ret);
                    return -2;
            }
            sockaddr_in src;
            unsigned int srcLength = sizeof(sockaddr_in);
            char buffer[MAXBUF] = {
       0};
            while(1)
            {
                    do_service(listenFd);
            }
            close(listenFd);
            listenFd = -1;
            return 0;
    }

[70]: /images/20220522/de4a63f321324886bd2a6c142c3c3a3a.png