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@97:d91f7db00235, 2019-01-02 (annotated)

Committer:
andrewboyson
Date:
Wed Jan 02 17:48:38 2019 +0000
Revision:
97:d91f7db00235
Parent:
96:43eb7a110f1a
Child:
98:b977424ec7f7
Added fault points

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 97:d91f7db00235 84 FaultPoint = FAULT_POINT_Tcp4HandleReceivedPacket;
andrewboyson 97:d91f7db00235 85
andrewboyson 37:793b39683406 86 pTraceBack = traceback;
andrewboyson 37:793b39683406 87 tracePacketProtocol = TCP;
andrewboyson 59:e0e556c8bd46 88 calculatedChecksum = calculateChecksum(TCP, *pSrcIp, *pDstIp, sizeRx, pPacketRx);
andrewboyson 10:f0854784e960 89
andrewboyson 80:4ef1500fca1d 90 int action = TcpHandleReceivedPacket(trace, sizeRx, pPacketRx, pSizeTx, pPacketTx, IPV4, remArIndex, 0);
andrewboyson 10:f0854784e960 91
andrewboyson 10:f0854784e960 92 *pDstIp = *pSrcIp;
andrewboyson 10:f0854784e960 93
andrewboyson 59:e0e556c8bd46 94 finalisePacket(TCP, action, pPacketTx, *pSizeTx, pSrcIp, pDstIp);
andrewboyson 10:f0854784e960 95
andrewboyson 97:d91f7db00235 96 FaultPoint = 0;
andrewboyson 10:f0854784e960 97 return action;
andrewboyson 10:f0854784e960 98 }
andrewboyson 10:f0854784e960 99
andrewboyson 59:e0e556c8bd46 100 int Udp4HandleReceivedPacket(void (*traceback)(void), void* pPacketRx, int sizeRx, void* pPacketTx, int* pSizeTx, uint32_t* pSrcIp, uint32_t* pDstIp)
andrewboyson 10:f0854784e960 101 {
andrewboyson 37:793b39683406 102 pTraceBack = traceback;
andrewboyson 37:793b39683406 103 tracePacketProtocol = UDP;
andrewboyson 59:e0e556c8bd46 104 calculatedChecksum = calculateChecksum(UDP, *pSrcIp, *pDstIp, sizeRx, pPacketRx);
andrewboyson 59:e0e556c8bd46 105
andrewboyson 59:e0e556c8bd46 106 int action = UdpHandleReceivedPacket(trace, sizeRx, pPacketRx, pSizeTx, pPacketTx);
andrewboyson 10:f0854784e960 107
andrewboyson 10:f0854784e960 108 *pDstIp = *pSrcIp;
andrewboyson 10:f0854784e960 109
andrewboyson 59:e0e556c8bd46 110 finalisePacket(UDP, action, pPacketTx, *pSizeTx, pSrcIp, pDstIp); //Note that the ports are reversed here
andrewboyson 86:55bc5ddac16c 111
andrewboyson 10:f0854784e960 112 return action;
andrewboyson 10:f0854784e960 113 }
andrewboyson 71:736a5747ade1 114 int Tcp4PollForPacketToSend(void* pPacket, int* pSize, uint32_t* pSrcIp, uint32_t* pDstIp)
andrewboyson 71:736a5747ade1 115 {
andrewboyson 74:c3756bfa960e 116 int remArIndex = -1;
andrewboyson 80:4ef1500fca1d 117 int action = TcpPollForPacketToSend(pSize, pPacket, IPV4, &remArIndex, NULL);
andrewboyson 74:c3756bfa960e 118 if (action && remArIndex >= 0) *pDstIp = Ar4IndexToIp(remArIndex);
andrewboyson 71:736a5747ade1 119
andrewboyson 71:736a5747ade1 120 finalisePacket(TCP, action, pPacket, *pSize, pSrcIp, pDstIp);
andrewboyson 71:736a5747ade1 121
andrewboyson 71:736a5747ade1 122 return action;
andrewboyson 71:736a5747ade1 123 }
andrewboyson 10:f0854784e960 124 int Udp4PollForPacketToSend(void* pPacket, int* pSize, uint32_t* pSrcIp, uint32_t* pDstIp)
andrewboyson 10:f0854784e960 125 {
andrewboyson 10:f0854784e960 126 int action = UdpPollForPacketToSend(IPV4, pSize, pPacket);
andrewboyson 10:f0854784e960 127
andrewboyson 10:f0854784e960 128 finalisePacket(UDP, action, pPacket, *pSize, pSrcIp, pDstIp);
andrewboyson 10:f0854784e960 129
andrewboyson 10:f0854784e960 130 return action;
andrewboyson 10:f0854784e960 131 }