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.cpp@14:e75a59c1123d, 2017-05-05 (annotated)

Committer:
andrewboyson
Date:
Fri May 05 17:44:16 2017 +0000
Revision:
14:e75a59c1123d
Parent:
13:9cd54f7db57a
Child:
15:6ca6778168b1
Made IP addresses and ports available to debug messages

Who changed what in which revision?

UserRevisionLine numberNew contents of line
andrewboyson 10:f0854784e960 1 #include "mbed.h"
andrewboyson 10:f0854784e960 2 #include "log.h"
andrewboyson 10:f0854784e960 3 #include "net.h"
andrewboyson 10:f0854784e960 4 #include "udp.h"
andrewboyson 10:f0854784e960 5 #include "ntp.h"
andrewboyson 10:f0854784e960 6 #include "dhcp.h"
andrewboyson 10:f0854784e960 7 #include "dns.h"
andrewboyson 10:f0854784e960 8 #include "ip4.h"
andrewboyson 10:f0854784e960 9 #include "ip6.h"
andrewboyson 10:f0854784e960 10 #include "slaac.h"
andrewboyson 10:f0854784e960 11 #include "ndp.h"
andrewboyson 13:9cd54f7db57a 12 #include "io.h"
andrewboyson 10:f0854784e960 13
andrewboyson 10:f0854784e960 14 #define UNKNOWN false
andrewboyson 10:f0854784e960 15
andrewboyson 10:f0854784e960 16 #define HEADER_SIZE 8
andrewboyson 10:f0854784e960 17 __packed struct header
andrewboyson 10:f0854784e960 18 {
andrewboyson 10:f0854784e960 19 uint16_t srcPort;
andrewboyson 10:f0854784e960 20 uint16_t dstPort;
andrewboyson 10:f0854784e960 21 uint16_t totalLength;
andrewboyson 10:f0854784e960 22 uint16_t checksum;
andrewboyson 10:f0854784e960 23 };
andrewboyson 14:e75a59c1123d 24 uint16_t UdpSrcPort;
andrewboyson 14:e75a59c1123d 25 uint16_t UdpDstPort;
andrewboyson 10:f0854784e960 26 static uint16_t checksum;
andrewboyson 10:f0854784e960 27 static uint16_t totalLength;
andrewboyson 10:f0854784e960 28
andrewboyson 10:f0854784e960 29 static int handlePort(int* pDataLength, void* pData)
andrewboyson 10:f0854784e960 30 {
andrewboyson 14:e75a59c1123d 31 switch (UdpDstPort)
andrewboyson 10:f0854784e960 32 {
andrewboyson 10:f0854784e960 33 //Handle these
andrewboyson 13:9cd54f7db57a 34 case DHCP_CLIENT_PORT: return DhcpHandleResponse ( pDataLength, pData); // 68
andrewboyson 14:e75a59c1123d 35 case NTP_PORT: return NtpHandleRequest ( pDataLength, pData); // 123
andrewboyson 14:e75a59c1123d 36 case DNS_UNICAST_CLIENT_PORT: return DnsHandlePacketReceived(DNS_PROTOCOL_UDNS, pDataLength, pData); //53053
andrewboyson 14:e75a59c1123d 37 case DNS_MDNS_PORT: return DnsHandlePacketReceived(DNS_PROTOCOL_MDNS, pDataLength, pData); // 5353
andrewboyson 14:e75a59c1123d 38 case DNS_LLMNR_CLIENT_PORT: return DnsHandlePacketReceived(DNS_PROTOCOL_LLMNR, pDataLength, pData); // 5355
andrewboyson 10:f0854784e960 39
andrewboyson 10:f0854784e960 40 //Quietly drop these
andrewboyson 10:f0854784e960 41 case DHCP_SERVER_PORT: //67
andrewboyson 10:f0854784e960 42 case 1900: //SSDP Simple Service Discovery Protocol (uPnP)
andrewboyson 10:f0854784e960 43 case 3076: //Call of Duty - Xbox
andrewboyson 10:f0854784e960 44 case 9997: //VLC
andrewboyson 10:f0854784e960 45 case 9998: //VLC
andrewboyson 10:f0854784e960 46 case 9999: //VLC
andrewboyson 10:f0854784e960 47 case 17500: //Dropbox LAN sync
andrewboyson 10:f0854784e960 48 return DO_NOTHING;
andrewboyson 10:f0854784e960 49
andrewboyson 10:f0854784e960 50 //Report anything else
andrewboyson 10:f0854784e960 51 default:
andrewboyson 14:e75a59c1123d 52 if (UNKNOWN) LogTimeF("UDP unknown port %d\r\n", UdpDstPort);
andrewboyson 10:f0854784e960 53 return DO_NOTHING;
andrewboyson 10:f0854784e960 54 }
andrewboyson 10:f0854784e960 55 }
andrewboyson 10:f0854784e960 56 int UdpHandleReceivedPacket(int* pSize, void* pPacket)
andrewboyson 10:f0854784e960 57 {
andrewboyson 10:f0854784e960 58 void* pData = (char*)pPacket + HEADER_SIZE;
andrewboyson 10:f0854784e960 59 int dataLength = *pSize - HEADER_SIZE;
andrewboyson 10:f0854784e960 60
andrewboyson 10:f0854784e960 61 int action = handlePort(&dataLength, pData);
andrewboyson 10:f0854784e960 62
andrewboyson 10:f0854784e960 63 *pSize = dataLength + HEADER_SIZE;
andrewboyson 10:f0854784e960 64
andrewboyson 14:e75a59c1123d 65 uint16_t tmpPort = UdpDstPort;
andrewboyson 14:e75a59c1123d 66 UdpDstPort = UdpSrcPort;
andrewboyson 14:e75a59c1123d 67 UdpSrcPort = tmpPort;
andrewboyson 10:f0854784e960 68
andrewboyson 10:f0854784e960 69 return action;
andrewboyson 10:f0854784e960 70 }
andrewboyson 10:f0854784e960 71 static int pollForPacketToSend(int type, int* pDataLength, void* pData)
andrewboyson 10:f0854784e960 72 {
andrewboyson 10:f0854784e960 73 int action = 0;
andrewboyson 10:f0854784e960 74
andrewboyson 10:f0854784e960 75 if (!action && type == IPV4) //DHCP only works under IPv4
andrewboyson 10:f0854784e960 76 {
andrewboyson 10:f0854784e960 77 action = DhcpPollForRequestToSend(pData, pDataLength);
andrewboyson 10:f0854784e960 78 if (action)
andrewboyson 10:f0854784e960 79 {
andrewboyson 14:e75a59c1123d 80 UdpSrcPort = DHCP_CLIENT_PORT;
andrewboyson 14:e75a59c1123d 81 UdpDstPort = DHCP_SERVER_PORT;
andrewboyson 10:f0854784e960 82 }
andrewboyson 10:f0854784e960 83 }
andrewboyson 10:f0854784e960 84
andrewboyson 10:f0854784e960 85 if (!action) //DNS is agnostic
andrewboyson 10:f0854784e960 86 {
andrewboyson 13:9cd54f7db57a 87 action = DnsPollForPacketToSend(pData, pDataLength);
andrewboyson 10:f0854784e960 88 if (action)
andrewboyson 10:f0854784e960 89 {
andrewboyson 10:f0854784e960 90 switch (action)
andrewboyson 10:f0854784e960 91 {
andrewboyson 14:e75a59c1123d 92 case UNICAST_DNS: UdpSrcPort = DNS_UNICAST_CLIENT_PORT; UdpDstPort = DNS_UNICAST_SERVER_PORT; break; //53053, 53
andrewboyson 14:e75a59c1123d 93 case MULTICAST_MDNS: UdpSrcPort = DNS_MDNS_PORT; UdpDstPort = DNS_MDNS_PORT; break; // 5353, 5353
andrewboyson 14:e75a59c1123d 94 case MULTICAST_LLMNR: UdpSrcPort = DNS_LLMNR_CLIENT_PORT; UdpDstPort = DNS_LLMNR_SERVER_PORT; break; //53055, 5355
andrewboyson 10:f0854784e960 95
andrewboyson 10:f0854784e960 96 //Report anything else
andrewboyson 10:f0854784e960 97 default:
andrewboyson 10:f0854784e960 98 LogTimeF("DNS unknown action %d\r\n", action);
andrewboyson 10:f0854784e960 99 return DO_NOTHING;
andrewboyson 10:f0854784e960 100 }
andrewboyson 10:f0854784e960 101 }
andrewboyson 10:f0854784e960 102 }
andrewboyson 10:f0854784e960 103 return action;
andrewboyson 10:f0854784e960 104 }
andrewboyson 10:f0854784e960 105 int UdpPollForPacketToSend(int type, int* pSize, void* pPacket)
andrewboyson 10:f0854784e960 106 {
andrewboyson 10:f0854784e960 107 void* pData = (char*)pPacket + HEADER_SIZE;
andrewboyson 10:f0854784e960 108 int dataLength = *pSize - HEADER_SIZE;
andrewboyson 10:f0854784e960 109
andrewboyson 10:f0854784e960 110 int action = pollForPacketToSend(type, &dataLength, pData);
andrewboyson 10:f0854784e960 111
andrewboyson 10:f0854784e960 112 *pSize = dataLength + HEADER_SIZE;
andrewboyson 10:f0854784e960 113 return action;
andrewboyson 10:f0854784e960 114 }
andrewboyson 10:f0854784e960 115
andrewboyson 10:f0854784e960 116 void UdpLogHeader(char* title, void* pPacket, uint16_t calculatedChecksum)
andrewboyson 10:f0854784e960 117 {
andrewboyson 10:f0854784e960 118 LogTimeF("UDP %s\r\n", title);
andrewboyson 14:e75a59c1123d 119 LogF(" Source port %hu\r\n", UdpSrcPort);
andrewboyson 14:e75a59c1123d 120 LogF(" Destination port %hu\r\n", UdpDstPort);
andrewboyson 10:f0854784e960 121 LogF(" Total length %hu\r\n", totalLength);
andrewboyson 10:f0854784e960 122 LogF(" Checksum (hex) %04hX\r\n", checksum);
andrewboyson 10:f0854784e960 123 LogF(" Calculated %04hX\r\n", calculatedChecksum);
andrewboyson 10:f0854784e960 124 }
andrewboyson 10:f0854784e960 125
andrewboyson 10:f0854784e960 126 void UdpMakeHeader(int size, void* pPacket)
andrewboyson 10:f0854784e960 127 {
andrewboyson 10:f0854784e960 128 struct header* pHeader = (header*)pPacket;
andrewboyson 10:f0854784e960 129
andrewboyson 14:e75a59c1123d 130 pHeader->dstPort = NetToHost16(UdpDstPort);
andrewboyson 14:e75a59c1123d 131 pHeader->srcPort = NetToHost16(UdpSrcPort);
andrewboyson 10:f0854784e960 132 pHeader->totalLength = NetToHost16(size);
andrewboyson 10:f0854784e960 133 pHeader->checksum = 0;
andrewboyson 10:f0854784e960 134
andrewboyson 10:f0854784e960 135 }
andrewboyson 10:f0854784e960 136 void UdpAddChecksum(void* pPacket, uint16_t checksum)
andrewboyson 10:f0854784e960 137 {
andrewboyson 10:f0854784e960 138 struct header* pHeader = (header*)pPacket;
andrewboyson 10:f0854784e960 139 pHeader->checksum = checksum;
andrewboyson 10:f0854784e960 140 }
andrewboyson 10:f0854784e960 141 void UdpReadHeader(void* pPacket, uint16_t size)
andrewboyson 10:f0854784e960 142 {
andrewboyson 10:f0854784e960 143 struct header* pHeader = (header*)pPacket;
andrewboyson 10:f0854784e960 144
andrewboyson 14:e75a59c1123d 145 UdpSrcPort = NetToHost16(pHeader->srcPort);
andrewboyson 14:e75a59c1123d 146 UdpDstPort = NetToHost16(pHeader->dstPort);
andrewboyson 10:f0854784e960 147 totalLength = NetToHost16(pHeader->totalLength);
andrewboyson 10:f0854784e960 148 checksum = NetToHost16(pHeader->checksum);
andrewboyson 10:f0854784e960 149 }