NetUdpSocket.hpp

//------------------------------------------------------------------------------
//This file is part of embKernel. 
//See license.txt for the full license governing this code.
//------------------------------------------------------------------------------

#ifndef NET_UDP_SOCKET_HPP_
#define NET_UDP_SOCKET_HPP_

#include "NetDefs.hpp"
#include "RtosSemaphore.hpp"
#include "LibLinkedList.hpp"
#include "RtosMsgBox.hpp"
#include <stdlib.h>

class NetUdpSocket: public LIB_LINKED_LIST_ITERABLE<NetUdpSocket> {
    friend class NetUdp;

public:
    typedef struct {
        NetDefs::IP_ADDR ip;
        NetDefs::UDP_PORT port;
        uint16_t dataLen;
    }__attribute__((packed)) PACKET_HEADER;

    typedef struct {
        PACKET_HEADER hdr;
        uint8_t data[NetDefs::UDP_MAX_DATA_LEN]; //
    }__attribute__((packed)) RX_PACKET;

    typedef struct {
        PACKET_HEADER hdr;
        uint8_t* data; //
    }__attribute__((packed)) TX_PACKET;

private:
    NetDefs::UDP_PORT mLocalPort;
    static uint16_t sAutomaticPort;
    RtosMsgBox<RX_PACKET*, 4> mRxMsgBox;
public:
    NetUdpSocket();
    ~NetUdpSocket();

    void open(NetDefs::UDP_PORT localPort = 0);
    void close();

    TX_PACKET* allocTxPacket(size_t maxDataLen);
    void discardTxPacket(TX_PACKET* pkt);
    bool sendPacket(TX_PACKET& pkt, size_t dataLen);

    RX_PACKET* recvPacket(Rtos::TICK timeout);
    void discardRxPacket(RX_PACKET* pkt);
};

#endif /* NET_UDP_SOCKET_HPP_ */