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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > Internet/IE编程 > 查看源码
DASocket.cpp
资源名称：IEC101-104RTU.rar　[点击查看]
上传用户：asikq0571
上传日期：2014-07-12
资源大小：528k
文件大小：13k
源码类别：

Internet/IE编程

开发平台：

Visual C++

DASocket.cpp：源码内容
  1. // DASocket.cpp: implementation of the CDASrvCnSocket class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #include "stdafx.h"
  6. #include "DASocket.h"
  7. #ifdef _DEBUG
  8. #undef THIS_FILE
  9. static char THIS_FILE[]=__FILE__;
  10. #define new DEBUG_NEW
  11. #endif
  13. /////////////////////////////////////////////////////////////////////////////
  14. //网络常用操作函数
  15. int GetLocalHostName(CString &sHostName)
  16. {
  17. char szHostName[256];
  18. int nRetCode;
  19. nRetCode=gethostname(szHostName,sizeof(szHostName));
  20. if(nRetCode!=0)
  21. {
  22. //产生错误
  23. sHostName=_T("没有取得本地用户名");
  24. return GetLastError();
  25. }
  26. sHostName=szHostName;
  27. return 0;
  28. }
  30. int GetIpAddress(const CString &sHostName, CString &sIpAddress)//获得本地IP
  31. {
  32. struct hostent FAR * lpHostEnt=gethostbyname(sHostName);
  33. if(lpHostEnt==NULL)
  34. {
  35. //产生错误
  36. sIpAddress=_T("");
  37. return GetLastError();
  38. }
  39. //获取IP
  40. LPSTR lpAddr=lpHostEnt->h_addr_list[0];
  41. if(lpAddr)
  42. {
  43. struct in_addr inAddr;
  44. memmove(&inAddr,lpAddr,4);
  45. //转换为标准格式
  46. sIpAddress=inet_ntoa(inAddr);
  47. if(sIpAddress.IsEmpty())
  48. sIpAddress=_T("没有取得本地IP地址");
  49. }
  50. return 0;
  51. }
  53. int GetNamebyAddress(const CString &IpAddress,CString &sYouName)//获得对方计算机名称
  54. {
  55. unsigned long addr;
  56. addr=inet_addr(IpAddress);
  57. struct hostent FAR * lpHostEnt=gethostbyaddr((char *)&addr,4,AF_INET);
  58. if(lpHostEnt==NULL)
  59. {
  60. //产生错误
  61. sYouName=_T("");
  62. AfxMessageBox("连接不上");//应该取得其错误
  63. return -1;
  64. }
  65. CString name=lpHostEnt->h_name;
  66. sYouName=name;
  67. return 0;
  68. }
  70. CString GetNetError(DWORD error) //返回错误信息
  71. {
  72. CString strError;
  73. switch(error)
  74. {
  75. case WSANOTINITIALISED:
  76. strError="初始化错误";
  77. break;
  78. case WSAENOTCONN:
  79. strError="对方没有启动";
  80. break;
  81. case WSAEWOULDBLOCK :
  82. strError="对方已经关闭";
  83. break;
  84. case WSAECONNREFUSED:
  85. strError="连接的尝试被拒绝";
  86. break;
  87. case WSAENOTSOCK:
  88. strError="在一个非套接字上尝试了一个操作";
  89. break;
  90. case WSAEADDRINUSE:
  91. strError="特定的地址已在使用中";
  92. break;
  93. case WSAECONNRESET:
  94. strError="与主机的连接被关闭";
  95. break;
  96. default:
  97. strError="一般错误";
  98. }
  99. return strError;
  100. }
  101. /////////////////////////////////////////////////////////////////////////////
  104. /////////////////////////////////////////////////////////////////////////////
  105. // CDASrvCnSocket
  107. CDASrvCnSocket::CDASrvCnSocket(CDAServerSocket* pOwner)
  108. {
  109. m_pOwner = pOwner;
  110. m_pEvtReceive = NULL;      
  111. m_pEvtSend = NULL;         
  112. m_pEvtOutOfBandData = NULL;
  113. m_pEvtAccept = NULL;       
  114. m_pEvtConnect = NULL;      
  115. m_pEvtClose = NULL;        
  116. }
  118. CDASrvCnSocket::~CDASrvCnSocket()
  119. {
  120. }
  123. // Do not edit the following lines, which are needed by ClassWizard.
  124. #if 0
  125. BEGIN_MESSAGE_MAP(CDASrvCnSocket, CAsyncSocket)
  126. //{{AFX_MSG_MAP(CDASrvCnSocket)
  127. //}}AFX_MSG_MAP
  128. END_MESSAGE_MAP()
  129. #endif // 0
  131. /////////////////////////////////////////////////////////////////////////////
  132. // CDASrvCnSocket member functions
  134. void CDASrvCnSocket::OnReceive(int nErrorCode) 
  135. {
  136. // TODO: Add your specialized code here and/or call the base class
  137. if(m_pEvtReceive)
  138. {
  139. (m_pOwner->*m_pEvtReceive)(this, nErrorCode);
  140. }
  141. AsyncSelect(FD_READ | FD_CLOSE);
  142. CAsyncSocket::OnReceive(nErrorCode);
  143. }
  145. void CDASrvCnSocket::OnClose(int nErrorCode) 
  146. {
  147. // TODO: Add your specialized code here and/or call the base class
  148. if(m_pEvtClose)
  149. (m_pOwner->*m_pEvtClose)(this, nErrorCode);
  150. CAsyncSocket::OnClose(nErrorCode);
  151. }
  153. void CDASrvCnSocket::OnConnect(int nErrorCode) 
  154. {
  155. // TODO: Add your specialized code here and/or call the base class
  156. if(m_pEvtConnect)
  157. {
  158. (m_pOwner->*m_pEvtConnect)(this, nErrorCode);
  159. }
  160. AsyncSelect(FD_READ | FD_CLOSE);
  161. CAsyncSocket::OnConnect(nErrorCode);
  162. }
  164. void CDASrvCnSocket::OnOutOfBandData(int nErrorCode) 
  165. {
  166. // TODO: Add your specialized code here and/or call the base class
  167. if(m_pEvtOutOfBandData)
  168. (m_pOwner->*m_pEvtOutOfBandData)(this, nErrorCode);
  169. CAsyncSocket::OnOutOfBandData(nErrorCode);
  170. }
  172. void CDASrvCnSocket::OnSend(int nErrorCode) 
  173. {
  174. // TODO: Add your specialized code here and/or call the base class
  175. if(m_pEvtSend)
  176. {
  177. (m_pOwner->*m_pEvtSend)(this, nErrorCode);
  178. }
  179. AsyncSelect(FD_READ | FD_CLOSE);
  180. CAsyncSocket::OnSend(nErrorCode);
  181. }
  183. BOOL CDASrvCnSocket::SendData(char *pSendBuffer, int nBufLen, int& dwBytesSended)
  184. {
  185. if(nBufLen < 0) return FALSE;
  186. int iRemainBytes = nBufLen;
  187. int iTotalSentBytes = 0;
  188. int iCurSent = 0;
  189. while(iRemainBytes)
  190. {
  191. iCurSent = Send(pSendBuffer + iTotalSentBytes,iRemainBytes);
  192. if(iCurSent < 0)
  193. {
  194. TRACE("Send Error!n");
  195. return FALSE;
  196. }
  197. iTotalSentBytes += iCurSent;
  198. iRemainBytes -= iCurSent; 
  199. }
  200. dwBytesSended=iTotalSentBytes;
  201. return TRUE;
  203. }
  205. BOOL CDASrvCnSocket::RecvData(char *pRecvBuffer, int nBufLen, int& dwBytesReaded)
  206. {
  207. if(nBufLen < 0) return FALSE;
  208. int iRemainBytes = nBufLen;
  209. int iTotalRecvBytes = 0;
  210. int iCurRecv = 0;
  211. while(iRemainBytes)
  212. {
  213. iCurRecv = Receive(pRecvBuffer + iTotalRecvBytes,iRemainBytes);
  214. if(iCurRecv < 0)
  215. {
  216. TRACE("Receive Error!n");
  217. return FALSE;
  218. }
  219. iTotalRecvBytes += iCurRecv;
  220. iRemainBytes -= iCurRecv; 
  221. }
  222. dwBytesReaded=(DWORD)iTotalRecvBytes;
  223. return TRUE;
  224. }
  226. /////////////////////////////////////////////////////////////////////////////
  227. // CDAServerSocket
  229. CDAServerSocket::CDAServerSocket(HWND pOwner)
  230. {
  231. m_hOwner = pOwner;
  232. m_bHasOpened = FALSE;
  233. m_bConnected = FALSE;
  234. }
  236. CDAServerSocket::~CDAServerSocket()
  237. {
  238. for(int i = 0; i < m_arConnections.GetSize(); i++) //释放所有连接
  239. {
  240. delete m_arConnections[i];
  241. }
  242. }
  245. // Do not edit the following lines, which are needed by ClassWizard.
  246. #if 0
  247. BEGIN_MESSAGE_MAP(CDAServerSocket, CAsyncSocket)
  248. //{{AFX_MSG_MAP(CDAServerSocket)
  249. //}}AFX_MSG_MAP
  250. END_MESSAGE_MAP()
  251. #endif // 0
  253. BOOL CDAServerSocket::OpenPort(UINT nSocketPort, const CString szAddr)
  254. {
  255. m_nSocketPort = nSocketPort;
  256. m_szAddr = szAddr;
  257. AfxSocketInit();
  258. if(!Create(nSocketPort))//, SOCK_STREAM, FD_ACCEPT))//, szAddr))
  259. {
  260. return FALSE;
  261. }
  262. if(!Listen()) 
  263. {
  264. Close();
  265. return FALSE;
  266. }
  267. m_bHasOpened = TRUE;
  268. return TRUE;
  269. }
  271. void CDAServerSocket::ClosePort()
  272. {
  273. CDASrvCnSocket* pSrvCnSocket = NULL;
  274. Close();
  275. m_bConnected = FALSE;
  276. for(int i = 0; i < m_arConnections.GetSize(); i++)
  277. {
  278. pSrvCnSocket = m_arConnections.ElementAt(i);
  279. delete pSrvCnSocket;
  280. pSrvCnSocket = NULL;
  281. }
  282. m_arConnections.RemoveAll();
  283. }
  284. /////////////////////////////////////////////////////////////////////////////
  285. // CDAServerSocket member functions
  286. void CDAServerSocket::OnAccept(int nErrorCode) 
  287. {
  288. // TODO: Add your specialized code here and/or call the base class
  289. if(!nErrorCode) 
  290. {
  291. CDASrvCnSocket* pcnSocket;
  292. SOCKADDR_IN client;
  293. int iAddrSize=sizeof(client);
  294. pcnSocket = new CDASrvCnSocket(this);
  295. if(Accept(*pcnSocket,(SOCKADDR *)&client,&iAddrSize))
  296. {
  297. pcnSocket->AsyncSelect(FD_READ|FD_WRITE|FD_CLOSE);
  298. pcnSocket->m_pEvtReceive = ReceiveEvt;
  299. pcnSocket->m_pEvtSend = SendEvt;
  300. pcnSocket->m_pEvtClose = CloseEvt;
  301. pcnSocket->m_strRemoIP = inet_ntoa(client.sin_addr);
  302. pcnSocket->m_nRemoPort = client.sin_port;
  303. m_arConnections.Add(pcnSocket);
  304. m_bConnected = TRUE;
  305. SendMessage(m_hOwner, WM_TCP_SERVER_CONNECTED, (WPARAM)nErrorCode, (LPARAM)pcnSocket);
  306. } 
  307. else 
  308. {
  309. delete pcnSocket;
  310. pcnSocket = NULL;
  311. }
  312. }
  313. AsyncSelect(FD_ACCEPT|FD_CLOSE);
  314. CAsyncSocket::OnAccept(nErrorCode);
  315. }
  317. void CDAServerSocket::ReceiveEvt(CAsyncSocket* pSrvCnSocket, int nErrorCode)
  318. {
  319. SendMessage(m_hOwner, WM_TCP_SERVER_RECEIVE, (WPARAM)nErrorCode, (LPARAM)pSrvCnSocket);
  320. }
  322. void CDAServerSocket::SendEvt(CAsyncSocket* pSrvCnSocket, int nErrorCode)
  323. {
  324. SendMessage(m_hOwner, WM_TCP_SERVER_SEND, (WPARAM)nErrorCode, (LPARAM)pSrvCnSocket);
  325. }
  327. void CDAServerSocket::CloseEvt(CAsyncSocket* pSrvCnSocket, int nErrorCode)
  328. {
  329. SendMessage(m_hOwner, WM_TCP_SERVER_CLOSED, (WPARAM)nErrorCode, (LPARAM)pSrvCnSocket);
  331. for(int i = 0; i < m_arConnections.GetSize(); i++)
  332. {
  333. if(m_arConnections[i]->m_hSocket == pSrvCnSocket->m_hSocket)
  334. {
  335. delete pSrvCnSocket;
  336. pSrvCnSocket = NULL;
  337. m_arConnections.RemoveAt(i);
  339. if(m_arConnections.GetSize() == 0)
  340. {
  341. m_bConnected = FALSE;
  342. }
  343. return;
  344. }
  345. }
  346. }
  347. /////////////////////////////////////////////////////////////////////////////
  348. // CDAClientSocket
  350. CDAClientSocket::CDAClientSocket(HWND pOwner)
  351. {
  352. m_pOwner = pOwner;
  353. m_bHasOpened = FALSE;
  354. m_bConnected = FALSE;
  355. }
  357. CDAClientSocket::~CDAClientSocket()
  358. {
  359. if(m_hSocket != INVALID_SOCKET) Close();
  360. }
  362. BOOL CDAClientSocket::OpenPort(UINT nHostPort, const CString szHostAddr)
  363. {
  364. m_nHostPort = nHostPort;
  365. m_szHostAddr = szHostAddr;
  366. if(m_hSocket != INVALID_SOCKET)
  367. {
  368. Close();
  369. }
  370. if(!Create(0,SOCK_STREAM,FD_READ |FD_CONNECT | FD_CLOSE))
  371. {
  372. m_bHasOpened = FALSE;
  373. m_bConnected = FALSE;
  374. return FALSE;
  375. }
  377.    Connect(szHostAddr, nHostPort);
  379. return TRUE;
  380. }
  382. void CDAClientSocket::ClosePort()
  383. {
  384. Close();
  385. }
  387. // Do not edit the following lines, which are needed by ClassWizard.
  388. #if 0
  389. BEGIN_MESSAGE_MAP(CDAClientSocket, CAsyncSocket)
  390. //{{AFX_MSG_MAP(CDAClientSocket)
  391. //}}AFX_MSG_MAP
  392. END_MESSAGE_MAP()
  393. #endif // 0
  395. /////////////////////////////////////////////////////////////////////////////
  396. // CDAClientSocket member functions
  398. void CDAClientSocket::OnConnect(int nErrorCode) 
  399. {
  400. // TODO: Add your specialized code here and/or call the base class
  401. if(!nErrorCode) 
  402. {
  403. m_bHasOpened = TRUE;
  404. m_bConnected = TRUE;
  406.        SendMessage(m_pOwner, WM_TCP_CLIENT_CONNECTED, (WPARAM)nErrorCode, (LPARAM)this);
  407. }
  408. else
  409.         AfxMessageBox("无法连接指定的服务器网络端口，请重试！");
  411. AsyncSelect(FD_READ | FD_CLOSE);
  412. CAsyncSocket::OnConnect(nErrorCode);
  413. }
  415. void CDAClientSocket::OnReceive(int nErrorCode) 
  416. {
  418. SendMessage(m_pOwner, WM_TCP_CLIENT_RECEIVE, (WPARAM)nErrorCode, (LPARAM)this);
  419. AsyncSelect(FD_READ | FD_CLOSE);
  420. CAsyncSocket::OnReceive(nErrorCode);
  421. }
  423. void CDAClientSocket::OnSend(int nErrorCode) 
  424. {
  425. SendMessage(m_pOwner, WM_TCP_CLIENT_SEND, (WPARAM)nErrorCode, (LPARAM)this);
  426. AsyncSelect(FD_READ | FD_CLOSE);
  427. CAsyncSocket::OnSend(nErrorCode);
  428. }
  430. void CDAClientSocket::OnClose(int nErrorCode) 
  431. {
  432.     // TODO: Add your specialized code here and/or call the base class
  433. m_bHasOpened = FALSE;
  434. m_bConnected = FALSE;
  435. SendMessage(m_pOwner, WM_TCP_CLIENT_CLOSED, (WPARAM)nErrorCode, (LPARAM)this);
  436. CAsyncSocket::OnClose(nErrorCode);
  438. }
  440. BOOL CDAClientSocket::SendData(char *pSendBuffer, int nBufLen)
  441. {
  442. if(nBufLen < 0) return FALSE;
  443. int iRemainBytes = nBufLen;
  444. int iTotalSentBytes = 0;
  445. int iCurSent = 0;
  446. while(iRemainBytes)
  447. {
  448. iCurSent = Send(pSendBuffer + iTotalSentBytes,iRemainBytes);
  449. if(iCurSent < 0)
  450. {
  451. TRACE("Send Error!n");
  452. return FALSE;
  453. }
  454. iTotalSentBytes += iCurSent;
  455. iRemainBytes -= iCurSent; 
  456. }
  457. return TRUE;
  458. }
  460. BOOL CDAClientSocket::RecvData(char *pRecvBuffer, int nBufLen)
  461. {
  462. if(nBufLen < 0) return FALSE;
  463. int iRemainBytes = nBufLen;
  464. int iTotalRecvBytes = 0;
  465. int iCurRecv = 0;
  466. while(iRemainBytes)
  467. {
  468. iCurRecv = Receive(pRecvBuffer + iTotalRecvBytes,iRemainBytes);
  469. if(iCurRecv < 0)
  470. {
  471. TRACE("Receive Error!n");
  472. return FALSE;
  473. }
  474. iTotalRecvBytes += iCurRecv;
  475. iRemainBytes -= iCurRecv; 
  476. }
  477. return TRUE;
  479. }
  481. ///////////////////////////////////////
  482. ///////////////////////////////////////
  483. CDAUDPSocket::CDAUDPSocket(HWND pOwner)
  484. {
  485. m_pOwner = pOwner;
  486. m_bHasOpened = FALSE;
  487. }
  489. CDAUDPSocket::~CDAUDPSocket()
  490. {
  491. if(m_hSocket != INVALID_SOCKET) Close();
  492. }
  494. BOOL CDAUDPSocket::OpenPort(UINT nLocalPort)
  495. {
  496. m_nLocalPort = nLocalPort;
  497. if(m_hSocket != INVALID_SOCKET)
  498. {
  499. Close();
  500. }
  501. if(!Create(nLocalPort,SOCK_DGRAM,FD_READ |FD_CONNECT | FD_CLOSE))
  502. {
  503. m_bHasOpened = FALSE;
  504. return FALSE;
  505. }
  506. m_bHasOpened = TRUE;
  507. return TRUE;
  508. }
  510. void CDAUDPSocket::ClosePort()
  511. {
  512. Close();
  513. }
  514. // Do not edit the following lines, which are needed by ClassWizard.
  515. #if 0
  516. BEGIN_MESSAGE_MAP(CDAUDPSocket, CAsyncSocket)
  517. //{{AFX_MSG_MAP(CDAUDPSocket)
  518. //}}AFX_MSG_MAP
  519. END_MESSAGE_MAP()
  520. #endif // 0
  522. void CDAUDPSocket::OnReceive(int nErrorCode) 
  523. {
  525. SendMessage(m_pOwner, WM_UDP_RECEIVE, (WPARAM)nErrorCode, (LPARAM)this);
  526. AsyncSelect(FD_READ | FD_CLOSE);
  527. CAsyncSocket::OnReceive(nErrorCode);
  528. }
  530. void CDAUDPSocket::OnSend(int nErrorCode) 
  531. {
  532. SendMessage(m_pOwner, WM_UDP_SEND, (WPARAM)nErrorCode, (LPARAM)this);
  533. AsyncSelect(FD_READ | FD_CLOSE);
  534. CAsyncSocket::OnSend(nErrorCode);
  535. }
  537. void CDAUDPSocket::OnClose(int nErrorCode) 
  538. {
  540. // TODO: Add your specialized code here and/or call the base class
  541. m_bHasOpened = FALSE;
  542. SendMessage(m_pOwner, WM_UDP_CLOSED, (WPARAM)nErrorCode, (LPARAM)this);
  543. CAsyncSocket::OnClose(nErrorCode);
  545. }