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.c
- Committer:
- andrewboyson
- Date:
- 2021-05-20
- Revision:
- 200:5acbc41bf469
- Parent:
- 187:122fc1996c86
File content as of revision 200:5acbc41bf469:
#include <stdint.h> #include "log.h" #include "net.h" #include "action.h" #include "dhcp.h" #include "eth.h" #include "ip.h" #include "ip4addr.h" #include "tcp.h" #include "tcprecv.h" #include "tcpsend.h" #include "tcphdr.h" #include "udp.h" #include "ar4.h" #include "led.h" #include "restart.h" #include "checksum.h" static uint16_t calculateChecksum(uint8_t pro, uint32_t srcIp, uint32_t dstIp, int size, void* pPacket) { uint32_t sum = 0; uint16_t pro16 = pro; sum = CheckSumAddDirect(sum, 4, &srcIp ); sum = CheckSumAddDirect(sum, 4, &dstIp ); sum = CheckSumAddInvert(sum, 2, &pro16 ); sum = CheckSumAddInvert(sum, 2, &size ); return CheckSumFinDirect(sum, size, pPacket); } static void finalisePacket(uint8_t pro, int action, void* pPacket, int size, uint32_t* pSrcIp, uint32_t* pDstIp) { if (!action) return; Ip4AddrFromDest(ActionGetDestPart(action), pDstIp); *pSrcIp = DhcpLocalIp; switch (pro) { case TCP: TcpHdrWriteToPacket(pPacket); break; case UDP: UdpMakeHeader(size, pPacket); break; } uint16_t checksum = calculateChecksum(pro, *pSrcIp, *pDstIp, size, pPacket); switch (pro) { case TCP: TcpHdrSetChecksum(pPacket, checksum); break; case UDP: UdpHdrSetChecksum(pPacket, checksum); break; } if (ActionGetTracePart(action)) { switch (pro) { case TCP: TcpHdrLog(0); break; case UDP: UdpLogHeader(0); break; } } } static void (*pTraceBack)(void); static int tracePacketProtocol; static uint16_t calculatedChecksum; static void trace() { pTraceBack(); switch(tracePacketProtocol) { case UDP: UdpLogHeader(calculatedChecksum); break; case TCP: TcpHdrLog(calculatedChecksum); break; default: LogTimeF("UdpTcp4 - traceback unrecognised protocol %d\r\n", tracePacketProtocol); break; } } int Tcp4HandleReceivedPacket(void (*traceback)(void), void* pPacketRx, int sizeRx, void* pPacketTx, int* pSizeTx, uint32_t* pSrcIp, uint32_t* pDstIp, int remArIndex) { int lastRestartPoint = RestartPoint; RestartPoint = FAULT_POINT_Tcp4HandleReceivedPacket; pTraceBack = traceback; tracePacketProtocol = TCP; calculatedChecksum = calculateChecksum(TCP, *pSrcIp, *pDstIp, sizeRx, pPacketRx); int action = TcpHandleReceivedPacket(trace, sizeRx, pPacketRx, pSizeTx, pPacketTx, ETH_IPV4, remArIndex, 0); *pDstIp = *pSrcIp; finalisePacket(TCP, action, pPacketTx, *pSizeTx, pSrcIp, pDstIp); RestartPoint = lastRestartPoint; return action; } int Udp4HandleReceivedPacket(void (*traceback)(void), void* pPacketRx, int sizeRx, void* pPacketTx, int* pSizeTx, uint32_t* pSrcIp, uint32_t* pDstIp) { int lastRestartPoint = RestartPoint; RestartPoint = FAULT_POINT_Udp4HandleReceivedPacket; pTraceBack = traceback; tracePacketProtocol = UDP; calculatedChecksum = calculateChecksum(UDP, *pSrcIp, *pDstIp, sizeRx, pPacketRx); int action = UdpHandleReceivedPacket(trace, sizeRx, pPacketRx, pSizeTx, pPacketTx); *pDstIp = *pSrcIp; finalisePacket(UDP, action, pPacketTx, *pSizeTx, pSrcIp, pDstIp); //Note that the ports are reversed here RestartPoint = lastRestartPoint; return action; } int Tcp4PollForPacketToSend(void* pPacket, int* pSize, uint32_t* pSrcIp, uint32_t* pDstIp) { int remArIndex = -1; int action = TcpPollForPacketToSend(pSize, pPacket, ETH_IPV4, &remArIndex, NULL); if (action && remArIndex >= 0) *pDstIp = Ar4IndexToIp(remArIndex); finalisePacket(TCP, action, pPacket, *pSize, pSrcIp, pDstIp); return action; } int Udp4PollForPacketToSend(void* pPacket, int* pSize, uint32_t* pSrcIp, uint32_t* pDstIp) { int action = UdpPollForPacketToSend(ETH_IPV4, pSize, pPacket); finalisePacket(UDP, action, pPacket, *pSize, pSrcIp, pDstIp); return action; }