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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 查看源码
COMM.C
资源名称：MSDN_VC98.zip　[点击查看]
上传用户：bangxh
上传日期：2007-01-31
资源大小：42235k
文件大小：3k
源码类别：

Windows编程

开发平台：

Visual C++

COMM.C：源码内容
  1. /*++
  3. Copyright 1996-1997 Microsoft Corporation
  5. Module Name:
  7.     comm.c
  9. Abstract:
  11.     Implements a set of common operations for socket communication
  13. Revision History:
  15. --*/
  17. #include <windows.h>
  18. #include <winsock.h>
  19. #include <stdio.h>
  20. #include <stdlib.h>
  21. #include "comm.h"
  23. BOOL InitWinsock ()
  24. {
  25. int nRes;
  26. WSADATA wsaData;
  27. WORD wVerRequested = 0x0101; // ver 1.1
  29. // Init the sockets interface
  30. nRes = WSAStartup (wVerRequested, &wsaData);
  31. if (nRes)  {
  32. fprintf (stderr, "Couldn't init winsock: %dn", nRes);
  33. return(FALSE);
  34. }
  36. return(TRUE);
  37. }
  39. BOOL TermWinsock ()
  40. {
  41. if (SOCKET_ERROR == WSACleanup ())
  42. return(FALSE);
  43. else
  44. return(TRUE);
  45. }
  47. BOOL SendMsg (SOCKET s, PBYTE pBuf, DWORD cbBuf)
  48. /*++
  50.  Routine Description:
  52.     Sends a message over the socket by first sending a DWORD that
  53. represents the size of the message followed by the message itself.
  55.  Return Value:
  57.     Returns TRUE is successful; otherwise FALSE is returned.
  59. --*/
  60. {
  61. if (0 == cbBuf)
  62. return(TRUE);
  64. // send the size of the message
  65. //
  66. if (!SendBytes (s, (PBYTE)&cbBuf, sizeof (cbBuf)))
  67. return(FALSE);
  69. // send the body of the message
  70. //
  71. if (!SendBytes (s, pBuf, cbBuf))
  72. return(FALSE);
  74. return(TRUE);
  75. }
  77. BOOL ReceiveMsg (SOCKET s, PBYTE pBuf, DWORD cbBuf, DWORD *pcbRead)
  78. /*++
  80.  Routine Description:
  82.     Receives a message over the socket.  The first DWORD in the message
  83. will be the message size.  The remainder of the bytes will be the
  84. actual message.
  86.  Return Value:
  88.     Returns TRUE is successful; otherwise FALSE is returned.
  90. --*/
  91. {
  92. DWORD cbRead;
  93. DWORD cbData;
  95. // find out how much data is in the message
  96. //
  97. if (!ReceiveBytes (s, (PBYTE)&cbData, sizeof (cbData), &cbRead))
  98. return(FALSE);
  100. if (sizeof (cbData) != cbRead)
  101. return(FALSE);
  103. // Read the full message
  104. //
  105. if (!ReceiveBytes (s, pBuf, cbData, &cbRead))
  106. return(FALSE);
  108. if (cbRead != cbData)
  109. return(FALSE);
  111. *pcbRead = cbRead;
  113. return(TRUE);
  114. }
  116. BOOL SendBytes (SOCKET s, PBYTE pBuf, DWORD cbBuf)
  117. {
  118. PBYTE pTemp = pBuf;
  119. int cbSent, cbRemaining = cbBuf;
  121. if (0 == cbBuf)
  122. return(TRUE);
  124. while (cbRemaining) {
  125. cbSent = send (s, pTemp, cbRemaining, 0);
  126. if (SOCKET_ERROR == cbSent) {
  127. fprintf (stderr, "send failed: %un", GetLastError ());
  128. return FALSE;
  129. }
  131. pTemp += cbSent;
  132. cbRemaining -= cbSent;
  133. }
  135. return TRUE;
  136. }
  138. BOOL ReceiveBytes (SOCKET s, PBYTE pBuf, DWORD cbBuf, DWORD *pcbRead)
  139. {
  140. PBYTE pTemp = pBuf;
  141. int cbRead, cbRemaining = cbBuf;
  143. while (cbRemaining) {
  144. cbRead = recv (s, pTemp, cbRemaining, 0);
  145. if (0 == cbRead)
  146. break;
  148. if (SOCKET_ERROR == cbRead) {
  149. fprintf (stderr, "recv failed: %un", GetLastError ());
  150. return FALSE;
  151. }
  153. cbRemaining -= cbRead;
  154. pTemp += cbRead;
  155. }
  157. *pcbRead = cbBuf - cbRemaining;
  159. return TRUE;
  160. }