#include <iostream>
#include <stdio.h>
#include <winsock2.h> // Wincosk2.h должен быть раньше windows!
#include <windows.h>

#include <iostream>
#include <conio.h>
#include <fstream>

using namespace std; 

#define MAX_PACKET_SIZE         65535
char* ip="127.0.0.1";
int port=5050;
SOCKET my_sock;

int otpravka_i_polysenie_dannih_CLIENT(string dannie_dla_otpravki, int MAX_count_popitok, int PAUSE_mezdy_popitok);
int recv_all(SOCKET s,  char* buf, int len);
int  otpravka_soobsenia(string text_msg, int MAX_count_popitok, int PAUSE_mezdy_popitok);
int polychenie_soobsenia(char buff[MAX_PACKET_SIZE], int MAX_count_popitok, int &len, int PAUSE_mezdy_popitok);
int podklysenie_k_servery(int MAX_count_popitok, int PAUSE_mezdy_popitok);


int main()
{
	int count_popitok=0, MAX_count_popitok=5, PAUSE_mezdy_popitok=2000, result=1;
	string dannie_dla_otpravki;

	printf("TCP DEMO CLIENT\n");

	result=podklysenie_k_servery(MAX_count_popitok, PAUSE_mezdy_popitok);
	
	if(result==1){
	printf("Soedinenie s %s uspeshno ustanovlenno\n\n", ip);
    dannie_dla_otpravki="MSG OT KLIENTA \n";
	result=otpravka_i_polysenie_dannih_CLIENT(dannie_dla_otpravki, MAX_count_popitok, PAUSE_mezdy_popitok);
		}	

	
	if(result==-1){MessageBoxA(NULL, "Ошибка", "Работа с сервером завершена неудачно", MB_OK);}

	return -1;

}

int otpravka_i_polysenie_dannih_CLIENT(string dannie_dla_otpravki, int MAX_count_popitok, int PAUSE_mezdy_popitok){

	int result=1;

	char  buff[MAX_PACKET_SIZE];

	Sleep(10000);

	int ret=0, i=0;
	

	result=otpravka_soobsenia(dannie_dla_otpravki, MAX_count_popitok, PAUSE_mezdy_popitok);
	
	int len=0;

	if(result==1){
 	while(strcmp(buff, "quit")!=0){ 
 		for (i = 0; i<MAX_PACKET_SIZE; i++){buff[i]=0;}
 
 		result=polychenie_soobsenia(buff, MAX_count_popitok, len, PAUSE_mezdy_popitok);
 		for (i = 0; i<len; i++){printf ("%c", buff [i]);}
 
 		if(len>0&&result==1){	
       	result=otpravka_soobsenia("quit", MAX_count_popitok, PAUSE_mezdy_popitok);
		}
	
	if(result==-1){break;}
	}
	
	
	
	}

closesocket(my_sock);
WSACleanup();

Sleep(10000);

return result;
}
int podklysenie_k_servery(int MAX_count_popitok, int PAUSE_mezdy_popitok){

	int result=-1, count_popitok=0;
	char buff[MAX_PACKET_SIZE];


	while(result==-1&&count_popitok<=MAX_count_popitok){
		
		result=1;

	// Шаг 1 - инициализация библиотеки Winsock
	if (WSAStartup(0x202, (WSADATA *)&buff[0]))
	{
		printf("WSAStart error %d\n", WSAGetLastError());
		result=-1;
	}
	// Шаг 2 - создание сокета
	//SOCKET my_sock;
	my_sock = socket(AF_INET, SOCK_STREAM, 0);
	if (my_sock < 0)
	{
		WSACleanup();
		printf("Socket() error %d\n", WSAGetLastError());
		result=-1;
	}

	// Шаг 3 - установка соединения
	// заполнение структуры sockaddr_in - указание адреса и порта сервера
	sockaddr_in dest_addr;
	dest_addr.sin_family = AF_INET;
	dest_addr.sin_port = htons(port);
	HOSTENT *hst;
	dest_addr.sin_addr.s_addr = inet_addr(ip);

	// адрес сервера получен - пытаемся установить соединение
	if (connect(my_sock, (sockaddr *)&dest_addr, sizeof(dest_addr)))
	{
		closesocket(my_sock);
		WSACleanup();
		printf("Connect error %d\n", WSAGetLastError());
		result=-1;
	}
	
	if(result!=1){		
		closesocket(my_sock);
		WSACleanup();

		printf("Popitka soedinenia #%d neydasna\n", count_popitok);
		Sleep(PAUSE_mezdy_popitok);}
	
	if(count_popitok==MAX_count_popitok&&result==-1){MessageBoxA(NULL, "Ошибка", "Не удалось подключиться к серверу", MB_OK);}
	
	count_popitok++;
	
	}

	


return result;
}
int polychenie_soobsenia(char buff[MAX_PACKET_SIZE], int MAX_count_popitok, int &len, int PAUSE_mezdy_popitok){
	
	int result=1, wsaError=1, recv1=0, count=0;

	while(wsaError/*==10054*/){
		

		if(count>0){
			closesocket(my_sock);
			WSACleanup();
			result=podklysenie_k_servery(MAX_count_popitok, PAUSE_mezdy_popitok);}
		if(result==-1){break;}

	if(recv1==0){
	wsaError=recv_all(my_sock, (char*)&len, sizeof(len));
	if(wsaError){printf("\nRecv error1 %d\n", wsaError);}else{recv1=1;}
		}
	

	
	if(recv1==1){
	wsaError=recv_all(my_sock, buff, len);
	if(wsaError){printf("\nRecv error2 %d\n", wsaError);}
	}
	 
	count++;
	}


	return result;
}
int otpravka_soobsenia(string text_msg, int MAX_count_popitok, int PAUSE_mezdy_popitok){
	int result=1, wsaError=1;
	int len = text_msg.length();
	int send1=0, count=0;

	while(wsaError/*==10054*/){

		if(count>0){
			closesocket(my_sock);
			WSACleanup();
			result=podklysenie_k_servery(MAX_count_popitok, PAUSE_mezdy_popitok);}
		if(result==-1){break;}

	if(send1==0){//Чтобы небыло повторных отправок в случае успешной отправки первого сообщения и ошибки во втором
	send(my_sock, (const char*)&len, sizeof(len), 0);

	wsaError = WSAGetLastError();
	if(wsaError){printf("\nSend error1 %d\n", WSAGetLastError()); printf("result %d\n", result);}else{send1=1;}
	
	}

	if(send1==1){//Если первое сообщение успешно отправлено пытаемся отправить второе
	send(my_sock, text_msg.c_str(), len, 0);

	wsaError = WSAGetLastError();
	if( wsaError )
	{	printf("\nSend error2 %d\n", WSAGetLastError());
		
	}}
	
	 

	count++;
	}
	
	return result;
}
int recv_all(SOCKET s,  char* buf, int len)
{
	int total_recv = 0;
	int bytes_recv;
	int i=0;

	while (len > 0 && (bytes_recv = recv(s, buf + total_recv , len, 0)) > 0) {

		i++;
		total_recv+= bytes_recv;
		len -= bytes_recv;
	}


	return WSAGetLastError();
}

	// Шаг 1 - инициализация библиотеки Winsock
	WSADATA wsadata;
	if( WSAStartup(0x202, &wsadata) )
	{
		printf("WSAStart error %d\n", WSAGetLastError());
		return -1;
	}

    // Шаг 1 - Инициализация Библиотеки Сокетов
    // т.к. возвращенная функцией информация не используется
    // ей передается указатель на рабочий буфер, преобразуемый к указателю
    // на структуру WSADATA.
    // Такой прием позволяет сэкономить одну переменную, однако, буфер
    // должен быть не менее полкилобайта размером (структура WSADATA
    // занимает 400 байт)
    if (WSAStartup(0x0202, (WSADATA *)&buff[0]))
    {
        // Ошибка!
        printf("Error WSAStartup %d\n", WSAGetLastError());
        return -1;
    }