#define eom EndOfMessage
#define gc GetCode

int EndOfMessage(char* s) {
	int i = 0;
	while ((s[0] != '\r' || s[1] != '\n') && i < 512 - 2)
		s++, i++;

	if (i < 512 - 2) {
		s[2] = '\0';
		return 0;
	}

	if (s[0] == '\r' && s[1] == '\n') {
		s[0] = '\n';
		s[1] = '\0';
		return 0;
	}

	if (s[1] == '\r')
		return 1;

	return 2;
}

//Lấy mã
char* GetCode(char* message, char* code) {
	char* tmp = code;
	while (message[0] != ' ') {
		code[0] = message[0];
		code++; message++;
	}
	code[0] = '\0';
	return tmp;
}

int Login(SOCKET& sock) {
	char user[256] = { 0 };
	char pass[256] = { 0 };
	char SendBuf[1024];
	char RecvBuf[1024];
	char Code[10];
	int val;

	//Nhap username
	cout << "User: ";
	cin >> user;

	sprintf(SendBuf, "USER %s\r\n", user);
	send(sock, SendBuf, strlen(SendBuf), 0);
	recv(sock, RecvBuf, 1024, 0);
	eom(RecvBuf);
	gc(RecvBuf, Code);
	cout << RecvBuf;

	//Nhap password
	cout << "Pass: ";
	cin >> pass;

	sprintf(SendBuf, "PASS %s\r\n", pass);
	send(sock, SendBuf, strlen(SendBuf), 0);
	recv(sock, RecvBuf, 1024, 0);
	eom(RecvBuf);
	gc(RecvBuf, Code);
	cout << RecvBuf;

	if (strcmp(Code, "230") == 0)
		val = 0;
	else
		val = -1;

	Sleep(1000);
	return val;
}