You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

140 lines
4.3 KiB

/*
* Copyright (C) 2016 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "wificond/net/mlme_event.h"
#include <vector>
#include <android-base/logging.h>
#include "wificond/net/kernel-header-latest/nl80211.h"
#include "wificond/net/nl80211_packet.h"
using std::array;
using std::unique_ptr;
using std::vector;
namespace android {
namespace wificond {
namespace {
bool GetCommonFields(const NL80211Packet* packet,
uint32_t* if_index,
array<uint8_t, ETH_ALEN>* bssid) {
if (!packet->GetAttributeValue(NL80211_ATTR_IFINDEX, if_index)) {
LOG(ERROR) << "Failed to get NL80211_ATTR_IFINDEX";
return false;
}
// Some MLME events do not contain MAC address.
if (!packet->GetAttributeValue(NL80211_ATTR_MAC, bssid)) {
LOG(DEBUG) << "Failed to get NL80211_ATTR_MAC";
}
return true;
}
} // namespace
unique_ptr<MlmeAssociateEvent> MlmeAssociateEvent::InitFromPacket(
const NL80211Packet* packet) {
if (packet->GetCommand() != NL80211_CMD_ASSOCIATE) {
return nullptr;
}
unique_ptr<MlmeAssociateEvent> associate_event(new MlmeAssociateEvent());
if (!GetCommonFields(packet,
&(associate_event->interface_index_),
&(associate_event->bssid_))){
return nullptr;
}
// According to wpa_supplicant, status code of an ASSOCIATE event should be
// parsed from NL80211_ATTR_FRAME attribute.
// TODO(nywang): Parse NL80211_ATTR_FRAME 80211 management frame and get
// status code.
associate_event->status_code_ = 0;
associate_event->is_timeout_ = packet->HasAttribute(NL80211_ATTR_TIMED_OUT);
return associate_event;
}
unique_ptr<MlmeConnectEvent> MlmeConnectEvent::InitFromPacket(
const NL80211Packet* packet) {
if (packet->GetCommand() != NL80211_CMD_CONNECT) {
return nullptr;
}
unique_ptr<MlmeConnectEvent> connect_event(new MlmeConnectEvent());
if (!GetCommonFields(packet,
&(connect_event->interface_index_),
&(connect_event->bssid_))){
return nullptr;
}
if (!packet->GetAttributeValue(NL80211_ATTR_STATUS_CODE,
&(connect_event->status_code_))) {
LOG(WARNING) << "Failed to get NL80211_ATTR_STATUS_CODE";
connect_event->status_code_ = 0;
}
connect_event->is_timeout_ = packet->HasAttribute(NL80211_ATTR_TIMED_OUT);
return connect_event;
}
unique_ptr<MlmeRoamEvent> MlmeRoamEvent::InitFromPacket(
const NL80211Packet* packet) {
if (packet->GetCommand() != NL80211_CMD_ROAM) {
return nullptr;
}
unique_ptr<MlmeRoamEvent> roam_event(new MlmeRoamEvent());
if (!GetCommonFields(packet,
&(roam_event->interface_index_),
&(roam_event->bssid_))){
return nullptr;
}
return roam_event;
}
unique_ptr<MlmeDisconnectEvent> MlmeDisconnectEvent::InitFromPacket(
const NL80211Packet* packet) {
if (packet->GetCommand() != NL80211_CMD_DISCONNECT) {
return nullptr;
}
unique_ptr<MlmeDisconnectEvent> disconnect_event(new MlmeDisconnectEvent());
if (!GetCommonFields(packet,
&(disconnect_event->interface_index_),
&(disconnect_event->bssid_))){
return nullptr;
}
return disconnect_event;
}
unique_ptr<MlmeDisassociateEvent> MlmeDisassociateEvent::InitFromPacket(
const NL80211Packet* packet) {
if (packet->GetCommand() != NL80211_CMD_DISASSOCIATE) {
return nullptr;
}
unique_ptr<MlmeDisassociateEvent> disassociate_event(new MlmeDisassociateEvent());
if (!GetCommonFields(packet,
&(disassociate_event->interface_index_),
&(disassociate_event->bssid_))){
return nullptr;
}
return disassociate_event;
}
} // namespace wificond
} // namespace android