StreamTestEntity.h

Go to the documentation of this file.

#ifndef STREAM_TEST_ENTITY_H
#define STREAM_TEST_ENTITY_H

#include "HdlcdClient.h"
#include "Statistic.h"
#include "SnetProbeRequest.h"
#include "SnetProbeReply.h"
#include "SnetPeerStatistic.h"

class StreamTestEntity {
public:
    StreamTestEntity(HdlcdClient& a_HdlcdClient, uint16_t a_SrcSSA, uint16_t a_DstSSA): m_SrcSSA(a_SrcSSA), m_DstSSA(a_DstSSA), m_HdlcdClient(a_HdlcdClient), m_LocalSeqNbr(0), m_RemoteStatistic("remote      "), m_LocalStatistic("local       ") {
        // Init
        srand(::time(NULL));
        m_LocalSeed = rand();
        m_RemoteSeed = 0;
        m_TotalBytesWritten = 0;
        m_HdlcdClient.SetOnDataCallback([this](const HdlcdPacketData& a_PacketData) { 
            PacketReceived(a_PacketData);
        });
    }
    
    void Start() {
        // Trigger activity
        SendNextProbeRequest();
    }

private:
    // Helpers
    void PacketReceived(const HdlcdPacketData& a_PacketData) {
        SnetAppMessage l_AppMessage;
        if (l_AppMessage.Deserialize(a_PacketData.GetData())) {
            // Checks: Unicast SSAs must match, but do not try to distinguish streams if non-unicast addresses were used!
            if (((l_AppMessage.GetSrcSSA() == m_DstSSA) || (m_DstSSA >= 0x3FF0)) && (l_AppMessage.GetSrcServiceId() == 0x22) && (l_AppMessage.GetDstServiceId() == 0x22)) {
                if (l_AppMessage.GetToken() == 0x01) {
                    HandleProbeReply(a_PacketData);
                } else if (l_AppMessage.GetToken() == 0x02) {
                    HandlePeerStatistics(a_PacketData);
                } // else if
            } // if
        } // if
    }

    void HandleProbeReply(const HdlcdPacketData& a_PacketData) {
        SnetProbeReply l_ProbeReply;
        if (l_ProbeReply.Deserialize(a_PacketData.GetData())) {
            UpdateStatistics(l_ProbeReply.GetRemoteSeed(), l_ProbeReply.GetLocalSeed(), l_ProbeReply.GetRemoteSeqNbr(), l_ProbeReply.GetLocalSeqNbr());
        } // if
    }
    
    void HandlePeerStatistics(const HdlcdPacketData& a_PacketData) {
        SnetPeerStatistic l_PeerStatistic;
        if (l_PeerStatistic.Deserialize(a_PacketData.GetData())) {        
            // Print
            std::cout << "New peer statistics, Gen = " << l_PeerStatistic.GetLocalGeneration() << ", Amount = " << l_PeerStatistic.GetAmountOfPackets();
            if (l_PeerStatistic.GetAmountOfPackets() == 256) {
                std::cout << std::endl;
            } else {
                std::cout << " (!!!)" << std::endl;
            } // else
        } // if
    }
    
    void UpdateStatistics(uint32_t a_RemoteSeed, uint32_t a_LocalSeed, uint32_t a_RemoteSeqNbr, uint32_t a_LocalSeqNbr) {
        // Update statistics
        if (m_LocalSeed != a_LocalSeed) {
            return;
        } // if
            
        if (m_RemoteSeed != a_RemoteSeed) {
            m_RemoteSeed = a_RemoteSeed;
            m_RemoteStatistic.Reset();
            m_LocalStatistic.Reset();
        } // if

        m_RemoteStatistic.Update(a_RemoteSeqNbr);
        m_LocalStatistic.Update(a_LocalSeqNbr);
    }

    void SendNextProbeRequest() {
        // Create probe request
        SnetProbeRequest l_ProbeRequest(m_SrcSSA, m_DstSSA);
        l_ProbeRequest.SetLocalSeed  (m_LocalSeed);
        l_ProbeRequest.SetLocalSeqNbr(m_LocalSeqNbr);

        // Send one probe request, and if done, call this method again via a provided lambda-callback
        if (m_HdlcdClient.Send(std::move(HdlcdPacketData::CreatePacket(l_ProbeRequest.Serialize(), true)), [this](){ SendNextProbeRequest(); })) {
            // One packet is on its way
            m_TotalBytesWritten += l_ProbeRequest.GetSize();
            if (((++m_LocalSeqNbr) % 0xFF) == 0) {
                std::cout << "256 Packets written to the TCP socket (" << std::dec << m_TotalBytesWritten << " bytes total)" << std::endl;
            } // if
        } // if
    }
    
    // Members
    uint16_t m_SrcSSA;
    uint16_t m_DstSSA;
    uint32_t m_RemoteSeed;
    uint32_t m_LocalSeed;
    HdlcdClient& m_HdlcdClient;
    uint32_t m_LocalSeqNbr;
    Statistic m_RemoteStatistic;
    Statistic m_LocalStatistic;
    unsigned long m_TotalBytesWritten;
};

#endif // STREAM_TEST_ENTITY_H