sta-wifi-mac.h

Go to the documentation of this file.

/*
 * Copyright (c) 2006, 2009 INRIA
 * Copyright (c) 2009 MIRKO BANCHI
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation;
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 * Authors: Mathieu Lacage <mathieu.lacage@sophia.inria.fr>
 *          Mirko Banchi <mk.banchi@gmail.com>
 */
 
#ifndef STA_WIFI_MAC_H
#define STA_WIFI_MAC_H
 
#include "mgt-headers.h"
#include "wifi-mac.h"
 
#include <set>
#include <variant>
 
class TwoLevelAggregationTest;
class AmpduAggregationTest;
class HeAggregationTest;
class MultiLinkOperationsTestBase;
 
namespace ns3
{
 
class SupportedRates;
class CapabilityInformation;
class RandomVariableStream;
class WifiAssocManager;
 
struct WifiScanParams
{
    struct Channel
    {
        uint16_t number{0};                          
        WifiPhyBand band{WIFI_PHY_BAND_UNSPECIFIED}; 
    };
 
    using ChannelList = std::list<Channel>;
 
    enum : uint8_t
    {
        ACTIVE = 0,
        PASSIVE
    } type; 
 
    Ssid ssid;                            
    std::vector<ChannelList> channelList; 
    Time probeDelay;                      
    Time minChannelTime;                  
    Time maxChannelTime;                  
};
 
class StaWifiMac : public WifiMac
{
  public:
    friend class ::TwoLevelAggregationTest;
    friend class ::AmpduAggregationTest;
    friend class ::HeAggregationTest;
    friend class ::MultiLinkOperationsTestBase;
 
    using MgtFrameType =
        std::variant<MgtBeaconHeader, MgtProbeResponseHeader, MgtAssocResponseHeader>;
 
    struct ApInfo
    {
        Mac48Address m_bssid;  
        Mac48Address m_apAddr; 
        double m_snr;          
        MgtFrameType m_frame;  
        WifiScanParams::Channel m_channel; 
        uint8_t m_linkId;                  
        std::list<std::pair<std::uint8_t, uint8_t>> m_setupLinks;
    };
 
    static TypeId GetTypeId();
 
    StaWifiMac();
    ~StaWifiMac() override;
 
    void Enqueue(Ptr<Packet> packet, Mac48Address to) override;
    bool CanForwardPacketsTo(Mac48Address to) const override;
 
    void SetWifiPhys(const std::vector<Ptr<WifiPhy>>& phys) override;
 
    void SetAssocManager(Ptr<WifiAssocManager> assocManager);
 
    void SendProbeRequest();
 
    void ScanningTimeout(const std::optional<ApInfo>& bestAp);
 
    bool IsAssociated() const;
 
    std::set<uint8_t> GetSetupLinkIds() const;
 
    uint16_t GetAssociationId() const;
 
    void NotifyChannelSwitching(uint8_t linkId) override;
 
    int64_t AssignStreams(int64_t stream);
 
  protected:
    struct StaLinkEntity : public WifiMac::LinkEntity
    {
        ~StaLinkEntity() override;
 
        bool sendAssocReq;                 
        std::optional<uint8_t> apLinkId;   
        std::optional<Mac48Address> bssid; 
        EventId beaconWatchdog;            
        Time beaconWatchdogEnd{0};         
    };
 
    StaLinkEntity& GetLink(uint8_t linkId) const;
 
  private:
    enum MacState
    {
        ASSOCIATED,
        SCANNING,
        WAIT_ASSOC_RESP,
        UNASSOCIATED,
        REFUSED
    };
 
    void SetActiveProbing(bool enable);
    bool GetActiveProbing() const;
 
    bool CheckSupportedRates(std::variant<MgtBeaconHeader, MgtProbeResponseHeader> frame,
                             uint8_t linkId);
 
    void Receive(Ptr<const WifiMpdu> mpdu, uint8_t linkId) override;
    std::unique_ptr<LinkEntity> CreateLinkEntity() const override;
    Mac48Address DoGetLocalAddress(const Mac48Address& remoteAddr) const override;
 
    void ReceiveBeacon(Ptr<const WifiMpdu> mpdu, uint8_t linkId);
 
    void ReceiveProbeResp(Ptr<const WifiMpdu> mpdu, uint8_t linkId);
 
    void ReceiveAssocResp(Ptr<const WifiMpdu> mpdu, uint8_t linkId);
 
    void UpdateApInfo(const MgtFrameType& frame,
                      const Mac48Address& apAddr,
                      const Mac48Address& bssid,
                      uint8_t linkId);
 
    std::variant<MgtAssocRequestHeader, MgtReassocRequestHeader> GetAssociationRequest(
        bool isReassoc,
        uint8_t linkId) const;
 
    void SendAssociationRequest(bool isReassoc);
    void TryToEnsureAssociated();
    void AssocRequestTimeout();
    void StartScanning();
    bool IsWaitAssocResp() const;
    void MissedBeacons(uint8_t linkId);
    void RestartBeaconWatchdog(Time delay, uint8_t linkId);
    void Disassociated(uint8_t linkId);
    SupportedRates GetSupportedRates(uint8_t linkId) const;
    MultiLinkElement GetMultiLinkElement(bool isReassoc, uint8_t linkId) const;
    void SetState(MacState value);
    void SetEdcaParameters(AcIndex ac,
                           uint32_t cwMin,
                           uint32_t cwMax,
                           uint8_t aifsn,
                           Time txopLimit);
    void SetMuEdcaParameters(AcIndex ac,
                             uint16_t cwMin,
                             uint16_t cwMax,
                             uint8_t aifsn,
                             Time muEdcaTimer);
    CapabilityInformation GetCapabilities(uint8_t linkId) const;
 
    void PhyCapabilitiesChanged();
 
    WifiScanParams::Channel GetCurrentChannel(uint8_t linkId) const;
 
    void DoInitialize() override;
    void DoDispose() override;
 
    MacState m_state;                       
    uint16_t m_aid;                         
    Ptr<WifiAssocManager> m_assocManager;   
    Time m_waitBeaconTimeout;               
    Time m_probeRequestTimeout;             
    Time m_assocRequestTimeout;             
    EventId m_assocRequestEvent;            
    uint32_t m_maxMissedBeacons;            
    bool m_activeProbing;                   
    Ptr<RandomVariableStream> m_probeDelay; 
 
    TracedCallback<Mac48Address> m_assocLogger;             
    TracedCallback<uint8_t, Mac48Address> m_setupCompleted; 
    TracedCallback<Mac48Address> m_deAssocLogger;           
    TracedCallback<uint8_t, Mac48Address> m_setupCanceled;  
    TracedCallback<Time> m_beaconArrival;                   
    TracedCallback<ApInfo> m_beaconInfo;                    
 
    using LinkSetupCallback = void (*)(uint8_t /* link ID */, Mac48Address /* AP address */);
};
 
std::ostream& operator<<(std::ostream& os, const StaWifiMac::ApInfo& apInfo);
 
} // namespace ns3
 
#endif /* STA_WIFI_MAC_H */