本文原地址: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;
}