Diff: ip4/udptcp4.c

Revision:

61:aad055f1b0d1

Parent:

59:e0e556c8bd46

Child:

71:736a5747ade1

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/ip4/udptcp4.c	Thu Jan 11 17:38:21 2018 +0000
@@ -0,0 +1,112 @@
+#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    "udp.h"
+
+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;
+    
+    Ip4AddressFromDest(ActionGetDestPart(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;
+    }
+    
+    if (ActionGetTracePart(action))
+    {
+        switch (pro)
+        {
+            case TCP: TcpLogHeader(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: TcpLogHeader(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)
+{   
+    pTraceBack = traceback;
+    tracePacketProtocol = TCP;
+    calculatedChecksum = calculateChecksum(TCP, *pSrcIp, *pDstIp, sizeRx, pPacketRx);
+        
+    int action = TcpHandleReceivedPacket(trace, sizeRx, pPacketRx, pSizeTx, pPacketTx);
+    
+    *pDstIp = *pSrcIp;
+    
+    finalisePacket(TCP, action, pPacketTx, *pSizeTx, pSrcIp, pDstIp);
+  
+    return action;
+}
+
+int Udp4HandleReceivedPacket(void (*traceback)(void), void* pPacketRx, int sizeRx, void* pPacketTx, int* pSizeTx, uint32_t* pSrcIp, uint32_t* pDstIp)
+{   
+    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
+  
+    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;
+}