diff options
Diffstat (limited to 'src/Net/Packets/HostListPacket.cpp')
-rw-r--r-- | src/Net/Packets/HostListPacket.cpp | 87 |
1 files changed, 87 insertions, 0 deletions
diff --git a/src/Net/Packets/HostListPacket.cpp b/src/Net/Packets/HostListPacket.cpp new file mode 100644 index 0000000..30a33b3 --- /dev/null +++ b/src/Net/Packets/HostListPacket.cpp @@ -0,0 +1,87 @@ +/* + * HostListPacket.cpp + * + * Copyright (C) 2008 Matthias Schiffer <matthias@gamezock.de> + * + * This program is free software: you can redistribute it and/or modify it + * under the terms of the GNU General Public License as published by the + * Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. + * See the GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with this program. If not, see <http://www.gnu.org/licenses/>. + */ + +#include "HostListPacket.h" +#include <cstring> +#include <sstream> + +namespace Mad { +namespace Net { +namespace Packets { + +void HostListPacket::assemblePacket() { + std::ostringstream stream; + + for(std::vector<Common::HostInfo>::iterator host = hostList.begin(); host != hostList.end(); ++host) { + stream << host->getName() << std::endl; + stream << host->getIP() << std::endl; + } + + std::string str = stream.str(); + + setLength(sizeof(uint16_t) + sizeof(HostData)*hostList.size() + str.length()); + + nHosts = (uint16_t*)rawData->data; + hostData = (HostData*)(rawData->data + sizeof(uint16_t)); + charData = (char*)(rawData->data + sizeof(uint16_t) + sizeof(HostData)*hostList.size()); + + std::memcpy(charData, str.c_str(), str.length()); + + *nHosts = htons(hostList.size()); + + for(size_t i = 0; i < hostList.size(); ++i) { + hostData[i].status = htons((uint16_t)hostList[i].getStatus()); + } +} + +void HostListPacket::parsePacket() { + hostList.clear(); + + if(getLength() < sizeof(uint16_t)) + return; + + nHosts = (uint16_t*)rawData->data; + hostList.resize(ntohs(*nHosts)); + + if(getLength() < sizeof(uint16_t) + sizeof(HostData)*hostList.size()) + setLength(sizeof(uint16_t) + sizeof(HostData)*hostList.size()); + + nHosts = (uint16_t*)rawData->data; + hostData = (HostData*)(rawData->data + sizeof(uint16_t)); + charData = (char*)(rawData->data + sizeof(uint16_t) + sizeof(HostData)*hostList.size()); + + std::istringstream stream(std::string(charData, getLength() - (sizeof(uint16_t)+sizeof(HostData)*hostList.size()))); + + for(size_t i = 0; i < hostList.size(); ++i) { + hostList[i].setStatus((Common::HostInfo::Status)ntohs(hostData[i].status)); + + if(!stream.eof()) { + std::string str; + + std::getline(stream, str); + hostList[i].setName(str); + std::getline(stream, str); + hostList[i].setIP(str); + } + } +} + +} +} +} |