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 #ifndef WIFICOND_CLIENT_INTERFACE_IMPL_H_
18 #define WIFICOND_CLIENT_INTERFACE_IMPL_H_
19 
20 #include <array>
21 #include <string>
22 
23 #include <linux/if_ether.h>
24 
25 #include <android-base/macros.h>
26 #include <utils/StrongPointer.h>
27 #include <wifi_system/interface_tool.h>
28 
29 #include "android/net/wifi/IClientInterface.h"
30 #include "android/net/wifi/ISendMgmtFrameEvent.h"
31 #include "wificond/net/mlme_event_handler.h"
32 #include "wificond/net/netlink_utils.h"
33 #include "wificond/scanning/scanner_impl.h"
34 
35 namespace android {
36 namespace wificond {
37 
38 class ClientInterfaceBinder;
39 class ClientInterfaceImpl;
40 class ScanUtils;
41 
42 class MlmeEventHandlerImpl : public MlmeEventHandler {
43  public:
44   MlmeEventHandlerImpl(ClientInterfaceImpl* client_interface);
45   ~MlmeEventHandlerImpl() override;
46   void OnConnect(std::unique_ptr<MlmeConnectEvent> event) override;
47   void OnRoam(std::unique_ptr<MlmeRoamEvent> event) override;
48   void OnAssociate(std::unique_ptr<MlmeAssociateEvent> event) override;
49   void OnDisconnect(std::unique_ptr<MlmeDisconnectEvent> event) override;
50   void OnDisassociate(std::unique_ptr<MlmeDisassociateEvent> event) override;
51 
52  private:
53   ClientInterfaceImpl* client_interface_;
54 };
55 
56 // Holds the guts of how we control network interfaces capable of connecting to
57 // access points via wpa_supplicant.
58 //
59 // Because remote processes may hold on to the corresponding
60 // binder object past the lifetime of the local object, we are forced to
61 // keep this object separate from the binder representation of itself.
62 class ClientInterfaceImpl {
63  public:
64   ClientInterfaceImpl(
65       uint32_t wiphy_index,
66       const std::string& interface_name,
67       uint32_t interface_index,
68       const std::array<uint8_t, ETH_ALEN>& interface_mac_addr,
69       android::wifi_system::InterfaceTool* if_tool,
70       NetlinkUtils* netlink_utils,
71       ScanUtils* scan_utils);
72   virtual ~ClientInterfaceImpl();
73 
74   // Get a pointer to the binder representing this ClientInterfaceImpl.
75   android::sp<android::net::wifi::IClientInterface> GetBinder() const;
76 
77   bool GetPacketCounters(std::vector<int32_t>* out_packet_counters);
78   bool SignalPoll(std::vector<int32_t>* out_signal_poll_results);
79   const std::array<uint8_t, ETH_ALEN>& GetMacAddress();
GetInterfaceName()80   const std::string& GetInterfaceName() const { return interface_name_; }
GetScanner()81   const android::sp<ScannerImpl> GetScanner() { return scanner_; };
82   bool SetMacAddress(const std::array<uint8_t, ETH_ALEN>& mac);
83   virtual bool IsAssociated() const;
84   void Dump(std::stringstream* ss) const;
85   void SendMgmtFrame(
86       const std::vector<uint8_t>& frame,
87       const sp<::android::net::wifi::ISendMgmtFrameEvent>& callback,
88       int32_t mcs);
89 
90  private:
91   bool RefreshAssociateFreq();
92 
93   const uint32_t wiphy_index_;
94   const std::string interface_name_;
95   const uint32_t interface_index_;
96   const std::array<uint8_t, ETH_ALEN> interface_mac_addr_;
97   android::wifi_system::InterfaceTool* const if_tool_;
98   NetlinkUtils* const netlink_utils_;
99   ScanUtils* const scan_utils_;
100   const std::unique_ptr<MlmeEventHandlerImpl> mlme_event_handler_;
101   const android::sp<ClientInterfaceBinder> binder_;
102   android::sp<ScannerImpl> scanner_;
103 
104   // Cached information for this connection.
105   bool is_associated_;
106   std::array<uint8_t, ETH_ALEN> bssid_;
107   uint32_t associate_freq_;
108 
109   // Capability information for this wiphy/interface.
110   BandInfo band_info_;
111   ScanCapabilities scan_capabilities_;
112   WiphyFeatures wiphy_features_;
113 
114   // handler for frame tx status messages
115   bool frame_tx_in_progress_;
116   uint64_t frame_tx_status_cookie_;
117   std::function<void(bool was_acked)> on_frame_tx_status_event_handler_;
118 
119   DISALLOW_COPY_AND_ASSIGN(ClientInterfaceImpl);
120   friend class MlmeEventHandlerImpl;
121 };
122 
123 }  // namespace wificond
124 }  // namespace android
125 
126 #endif  // WIFICOND_CLIENT_INTERFACE_IMPL_H_
127