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/dns/dnsreply.cpp@32:679654f2d023, 2017-08-17 (annotated)

Committer:
andrewboyson
Date:
Thu Aug 17 14:21:02 2017 +0000
Revision:
32:679654f2d023
Parent:
30:e34173b7585c
Child:
35:93c39d260a83
Corrected issues with .local in llmnr and mdns

Who changed what in which revision?

UserRevisionLine numberNew contents of line
andrewboyson 13:9cd54f7db57a 1 #include "mbed.h"
andrewboyson 13:9cd54f7db57a 2 #include "log.h"
andrewboyson 13:9cd54f7db57a 3 #include "net.h"
andrewboyson 14:e75a59c1123d 4 #include "ip4.h"
andrewboyson 14:e75a59c1123d 5 #include "ip6.h"
andrewboyson 30:e34173b7585c 6 #include "ar.h"
andrewboyson 32:679654f2d023 7 #include "dnscache.h"
andrewboyson 30:e34173b7585c 8 #include "dns.h"
andrewboyson 30:e34173b7585c 9 #include "dnshdr.h"
andrewboyson 30:e34173b7585c 10 #include "dnsname.h"
andrewboyson 32:679654f2d023 11 #include "dhcp.h"
andrewboyson 13:9cd54f7db57a 12
andrewboyson 22:914b970356f0 13 #define DEBUG false
andrewboyson 13:9cd54f7db57a 14
andrewboyson 13:9cd54f7db57a 15 char DnsReplyRecordName[256];
andrewboyson 13:9cd54f7db57a 16 uint32_t DnsReplyRecordNameAsIp4 = 0;
andrewboyson 13:9cd54f7db57a 17 char DnsReplyRecordNameAsIp6[16];
andrewboyson 13:9cd54f7db57a 18 char DnsReplyName[64];
andrewboyson 13:9cd54f7db57a 19 uint32_t DnsReplyIp4 = 0;
andrewboyson 13:9cd54f7db57a 20 char DnsReplyIp6[16];
andrewboyson 13:9cd54f7db57a 21
andrewboyson 13:9cd54f7db57a 22 static char *p;
andrewboyson 13:9cd54f7db57a 23 static int recordNameOffset;
andrewboyson 13:9cd54f7db57a 24 static int recordType;
andrewboyson 13:9cd54f7db57a 25 static int recordDataLength;
andrewboyson 13:9cd54f7db57a 26 static char* pRecordData;
andrewboyson 13:9cd54f7db57a 27
andrewboyson 13:9cd54f7db57a 28 static int scanQuery()
andrewboyson 13:9cd54f7db57a 29 {
andrewboyson 13:9cd54f7db57a 30 int recordNameLength = DnsNameLength(p);
andrewboyson 13:9cd54f7db57a 31 if (!recordNameLength)
andrewboyson 13:9cd54f7db57a 32 {
andrewboyson 22:914b970356f0 33 if (DEBUG) LogTimeF("DnsReply scanRecord name length of zero\r\n");
andrewboyson 13:9cd54f7db57a 34 return -1; //failure
andrewboyson 13:9cd54f7db57a 35 }
andrewboyson 13:9cd54f7db57a 36
andrewboyson 13:9cd54f7db57a 37 recordNameOffset = DnsNameIndexFromPointer(p);
andrewboyson 13:9cd54f7db57a 38 p += recordNameLength;
andrewboyson 13:9cd54f7db57a 39
andrewboyson 13:9cd54f7db57a 40 p++ ; //skip the first byte of the type
andrewboyson 13:9cd54f7db57a 41 recordType = *p++;
andrewboyson 13:9cd54f7db57a 42
andrewboyson 13:9cd54f7db57a 43 p += 2; //skip the class
andrewboyson 13:9cd54f7db57a 44
andrewboyson 13:9cd54f7db57a 45 return 0; //success
andrewboyson 13:9cd54f7db57a 46 }
andrewboyson 13:9cd54f7db57a 47
andrewboyson 13:9cd54f7db57a 48 static int scanAnswer()
andrewboyson 13:9cd54f7db57a 49 {
andrewboyson 13:9cd54f7db57a 50 int recordNameLength = DnsNameLength(p);
andrewboyson 13:9cd54f7db57a 51 if (!recordNameLength)
andrewboyson 13:9cd54f7db57a 52 {
andrewboyson 22:914b970356f0 53 if (DEBUG) LogTimeF("DnsReply scanRecord name length of zero\r\n");
andrewboyson 13:9cd54f7db57a 54 return -1; //failure
andrewboyson 13:9cd54f7db57a 55 }
andrewboyson 13:9cd54f7db57a 56
andrewboyson 13:9cd54f7db57a 57 recordNameOffset = DnsNameIndexFromPointer(p);
andrewboyson 13:9cd54f7db57a 58 p += recordNameLength;
andrewboyson 13:9cd54f7db57a 59
andrewboyson 13:9cd54f7db57a 60 p++ ; //skip the first byte of the type
andrewboyson 13:9cd54f7db57a 61 recordType = *p++;
andrewboyson 13:9cd54f7db57a 62
andrewboyson 13:9cd54f7db57a 63 p += 6; //skip the class, TTL
andrewboyson 13:9cd54f7db57a 64 recordDataLength = 0;
andrewboyson 13:9cd54f7db57a 65 recordDataLength |= *p++ << 8;
andrewboyson 13:9cd54f7db57a 66 recordDataLength |= *p++;
andrewboyson 13:9cd54f7db57a 67
andrewboyson 13:9cd54f7db57a 68 pRecordData = p; //record the start of the data
andrewboyson 13:9cd54f7db57a 69
andrewboyson 13:9cd54f7db57a 70 p += recordDataLength; //Move to the start of the next record
andrewboyson 13:9cd54f7db57a 71
andrewboyson 13:9cd54f7db57a 72 return 0; //success
andrewboyson 13:9cd54f7db57a 73 }
andrewboyson 13:9cd54f7db57a 74 static void readAnswer()
andrewboyson 13:9cd54f7db57a 75 {
andrewboyson 13:9cd54f7db57a 76 DnsReplyRecordName[0] = 0;
andrewboyson 13:9cd54f7db57a 77 DnsReplyRecordNameAsIp4 = 0;
andrewboyson 13:9cd54f7db57a 78 DnsReplyRecordNameAsIp6[0] = 0;
andrewboyson 13:9cd54f7db57a 79 DnsReplyName[0] = 0;
andrewboyson 13:9cd54f7db57a 80 DnsReplyIp4 = 0;
andrewboyson 13:9cd54f7db57a 81 DnsReplyIp6[0] = 0;
andrewboyson 13:9cd54f7db57a 82
andrewboyson 13:9cd54f7db57a 83 switch (recordType)
andrewboyson 13:9cd54f7db57a 84 {
andrewboyson 13:9cd54f7db57a 85 case DNS_RECORD_A:
andrewboyson 13:9cd54f7db57a 86 case DNS_RECORD_AAAA:
andrewboyson 13:9cd54f7db57a 87 case DNS_RECORD_PTR:
andrewboyson 15:6ca6778168b1 88 case DNS_RECORD_SRV:
andrewboyson 15:6ca6778168b1 89 case DNS_RECORD_TXT:
andrewboyson 13:9cd54f7db57a 90 break;
andrewboyson 13:9cd54f7db57a 91 default:
andrewboyson 13:9cd54f7db57a 92 LogTimeF("DnsReply readAnswer unrecognised record type %d\r\n", recordType);
andrewboyson 15:6ca6778168b1 93 return;
andrewboyson 15:6ca6778168b1 94 }
andrewboyson 13:9cd54f7db57a 95
andrewboyson 13:9cd54f7db57a 96 DnsNameDecode (recordNameOffset, sizeof(DnsReplyRecordName), DnsReplyRecordName);
andrewboyson 13:9cd54f7db57a 97 DnsNameDecodeIp4(recordNameOffset, &DnsReplyRecordNameAsIp4);
andrewboyson 13:9cd54f7db57a 98 DnsNameDecodeIp6(recordNameOffset, DnsReplyRecordNameAsIp6);
andrewboyson 15:6ca6778168b1 99
andrewboyson 13:9cd54f7db57a 100 switch (recordType)
andrewboyson 13:9cd54f7db57a 101 {
andrewboyson 13:9cd54f7db57a 102 case DNS_RECORD_A:
andrewboyson 13:9cd54f7db57a 103 if (recordDataLength != 4)
andrewboyson 13:9cd54f7db57a 104 {
andrewboyson 13:9cd54f7db57a 105 LogTimeF("DnsReply A type length of %d\r\n", recordDataLength);
andrewboyson 13:9cd54f7db57a 106 return;
andrewboyson 13:9cd54f7db57a 107 }
andrewboyson 13:9cd54f7db57a 108 memcpy(&DnsReplyIp4, pRecordData, 4);
andrewboyson 13:9cd54f7db57a 109 break;
andrewboyson 13:9cd54f7db57a 110 case DNS_RECORD_AAAA:
andrewboyson 13:9cd54f7db57a 111 if (recordDataLength != 16)
andrewboyson 13:9cd54f7db57a 112 {
andrewboyson 13:9cd54f7db57a 113 LogTimeF("DnsReply AAAA type length of %d\r\n", recordDataLength);
andrewboyson 13:9cd54f7db57a 114 return;
andrewboyson 13:9cd54f7db57a 115 }
andrewboyson 13:9cd54f7db57a 116 memcpy(DnsReplyIp6, pRecordData, 16);
andrewboyson 13:9cd54f7db57a 117 break;
andrewboyson 13:9cd54f7db57a 118 case DNS_RECORD_PTR:
andrewboyson 13:9cd54f7db57a 119 if (recordDataLength > 63)
andrewboyson 13:9cd54f7db57a 120 {
andrewboyson 13:9cd54f7db57a 121 LogTimeF("DnsReply PTR type length of %d\r\n", recordDataLength);
andrewboyson 13:9cd54f7db57a 122 return;
andrewboyson 13:9cd54f7db57a 123 }
andrewboyson 13:9cd54f7db57a 124 DnsNameDecode(DnsNameIndexFromPointer(pRecordData), sizeof(DnsReplyName), DnsReplyName);
andrewboyson 13:9cd54f7db57a 125 break;
andrewboyson 13:9cd54f7db57a 126 }
andrewboyson 15:6ca6778168b1 127
andrewboyson 13:9cd54f7db57a 128 if (DEBUG)
andrewboyson 13:9cd54f7db57a 129 {
andrewboyson 15:6ca6778168b1 130 LogF(" answer: %s", DnsReplyRecordName);
andrewboyson 13:9cd54f7db57a 131 char text[100];
andrewboyson 15:6ca6778168b1 132 if (DnsReplyRecordNameAsIp4)
andrewboyson 15:6ca6778168b1 133 {
andrewboyson 15:6ca6778168b1 134 Ip4AddressToString(DnsReplyRecordNameAsIp4, sizeof(text), text);
andrewboyson 15:6ca6778168b1 135 LogF(" (%s)", text);
andrewboyson 15:6ca6778168b1 136 }
andrewboyson 15:6ca6778168b1 137 if (DnsReplyRecordNameAsIp6[0])
andrewboyson 15:6ca6778168b1 138 {
andrewboyson 15:6ca6778168b1 139 Ip6AddressToString(DnsReplyRecordNameAsIp6, sizeof(text), text);
andrewboyson 15:6ca6778168b1 140 LogF(" (%s)", text);
andrewboyson 15:6ca6778168b1 141 }
andrewboyson 15:6ca6778168b1 142 LogF(" == ");
andrewboyson 13:9cd54f7db57a 143 switch (recordType)
andrewboyson 13:9cd54f7db57a 144 {
andrewboyson 13:9cd54f7db57a 145 case DNS_RECORD_A:
andrewboyson 14:e75a59c1123d 146 Ip4AddressToString(DnsReplyIp4, sizeof(text), text);
andrewboyson 15:6ca6778168b1 147 LogF("%s\r\n", text);
andrewboyson 13:9cd54f7db57a 148 break;
andrewboyson 13:9cd54f7db57a 149 case DNS_RECORD_AAAA:
andrewboyson 14:e75a59c1123d 150 Ip6AddressToString(DnsReplyIp6, sizeof(text), text);
andrewboyson 15:6ca6778168b1 151 LogF("%s\r\n", text);
andrewboyson 13:9cd54f7db57a 152 break;
andrewboyson 13:9cd54f7db57a 153 case DNS_RECORD_PTR:
andrewboyson 15:6ca6778168b1 154 LogF("%s\r\n", DnsReplyName);
andrewboyson 15:6ca6778168b1 155 break;
andrewboyson 15:6ca6778168b1 156 default:
andrewboyson 15:6ca6778168b1 157 DnsRecordTypeToString(recordType, sizeof(text), text);
andrewboyson 15:6ca6778168b1 158 LogF("%d bytes of %s\r\n", recordDataLength, text);
andrewboyson 13:9cd54f7db57a 159 break;
andrewboyson 13:9cd54f7db57a 160 }
andrewboyson 13:9cd54f7db57a 161 }
andrewboyson 13:9cd54f7db57a 162 }
andrewboyson 13:9cd54f7db57a 163
andrewboyson 13:9cd54f7db57a 164 static void sendToAr(int dnsProtocol)
andrewboyson 13:9cd54f7db57a 165 {
andrewboyson 13:9cd54f7db57a 166 if (DnsReplyIp4 && DnsReplyRecordName[0]) ArAddName4(DnsReplyIp4, DnsReplyRecordName, dnsProtocol);
andrewboyson 13:9cd54f7db57a 167 if (DnsReplyIp6[0] && DnsReplyRecordName[0]) ArAddName6(DnsReplyIp6, DnsReplyRecordName, dnsProtocol);
andrewboyson 13:9cd54f7db57a 168 if (DnsReplyRecordNameAsIp4 && DnsReplyName[0] ) ArAddName4(DnsReplyRecordNameAsIp4, DnsReplyName, dnsProtocol);
andrewboyson 13:9cd54f7db57a 169 if (DnsReplyRecordNameAsIp6[0] && DnsReplyName[0] ) ArAddName6(DnsReplyRecordNameAsIp6, DnsReplyName, dnsProtocol);
andrewboyson 13:9cd54f7db57a 170 }
andrewboyson 32:679654f2d023 171 static void sendToDnsCache(int dnsProtocol)
andrewboyson 30:e34173b7585c 172 {
andrewboyson 32:679654f2d023 173 char strippedName[100];
andrewboyson 32:679654f2d023 174 if (DnsReplyRecordName[0]) DnsStripNameFromFullName(dnsProtocol, DnsReplyRecordName, sizeof(strippedName), strippedName);
andrewboyson 32:679654f2d023 175 if (DnsReplyName[0] ) DnsStripNameFromFullName(dnsProtocol, DnsReplyName , sizeof(strippedName), strippedName);
andrewboyson 32:679654f2d023 176
andrewboyson 32:679654f2d023 177 if (DnsReplyIp4 && DnsReplyRecordName[0]) DnsCacheAddIp4Record(DnsReplyIp4, strippedName, dnsProtocol);
andrewboyson 32:679654f2d023 178 if (DnsReplyIp6[0] && DnsReplyRecordName[0]) DnsCacheAddIp6Record(DnsReplyIp6, strippedName, dnsProtocol);
andrewboyson 32:679654f2d023 179 if (DnsReplyRecordNameAsIp4 && DnsReplyName[0] ) DnsCacheAddIp4Record(DnsReplyRecordNameAsIp4, strippedName, dnsProtocol);
andrewboyson 32:679654f2d023 180 if (DnsReplyRecordNameAsIp6[0] && DnsReplyName[0] ) DnsCacheAddIp6Record(DnsReplyRecordNameAsIp6, strippedName, dnsProtocol);
andrewboyson 30:e34173b7585c 181 }
andrewboyson 13:9cd54f7db57a 182 int DnsReplyHandle(int dnsProtocol, int *pSize)
andrewboyson 13:9cd54f7db57a 183 {
andrewboyson 13:9cd54f7db57a 184 if (!DnsHdrAncount) return DO_NOTHING;
andrewboyson 13:9cd54f7db57a 185 if (DEBUG) DnsHdrLog("received reply", dnsProtocol);
andrewboyson 13:9cd54f7db57a 186
andrewboyson 13:9cd54f7db57a 187 p = DnsHdrData;
andrewboyson 13:9cd54f7db57a 188
andrewboyson 13:9cd54f7db57a 189 for (int q = 0; q < DnsHdrQdcount; q++)
andrewboyson 13:9cd54f7db57a 190 {
andrewboyson 13:9cd54f7db57a 191 if (scanQuery()) return DO_NOTHING;
andrewboyson 13:9cd54f7db57a 192 }
andrewboyson 13:9cd54f7db57a 193 for (int a = 0; a < DnsHdrAncount; a++)
andrewboyson 13:9cd54f7db57a 194 {
andrewboyson 13:9cd54f7db57a 195 if (scanAnswer()) return DO_NOTHING;
andrewboyson 13:9cd54f7db57a 196 readAnswer();
andrewboyson 13:9cd54f7db57a 197 sendToAr(dnsProtocol);
andrewboyson 32:679654f2d023 198 sendToDnsCache(dnsProtocol);
andrewboyson 13:9cd54f7db57a 199 }
andrewboyson 13:9cd54f7db57a 200
andrewboyson 13:9cd54f7db57a 201 return DO_NOTHING;
andrewboyson 13:9cd54f7db57a 202 }