A stack which works with or without an Mbed os library. Provides IPv4 or IPv6 with a full 1500 byte buffer.
Dependents: oldheating gps motorhome heating
ip4/udptcp4.cpp
- Committer:
- andrewboyson
- Date:
- 2017-05-09
- Revision:
- 15:6ca6778168b1
- Parent:
- 11:c051adb70c5a
- Child:
- 37:793b39683406
File content as of revision 15:6ca6778168b1:
#include "mbed.h" #include "log.h" #include "net.h" #include "dhcp.h" #include "eth.h" #include "ip.h" #include "ip4.h" #include "tcp.h" #include "udp.h" #define DEBUG false static uint16_t calculateChecksum(uint8_t pro, uint32_t srcIp, uint32_t dstIp, int size, void* pPacket) { __packed struct pseudo { uint32_t src; uint32_t dst; uint8_t zer; uint8_t pro; uint16_t len; } pseudo; pseudo.src = srcIp; pseudo.dst = dstIp; pseudo.zer = 0; pseudo.pro = pro; pseudo.len = NetToHost16(size); return NetCheckSumTwo(sizeof(pseudo), &pseudo, size, pPacket); } static void finalisePacket(uint8_t pro, int action, void* pPacket, int size, uint32_t* pSrcIp, uint32_t* pDstIp) { if (!action) return; Ip4DestIpFromAction(action, pDstIp); *pSrcIp = DhcpLocalIp; switch (pro) { case TCP: TcpMakeHeader(size, pPacket); break; case UDP: UdpMakeHeader(size, pPacket); break; } uint16_t checksum = calculateChecksum(pro, *pSrcIp, *pDstIp, size, pPacket); switch (pro) { case TCP: TcpAddChecksum(pPacket, checksum); break; case UDP: UdpAddChecksum(pPacket, checksum); break; } switch (pro) { case TCP: if (DEBUG)TcpLogHeader("IPv4 packet sent", pPacket, 0); break; case UDP: if (DEBUG)UdpLogHeader("IPv4 packet sent", pPacket, 0); break; } } int Tcp4HandleReceivedPacket(uint32_t* pSrcIp, uint32_t* pDstIp, int* pSize, void * pPacket) { uint16_t calculatedChecksum = calculateChecksum(TCP, *pSrcIp, *pDstIp, *pSize, pPacket); if (DEBUG) TcpLogHeader("IPv4 packet received", pPacket, calculatedChecksum); TcpReadHeader(pPacket, *pSize); int action = TcpHandleReceivedPacket(pSize, pPacket); *pDstIp = *pSrcIp; finalisePacket(TCP, action, pPacket, *pSize, pSrcIp, pDstIp); return action; } int Udp4HandleReceivedPacket(uint32_t* pSrcIp, uint32_t* pDstIp, int* pSize, void * pPacket) { uint16_t calculatedChecksum = calculateChecksum(UDP, *pSrcIp, *pDstIp, *pSize, pPacket); if (DEBUG) UdpLogHeader("IPv4 packet received", pPacket, calculatedChecksum); UdpReadHeader(pPacket, *pSize); int action = UdpHandleReceivedPacket(pSize, pPacket); *pDstIp = *pSrcIp; finalisePacket(UDP, action, pPacket, *pSize, pSrcIp, pDstIp); //Note that the ports are reversed here return action; } int Udp4PollForPacketToSend(void* pPacket, int* pSize, uint32_t* pSrcIp, uint32_t* pDstIp) { int action = UdpPollForPacketToSend(IPV4, pSize, pPacket); finalisePacket(UDP, action, pPacket, *pSize, pSrcIp, pDstIp); return action; }