TCP.UDP_gaojibiancheng.rar_TCP-UDP_c# tcp/udp_tcp/udp_visual c

/////////////////////////////////////////////////////////////////////////////// // File: SocketComm.cpp // Version: 1.1 // // Author: Ernest Laurentin // E-mail: [email protected] // // Implementation of the CSocketComm and associated classes. // // This code may be used in compiled form in any way you desire. This // file may be redistributed unmodified by any means PROVIDING it is // not sold for profit without the authors written consent, and // providing that this notice and the authors name and all copyright // notices remains intact. // // An email letting me know how you are using it would be nice as well. // // This file is provided "as is" with no expressed or implied warranty. // The author accepts no liability for any damage/loss of business that // this c++ class may cause. // // Version history // // 1.0 - Initial release. // 1.1 - Add support for Smart Addressing mode /////////////////////////////////////////////////////////////////////////////// #include "stdafx.h" #include <stdio.h> #include <tchar.h> #include <process.h> #include <crtdbg.h> #include "SocketComm.h" const DWORD DEFAULT_TIMEOUT = 100L; /////////////////////////////////////////////////////////////////////////////// // SockAddrIn Struct /////////////////////////////////////////////////////////////////////////////// // Copy SockAddrIn& SockAddrIn::Copy(const SockAddrIn& sin) { memcpy(&this->sockAddrIn, &sin.sockAddrIn, Size()); return *this; } /////////////////////////////////////////////////////////////////////////////// // IsEqual bool SockAddrIn::IsEqual(const SockAddrIn& sin) { // Is it Equal? - ignore 'sin_zero' return (memcmp(&this->sockAddrIn, &sin.sockAddrIn, Size()-sizeof(sockAddrIn.sin_zero)) == 0); } /////////////////////////////////////////////////////////////////////////////// // IsGreater bool SockAddrIn::IsGreater(const SockAddrIn& sin) { // Is it Greater? - ignore 'sin_zero' return (memcmp(&this->sockAddrIn, &sin.sockAddrIn, Size()-sizeof(sockAddrIn.sin_zero)) > 0); } /////////////////////////////////////////////////////////////////////////////// // IsLower bool SockAddrIn::IsLower(const SockAddrIn& sin) { // Is it Lower? - ignore 'sin_zero' return (memcmp(&this->sockAddrIn, &sin.sockAddrIn, Size()-sizeof(sockAddrIn.sin_zero)) < 0); } /////////////////////////////////////////////////////////////////////////////// // CreateFrom bool SockAddrIn::CreateFrom(LPCTSTR sAddr, LPCTSTR sService) { sockAddrIn.sin_addr.s_addr = htonl( CSocketComm::GetIPAddress(sAddr) ); sockAddrIn.sin_port = htons( CSocketComm::GetPortNumber( sService ) ); sockAddrIn.sin_family = AF_INET; return true; } /////////////////////////////////////////////////////////////////////////////// // Construct & Destruct //构造函数 CSocketComm::CSocketComm() : m_hComm(INVALID_HANDLE_VALUE), m_hMutex(NULL), m_bServer(false), m_bBroadcast(false), m_hThread(NULL) { } CSocketComm::~CSocketComm() { //停止通信 StopComm(); } /////////////////////////////////////////////////////////////////////////////// // Members /////////////////////////////////////////////////////////////////////////////// /////////////////////////////////////////////////////////////////////////////// // IsOpen bool CSocketComm::IsOpen() const { return ( INVALID_HANDLE_VALUE != m_hComm ); } /////////////////////////////////////////////////////////////////////////////// //判断是否启动 bool CSocketComm::IsStart() const { return ( NULL != m_hThread ); } /////////////////////////////////////////////////////////////////////////////// // IsServer bool CSocketComm::IsServer() const { return m_bServer; } /////////////////////////////////////////////////////////////////////////////// // IsBroadcast bool CSocketComm::IsBroadcast() const { return m_bBroadcast; } /////////////////////////////////////////////////////////////////////////////// // IsSmartAddressing bool CSocketComm::IsSmartAddressing() const { return m_bSmartAddressing; } /////////////////////////////////////////////////////////////////////////////// // GetSocket //获得socket句柄 SOCKET CSocketComm::GetSocket() const { return (SOCKET) m_hComm; } /////////////////////////////////////////////////////////////////////////////// // LockList //进程互斥 void CSocketComm::LockList() { if (NULL != m_hMutex) WaitForSingleObject(m_hMutex, INFINITE); } /////////////////////////////////////////////////////////////////////////////// // UnlockList //解开进程互斥 void CSocketComm::UnlockList() { if (NULL != m_hMutex) ReleaseMutex(m_hMutex); } /////////////////////////////////////////////////////////////////////////////// // AddToList void CSocketComm::AddToList(const SockAddrIn& saddr_in) { LockList(); m_AddrList.insert( m_AddrList.end(), saddr_in ); UnlockList(); } /////////////////////////////////////////////////////////////////////////////// // RemoveFromList void CSocketComm::RemoveFromList(const SockAddrIn& saddr_in) { LockList(); m_AddrList.remove( saddr_in ); UnlockList(); } /////////////////////////////////////////////////////////////////////////////// // SetServerState void CSocketComm::SetServerState(bool bServer) { if (!IsStart()) m_bServer = bServer; } /////////////////////////////////////////////////////////////////////////////// // SetSmartAddressing : Address is included with message //设定 void CSocketComm::SetSmartAddressing(bool bSmartAddressing) { if (!IsStart()) m_bSmartAddressing = bSmartAddressing; } /////////////////////////////////////////////////////////////////////////////// // OnDataReceived /////////////////////////////////////////////////////////////////////////////// // DESCRIPTION: // This function is PURE Virtual, you MUST overwrite it. This is // called every time new data is available. // PARAMETERS: /////////////////////////////////////////////////////////////////////////////// void CSocketComm::OnDataReceived(const LPBYTE lpBuffer, DWORD dwCount) { } /////////////////////////////////////////////////////////////////////////////// // OnEvent /////////////////////////////////////////////////////////////////////////////// // DESCRIPTION: // This function reports events & errors // PARAMETERS: // UINT uEvent: can be one of the event value EVT_(events) /////////////////////////////////////////////////////////////////////////////// void CSocketComm::OnEvent(UINT uEvent) { } /////////////////////////////////////////////////////////////////////////////// // GetPortNumber /////////////////////////////////////////////////////////////////////////////// // DESCRIPTION: // Returns a port number based on service name or port number string // PARAMETERS: // LPCTSTR strServiceName: Service name or port string /////////////////////////////////////////////////////////////////////////////// //获得端口号 USHORT CSocketComm::GetPortNumber( LPCTSTR strServiceName ) { LPSERVENT lpservent; USHORT nPortNumber = 0; if ( _istdigit( strServiceName[0] ) ) { nPortNumber = (USHORT) _ttoi( strServiceName ); } else { #ifdef _UNICODE char pstrService[HOSTNAME_SIZE]; WideCharToMultiByte(CP_ACP, 0, pstrService, -1, strServiceName, sizeof(pstrService), NULL, NULL ); #else LPCTSTR pstrDevice = strServiceName; #endif // 转化网络字节到主机字节 if ( (lpservent = getservbyname( pstrDevice, NULL )) != NULL ) nPortNumber = ntohs( lpservent->s_port ); } return nPortNumber; } /////////////////////////////////////////////////////////////////////////////// // GetIPAddress /////////////////////////////////////////////////////////////////////////////// // DESCRIPTION: // Returns an IP address. // - It tries to convert the string directly // - If that fails, it tries to resolve it as a hostname // PARAMETER