summaryrefslogtreecommitdiffstats
path: root/src/Net/Packets/HostListPacket.cpp
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);
    }
  }
}

}
}
}