blob: 30a33b3b0353d5751fbd9d643fb1613a6468d05d (
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
84
85
86
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);
}
}
}
}
}
}
|
