Home | History | Annotate | Download | only in apmanager 
      1 //
      2 // Copyright (C) 2015 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 APMANAGER_HOSTAPD_MONITOR_H_
     18 #define APMANAGER_HOSTAPD_MONITOR_H_
     19 
     20 #include <string>
     21 
     22 #include <base/cancelable_callback.h>
     23 #include <base/macros.h>
     24 #include <base/memory/weak_ptr.h>
     25 
     26 #include "apmanager/event_dispatcher.h"
     27 
     28 namespace shill {
     29 
     30 struct InputData;
     31 class IOHandler;
     32 class IOHandlerFactory;
     33 class Sockets;
     34 
     35 }  // namespace shill
     36 
     37 namespace apmanager {
     38 
     39 // Class for monitoring events from hostapd control interface.
     40 class HostapdMonitor {
     41  public:
     42   enum Event {
     43     kHostapdFailed,
     44     kHostapdStarted,
     45     kStationConnected,
     46     kStationDisconnected,
     47   };
     48 
     49   typedef base::Callback<void(Event event, const std::string& data)>
     50       EventCallback;
     51 
     52   HostapdMonitor(const EventCallback& callback_,
     53                  const std::string& control_interface_path,
     54                  const std::string& network_interface_name);
     55   virtual ~HostapdMonitor();
     56 
     57   virtual void Start();
     58 
     59  private:
     60   friend class HostapdMonitorTest;
     61 
     62   static const char kLocalPathFormat[];
     63   static const char kHostapdCmdAttach[];
     64   static const char kHostapdRespOk[];
     65   static const char kHostapdEventStationConnected[];
     66   static const char kHostapdEventStationDisconnected[];
     67   static const int kHostapdCtrlIfaceCheckIntervalMs;
     68   static const int kHostapdCtrlIfaceCheckMaxAttempts;
     69   static const int kHostapdAttachTimeoutMs;
     70   static const int kInvalidSocket;
     71 
     72   // Task for checking if hostapd control interface is up or not.
     73   void HostapdCtrlIfaceCheckTask();
     74 
     75   // Attach to hostapd control interface to receive unsolicited event
     76   // notifications.
     77   void AttachToHostapd();
     78   void AttachTimeoutHandler();
     79 
     80   bool SendMessage(const char* message, size_t length);
     81   void ParseMessage(shill::InputData* data);
     82   void OnReadError(const std::string& error_msg);
     83 
     84   std::unique_ptr<shill::Sockets> sockets_;
     85   EventCallback event_callback_;
     86 
     87   // File path for interprocess communication with hostapd.
     88   std::string dest_path_;
     89   std::string local_path_;
     90 
     91   // Socket descriptor for communication with hostapd.
     92   int hostapd_socket_;
     93 
     94   base::Callback<void(shill::InputData *)> hostapd_callback_;
     95   std::unique_ptr<shill::IOHandler> hostapd_input_handler_;
     96   shill::IOHandlerFactory *io_handler_factory_;
     97   EventDispatcher* event_dispatcher_;
     98   base::WeakPtrFactory<HostapdMonitor> weak_ptr_factory_;
     99 
    100   int hostapd_ctrl_iface_check_count_;
    101   base::CancelableClosure attach_timeout_callback_;
    102 
    103   bool started_;
    104 
    105   DISALLOW_COPY_AND_ASSIGN(HostapdMonitor);
    106 };
    107 
    108 }  // namespace apmanager
    109 
    110 #endif  // APMANAGER_HOSTAPD_MONITOR_H_
    111