开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 多媒体 > IP电话/视频会议 > 查看源码
Multicast.cpp
资源名称：视频会议系统.rar　[点击查看]
上传用户：popouu88
上传日期：2013-02-11
资源大小：2894k
文件大小：3k
源码类别：

IP电话/视频会议

开发平台：

Visual C++

Multicast.cpp：源码内容
  1. // Multicast.cpp: implementation of the CMulticast class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #include "..stdafx.h"
  6. #include "Multicast.h"
  8. #ifdef _DEBUG
  9. #undef THIS_FILE
  10. static char THIS_FILE[]=__FILE__;
  11. #define new DEBUG_NEW
  12. #endif
  14. #pragma comment( lib , "ws2_32.lib")
  15. //////////////////////////////////////////////////////////////////////
  16. // Construction/Destruction
  17. //////////////////////////////////////////////////////////////////////
  19. int CMulticast::InitWSAStatup = 0;
  21. CMulticast::CMulticast()
  22. {
  23. if( ! CMulticast::InitWSAStatup )
  24. {
  25. WSADATA data;
  27. ::WSAStartup( MAKEWORD( 2 , 2 ) , &data );
  28. }
  29. CMulticast::InitWSAStatup ++;
  31. this->m_sock = INVALID_SOCKET;
  33. memset( &this->addr , 0 , sizeof( this->addr ) );
  35. memset( &this->command , 0 , sizeof( this->command ) );
  37. this->user_data = 0;
  39. memset( &this->recvBuffer , 0 , sizeof( this->recvBuffer ) );
  40. }
  42. CMulticast::~CMulticast()
  43. {
  44. CMulticast::InitWSAStatup -- ;
  46. if( ! CMulticast::InitWSAStatup )
  48. ::WSACleanup( );
  49. }
  50. //加入组播
  51. bool CMulticast::AddMemberShip( const char * ip , const int port )
  52. {
  53. this->DropMemberShip( );
  55. if( ( this->m_sock = ::socket( AF_INET , SOCK_DGRAM ,  0 ) ) == INVALID_SOCKET )
  57. return false;
  59. this->addr.sin_family = AF_INET;
  61. this->addr.sin_addr.s_addr = INADDR_ANY;
  63. this->addr.sin_port = ::htons( port );
  65. int on = 1;
  66.     //设定端口可重用
  67.     if( ::setsockopt( this->m_sock , SOL_SOCKET , SO_REUSEADDR , ( const char * )&on , sizeof( on ) ) < 0 )
  68. {
  69. this->DropMemberShip( );
  71. return false;
  72. }//绑定端口
  73. if( ::bind( this->m_sock , ( struct sockaddr * )&this->addr , sizeof( this->addr ) ) == SOCKET_ERROR )
  74. {
  75. this->DropMemberShip( );
  77. return false;
  78. }
  79. this->command.imr_multiaddr.s_addr = inet_addr( ip ); 
  81.     this->command.imr_interface.s_addr = INADDR_ANY;
  83. if( this->command.imr_multiaddr.s_addr == -1 ) 
  84. {
  85. this->DropMemberShip( );
  87. return false;
  88. }//加入组播
  89. if( ::setsockopt( this->m_sock , IPPROTO_IP , IP_ADD_MEMBERSHIP , ( const char * )&this->command , sizeof( this->command ) ) < 0 )
  90. {
  91. this->DropMemberShip( );
  93. return false;
  94. }//保存组播地址
  95. this->addr.sin_addr.s_addr = ::inet_addr( ip );
  97. return true;
  98. }
  99. //从组播中移出
  100. void CMulticast::DropMemberShip( void )
  101. {
  102. if( this->IsConnected( ) )
  103. {   //移出租播
  104. ::setsockopt( this->m_sock , IPPROTO_IP , IP_DROP_MEMBERSHIP , ( const char * )&this->command , sizeof( this->command ) );
  106. ::closesocket( this->m_sock );
  108. this->m_sock = INVALID_SOCKET;
  109. }
  110. memset( &this->command , 0 , sizeof( this->command ) );
  111. }
  112. //发送组播数据
  113. bool CMulticast::SendTo( CBuffer & buffer )
  114. {
  115. if( this->IsConnected( ) )
  117. return ::sendto( this->m_sock , buffer.GetBuffer( ) , *( int * )buffer.GetBuffer( ) , 0 , ( struct sockaddr * )&this->addr , sizeof( this->addr ) );
  119. return false;
  120. }
  121. //接收组播数据
  122. bool CMulticast::ReceiveFrom( CBuffer & buffer )
  123. {
  124. if( ! this->IsConnected( ) )
  126. return false;
  128. struct sockaddr_in addr;
  130. int len = sizeof( addr );
  132. if( ::recvfrom( this->m_sock , this->recvBuffer , MAX_BUFFER_SIZE , 0 , ( struct sockaddr * )&addr , &len ) >= *( int * )this->recvBuffer )
  133. {
  134. if( buffer.GetSize( ) < *( int * )this->recvBuffer )
  136. buffer.Resize( *( int * )this->recvBuffer );
  138. memcpy( buffer.GetBuffer( ) , this->recvBuffer , *( int * )this->recvBuffer );
  140. return true;
  141. }
  142. return false;
  143. }