Adressed feedback

This commit is contained in:
B3n30 2018-10-04 12:34:51 +02:00
parent 38f136048e
commit 458346d1e8

View File

@ -9,9 +9,9 @@
#include <list> #include <list>
#include <map> #include <map>
#include <mutex> #include <mutex>
#include <optional>
#include <unordered_map> #include <unordered_map>
#include <vector> #include <vector>
#include <boost/optional.hpp>
#include <cryptopp/osrng.h> #include <cryptopp/osrng.h>
#include "common/common_types.h" #include "common/common_types.h"
#include "common/logging/log.h" #include "common/logging/log.h"
@ -104,16 +104,16 @@ static std::mutex beacon_mutex;
// Number of beacons to store before we start dropping the old ones. // Number of beacons to store before we start dropping the old ones.
// TODO(Subv): Find a more accurate value for this limit. // TODO(Subv): Find a more accurate value for this limit.
static constexpr std::size_t MaxBeaconFrames = 15; constexpr std::size_t MaxBeaconFrames = 15;
// List of the last <MaxBeaconFrames> beacons received from the network. // List of the last <MaxBeaconFrames> beacons received from the network.
static std::list<Network::WifiPacket> received_beacons; static std::list<Network::WifiPacket> received_beacons;
// Network node id used when a SecureData packet is addressed to every connected node. // Network node id used when a SecureData packet is addressed to every connected node.
static constexpr u16 BroadcastNetworkNodeId = 0xFFFF; constexpr u16 BroadcastNetworkNodeId = 0xFFFF;
// The Host has always dest_node_id 1 // The Host has always dest_node_id 1
static constexpr u16 HostDestNodeId = 1; constexpr u16 HostDestNodeId = 1;
/** /**
* Returns a list of received 802.11 beacon frames from the specified sender since the last call. * Returns a list of received 802.11 beacon frames from the specified sender since the last call.
@ -167,11 +167,13 @@ static void BroadcastNodeMap() {
packet.channel = network_channel; packet.channel = network_channel;
packet.type = Network::WifiPacket::PacketType::NodeMap; packet.type = Network::WifiPacket::PacketType::NodeMap;
packet.destination_address = Network::BroadcastMac; packet.destination_address = Network::BroadcastMac;
std::size_t size = node_map.size(); std::size_t num_entries = std::count_if(node_map.begin(), node_map.end(),
[](const auto& node) { return node.second.connected; });
using node_t = decltype(node_map)::value_type; using node_t = decltype(node_map)::value_type;
packet.data.resize(sizeof(size) + (sizeof(node_t::first) + sizeof(node_t::second)) * size); packet.data.resize(sizeof(num_entries) +
std::memcpy(packet.data.data(), &size, sizeof(size)); (sizeof(node_t::first) + sizeof(node_t::second.node_id)) * num_entries);
std::size_t offset = sizeof(size); std::memcpy(packet.data.data(), &num_entries, sizeof(num_entries));
std::size_t offset = sizeof(num_entries);
for (const auto& node : node_map) { for (const auto& node : node_map) {
if (node.second.connected) { if (node.second.connected) {
std::memcpy(packet.data.data() + offset, node.first.data(), sizeof(node.first)); std::memcpy(packet.data.data() + offset, node.first.data(), sizeof(node.first));
@ -542,29 +544,28 @@ void HandleDeauthenticationFrame(const Network::WifiPacket& packet) {
return; return;
} }
s32 node_id = node_map[packet.transmitter_address].node_id; Node node = node_map[packet.transmitter_address];
bool connected = node_map[packet.transmitter_address].connected;
node_map.erase(packet.transmitter_address); node_map.erase(packet.transmitter_address);
if (!connected) { if (!node.connected) {
LOG_DEBUG(Service_NWM, "Received DeauthenticationFrame from a not connected MAC Address"); LOG_DEBUG(Service_NWM, "Received DeauthenticationFrame from a not connected MAC Address");
return; return;
} }
auto node = std::find_if(node_info.begin(), node_info.end(), [&node_id](const NodeInfo& info) { auto node_it = std::find_if(node_info.begin(), node_info.end(), [&node](const NodeInfo& info) {
return info.network_node_id == node_id; return info.network_node_id == node.node_id;
}); });
ASSERT(node != node_info.end()); ASSERT(node_it != node_info.end());
connection_status.node_bitmask &= ~(1 << (node_id - 1)); connection_status.node_bitmask &= ~(1 << (node.node_id - 1));
connection_status.changed_nodes |= 1 << (node_id - 1); connection_status.changed_nodes |= 1 << (node.node_id - 1);
connection_status.total_nodes--; connection_status.total_nodes--;
connection_status.nodes[node_id - 1] = 0; connection_status.nodes[node.node_id - 1] = 0;
network_info.total_nodes--; network_info.total_nodes--;
// TODO(B3N30): broadcast new connection_status to clients // TODO(B3N30): broadcast new connection_status to clients
node->Reset(); node_it->Reset();
connection_status_event->Signal(); connection_status_event->Signal();
} }
@ -604,7 +605,7 @@ void OnWifiPacketReceived(const Network::WifiPacket& packet) {
} }
} }
static std::optional<Network::MacAddress> GetNodeMacAddress(u16 dest_node_id, u8 flags) { static boost::optional<Network::MacAddress> GetNodeMacAddress(u16 dest_node_id, u8 flags) {
constexpr u8 BroadcastFlag = 0x2; constexpr u8 BroadcastFlag = 0x2;
if ((flags & BroadcastFlag) || dest_node_id == BroadcastNetworkNodeId) { if ((flags & BroadcastFlag) || dest_node_id == BroadcastNetworkNodeId) {
// Broadcast // Broadcast