blob: 659b6215a290faade0439307d5e38930c872d04f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
/*
* 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::string str;
for(std::vector<Common::HostInfo>::iterator host = hostList.begin(); host != hostList.end(); ++host)
str += host->getName() + "\n" + host->getIP() + "\n";
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].state = htons((uint16_t)hostList[i].getState());
}
}
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].setState((Common::HostInfo::State)ntohs(hostData[i].state));
if(!stream.eof()) {
std::string str;
std::getline(stream, str);
hostList[i].setName(str);
std::getline(stream, str);
hostList[i].setIP(str);
}
}
}
}
}
}
|
