UdpComponentImpl.hpp

Go to the documentation of this file.

// ======================================================================
// \title  UdpComponentImpl.hpp
// \author mstarch
// \brief  hpp file for UdpComponentImpl component implementation
//
// \copyright
// Copyright 2009-2020, by the California Institute of Technology.
// ALL RIGHTS RESERVED.  United States Government Sponsorship
// acknowledged.
//
// ======================================================================

#ifndef UdpComponentImpl_HPP
#define UdpComponentImpl_HPP

#include <Drv/Ip/IpSocket.hpp>
#include <Drv/Ip/SocketComponentHelper.hpp>
#include <Drv/Ip/UdpSocket.hpp>
#include "Drv/Udp/UdpComponentAc.hpp"

namespace Drv {

class UdpComponentImpl : public UdpComponentBase, public SocketComponentHelper {
    friend class UdpTester;

  public:
    // ----------------------------------------------------------------------
    // Construction, initialization, and destruction
    // ----------------------------------------------------------------------

    UdpComponentImpl(const char* const compName);

    ~UdpComponentImpl();

    // ----------------------------------------------------------------------
    // Helper methods to start and stop socket
    // ----------------------------------------------------------------------

    SocketIpStatus configureSend(const char* const ipv4_address,
                                 const U16 port,
                                 const U32 send_timeout_seconds = SOCKET_SEND_TIMEOUT_SECONDS,
                                 const U32 send_timeout_microseconds = SOCKET_SEND_TIMEOUT_MICROSECONDS);

    SocketIpStatus configureRecv(const char* const ipv4_address, const U16 port, FwSizeType buffer_size = 1024);

    U16 getRecvPort();

  protected:
    // ----------------------------------------------------------------------
    // Implementations for socket read task virtual methods
    // ----------------------------------------------------------------------

    IpSocket& getSocketHandler() override;

    Fw::Buffer getBuffer() override;

    void sendBuffer(Fw::Buffer buffer, SocketIpStatus status) override;

    void connected() override;

  private:
    // ----------------------------------------------------------------------
    // Handler implementations for user-defined typed input ports
    // ----------------------------------------------------------------------

    Drv::ByteStreamStatus send_handler(const FwIndexType portNum, Fw::Buffer& fwBuffer) override;

    void recvReturnIn_handler(FwIndexType portNum,  
                              Fw::Buffer& fwBuffer  
                              ) override;

    Drv::UdpSocket m_socket;  

    FwSizeType m_allocation_size;  
};

}  // end namespace Drv

#endif  // end UdpComponentImpl