WinSock客户端和服务端（select模型）

客户端：

#include <stdio.h>
#include <winsock2.h>
#include <iostream>

using namespace std;

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


const char SERVER[] = "172.18.9.71";
const int PORT = 5555;
const int MSGSIZE = 1024;


int main()
{
	WSADATA wsaData;
	SOCKET sClient;
	SOCKADDR_IN server;
	char szMessage[MSGSIZE];
	int ret;

	WSAStartup(0x0202, &wsaData);

	sClient = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

	memset(&server, 0, sizeof(SOCKADDR_IN));
	server.sin_family = AF_INET;
	server.sin_addr.S_un.S_addr = inet_addr(SERVER);
	server.sin_port = htons(PORT);

    cout << "Waiting for connect!" << endl;
	if (connect(sClient, (struct sockaddr*)(&server), sizeof(SOCKADDR_IN)) == 0)
	{
	    cout << "connect!!~" << endl;
		while (TRUE)
		{
			cout << "send:" << endl;
			cin >> szMessage;

			send(sClient, szMessage, strlen(szMessage), 0);

			ret = recv(sClient, szMessage, MSGSIZE, 0);
			szMessage[ret] = '\0';

			cout << "receive: " << szMessage << endl;
		}
	}

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

服务端：

#include <stdio.h>
#include <process.h>
#include <winsock.h>
#include <iostream>

using namespace std;

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

const char SERVER[] = "172.18.9.71";
const int PORT = 5555;
const int MSGSIZE = 1024;

int cnt = 0;
SOCKET sockets[FD_SETSIZE];

DWORD WINAPI WorkerThread(LPVOID lpParam)
{
	fd_set fdread;
	int ret;
	struct timeval tv = { 1, 0 };
	char szMessage[MSGSIZE];
	while (TRUE)
	{
		FD_ZERO(&fdread);
		for (int i = 0; i < cnt; i++)
		{
			FD_SET(sockets[i], &fdread);//将套接口加入集合中
		}
		//检查可读性的套接口
		ret = select(0, &fdread, NULL, NULL, &tv);

		if (ret == 0)
		{
		    cout << "ret == 0?" << endl;
			continue;
		}

		for (int i = 0; i < cnt; i++)
		{
			if (FD_ISSET(sockets[i], &fdread)) //如果可读, 奶蛋的，写错了。
			{
				ret = recv(sockets[i], szMessage, MSGSIZE, NULL);
				if (ret == 0 || (ret == SOCKET_ERROR && WSAGetLastError() == WSAECONNRESET))
				{
					printf("Client socket %d closed.\n", sockets[i]);
					closesocket(sockets[i]);
					if (i < cnt - 1)
					{
						sockets[i--] = sockets[--cnt];
					}
				}
				else
				{
					szMessage[ret] = '\0';
					cout << "The message that client send is :" << szMessage << endl;
					char szBuffer[1001];
					sprintf(szBuffer, "I have bean received your messages, %s", szMessage);
					send(sockets[i], szBuffer, strlen(szBuffer), NULL);
				}
			}
		}

	}
	return 0;
}


int main()
{
	WSADATA wsaData; //WSADATA framework 用于存储 Windows 套接字调用返回的初始化信息
	SOCKET sListen, sClient;
	SOCKADDR_IN local, client;
	char szMessage[MSGSIZE];
	int ret;
	DWORD dwThreadId;
	int iaddrSize = sizeof(SOCKADDR_IN);

	WSAStartup(0x0202, &wsaData); //初始化ws2_32.lib

	//create listening socket
	sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

	//bind
	local.sin_family = AF_INET;
	local.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
	local.sin_port = htons(PORT);
	if (bind(sListen, reinterpret_cast<sockaddr*>(&local), sizeof(SOCKADDR_IN)) == SOCKET_ERROR)
    {
        cout << "bind failed!" << endl;
    }
	//listen
	if (listen(sListen, 3) == SOCKET_ERROR)
    {
        cout << "listen failed!" << endl;
    }
	//create WorkerThread
	CreateThread(NULL, 0, WorkerThread, NULL, 0, &dwThreadId);
	cout << "PID = " << dwThreadId << endl;

	while (TRUE)
	{
		//accept a connection
		sClient = accept(sListen, reinterpret_cast<sockaddr*>(&client), &iaddrSize);
		if (sClient == INVALID_SOCKET)
        {
            cout << "accept error!" << endl;
            break;
        }
        else
        {
            printf("Accept client: %s: %d\n", inet_ntoa(client.sin_addr), ntohs(client.sin_port));
            //加入到socket的数组中
            sockets[cnt++] = sClient;
        }

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