#pragma comment(lib, "ws2_32")
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#include <WinSock2.h>
#include <stdlib.h>
#include <stdio.h>
#include <iostream>

#define SERVERPORT 9000
#define BUFSIZE 512
#include "error_handle.h"

int main(int argc, char* argv[])
{
	int retval;             // 반환값 중간 저장
    int addrlen;            // client 통신시 SOCKADDR_IN길이를 저장할 변수
	char buf[BUFSIZE + 1];  // 데이터 담을 버퍼
	SOCKADDR_IN serveraddr; // 서버
    SOCKET client_sock;
	SOCKADDR_IN clientaddr;

	ZeroMemory(&serveraddr, sizeof(serveraddr));    // 설정 안할거 전부 0
	serveraddr.sin_family = AF_INET;                // IPv4 TCP/IP
	serveraddr.sin_addr.s_addr = htonl(INADDR_ANY); // 네트워크 데이터정렬 long
	serveraddr.sin_port = htons(SERVERPORT);        // 네트워크 데이터 정렬 short

	WSADATA wsa;
	if (WSAStartup(MAKEWORD(2, 2), &wsa) != 0)      // 윈속 ddl 메모리 올림
		return 1;
	SOCKET listen_sock = socket(AF_INET, SOCK_STREAM, 0);                   // listen소켓 생성
	if (listen_sock == INVALID_SOCKET) err_quit("socket()");    
	retval = bind(listen_sock, (SOCKADDR*)&serveraddr, sizeof(serveraddr)); // listen소켓 port에 바인딩
	if (retval == SOCKET_ERROR) err_quit("bind()");
	retval = listen(listen_sock, SOMAXCONN);                                // listen소켓의 port listen형태로 변환
	if (retval == SOCKET_ERROR) err_quit("listen()");
	while (1)
	{
		addrlen = sizeof(clientaddr);                                           // 이거 왜 여기다가?
		client_sock = accept(listen_sock, (SOCKADDR*)&clientaddr, &addrlen);    // 블락. 클라이언트와 통신 수락 후 기술자 반환
		if (client_sock == INVALID_SOCKET)
		{
			err_display("accept()");
			break;
		}
		std::cout << "\n[TCP 서버] 클라이언트 접속 : IP주소=" << inet_ntoa(clientaddr.sin_addr) << "포트번호=" << ntohs(clientaddr.sin_port) << std::endl;
		while (1)
		{
			retval = recv(client_sock, buf, BUFSIZE, 0);    // 블락. 버퍼에 길이 만큼 데이터 받음
			if (retval == SOCKET_ERROR) {
				err_display("send()");
				break;
			}
			else if (retval == 0)   // recv시 받은 길이가 0이면 클라이언트가 통신 끊은거임 -> 반복문 탈출
				break;
			else
			{
				buf[retval] = '\0';
				std::cout << "[TCP / " << inet_ntoa(clientaddr.sin_addr) << ":" << ntohs(clientaddr.sin_port) << "] " << buf << std::endl;
				retval = send(client_sock, buf, retval, 0); // 블락. 버퍼에 길이만큼의 데이터 보냄
				if (retval == SOCKET_ERROR) {
					err_display("send()");
					break;
				}
				buf[0] = '\0';
			}
		}
		closesocket(client_sock);   // 클라이언트에서 통신 종료 확인시 클라소켓 닫음
		std::cout << "\n[TCP 서버] 클라이언트 종료 : IP주소=" << inet_ntoa(clientaddr.sin_addr) << "포트번호=" << ntohs(clientaddr.sin_port) << std::endl;
	}
	closesocket(listen_sock);       // 서버의 대기가 끝나면 listen소켓을 닫음 
	WSACleanup();   // 윈속 메모리에서 내림
	return 0;
}

int WSAAPI bind(
  [in] SOCKET         s,
  [in] const sockaddr *name,
  [in] int            namelen
);

int WSAAPI listen(
  [in] SOCKET s,
  [in] int    backlog
);

SOCKET WSAAPI accept(
  [in]      SOCKET   s,
  [out]     sockaddr *addr,
  [in, out] int      *addrlen
);

int recv(
  [in]  SOCKET s,
  [out] char   *buf,
  [in]  int    len,
  [in]  int    flags
);

int WSAAPI send(
  [in] SOCKET     s,
  [in] const char *buf,
  [in] int        len,
  [in] int        flags
);