#include <winsock2.h>
#include <stdlib.h>
#include <stdio.h>

#define BUFSIZE 4096

// 소켓 함수 오류 출력 후 종료
void err_quit(char *msg)
{
	LPVOID lpMsgBuf;
	FormatMessage( 
		FORMAT_MESSAGE_ALLOCATE_BUFFER|
		FORMAT_MESSAGE_FROM_SYSTEM,
		NULL, WSAGetLastError(),
		MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
		(LPTSTR)&lpMsgBuf, 0, NULL);
	MessageBox(NULL, (LPCTSTR)lpMsgBuf, msg, MB_ICONERROR);
	LocalFree(lpMsgBuf);
	exit(-1);
}

// 소켓 함수 오류 출력
void err_display(char *msg) 
{
	LPVOID lpMsgBuf;
	FormatMessage( 
		FORMAT_MESSAGE_ALLOCATE_BUFFER|
		FORMAT_MESSAGE_FROM_SYSTEM,
		NULL, WSAGetLastError(),
		MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
		(LPTSTR)&lpMsgBuf, 0, NULL);
	printf("[%s] %s", msg, (LPCTSTR)lpMsgBuf);
	LocalFree(lpMsgBuf);
}

// 사용자 정의 데이터 수신 함수
int recvn(SOCKET s, char *buf, int len, int flags)
{
	int received;
	char *ptr = buf;
	int left = len;

	while(left > 0){
		received = recv(s, ptr, left, flags);
		if(received == SOCKET_ERROR) 
			return SOCKET_ERROR;
		else if(received == 0) 
			break;
		left -= received;
		ptr += received;
	}

	return (len - left);
}

DWORD WINAPI ProcessClient(LPVOID arg);
class Cilent_Connect
{
private:
	int retval;
	SOCKET client_sock;		// 접속하는 클라이언트의 소켓
	SOCKADDR_IN clientaddr;	// 접속하는 클라이언트의 구조체
	char buf[BUFSIZE];		// 버퍼
	char filename[256];		// 파일 이름
	int totalbytes;			// 전송 받을 데이터 크기 받기
	int currbytes;			// 이어받기를 할때 필요한 변수, 파일크기가 저장됨.

	void Client_Disconnect();			// 파일을 모두 받았으면 해당 클라이언트를 종료하는 함수
	bool Client_Send_FilePosition();	// 전송 시작할 위치(=현재의 파일 크기) 보내기
	bool Client_Recv_Filesize();		// 클라이언트에게로부터 받으려는 파일 사이즈를 받는다.
	bool Check_File();					// 파일이 이미 있는지 검사하기
	bool File_Download();				// 파일 데이터 받기
public:
	Cilent_Connect(SOCKET client)
	{
		client_sock = client;

		int addrlen = sizeof(clientaddr);
		getpeername(client_sock, (SOCKADDR*)&clientaddr, &addrlen);

		totalbytes = 0;  // 초기화;;
	}
	~Cilent_Connect()
	{
		Client_Disconnect();		// 다운로드 완료시, 연결된 클라이언트의 접속을 끊고 다시 listen 상태로...
	}
	bool Download_Proc()
	{
		if(!Check_File())			// 받으려는 파일이 있는지 확인, 있으면 접속을 끊고 listen 상태로..
			return false;
		if(!Client_Send_FilePosition())	// 받으려는 파일이 서버에 없다면, or 제대로 안받았다면, 그파일의 마지막 지점을 구해서 클라이언트에 보낸다
			return false;
		if(!Client_Recv_Filesize())	// 받으려는 파일의 전체 크기를 받는다.
			return false;
		if(!File_Download())		// 다운로드 시작
			return false;
		return true;
	}

};
class File_Server
{
private:
	int retval;								// 공용 변수 ㅡㅡ.. 대략 어느 함수의 리턴값에 활용
	WSADATA wsa;							// 윈소켓
	SOCKADDR_IN serveraddr;					// 서버 소켓의 구조체
	SOCKET listen_sock;						// 대기중인 서버 소켓
	SOCKET client_sock;						// 접속하는 클라이언트의 소켓
	SOCKADDR_IN clientaddr;					// 접속하는 클라이언트의 구조체
	bool Client_Accept();					// 클라이언트 접근시 처리 함수
	
public:
	File_Server(char *ip, int port);	// socket 선언에서 listen 단계까지 일괄처리 단계
	~File_Server();						// 프로그램 종료시 마무리 작업
	void Server_Loop();					// 서버 루프
};