140 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			140 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			C++
		
	
	
	
| /*
 | |
|  * 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
 |