#include <string>
#include <iostream>
#include <enet/enet.h>
using namespace std;

#include "scope_guard.h"
void main()
{
	if(enet_initialize()) {
		cout << "fail init" << endl;
		return;
	}


	int success = 0;
	int fail = 0;

	while(1) {
		ENetHost* client = enet_host_create(NULL,
							1,
							2,
							0,
							0);
		if(client == NULL) {
			cout << "fail create client" << endl;
			fail++;
			continue;
		}
		scope_guard g1([&]{ enet_host_destroy(client); });

		ENetAddress svraddr;
		enet_address_set_host(&svraddr, "127.0.0.1");
		svraddr.port = 9000;

		ENetPeer* server = enet_host_connect(client, &svraddr, 2, 0);

		if(server == NULL) {
			fail++;
			continue;
		}

		ENetEvent event;
		if(enet_host_service (client, &event, 5000) > 0 &&
		   event.type == ENET_EVENT_TYPE_CONNECT) {
		} else {
			enet_peer_reset (server);
			fail++;
			continue;
		}

		string data(2 * 1024, 'a'); // not ok
		ENetPacket* packet1 = enet_packet_create(data.c_str(), data.length(), ENET_PACKET_FLAG_RELIABLE);
		if(!packet1) {
			fail++;
			continue;
		}
		if(enet_peer_send(server, 0, packet1) < 0) {
			fail++;
			continue;
		}

		enet_host_flush (client);

		bool disconnected = false;
		while(enet_host_service (client, &event, 5000) > 0) {
			if(event.type == ENET_EVENT_TYPE_RECEIVE) {
				if(event.packet->dataLength == 2*1024) {
					success++;
				} else {
					fail++;
				}
				enet_packet_destroy (event.packet);
				break;
			} else if(event.type == ENET_EVENT_TYPE_DISCONNECT) {
				disconnected = true;
				break;
			}
		}
		if(!disconnected) {
			enet_peer_disconnect(server, 0);
			while(enet_host_service (client, &event, 5000) > 0) {
				if(event.type == ENET_EVENT_TYPE_DISCONNECT) {
					break;
				}
			}
		}

		enet_peer_reset(server);
		cout << "success: " << success << ", fail: " << fail << endl;
	}


	enet_deinitialize();
}