C语言udp的封装和使用

本文原地址：C语言udp的封装和使用

使用C++ 将udp做一个简单的封装，实现及时通讯，以供大家参考学习和自己备用

本项目由QT编写， windows下需要在 .pro文件需要加入以下配置：
#-lWs2_32 添加库Ws2_32
LIBS += -lWs2_32

测试方法：编译通过后，win+R 输入cmd 回车打开命令窗口
发送端：
输入.exe文件名后边加IP地址
接受端：
输入.exe文件名直接运行
退出：
输入exit回车退出

代码如下：
udp.h

#ifndef UDP_H
#define UDP_H

#ifdef MYLINUX

#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#define SOCKET int

#else

#include <winsock2.h>

#endif

//明确此函数是C语言函数，编译时候不要转换成C++格式了
//extern "C"
//{

//}

class myudp{
private:
    SOCKET st;

public:
    myudp();
    ~myudp();
    int socket_bind(short int port);
    int socket_send(const char *IP, const short int port, const char *buf, int len);
    int socket_recv(char *buf, int len, char *srcIP);

};


#endif // UDP_H

udp.cpp

#include <string.h>
#include <stdio.h>
#include "udp.h"

int myudp::socket_send(const char *IP, const short int port, const char *buf, int len){


    /**
     * 建立socket
     * AF_INET 指定使用TCP/IP协议
     * SOCK_DGRAM 指定使用UDP协议
     */
    st = socket(AF_INET, SOCK_DGRAM, 0);
    struct sockaddr_in addr;
    memset(&addr, 0, sizeof(addr));//初始化结构
    addr.sin_family = AF_INET;
    addr.sin_port = htons(port);//host to net short
    addr.sin_addr.s_addr = inet_addr(IP);

    //IP 内存原理解析
//    unsigned long laddr = inet_addr("192.168.6.200");
//    unsigned char *p = &laddr;
//    printf("IP_long:%x\n", laddr);
//    printf("IP_char:%u,%u,%u,%u\n", *(p),*(p+1),*(p+2),*(p+3));
    size_t rc = sendto(st, buf, len, 0, (struct sockaddr *)&addr, sizeof(addr));


    return rc;
}

int myudp::socket_recv(char *buf, int len, char *srcIP){
    struct sockaddr_in sendaddr;
    memset(&sendaddr, 0, sizeof(sendaddr));

#ifdef MYLINUX
    socklen_t addrlen;
#else
    int addrlen;
#endif

    addrlen = sizeof(sendaddr);

    memset(buf, 0, len);
    //接受UDP数据
    size_t rc = recvfrom(st, buf, len, 0, (struct sockaddr *)&sendaddr, &addrlen);
//    printf("%s:%s\n", inet_ntoa(sendaddr.sin_addr), buf);
    if(srcIP)
    strcpy(srcIP, inet_ntoa(sendaddr.sin_addr));
    return rc;
}

myudp::myudp()
{
#ifndef MYLINUX

    /**
     * 初始化socket
     */
    DWORD ver;
    WSADATA wsaData;

    ver = MAKEWORD(1, 1);//在WSAStartup之前，告诉windows使用Socket的版本号
    WSAStartup(ver, &wsaData);//windows要求，使用socket必须调用这个函数
#endif

    /**
     * 建立socket
     * AF_INET 指定使用TCP/IP协议
     * SOCK_DGRAM 指定使用UDP协议
     */
    st = socket(AF_INET, SOCK_DGRAM, 0);
}

myudp::~myudp()
{
#ifdef MYLINUX
    close(st);
#else
    closesocket(st);//关闭socket
    WSACleanup();//释放win中 socket相关资源
#endif
}

//将接收方和端口号绑定
int myudp::socket_bind(short port)
{

    struct sockaddr_in addr;
    memset(&addr, 0, sizeof(addr));//初始化结构
    addr.sin_family = AF_INET;
    addr.sin_port = htons(port);//host to net short
    addr.sin_addr.s_addr = htonl(INADDR_ANY);//INADDR_ANY接收方的主机ip

    int localBind = bind(st, (struct sockaddr *)&addr, sizeof(addr));

    return localBind;
}

main.cpp

#include <iostream>
#include "udp.h"

#define PORT 8080

using namespace std;

int main(int argc, char *args[])
{
    if(argc > 1){
    myudp udp;
    char buf[1024] = {0};
    while(1){
        memset(buf, 0, sizeof(buf));
        gets(buf);
        udp.socket_send(args[1], PORT, buf, strlen(buf));
        if(strncmp(buf, "exit", 4) == 0)
            break;
    }

    }else{
    myudp udp;
    char buf[1024] = {0};
    if(udp.socket_bind(PORT) > -1){
        cout << "bind success !" << endl;
        char ip[100] = {0};
        while(1){
            memset(buf, 0, sizeof(buf));
            memset(ip, 0, sizeof(ip));

            udp.socket_recv(buf, sizeof(buf), ip);
            if(strncmp(buf, "exit", 4) == 0)
                break;
            cout << buf << endl;
            cout << ip << endl;
        }

    }

    }

    cout << "over" << endl;
    return 0;
}