client_interface_impl.cpp

   1 /*
   2  * Copyright (C) 2016 The Android Open Source Project
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include "wificond/client_interface_impl.h"
  18
  19 #include <vector>
  20
  21 #include <android-base/logging.h>
  22 #include <wifi_system/supplicant_manager.h>
  23 #include <wifi_system/wifi.h>
  24
  25 #include "wificond/client_interface_binder.h"
  26 #include "wificond/net/mlme_event.h"
  27 #include "wificond/net/netlink_utils.h"
  28 #include "wificond/scanning/scan_result.h"
  29 #include "wificond/scanning/scan_utils.h"
  30 #include "wificond/scanning/scanner_impl.h"
  31
  32 using android::net::wifi::IClientInterface;
  33 using com::android::server::wifi::wificond::NativeScanResult;
  34 using android::sp;
  35 using android::wifi_system::InterfaceTool;
  36 using android::wifi_system::SupplicantManager;
  37
  38 using std::string;
  39 using std::unique_ptr;
  40 using std::vector;
  41
  42 namespace android {
  43 namespace wificond {
  44
  45 MlmeEventHandlerImpl::MlmeEventHandlerImpl(ClientInterfaceImpl* client_interface)
  46     : client_interface_(client_interface) {
  47 }
  48
  49 MlmeEventHandlerImpl::~MlmeEventHandlerImpl() {
  50 }
  51
  52 void MlmeEventHandlerImpl::OnConnect(unique_ptr<MlmeConnectEvent> event) {
  53   if (!event->IsTimeout() && event->GetStatusCode() == 0) {
  54     client_interface_->is_associated_ = true;
  55     client_interface_->RefreshAssociateFreq();
  56     client_interface_->bssid_ = event->GetBSSID();
  57   } else {
  58     if (event->IsTimeout()) {
  59       LOG(INFO) << "Connect timeout";
  60     }
  61     client_interface_->is_associated_ = false;
  62     client_interface_->bssid_.clear();
  63   }
  64 }
  65
  66 void MlmeEventHandlerImpl::OnRoam(unique_ptr<MlmeRoamEvent> event) {
  67   if (event->GetStatusCode() == 0) {
  68     client_interface_->is_associated_ = true;
  69     client_interface_->RefreshAssociateFreq();
  70     client_interface_->bssid_ = event->GetBSSID();
  71   } else {
  72     client_interface_->is_associated_ = false;
  73     client_interface_->bssid_.clear();
  74   }
  75 }
  76
  77 void MlmeEventHandlerImpl::OnAssociate(unique_ptr<MlmeAssociateEvent> event) {
  78   if (!event->IsTimeout() && event->GetStatusCode() == 0) {
  79     client_interface_->is_associated_ = true;
  80     client_interface_->RefreshAssociateFreq();
  81     client_interface_->bssid_ = event->GetBSSID();
  82   } else {
  83     if (event->IsTimeout()) {
  84       LOG(INFO) << "Associate timeout";
  85     }
  86     client_interface_->is_associated_ = false;
  87     client_interface_->bssid_.clear();
  88   }
  89 }
  90
  91 void MlmeEventHandlerImpl::OnDisconnect(unique_ptr<MlmeDisconnectEvent> event) {
  92   client_interface_->is_associated_ = false;
  93   client_interface_->bssid_.clear();
  94 }
  95
  96 void MlmeEventHandlerImpl::OnDisassociate(unique_ptr<MlmeDisassociateEvent> event) {
  97   client_interface_->is_associated_ = false;
  98   client_interface_->bssid_.clear();
  99 }
 100
 101
 102 ClientInterfaceImpl::ClientInterfaceImpl(
 103     uint32_t wiphy_index,
 104     const std::string& interface_name,
 105     uint32_t interface_index,
 106     const std::vector<uint8_t>& interface_mac_addr,
 107     InterfaceTool* if_tool,
 108     SupplicantManager* supplicant_manager,
 109     NetlinkUtils* netlink_utils,
 110     ScanUtils* scan_utils)
 111     : wiphy_index_(wiphy_index),
 112       interface_name_(interface_name),
 113       interface_index_(interface_index),
 114       interface_mac_addr_(interface_mac_addr),
 115       if_tool_(if_tool),
 116       supplicant_manager_(supplicant_manager),
 117       netlink_utils_(netlink_utils),
 118       scan_utils_(scan_utils),
 119       mlme_event_handler_(new MlmeEventHandlerImpl(this)),
 120       binder_(new ClientInterfaceBinder(this)),
 121       is_associated_(false) {
 122   netlink_utils_->SubscribeMlmeEvent(
 123       interface_index_,
 124       mlme_event_handler_.get());
 125   if (!netlink_utils_->GetWiphyInfo(wiphy_index_,
 126                                &band_info_,
 127                                &scan_capabilities_,
 128                                &wiphy_features_)) {
 129     LOG(ERROR) << "Failed to get wiphy info from kernel";
 130   }
 131   LOG(INFO) << "create scanner for interface with index: "
 132             << (int)interface_index_;
 133   scanner_ = new ScannerImpl(wiphy_index,
 134                              interface_index_,
 135                              scan_capabilities_,
 136                              wiphy_features_,
 137                              this,
 138                              netlink_utils_,
 139                              scan_utils_);
 140 }
 141
 142 ClientInterfaceImpl::~ClientInterfaceImpl() {
 143   binder_->NotifyImplDead();
 144   scanner_->Invalidate();
 145   DisableSupplicant();
 146   netlink_utils_->UnsubscribeMlmeEvent(interface_index_);
 147   if_tool_->SetUpState(interface_name_.c_str(), false);
 148 }
 149
 150 sp<android::net::wifi::IClientInterface> ClientInterfaceImpl::GetBinder() const {
 151   return binder_;
 152 }
 153
 154 bool ClientInterfaceImpl::EnableSupplicant() {
 155   return supplicant_manager_->StartSupplicant();
 156 }
 157
 158 bool ClientInterfaceImpl::DisableSupplicant() {
 159   return supplicant_manager_->StopSupplicant();
 160 }
 161
 162 bool ClientInterfaceImpl::GetPacketCounters(vector<int32_t>* out_packet_counters) {
 163   StationInfo station_info;
 164   if (!netlink_utils_->GetStationInfo(interface_index_,
 165                                       interface_mac_addr_,
 166                                       &station_info)) {
 167     return false;
 168   }
 169   out_packet_counters->push_back(station_info.station_tx_packets);
 170   out_packet_counters->push_back(station_info.station_tx_failed);
 171
 172   return true;
 173 }
 174
 175 bool ClientInterfaceImpl::SignalPoll(vector<int32_t>* out_signal_poll_results) {
 176   StationInfo station_info;
 177   if (!netlink_utils_->GetStationInfo(interface_index_,
 178                                       bssid_,
 179                                       &station_info)) {
 180     return false;
 181   }
 182   out_signal_poll_results->push_back(
 183       static_cast<int32_t>(station_info.current_rssi));
 184   // Convert from 100kbit/s to Mbps.
 185   out_signal_poll_results->push_back(
 186       static_cast<int32_t>(station_info.station_tx_bitrate/10));
 187   // Association frequency.
 188   out_signal_poll_results->push_back(
 189       static_cast<int32_t>(associate_freq_));
 190
 191   return true;
 192 }
 193
 194 const vector<uint8_t>& ClientInterfaceImpl::GetMacAddress() {
 195   return interface_mac_addr_;
 196 }
 197
 198 bool ClientInterfaceImpl::requestANQP(
 199       const ::std::vector<uint8_t>& bssid,
 200       const ::android::sp<::android::net::wifi::IANQPDoneCallback>& callback) {
 201   // TODO(nywang): query ANQP information from wpa_supplicant.
 202   return true;
 203 }
 204
 205 bool ClientInterfaceImpl::RefreshAssociateFreq() {
 206   // wpa_supplicant fetches associate frequency using the latest scan result.
 207   // We should follow the same method here before we find a better solution.
 208   std::vector<NativeScanResult> scan_results;
 209   if (!scan_utils_->GetScanResult(interface_index_, &scan_results)) {
 210     return false;
 211   }
 212   for (auto& scan_result : scan_results) {
 213     if (scan_result.associated) {
 214       associate_freq_ = scan_result.frequency;
 215     }
 216   }
 217   return false;
 218 }
 219
 220 bool ClientInterfaceImpl::IsAssociated() {
 221   return is_associated_;
 222 }
 223
 224 }  // namespace wificond
 225 }  // namespace android