udp的封装和使用

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

本项目由QT编写, windows下需要在 .pro文件需要加入以下配置:

#-lWs2_32 添加库Ws2_32

LIBS += -lWs2_32

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

代码如下:
udp.h

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
#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

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
#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

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#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;
}