multichatting_program.zip server.cpp

www.pudn.com > multichatting_program.zip > server.cpp, change:2008-06-12,size:4818b
#include <stdio.h> 
#include <process.h> 
#include <winsock.h> 
#include <windows.h> 
 
#pragma comment(lib, "wsock32.lib")	//윈속 추가 
#define MAXCLIENT 5		// 최대 허용 인원 수 설정 매크로 
#define MAXSOCKETNUM 65535		// 최대 생성 가능 소켓 번호 설정 매크로 
 
void receive_client(void *ns);		// 스레드 함수 프로토 타입 
void SetColor(int color){ //색깔 지정 함수 
	SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), color); 
} 
 
int client_num = 0;			// 클라이언트 갯수 
int num = 0;			// 클라이언트 번호 
char msg_client[] = "번 클라이언트, 환영합니다.\n\0";	// Welcome  
char msg_full[] = "가득찼습니다.\n";		     	// 사용자 초과시 
int client_sock[MAXSOCKETNUM];		// client_sock  
HANDLE hMutex = 0;		// 뮤택스 
 
int main() 
{	 
	SOCKADDR_IN clientaddr; 
 
	int listen_sock; 
	int addrlen; 
 
	SetColor(8); 
	printf("┌───────────────────────────┐\n"); 
	printf("│               Multi Chatting Program                 │\n"); 
	printf("│                    - Server -                        │\n"); 
	 
	 
	int retval; 
 
	// 뮤택스 오브젝트 생성 
	hMutex = CreateMutex(NULL, FALSE, NULL); //인자값(보안관련특성, 뮤텍스 소유자 유무, 생성된 뮤텍스에 이름을 줄건지 여부) 
	if(!hMutex) 
	{ 
		printf("Mutex Error\n"); 
		CloseHandle(hMutex); 
		return -1; 
	} 
 
	// 윈속 초기화 
	WSADATA wsa; 
	if(WSAStartup(MAKEWORD(2,2), &wsa) != 0)	 
		return -1; 
 
	// socket() 
	listen_sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); 
	if (listen_sock == SOCKET_ERROR) 
	{ 
		printf("socket() Error\n"); 
		return -1; 
	} 
 
	// bind() 
	SOCKADDR_IN serveraddr ; 
	serveraddr.sin_family = AF_INET; 
	serveraddr.sin_port = htons(6000); 
	serveraddr.sin_addr.s_addr = htonl(INADDR_ANY); 
 
	retval = bind(listen_sock, (sockaddr*)&serveraddr, sizeof(serveraddr)); 
 
	if( retval == SOCKET_ERROR) 
	{ 
		printf("bind() Error\n"); 
		return -1; 
	} 
 
 
	printf("│           접속 가능한 클라이언트의 수 : %d            │\n", MAXCLIENT - client_num); 
	printf("└───────────────────────────┘\n"); 
	listen(listen_sock,10); 
 
	SetColor(15); 
 
	//클라이언트가 연결되기를 기다린다. 
	while(1) 
	{ 
		//accept() 
		addrlen = sizeof(clientaddr); 
		client_sock[num] = accept(listen_sock, (sockaddr*)&clientaddr, &addrlen); 
 
		//클라이언트의 최대허용갯수까지 받아들인다. 
		if(client_num  MAXCLIENT)	 
		{ 
			_beginthread(receive_client, 0, &client_sock[num]); //스레드 생성 
			Sleep(10); 
			printf("%d번 클라이언트 %s:  %d포트에서 접속하였습니다\n", num, inet_ntoa(clientaddr.sin_addr), ntohs(clientaddr.sin_port)); 
		} 
		//최대허용을 초과시 msg_full메세지를 보내고 접속을 끊는다. 
		else 
		{ 
			addrlen = sizeof(clientaddr); 
			if (client_sock[num] == INVALID_SOCKET) 
			{ 
				printf("accept() Error\n"); 
				closesocket(client_sock[num]); 
				closesocket(listen_sock); 
				WSACleanup(); 
				return -1; 
			} 
			retval = send(client_sock[num], msg_full, sizeof(msg_full), 0); 
			closesocket(client_sock[num]); 
		} 
	} 
	return 0; 
} 
 
void receive_client(void *ns) 
{ 
	int nTmp; 
	char welcome[100] = {0};	 
	char send_clientnum[5] = {0}; 
	char buff[1024] = {0}; 
	char temp[1024] = {0}; 
	int retval, i; 
 
	WaitForSingleObject(hMutex,INFINITE);//뮤텍스를 소유하는 역할. 
	//정상적인 접속시 스레드를 실행하고 클라이언트의 번호를 증가시킴 
	client_num++;			 
	num++;	 
	nTmp = num; 
	printf("접속 가능한 클라이언트의 수 : %d\n", MAXCLIENT - client_num);	 
	ReleaseMutex(hMutex); 
	 
	itoa(num,welcome,10);	 
	itoa(num,send_clientnum,10); 
	strcat(welcome,msg_client);	 
	 
	retval = send(*(SOCKET*)ns, welcome, sizeof(welcome), 0);	//서버접속 성공했다는 메세지 전송 
 
	retval = send(*(SOCKET*)ns, send_clientnum, sizeof(send_clientnum), 0);	//클라이언트 번호 전송 
 
	while(retval != SOCKET_ERROR || retval != INVALID_SOCKET) 
	{ 
		recv(*(SOCKET*)ns, temp, 1024, 0);		// 데이터 받기 
		retval = recv(*(SOCKET*)ns, buff, 1024, 0);		// 데이터 받기 
 
		// broadcast 부분 
		for(i = 0; i  MAXSOCKETNUM; i++) 
		{	 
			// 받은 클라이언트와 보내는 클라이언트의 소켓 메모리 주소가 다를때만 전송 
			WaitForSingleObject(hMutex,INFINITE);		 
			if( ((unsigned*)&client_sock[i] != (SOCKET*)ns)) 
			{ 
				send(client_sock[i], temp, strlen(buff), 0); 
				send(client_sock[i], buff, strlen(buff), 0); 
			} 
			ReleaseMutex(hMutex); 
		} 
		// 서버에 데이터 송수신 정보를 기록 
		 
		if(strlen(buff) != 0) 
		{ 
			printf("%s번 클라이언트 >> ", temp); 
			SetColor(nTmp); 
			printf("%s", buff); 
			SetColor(15); 
		} 
		memset(buff, 0, 1024); 
		memset(temp, 0, 1024); 
	} 
 
 
	// 클라이언트의 연결 해제시 
	WaitForSingleObject(hMutex,INFINITE); 
		client_num--; 
	printf("%d 클라이언트가 접속을 끊었습니다\n접속 가능한 클라이언트의 수 : %d\n", num, MAXCLIENT - client_num); 
	ReleaseMutex(hMutex); 
 
	closesocket(*(int*)ns); 
	 
	return; 
}