UDP服务器和客户端的编写(原创)(linux系统)

服务器:server
#include<stdio.h>
#include<stdlib.h>
#include<pthread.h>
#include<string.h>
#include<sys/socket.h>
#include <sys/epoll.h>
#include <arpa/inet.h>

#define DEFAULT_HEADBEAT_PORT 7090
#define RECV_BUF_SIZE 1024

int epfd_heatbeat = 0;
int heatBeatFd = 0;

int headBeatUdpSocketInit(){
    int ret = 0;
    int len = 0 , size = 0;
    
    struct epoll_event event;
    struct sockaddr_in heatbeadAddr;

    epfd_heatbeat = epoll_create(1);
    if(epfd_heatbeat == -1){
        printf("[headBeatRecv] Epoll Init Failed.\n");
        return -1;
    }

    if ((heatBeatFd = socket(AF_INET, SOCK_DGRAM, 0)) < 0)
    {
        return -1;
    }

    memset(&heatbeadAddr, 0, sizeof(heatbeadAddr));
    heatbeadAddr.sin_family = AF_INET;
    heatbeadAddr.sin_addr.s_addr = htonl(INADDR_ANY); //自动获取IP地址
    heatbeadAddr.sin_port = htons(DEFAULT_HEADBEAT_PORT);

    if (bind(heatBeatFd, (struct sockaddr *)&heatbeadAddr, sizeof(heatbeadAddr)) < 0)
    {
        close(heatBeatFd);
        return -1;
    }

    memset(&event, 0, sizeof(event));

    if(heatBeatFd > 0){
        event.data.fd = heatBeatFd;
        event.events = EPOLLIN;
        ret = epoll_ctl(epfd_heatbeat, EPOLL_CTL_ADD, heatBeatFd, &event);
        if (ret < 0)
        {
            close(heatBeatFd);
            heatBeatFd = -1;
            return -1;
        }
    }

    len = sizeof(size);
    if (getsockopt(heatBeatFd, SOL_SOCKET, SO_RCVBUF, (void *) &size, &len) == 0)
    {
        size *= 4;
        setsockopt(heatBeatFd, SOL_SOCKET, SO_RCVBUF, (void *) &size, sizeof (size));
    }
    else
    {
    }
    
    if (getsockopt(heatBeatFd, SOL_SOCKET, SO_SNDBUF, (void *) &size, &len) == 0)
    {
        size *= 4;
        setsockopt(heatBeatFd, SOL_SOCKET, SO_SNDBUF, (void *) &size, sizeof (size));
    }
    else
    {
    }

    return 0;
}

void *_headBeatRecvThread(void *data){
    int ret = 0, res = 0;
    int readfd = 0;
    int fdnum = 0;
    int recvfd = 0;
    int client_addr_len = 0;
    struct sockaddr_in clientaddr;
    struct epoll_event  heatBeatEventArray[1];
    char recv_buf[RECV_BUF_SIZE] = {'\0'};

    ret = headBeatUdpSocketInit();
    if(ret != 0){
        pthread_exit(NULL);
    }

    while(1){
        if(heatBeatFd > 0){
            readfd = epoll_wait(epfd_heatbeat, heatBeatEventArray, 1, 1000);
            if(readfd > 0){
                for(fdnum = 0; fdnum < readfd; fdnum++){
                    bzero(recv_buf, sizeof(recv_buf));
                    memset(&clientaddr, 0, sizeof(clientaddr));
                    client_addr_len = sizeof(clientaddr);

                    recvfd = heatBeatEventArray[fdnum].data.fd;
                    if(recvfd = heatBeatFd){
                        if(heatBeatEventArray[fdnum].events & EPOLLIN){
                            res = recvfrom(heatBeatFd, recv_buf, RECV_BUF_SIZE - 1, 0, (struct sockaddr *)&clientaddr, &client_addr_len);
                            if (res == -1)
                            {
                            }else{
                                printf("recv value:%s\n",recv_buf);
                                sendto(heatBeatFd, recv_buf, strlen(recv_buf), 0, (struct sockaddr*)&clientaddr, sizeof(clientaddr));
                            }
                        }
                    }    
                }
            }
        }
    }
    
    return NULL;
}

int main(){
   
    pthread_t _sipHeatBeatThread = -1;
    pthread_attr_t _sipHeatBeatThreadAttr;

    pthread_attr_init(&_sipHeatBeatThreadAttr);
    pthread_attr_setdetachstate(&_sipHeatBeatThreadAttr, PTHREAD_CREATE_DETACHED);

    if(pthread_create(&_sipHeatBeatThread, &_sipHeatBeatThreadAttr, _headBeatRecvThread, NULL) < 0){
        printf("pthread create faild\n");
        return -1;
    }
    while(1){
        sleep(5);
    } 
    return 0;
}
多线程:编译时加gcc  -lpthread

客户端:client
#include<stdio.h>
#include<stdlib.h>
#include<pthread.h>
#include<string.h>
#include<sys/socket.h>
#include<sys/epoll.h>
#include<arpa/inet.h>
#include<time.h>

int main(){

    char buffer[]="abc";
    char recv_buf[1024] = {'\0'};
    int client_addr_len = 0;

    int fd = socket(AF_INET, SOCK_DGRAM, 0);
    //ssize_t sendto(int sock, void *buf, size_t nbytes, int flags, struct sockaddr *to, socklen_t addrlen);  
    
    struct sockaddr_in servAddr;
    struct sockaddr_in clientaddr;
    
    memset(&servAddr, 0, sizeof(servAddr));  
    memset(&clientaddr, 0, sizeof(clientaddr));  
    
    servAddr.sin_family = AF_INET;
    servAddr.sin_addr.s_addr = inet_addr("127.0.0.1");
    servAddr.sin_port = htons(7090);
    while(1){
        //bzero(buffer, sizeof(buffer));
        //printf("输入字符串:");
        //scanf("%s",buffer);
        bzero(recv_buf, sizeof(recv_buf));
        sendto(fd, buffer, strlen(buffer), 0, (struct sockaddr*)&servAddr, sizeof(servAddr));
        recvfrom(fd, recv_buf, 1024 - 1, 0, (struct sockaddr *)&clientaddr, &client_addr_len);
        printf("recv:%s\n", recv_buf);
        sleep(1);
    }
    close(fd);

    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值