#include <winsock2.h>
#include <stdlib.h>
#include <stdio.h>
#include <sys/timeb.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);
}

class File_Client
{
private:
	int retval;					// 여러 통신간 리턴값을 저장하는 변수
	WSADATA wsa;				// 윈속
	SOCKET sock;				// 클라이언트 소켓
	SOCKADDR_IN serveraddr;		// 소켓 구조체
	struct _timeb start, finish;// 전송 소요 시간을 계산하기 위한 변수

	// 파일 데이터 전송에 사용할 변수
	char filename[256];			// 파일 이름
	int currbytes;				// 서버에서 받은 파일크기
	int totalbytes;				// 파일의 전체 크기
	FILE *fp;					// 파일 구조체

	void File_SendName();		// 서버에 어떤파일을 보낸다고 통보
	void File_RecvSize();		// 같은 파일이 있다면, 그 파일의 크기가 오고, 아니면 0이 옴
	bool File_SendSize();		// 보낼 파일 사이즈 정해서 서버에 통해
	void File_Upload();			// 파일 업로드 과정

public:
	File_Client();	// 파일 클라이언트 connect 작업까지 처리
	~File_Client();						// 기타 소켓 종료등. 마무리
	void File_SendProc();				// 구동 부분
};