Winsock MSDN Sample (Server)

注意：

1.服务端最好也以控制台方式启动（运行->CMD）

2.参照MSDN官方Winsock示例程序 Site : http://msdn.microsoft.com/en-us/library/ms738545(v=VS.85).aspx

3.处理1个客户端请求

4.可直接用Visual Studio 2010创建空控制台项目，添加CPP文件编译运行

Source Code :

/*
 * File name : WinsockServer.cpp
 * Last updated date : 2011.9.19
 * Author : Dormy.ET.C
 * Description :
 *
 */

#ifndef WIN_LEAN_AND_MEAN
#define WIN_LEAN_AND_MEAN
#endif

#undef UNICODE

// 连接"ws2_32.lib"库
#pragma comment(lib, "ws2_32.lib")
//#pragma comment(lib, "Mswsock.lib")

// 头文件集
#include <WinSock2.h>
#include <Windows.h>
#include <WS2tcpip.h>
#include <IPHlpApi.h>
#include <stdio.h>
#include <stdlib.h>

#define DEFAULT_PORT "27015"
#define DEFAULT_BUFLEN 512

int __cdecl main(int argc, char *argv[])
{
	/*
	 * Name : 重要变量声明
	 * Description :
	 * 1.WSADATA
	 * 2.SOCKET(创建客户端套接字为了接收数据）
	 * 3.BYTE（接收缓冲区长度）
	 */
	WSADATA wsaData;
	int		iResult;

	SOCKET _listenSocket = INVALID_SOCKET;
	SOCKET _clientSocket = INVALID_SOCKET;

	struct addrinfo *pResult = NULL;
	struct addrinfo  hints;

	int		iSendResult;
	char	recvBuf[DEFAULT_BUFLEN];
	int		recvBufLen = DEFAULT_BUFLEN;

	// WSADATA 结构体初始化，并开启一个使用 Winsock.DLL的进程
	// WSAStartup() 成功返回 0，否则返回错误代码
	iResult = ::WSAStartup(MAKEWORD(2, 2), &wsaData);

	// 检查函数调用返回值
	if( iResult != 0 )
	{
		printf("Error WSAStartup(), code : %d\n", iResult);
		return 1;
	}

	ZeroMemory(&hints, sizeof(hints));
	hints.ai_family = AF_INET;
	hints.ai_socktype = SOCK_STREAM;
	hints.ai_protocol = IPPROTO_TCP;
	hints.ai_flags = AI_PASSIVE;

	// 解析服务端地址与端口
	iResult = ::getaddrinfo(NULL, DEFAULT_PORT, &hints, &pResult);

	if( iResult != 0 )
	{
		printf("Error getaddrinfo(), code : %d\n", iResult);
		::WSACleanup();
		return 1;
	}

	// 创建一个连接服务端的套接字（负责监听端口）
	_listenSocket = ::socket(pResult->ai_family, pResult->ai_socktype, pResult->ai_protocol);
	
	if( _listenSocket == INVALID_SOCKET )
	{
		printf("Error socket() - %ld\n", ::WSAGetLastError());
		::freeaddrinfo(pResult);
		::WSACleanup();
		return 1;
	}

	// 设置监听套接字
	iResult = ::bind(_listenSocket, pResult->ai_addr, (int)pResult->ai_addrlen);

	if( iResult == SOCKET_ERROR )
	{
		printf("Error bind(), code : %d\n", ::WSAGetLastError());
		::freeaddrinfo(pResult);
		::closesocket(_listenSocket);
		::WSACleanup();
		return 1;
	}

	::freeaddrinfo(pResult);

	// 开始监听“请求”
	iResult = ::listen(_listenSocket, SOMAXCONN);

	if( iResult == SOCKET_ERROR )
	{
		printf("Error listen(), code : %d\n", ::WSAGetLastError());
		::freeaddrinfo(pResult);
		::closesocket(_listenSocket);
		::WSACleanup();
		return 1;
	}

	// 接受监听套接字监听到的“请求”
	printf("Accept request and ready to deal with the client data package\n");

	_clientSocket = ::accept(_listenSocket, NULL, NULL);

	if( _clientSocket == INVALID_SOCKET )
	{
		printf("Error accept(), code : %d\n", ::WSAGetLastError());
		::closesocket(_listenSocket);
		::WSACleanup();
		return 1;
	}

	// 不需要服务端监听套接字了
	::closesocket(_listenSocket);

	// 接收来自客户端的数据包，并做回应
	do
	{
		iResult = ::recv(_clientSocket, recvBuf, recvBufLen, 0);

		if( iResult > 0 )
		{
			printf("Bytes received : %d\n", iResult);

			iSendResult = ::send(_clientSocket, recvBuf, iResult, 0);

			if( iSendResult == SOCKET_ERROR )
			{
				printf("Error send(), code : %d\n", ::WSAGetLastError());
				::closesocket(_clientSocket);
				::WSACleanup();
				return 1;
			}

			printf("Bytes sent : %d\n", iSendResult);
		}

		else if ( iResult == 0 )
			printf("Connection closing...\n");
		else
		{
			printf("Error recv(), code : %d\n", ::WSAGetLastError());
			::closesocket(_clientSocket);
			::WSACleanup();
			return 1;
		}

	}while(iResult > 0);

	// 关闭连接的客户端
	iResult = ::shutdown(_clientSocket, SD_SEND);

	if( iResult == SOCKET_ERROR )
	{
		printf("Error shutdown(), code : %d\n", ::WSAGetLastError());
		::closesocket(_clientSocket);
		::WSACleanup();
		return 1;
	}

	// 关闭客户端套接字，清除WSADATA对象，释放内存
	::closesocket(_clientSocket);
	::WSACleanup();

	return 0;
}