boost库之udp广播实例

//UdpLinkServer.h

//udp服务#pragma once#include <boost/asio/ip/tcp.hpp>#include <boost/asio.hpp>#include <boost/function.hpp>#include <boost/bind.hpp>#include <boost/enable_shared_from_this.hpp>#include <boost/shared_ptr.hpp>#include <boost/array.hpp>using boost::asio::ip::udp;#define  UDP_DATA_PACKAGE_MAX_LENGTH1024//发送数据回调函数typedef void (CALLBACK *SendDataCallback)(const boost::system::error_code& error,std::size_t bytes_transferred,DWORD dwUserData1,DWORD dwUserData2);//接收数据回调函数typedef void (CALLBACK *RecvDataCallback)(const boost::system::error_code& error,char *pData,int nDataLength,char *pPeerIp,unsigned short usPeerPort,DWORD dwUserData1,DWORD dwUserData2);class UdpLinkServer{public:UdpLinkServer(unsigned short usPort,bool bBroadcast);virtual ~UdpLinkServer(void);//发送数据回调函数 boost function定义typedef boost::function<void* (const boost::system::error_code& error,std::size_t bytes_transferred,DWORD dwUserData1,DWORD dwUserData2)>SendDataCallbackHandler;/*功能：设置接收数据回调函数参数：bAutoRecvData：是否自动接收数据；pfunc：接收数据回调函数；dwUserData1：接收数据回调函数用户数据1；dwUserData2：接收数据回调函数用户数据2返回值：无*/void SetRecvDataCallback(bool bAutoRecvData,RecvDataCallback pfunc,DWORD dwUserData1,DWORD dwUserData2);//开始int Start(boost::asio::io_service& ioService);//停止int Stop();//发送数据int SendDataEx(udp::endpoint endpointRemote,char *pBuffer,int nBufferSize,SendDataCallback pfunc,DWORD dwUserData1,DWORD dwUserData2);//发送数据int SendData(char *pBuffer,int nBufferSize,bool bAsync);//获取广播端点udp::endpoint & GetBroadcastEndPoint();//接收数据处理（手动）void handleRecvDataByManual(RecvDataCallback pfunc=NULL,DWORD dwUserData1=0,DWORD dwUserData2=0);//当发送数据给客户端成功之后响应处理void handleSendData(char *pBuffer,int nBufferSize,const boost::system::error_code& error,std::size_t bytes_transferred);void handleSendDataInner(SendDataCallback pfunc,DWORD dwUserData1,DWORD dwUserData2,const boost::system::error_code& error,std::size_t bytes_transferred);//void handleSendData(boost::shared_ptr<std::string> strMessage,const boost::system::error_code& error,std::size_t bytes_transferred);static void WINAPI SendDataCallbackOuter(const boost::system::error_code& error,std::size_t bytes_transferred,DWORD dwUserData1,DWORD dwUserData2);protected://接收数据void RecvDataProcess(RecvDataCallback pfunc,DWORD dwUserData1,DWORD dwUserData2);//接收数据处理（自动）void handleRecvData(const boost::system::error_code& error,std::size_t bytes_transferred,RecvDataCallback pfunc,DWORD dwUserData1,DWORD dwUserData2);//是否停止服务bool IsStop();private:udp::socket*m_sockUdp;//服务器的SOCKETudp::endpoint m_endpointRemote;//收到数据时的端点信息udp::endpoint m_endpointBroadcast;//广播端点信息boost::array<char,UDP_DATA_PACKAGE_MAX_LENGTH> m_recvBuf;//接收数据缓冲区bool m_bStop;//停止服务bool  m_bBroadcast;//是否广播unsigned short m_usPort;//端口bool m_bAutoRecvData;//是否自动接收数据，true，表示自动接收数据；false，表示外部手动调用函数接收数据RecvDataCallbackm_pfuncRecvDataCallback;//接收数据回调函数DWORDm_dwRecvDataCallbackUserData1;//接收数据回调函数用户数据1DWORDm_dwRecvDataCallbackUserData2;  //接收数据回调函数用户数据2};

//UdpLinkServer.cpp

#include "StdAfx.h"#include "UdpLinkServer.h"#include <boost/exception/all.hpp>UdpLinkServer::UdpLinkServer(unsigned short usPort,bool bBroadcast){m_bStop = false;m_bBroadcast = bBroadcast;m_usPort = usPort;m_sockUdp = NULL;m_bAutoRecvData = true;m_pfuncRecvDataCallback = NULL;m_dwRecvDataCallbackUserData1 = 0;m_dwRecvDataCallbackUserData2 = 0;}UdpLinkServer::~UdpLinkServer(void){if(m_sockUdp != NULL){m_sockUdp->close();delete m_sockUdp;m_sockUdp = NULL;}}//设置接收数据回调函数void UdpLinkServer::SetRecvDataCallback(bool bAutoRecvData,RecvDataCallback pfunc,DWORD dwUserData1,DWORD dwUserData2){m_bAutoRecvData = bAutoRecvData;m_pfuncRecvDataCallback = pfunc;m_dwRecvDataCallbackUserData1 = dwUserData1;m_dwRecvDataCallbackUserData2 = dwUserData2;}//开始int UdpLinkServer::Start(boost::asio::io_service& ioService){try{if(m_bBroadcast){//广播m_sockUdp = new udp::socket(ioService,udp::endpoint(udp::v4(),0));m_sockUdp->set_option(boost::asio::socket_base::reuse_address(true));m_sockUdp->set_option(boost::asio::socket_base::broadcast(true));m_endpointBroadcast = udp::endpoint(boost::asio::ip::address_v4::broadcast(), m_usPort);}else{m_sockUdp = new udp::socket(ioService,udp::endpoint(udp::v4(),m_usPort));if(!m_sockUdp->is_open()){//端口被占用return -1;}m_sockUdp->set_option(boost::asio::socket_base::reuse_address(true));}}catch (boost::exception& e){return -1;}    m_bStop = false;if(m_bAutoRecvData){RecvDataProcess(m_pfuncRecvDataCallback,m_dwRecvDataCallbackUserData1,m_dwRecvDataCallbackUserData2);}return 0;}//停止int UdpLinkServer::Stop(){m_bStop = true;return 0;}//是否停止服务bool UdpLinkServer::IsStop(){return m_bStop;}//获取广播端点udp::endpoint & UdpLinkServer::GetBroadcastEndPoint(){return m_endpointBroadcast;}void UdpLinkServer::SendDataCallbackOuter(const boost::system::error_code& error,std::size_t bytes_transferred,DWORD dwUserData1,DWORD dwUserData2){int i = 0;}//发送数据int UdpLinkServer::SendDataEx(udp::endpoint endpointRemote,char *pBuffer,int nBufferSize,SendDataCallback pfunc,DWORD dwUserData1,DWORD dwUserData2){m_sockUdp->async_send_to(boost::asio::buffer(pBuffer,nBufferSize),endpointRemote,boost::bind(&UdpLinkServer::handleSendDataInner,this,pfunc,dwUserData1,dwUserData2,boost::asio::placeholders::error,boost::asio::placeholders::bytes_transferred));return 0;}//发送数据int UdpLinkServer::SendData(char *pBuffer,int nBufferSize,bool bAsync){if(!m_bBroadcast){if(bAsync){//异步发送m_sockUdp->async_send_to(boost::asio::buffer(pBuffer,nBufferSize),m_endpointRemote,boost::bind(&UdpLinkServer::handleSendData,this,pBuffer,nBufferSize,boost::asio::placeholders::error,boost::asio::placeholders::bytes_transferred));}else{//同步发送m_sockUdp->send_to(boost::asio::buffer(pBuffer,nBufferSize),m_endpointRemote);}}else{//广播if(bAsync){//异步发送m_sockUdp->async_send_to(boost::asio::buffer(pBuffer,nBufferSize),m_endpointBroadcast,boost::bind(&UdpLinkServer::handleSendData,this,pBuffer,nBufferSize,boost::asio::placeholders::error,boost::asio::placeholders::bytes_transferred));}else{//同步发送m_sockUdp->send_to(boost::asio::buffer(pBuffer,nBufferSize),m_endpointBroadcast);}}return 0;}//接收数据void UdpLinkServer::RecvDataProcess(RecvDataCallback pfunc,DWORD dwUserData1,DWORD dwUserData2){//异步接收数据m_sockUdp->async_receive_from(boost::asio::buffer(m_recvBuf),m_endpointRemote,boost::bind(&UdpLinkServer::handleRecvData,this,boost::asio::placeholders::error,boost::asio::placeholders::bytes_transferred,pfunc,dwUserData1,dwUserData2));}//接收数据处理（手动），就进入本函数响应处理void UdpLinkServer::handleRecvDataByManual(RecvDataCallback pfunc,DWORD dwUserData1,DWORD dwUserData2){if(IsStop())return;//下一次接收RecvDataProcess(pfunc,dwUserData1,dwUserData2);}//当收到客户端数据时，就进入本函数响应处理  void UdpLinkServer::handleRecvData(const boost::system::error_code& error,std::size_t bytes_transferred,RecvDataCallback pfunc,DWORD dwUserData1,DWORD dwUserData2){if(IsStop())return;//如果没有出错if(!error || error==boost::asio::error::message_size){if(bytes_transferred > UDP_DATA_PACKAGE_MAX_LENGTH){//超过最大数据长度return;}/*//打印接收的内容char szTmp[UDP_DATA_PACKAGE_MAX_LENGTH] = {0};memcpy(szTmp,m_recvBuf.data(),bytes_transferred);printf("%s\n",szTmp);//boost::shared_ptr<std::string> strMessage(new std::string("aaaaaa"));//SendData((char *)strMessage.data(),strMessage.size());std::string strMessage = "aaaaaaaaaa";SendDataEx(m_endpointRemote,(char *)strMessage.data(),strMessage.size(),SendDataCallbackOuter,(int)this,0);*///回调数据if(pfunc != NULL){pfunc(error,m_recvBuf.data(),bytes_transferred,(char *)m_endpointRemote.address().to_string().c_str(),m_endpointRemote.port(),dwUserData1,dwUserData2);}//下一次接收RecvDataProcess(pfunc,dwUserData1,dwUserData2);}else{//出错if(pfunc != NULL){pfunc(error,NULL,bytes_transferred,NULL,0,dwUserData1,dwUserData2);}}}//当发送数据给客户端成功之后响应。  void UdpLinkServer::handleSendData(char *pBuffer,int nBufferSize,const boost::system::error_code& error,std::size_t bytes_transferred){int n = 0;}void UdpLinkServer::handleSendDataInner(SendDataCallback pfunc,DWORD dwUserData1,DWORD dwUserData2,const boost::system::error_code& error,std::size_t bytes_transferred){if(error != NULL){//打印错误信息printf("%s", boost::system::system_error(error).what());}if(pfunc != NULL){pfunc(error,bytes_transferred,dwUserData1,dwUserData2);}int n = 0;}/*void UdpLinkServer::handleSendData(boost::shared_ptr<std::string> strMessage,const boost::system::error_code& error,std::size_t bytes_transferred){ int n = 0;}*/

//广播发送代码

// UdpbroastSender.cpp : 定义控制台应用程序的入口点。//Udpbroast Sender#include "stdafx.h"#include "UdpLinkServer.h"#include <boost/thread/thread.hpp>//设备信息typedef struct tagDeviceInfo{unsigned shortusFunction;//功能码unsigned shortusVersionFlag;//版本标记unsigned intuiCompanyId;//企业idcharszDeviceSerialNo[24];//设备序列号unsigned shortusServicePort;//数据服务端口charszExtend[38];//扩展}DeviceInfo;//工作线程参数typedef struct tagWorkThreadParameter{boost::asio::io_service * pIoService;UdpLinkServer * pUdpService;}WorkThreadParameter;//工作线程函数处理函数bool g_WorkThreadExit = false;int  g_nBroastDataSendInteral = 5000;//广播发送间隔，单位：毫秒DeviceInfo g_diDeviceInfo = {0};//设备信息unsigned int __stdcall WorkThreadFunByDeviceServiceProcess(PVOID pParam){int nn = 0;int nDataSize = sizeof(DeviceInfo);WorkThreadParameter *pAllParameter = (WorkThreadParameter *)pParam;while(true){if(g_WorkThreadExit){break;}//发送数据pAllParameter->pUdpService->SendData((char *)&g_diDeviceInfo,nDataSize,true);pAllParameter->pIoService->poll();for(nn=g_nBroastDataSendInteral;nn>0; nn-=200){if(g_WorkThreadExit){break;}Sleep(200);}}return 0;}int main(int argc, char* argv[]){boost::asio::io_service ioService;UdpLinkServer usUdpService(9090,true);//true，为广播模式；false，非广播模式usUdpService.Start(ioService);g_diDeviceInfo.usFunction = 1;g_diDeviceInfo.usVersionFlag = 0x0001;strcpy(g_diDeviceInfo.szDeviceSerialNo,"ABCDEFG111111111");g_diDeviceInfo.usServicePort = 9001;WorkThreadParameter wtpWorkThreadParameter;wtpWorkThreadParameter.pIoService = &ioService;wtpWorkThreadParameter.pUdpService = &usUdpService;//开启线程boost::thread thrd(WorkThreadFunByDeviceServiceProcess,&wtpWorkThreadParameter);//线程poll等待thrd.join();g_WorkThreadExit = true;usUdpService.Stop();ioService.stop();return 0;}

//广播接收代码

// UdpbroastRevicer.cpp : 定义控制台应用程序的入口点。//Udpbroast Revicer#include "stdafx.h"#include "UdpLinkServer.h"#include <boost/thread/thread.hpp>//设备信息typedef struct tagDeviceInfo{unsigned shortusFunction;//功能码unsigned shortusVersionFlag;//版本标记unsigned intuiCompanyId;//企业idcharszDeviceSerialNo[24];//设备序列号unsigned shortusServicePort;//数据服务端口charszExtend[38];//扩展}DeviceInfo;//工作线程函数处理函数bool g_WorkThreadExit = false;int  g_nBroastDataSendInteral = 5000;//广播发送间隔，单位：毫秒DeviceInfo g_diDeviceInfo = {0};//设备信息unsigned int __stdcall WorkThreadFunByDeviceServiceProcess(PVOID pParam){int nn = 0;boost::asio::io_service * pIoService  = (boost::asio::io_service *)pParam;while(true){if(g_WorkThreadExit){break;}pIoService->poll();for(nn=g_nBroastDataSendInteral;nn>0; nn-=200){if(g_WorkThreadExit){break;}Sleep(200);}}return 0;}static void WINAPI BroastDeviceInfoRecvDataCallback(const boost::system::error_code& error,char *pData,int nDataLength,char *pPeerIp,unsigned short usPeerPort,DWORD dwUserData1,DWORD dwUserData2){SYSTEMTIME sm;GetLocalTime(&sm);char szInfo[256] = {0};DeviceInfo *pDeviceInfo = (DeviceInfo *)pData;sprintf(szInfo,"%s %s:%d time:%04d-%02d-%0d %02d:%02d:%02d\n",pDeviceInfo->szDeviceSerialNo,pPeerIp,usPeerPort,sm.wYear,sm.wMonth,sm.wDay,sm.wHour,sm.wMinute,sm.wSecond);printf(szInfo);}int main(int argc, char* argv[]){boost::asio::io_service ioService;UdpLinkServer usUdpService(9090,false);usUdpService.SetRecvDataCallback(true,BroastDeviceInfoRecvDataCallback,0,0);usUdpService.Start(ioService);//开启线程boost::thread thrd(WorkThreadFunByDeviceServiceProcess,&ioService);//线程poll等待thrd.join();g_WorkThreadExit = true;usUdpService.Stop();ioService.stop();return 0;}