Dirk-Willem van Gulik (NXP/mbed)
/
AmpereMeter
Example self-announcing webserver which controls a servo through a smallHTML userinterface.
services/mDNS/mDNSResponder.cpp@0:a259777c45a3, 2010-08-14 (annotated)
- Committer:
- dirkx
- Date:
- Sat Aug 14 15:56:01 2010 +0000
- Revision:
- 0:a259777c45a3
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
dirkx | 0:a259777c45a3 | 1 | /* Class: mDNSResponder |
dirkx | 0:a259777c45a3 | 2 | * Copyright 1991, 2003, 2010 Dirk-Willem van Gulik <dirkx(at)apache(punto)org> |
dirkx | 0:a259777c45a3 | 3 | * |
dirkx | 0:a259777c45a3 | 4 | * License: Any BSD or ASF License. |
dirkx | 0:a259777c45a3 | 5 | * |
dirkx | 0:a259777c45a3 | 6 | * Rough and ready port of some mDNS code. |
dirkx | 0:a259777c45a3 | 7 | * |
dirkx | 0:a259777c45a3 | 8 | * Typical use is something like |
dirkx | 0:a259777c45a3 | 9 | * |
dirkx | 0:a259777c45a3 | 10 | * EthernetNetIf eth; |
dirkx | 0:a259777c45a3 | 11 | * HTTPServer svr; |
dirkx | 0:a259777c45a3 | 12 | * mDNSResponder mdns; |
dirkx | 0:a259777c45a3 | 13 | * |
dirkx | 0:a259777c45a3 | 14 | * int main()... |
dirkx | 0:a259777c45a3 | 15 | * |
dirkx | 0:a259777c45a3 | 16 | * // get internet |
dirkx | 0:a259777c45a3 | 17 | * EthernetErr ethErr = eth.setup(); |
dirkx | 0:a259777c45a3 | 18 | * ... etc .. |
dirkx | 0:a259777c45a3 | 19 | * |
dirkx | 0:a259777c45a3 | 20 | * // set up some server |
dirkx | 0:a259777c45a3 | 21 | * svr.addHandler<SimpleHandler>("/"); //Default handler |
dirkx | 0:a259777c45a3 | 22 | * svr.bind(80); * * |
dirkx | 0:a259777c45a3 | 23 | |
dirkx | 0:a259777c45a3 | 24 | * // Extract the IP address. |
dirkx | 0:a259777c45a3 | 25 | * IpAddr ip = eth.getIp(); |
dirkx | 0:a259777c45a3 | 26 | * printf("mbed IP Address is %d.%d.%d.%d\r\n", ip[0], ip[1], ip[2], ip[3]); |
dirkx | 0:a259777c45a3 | 27 | * |
dirkx | 0:a259777c45a3 | 28 | * // Announce ourselves. |
dirkx | 0:a259777c45a3 | 29 | * mdns.announce(ip, "fred", "_http._tcp", 80, "The Little Server that Could", "path=/demo"); |
dirkx | 0:a259777c45a3 | 30 | * |
dirkx | 0:a259777c45a3 | 31 | * while()... enter some run loop |
dirkx | 0:a259777c45a3 | 32 | * ... |
dirkx | 0:a259777c45a3 | 33 | * |
dirkx | 0:a259777c45a3 | 34 | * This will cause http://fred.local./demo to be announced as 'The Little Server that Could'. |
dirkx | 0:a259777c45a3 | 35 | * |
dirkx | 0:a259777c45a3 | 36 | * Or as another example: (http://files.dns-sd.org/draft-cheshire-dnsext-dns-sd.txt) |
dirkx | 0:a259777c45a3 | 37 | * and the various RFCs: |
dirkx | 0:a259777c45a3 | 38 | * |
dirkx | 0:a259777c45a3 | 39 | * mdns.announce(ip, "_ssh._tcp", 22, SSH to Serial Gateway", NULL); |
dirkx | 0:a259777c45a3 | 40 | * |
dirkx | 0:a259777c45a3 | 41 | * CAVEAT - a lot of the buffer overrun and copy checks |
dirkx | 0:a259777c45a3 | 42 | * where removed; and this is not anywhere near |
dirkx | 0:a259777c45a3 | 43 | * threadsafve or sane. Not for production use. |
dirkx | 0:a259777c45a3 | 44 | */ |
dirkx | 0:a259777c45a3 | 45 | #include "lwip/opt.h" |
dirkx | 0:a259777c45a3 | 46 | #if LWIP_DNS /* don't build if not configured for use in lwipopts.h */ |
dirkx | 0:a259777c45a3 | 47 | |
dirkx | 0:a259777c45a3 | 48 | #include "dbg/dbg.h" |
dirkx | 0:a259777c45a3 | 49 | |
dirkx | 0:a259777c45a3 | 50 | #include "mDNSResponder.h" |
dirkx | 0:a259777c45a3 | 51 | #include <stdio.h> |
dirkx | 0:a259777c45a3 | 52 | |
dirkx | 0:a259777c45a3 | 53 | #ifndef DNS_RRTYPE_SRV |
dirkx | 0:a259777c45a3 | 54 | #define DNS_RRTYPE_SRV (33) |
dirkx | 0:a259777c45a3 | 55 | #endif |
dirkx | 0:a259777c45a3 | 56 | |
dirkx | 0:a259777c45a3 | 57 | #ifndef DNS_LOCAL |
dirkx | 0:a259777c45a3 | 58 | #define DNS_LOCAL "local" |
dirkx | 0:a259777c45a3 | 59 | #endif |
dirkx | 0:a259777c45a3 | 60 | |
dirkx | 0:a259777c45a3 | 61 | mDNSResponder::mDNSResponder() : NetService(false), |
dirkx | 0:a259777c45a3 | 62 | moi_txt(NULL), moi_port(0), moi_ip(0), |
dirkx | 0:a259777c45a3 | 63 | moi_name(NULL), moi_proto(NULL), |
dirkx | 0:a259777c45a3 | 64 | moi_local_proto(NULL), moi_local_name(NULL), moi_local_pglue(NULL), |
dirkx | 0:a259777c45a3 | 65 | #ifdef MDNS_QCACHE |
dirkx | 0:a259777c45a3 | 66 | labelCacheCnt(0), labelCacheOffset(0), |
dirkx | 0:a259777c45a3 | 67 | #endif |
dirkx | 0:a259777c45a3 | 68 | announcer() |
dirkx | 0:a259777c45a3 | 69 | { |
dirkx | 0:a259777c45a3 | 70 | initLabelCache(0); |
dirkx | 0:a259777c45a3 | 71 | } |
dirkx | 0:a259777c45a3 | 72 | |
dirkx | 0:a259777c45a3 | 73 | mDNSResponder::~mDNSResponder() { |
dirkx | 0:a259777c45a3 | 74 | announcer.stop(); |
dirkx | 0:a259777c45a3 | 75 | close(); |
dirkx | 0:a259777c45a3 | 76 | } |
dirkx | 0:a259777c45a3 | 77 | |
dirkx | 0:a259777c45a3 | 78 | void mDNSResponder::announce(IpAddr ip, const char * ldn, const char * proto, uint16_t port, const char * name, char ** txt) { |
dirkx | 0:a259777c45a3 | 79 | Host localhost(IpAddr(MCAST), MDNS_PORT, NULL /* fqdn */); |
dirkx | 0:a259777c45a3 | 80 | |
dirkx | 0:a259777c45a3 | 81 | m_pUDPSocket = new UDPSocket; |
dirkx | 0:a259777c45a3 | 82 | m_pUDPSocket->bind(localhost); |
dirkx | 0:a259777c45a3 | 83 | |
dirkx | 0:a259777c45a3 | 84 | m_pUDPSocket->setOnEvent(this, &mDNSResponder::onUDPSocketEvent); |
dirkx | 0:a259777c45a3 | 85 | |
dirkx | 0:a259777c45a3 | 86 | moi_port = port; |
dirkx | 0:a259777c45a3 | 87 | moi_txt = txt; |
dirkx | 0:a259777c45a3 | 88 | moi_ip = ip; |
dirkx | 0:a259777c45a3 | 89 | moi_proto = proto; |
dirkx | 0:a259777c45a3 | 90 | moi_name = name; |
dirkx | 0:a259777c45a3 | 91 | |
dirkx | 0:a259777c45a3 | 92 | moi_local_proto = (char *)malloc(128); |
dirkx | 0:a259777c45a3 | 93 | moi_local_name = (char *)malloc(128); |
dirkx | 0:a259777c45a3 | 94 | moi_local_pglue = (char *)malloc(128); |
dirkx | 0:a259777c45a3 | 95 | |
dirkx | 0:a259777c45a3 | 96 | snprintf(moi_local_proto,128,"%s.%s.", moi_proto, DNS_LOCAL); |
dirkx | 0:a259777c45a3 | 97 | snprintf(moi_local_name,128,"%s.%s.", ldn, DNS_LOCAL); |
dirkx | 0:a259777c45a3 | 98 | snprintf(moi_local_pglue,128,"%s.%s.%s.", moi_name,moi_proto, DNS_LOCAL); |
dirkx | 0:a259777c45a3 | 99 | |
dirkx | 0:a259777c45a3 | 100 | // Gratuis intro - and repeat such regularly.. taking some |
dirkx | 0:a259777c45a3 | 101 | // care to not go beyond DHCP limits - but not more often |
dirkx | 0:a259777c45a3 | 102 | // than makes sense for our TTL. See the .h file for the |
dirkx | 0:a259777c45a3 | 103 | // various tradeoffs. |
dirkx | 0:a259777c45a3 | 104 | // |
dirkx | 0:a259777c45a3 | 105 | mDNSResponder::sendReply(DNS_RRTYPE_PTR, 0, localhost); |
dirkx | 0:a259777c45a3 | 106 | announcer.start(); |
dirkx | 0:a259777c45a3 | 107 | } |
dirkx | 0:a259777c45a3 | 108 | |
dirkx | 0:a259777c45a3 | 109 | void mDNSResponder::close() { |
dirkx | 0:a259777c45a3 | 110 | m_pUDPSocket->resetOnEvent(); |
dirkx | 0:a259777c45a3 | 111 | m_pUDPSocket->close(); |
dirkx | 0:a259777c45a3 | 112 | delete m_pUDPSocket; |
dirkx | 0:a259777c45a3 | 113 | } |
dirkx | 0:a259777c45a3 | 114 | |
dirkx | 0:a259777c45a3 | 115 | void mDNSResponder::poll() { //Called by NetServices |
dirkx | 0:a259777c45a3 | 116 | if (announcer.read_ms() > 1000 * MDNS_INTERVAL) { |
dirkx | 0:a259777c45a3 | 117 | |
dirkx | 0:a259777c45a3 | 118 | Host mch(IpAddr(MCAST), MDNS_PORT, NULL); |
dirkx | 0:a259777c45a3 | 119 | mDNSResponder::sendReply(DNS_RRTYPE_PTR, 0, mch); |
dirkx | 0:a259777c45a3 | 120 | |
dirkx | 0:a259777c45a3 | 121 | announcer.reset(); |
dirkx | 0:a259777c45a3 | 122 | } |
dirkx | 0:a259777c45a3 | 123 | } |
dirkx | 0:a259777c45a3 | 124 | |
dirkx | 0:a259777c45a3 | 125 | char * index(char * str, char c) { |
dirkx | 0:a259777c45a3 | 126 | for (;str && *str;str++) { |
dirkx | 0:a259777c45a3 | 127 | if (*str == c) return str; |
dirkx | 0:a259777c45a3 | 128 | }; |
dirkx | 0:a259777c45a3 | 129 | return NULL; |
dirkx | 0:a259777c45a3 | 130 | } |
dirkx | 0:a259777c45a3 | 131 | |
dirkx | 0:a259777c45a3 | 132 | #ifdef MDNS_QCACHE |
dirkx | 0:a259777c45a3 | 133 | void mDNSResponder::initLabelCache(char * off) { |
dirkx | 0:a259777c45a3 | 134 | labelCacheOffset = off; |
dirkx | 0:a259777c45a3 | 135 | labelCacheCnt = 0; |
dirkx | 0:a259777c45a3 | 136 | memset(_cache_off,0, sizeof(_cache_off)); // Rely on min value to be 12 |
dirkx | 0:a259777c45a3 | 137 | }; |
dirkx | 0:a259777c45a3 | 138 | |
dirkx | 0:a259777c45a3 | 139 | uint16_t mDNSResponder::checkLabelCache(char * name) { |
dirkx | 0:a259777c45a3 | 140 | // Bail out if not initialized with a valid (>12) offset. |
dirkx | 0:a259777c45a3 | 141 | // |
dirkx | 0:a259777c45a3 | 142 | if (!labelCacheOffset) |
dirkx | 0:a259777c45a3 | 143 | return 0; |
dirkx | 0:a259777c45a3 | 144 | |
dirkx | 0:a259777c45a3 | 145 | for (int i=0; i < MDNS_MAXCACHEDNSLABELS; i++) |
dirkx | 0:a259777c45a3 | 146 | if (_cache_off[i] && !strcmp(_cache_och[i],name)) |
dirkx | 0:a259777c45a3 | 147 | return _cache_off[i]; |
dirkx | 0:a259777c45a3 | 148 | return 0; |
dirkx | 0:a259777c45a3 | 149 | }; |
dirkx | 0:a259777c45a3 | 150 | |
dirkx | 0:a259777c45a3 | 151 | void mDNSResponder::addLabelCache(char * p, char * name) { |
dirkx | 0:a259777c45a3 | 152 | // Bail out if not initialized with a valid (>12) offset. |
dirkx | 0:a259777c45a3 | 153 | // |
dirkx | 0:a259777c45a3 | 154 | if (!labelCacheOffset) |
dirkx | 0:a259777c45a3 | 155 | return; |
dirkx | 0:a259777c45a3 | 156 | |
dirkx | 0:a259777c45a3 | 157 | _cache_off[labelCacheCnt] = p - labelCacheOffset; |
dirkx | 0:a259777c45a3 | 158 | _cache_och[labelCacheCnt] = name; |
dirkx | 0:a259777c45a3 | 159 | |
dirkx | 0:a259777c45a3 | 160 | labelCacheCnt++; |
dirkx | 0:a259777c45a3 | 161 | // we intentionally do not wack the first entries - as they are the most |
dirkx | 0:a259777c45a3 | 162 | // likely ones to be used. |
dirkx | 0:a259777c45a3 | 163 | if (labelCacheCnt>= MDNS_MAXCACHEDNSLABELS) |
dirkx | 0:a259777c45a3 | 164 | labelCacheCnt = MDNS_MAXCACHEDNSLABELS / 3; |
dirkx | 0:a259777c45a3 | 165 | }; |
dirkx | 0:a259777c45a3 | 166 | #else |
dirkx | 0:a259777c45a3 | 167 | #define initLabelCache(x) {} /* Ignored */ |
dirkx | 0:a259777c45a3 | 168 | #define checkLabelCache(x) (0) /* never a hit */ |
dirkx | 0:a259777c45a3 | 169 | #define addLabelCache(x,y) {} /* Ignored */ |
dirkx | 0:a259777c45a3 | 170 | #endif |
dirkx | 0:a259777c45a3 | 171 | |
dirkx | 0:a259777c45a3 | 172 | char * mDNSResponder::breakname(char *p, char *name) { |
dirkx | 0:a259777c45a3 | 173 | int l = 0, de = 1; |
dirkx | 0:a259777c45a3 | 174 | char * q; |
dirkx | 0:a259777c45a3 | 175 | uint16_t off; |
dirkx | 0:a259777c45a3 | 176 | |
dirkx | 0:a259777c45a3 | 177 | do { |
dirkx | 0:a259777c45a3 | 178 | if ((off = checkLabelCache(name)) != 0) { |
dirkx | 0:a259777c45a3 | 179 | *p++ = 192 + (off >> 8); |
dirkx | 0:a259777c45a3 | 180 | *p++ = (off & 0xFF); |
dirkx | 0:a259777c45a3 | 181 | return p; |
dirkx | 0:a259777c45a3 | 182 | } else { |
dirkx | 0:a259777c45a3 | 183 | addLabelCache(p, name); |
dirkx | 0:a259777c45a3 | 184 | }; |
dirkx | 0:a259777c45a3 | 185 | |
dirkx | 0:a259777c45a3 | 186 | q = index(name,'.'); |
dirkx | 0:a259777c45a3 | 187 | if (!q) { |
dirkx | 0:a259777c45a3 | 188 | q = name + strlen(name); |
dirkx | 0:a259777c45a3 | 189 | de = 0; |
dirkx | 0:a259777c45a3 | 190 | }; |
dirkx | 0:a259777c45a3 | 191 | l = q - name; |
dirkx | 0:a259777c45a3 | 192 | *p++ = l; |
dirkx | 0:a259777c45a3 | 193 | memcpy(p, name, l); |
dirkx | 0:a259777c45a3 | 194 | p+=l; |
dirkx | 0:a259777c45a3 | 195 | name = q + 1; |
dirkx | 0:a259777c45a3 | 196 | } while (l && *name && de); |
dirkx | 0:a259777c45a3 | 197 | |
dirkx | 0:a259777c45a3 | 198 | // terminating root field if any (not the case for |
dirkx | 0:a259777c45a3 | 199 | // things like TXTs). |
dirkx | 0:a259777c45a3 | 200 | if (de) *p++ = 0; |
dirkx | 0:a259777c45a3 | 201 | return p; |
dirkx | 0:a259777c45a3 | 202 | } |
dirkx | 0:a259777c45a3 | 203 | |
dirkx | 0:a259777c45a3 | 204 | char * mDNSResponder::mRR(char *p, char * name, uint16_t tpe, uint16_t cls, uint32_t ttl) { |
dirkx | 0:a259777c45a3 | 205 | uint16_t i = 0; |
dirkx | 0:a259777c45a3 | 206 | uint32_t j = 0; |
dirkx | 0:a259777c45a3 | 207 | |
dirkx | 0:a259777c45a3 | 208 | p = breakname(p, name); |
dirkx | 0:a259777c45a3 | 209 | |
dirkx | 0:a259777c45a3 | 210 | // NOTE: Cannot assume proper byte boundaries; so |
dirkx | 0:a259777c45a3 | 211 | // we assume the compiler does not allow that for |
dirkx | 0:a259777c45a3 | 212 | // casts - and write it out with memcpy's |
dirkx | 0:a259777c45a3 | 213 | // |
dirkx | 0:a259777c45a3 | 214 | i = htons(tpe); // Type |
dirkx | 0:a259777c45a3 | 215 | memcpy(p, &i, 2); |
dirkx | 0:a259777c45a3 | 216 | p+=2; |
dirkx | 0:a259777c45a3 | 217 | |
dirkx | 0:a259777c45a3 | 218 | i = htons(cls); // Class |
dirkx | 0:a259777c45a3 | 219 | memcpy(p, &i, 2); |
dirkx | 0:a259777c45a3 | 220 | p+=2; |
dirkx | 0:a259777c45a3 | 221 | |
dirkx | 0:a259777c45a3 | 222 | j = htonl(ttl); // TTL (4 bytes) |
dirkx | 0:a259777c45a3 | 223 | memcpy(p, &j, 4); |
dirkx | 0:a259777c45a3 | 224 | p+=4; |
dirkx | 0:a259777c45a3 | 225 | |
dirkx | 0:a259777c45a3 | 226 | return p; |
dirkx | 0:a259777c45a3 | 227 | } |
dirkx | 0:a259777c45a3 | 228 | |
dirkx | 0:a259777c45a3 | 229 | char * mDNSResponder::mRRLABEL(char *p, char * name, uint16_t tpe, char ** rr) { |
dirkx | 0:a259777c45a3 | 230 | uint16_t i = 0; |
dirkx | 0:a259777c45a3 | 231 | |
dirkx | 0:a259777c45a3 | 232 | p = mRR(p, name, tpe, DNS_RRCLASS_IN, MDNS_TTL); // Type, IN, TTL |
dirkx | 0:a259777c45a3 | 233 | |
dirkx | 0:a259777c45a3 | 234 | // RR String |
dirkx | 0:a259777c45a3 | 235 | char * q = p + 2; |
dirkx | 0:a259777c45a3 | 236 | |
dirkx | 0:a259777c45a3 | 237 | for (;*rr;rr++) |
dirkx | 0:a259777c45a3 | 238 | q = breakname(q, *rr); |
dirkx | 0:a259777c45a3 | 239 | |
dirkx | 0:a259777c45a3 | 240 | i = htons(q - p - 2); // RDLEN |
dirkx | 0:a259777c45a3 | 241 | memcpy(p, &i, 2); |
dirkx | 0:a259777c45a3 | 242 | return q; |
dirkx | 0:a259777c45a3 | 243 | } |
dirkx | 0:a259777c45a3 | 244 | |
dirkx | 0:a259777c45a3 | 245 | char * mDNSResponder::mPTR(char *p, char * name, char * r) { |
dirkx | 0:a259777c45a3 | 246 | char *rr[] = { r, NULL }; |
dirkx | 0:a259777c45a3 | 247 | return mRRLABEL(p, name, DNS_RRTYPE_PTR, rr ); |
dirkx | 0:a259777c45a3 | 248 | } |
dirkx | 0:a259777c45a3 | 249 | |
dirkx | 0:a259777c45a3 | 250 | char * mDNSResponder::mTXT(char *p, char * name, char ** rr) { |
dirkx | 0:a259777c45a3 | 251 | return mRRLABEL(p, name, DNS_RRTYPE_TXT, rr); |
dirkx | 0:a259777c45a3 | 252 | } |
dirkx | 0:a259777c45a3 | 253 | |
dirkx | 0:a259777c45a3 | 254 | char * mDNSResponder::mARR(char *p, char * name, IpAddr ip) { |
dirkx | 0:a259777c45a3 | 255 | uint16_t i = 0; |
dirkx | 0:a259777c45a3 | 256 | |
dirkx | 0:a259777c45a3 | 257 | p = mRR(p, name, DNS_RRTYPE_A, DNS_RRCLASS_IN, MDNS_TTL ); // A, IN |
dirkx | 0:a259777c45a3 | 258 | |
dirkx | 0:a259777c45a3 | 259 | i = htons(4); // RDLEN - we're just doing a single IPv4 address - our primary link ? |
dirkx | 0:a259777c45a3 | 260 | memcpy(p, &i, 2); |
dirkx | 0:a259777c45a3 | 261 | |
dirkx | 0:a259777c45a3 | 262 | // IP already in network order. |
dirkx | 0:a259777c45a3 | 263 | memcpy(p+2, &ip, 4); |
dirkx | 0:a259777c45a3 | 264 | |
dirkx | 0:a259777c45a3 | 265 | return p + 2 + 4; |
dirkx | 0:a259777c45a3 | 266 | } |
dirkx | 0:a259777c45a3 | 267 | |
dirkx | 0:a259777c45a3 | 268 | char * mDNSResponder::mSRV(char *p, char * name, uint16_t port, char * rr) { |
dirkx | 0:a259777c45a3 | 269 | uint16_t i = 0; |
dirkx | 0:a259777c45a3 | 270 | char * q; |
dirkx | 0:a259777c45a3 | 271 | |
dirkx | 0:a259777c45a3 | 272 | p = mRR(p, name, DNS_RRTYPE_SRV, DNS_RRCLASS_IN, MDNS_TTL); // SRV, IN |
dirkx | 0:a259777c45a3 | 273 | |
dirkx | 0:a259777c45a3 | 274 | // Keep space for RDLEN |
dirkx | 0:a259777c45a3 | 275 | q = p; |
dirkx | 0:a259777c45a3 | 276 | p+=2; |
dirkx | 0:a259777c45a3 | 277 | |
dirkx | 0:a259777c45a3 | 278 | i = htons(1); // Priority |
dirkx | 0:a259777c45a3 | 279 | memcpy(p, &i, 2); |
dirkx | 0:a259777c45a3 | 280 | p+=2; |
dirkx | 0:a259777c45a3 | 281 | |
dirkx | 0:a259777c45a3 | 282 | i = htons(0); // Weight (see rfc2782) |
dirkx | 0:a259777c45a3 | 283 | memcpy(p, &i, 2); |
dirkx | 0:a259777c45a3 | 284 | p+=2; |
dirkx | 0:a259777c45a3 | 285 | |
dirkx | 0:a259777c45a3 | 286 | i = htons(port); // Port |
dirkx | 0:a259777c45a3 | 287 | memcpy(p, &i, 2); |
dirkx | 0:a259777c45a3 | 288 | p+=2; |
dirkx | 0:a259777c45a3 | 289 | |
dirkx | 0:a259777c45a3 | 290 | p = breakname(p, rr); |
dirkx | 0:a259777c45a3 | 291 | |
dirkx | 0:a259777c45a3 | 292 | i = htons(p - q - 2); // RDLEN |
dirkx | 0:a259777c45a3 | 293 | memcpy(q, &i, 2); |
dirkx | 0:a259777c45a3 | 294 | |
dirkx | 0:a259777c45a3 | 295 | return p; |
dirkx | 0:a259777c45a3 | 296 | } |
dirkx | 0:a259777c45a3 | 297 | |
dirkx | 0:a259777c45a3 | 298 | char * mDNSResponder::qANY(char *p, char * name, uint16_t tpe, uint16_t cls) { |
dirkx | 0:a259777c45a3 | 299 | uint16_t i = 0; |
dirkx | 0:a259777c45a3 | 300 | p = breakname(p, name); // QNAME |
dirkx | 0:a259777c45a3 | 301 | |
dirkx | 0:a259777c45a3 | 302 | i = htons(tpe); // QTYPE |
dirkx | 0:a259777c45a3 | 303 | memcpy(p, &i, 2); |
dirkx | 0:a259777c45a3 | 304 | p+=2; |
dirkx | 0:a259777c45a3 | 305 | |
dirkx | 0:a259777c45a3 | 306 | i = htons(cls); // QCLASS |
dirkx | 0:a259777c45a3 | 307 | memcpy(p, &i, 2); |
dirkx | 0:a259777c45a3 | 308 | p+=2; |
dirkx | 0:a259777c45a3 | 309 | |
dirkx | 0:a259777c45a3 | 310 | return p; |
dirkx | 0:a259777c45a3 | 311 | } |
dirkx | 0:a259777c45a3 | 312 | |
dirkx | 0:a259777c45a3 | 313 | char * mDNSResponder::qPTR(char *p, char *name) { |
dirkx | 0:a259777c45a3 | 314 | return qANY(p,name,DNS_RRTYPE_PTR,DNS_RRCLASS_IN); |
dirkx | 0:a259777c45a3 | 315 | } |
dirkx | 0:a259777c45a3 | 316 | |
dirkx | 0:a259777c45a3 | 317 | char * mDNSResponder::qA(char *p, char *name) { |
dirkx | 0:a259777c45a3 | 318 | return qANY(p,name,DNS_RRTYPE_A,DNS_RRCLASS_IN); |
dirkx | 0:a259777c45a3 | 319 | } |
dirkx | 0:a259777c45a3 | 320 | |
dirkx | 0:a259777c45a3 | 321 | char * mDNSResponder::qSRV(char *p, char *name) { |
dirkx | 0:a259777c45a3 | 322 | return qANY(p,name,DNS_RRTYPE_SRV,DNS_RRCLASS_IN); |
dirkx | 0:a259777c45a3 | 323 | } |
dirkx | 0:a259777c45a3 | 324 | |
dirkx | 0:a259777c45a3 | 325 | void mDNSResponder::mDNSResponder::sendReply(uint16_t t, uint16_t tid, Host dst) { |
dirkx | 0:a259777c45a3 | 326 | // sent a reply - basically a precooked A/PTR/SRV with the TransactionID |
dirkx | 0:a259777c45a3 | 327 | // and my URI squeezed in. |
dirkx | 0:a259777c45a3 | 328 | char out[ 1500 ], *p; |
dirkx | 0:a259777c45a3 | 329 | DNSPacket * op = (DNSPacket *) out; |
dirkx | 0:a259777c45a3 | 330 | |
dirkx | 0:a259777c45a3 | 331 | initLabelCache(out); // Offsets are relative to the ID header. |
dirkx | 0:a259777c45a3 | 332 | |
dirkx | 0:a259777c45a3 | 333 | op->tid = ntohs(tid); |
dirkx | 0:a259777c45a3 | 334 | op->flags = ntohs(0x8000U); // Answer |
dirkx | 0:a259777c45a3 | 335 | op->question_count = ntohs(1); // Courtesy copy of the question. |
dirkx | 0:a259777c45a3 | 336 | op->answer_count = ntohs(1); // The record asked for |
dirkx | 0:a259777c45a3 | 337 | op->a_count = ntohs(0); |
dirkx | 0:a259777c45a3 | 338 | p = out + 12; |
dirkx | 0:a259777c45a3 | 339 | |
dirkx | 0:a259777c45a3 | 340 | switch (t) { |
dirkx | 0:a259777c45a3 | 341 | case DNS_RRTYPE_PTR: // PTR record, SRV, optional TXT and A with my own address. |
dirkx | 0:a259777c45a3 | 342 | op->aa_count = ntohs(moi_txt ? 3 : 2); |
dirkx | 0:a259777c45a3 | 343 | p = qPTR(p,moi_local_proto); |
dirkx | 0:a259777c45a3 | 344 | |
dirkx | 0:a259777c45a3 | 345 | p = mPTR(p,moi_local_proto, moi_local_pglue); |
dirkx | 0:a259777c45a3 | 346 | p = mSRV(p,moi_local_pglue, moi_port, moi_local_name); |
dirkx | 0:a259777c45a3 | 347 | |
dirkx | 0:a259777c45a3 | 348 | if (moi_txt && * moi_txt) |
dirkx | 0:a259777c45a3 | 349 | p = mTXT(p,moi_local_pglue,moi_txt); |
dirkx | 0:a259777c45a3 | 350 | |
dirkx | 0:a259777c45a3 | 351 | break; |
dirkx | 0:a259777c45a3 | 352 | case DNS_RRTYPE_A: // A record (and nothing else) |
dirkx | 0:a259777c45a3 | 353 | op->aa_count = ntohs(1); |
dirkx | 0:a259777c45a3 | 354 | p = qA(p,moi_local_name); |
dirkx | 0:a259777c45a3 | 355 | break; |
dirkx | 0:a259777c45a3 | 356 | |
dirkx | 0:a259777c45a3 | 357 | case DNS_RRTYPE_SRV: // SRV record, optional TXT and a gratious A record to complete. |
dirkx | 0:a259777c45a3 | 358 | op->aa_count = ntohs(moi_txt ? 2 : 1); |
dirkx | 0:a259777c45a3 | 359 | p = qSRV(p,moi_local_pglue); |
dirkx | 0:a259777c45a3 | 360 | |
dirkx | 0:a259777c45a3 | 361 | p = mSRV(p,moi_local_pglue, moi_port, moi_local_name); |
dirkx | 0:a259777c45a3 | 362 | if (moi_txt && * moi_txt) |
dirkx | 0:a259777c45a3 | 363 | p = mTXT(p,moi_local_pglue,moi_txt); |
dirkx | 0:a259777c45a3 | 364 | break; |
dirkx | 0:a259777c45a3 | 365 | } |
dirkx | 0:a259777c45a3 | 366 | p = mARR(p,moi_local_name,moi_ip); |
dirkx | 0:a259777c45a3 | 367 | m_pUDPSocket->sendto(out,p-out,&dst); |
dirkx | 0:a259777c45a3 | 368 | } |
dirkx | 0:a259777c45a3 | 369 | |
dirkx | 0:a259777c45a3 | 370 | char * mDNSResponder::decodeQBlock(char * udp, int len, char *p, char * fqdn, int fqdn_size, uint32_t *Qclass, uint32_t *Qtype) { |
dirkx | 0:a259777c45a3 | 371 | char * endp = udp + len; |
dirkx | 0:a259777c45a3 | 372 | int depth = 0; |
dirkx | 0:a259777c45a3 | 373 | int l, ll = 0; |
dirkx | 0:a259777c45a3 | 374 | char * q = 0; |
dirkx | 0:a259777c45a3 | 375 | char * ep = fqdn; |
dirkx | 0:a259777c45a3 | 376 | |
dirkx | 0:a259777c45a3 | 377 | do { |
dirkx | 0:a259777c45a3 | 378 | while (((l = *p++) < 192) && (l > 0)) { |
dirkx | 0:a259777c45a3 | 379 | if (p + l >= endp) { |
dirkx | 0:a259777c45a3 | 380 | DBG("Malformed packet or bug, RR field larger than packet itself\n\r"); |
dirkx | 0:a259777c45a3 | 381 | return NULL; |
dirkx | 0:a259777c45a3 | 382 | }; |
dirkx | 0:a259777c45a3 | 383 | |
dirkx | 0:a259777c45a3 | 384 | if (ll + l + 1 > fqdn_size) { |
dirkx | 0:a259777c45a3 | 385 | DBG("Malformed packet or bug, FQDN exceeds %d bytes\n\r", fqdn_size); |
dirkx | 0:a259777c45a3 | 386 | return NULL; |
dirkx | 0:a259777c45a3 | 387 | } |
dirkx | 0:a259777c45a3 | 388 | |
dirkx | 0:a259777c45a3 | 389 | memcpy(ep,p,l); |
dirkx | 0:a259777c45a3 | 390 | ep[l]='.'; |
dirkx | 0:a259777c45a3 | 391 | ep += l + 1; |
dirkx | 0:a259777c45a3 | 392 | p += l; |
dirkx | 0:a259777c45a3 | 393 | ll += l + 1; |
dirkx | 0:a259777c45a3 | 394 | }; |
dirkx | 0:a259777c45a3 | 395 | |
dirkx | 0:a259777c45a3 | 396 | if (l >= 192) { |
dirkx | 0:a259777c45a3 | 397 | // construct an offset pointer by wiping the top 1 bits |
dirkx | 0:a259777c45a3 | 398 | // and then getting the remaining 8 bytes to make 14. |
dirkx | 0:a259777c45a3 | 399 | l -= 192; |
dirkx | 0:a259777c45a3 | 400 | l = l << 8; |
dirkx | 0:a259777c45a3 | 401 | l += *p++; |
dirkx | 0:a259777c45a3 | 402 | |
dirkx | 0:a259777c45a3 | 403 | // rescue our reference; as we've not gotten the Qt's yet |
dirkx | 0:a259777c45a3 | 404 | // and these follow the last entry in our record (and not |
dirkx | 0:a259777c45a3 | 405 | // that from the compressed values. |
dirkx | 0:a259777c45a3 | 406 | // |
dirkx | 0:a259777c45a3 | 407 | if (!q) q = p; |
dirkx | 0:a259777c45a3 | 408 | |
dirkx | 0:a259777c45a3 | 409 | // printf(" [->%d] ",l); |
dirkx | 0:a259777c45a3 | 410 | p = udp + l; |
dirkx | 0:a259777c45a3 | 411 | |
dirkx | 0:a259777c45a3 | 412 | if (p >= udp + len || p < udp + 12) { |
dirkx | 0:a259777c45a3 | 413 | DBG("Malformed packet or bug, pointer outside UDP packet\n\r"); |
dirkx | 0:a259777c45a3 | 414 | return NULL; |
dirkx | 0:a259777c45a3 | 415 | }; |
dirkx | 0:a259777c45a3 | 416 | }; |
dirkx | 0:a259777c45a3 | 417 | |
dirkx | 0:a259777c45a3 | 418 | if (depth++ >= 128) { |
dirkx | 0:a259777c45a3 | 419 | DBG("Malformed packet or bug, depth too high\n\r"); |
dirkx | 0:a259777c45a3 | 420 | return NULL; |
dirkx | 0:a259777c45a3 | 421 | }; |
dirkx | 0:a259777c45a3 | 422 | } while (l); |
dirkx | 0:a259777c45a3 | 423 | |
dirkx | 0:a259777c45a3 | 424 | // Terminate the FQDN string. |
dirkx | 0:a259777c45a3 | 425 | *ep = 0; |
dirkx | 0:a259777c45a3 | 426 | |
dirkx | 0:a259777c45a3 | 427 | // in case no compression was used at all. |
dirkx | 0:a259777c45a3 | 428 | if (!q) |
dirkx | 0:a259777c45a3 | 429 | q = p; |
dirkx | 0:a259777c45a3 | 430 | |
dirkx | 0:a259777c45a3 | 431 | *Qtype = htons(q[0] + q[1]*256); |
dirkx | 0:a259777c45a3 | 432 | *Qclass = htons(q[2] + q[3]*256); |
dirkx | 0:a259777c45a3 | 433 | return p; |
dirkx | 0:a259777c45a3 | 434 | } |
dirkx | 0:a259777c45a3 | 435 | |
dirkx | 0:a259777c45a3 | 436 | void mDNSResponder::onUDPSocketEvent(UDPSocketEvent e) { |
dirkx | 0:a259777c45a3 | 437 | char udp[ 1500 ]; |
dirkx | 0:a259777c45a3 | 438 | Host from; |
dirkx | 0:a259777c45a3 | 439 | int len; |
dirkx | 0:a259777c45a3 | 440 | |
dirkx | 0:a259777c45a3 | 441 | //The only event for now |
dirkx | 0:a259777c45a3 | 442 | if (e != UDPSOCKET_READABLE) |
dirkx | 0:a259777c45a3 | 443 | return; |
dirkx | 0:a259777c45a3 | 444 | |
dirkx | 0:a259777c45a3 | 445 | // parse through the packet; find any PTR requests |
dirkx | 0:a259777c45a3 | 446 | // and respond to any for _http._tcp._local. |
dirkx | 0:a259777c45a3 | 447 | // |
dirkx | 0:a259777c45a3 | 448 | while ((len = m_pUDPSocket->recvfrom( (char*)&udp, sizeof(udp), &from )) != 0) { |
dirkx | 0:a259777c45a3 | 449 | DNSPacket * dp = (DNSPacket *) udp; |
dirkx | 0:a259777c45a3 | 450 | |
dirkx | 0:a259777c45a3 | 451 | unsigned int tid = ntohs(dp->tid); |
dirkx | 0:a259777c45a3 | 452 | unsigned int flags = ntohs(dp->flags); |
dirkx | 0:a259777c45a3 | 453 | unsigned int nQ = ntohs(dp->question_count); |
dirkx | 0:a259777c45a3 | 454 | |
dirkx | 0:a259777c45a3 | 455 | if (flags & 2 != 0 || nQ < 1) |
dirkx | 0:a259777c45a3 | 456 | continue; // we only want questions |
dirkx | 0:a259777c45a3 | 457 | |
dirkx | 0:a259777c45a3 | 458 | // assume nQ NON terminated fields followed by Qtype & Qclass |
dirkx | 0:a259777c45a3 | 459 | // |
dirkx | 0:a259777c45a3 | 460 | char * p = udp + 12; |
dirkx | 0:a259777c45a3 | 461 | for (int i = 0; i < nQ; i++) { |
dirkx | 0:a259777c45a3 | 462 | char fqdn[ 256 ]; |
dirkx | 0:a259777c45a3 | 463 | uint32_t Qt, Qc; |
dirkx | 0:a259777c45a3 | 464 | |
dirkx | 0:a259777c45a3 | 465 | if ( (p = decodeQBlock(udp,len,p,fqdn,sizeof(fqdn),&Qc, &Qt)) == 0) |
dirkx | 0:a259777c45a3 | 466 | return; |
dirkx | 0:a259777c45a3 | 467 | |
dirkx | 0:a259777c45a3 | 468 | // We only deal with INternet. |
dirkx | 0:a259777c45a3 | 469 | // |
dirkx | 0:a259777c45a3 | 470 | if (Qc != DNS_RRCLASS_IN) |
dirkx | 0:a259777c45a3 | 471 | continue; |
dirkx | 0:a259777c45a3 | 472 | |
dirkx | 0:a259777c45a3 | 473 | DBG("IN query 0x%x for %s\n\r", Qt, fqdn); |
dirkx | 0:a259777c45a3 | 474 | |
dirkx | 0:a259777c45a3 | 475 | // Normally we'll respond to a mCast query for the PTR of our conceptual |
dirkx | 0:a259777c45a3 | 476 | // service; to which we reply with the services we run; and then the A |
dirkx | 0:a259777c45a3 | 477 | // records for our actual endpoint. However if one of the requistors their |
dirkx | 0:a259777c45a3 | 478 | // cashing is not up to scratch - we may (also) get hit up a bit later for |
dirkx | 0:a259777c45a3 | 479 | // the secondary records under their logical names. So we respond to |
dirkx | 0:a259777c45a3 | 480 | // all 3. As this is likely to happen due to resource constraints on the |
dirkx | 0:a259777c45a3 | 481 | // side of the requestor; we are careful and reply as 'narrow' as possible. |
dirkx | 0:a259777c45a3 | 482 | // |
dirkx | 0:a259777c45a3 | 483 | // Note that we also respond to the ANY query - as to make a quick 'dig' |
dirkx | 0:a259777c45a3 | 484 | // testing command easier. |
dirkx | 0:a259777c45a3 | 485 | // |
dirkx | 0:a259777c45a3 | 486 | if ((Qt == DNS_RRTYPE_PTR || Qt == 255) && !(strcmp(fqdn,moi_local_proto))) |
dirkx | 0:a259777c45a3 | 487 | sendReply(DNS_RRTYPE_PTR,tid,from); |
dirkx | 0:a259777c45a3 | 488 | else if ((Qt == DNS_RRTYPE_A || Qt == 255) && !(strcmp(fqdn,moi_local_name))) |
dirkx | 0:a259777c45a3 | 489 | sendReply(DNS_RRTYPE_A,tid,from); |
dirkx | 0:a259777c45a3 | 490 | else if ((Qt == DNS_RRTYPE_SRV || Qt == 255) && !(strcmp(fqdn,moi_local_pglue))) |
dirkx | 0:a259777c45a3 | 491 | sendReply(DNS_RRTYPE_SRV,tid,from); |
dirkx | 0:a259777c45a3 | 492 | else |
dirkx | 0:a259777c45a3 | 493 | DBG(".. which was ignored\n\r", Qt, fqdn); |
dirkx | 0:a259777c45a3 | 494 | |
dirkx | 0:a259777c45a3 | 495 | }; |
dirkx | 0:a259777c45a3 | 496 | } |
dirkx | 0:a259777c45a3 | 497 | } |
dirkx | 0:a259777c45a3 | 498 | #endif |