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

udp/udp.c@83:08c983006a6e, 2018-11-17 (annotated)

Committer:
andrewboyson
Date:
Sat Nov 17 15:49:03 2018 +0000
Revision:
83:08c983006a6e
Parent:
61:aad055f1b0d1
Child:
95:b9a99049016a
Added "send requests via IPv4" toggles for DNS, NTP and TFTP.

Who changed what in which revision?

UserRevisionLine numberNew contents of line
andrewboyson 61:aad055f1b0d1 1 #include <stdint.h>
andrewboyson 61:aad055f1b0d1 2 #include <stdbool.h>
andrewboyson 61:aad055f1b0d1 3
andrewboyson 37:793b39683406 4 #include "log.h"
andrewboyson 37:793b39683406 5 #include "net.h"
andrewboyson 37:793b39683406 6 #include "action.h"
andrewboyson 37:793b39683406 7 #include "udp.h"
andrewboyson 37:793b39683406 8 #include "ntp.h"
andrewboyson 57:e0fb648acf48 9 #include "tftp.h"
andrewboyson 37:793b39683406 10 #include "dhcp.h"
andrewboyson 37:793b39683406 11 #include "dns.h"
andrewboyson 37:793b39683406 12 #include "eth.h"
andrewboyson 37:793b39683406 13 #include "ip4.h"
andrewboyson 37:793b39683406 14 #include "ip6.h"
andrewboyson 37:793b39683406 15 #include "slaac.h"
andrewboyson 37:793b39683406 16 #include "ns.h"
andrewboyson 10:f0854784e960 17
andrewboyson 52:fbc5a46b5e16 18 bool UdpTrace = true;
andrewboyson 10:f0854784e960 19
andrewboyson 10:f0854784e960 20 __packed struct header
andrewboyson 10:f0854784e960 21 {
andrewboyson 10:f0854784e960 22 uint16_t srcPort;
andrewboyson 10:f0854784e960 23 uint16_t dstPort;
andrewboyson 10:f0854784e960 24 uint16_t totalLength;
andrewboyson 10:f0854784e960 25 uint16_t checksum;
andrewboyson 10:f0854784e960 26 };
andrewboyson 37:793b39683406 27 static uint16_t srcPort;
andrewboyson 37:793b39683406 28 static uint16_t dstPort;
andrewboyson 10:f0854784e960 29 static uint16_t checksum;
andrewboyson 10:f0854784e960 30 static uint16_t totalLength;
andrewboyson 10:f0854784e960 31
andrewboyson 59:e0e556c8bd46 32 static void readHeader(void* pPacket, uint16_t size)
andrewboyson 59:e0e556c8bd46 33 {
andrewboyson 61:aad055f1b0d1 34 struct header* pHeader = (struct header*)pPacket;
andrewboyson 59:e0e556c8bd46 35
andrewboyson 59:e0e556c8bd46 36 srcPort = NetToHost16(pHeader->srcPort);
andrewboyson 59:e0e556c8bd46 37 dstPort = NetToHost16(pHeader->dstPort);
andrewboyson 59:e0e556c8bd46 38 totalLength = NetToHost16(pHeader->totalLength);
andrewboyson 59:e0e556c8bd46 39 checksum = NetToHost16(pHeader->checksum);
andrewboyson 59:e0e556c8bd46 40 }
andrewboyson 59:e0e556c8bd46 41 static int handlePort(void (*traceback)(void), int dataLengthRx, void* pDataRx, int* pPataLengthTx, void* pDataTx)
andrewboyson 10:f0854784e960 42 {
andrewboyson 37:793b39683406 43 switch (dstPort)
andrewboyson 10:f0854784e960 44 {
andrewboyson 10:f0854784e960 45 //Handle these
andrewboyson 59:e0e556c8bd46 46 case DHCP_CLIENT_PORT: return DhcpHandleResponse (traceback, dataLengthRx, pDataRx, pPataLengthTx, pDataTx); // 68
andrewboyson 59:e0e556c8bd46 47 case NTP_PORT: return NtpHandlePacketReceived(traceback, dataLengthRx, pDataRx, pPataLengthTx, pDataTx); // 123
andrewboyson 59:e0e556c8bd46 48 case DNS_UNICAST_CLIENT_PORT: return DnsHandlePacketReceived(traceback, DNS_PROTOCOL_UDNS, dataLengthRx, pDataRx, pPataLengthTx, pDataTx); //53053
andrewboyson 59:e0e556c8bd46 49 case DNS_MDNS_PORT: return DnsHandlePacketReceived(traceback, DNS_PROTOCOL_MDNS, dataLengthRx, pDataRx, pPataLengthTx, pDataTx); // 5353
andrewboyson 59:e0e556c8bd46 50 case DNS_LLMNR_CLIENT_PORT: return DnsHandlePacketReceived(traceback, DNS_PROTOCOL_LLMNR, dataLengthRx, pDataRx, pPataLengthTx, pDataTx); //53055
andrewboyson 59:e0e556c8bd46 51 case DNS_LLMNR_SERVER_PORT: return DnsHandlePacketReceived(traceback, DNS_PROTOCOL_LLMNR, dataLengthRx, pDataRx, pPataLengthTx, pDataTx); // 5355
andrewboyson 59:e0e556c8bd46 52 case TFTP_CLIENT_PORT: return TftpHandlePacketReceived(traceback, dataLengthRx, pDataRx, pPataLengthTx, pDataTx); //60690
andrewboyson 10:f0854784e960 53
andrewboyson 10:f0854784e960 54 //Quietly drop these
andrewboyson 42:222a4f45f916 55 case DHCP_SERVER_PORT: //67
andrewboyson 57:e0fb648acf48 56 case TFTP_SERVER_PORT: //69
andrewboyson 42:222a4f45f916 57 case 137: //NETBIOS name service
andrewboyson 42:222a4f45f916 58 case 138: //NETBIOS datagram service
andrewboyson 42:222a4f45f916 59 case 139: //NETBIOS session service
andrewboyson 42:222a4f45f916 60 case 500: //Key exchange - Xbox live
andrewboyson 42:222a4f45f916 61 case 1900: //SSDP Simple Service Discovery Protocol (uPnP)
andrewboyson 42:222a4f45f916 62 case 3074: //Xbox live
andrewboyson 42:222a4f45f916 63 case 3076: //Call of Duty - Xbox
andrewboyson 52:fbc5a46b5e16 64 case 5050: //Don't know but have been sent by Kate and my android phones.
andrewboyson 42:222a4f45f916 65 case 5224: //Don't know but a burst was broadcast by Kate's phone containing '_logitech-reverse-bonjour._tcp.local.5446 192.168.1.36 string'
andrewboyson 52:fbc5a46b5e16 66 case 9956: //Alljoyn part of Allseen IoT services
andrewboyson 42:222a4f45f916 67 case 9997: //VLC
andrewboyson 42:222a4f45f916 68 case 9998: //VLC
andrewboyson 42:222a4f45f916 69 case 9999: //VLC
andrewboyson 42:222a4f45f916 70 case 17500: //Dropbox LAN sync
andrewboyson 10:f0854784e960 71 return DO_NOTHING;
andrewboyson 10:f0854784e960 72
andrewboyson 10:f0854784e960 73 //Report anything else
andrewboyson 10:f0854784e960 74 default:
andrewboyson 52:fbc5a46b5e16 75 if (UdpTrace)
andrewboyson 52:fbc5a46b5e16 76 {
andrewboyson 52:fbc5a46b5e16 77 LogTimeF("UDP unknown port %d\r\n", dstPort);
andrewboyson 52:fbc5a46b5e16 78 traceback(); //This will already include the UDP header
andrewboyson 52:fbc5a46b5e16 79 }
andrewboyson 10:f0854784e960 80 return DO_NOTHING;
andrewboyson 10:f0854784e960 81 }
andrewboyson 10:f0854784e960 82 }
andrewboyson 59:e0e556c8bd46 83 int UdpHandleReceivedPacket(void (*traceback)(void), int sizeRx, void* pPacketRx, int* pSizeTx, void* pPacketTx)
andrewboyson 10:f0854784e960 84 {
andrewboyson 59:e0e556c8bd46 85 readHeader(pPacketRx, sizeRx);
andrewboyson 59:e0e556c8bd46 86
andrewboyson 61:aad055f1b0d1 87 void* pDataRx = (char*)pPacketRx + sizeof(struct header);
andrewboyson 61:aad055f1b0d1 88 void* pDataTx = (char*)pPacketTx + sizeof(struct header);
andrewboyson 61:aad055f1b0d1 89 int dataLengthRx = sizeRx - sizeof(struct header);
andrewboyson 61:aad055f1b0d1 90 int dataLengthTx = *pSizeTx - sizeof(struct header);
andrewboyson 10:f0854784e960 91
andrewboyson 59:e0e556c8bd46 92 int action = handlePort(traceback, dataLengthRx, pDataRx, &dataLengthTx, pDataTx);
andrewboyson 10:f0854784e960 93
andrewboyson 61:aad055f1b0d1 94 *pSizeTx = dataLengthTx + sizeof(struct header);
andrewboyson 10:f0854784e960 95
andrewboyson 37:793b39683406 96 uint16_t tmpPort = dstPort;
andrewboyson 37:793b39683406 97 dstPort = srcPort;
andrewboyson 37:793b39683406 98 srcPort = tmpPort;
andrewboyson 10:f0854784e960 99
andrewboyson 10:f0854784e960 100 return action;
andrewboyson 10:f0854784e960 101 }
andrewboyson 10:f0854784e960 102 static int pollForPacketToSend(int type, int* pDataLength, void* pData)
andrewboyson 10:f0854784e960 103 {
andrewboyson 37:793b39683406 104 int action = DO_NOTHING;
andrewboyson 10:f0854784e960 105
andrewboyson 10:f0854784e960 106 if (!action && type == IPV4) //DHCP only works under IPv4
andrewboyson 10:f0854784e960 107 {
andrewboyson 10:f0854784e960 108 action = DhcpPollForRequestToSend(pData, pDataLength);
andrewboyson 10:f0854784e960 109 if (action)
andrewboyson 10:f0854784e960 110 {
andrewboyson 37:793b39683406 111 srcPort = DHCP_CLIENT_PORT;
andrewboyson 37:793b39683406 112 dstPort = DHCP_SERVER_PORT;
andrewboyson 10:f0854784e960 113 }
andrewboyson 10:f0854784e960 114 }
andrewboyson 10:f0854784e960 115
andrewboyson 83:08c983006a6e 116 if (!action && type == (DnsSendRequestsViaIp4 ? IPV4 : IPV6)) //DNS is agnostic
andrewboyson 10:f0854784e960 117 {
andrewboyson 13:9cd54f7db57a 118 action = DnsPollForPacketToSend(pData, pDataLength);
andrewboyson 37:793b39683406 119 int dest = ActionGetDestPart(action);
andrewboyson 37:793b39683406 120 if (dest)
andrewboyson 10:f0854784e960 121 {
andrewboyson 37:793b39683406 122 switch (dest)
andrewboyson 10:f0854784e960 123 {
andrewboyson 37:793b39683406 124 case UNICAST_DNS: srcPort = DNS_UNICAST_CLIENT_PORT; dstPort = DNS_UNICAST_SERVER_PORT; break; //53053, 53
andrewboyson 37:793b39683406 125 case MULTICAST_MDNS: srcPort = DNS_MDNS_PORT; dstPort = DNS_MDNS_PORT; break; // 5353, 5353
andrewboyson 37:793b39683406 126 case MULTICAST_LLMNR: srcPort = DNS_LLMNR_CLIENT_PORT; dstPort = DNS_LLMNR_SERVER_PORT; break; //53055, 5355
andrewboyson 10:f0854784e960 127
andrewboyson 10:f0854784e960 128 //Report anything else
andrewboyson 10:f0854784e960 129 default:
andrewboyson 37:793b39683406 130 LogTimeF("DNS unknown dest %d\r\n", dest);
andrewboyson 10:f0854784e960 131 return DO_NOTHING;
andrewboyson 10:f0854784e960 132 }
andrewboyson 10:f0854784e960 133 }
andrewboyson 10:f0854784e960 134 }
andrewboyson 83:08c983006a6e 135 if (!action && type == (NtpSendRequestsViaIp4 ? IPV4 : IPV6)) //NTP is agnostic
andrewboyson 22:914b970356f0 136 {
andrewboyson 22:914b970356f0 137 action = NtpPollForPacketToSend(type, pData, pDataLength);
andrewboyson 22:914b970356f0 138 if (action)
andrewboyson 22:914b970356f0 139 {
andrewboyson 37:793b39683406 140 srcPort = NTP_PORT;
andrewboyson 37:793b39683406 141 dstPort = NTP_PORT;
andrewboyson 22:914b970356f0 142 }
andrewboyson 22:914b970356f0 143 }
andrewboyson 83:08c983006a6e 144 if (!action && type == (TftpSendRequestsViaIp4 ? IPV4 : IPV6)) //TFTP is agnostic
andrewboyson 57:e0fb648acf48 145 {
andrewboyson 57:e0fb648acf48 146 action = TftpPollForPacketToSend(type, pData, pDataLength);
andrewboyson 57:e0fb648acf48 147 if (action)
andrewboyson 57:e0fb648acf48 148 {
andrewboyson 57:e0fb648acf48 149 srcPort = TFTP_CLIENT_PORT;
andrewboyson 57:e0fb648acf48 150 dstPort = TFTP_SERVER_PORT;
andrewboyson 57:e0fb648acf48 151 }
andrewboyson 57:e0fb648acf48 152 }
andrewboyson 37:793b39683406 153
andrewboyson 10:f0854784e960 154 return action;
andrewboyson 10:f0854784e960 155 }
andrewboyson 10:f0854784e960 156 int UdpPollForPacketToSend(int type, int* pSize, void* pPacket)
andrewboyson 10:f0854784e960 157 {
andrewboyson 61:aad055f1b0d1 158 void* pData = (char*)pPacket + sizeof(struct header);
andrewboyson 61:aad055f1b0d1 159 int dataLength = *pSize - sizeof(struct header);
andrewboyson 10:f0854784e960 160
andrewboyson 10:f0854784e960 161 int action = pollForPacketToSend(type, &dataLength, pData);
andrewboyson 10:f0854784e960 162
andrewboyson 61:aad055f1b0d1 163 *pSize = dataLength + sizeof(struct header);
andrewboyson 10:f0854784e960 164 return action;
andrewboyson 10:f0854784e960 165 }
andrewboyson 10:f0854784e960 166
andrewboyson 37:793b39683406 167 void UdpLogHeader(uint16_t calculatedChecksum)
andrewboyson 43:bc028d5a6424 168 {
andrewboyson 43:bc028d5a6424 169 if (NetTraceVerbose)
andrewboyson 43:bc028d5a6424 170 {
andrewboyson 43:bc028d5a6424 171 Log ("UDP header\r\n");
andrewboyson 43:bc028d5a6424 172 LogF(" Source port %hu\r\n", srcPort);
andrewboyson 43:bc028d5a6424 173 LogF(" Destination port %hu\r\n", dstPort);
andrewboyson 43:bc028d5a6424 174 LogF(" Total length %hu\r\n", totalLength);
andrewboyson 43:bc028d5a6424 175 LogF(" Checksum (hex) %04hX\r\n", checksum);
andrewboyson 43:bc028d5a6424 176 LogF(" Calculated %04hX\r\n", calculatedChecksum);
andrewboyson 43:bc028d5a6424 177 }
andrewboyson 43:bc028d5a6424 178 else
andrewboyson 43:bc028d5a6424 179 {
andrewboyson 44:83ce5ace337b 180 LogF("UDP header %hu >>> %hu\r\n", srcPort, dstPort);
andrewboyson 43:bc028d5a6424 181 }
andrewboyson 10:f0854784e960 182 }
andrewboyson 10:f0854784e960 183
andrewboyson 10:f0854784e960 184 void UdpMakeHeader(int size, void* pPacket)
andrewboyson 10:f0854784e960 185 {
andrewboyson 61:aad055f1b0d1 186 struct header* pHeader = (struct header*)pPacket;
andrewboyson 10:f0854784e960 187
andrewboyson 37:793b39683406 188 pHeader->dstPort = NetToHost16(dstPort);
andrewboyson 37:793b39683406 189 pHeader->srcPort = NetToHost16(srcPort);
andrewboyson 10:f0854784e960 190 pHeader->totalLength = NetToHost16(size);
andrewboyson 10:f0854784e960 191 pHeader->checksum = 0;
andrewboyson 10:f0854784e960 192
andrewboyson 10:f0854784e960 193 }
andrewboyson 10:f0854784e960 194 void UdpAddChecksum(void* pPacket, uint16_t checksum)
andrewboyson 10:f0854784e960 195 {
andrewboyson 61:aad055f1b0d1 196 struct header* pHeader = (struct header*)pPacket;
andrewboyson 10:f0854784e960 197 pHeader->checksum = checksum;
andrewboyson 10:f0854784e960 198 }