TCP协议代码

TCP ECHO

服务器端

// server.cpp : 定义控制台应用程序的入口点。

#include "stdafx.h"

#pragma comment(lib,"ws2_32.lib")

#include

#include

#include

#include

int _tmain(int argc, _TCHAR* argv[])

{WSAData wsaData;

int err = WSAStartup(WINSOCK_VERSION,&wsaData);

if(0!=err)

{return -1;}

SOCKET sock= socket(AF_INET, SOCK_STREAM, 0);

if(INV ALID_SOCKET==sock) {

printf("socket() Failed: %d\n",WSAGetLastError()); WSACleanup();return -1;}

sockaddr_in LocaAddr;

LocaAddr.sin_family = AF_INET;

LocaAddr.sin_port=htons(10000);

LocaAddr.sin_addr.s_addr=htonl(INADDR_ANY);

err = bind(sock, (sockaddr *)&LocaAddr, sizeof(LocaAddr));

if(SOCKET_ERROR==err) {

printf("bind() Failed: %d\n",WSAGetLastError());

closesocket(sock);WSACleanup();return -1;}

err=listen(sock,5);

if(SOCKET_ERROR==err)

{printf("listen() Failed:%d\n", WSAGetLastError());

closesocket(sock);WSACleanup();return -1;}

sockaddr_in ClientAddr;

int AddrLen =sizeof(ClientAddr);

SOCKET sockConnected=accept(sock,(sockaddr*)

&ClientAddr,&AddrLen);

if(INV ALID_SOCKET==sockConnected)

{printf("accpet() Failed:%d\n", WSAGetLastError());

closesocket(sock);WSACleanup();return -1;} char rbuf[1024];

while(1)

{memset(rbuf,0,1024);

int rByte = recv(sockConnected,rbuf,1024,0);

if(SOCKET_ERROR==rByte) /*出错处理*/

{printf("recv Failed: %d\n",WSAGetLastError());

closesocket(sockConnected);break;}

if(0==rByte)

{printf("Connection is closed.\n"); shutdown

(sockConnected,SD_SEND); closesocket

(sockConnected);break;}

printf("TCP recv [%d]DATA:%s\n",rByte,rbuf);

int sByte=send(sockConnected,rbuf,rByte,0);

if(INV ALID_SOCKET==sByte)

{ printf("send() Failed:%d\n", WSAGetLastError());

closesocket(sockConnected);break;}} closesocket(sock);WSACleanup();return 0;}

客户端

// SOCKET.cpp : 定义控制台应用程序的入口点。

#include "stdafx.h"

#pragma comment(lib,"ws2_32.lib")

#include

#include

#include

#include

int _tmain(int argc, _TCHAR* argv[])

{WSAData wsaData;

int err = WSAStartup(WINSOCK_VERSION,&wsaData); if(0!=err){return -1;}

SOCKET sock;

sock = socket(AF_INET, SOCK_STREAM, 0);

if(INV ALID_SOCKET==sock) //出错处理{printf("socket() Failed: %d\n",WSAGetLastError());

WSACleanup();return -1;}

sockaddr_in ServerAddr;

ServerAddr.sin_family = AF_INET;

ServerAddr.sin_port = htons(10000);

ServerAddr.sin_addr.s_addr=inet_addr("127.0.0.1");

err = connect(sock, (sockaddr *)&ServerAddr, sizeof

(ServerAddr));

if(SOCKET_ERROR==err)

{printf("connect() Failed: %d\n",WSAGetLastError()); closesocket(sock);WSACleanup();return -1;}

char data[]="Hello World!";

int sByte=send(sock,data,strlen(data),0);

if(SOCKET_ERROR==sByte)

{printf("send() Failed: %d\n",WSAGetLastError());

closesocket(sock);WSACleanup();return -1;}

printf("TCP send [%d]DATA:%s\n",sByte,data);

char rbuf[1024];

while(1)

{memset(rbuf,0,1024);

int rByte = recv(sock,rbuf,1024,0);

if(SOCKET_ERROR==rByte)

{printf("recv Failed: %d\n",WSAGetLastError());

closesocket(sock);WSACleanup();return -1;} if(0==rByte) //对方关闭了连接

{printf("Connection is closed.\n");break;}

printf("TCP recv [%d]DATA:%s\n",rByte,rbuf);}

closesocket(sock);WSACleanup();return 0;}