lib/wlan/mlme/include/wlan/mlme/ap/beacon_sender.h

// Copyright 2017 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef GARNET_LIB_WLAN_MLME_INCLUDE_WLAN_MLME_AP_BEACON_SENDER_H_
#define GARNET_LIB_WLAN_MLME_INCLUDE_WLAN_MLME_AP_BEACON_SENDER_H_

#include <wlan/mlme/device_interface.h>

#include <fuchsia/wlan/mlme/cpp/fidl.h>

#include <zircon/types.h>

namespace wlan {

class BssInterface;
class PsCfg;
template <typename T> class MlmeMsg;

// Configures the driver to send Beacon frames periodically.
class BeaconSender {
   public:
    explicit BeaconSender(DeviceInterface* device);
    ~BeaconSender();

    void Start(BssInterface* bss, const PsCfg& ps_cfg,
               const MlmeMsg<::fuchsia::wlan::mlme::StartRequest>& req);
    void Stop();
    zx_status_t UpdateBeacon(const PsCfg& ps_cfg);
    void SendProbeResponse(const common::MacAddr& recv_addr, Span<const uint8_t> ie_chain);

   private:
    zx_status_t BuildBeacon(const PsCfg& ps_cfg, MgmtFrame<Beacon>* frame, size_t* tim_ele_offset);

    bool IsStarted();

    DeviceInterface* const device_;
    ::fuchsia::wlan::mlme::StartRequest req_;
    BssInterface* bss_ = nullptr;
};

// Visible for testing
bool ShouldSendProbeResponse(Span<const uint8_t> ie_chain, Span<const uint8_t> our_ssid);

}  // namespace wlan

#endif  // GARNET_LIB_WLAN_MLME_INCLUDE_WLAN_MLME_AP_BEACON_SENDER_H_