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 
24 #include "wificond/client_interface_binder.h"
25 #include "wificond/net/mlme_event.h"
26 #include "wificond/net/netlink_utils.h"
27 #include "wificond/scanning/scan_result.h"
28 #include "wificond/scanning/scan_utils.h"
29 #include "wificond/scanning/scanner_impl.h"
30 
31 using android::net::wifi::IClientInterface;
32 using com::android::server::wifi::wificond::NativeScanResult;
33 using android::sp;
34 using android::wifi_system::InterfaceTool;
35 using android::wifi_system::SupplicantManager;
36 
37 using std::endl;
38 using std::string;
39 using std::unique_ptr;
40 using std::vector;
41 
42 namespace android {
43 namespace wificond {
44 
MlmeEventHandlerImpl(ClientInterfaceImpl * client_interface)45 MlmeEventHandlerImpl::MlmeEventHandlerImpl(ClientInterfaceImpl* client_interface)
46     : client_interface_(client_interface) {
47 }
48 
~MlmeEventHandlerImpl()49 MlmeEventHandlerImpl::~MlmeEventHandlerImpl() {
50 }
51 
OnConnect(unique_ptr<MlmeConnectEvent> event)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 
OnRoam(unique_ptr<MlmeRoamEvent> event)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 
OnAssociate(unique_ptr<MlmeAssociateEvent> event)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 
OnDisconnect(unique_ptr<MlmeDisconnectEvent> event)91 void MlmeEventHandlerImpl::OnDisconnect(unique_ptr<MlmeDisconnectEvent> event) {
92   client_interface_->is_associated_ = false;
93   client_interface_->bssid_.clear();
94 }
95 
OnDisassociate(unique_ptr<MlmeDisassociateEvent> event)96 void MlmeEventHandlerImpl::OnDisassociate(unique_ptr<MlmeDisassociateEvent> event) {
97   client_interface_->is_associated_ = false;
98   client_interface_->bssid_.clear();
99 }
100 
101 
ClientInterfaceImpl(uint32_t wiphy_index,const std::string & interface_name,uint32_t interface_index,const std::vector<uint8_t> & interface_mac_addr,InterfaceTool * if_tool,SupplicantManager * supplicant_manager,NetlinkUtils * netlink_utils,ScanUtils * scan_utils)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 
~ClientInterfaceImpl()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 
GetBinder() const150 sp<android::net::wifi::IClientInterface> ClientInterfaceImpl::GetBinder() const {
151   return binder_;
152 }
153 
Dump(std::stringstream * ss) const154 void ClientInterfaceImpl::Dump(std::stringstream* ss) const {
155   *ss << "------- Dump of client interface with index: "
156       << interface_index_ << " and name: " << interface_name_
157       << "-------" << endl;
158   *ss << "Max number of ssids for single shot scan: "
159       << static_cast<int>(scan_capabilities_.max_num_scan_ssids) << endl;
160   *ss << "Max number of ssids for scheduled scan: "
161       << static_cast<int>(scan_capabilities_.max_num_sched_scan_ssids) << endl;
162   *ss << "Max number of match sets for scheduled scan: "
163       << static_cast<int>(scan_capabilities_.max_match_sets) << endl;
164   *ss << "Device supports random MAC for single shot scan: "
165       << wiphy_features_.supports_random_mac_oneshot_scan << endl;
166   *ss << "Device supports random MAC for scheduled scan: "
167       << wiphy_features_.supports_random_mac_sched_scan << endl;
168   *ss << "------- Dump End -------" << endl;
169 }
170 
EnableSupplicant()171 bool ClientInterfaceImpl::EnableSupplicant() {
172   return supplicant_manager_->StartSupplicant();
173 }
174 
DisableSupplicant()175 bool ClientInterfaceImpl::DisableSupplicant() {
176   return supplicant_manager_->StopSupplicant();
177 }
178 
GetPacketCounters(vector<int32_t> * out_packet_counters)179 bool ClientInterfaceImpl::GetPacketCounters(vector<int32_t>* out_packet_counters) {
180   StationInfo station_info;
181   if (!netlink_utils_->GetStationInfo(interface_index_,
182                                       interface_mac_addr_,
183                                       &station_info)) {
184     return false;
185   }
186   out_packet_counters->push_back(station_info.station_tx_packets);
187   out_packet_counters->push_back(station_info.station_tx_failed);
188 
189   return true;
190 }
191 
SignalPoll(vector<int32_t> * out_signal_poll_results)192 bool ClientInterfaceImpl::SignalPoll(vector<int32_t>* out_signal_poll_results) {
193   StationInfo station_info;
194   if (!netlink_utils_->GetStationInfo(interface_index_,
195                                       bssid_,
196                                       &station_info)) {
197     return false;
198   }
199   out_signal_poll_results->push_back(
200       static_cast<int32_t>(station_info.current_rssi));
201   // Convert from 100kbit/s to Mbps.
202   out_signal_poll_results->push_back(
203       static_cast<int32_t>(station_info.station_tx_bitrate/10));
204   // Association frequency.
205   out_signal_poll_results->push_back(
206       static_cast<int32_t>(associate_freq_));
207 
208   return true;
209 }
210 
GetMacAddress()211 const vector<uint8_t>& ClientInterfaceImpl::GetMacAddress() {
212   return interface_mac_addr_;
213 }
214 
requestANQP(const::std::vector<uint8_t> & bssid,const::android::sp<::android::net::wifi::IANQPDoneCallback> & callback)215 bool ClientInterfaceImpl::requestANQP(
216       const ::std::vector<uint8_t>& bssid,
217       const ::android::sp<::android::net::wifi::IANQPDoneCallback>& callback) {
218   // TODO(nywang): query ANQP information from wpa_supplicant.
219   return true;
220 }
221 
RefreshAssociateFreq()222 bool ClientInterfaceImpl::RefreshAssociateFreq() {
223   // wpa_supplicant fetches associate frequency using the latest scan result.
224   // We should follow the same method here before we find a better solution.
225   std::vector<NativeScanResult> scan_results;
226   if (!scan_utils_->GetScanResult(interface_index_, &scan_results)) {
227     return false;
228   }
229   for (auto& scan_result : scan_results) {
230     if (scan_result.associated) {
231       associate_freq_ = scan_result.frequency;
232     }
233   }
234   return false;
235 }
236 
IsAssociated() const237 bool ClientInterfaceImpl::IsAssociated() const {
238   return is_associated_;
239 }
240 
241 }  // namespace wificond
242 }  // namespace android
243