udp6.hpp

Go to the documentation of this file.

/*
 *  Copyright (c) 2016, The OpenThread Authors.
 *  All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions are met:
 *  1. Redistributions of source code must retain the above copyright
 *     notice, this list of conditions and the following disclaimer.
 *  2. Redistributions in binary form must reproduce the above copyright
 *     notice, this list of conditions and the following disclaimer in the
 *     documentation and/or other materials provided with the distribution.
 *  3. Neither the name of the copyright holder nor the
 *     names of its contributors may be used to endorse or promote products
 *     derived from this software without specific prior written permission.
 *
 *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 *  AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 *  IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 *  ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 *  LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 *  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 *  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 *  INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 *  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 *  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 *  POSSIBILITY OF SUCH DAMAGE.
 */

#ifndef UDP6_HPP_
#define UDP6_HPP_

#include "openthread-core-config.h"

#include <openthread/udp.h>

#include "common/locator.hpp"
#include "net/ip6_headers.hpp"

namespace ot {
namespace Ip6 {

class Udp;

class UdpSocket: public otUdpSocket
{
    friend class Udp;

public:
    UdpSocket(Udp &aUdp);

    Message *NewMessage(uint16_t aReserved);

    otError Open(otUdpReceive aHandler, void *aContext);

    otError Bind(const SockAddr &aSockAddr);

    otError Connect(const SockAddr &aSockAddr);

    otError Close(void);

    otError SendTo(Message &aMessage, const MessageInfo &aMessageInfo);

    SockAddr &GetSockName(void) { return *static_cast<SockAddr *>(&mSockName); }

    SockAddr &GetPeerName(void) { return *static_cast<SockAddr *>(&mPeerName); }

private:
    UdpSocket *GetNext(void) { return static_cast<UdpSocket *>(mNext); }
    void SetNext(UdpSocket *socket) { mNext = static_cast<otUdpSocket *>(socket); }

    void HandleUdpReceive(Message &aMessage, const MessageInfo &aMessageInfo) {
        mHandler(mContext, &aMessage, &aMessageInfo);
    }
};

class Udp: public InstanceLocator
{
    friend class UdpSocket;

public:
    Udp(otInstance &aInstance);

    otError AddSocket(UdpSocket &aSocket);

    otError RemoveSocket(UdpSocket &aSocket);

    uint16_t GetEphemeralPort(void);

    Message *NewMessage(uint16_t aReserved);

    otError SendDatagram(Message &aMessage, MessageInfo &aMessageInfo, IpProto aIpProto);

    otError HandleMessage(Message &aMessage, MessageInfo &aMessageInfo);

    otError UpdateChecksum(Message &aMessage, uint16_t aPseudoHeaderChecksum);

private:
    enum
    {
        kDynamicPortMin = 49152,  
        kDynamicPortMax = 65535,  
    };
    uint16_t mEphemeralPort;
    UdpSocket *mSockets;
};

OT_TOOL_PACKED_BEGIN
struct UdpHeaderPoD
{
    uint16_t mSource;
    uint16_t mDestination;
    uint16_t mLength;
    uint16_t mChecksum;
} OT_TOOL_PACKED_END;

OT_TOOL_PACKED_BEGIN
class UdpHeader: private UdpHeaderPoD
{
public:
    uint16_t GetSourcePort(void) const { return HostSwap16(mSource); }

    void SetSourcePort(uint16_t aPort) { mSource = HostSwap16(aPort); }

    uint16_t GetDestinationPort(void) const { return HostSwap16(mDestination); }

    void SetDestinationPort(uint16_t aPort) { mDestination = HostSwap16(aPort); }

    uint16_t GetLength(void) const { return HostSwap16(mLength); }

    void SetLength(uint16_t aLength) { mLength = HostSwap16(aLength); }

    uint16_t GetChecksum(void) const { return HostSwap16(mChecksum); }

    void SetChecksum(uint16_t aChecksum) { mChecksum = HostSwap16(aChecksum); }

    static uint8_t GetLengthOffset(void) { return offsetof(UdpHeaderPoD, mLength); }

    static uint8_t GetChecksumOffset(void) { return offsetof(UdpHeaderPoD, mChecksum); }

} OT_TOOL_PACKED_END;

}  // namespace Ip6
}  // namespace ot

#endif  // NET_UDP6_HPP_