DarkflameServer/dNet/PacketUtils.cpp

#include "PacketUtils.h"
#include <MessageIdentifiers.h>
#include <vector>
#include <fstream>
#include "dLogger.h"
#include "Game.h"

void PacketUtils::WriteHeader(RakNet::BitStream& bitStream, uint16_t connectionType, uint32_t internalPacketID) {
	bitStream.Write(MessageID(ID_USER_PACKET_ENUM));
	bitStream.Write(connectionType);
	bitStream.Write(internalPacketID);
	bitStream.Write(uint8_t(0));
}

uint16_t PacketUtils::ReadPacketU16(uint32_t startLoc, Packet* packet) {
	if (startLoc + 2 > packet->length) return 0;

	std::vector<unsigned char> t;
	for (uint32_t i = startLoc; i < startLoc + 2; i++) t.push_back(packet->data[i]);
	return *(uint16_t*)t.data();
}

uint32_t PacketUtils::ReadPacketU32(uint32_t startLoc, Packet* packet) {
	if (startLoc + 4 > packet->length) return 0;

	std::vector<unsigned char> t;
	for (uint32_t i = startLoc; i < startLoc + 4; i++) {
		t.push_back(packet->data[i]);
	}
	return *(uint32_t*)t.data();
}

uint64_t PacketUtils::ReadPacketU64(uint32_t startLoc, Packet* packet) {
	if (startLoc + 8 > packet->length) return 0;

	std::vector<unsigned char> t;
	for (uint32_t i = startLoc; i < startLoc + 8; i++) t.push_back(packet->data[i]);
	return *(uint64_t*)t.data();
}

int64_t PacketUtils::ReadPacketS64(uint32_t startLoc, Packet* packet) {
	if (startLoc + 8 > packet->length) return 0;

	std::vector<unsigned char> t;
	for (size_t i = startLoc; i < startLoc + 8; i++) t.push_back(packet->data[i]);
	return *(int64_t*)t.data();
}

std::string PacketUtils::ReadString(uint32_t startLoc, Packet* packet, bool wide, uint32_t maxLen) {
	std::string readString = "";

	if (wide) maxLen *= 2;

	if (packet->length > startLoc) {
		uint32_t i = 0;
		while (packet->data[startLoc + i] != '\0' && packet->length > (uint32_t)(startLoc + i) && maxLen > i) {
			readString.push_back(packet->data[startLoc + i]);

			if (wide) {
				i += 2;     // Wide-char string
			} else {
				i++;        // Regular string
			}
		}
	}

	return readString;
}

void PacketUtils::WritePacketString(const std::string& string, uint32_t maxSize, RakNet::BitStream* bitStream) {
	uint32_t size = static_cast<uint32_t>(string.size());
	uint32_t remSize = static_cast<uint32_t>(maxSize - size);

	if (size > maxSize) size = maxSize;

	for (uint32_t i = 0; i < size; ++i) {
		bitStream->Write(static_cast<char>(string[i]));
	}

	for (uint32_t j = 0; j < remSize; ++j) {
		bitStream->Write(static_cast<char>(0));
	}
}

void PacketUtils::WriteString(RakNet::BitStream& bitStream, const std::string& s, uint32_t maxSize) {
	uint32_t size = s.size();
	uint32_t emptySize = maxSize - size;

	if (size > maxSize) size = maxSize;

	for (uint32_t i = 0; i < size; i++) {
		bitStream.Write((char)s[i]);
	}

	for (uint32_t i = 0; i < emptySize; i++) {
		bitStream.Write((char)0);
	}
}

void PacketUtils::WriteWString(RakNet::BitStream& bitStream, const std::string& string, uint32_t maxSize) {
	uint32_t size = static_cast<uint32_t>(string.length());
	uint32_t remSize = static_cast<uint32_t>(maxSize - size);

	if (size > maxSize) size = maxSize;

	for (uint32_t i = 0; i < size; ++i) {
		bitStream.Write(static_cast<uint16_t>(string[i]));
	}

	for (uint32_t j = 0; j < remSize; ++j) {
		bitStream.Write(static_cast<uint16_t>(0));
	}
}

void PacketUtils::WriteWString(RakNet::BitStream& bitStream, const std::u16string& string, uint32_t maxSize) {
	uint32_t size = static_cast<uint32_t>(string.length());
	uint32_t remSize = static_cast<uint32_t>(maxSize - size);

	if (size > maxSize) size = maxSize;

	for (uint32_t i = 0; i < size; ++i) {
		bitStream.Write(static_cast<uint16_t>(string[i]));
	}

	for (uint32_t j = 0; j < remSize; ++j) {
		bitStream.Write(static_cast<uint16_t>(0));
	}
}

void PacketUtils::WritePacketWString(const std::string& string, uint32_t maxSize, RakNet::BitStream* bitStream) {
	uint32_t size = static_cast<uint32_t>(string.length());
	uint32_t remSize = static_cast<uint32_t>(maxSize - size);

	if (size > maxSize) size = maxSize;

	for (uint32_t i = 0; i < size; ++i) {
		bitStream->Write(static_cast<uint16_t>(string[i]));
	}

	for (uint32_t j = 0; j < remSize; ++j) {
		bitStream->Write(static_cast<uint16_t>(0));
	}
}

//! Saves a packet to the filesystem
void PacketUtils::SavePacket(const std::string& filename, const char* data, size_t length) {
	//If we don't log to the console, don't save the bin files either. This takes up a lot of time.
	if (!Game::logger->GetIsLoggingToConsole()) return;

	std::string path = "packets/" + filename;

	std::ofstream file(path, std::ios::binary);
	if (!file.is_open()) return;

	file.write(data, length);
	file.close();
}