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@113:904b40231907, 2019-01-22 (annotated)

Committer:
andrewboyson
Date:
Tue Jan 22 15:41:12 2019 +0000
Revision:
113:904b40231907
Parent:
112:f8694d0b8858
Child:
121:bc048b65a630
Incorporated ntp server module

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