From ac8905602d1f221a71ddcdc86796344101276321 Mon Sep 17 00:00:00 2001
From: LaG1924 <12997935+LaG1924@users.noreply.github.com>
Date: Wed, 14 Jun 2017 19:22:52 +0500
Subject: 2017-06-14

---
 src/network/Network.cpp | 144 ++++++++++++++++++++++++++++--------------------
 1 file changed, 83 insertions(+), 61 deletions(-)

(limited to 'src/network')

diff --git a/src/network/Network.cpp b/src/network/Network.cpp
index 399ce20..4ce424c 100644
--- a/src/network/Network.cpp
+++ b/src/network/Network.cpp
@@ -1,79 +1,101 @@
 #include "Network.hpp"
 
 Network::Network(std::string address, unsigned short port) : m_address(address), m_port(port) {
-    LOG(INFO) << "Connecting to server " << m_address << ":" << m_port;
-    sf::Socket::Status status = m_socket.connect(sf::IpAddress(m_address), m_port);
-    m_socket.setBlocking(true);
-    if (status != sf::Socket::Done) {
-        if (status == sf::Socket::Error) {
-            LOG(ERROR) << "Can't connect to remote server";
-        } else {
-            LOG(ERROR) << "Connection failed with unknown reason";
+	LOG(INFO) << "Connecting to server " << m_address << ":" << m_port;
+	sf::Socket::Status status = m_socket.connect(sf::IpAddress(m_address), m_port);
+	m_socket.setBlocking(true);
+	if (status != sf::Socket::Done) {
+		if (status == sf::Socket::Error) {
+			LOG(ERROR) << "Can't connect to remote server";
+		} else {
+			LOG(ERROR) << "Connection failed with unknown reason";
 			throw std::runtime_error("Connection is failed");
-            throw 13;
-        }
-    }
-    LOG(INFO) << "Connected to server";
+			throw 13;
+		}
+	}
+	LOG(INFO) << "Connected to server";
 }
 
 Network::~Network() {
-    m_socket.disconnect();
-    LOG(INFO) << "Disconnected";
+	m_socket.disconnect();
+	LOG(INFO) << "Disconnected";
 }
 
 void Network::SendHandshake(std::string username) {
-    //Handshake packet
-    Packet handshakePacket = PacketBuilder::CHandshaking0x00(316, m_address, m_port, 2);
-    SendPacket(handshakePacket);
+	//Handshake packet
+	Packet handshakePacket = PacketBuilder::CHandshaking0x00(316, m_address, m_port, 2);
+	SendPacket(handshakePacket);
 
-    //LoginStart packet
-    Field fName;
-    fName.SetString(username);
-    Packet loginPacket(0);
-    loginPacket.AddField(fName);
-    SendPacket(loginPacket);
+	//LoginStart packet
+	Field fName;
+	fName.SetString(username);
+	Packet loginPacket(0);
+	loginPacket.AddField(fName);
+	SendPacket(loginPacket);
 }
 
+void DumpPacket(Packet &packet, std::string DumpName) {
+	return;
+	byte *buff = new byte[packet.GetLength()];
+	packet.CopyToBuff(buff);
+	std::ofstream fs(DumpName, std::ios::out | std::ios::binary);
+	fs.write(reinterpret_cast<const char *>(buff), packet.GetLength());
+	fs.close();
+	delete buff;
+}
+
+static int pn = 0;
+
 void Network::SendPacket(Packet &packet) {
-    m_socket.setBlocking(true);
-    byte *packetData = new byte[packet.GetLength()];
-    packet.CopyToBuff(packetData);
-    m_socket.send(packetData, packet.GetLength());
-    delete[] packetData;
+	m_socket.setBlocking(true);
+	byte *packetData = new byte[packet.GetLength()];
+	packet.CopyToBuff(packetData);
+	m_socket.send(packetData, packet.GetLength());
+	std::ostringstream out;
+	out << "s" << pn++ << "-";
+	out << "0x" << (packet.GetId() < 15 ? "0" : "") << std::hex << packet.GetId() << std::dec;
+	DumpPacket(packet, out.str());
+
+	delete[] packetData;
 }
 
 Packet Network::ReceivePacket() {
-    byte bufLen[5] = {0};
-    size_t rec = 0;
-    for (int i = 0; i < 5; i++) {
-        byte buff = 0;
-        size_t r = 0;
-        m_socket.receive(&buff, 1, r);
-        rec += r;
-        bufLen[i] = buff;
-        if ((buff & 0b10000000) == 0) {
-            break;
-        }
-    }
-    Field fLen = FieldParser::Parse(VarIntType, bufLen);
-    size_t packetLen = fLen.GetVarInt() + fLen.GetLength();
-    if (packetLen > 1024 * 1024 * 15)
-        LOG(WARNING)<<"OMG SIZEOF PACKAGE IS "<<packetLen;
-    if (packetLen < rec) {
-        return Packet(bufLen);
-    }
-    byte *bufPack = new byte[packetLen];
-    std::copy(bufLen, bufLen + rec, bufPack);
-    size_t dataLen = rec;
-    while (m_socket.receive(bufPack + dataLen, packetLen - dataLen, rec) == sf::Socket::Done && dataLen < packetLen) {
-        dataLen += rec;
-    }
-    if (dataLen < packetLen) {
-        LOG(ERROR) << "Received data is "<<dataLen<<" but "<<packetLen<<" is promoted";
-        throw std::runtime_error("Data is losted");
-    } else {
-        Packet p(bufPack);
-        delete[] bufPack;
-        return p;
-    }
+	byte bufLen[5] = {0};
+	size_t rec = 0;
+	for (int i = 0; i < 5; i++) {
+		byte buff = 0;
+		size_t r = 0;
+		m_socket.receive(&buff, 1, r);
+		rec += r;
+		bufLen[i] = buff;
+		if ((buff & 0b10000000) == 0) {
+			break;
+		}
+	}
+	Field fLen = FieldParser::Parse(VarIntType, bufLen);
+	size_t packetLen = fLen.GetVarInt() + fLen.GetLength();
+	if (packetLen > 1024 * 1024 * 15)
+		LOG(WARNING) << "OMG SIZEOF PACKAGE IS " << packetLen;
+	if (packetLen < rec) {
+		return Packet(bufLen);
+	}
+	byte *bufPack = new byte[packetLen];
+	std::copy(bufLen, bufLen + rec, bufPack);
+	size_t dataLen = rec;
+	while (m_socket.receive(bufPack + dataLen, packetLen - dataLen, rec) == sf::Socket::Done && dataLen < packetLen) {
+		dataLen += rec;
+	}
+	if (dataLen < packetLen) {
+		LOG(ERROR) << "Received data is " << dataLen << " but " << packetLen << " is promoted";
+		throw std::runtime_error("Data is losted");
+	} else {
+		Packet p(bufPack);
+		delete[] bufPack;
+
+		std::ostringstream out;
+		out << "r" << pn++ << "-";
+		out << "0x" << (p.GetId() < 15 ? "0" : "") << std::hex << p.GetId() << std::dec;
+		DumpPacket(p, out.str());
+		return p;
+	}
 }
-- 
cgit v1.2.3