selss.cpp

select socket 完成的聊天室 简单的代码 完成对select socket的熟悉

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

#include "stdafx.h"
#include <winsock2.h>
#include <iostream>
using namespace std;
#pragma comment(lib, "ws2_32.lib") 
int g_iTotalConn = 0;
int temp = 0;
SOCKET g_CliSocketArr[FD_SETSIZE];
SOCKADDR_IN addr[FD_SETSIZE];


DWORD WINAPI Thread(LPVOID lpParameter);

int _tmain(int argc, _TCHAR* argv[])
{
	WSADATA wsadata;
	SOCKADDR_IN saddr,caddr;
	DWORD dwThreadId;
	
	WSAStartup(MAKEWORD(2,2),&wsadata);
	int icaddrsize =sizeof(SOCKADDR_IN);
	SOCKET s,c;
	s =socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
	if(INVALID_SOCKET  == s)
	{
		cout<<"socket 创建失败!"<<endl;
		return 0;
	}

	saddr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
	saddr.sin_port             = htons(8888);
	saddr.sin_family           = AF_INET;
	
	if (SOCKET_ERROR == bind(s,(struct sockaddr *)&saddr,sizeof(SOCKADDR_IN)))
	{
		cout<<"socket 绑定失败!"<<endl;
		closesocket(s);
		return 0;
	}

	if(SOCKET_ERROR == listen(s,5))
	{
		cout<<"socket 监听失败!"<<endl;
		closesocket(s);
		return 0;
	}

	CreateThread(NULL,0,&Thread,NULL,0,&dwThreadId);

	while(1)
	{
		c =accept(s,(struct sockaddr*)&caddr,&icaddrsize);
	//	cout<<"Accepted client:"<<inet_ntoa(caddr.sin_addr)<<ntohs(caddr.sin_port)<<endl;
		printf("Accepted client:%s:%d\n", inet_ntoa(caddr.sin_addr), ntohs(caddr.sin_port));
		temp =g_iTotalConn;
		g_CliSocketArr[temp] = c;
		addr[temp] = caddr;
		g_iTotalConn++;

	}


	return 0;
}



DWORD WINAPI Thread(LPVOID lpParameter)
{
	int i,ret;
	fd_set fdread;
	struct timeval tv ={0,1};
	char szMessage[1024];
	memset(szMessage,0,1024);

	while(1)
	{
		FD_ZERO(&fdread);
		for( i =0; i< g_iTotalConn; i++)
		{
			FD_SET(g_CliSocketArr[i],&fdread);
		}

		ret = select(0,&fdread,NULL,NULL,&tv);
		if(ret == 0)
		{
	//		closesocket(g_CliSocketArr[i]);
			continue;
		}
		for( i =0; i< g_iTotalConn; i++)
		{
			if(FD_ISSET(g_CliSocketArr[i], &fdread))
			{
				ret = recv(g_CliSocketArr[i], szMessage, 1024, 0);
			//	if(ret == 0 || (ret == SOCKET_ERROR && WSAGetLastError() == WSAECONNRESET))
				if( ret == 0 || ret == SOCKET_ERROR )
				{
					closesocket(g_CliSocketArr[i]);
				//	printf("client away:%s:%d\n", inet_ntoa(addr[i].sin_addr),  ntohs(addr[i].sin_port));

					if(i < g_iTotalConn - 1)
					{
						g_CliSocketArr[i--] = g_CliSocketArr[--g_iTotalConn];
					}
				else
				{
					szMessage[ret] ='\0';
					printf("client:%s:%s\n", inet_ntoa(addr[i].sin_addr), szMessage);

				}
			}
		}
	}
	return 0;
}