Andrew Boyson / net

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@125:8c84daac38ab, 2019-02-22 (annotated)

Committer:
andrewboyson
Date:
Fri Feb 22 11:26:55 2019 +0000
Revision:
125:8c84daac38ab
Parent:
121:bc048b65a630
Child:
136:8a65abb0dc63
tidied up ntp data types for calculating ms from clktime.

Who changed what in which revision?

UserRevisionLine numberNew contents of line
andrewboyson 61:aad055f1b0d1 1 #include <stdint.h>
andrewboyson 61:aad055f1b0d1 2
andrewboyson 97:d91f7db00235 3 #include "log.h"
andrewboyson 97:d91f7db00235 4 #include "net.h"
andrewboyson 97:d91f7db00235 5 #include "action.h"
andrewboyson 97:d91f7db00235 6 #include "dhcp.h"
andrewboyson 97:d91f7db00235 7 #include "eth.h"
andrewboyson 97:d91f7db00235 8 #include "ip.h"
andrewboyson 49:1a6336f2b3f9 9 #include "ip4addr.h"
andrewboyson 74:c3756bfa960e 10 #include "tcp.h"
andrewboyson 74:c3756bfa960e 11 #include "tcprecv.h"
andrewboyson 74:c3756bfa960e 12 #include "tcpsend.h"
andrewboyson 74:c3756bfa960e 13 #include "tcphdr.h"
andrewboyson 74:c3756bfa960e 14 #include "udp.h"
andrewboyson 86:55bc5ddac16c 15 #include "ar4.h"
andrewboyson 86:55bc5ddac16c 16 #include "led.h"
andrewboyson 97:d91f7db00235 17 #include "fault.h"
andrewboyson 10:f0854784e960 18
andrewboyson 10:f0854784e960 19 static uint16_t calculateChecksum(uint8_t pro, uint32_t srcIp, uint32_t dstIp, int size, void* pPacket)
andrewboyson 10:f0854784e960 20 {
andrewboyson 10:f0854784e960 21 __packed struct pseudo
andrewboyson 10:f0854784e960 22 {
andrewboyson 10:f0854784e960 23 uint32_t src;
andrewboyson 10:f0854784e960 24 uint32_t dst;
andrewboyson 10:f0854784e960 25 uint8_t zer;
andrewboyson 10:f0854784e960 26 uint8_t pro;
andrewboyson 10:f0854784e960 27 uint16_t len;
andrewboyson 10:f0854784e960 28 } pseudo;
andrewboyson 10:f0854784e960 29
andrewboyson 10:f0854784e960 30 pseudo.src = srcIp;
andrewboyson 10:f0854784e960 31 pseudo.dst = dstIp;
andrewboyson 10:f0854784e960 32 pseudo.zer = 0;
andrewboyson 10:f0854784e960 33 pseudo.pro = pro;
andrewboyson 10:f0854784e960 34 pseudo.len = NetToHost16(size);
andrewboyson 10:f0854784e960 35
andrewboyson 10:f0854784e960 36 return NetCheckSumTwo(sizeof(pseudo), &pseudo, size, pPacket);
andrewboyson 10:f0854784e960 37 }
andrewboyson 10:f0854784e960 38 static void finalisePacket(uint8_t pro, int action, void* pPacket, int size, uint32_t* pSrcIp, uint32_t* pDstIp)
andrewboyson 10:f0854784e960 39 {
andrewboyson 10:f0854784e960 40 if (!action) return;
andrewboyson 10:f0854784e960 41
andrewboyson 49:1a6336f2b3f9 42 Ip4AddressFromDest(ActionGetDestPart(action), pDstIp);
andrewboyson 10:f0854784e960 43 *pSrcIp = DhcpLocalIp;
andrewboyson 10:f0854784e960 44
andrewboyson 10:f0854784e960 45 switch (pro)
andrewboyson 10:f0854784e960 46 {
andrewboyson 74:c3756bfa960e 47 case TCP: TcpHdrWriteToPacket(pPacket); break;
andrewboyson 10:f0854784e960 48 case UDP: UdpMakeHeader(size, pPacket); break;
andrewboyson 10:f0854784e960 49 }
andrewboyson 10:f0854784e960 50
andrewboyson 10:f0854784e960 51 uint16_t checksum = calculateChecksum(pro, *pSrcIp, *pDstIp, size, pPacket);
andrewboyson 10:f0854784e960 52
andrewboyson 10:f0854784e960 53 switch (pro)
andrewboyson 10:f0854784e960 54 {
andrewboyson 74:c3756bfa960e 55 case TCP: TcpHdrAddChecksumToPacket(pPacket, checksum); break;
andrewboyson 10:f0854784e960 56 case UDP: UdpAddChecksum(pPacket, checksum); break;
andrewboyson 10:f0854784e960 57 }
andrewboyson 10:f0854784e960 58
andrewboyson 37:793b39683406 59 if (ActionGetTracePart(action))
andrewboyson 11:c051adb70c5a 60 {
andrewboyson 37:793b39683406 61 switch (pro)
andrewboyson 37:793b39683406 62 {
andrewboyson 73:43e3d7fb3d60 63 case TCP: TcpHdrLog(0); break;
andrewboyson 56:35117a8b5c65 64 case UDP: UdpLogHeader(0); break;
andrewboyson 37:793b39683406 65 }
andrewboyson 11:c051adb70c5a 66 }
andrewboyson 10:f0854784e960 67 }
andrewboyson 10:f0854784e960 68
andrewboyson 37:793b39683406 69 static void (*pTraceBack)(void);
andrewboyson 37:793b39683406 70 static int tracePacketProtocol;
andrewboyson 37:793b39683406 71 static uint16_t calculatedChecksum;
andrewboyson 37:793b39683406 72 static void trace()
andrewboyson 37:793b39683406 73 {
andrewboyson 37:793b39683406 74 pTraceBack();
andrewboyson 37:793b39683406 75 switch(tracePacketProtocol)
andrewboyson 37:793b39683406 76 {
andrewboyson 56:35117a8b5c65 77 case UDP: UdpLogHeader(calculatedChecksum); break;
andrewboyson 73:43e3d7fb3d60 78 case TCP: TcpHdrLog(calculatedChecksum); break;
andrewboyson 37:793b39683406 79 default: LogTimeF("UdpTcp4 - traceback unrecognised protocol %d\r\n", tracePacketProtocol); break;
andrewboyson 37:793b39683406 80 }
andrewboyson 37:793b39683406 81 }
andrewboyson 74:c3756bfa960e 82 int Tcp4HandleReceivedPacket(void (*traceback)(void), void* pPacketRx, int sizeRx, void* pPacketTx, int* pSizeTx, uint32_t* pSrcIp, uint32_t* pDstIp, int remArIndex)
andrewboyson 97:d91f7db00235 83 {
andrewboyson 98:b977424ec7f7 84 int lastFaultPoint = FaultPoint;
andrewboyson 97:d91f7db00235 85 FaultPoint = FAULT_POINT_Tcp4HandleReceivedPacket;
andrewboyson 97:d91f7db00235 86
andrewboyson 37:793b39683406 87 pTraceBack = traceback;
andrewboyson 37:793b39683406 88 tracePacketProtocol = TCP;
andrewboyson 59:e0e556c8bd46 89 calculatedChecksum = calculateChecksum(TCP, *pSrcIp, *pDstIp, sizeRx, pPacketRx);
andrewboyson 10:f0854784e960 90
andrewboyson 80:4ef1500fca1d 91 int action = TcpHandleReceivedPacket(trace, sizeRx, pPacketRx, pSizeTx, pPacketTx, IPV4, remArIndex, 0);
andrewboyson 10:f0854784e960 92
andrewboyson 10:f0854784e960 93 *pDstIp = *pSrcIp;
andrewboyson 10:f0854784e960 94
andrewboyson 59:e0e556c8bd46 95 finalisePacket(TCP, action, pPacketTx, *pSizeTx, pSrcIp, pDstIp);
andrewboyson 10:f0854784e960 96
andrewboyson 98:b977424ec7f7 97 FaultPoint = lastFaultPoint;
andrewboyson 10:f0854784e960 98 return action;
andrewboyson 10:f0854784e960 99 }
andrewboyson 10:f0854784e960 100
andrewboyson 59:e0e556c8bd46 101 int Udp4HandleReceivedPacket(void (*traceback)(void), void* pPacketRx, int sizeRx, void* pPacketTx, int* pSizeTx, uint32_t* pSrcIp, uint32_t* pDstIp)
andrewboyson 10:f0854784e960 102 {
andrewboyson 121:bc048b65a630 103 int lastFaultPoint = FaultPoint;
andrewboyson 121:bc048b65a630 104 FaultPoint = FAULT_POINT_Udp4HandleReceivedPacket;
andrewboyson 121:bc048b65a630 105
andrewboyson 37:793b39683406 106 pTraceBack = traceback;
andrewboyson 37:793b39683406 107 tracePacketProtocol = UDP;
andrewboyson 59:e0e556c8bd46 108 calculatedChecksum = calculateChecksum(UDP, *pSrcIp, *pDstIp, sizeRx, pPacketRx);
andrewboyson 59:e0e556c8bd46 109
andrewboyson 59:e0e556c8bd46 110 int action = UdpHandleReceivedPacket(trace, sizeRx, pPacketRx, pSizeTx, pPacketTx);
andrewboyson 10:f0854784e960 111
andrewboyson 10:f0854784e960 112 *pDstIp = *pSrcIp;
andrewboyson 10:f0854784e960 113
andrewboyson 59:e0e556c8bd46 114 finalisePacket(UDP, action, pPacketTx, *pSizeTx, pSrcIp, pDstIp); //Note that the ports are reversed here
andrewboyson 86:55bc5ddac16c 115
andrewboyson 121:bc048b65a630 116 FaultPoint = lastFaultPoint;
andrewboyson 10:f0854784e960 117 return action;
andrewboyson 10:f0854784e960 118 }
andrewboyson 71:736a5747ade1 119 int Tcp4PollForPacketToSend(void* pPacket, int* pSize, uint32_t* pSrcIp, uint32_t* pDstIp)
andrewboyson 71:736a5747ade1 120 {
andrewboyson 74:c3756bfa960e 121 int remArIndex = -1;
andrewboyson 80:4ef1500fca1d 122 int action = TcpPollForPacketToSend(pSize, pPacket, IPV4, &remArIndex, NULL);
andrewboyson 74:c3756bfa960e 123 if (action && remArIndex >= 0) *pDstIp = Ar4IndexToIp(remArIndex);
andrewboyson 71:736a5747ade1 124
andrewboyson 71:736a5747ade1 125 finalisePacket(TCP, action, pPacket, *pSize, pSrcIp, pDstIp);
andrewboyson 71:736a5747ade1 126
andrewboyson 71:736a5747ade1 127 return action;
andrewboyson 71:736a5747ade1 128 }
andrewboyson 10:f0854784e960 129 int Udp4PollForPacketToSend(void* pPacket, int* pSize, uint32_t* pSrcIp, uint32_t* pDstIp)
andrewboyson 10:f0854784e960 130 {
andrewboyson 10:f0854784e960 131 int action = UdpPollForPacketToSend(IPV4, pSize, pPacket);
andrewboyson 10:f0854784e960 132
andrewboyson 10:f0854784e960 133 finalisePacket(UDP, action, pPacket, *pSize, pSrcIp, pDstIp);
andrewboyson 10:f0854784e960 134
andrewboyson 10:f0854784e960 135 return action;
andrewboyson 10:f0854784e960 136 }