NetUdpSocket.cpp

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

#include "NetUdpSocket.hpp"
#include "Net.hpp"
#include "NetUdp.hpp"
#include "NetArp.hpp"

uint16_t NetUdpSocket::sAutomaticPort = 0xC000;

NetUdpSocket::NetUdpSocket() {
    mLocalPort = 0;
    NetUdp::addSocket(*this);
}

NetUdpSocket::~NetUdpSocket() {
    NetUdp::removeSocket(*this);
    RX_PACKET* pkt;
    while (mRxMsgBox.take(pkt, 0)) {
        discardRxPacket(pkt);
    }
}

void NetUdpSocket::open(NetDefs::UDP_PORT localPort) {
    if (localPort == 0) {
        mLocalPort = sAutomaticPort++;
        if (sAutomaticPort == 0) {
            sAutomaticPort = 0xC000;
        }
    }
    else {
        mLocalPort = localPort;
    }
}

void NetUdpSocket::close() {
    mLocalPort = 0;
}

NetUdpSocket::TX_PACKET* NetUdpSocket::allocTxPacket(size_t maxDataLen) {
    if ((sizeof(NetDefs::MAC_HEADER) + sizeof(NetDefs::IP_HEADER) + sizeof(NetDefs::UDP_HEADER) + maxDataLen) > NetDefs::MAX_FRAME_LEN) {
        return 0;
    }
    NetDefs::FRAME* frame = Net::allocFrame(sizeof(NetDefs::MAC_HEADER) + sizeof(NetDefs::IP_HEADER) + sizeof(NetDefs::UDP_HEADER) + maxDataLen);

    TX_PACKET* pkt = (TX_PACKET*) frame;
    if (pkt) {
        pkt->data = frame->types.udp.data;
    }

    return pkt;
}

void NetUdpSocket::discardTxPacket(TX_PACKET* pkt) {
    Net::freeFrame(*(NetDefs::FRAME*) pkt);
}

bool NetUdpSocket::sendPacket(TX_PACKET& pkt, size_t dataLen) {
    NetDefs::MAC_ADDR macAddr;
    NetDefs::IP_ADDR dstIpAddr = pkt.hdr.ip;
    if (!NetArp::resolve(dstIpAddr, macAddr)) {
        Net::freeFrame((NetDefs::FRAME&) pkt);
        return false;
    }
    NetDefs::UDP_PORT dstPort = pkt.hdr.port;
    NetUdp::putHeader((NetDefs::FRAME&) pkt, macAddr, dstIpAddr, mLocalPort, dstPort, dataLen);
    NetUdp::flush((NetDefs::FRAME&) pkt, dataLen, true);

    return true;
}

NetUdpSocket::RX_PACKET* NetUdpSocket::recvPacket(Rtos::TICK timeout) {
    RX_PACKET* pkt;
    if (!mRxMsgBox.take(pkt, timeout)) {
        return 0;
    }
    return pkt;
}

void NetUdpSocket::discardRxPacket(RX_PACKET* pkt) {
    Net::freeFrame((NetDefs::FRAME&) *pkt);
}