]> git.saurik.com Git - apple/mdnsresponder.git/blob - mDNSShared/uds_daemon.c
604f39b7b094217e20981b2c540f19bf75663e24
[apple/mdnsresponder.git] / mDNSShared / uds_daemon.c
1 /* -*- Mode: C; tab-width: 4 -*-
2 *
3 * Copyright (c) 2003-2006 Apple Computer, Inc. All rights reserved.
4 *
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
8 *
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
16
17 Change History (most recent first):
18
19 $Log: uds_daemon.c,v $
20 Revision 1.366 2007/10/10 00:48:54 cheshire
21 <rdar://problem/5526379> Daemon spins in an infinite loop when it doesn't get the control message it's expecting
22
23 Revision 1.365 2007/10/06 03:25:23 cheshire
24 <rdar://problem/5525267> MacBuddy exits abnormally when clicking "Continue" in AppleConnect pane
25
26 Revision 1.364 2007/10/06 03:20:16 cheshire
27 Improved LogOperation debugging messages
28
29 Revision 1.363 2007/10/05 23:24:52 cheshire
30 Improved LogOperation messages about separate error return socket
31
32 Revision 1.362 2007/10/05 22:11:58 cheshire
33 Improved "send_msg ERROR" debugging message
34
35 Revision 1.361 2007/10/04 20:45:18 cheshire
36 <rdar://problem/5518381> Race condition in kDNSServiceFlagsShareConnection-mode call handling
37
38 Revision 1.360 2007/10/01 23:24:46 cheshire
39 SIGINFO output was mislabeling mDNSInterface_Any queries as unicast queries
40
41 Revision 1.359 2007/09/30 00:09:27 cheshire
42 <rdar://problem/5492315> Pass socket fd via SCM_RIGHTS sendmsg instead of using named UDS in the filesystem
43
44 Revision 1.358 2007/09/29 20:08:06 cheshire
45 Fixed typo in comment
46
47 Revision 1.357 2007/09/27 22:10:04 cheshire
48 Add LogOperation line for DNSServiceRegisterRecord callbacks
49
50 Revision 1.356 2007/09/26 21:29:30 cheshire
51 Improved question list SIGINFO output
52
53 Revision 1.355 2007/09/26 01:54:34 mcguire
54 Debugging: In SIGINFO output, show ClientTunnel query interval, which is how we determine whether a query is still active
55
56 Revision 1.354 2007/09/26 01:26:31 cheshire
57 <rdar://problem/5501567> BTMM: mDNSResponder crashes in free_service_instance enabling/disabling BTMM
58 Need to call SendServiceRemovalNotification *before* backpointer is cleared
59
60 Revision 1.353 2007/09/25 20:46:33 cheshire
61 Include DNSServiceRegisterRecord operations in SIGINFO output
62
63 Revision 1.352 2007/09/25 20:23:40 cheshire
64 <rdar://problem/5501567> BTMM: mDNSResponder crashes in free_service_instance enabling/disabling BTMM
65 Need to clear si->request backpointer before calling mDNS_DeregisterService(&mDNSStorage, &si->srs);
66
67 Revision 1.351 2007/09/25 18:20:34 cheshire
68 Changed name of "free_service_instance" to more accurate "unlink_and_free_service_instance"
69
70 Revision 1.350 2007/09/24 23:54:52 mcguire
71 Additional list checking in uds_validatelists()
72
73 Revision 1.349 2007/09/24 06:01:00 cheshire
74 Debugging: In SIGINFO output, show NAT Traversal time values in seconds rather than platform ticks
75
76 Revision 1.348 2007/09/24 05:02:41 cheshire
77 Debugging: In SIGINFO output, indicate explicitly when a given section is empty
78
79 Revision 1.347 2007/09/21 02:04:33 cheshire
80 <rdar://problem/5440831> BTMM: mDNSResponder crashes in free_service_instance enabling/disabling BTMM
81
82 Revision 1.346 2007/09/19 22:47:25 cheshire
83 <rdar://problem/5490182> Memory corruption freeing a "no such service" service record
84
85 Revision 1.345 2007/09/19 20:32:29 cheshire
86 <rdar://problem/5482322> BTMM: Don't advertise SMB with BTMM because it doesn't support IPv6
87
88 Revision 1.344 2007/09/19 19:27:50 cheshire
89 <rdar://problem/5492182> Improved diagnostics when daemon can't connect to error return path socket
90
91 Revision 1.343 2007/09/18 21:42:30 cheshire
92 To reduce programming mistakes, renamed ExtPort to RequestedPort
93
94 Revision 1.342 2007/09/14 22:38:20 cheshire
95 Additional list checking in uds_validatelists()
96
97 Revision 1.341 2007/09/13 00:16:43 cheshire
98 <rdar://problem/5468706> Miscellaneous NAT Traversal improvements
99
100 Revision 1.340 2007/09/12 23:03:08 cheshire
101 <rdar://problem/5476978> DNSServiceNATPortMappingCreate callback not giving correct interface index
102
103 Revision 1.339 2007/09/12 19:22:21 cheshire
104 Variable renaming in preparation for upcoming fixes e.g. priv/pub renamed to intport/extport
105 Made NAT Traversal packet handlers take typed data instead of anonymous "mDNSu8 *" byte pointers
106
107 Revision 1.338 2007/09/12 01:22:13 cheshire
108 Improve validatelists() checking to detect when 'next' pointer gets smashed to ~0
109
110 Revision 1.337 2007/09/07 23:05:04 cheshire
111 Add display of client_context field in handle_cancel_request() LogOperation message
112 While loop was checking client_context.u32[2] instead of client_context.u32[1]
113
114 Revision 1.336 2007/09/07 20:56:03 cheshire
115 Renamed uint32_t field in client_context_t from "ptr64" to more accurate name "u32"
116
117 Revision 1.335 2007/09/05 22:25:01 vazquez
118 <rdar://problem/5400521> update_record mDNSResponder leak
119
120 Revision 1.334 2007/09/05 20:43:57 cheshire
121 Added LogOperation message showing fd of socket listening for incoming Unix Domain Socket client requests
122
123 Revision 1.333 2007/08/28 23:32:35 cheshire
124 Added LogOperation messages for DNSServiceNATPortMappingCreate() operations
125
126 Revision 1.332 2007/08/27 22:59:31 cheshire
127 Show reg_index in DNSServiceRegisterRecord/DNSServiceRemoveRecord messages
128
129 Revision 1.331 2007/08/27 20:29:57 cheshire
130 Added SIGINFO listing of TunnelClients
131
132 Revision 1.330 2007/08/24 23:46:50 cheshire
133 Added debugging messages and SIGINFO listing of DomainAuthInfo records
134
135 Revision 1.329 2007/08/18 01:02:04 mcguire
136 <rdar://problem/5415593> No Bonjour services are getting registered at boot
137
138 Revision 1.328 2007/08/15 20:18:28 vazquez
139 <rdar://problem/5400521> update_record mDNSResponder leak
140 Make sure we free all ExtraResourceRecords
141
142 Revision 1.327 2007/08/08 22:34:59 mcguire
143 <rdar://problem/5197869> Security: Run mDNSResponder as user id mdnsresponder instead of root
144
145 Revision 1.326 2007/08/01 16:09:14 cheshire
146 Removed unused NATTraversalInfo substructure from AuthRecord; reduced structure sizecheck values accordingly
147
148 Revision 1.325 2007/07/31 21:29:41 cheshire
149 <rdar://problem/5372207> System Default registration domain(s) not listed in Domain Enumeration ("dns-sd -E")
150
151 Revision 1.324 2007/07/31 01:56:21 cheshire
152 Corrected function name in log message
153
154 Revision 1.323 2007/07/27 23:57:23 cheshire
155 Added compile-time structure size checks
156
157 Revision 1.322 2007/07/27 19:37:19 cheshire
158 Moved AutomaticBrowseDomainQ into main mDNS object
159
160 Revision 1.321 2007/07/27 19:30:41 cheshire
161 Changed mDNSQuestionCallback parameter from mDNSBool to QC_result,
162 to properly reflect tri-state nature of the possible responses
163
164 Revision 1.320 2007/07/27 00:48:27 cheshire
165 <rdar://problem/4700198> BTMM: Services should only get registered in .Mac domain of current user
166 <rdar://problem/4731180> BTMM: Only browse in the current user's .Mac domain by default
167
168 Revision 1.319 2007/07/24 17:23:33 cheshire
169 <rdar://problem/5357133> Add list validation checks for debugging
170
171 Revision 1.318 2007/07/23 23:09:51 cheshire
172 <rdar://problem/5351997> Reject oversized client requests
173
174 Revision 1.317 2007/07/23 22:24:47 cheshire
175 <rdar://problem/5352299> Make mDNSResponder more defensive against malicious local clients
176 Additional refinements
177
178 Revision 1.316 2007/07/23 22:12:53 cheshire
179 <rdar://problem/5352299> Make mDNSResponder more defensive against malicious local clients
180
181 Revision 1.315 2007/07/21 01:36:13 cheshire
182 Need to also add ".local" as automatic browsing domain
183
184 Revision 1.314 2007/07/20 20:12:37 cheshire
185 Rename "mDNS_DomainTypeBrowseLegacy" as "mDNS_DomainTypeBrowseAutomatic"
186
187 Revision 1.313 2007/07/20 00:54:21 cheshire
188 <rdar://problem/4641118> Need separate SCPreferences for per-user .Mac settings
189
190 Revision 1.312 2007/07/11 03:06:43 cheshire
191 <rdar://problem/5303807> Register IPv6-only hostname and don't create port mappings for AutoTunnel services
192
193 Revision 1.311 2007/07/06 21:19:18 cheshire
194 Add list of NAT traversals to SIGINFO output
195
196 Revision 1.310 2007/07/03 19:56:50 cheshire
197 Add LogOperation message for DNSServiceSetDefaultDomainForUser
198
199 Revision 1.309 2007/06/29 23:12:49 vazquez
200 <rdar://problem/5294103> Stop using generate_final_fatal_reply_with_garbage
201
202 Revision 1.308 2007/06/29 00:10:07 vazquez
203 <rdar://problem/5301908> Clean up NAT state machine (necessary for 6 other fixes)
204
205 Revision 1.307 2007/05/25 00:25:44 cheshire
206 <rdar://problem/5227737> Need to enhance putRData to output all current known types
207
208 Revision 1.306 2007/05/24 22:31:35 vazquez
209 Bug #: 4272956
210 Reviewed by: Stuart Cheshire
211 <rdar://problem/4272956> WWDC API: Return ADD/REMOVE events in registration callback
212
213 Revision 1.305 2007/05/23 18:59:22 cheshire
214 Remove unnecessary IPC_FLAGS_REUSE_SOCKET
215
216 Revision 1.304 2007/05/22 01:07:42 cheshire
217 <rdar://problem/3563675> API: Need a way to get version/feature information
218
219 Revision 1.303 2007/05/22 00:32:58 cheshire
220 Make a send_all() subroutine -- will be helpful for implementing DNSServiceGetProperty(DaemonVersion)
221
222 Revision 1.302 2007/05/21 18:54:54 cheshire
223 Add "Cancel" LogOperation message when we get a cancel_request command over the UDS
224
225 Revision 1.301 2007/05/18 23:55:22 cheshire
226 <rdar://problem/4454655> Allow multiple register/browse/resolve operations to share single Unix Domain Socket
227
228 Revision 1.300 2007/05/18 21:27:11 cheshire
229 Rename connected_registration_termination to connection_termination
230
231 Revision 1.299 2007/05/18 21:24:34 cheshire
232 Rename rstate to request
233
234 Revision 1.298 2007/05/18 21:22:35 cheshire
235 Convert uint16_t etc. to their locally-defined equivalents, like the rest of the core code
236
237 Revision 1.297 2007/05/18 20:33:11 cheshire
238 Avoid declaring lots of uninitialized variables in read_rr_from_ipc_msg
239
240 Revision 1.296 2007/05/18 19:04:19 cheshire
241 Rename msgdata to msgptr (may be modified); rename (currently unused) bufsize to msgend
242
243 Revision 1.295 2007/05/18 17:57:13 cheshire
244 Reorder functions in file to arrange them in logical groups; added "#pragma mark" headers for each group
245
246 Revision 1.294 2007/05/17 20:58:22 cheshire
247 <rdar://problem/4647145> DNSServiceQueryRecord should return useful information with NXDOMAIN
248
249 Revision 1.293 2007/05/17 19:46:20 cheshire
250 Routine name deliver_async_error() is misleading. What it actually does is write a message header
251 (containing an error code) followed by 256 bytes of garbage zeroes onto a client connection,
252 thereby trashing it and making it useless for any subsequent communication. It's destructive,
253 and not very useful. Changing name to generate_final_fatal_reply_with_garbage().
254
255 Revision 1.292 2007/05/16 01:06:52 cheshire
256 <rdar://problem/4471320> Improve reliability of kDNSServiceFlagsMoreComing flag on multiprocessor machines
257
258 Revision 1.291 2007/05/15 21:57:16 cheshire
259 <rdar://problem/4608220> Use dnssd_SocketValid(x) macro instead of just
260 assuming that all negative values (or zero!) are invalid socket numbers
261
262 Revision 1.290 2007/05/10 23:30:57 cheshire
263 <rdar://problem/4084490> Only one browse gets remove events when disabling browse domain
264
265 Revision 1.289 2007/05/02 22:18:08 cheshire
266 Renamed NATTraversalInfo_struct context to NATTraversalContext
267
268 Revision 1.288 2007/04/30 21:33:39 cheshire
269 Fix crash when a callback unregisters a service while the UpdateSRVRecords() loop
270 is iterating through the m->ServiceRegistrations list
271
272 Revision 1.287 2007/04/27 19:03:22 cheshire
273 Check q->LongLived not q->llq to tell if a query is LongLived
274
275 Revision 1.286 2007/04/26 16:00:01 cheshire
276 Show interface number in DNSServiceBrowse RESULT output
277
278 Revision 1.285 2007/04/22 19:03:39 cheshire
279 Minor code tidying
280
281 Revision 1.284 2007/04/22 06:02:03 cheshire
282 <rdar://problem/4615977> Query should immediately return failure when no server
283
284 Revision 1.283 2007/04/21 21:47:47 cheshire
285 <rdar://problem/4376383> Daemon: Add watchdog timer
286
287 Revision 1.282 2007/04/20 21:17:24 cheshire
288 For naming consistency, kDNSRecordTypeNegative should be kDNSRecordTypePacketNegative
289
290 Revision 1.281 2007/04/19 23:25:20 cheshire
291 Added debugging message
292
293 Revision 1.280 2007/04/17 19:21:29 cheshire
294 <rdar://problem/5140339> Domain discovery not working over VPN
295
296 Revision 1.279 2007/04/16 21:53:49 cheshire
297 Improve display of negative cache entries
298
299 Revision 1.278 2007/04/16 20:49:40 cheshire
300 Fix compile errors for mDNSPosix build
301
302 Revision 1.277 2007/04/05 22:55:36 cheshire
303 <rdar://problem/5077076> Records are ending up in Lighthouse without expiry information
304
305 Revision 1.276 2007/04/05 19:20:13 cheshire
306 Non-blocking mode not being set correctly -- was clobbering other flags
307
308 Revision 1.275 2007/04/04 21:21:25 cheshire
309 <rdar://problem/4546810> Fix crash: In regservice_callback service_instance was being referenced after being freed
310
311 Revision 1.274 2007/04/04 01:30:42 cheshire
312 <rdar://problem/5075200> DNSServiceAddRecord is failing to advertise NULL record
313 Add SIGINFO output lising our advertised Authoritative Records
314
315 Revision 1.273 2007/04/04 00:03:27 cheshire
316 <rdar://problem/5089862> DNSServiceQueryRecord is returning kDNSServiceErr_NoSuchRecord for empty rdata
317
318 Revision 1.272 2007/04/03 20:10:32 cheshire
319 Show ADD/RMV in DNSServiceQueryRecord log message instead of just "RESULT"
320
321 Revision 1.271 2007/04/03 19:22:32 cheshire
322 Use mDNSSameIPv4Address (and similar) instead of accessing internal fields directly
323
324 Revision 1.270 2007/03/30 21:55:30 cheshire
325 Added comments
326
327 Revision 1.269 2007/03/29 01:31:44 cheshire
328 Faulty logic was incorrectly suppressing some NAT port mapping callbacks
329
330 Revision 1.268 2007/03/29 00:13:58 cheshire
331 Remove unnecessary fields from service_instance structure: autoname, autorename, allowremotequery, name
332
333 Revision 1.267 2007/03/28 20:59:27 cheshire
334 <rdar://problem/4743285> Remove inappropriate use of IsPrivateV4Addr()
335
336 Revision 1.266 2007/03/28 15:56:37 cheshire
337 <rdar://problem/5085774> Add listing of NAT port mapping and GetAddrInfo requests in SIGINFO output
338
339 Revision 1.265 2007/03/27 22:52:07 cheshire
340 Fix crash in udsserver_automatic_browse_domain_changed
341
342 Revision 1.264 2007/03/27 00:49:40 cheshire
343 Should use mallocL, not plain malloc
344
345 Revision 1.263 2007/03/27 00:45:01 cheshire
346 Removed unnecessary "void *termination_context" pointer
347
348 Revision 1.262 2007/03/27 00:40:43 cheshire
349 Eliminate resolve_termination_t as a separately-allocated structure, and make it part of the request_state union
350
351 Revision 1.261 2007/03/27 00:29:00 cheshire
352 Eliminate queryrecord_request data as a separately-allocated structure, and make it part of the request_state union
353
354 Revision 1.260 2007/03/27 00:18:42 cheshire
355 Eliminate enum_termination_t and domain_enum_t as separately-allocated structures,
356 and make them part of the request_state union
357
358 Revision 1.259 2007/03/26 23:48:16 cheshire
359 <rdar://problem/4848295> Advertise model information via Bonjour
360 Refinements to reduce unnecessary transmissions of the DeviceInfo TXT record
361
362 Revision 1.258 2007/03/24 00:40:04 cheshire
363 Minor code cleanup
364
365 Revision 1.257 2007/03/24 00:23:12 cheshire
366 Eliminate port_mapping_info_t as a separately-allocated structure, and make it part of the request_state union
367
368 Revision 1.256 2007/03/24 00:07:18 cheshire
369 Eliminate addrinfo_info_t as a separately-allocated structure, and make it part of the request_state union
370
371 Revision 1.255 2007/03/23 23:56:14 cheshire
372 Move list of record registrations into the request_state union
373
374 Revision 1.254 2007/03/23 23:48:56 cheshire
375 Eliminate service_info as a separately-allocated structure, and make it part of the request_state union
376
377 Revision 1.253 2007/03/23 23:04:29 cheshire
378 Eliminate browser_info_t as a separately-allocated structure, and make it part of request_state
379
380 Revision 1.252 2007/03/23 22:59:58 cheshire
381 <rdar://problem/4848295> Advertise model information via Bonjour
382 Use kStandardTTL, not kHostNameTTL
383
384 Revision 1.251 2007/03/23 22:44:07 cheshire
385 Instead of calling AbortUnlinkAndFree() haphazardly all over the place, make the handle* routines
386 return an error code, and then request_callback() does all necessary cleanup in one place.
387
388 Revision 1.250 2007/03/22 20:30:07 cheshire
389 Remove pointless "if (request->ts != t_complete) ..." checks
390
391 Revision 1.249 2007/03/22 20:13:27 cheshire
392 Delete unused client_context field
393
394 Revision 1.248 2007/03/22 20:03:37 cheshire
395 Rename variables for clarity: instead of using variable rs for both request_state
396 and reply_state, use req for request_state and rep for reply_state
397
398 Revision 1.247 2007/03/22 19:31:42 cheshire
399 <rdar://problem/4848295> Advertise model information via Bonjour
400 Add missing "model=" at start of DeviceInfo data
401
402 Revision 1.246 2007/03/22 18:31:48 cheshire
403 Put dst parameter first in mDNSPlatformStrCopy/mDNSPlatformMemCopy, like conventional Posix strcpy/memcpy
404
405 Revision 1.245 2007/03/22 00:49:20 cheshire
406 <rdar://problem/4848295> Advertise model information via Bonjour
407
408 Revision 1.244 2007/03/21 21:01:48 cheshire
409 <rdar://problem/4789793> Leak on error path in regrecord_callback, uds_daemon.c
410
411 Revision 1.243 2007/03/21 19:01:57 cheshire
412 <rdar://problem/5078494> IPC code not 64-bit-savvy: assumes long=32bits, and short=16bits
413
414 Revision 1.242 2007/03/21 18:51:21 cheshire
415 <rdar://problem/4549320> Code in uds_daemon.c passes function name instead of type name to mallocL/freeL
416
417 Revision 1.241 2007/03/20 00:04:50 cheshire
418 <rdar://problem/4837929> Should allow "udp" or "tcp" for protocol command-line arg
419 Fix LogOperation("DNSServiceNATPortMappingCreate(...)") message to actually show client arguments
420
421 Revision 1.240 2007/03/16 23:25:35 cheshire
422 <rdar://problem/5067001> NAT-PMP: Parameter validation not working correctly
423
424 Revision 1.239 2007/03/10 02:29:36 cheshire
425 Added comment about port_mapping_create_reply()
426
427 Revision 1.238 2007/03/07 00:26:48 cheshire
428 <rdar://problem/4426754> DNSServiceRemoveRecord log message should include record type
429
430 Revision 1.237 2007/02/28 01:44:29 cheshire
431 <rdar://problem/5027863> Byte order bugs in uDNS.c, uds_daemon.c, dnssd_clientstub.c
432
433 Revision 1.236 2007/02/14 01:58:19 cheshire
434 <rdar://problem/4995831> Don't delete Unix Domain Socket on exit if we didn't create it on startup
435
436 Revision 1.235 2007/02/08 21:12:28 cheshire
437 <rdar://problem/4386497> Stop reading /etc/mDNSResponder.conf on every sleep/wake
438
439 Revision 1.234 2007/02/06 19:06:49 cheshire
440 <rdar://problem/3956518> Need to go native with launchd
441
442 Revision 1.233 2007/01/10 20:49:37 cheshire
443 Remove unnecessary setting of q->Private fields
444
445 Revision 1.232 2007/01/09 00:03:23 cheshire
446 Call udsserver_handle_configchange() once at the end of udsserver_init()
447 to set up the automatic registration and browsing domains.
448
449 Revision 1.231 2007/01/06 02:50:19 cheshire
450 <rdar://problem/4632919> Instead of copying SRV and TXT record data, just store pointers to cache entities
451
452 Revision 1.230 2007/01/06 01:00:35 cheshire
453 Improved SIGINFO output
454
455 Revision 1.229 2007/01/05 08:30:56 cheshire
456 Trim excessive "$Log" checkin history from before 2006
457 (checkin history still available via "cvs log ..." of course)
458
459 Revision 1.228 2007/01/05 08:09:05 cheshire
460 Reorder code into functional sections, with "#pragma mark" headers
461
462 Revision 1.227 2007/01/05 07:04:24 cheshire
463 Minor code tidying
464
465 Revision 1.226 2007/01/05 05:44:35 cheshire
466 Move automatic browse/registration management from uDNS.c to mDNSShared/uds_daemon.c,
467 so that mDNSPosix embedded clients will compile again
468
469 Revision 1.225 2007/01/04 23:11:15 cheshire
470 <rdar://problem/4720673> uDNS: Need to start caching unicast records
471 When an automatic browsing domain is removed, generate appropriate "remove" events for legacy queries
472
473 Revision 1.224 2007/01/04 20:57:49 cheshire
474 Rename ReturnCNAME to ReturnIntermed (for ReturnIntermediates)
475
476 Revision 1.223 2006/12/21 01:25:49 cheshire
477 Tidy up SIGINFO state log
478
479 Revision 1.222 2006/12/21 00:15:22 cheshire
480 Get rid of gmDNS macro; fixed a crash in udsserver_info()
481
482 Revision 1.221 2006/12/20 04:07:38 cheshire
483 Remove uDNS_info substructure from AuthRecord_struct
484
485 Revision 1.220 2006/12/19 22:49:25 cheshire
486 Remove uDNS_info substructure from ServiceRecordSet_struct
487
488 Revision 1.219 2006/12/14 03:02:38 cheshire
489 <rdar://problem/4838433> Tools: dns-sd -G 0 only returns IPv6 when you have a routable IPv6 address
490
491 Revision 1.218 2006/11/18 05:01:33 cheshire
492 Preliminary support for unifying the uDNS and mDNS code,
493 including caching of uDNS answers
494
495 Revision 1.217 2006/11/15 19:27:53 mkrochma
496 <rdar://problem/4838433> Tools: dns-sd -G 0 only returns IPv6 when you have a routable IPv6 address
497
498 Revision 1.216 2006/11/10 00:54:16 cheshire
499 <rdar://problem/4816598> Changing case of Computer Name doesn't work
500
501 Revision 1.215 2006/10/27 01:30:23 cheshire
502 Need explicitly to set ReturnIntermed = mDNSfalse
503
504 Revision 1.214 2006/10/20 05:37:23 herscher
505 Display question list information in udsserver_info()
506
507 Revision 1.213 2006/10/05 03:54:31 herscher
508 Remove embedded uDNS_info struct from DNSQuestion_struct
509
510 Revision 1.212 2006/09/30 01:22:35 cheshire
511 Put back UTF-8 curly quotes in log messages
512
513 Revision 1.211 2006/09/27 00:44:55 herscher
514 <rdar://problem/4249761> API: Need DNSServiceGetAddrInfo()
515
516 Revision 1.210 2006/09/26 01:52:41 herscher
517 <rdar://problem/4245016> NAT Port Mapping API (for both NAT-PMP and UPnP Gateway Protocol)
518
519 Revision 1.209 2006/09/21 21:34:09 cheshire
520 <rdar://problem/4100000> Allow empty string name when using kDNSServiceFlagsNoAutoRename
521
522 Revision 1.208 2006/09/21 21:28:24 cheshire
523 Code cleanup to make it consistent with daemon.c: change rename_on_memfree to renameonmemfree
524
525 Revision 1.207 2006/09/15 21:20:16 cheshire
526 Remove uDNS_info substructure from mDNS_struct
527
528 Revision 1.206 2006/08/14 23:24:56 cheshire
529 Re-licensed mDNSResponder daemon source code under Apache License, Version 2.0
530
531 Revision 1.205 2006/07/20 22:07:30 mkrochma
532 <rdar://problem/4633196> Wide-area browsing is currently broken in TOT
533 More fixes for uninitialized variables
534
535 Revision 1.204 2006/07/15 02:01:33 cheshire
536 <rdar://problem/4472014> Add Private DNS client functionality to mDNSResponder
537 Fix broken "empty string" browsing
538
539 Revision 1.203 2006/07/07 01:09:13 cheshire
540 <rdar://problem/4472013> Add Private DNS server functionality to dnsextd
541 Only use mallocL/freeL debugging routines when building mDNSResponder, not dnsextd
542
543 Revision 1.202 2006/07/05 22:00:10 cheshire
544 Wide-area cleanup: Rename mDNSPlatformGetRegDomainList() to uDNS_GetDefaultRegDomainList()
545
546 Revision 1.201 2006/06/29 03:02:47 cheshire
547 <rdar://problem/4607042> mDNSResponder NXDOMAIN and CNAME support
548
549 Revision 1.200 2006/06/28 08:56:26 cheshire
550 Added "_op" to the end of the operation code enum values,
551 to differentiate them from the routines with the same names
552
553 Revision 1.199 2006/06/28 08:53:39 cheshire
554 Added (commented out) debugging messages
555
556 Revision 1.198 2006/06/27 20:16:07 cheshire
557 Fix code layout
558
559 Revision 1.197 2006/05/18 01:32:35 cheshire
560 <rdar://problem/4472706> iChat: Lost connection with Bonjour
561 (mDNSResponder insufficiently defensive against malformed browsing PTR responses)
562
563 Revision 1.196 2006/05/05 07:07:13 cheshire
564 <rdar://problem/4538206> mDNSResponder fails when UDS reads deliver partial data
565
566 Revision 1.195 2006/04/25 20:56:28 mkrochma
567 Added comment about previous checkin
568
569 Revision 1.194 2006/04/25 18:29:36 mkrochma
570 Workaround for warning: unused variable 'status' when building mDNSPosix
571
572 Revision 1.193 2006/03/19 17:14:38 cheshire
573 <rdar://problem/4483117> Need faster purging of stale records
574 read_rr_from_ipc_msg was not setting namehash and rdatahash
575
576 Revision 1.192 2006/03/18 20:58:32 cheshire
577 Misplaced curly brace
578
579 Revision 1.191 2006/03/10 22:19:43 cheshire
580 Update debugging message in resolve_result_callback() to indicate whether event is ADD or RMV
581
582 Revision 1.190 2006/03/10 21:56:12 cheshire
583 <rdar://problem/4111464> After record update, old record sometimes remains in cache
584 When service TXT and SRV record both change, clients with active resolve calls get *two* callbacks, one
585 when the TXT data changes, and then immediately afterwards a second callback with the new port number
586 This change suppresses the first unneccessary (and confusing) callback
587
588 Revision 1.189 2006/01/06 00:56:31 cheshire
589 <rdar://problem/4400573> Should remove PID file on exit
590
591 */
592
593 #if defined(_WIN32)
594 #include <process.h>
595 #define dnssd_strerror(X) win32_strerror(X)
596 #define usleep(X) Sleep(((X)+999)/1000)
597 mDNSlocal char *win32_strerror(int inErrorCode)
598 {
599 static char buffer[1024];
600 DWORD n;
601 memset(buffer, 0, sizeof(buffer));
602 n = FormatMessageA(
603 FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
604 NULL,
605 (DWORD) inErrorCode,
606 MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
607 buffer,
608 sizeof(buffer),
609 NULL);
610 if (n > 0)
611 {
612 // Remove any trailing CR's or LF's since some messages have them.
613 while ((n > 0) && isspace(((unsigned char *) buffer)[n - 1]))
614 buffer[--n] = '\0';
615 }
616 return buffer;
617 }
618 #else
619 #include <fcntl.h>
620 #include <errno.h>
621 #include <sys/ioctl.h>
622 #include <sys/types.h>
623 #include <sys/time.h>
624 #include <sys/resource.h>
625 #define dnssd_strerror(X) strerror(X)
626 #endif
627
628 #include <stdlib.h>
629 #include <stdio.h>
630 #include "mDNSEmbeddedAPI.h"
631 #include "DNSCommon.h"
632 #include "uDNS.h"
633 #include "uds_daemon.h"
634
635 // Apple-specific functionality, not required for other platforms
636 #if APPLE_OSX_mDNSResponder
637 #include <sys/ucred.h>
638 #ifndef PID_FILE
639 #define PID_FILE ""
640 #endif
641 #endif
642
643 // User IDs 0-500 are system-wide processes, not actual users in the usual sense
644 // User IDs for real user accounts start at 501 and count up from there
645 #define SystemUID(X) ((X) <= 500)
646
647 // ***************************************************************************
648 #if COMPILER_LIKES_PRAGMA_MARK
649 #pragma mark -
650 #pragma mark - Types and Data Structures
651 #endif
652
653 typedef enum
654 {
655 t_uninitialized,
656 t_morecoming,
657 t_complete,
658 t_error,
659 t_terminated
660 } transfer_state;
661
662 typedef struct request_state request_state;
663
664 typedef void (*req_termination_fn)(request_state *request);
665
666 typedef struct registered_record_entry
667 {
668 struct registered_record_entry *next;
669 mDNSu32 key;
670 AuthRecord *rr; // Variable-sized AuthRecord
671 client_context_t client_context;
672 request_state *request;
673 } registered_record_entry;
674
675 // A single registered service: ServiceRecordSet + bookkeeping
676 // Note that we duplicate some fields from parent service_info object
677 // to facilitate cleanup, when instances and parent may be deallocated at different times.
678 typedef struct service_instance
679 {
680 struct service_instance *next;
681 request_state *request;
682 dnssd_sock_t sd;
683 AuthRecord *subtypes;
684 mDNSBool renameonmemfree; // Set on config change when we deregister original name
685 mDNSBool clientnotified; // Has client been notified of successful registration yet?
686 mDNSBool default_local; // is this the "local." from an empty-string registration?
687 domainname domain;
688 ServiceRecordSet srs; // note - must be last field in struct
689 } service_instance;
690
691 // for multi-domain default browsing
692 typedef struct browser_t
693 {
694 struct browser_t *next;
695 domainname domain;
696 DNSQuestion q;
697 } browser_t;
698
699 struct request_state
700 {
701 request_state *next;
702 request_state *primary; // If this operation is on a shared socket, pointer to
703 // primary request_state for the original DNSServiceConnect() operation
704 dnssd_sock_t sd;
705 dnssd_sock_t errsd;
706 mDNSu32 uid;
707
708 // NOTE: On a shared connection these fields in the primary structure, including hdr, are re-used
709 // for each new request. This is because, until we've read the ipc_msg_hdr to find out what the
710 // operation is, we don't know if we're going to need to allocate a new request_state or not.
711 transfer_state ts;
712 mDNSu32 hdr_bytes; // bytes of header already read
713 ipc_msg_hdr hdr;
714 mDNSu32 data_bytes; // bytes of message data already read
715 char *msgbuf; // pointer to data storage to pass to free()
716 char *msgptr; // pointer to data to be read from (may be modified)
717 char *msgend; // pointer to byte after last byte of message
718
719 // reply, termination, error, and client context info
720 int no_reply; // don't send asynchronous replies to client
721 int time_blocked; // record time of a blocked client
722 struct reply_state *replies; // corresponding (active) reply list
723 req_termination_fn terminate;
724
725 union
726 {
727 registered_record_entry *reg_recs; // list of registrations for a connection-oriented request
728 struct
729 {
730 mDNSInterfaceID interface_id;
731 mDNSBool default_domain;
732 mDNSBool ForceMCast;
733 domainname regtype;
734 browser_t *browsers;
735 } browser;
736 struct
737 {
738 mDNSInterfaceID InterfaceID;
739 mDNSu16 txtlen;
740 void *txtdata;
741 mDNSIPPort port;
742 domainlabel name;
743 char type_as_string[MAX_ESCAPED_DOMAIN_NAME];
744 domainname type;
745 mDNSBool default_domain;
746 domainname host;
747 mDNSBool autoname; // Set if this name is tied to the Computer Name
748 mDNSBool autorename; // Set if this client wants us to automatically rename on conflict
749 mDNSBool allowremotequery; // Respond to unicast queries from outside the local link?
750 int num_subtypes;
751 service_instance *instances;
752 } servicereg;
753 struct
754 {
755 mDNSInterfaceID interface_id;
756 mDNSu32 flags;
757 mDNSu32 protocol;
758 DNSQuestion q4;
759 DNSQuestion q6;
760 } addrinfo;
761 struct
762 {
763 mDNSIPPort ReqExt; // External port we originally requested, for logging purposes
764 NATTraversalInfo NATinfo;
765 } pm;
766 struct
767 {
768 DNSQuestion q_all;
769 DNSQuestion q_default;
770 } enumeration;
771 struct
772 {
773 DNSQuestion q;
774 } queryrecord;
775 struct
776 {
777 DNSQuestion qtxt;
778 DNSQuestion qsrv;
779 const ResourceRecord *txt;
780 const ResourceRecord *srv;
781 } resolve;
782 ;
783 } u;
784 };
785
786 // struct physically sits between ipc message header and call-specific fields in the message buffer
787 typedef struct
788 {
789 DNSServiceFlags flags; // Note: This field is in NETWORK byte order
790 mDNSu32 ifi; // Note: This field is in NETWORK byte order
791 DNSServiceErrorType error; // Note: This field is in NETWORK byte order
792 } reply_hdr;
793
794 typedef struct reply_state
795 {
796 dnssd_sock_t sd;
797 transfer_state ts;
798 mDNSu32 nwriten;
799 mDNSu32 len;
800 request_state *request; // the request that this answers
801 struct reply_state *next; // if there are multiple unsent replies
802 char *msgbuf; // pointer to malloc'd buffer
803 ipc_msg_hdr *mhdr; // pointer into message buffer - allows fields to be changed after message is formatted
804 reply_hdr *rhdr;
805 char *sdata; // pointer to start of call-specific data
806 } reply_state;
807
808 // ***************************************************************************
809 #if COMPILER_LIKES_PRAGMA_MARK
810 #pragma mark -
811 #pragma mark - Globals
812 #endif
813
814 // globals
815 mDNSexport mDNS mDNSStorage;
816 mDNSexport const char ProgramName[] = "mDNSResponder";
817
818 static dnssd_sock_t listenfd = dnssd_InvalidSocket;
819 static request_state *all_requests = NULL;
820
821 static DNameListElem *SCPrefBrowseDomains; // List of automatic browsing domains read from SCPreferences for "empty string" browsing
822 static ARListElem *LocalDomainEnumRecords; // List of locally-generated PTR records to augment those we learn from the network
823 mDNSexport DNameListElem *AutoBrowseDomains; // List created from those local-only PTR records plus records we get from the network
824
825 mDNSexport DNameListElem *AutoRegistrationDomains; // Domains where we automatically register for empty-string registrations
826
827 #define MSG_PAD_BYTES 5 // pad message buffer (read from client) with n zero'd bytes to guarantee
828 // n get_string() calls w/o buffer overrun
829 // initialization, setup/teardown functions
830
831 // If a platform specifies its own PID file name, we use that
832 #ifndef PID_FILE
833 #define PID_FILE "/var/run/mDNSResponder.pid"
834 #endif
835
836 // ***************************************************************************
837 #if COMPILER_LIKES_PRAGMA_MARK
838 #pragma mark -
839 #pragma mark - General Utility Functions
840 #endif
841
842 mDNSlocal void FatalError(char *errmsg)
843 {
844 LogMsg("%s: %s", errmsg, dnssd_strerror(dnssd_errno()));
845 *(long*)0 = 0; // On OS X abort() doesn't generate a crash log, but writing to zero does
846 abort(); // On platforms where writing to zero doesn't generate an exception, abort instead
847 }
848
849 mDNSlocal mDNSu32 dnssd_htonl(mDNSu32 l)
850 {
851 mDNSu32 ret;
852 char *data = (char*) &ret;
853 put_uint32(l, &data);
854 return ret;
855 }
856
857 // hack to search-replace perror's to LogMsg's
858 mDNSlocal void my_perror(char *errmsg)
859 {
860 LogMsg("%s: %s", errmsg, dnssd_strerror(dnssd_errno()));
861 }
862
863 mDNSlocal void abort_request(request_state *req)
864 {
865 // First stop whatever mDNSCore operation we were doing
866 if (req->terminate) req->terminate(req);
867
868 // Now, if this request_state is not subbordinate to some other primary, close file descriptor and discard replies
869 if (!req->primary)
870 {
871 if (req->errsd != req->sd) LogOperation("%3d: Removing FD and closing errsd %d", req->sd, req->errsd);
872 else LogOperation("%3d: Removing FD", req->sd);
873 udsSupportRemoveFDFromEventLoop(req->sd); // Note: This also closes file descriptor req->sd for us
874 if (req->errsd != req->sd) { dnssd_close(req->errsd); req->errsd = req->sd; }
875
876 while (req->replies) // free pending replies
877 {
878 reply_state *ptr = req->replies;
879 req->replies = req->replies->next;
880 if (ptr->msgbuf) freeL("reply_state msgbuf (abort)", ptr->msgbuf);
881 freeL("reply_state (abort)", ptr);
882 }
883 }
884
885 // Set req->sd to something invalid, so that udsserver_idle knows to unlink and free this structure
886 #if APPLE_OSX_mDNSResponder && MACOSX_MDNS_MALLOC_DEBUGGING
887 // Don't use dnssd_InvalidSocket (-1) because that's the sentinel value MACOSX_MDNS_MALLOC_DEBUGGING uses
888 // for detecting when the memory for an object is inadvertently freed while the object is still on some list
889 req->sd = -2;
890 #else
891 req->sd = dnssd_InvalidSocket;
892 #endif
893 }
894
895 mDNSlocal void AbortUnlinkAndFree(request_state *req)
896 {
897 request_state **p = &all_requests;
898 abort_request(req);
899 while (*p && *p != req) p=&(*p)->next;
900 if (*p) { *p = req->next; freeL("request_state/AbortUnlinkAndFree", req); }
901 }
902
903 mDNSlocal reply_state *create_reply(const reply_op_t op, const size_t datalen, request_state *const request)
904 {
905 reply_state *reply;
906 int totallen = (int) (datalen + sizeof(ipc_msg_hdr));
907
908 if ((unsigned)datalen < sizeof(reply_hdr))
909 {
910 LogMsg("ERROR: create_reply - data length less than lenght of required fields");
911 return NULL;
912 }
913
914 reply = mallocL("reply_state", sizeof(reply_state));
915 if (!reply) FatalError("ERROR: malloc");
916 mDNSPlatformMemZero(reply, sizeof(reply_state));
917 reply->ts = t_morecoming;
918 reply->sd = request->sd;
919 reply->request = request;
920 reply->len = totallen;
921 reply->msgbuf = mallocL("reply_state msgbuf", totallen);
922 if (!reply->msgbuf) FatalError("ERROR: malloc");
923 mDNSPlatformMemZero(reply->msgbuf, totallen);
924 reply->mhdr = (ipc_msg_hdr *)reply->msgbuf;
925 reply->rhdr = (reply_hdr *)(reply->msgbuf + sizeof(ipc_msg_hdr));
926 reply->sdata = reply->msgbuf + sizeof(ipc_msg_hdr) + sizeof(reply_hdr);
927 reply->mhdr->version = VERSION;
928 reply->mhdr->datalen = datalen;
929 reply->mhdr->ipc_flags = 0;
930 reply->mhdr->op = op;
931 reply->mhdr->client_context = request->hdr.client_context;
932 reply->mhdr->reg_index = 0;
933 return reply;
934 }
935
936 // Append a reply to the list in a request object
937 // If our request is sharing a connection, then we append our reply_state onto the primary's list
938 mDNSlocal void append_reply(request_state *req, reply_state *rep)
939 {
940 request_state *r = req->primary ? req->primary : req;
941 reply_state **ptr = &r->replies;
942 while (*ptr) ptr = &(*ptr)->next;
943 *ptr = rep;
944 rep->next = NULL;
945 }
946
947 // Generates a response message giving name, type, domain, plus interface index,
948 // suitable for a browse result or service registration result.
949 // On successful completion rep is set to point to a malloc'd reply_state struct
950 mDNSlocal mStatus GenerateNTDResponse(const domainname *const servicename, const mDNSInterfaceID id,
951 request_state *const request, reply_state **const rep, reply_op_t op, DNSServiceFlags flags, mStatus err)
952 {
953 domainlabel name;
954 domainname type, dom;
955 *rep = NULL;
956 if (!DeconstructServiceName(servicename, &name, &type, &dom))
957 return kDNSServiceErr_Invalid;
958 else
959 {
960 char namestr[MAX_DOMAIN_LABEL+1];
961 char typestr[MAX_ESCAPED_DOMAIN_NAME];
962 char domstr [MAX_ESCAPED_DOMAIN_NAME];
963 int len;
964 char *data;
965
966 ConvertDomainLabelToCString_unescaped(&name, namestr);
967 ConvertDomainNameToCString(&type, typestr);
968 ConvertDomainNameToCString(&dom, domstr);
969
970 // Calculate reply data length
971 len = sizeof(DNSServiceFlags);
972 len += sizeof(mDNSu32); // if index
973 len += sizeof(DNSServiceErrorType);
974 len += (int) (strlen(namestr) + 1);
975 len += (int) (strlen(typestr) + 1);
976 len += (int) (strlen(domstr) + 1);
977
978 // Build reply header
979 *rep = create_reply(op, len, request);
980 (*rep)->rhdr->flags = dnssd_htonl(flags);
981 (*rep)->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(&mDNSStorage, id));
982 (*rep)->rhdr->error = dnssd_htonl(err);
983
984 // Build reply body
985 data = (*rep)->sdata;
986 put_string(namestr, &data);
987 put_string(typestr, &data);
988 put_string(domstr, &data);
989
990 return mStatus_NoError;
991 }
992 }
993
994 // Returns a resource record (allocated w/ malloc) containing the data found in an IPC message
995 // Data must be in the following format: flags, interfaceIndex, name, rrtype, rrclass, rdlen, rdata, (optional) ttl
996 // (ttl only extracted/set if ttl argument is non-zero). Returns NULL for a bad-parameter error
997 mDNSlocal AuthRecord *read_rr_from_ipc_msg(request_state *request, int GetTTL, int validate_flags)
998 {
999 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
1000 mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend);
1001 char name[256];
1002 int str_err = get_string(&request->msgptr, request->msgend, name, sizeof(name));
1003 mDNSu16 type = get_uint16(&request->msgptr, request->msgend);
1004 mDNSu16 class = get_uint16(&request->msgptr, request->msgend);
1005 mDNSu16 rdlen = get_uint16(&request->msgptr, request->msgend);
1006 char *rdata = get_rdata(&request->msgptr, request->msgend, rdlen);
1007 mDNSu32 ttl = GetTTL ? get_uint32(&request->msgptr, request->msgend) : 0;
1008 int storage_size = rdlen > sizeof(RDataBody) ? rdlen : sizeof(RDataBody);
1009 AuthRecord *rr;
1010
1011 if (str_err) { LogMsg("ERROR: read_rr_from_ipc_msg - get_string"); return NULL; }
1012
1013 if (!request->msgptr) { LogMsg("Error reading Resource Record from client"); return NULL; }
1014
1015 if (validate_flags &&
1016 !((flags & kDNSServiceFlagsShared) == kDNSServiceFlagsShared) &&
1017 !((flags & kDNSServiceFlagsUnique) == kDNSServiceFlagsUnique))
1018 {
1019 LogMsg("ERROR: Bad resource record flags (must be kDNSServiceFlagsShared or kDNSServiceFlagsUnique)");
1020 return NULL;
1021 }
1022
1023 rr = mallocL("AuthRecord/read_rr_from_ipc_msg", sizeof(AuthRecord) - sizeof(RDataBody) + storage_size);
1024 if (!rr) FatalError("ERROR: malloc");
1025 mDNS_SetupResourceRecord(rr, mDNSNULL, mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex),
1026 type, 0, (mDNSu8) ((flags & kDNSServiceFlagsShared) ? kDNSRecordTypeShared : kDNSRecordTypeUnique), mDNSNULL, mDNSNULL);
1027
1028 if (!MakeDomainNameFromDNSNameString(&rr->namestorage, name))
1029 {
1030 LogMsg("ERROR: bad name: %s", name);
1031 freeL("AuthRecord/read_rr_from_ipc_msg", rr);
1032 return NULL;
1033 }
1034
1035 if (flags & kDNSServiceFlagsAllowRemoteQuery) rr->AllowRemoteQuery = mDNStrue;
1036 rr->resrec.rrclass = class;
1037 rr->resrec.rdlength = rdlen;
1038 rr->resrec.rdata->MaxRDLength = rdlen;
1039 mDNSPlatformMemCopy(rr->resrec.rdata->u.data, rdata, rdlen);
1040 if (GetTTL) rr->resrec.rroriginalttl = ttl;
1041 rr->resrec.namehash = DomainNameHashValue(rr->resrec.name);
1042 SetNewRData(&rr->resrec, mDNSNULL, 0); // Sets rr->rdatahash for us
1043 return rr;
1044 }
1045
1046 mDNSlocal int build_domainname_from_strings(domainname *srv, char *name, char *regtype, char *domain)
1047 {
1048 domainlabel n;
1049 domainname d, t;
1050
1051 if (!MakeDomainLabelFromLiteralString(&n, name)) return -1;
1052 if (!MakeDomainNameFromDNSNameString(&t, regtype)) return -1;
1053 if (!MakeDomainNameFromDNSNameString(&d, domain)) return -1;
1054 if (!ConstructServiceName(srv, &n, &t, &d)) return -1;
1055 return 0;
1056 }
1057
1058 mDNSlocal void send_all(dnssd_sock_t s, const char *ptr, int len)
1059 {
1060 int n = send(s, ptr, len, 0);
1061 // On a freshly-created Unix Domain Socket, the kernel should *never* fail to buffer a small write for us
1062 // (four bytes for a typical error code return, 12 bytes for DNSServiceGetProperty(DaemonVersion)).
1063 // If it does fail, we don't attempt to handle this failure, but we do log it so we know something is wrong.
1064 if (n < len)
1065 LogMsg("ERROR: send_all(%d) wrote %d of %d errno %d %s",
1066 s, n, len, dnssd_errno(), dnssd_strerror(dnssd_errno()));
1067 }
1068
1069 // ***************************************************************************
1070 #if COMPILER_LIKES_PRAGMA_MARK
1071 #pragma mark -
1072 #pragma mark - DNSServiceRegister
1073 #endif
1074
1075 mDNSexport void FreeExtraRR(mDNS *const m, AuthRecord *const rr, mStatus result)
1076 {
1077 ExtraResourceRecord *extra = (ExtraResourceRecord *)rr->RecordContext;
1078 (void)m; //unused
1079
1080 if (result != mStatus_MemFree) { LogMsg("Error: FreeExtraRR invoked with unexpected error %d", result); return; }
1081
1082 LogOperation(" FreeExtraRR %s", RRDisplayString(m, &rr->resrec));
1083
1084 if (rr->resrec.rdata != &rr->rdatastorage)
1085 freeL("Extra RData", rr->resrec.rdata);
1086 freeL("ExtraResourceRecord/FreeExtraRR", extra);
1087 }
1088
1089 mDNSlocal void unlink_and_free_service_instance(service_instance *srv)
1090 {
1091 ExtraResourceRecord *e = srv->srs.Extras, *tmp;
1092
1093 // clear pointers from parent struct
1094 if (srv->request)
1095 {
1096 service_instance **p = &srv->request->u.servicereg.instances;
1097 while (*p)
1098 {
1099 if (*p == srv) { *p = (*p)->next; break; }
1100 p = &(*p)->next;
1101 }
1102 }
1103
1104 while (e)
1105 {
1106 e->r.RecordContext = e;
1107 tmp = e;
1108 e = e->next;
1109 FreeExtraRR(&mDNSStorage, &tmp->r, mStatus_MemFree);
1110 }
1111
1112 if (srv->srs.RR_TXT.resrec.rdata != &srv->srs.RR_TXT.rdatastorage)
1113 freeL("TXT RData", srv->srs.RR_TXT.resrec.rdata);
1114
1115 if (srv->subtypes) { freeL("ServiceSubTypes", srv->subtypes); srv->subtypes = NULL; }
1116 freeL("service_instance", srv);
1117 }
1118
1119 // Count how many other service records we have locally with the same name, but different rdata.
1120 // For auto-named services, we can have at most one per machine -- if we allowed two auto-named services of
1121 // the same type on the same machine, we'd get into an infinite autoimmune-response loop of continuous renaming.
1122 mDNSexport int CountPeerRegistrations(mDNS *const m, ServiceRecordSet *const srs)
1123 {
1124 int count = 0;
1125 ResourceRecord *r = &srs->RR_SRV.resrec;
1126 AuthRecord *rr;
1127 ServiceRecordSet *s;
1128
1129 for (rr = m->ResourceRecords; rr; rr=rr->next)
1130 if (rr->resrec.rrtype == kDNSType_SRV && SameDomainName(rr->resrec.name, r->name) && !SameRData(&rr->resrec, r))
1131 count++;
1132
1133 for (s = m->ServiceRegistrations; s; s = s->uDNS_next)
1134 if (s->state != regState_Unregistered && SameDomainName(s->RR_SRV.resrec.name, r->name) && !SameRData(&s->RR_SRV.resrec, r))
1135 count++;
1136
1137 verbosedebugf("%d peer registrations for %##s", count, r->name->c);
1138 return(count);
1139 }
1140
1141 mDNSexport int CountExistingRegistrations(domainname *srv, mDNSIPPort port)
1142 {
1143 int count = 0;
1144 AuthRecord *rr;
1145 for (rr = mDNSStorage.ResourceRecords; rr; rr=rr->next)
1146 if (rr->resrec.rrtype == kDNSType_SRV &&
1147 mDNSSameIPPort(rr->resrec.rdata->u.srv.port, port) &&
1148 SameDomainName(rr->resrec.name, srv))
1149 count++;
1150 return(count);
1151 }
1152
1153 mDNSlocal void SendServiceRemovalNotification(ServiceRecordSet *const srs)
1154 {
1155 reply_state *rep;
1156 service_instance *instance = srs->ServiceContext;
1157 if (GenerateNTDResponse(srs->RR_SRV.resrec.name, srs->RR_SRV.resrec.InterfaceID, instance->request, &rep, reg_service_reply_op, 0, mStatus_NoError) != mStatus_NoError)
1158 LogMsg("%3d: SendServiceRemovalNotification: %##s is not valid DNS-SD SRV name", instance->sd, srs->RR_SRV.resrec.name->c);
1159 else { append_reply(instance->request, rep); instance->clientnotified = mDNSfalse; }
1160 }
1161
1162 // service registration callback performs three duties - frees memory for deregistered services,
1163 // handles name conflicts, and delivers completed registration information to the client (via
1164 // process_service_registraion())
1165 mDNSlocal void regservice_callback(mDNS *const m, ServiceRecordSet *const srs, mStatus result)
1166 {
1167 mStatus err;
1168 mDNSBool SuppressError = mDNSfalse;
1169 service_instance *instance = srs->ServiceContext;
1170 reply_state *rep;
1171 (void)m; // Unused
1172 if (!srs) { LogMsg("regservice_callback: srs is NULL %d", result); return; }
1173 if (!instance) { LogMsg("regservice_callback: srs->ServiceContext is NULL %d", result); return; }
1174
1175 // don't send errors up to client for wide-area, empty-string registrations
1176 if (instance->request &&
1177 instance->request->u.servicereg.default_domain &&
1178 !instance->default_local)
1179 SuppressError = mDNStrue;
1180
1181 if (result == mStatus_NoError)
1182 LogOperation("%3d: DNSServiceRegister(%##s, %u) REGISTERED", instance->sd, srs->RR_SRV.resrec.name->c, mDNSVal16(srs->RR_SRV.resrec.rdata->u.srv.port));
1183 else if (result == mStatus_MemFree)
1184 LogOperation("%3d: DNSServiceRegister(%##s, %u) DEREGISTERED", instance->sd, srs->RR_SRV.resrec.name->c, mDNSVal16(srs->RR_SRV.resrec.rdata->u.srv.port));
1185 else if (result == mStatus_NameConflict)
1186 LogOperation("%3d: DNSServiceRegister(%##s, %u) NAME CONFLICT", instance->sd, srs->RR_SRV.resrec.name->c, mDNSVal16(srs->RR_SRV.resrec.rdata->u.srv.port));
1187 else
1188 LogOperation("%3d: DNSServiceRegister(%##s, %u) CALLBACK %d", instance->sd, srs->RR_SRV.resrec.name->c, mDNSVal16(srs->RR_SRV.resrec.rdata->u.srv.port), result);
1189
1190 if (!instance->request && result != mStatus_MemFree) { LogMsg("regservice_callback: instance->request is NULL %d", result); return; }
1191
1192 if (result == mStatus_NoError)
1193 {
1194 if (instance->request->u.servicereg.allowremotequery)
1195 {
1196 ExtraResourceRecord *e;
1197 srs->RR_ADV.AllowRemoteQuery = mDNStrue;
1198 srs->RR_PTR.AllowRemoteQuery = mDNStrue;
1199 srs->RR_SRV.AllowRemoteQuery = mDNStrue;
1200 srs->RR_TXT.AllowRemoteQuery = mDNStrue;
1201 for (e = instance->srs.Extras; e; e = e->next) e->r.AllowRemoteQuery = mDNStrue;
1202 }
1203
1204 if (GenerateNTDResponse(srs->RR_SRV.resrec.name, srs->RR_SRV.resrec.InterfaceID, instance->request, &rep, reg_service_reply_op, kDNSServiceFlagsAdd, result) != mStatus_NoError)
1205 LogMsg("%3d: regservice_callback: %##s is not valid DNS-SD SRV name", instance->sd, srs->RR_SRV.resrec.name->c);
1206 else { append_reply(instance->request, rep); instance->clientnotified = mDNStrue; }
1207
1208 if (instance->request->u.servicereg.autoname && CountPeerRegistrations(m, srs) == 0)
1209 RecordUpdatedNiceLabel(m, 0); // Successfully got new name, tell user immediately
1210 }
1211 else if (result == mStatus_MemFree)
1212 {
1213 if (instance->request && instance->renameonmemfree)
1214 {
1215 instance->renameonmemfree = 0;
1216 err = mDNS_RenameAndReregisterService(m, srs, &instance->request->u.servicereg.name);
1217 if (err) LogMsg("ERROR: regservice_callback - RenameAndReregisterService returned %ld", err);
1218 // error should never happen - safest to log and continue
1219 }
1220 else
1221 unlink_and_free_service_instance(instance);
1222 }
1223 else if (result == mStatus_NameConflict)
1224 {
1225 if (instance->request->u.servicereg.autorename)
1226 {
1227 if (instance->request->u.servicereg.autoname && CountPeerRegistrations(m, srs) == 0)
1228 {
1229 // On conflict for an autoname service, rename and reregister *all* autoname services
1230 IncrementLabelSuffix(&m->nicelabel, mDNStrue);
1231 m->MainCallback(m, mStatus_ConfigChanged); // will call back into udsserver_handle_configchange()
1232 }
1233 else // On conflict for a non-autoname service, rename and reregister just that one service
1234 {
1235 if (instance->clientnotified) SendServiceRemovalNotification(srs);
1236 mDNS_RenameAndReregisterService(m, srs, mDNSNULL);
1237 }
1238 }
1239 else
1240 {
1241 if (!SuppressError)
1242 {
1243 if (GenerateNTDResponse(srs->RR_SRV.resrec.name, srs->RR_SRV.resrec.InterfaceID, instance->request, &rep, reg_service_reply_op, kDNSServiceFlagsAdd, result) != mStatus_NoError)
1244 LogMsg("%3d: regservice_callback: %##s is not valid DNS-SD SRV name", instance->sd, srs->RR_SRV.resrec.name->c);
1245 else { append_reply(instance->request, rep); instance->clientnotified = mDNStrue; }
1246 }
1247 unlink_and_free_service_instance(instance);
1248 }
1249 }
1250 else
1251 {
1252 if (result != mStatus_NATTraversal)
1253 LogMsg("regservice_callback: Error %d%s for %s", result, SuppressError ? " (suppressed)" : "", ARDisplayString(m, &srs->RR_SRV));
1254 if (!SuppressError)
1255 {
1256 if (GenerateNTDResponse(srs->RR_SRV.resrec.name, srs->RR_SRV.resrec.InterfaceID, instance->request, &rep, reg_service_reply_op, kDNSServiceFlagsAdd, result) != mStatus_NoError)
1257 LogMsg("%3d: regservice_callback: %##s is not valid DNS-SD SRV name", instance->sd, srs->RR_SRV.resrec.name->c);
1258 else { append_reply(instance->request, rep); instance->clientnotified = mDNStrue; }
1259 }
1260 unlink_and_free_service_instance(instance);
1261 }
1262 }
1263
1264 mDNSlocal void regrecord_callback(mDNS *const m, AuthRecord *rr, mStatus result)
1265 {
1266 (void)m; // Unused
1267 if (!rr->RecordContext) // parent struct already freed by termination callback
1268 {
1269 if (result == mStatus_NoError)
1270 LogMsg("Error: regrecord_callback: successful registration of orphaned record");
1271 else
1272 {
1273 if (result != mStatus_MemFree) LogMsg("regrecord_callback: error %d received after parent termination", result);
1274 freeL("AuthRecord/regrecord_callback", rr);
1275 }
1276 }
1277 else
1278 {
1279 registered_record_entry *re = rr->RecordContext;
1280 request_state *request = re->request;
1281 int len = sizeof(DNSServiceFlags) + sizeof(mDNSu32) + sizeof(DNSServiceErrorType);
1282 reply_state *reply = create_reply(reg_record_reply_op, len, request);
1283 reply->mhdr->client_context = re->client_context;
1284 reply->rhdr->flags = dnssd_htonl(0);
1285 reply->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(m, rr->resrec.InterfaceID));
1286 reply->rhdr->error = dnssd_htonl(result);
1287
1288 LogOperation("%3d: DNSServiceRegisterRecord(%u) result %d", request->sd, request->hdr.reg_index, result);
1289 if (result)
1290 {
1291 // unlink from list, free memory
1292 registered_record_entry **ptr = &request->u.reg_recs;
1293 while (*ptr && (*ptr) != re) ptr = &(*ptr)->next;
1294 if (!*ptr) { LogMsg("regrecord_callback - record not in list!"); return; }
1295 *ptr = (*ptr)->next;
1296 freeL("registered_record_entry AuthRecord regrecord_callback", re->rr);
1297 freeL("registered_record_entry regrecord_callback", re);
1298 }
1299 append_reply(request, reply);
1300 }
1301 }
1302
1303 mDNSlocal void connection_termination(request_state *request)
1304 {
1305 request_state **req = &all_requests;
1306 while (*req)
1307 {
1308 if ((*req)->primary == request)
1309 {
1310 // Since we're already doing a list traversal, we unlink the request directly instead of using AbortUnlinkAndFree()
1311 request_state *tmp = *req;
1312 abort_request(tmp);
1313 *req = tmp->next;
1314 freeL("request_state/connection_termination", tmp);
1315 }
1316 else
1317 req = &(*req)->next;
1318 }
1319
1320 while (request->u.reg_recs)
1321 {
1322 registered_record_entry *ptr = request->u.reg_recs;
1323 request->u.reg_recs = request->u.reg_recs->next;
1324 ptr->rr->RecordContext = NULL;
1325 mDNS_Deregister(&mDNSStorage, ptr->rr); // Will free ptr->rr for us
1326 freeL("registered_record_entry/connection_termination", ptr);
1327 }
1328 }
1329
1330 mDNSlocal void handle_cancel_request(request_state *request)
1331 {
1332 request_state **req = &all_requests;
1333 LogOperation("%3d: Cancel %X%08X", request->sd, request->hdr.client_context.u32[1], request->hdr.client_context.u32[0]);
1334 while (*req)
1335 {
1336 if ((*req)->primary == request &&
1337 (*req)->hdr.client_context.u32[0] == request->hdr.client_context.u32[0] &&
1338 (*req)->hdr.client_context.u32[1] == request->hdr.client_context.u32[1])
1339 {
1340 // Since we're already doing a list traversal, we unlink the request directly instead of using AbortUnlinkAndFree()
1341 request_state *tmp = *req;
1342 abort_request(tmp);
1343 *req = tmp->next;
1344 freeL("request_state/handle_cancel_request", tmp);
1345 }
1346 else
1347 req = &(*req)->next;
1348 }
1349 }
1350
1351 mDNSlocal mStatus handle_regrecord_request(request_state *request)
1352 {
1353 mStatus err = mStatus_BadParamErr;
1354 AuthRecord *rr = read_rr_from_ipc_msg(request, 1, 1);
1355 if (rr)
1356 {
1357 // allocate registration entry, link into list
1358 registered_record_entry *re = mallocL("registered_record_entry", sizeof(registered_record_entry));
1359 if (!re) FatalError("ERROR: malloc");
1360 re->key = request->hdr.reg_index;
1361 re->rr = rr;
1362 re->request = request;
1363 re->client_context = request->hdr.client_context;
1364 rr->RecordContext = re;
1365 rr->RecordCallback = regrecord_callback;
1366 re->next = request->u.reg_recs;
1367 request->u.reg_recs = re;
1368
1369 if (rr->resrec.rroriginalttl == 0)
1370 rr->resrec.rroriginalttl = DefaultTTLforRRType(rr->resrec.rrtype);
1371
1372 LogOperation("%3d: DNSServiceRegisterRecord(%u %s)", request->sd, request->hdr.reg_index, RRDisplayString(&mDNSStorage, &rr->resrec));
1373 err = mDNS_Register(&mDNSStorage, rr);
1374 }
1375 return(err);
1376 }
1377
1378 mDNSlocal mStatus add_record_to_service(request_state *request, service_instance *instance, mDNSu16 rrtype, mDNSu16 rdlen, char *rdata, mDNSu32 ttl)
1379 {
1380 ServiceRecordSet *srs = &instance->srs;
1381 mStatus result;
1382 int size = rdlen > sizeof(RDataBody) ? rdlen : sizeof(RDataBody);
1383 ExtraResourceRecord *extra = mallocL("ExtraResourceRecord", sizeof(*extra) - sizeof(RDataBody) + size);
1384 if (!extra) { my_perror("ERROR: malloc"); return mStatus_NoMemoryErr; }
1385
1386 mDNSPlatformMemZero(extra, sizeof(ExtraResourceRecord)); // OK if oversized rdata not zero'd
1387 extra->r.resrec.rrtype = rrtype;
1388 extra->r.rdatastorage.MaxRDLength = (mDNSu16) size;
1389 extra->r.resrec.rdlength = rdlen;
1390 mDNSPlatformMemCopy(&extra->r.rdatastorage.u.data, rdata, rdlen);
1391
1392 result = mDNS_AddRecordToService(&mDNSStorage, srs, extra, &extra->r.rdatastorage, ttl);
1393 if (result) { freeL("ExtraResourceRecord/add_record_to_service", extra); return result; }
1394
1395 extra->ClientID = request->hdr.reg_index;
1396 return result;
1397 }
1398
1399 mDNSlocal mStatus handle_add_request(request_state *request)
1400 {
1401 service_instance *i;
1402 mStatus result = mStatus_UnknownErr;
1403 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
1404 mDNSu16 rrtype = get_uint16(&request->msgptr, request->msgend);
1405 mDNSu16 rdlen = get_uint16(&request->msgptr, request->msgend);
1406 char *rdata = get_rdata(&request->msgptr, request->msgend, rdlen);
1407 mDNSu32 ttl = get_uint32(&request->msgptr, request->msgend);
1408 if (!ttl) ttl = DefaultTTLforRRType(rrtype);
1409 (void)flags; // Unused
1410
1411 if (!request->msgptr) { LogMsg("%3d: DNSServiceAddRecord(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
1412
1413 LogOperation("%3d: DNSServiceAddRecord(%##s, %s, %d)", request->sd,
1414 (request->u.servicereg.instances) ? request->u.servicereg.instances->srs.RR_SRV.resrec.name->c : NULL, DNSTypeName(rrtype), rdlen);
1415
1416 for (i = request->u.servicereg.instances; i; i = i->next)
1417 {
1418 result = add_record_to_service(request, i, rrtype, rdlen, rdata, ttl);
1419 if (result && i->default_local) break;
1420 else result = mStatus_NoError; // suppress non-local default errors
1421 }
1422
1423 return(result);
1424 }
1425
1426 mDNSlocal void update_callback(mDNS *const m, AuthRecord *const rr, RData *oldrd)
1427 {
1428 (void)m; // Unused
1429 if (oldrd != &rr->rdatastorage) freeL("RData/update_callback", oldrd);
1430 }
1431
1432 mDNSlocal mStatus update_record(AuthRecord *rr, mDNSu16 rdlen, char *rdata, mDNSu32 ttl)
1433 {
1434 int rdsize;
1435 RData *newrd;
1436 mStatus result;
1437
1438 if (rdlen > sizeof(RDataBody)) rdsize = rdlen;
1439 else rdsize = sizeof(RDataBody);
1440 newrd = mallocL("RData/update_record", sizeof(RData) - sizeof(RDataBody) + rdsize);
1441 if (!newrd) FatalError("ERROR: malloc");
1442 newrd->MaxRDLength = (mDNSu16) rdsize;
1443 mDNSPlatformMemCopy(&newrd->u, rdata, rdlen);
1444
1445 // BIND named (name daemon) doesn't allow TXT records with zero-length rdata. This is strictly speaking correct,
1446 // since RFC 1035 specifies a TXT record as "One or more <character-string>s", not "Zero or more <character-string>s".
1447 // Since some legacy apps try to create zero-length TXT records, we'll silently correct it here.
1448 if (rr->resrec.rrtype == kDNSType_TXT && rdlen == 0) { rdlen = 1; newrd->u.txt.c[0] = 0; }
1449
1450 result = mDNS_Update(&mDNSStorage, rr, ttl, rdlen, newrd, update_callback);
1451 if (result) { LogMsg("ERROR: mDNS_Update - %ld", result); freeL("RData/update_record", newrd); }
1452 return result;
1453 }
1454
1455 mDNSlocal mStatus handle_update_request(request_state *request)
1456 {
1457 mStatus result = mStatus_BadReferenceErr;
1458 service_instance *i;
1459 AuthRecord *rr = NULL;
1460
1461 // get the message data
1462 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); // flags unused
1463 mDNSu16 rdlen = get_uint16(&request->msgptr, request->msgend);
1464 char *rdata = get_rdata(&request->msgptr, request->msgend, rdlen);
1465 mDNSu32 ttl = get_uint32(&request->msgptr, request->msgend);
1466 (void)flags; // Unused
1467
1468 if (!request->msgptr) { LogMsg("%3d: DNSServiceUpdateRecord(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
1469
1470 if (request->terminate == connection_termination)
1471 {
1472 // update an individually registered record
1473 registered_record_entry *reptr;
1474 for (reptr = request->u.reg_recs; reptr; reptr = reptr->next)
1475 {
1476 if (reptr->key == request->hdr.reg_index)
1477 {
1478 result = update_record(reptr->rr, rdlen, rdata, ttl);
1479 goto end;
1480 }
1481 }
1482 result = mStatus_BadReferenceErr;
1483 goto end;
1484 }
1485
1486 // update a record from a service record set
1487 for (i = request->u.servicereg.instances; i; i = i->next)
1488 {
1489 if (request->hdr.reg_index == TXT_RECORD_INDEX) rr = &i->srs.RR_TXT;
1490 else
1491 {
1492 ExtraResourceRecord *e;
1493 for (e = i->srs.Extras; e; e = e->next)
1494 if (e->ClientID == request->hdr.reg_index) { rr = &e->r; break; }
1495 }
1496
1497 if (!rr) { result = mStatus_BadReferenceErr; goto end; }
1498 result = update_record(rr, rdlen, rdata, ttl);
1499 if (result && i->default_local) goto end;
1500 else result = mStatus_NoError; // suppress non-local default errors
1501 }
1502
1503 end:
1504 LogOperation("%3d: DNSServiceUpdateRecord(%##s, %s)", request->sd,
1505 (request->u.servicereg.instances) ? request->u.servicereg.instances->srs.RR_SRV.resrec.name->c : NULL,
1506 rr ? DNSTypeName(rr->resrec.rrtype) : "<NONE>");
1507
1508 return(result);
1509 }
1510
1511 // remove a resource record registered via DNSServiceRegisterRecord()
1512 mDNSlocal mStatus remove_record(request_state *request)
1513 {
1514 mStatus err = mStatus_UnknownErr;
1515 registered_record_entry *e, **ptr = &request->u.reg_recs;
1516
1517 while (*ptr && (*ptr)->key != request->hdr.reg_index) ptr = &(*ptr)->next;
1518 if (!*ptr) { LogMsg("%3d: DNSServiceRemoveRecord(%u) not found", request->sd, request->hdr.reg_index); return mStatus_BadReferenceErr; }
1519 e = *ptr;
1520 *ptr = e->next; // unlink
1521
1522 LogOperation("%3d: DNSServiceRemoveRecord(%u %s)", request->sd, request->hdr.reg_index, RRDisplayString(&mDNSStorage, &e->rr->resrec));
1523 e->rr->RecordContext = NULL;
1524 err = mDNS_Deregister(&mDNSStorage, e->rr);
1525 if (err)
1526 {
1527 LogMsg("ERROR: remove_record, mDNS_Deregister: %ld", err);
1528 freeL("registered_record_entry AuthRecord remove_record", e->rr);
1529 }
1530 freeL("registered_record_entry remove_record", e);
1531 return err;
1532 }
1533
1534 mDNSlocal mStatus remove_extra(const request_state *const request, service_instance *const serv, mDNSu16 *const rrtype)
1535 {
1536 mStatus err = mStatus_BadReferenceErr;
1537 ExtraResourceRecord *ptr;
1538
1539 for (ptr = serv->srs.Extras; ptr; ptr = ptr->next)
1540 {
1541 if (ptr->ClientID == request->hdr.reg_index) // found match
1542 {
1543 *rrtype = ptr->r.resrec.rrtype;
1544 return mDNS_RemoveRecordFromService(&mDNSStorage, &serv->srs, ptr, FreeExtraRR, ptr);
1545 }
1546 }
1547 return err;
1548 }
1549
1550 mDNSlocal mStatus handle_removerecord_request(request_state *request)
1551 {
1552 mStatus err = mStatus_BadReferenceErr;
1553 get_flags(&request->msgptr, request->msgend); // flags unused
1554
1555 if (!request->msgptr) { LogMsg("%3d: DNSServiceRemoveRecord(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
1556
1557 if (request->terminate == connection_termination)
1558 err = remove_record(request); // remove individually registered record
1559 else
1560 {
1561 service_instance *i;
1562 mDNSu16 rrtype = 0;
1563 LogOperation("%3d: DNSServiceRemoveRecord(%##s, %s)", request->sd,
1564 (request->u.servicereg.instances) ? request->u.servicereg.instances->srs.RR_SRV.resrec.name->c : NULL,
1565 rrtype ? DNSTypeName(rrtype) : "<NONE>");
1566 for (i = request->u.servicereg.instances; i; i = i->next)
1567 {
1568 err = remove_extra(request, i, &rrtype);
1569 if (err && i->default_local) break;
1570 else err = mStatus_NoError; // suppress non-local default errors
1571 }
1572 }
1573
1574 return(err);
1575 }
1576
1577 // If there's a comma followed by another character,
1578 // FindFirstSubType overwrites the comma with a nul and returns the pointer to the next character.
1579 // Otherwise, it returns a pointer to the final nul at the end of the string
1580 mDNSlocal char *FindFirstSubType(char *p)
1581 {
1582 while (*p)
1583 {
1584 if (p[0] == '\\' && p[1]) p += 2;
1585 else if (p[0] == ',' && p[1]) { *p++ = 0; return(p); }
1586 else p++;
1587 }
1588 return(p);
1589 }
1590
1591 // If there's a comma followed by another character,
1592 // FindNextSubType overwrites the comma with a nul and returns the pointer to the next character.
1593 // If it finds an illegal unescaped dot in the subtype name, it returns mDNSNULL
1594 // Otherwise, it returns a pointer to the final nul at the end of the string
1595 mDNSlocal char *FindNextSubType(char *p)
1596 {
1597 while (*p)
1598 {
1599 if (p[0] == '\\' && p[1]) // If escape character
1600 p += 2; // ignore following character
1601 else if (p[0] == ',') // If we found a comma
1602 {
1603 if (p[1]) *p++ = 0;
1604 return(p);
1605 }
1606 else if (p[0] == '.')
1607 return(mDNSNULL);
1608 else p++;
1609 }
1610 return(p);
1611 }
1612
1613 // Returns -1 if illegal subtype found
1614 mDNSexport mDNSs32 ChopSubTypes(char *regtype)
1615 {
1616 mDNSs32 NumSubTypes = 0;
1617 char *stp = FindFirstSubType(regtype);
1618 while (stp && *stp) // If we found a comma...
1619 {
1620 if (*stp == ',') return(-1);
1621 NumSubTypes++;
1622 stp = FindNextSubType(stp);
1623 }
1624 if (!stp) return(-1);
1625 return(NumSubTypes);
1626 }
1627
1628 mDNSexport AuthRecord *AllocateSubTypes(mDNSs32 NumSubTypes, char *p)
1629 {
1630 AuthRecord *st = mDNSNULL;
1631 if (NumSubTypes)
1632 {
1633 mDNSs32 i;
1634 st = mallocL("ServiceSubTypes", NumSubTypes * sizeof(AuthRecord));
1635 if (!st) return(mDNSNULL);
1636 for (i = 0; i < NumSubTypes; i++)
1637 {
1638 mDNS_SetupResourceRecord(&st[i], mDNSNULL, mDNSInterface_Any, kDNSQType_ANY, kStandardTTL, 0, mDNSNULL, mDNSNULL);
1639 while (*p) p++;
1640 p++;
1641 if (!MakeDomainNameFromDNSNameString(&st[i].namestorage, p))
1642 { freeL("ServiceSubTypes", st); return(mDNSNULL); }
1643 }
1644 }
1645 return(st);
1646 }
1647
1648 mDNSlocal mStatus register_service_instance(request_state *request, const domainname *domain)
1649 {
1650 service_instance **ptr, *instance;
1651 int instance_size;
1652 mStatus result;
1653
1654 for (ptr = &request->u.servicereg.instances; *ptr; ptr = &(*ptr)->next)
1655 {
1656 if (SameDomainName(&(*ptr)->domain, domain))
1657 { LogMsg("register_service_instance: domain %##s already registered", domain->c); return mStatus_AlreadyRegistered; }
1658 }
1659
1660 // Special-case hack: We don't advertise SMB service in AutoTunnel domains, because AutoTunnel services have to support IPv6, and our SMB server does not
1661 // <rdar://problem/5482322> BTMM: Don't advertise SMB with BTMM because it doesn't support IPv6
1662 if (SameDomainName(&request->u.servicereg.type, (const domainname *) "\x4" "_smb" "\x4" "_tcp"))
1663 {
1664 DomainAuthInfo *AuthInfo = GetAuthInfoForName(&mDNSStorage, domain);
1665 if (AuthInfo && AuthInfo->AutoTunnel) return(kDNSServiceErr_Unsupported);
1666 }
1667
1668 instance_size = sizeof(*instance);
1669 if (request->u.servicereg.txtlen > sizeof(RDataBody)) instance_size += (request->u.servicereg.txtlen - sizeof(RDataBody));
1670 instance = mallocL("service_instance", instance_size);
1671 if (!instance) { my_perror("ERROR: malloc"); return mStatus_NoMemoryErr; }
1672
1673 instance->next = mDNSNULL;
1674 instance->request = request;
1675 instance->sd = request->sd;
1676 instance->subtypes = AllocateSubTypes(request->u.servicereg.num_subtypes, request->u.servicereg.type_as_string);
1677 instance->renameonmemfree = 0;
1678 instance->clientnotified = mDNSfalse;
1679 instance->default_local = (request->u.servicereg.default_domain && SameDomainName(domain, &localdomain));
1680 AssignDomainName(&instance->domain, domain);
1681
1682 if (request->u.servicereg.num_subtypes && !instance->subtypes)
1683 { unlink_and_free_service_instance(instance); instance = NULL; FatalError("ERROR: malloc"); }
1684
1685 LogOperation("%3d: DNSServiceRegister(%#s.%##s%##s, %u) ADDING",
1686 instance->sd, &request->u.servicereg.name, &request->u.servicereg.type, domain->c, mDNSVal16(request->u.servicereg.port));
1687
1688 result = mDNS_RegisterService(&mDNSStorage, &instance->srs,
1689 &request->u.servicereg.name, &request->u.servicereg.type, domain,
1690 request->u.servicereg.host.c[0] ? &request->u.servicereg.host : NULL,
1691 request->u.servicereg.port,
1692 request->u.servicereg.txtdata, request->u.servicereg.txtlen,
1693 instance->subtypes, request->u.servicereg.num_subtypes,
1694 request->u.servicereg.InterfaceID, regservice_callback, instance);
1695
1696 if (!result) *ptr = instance; // Append this to the end of our request->u.servicereg.instances list
1697 else
1698 {
1699 LogMsg("register_service_instance %#s.%##s%##s error %d",
1700 &request->u.servicereg.name, &request->u.servicereg.type, domain->c, result);
1701 unlink_and_free_service_instance(instance);
1702 }
1703
1704 return result;
1705 }
1706
1707 mDNSlocal void UpdateDeviceInfoRecord(mDNS *const m);
1708
1709 mDNSlocal void regservice_termination_callback(request_state *request)
1710 {
1711 if (!request) { LogMsg("regservice_termination_callback context is NULL"); return; }
1712 while (request->u.servicereg.instances)
1713 {
1714 service_instance *p = request->u.servicereg.instances;
1715 request->u.servicereg.instances = request->u.servicereg.instances->next;
1716 // only safe to free memory if registration is not valid, i.e. deregister fails (which invalidates p)
1717 LogOperation("%3d: DNSServiceRegister(%##s, %u) STOP",
1718 request->sd, p->srs.RR_SRV.resrec.name->c, mDNSVal16(p->srs.RR_SRV.resrec.rdata->u.srv.port));
1719
1720 // Clear backpointer *before* calling mDNS_DeregisterService/unlink_and_free_service_instance
1721 // We don't need unlink_and_free_service_instance to cut its element from the list, because we're already advancing
1722 // request->u.servicereg.instances as we work our way through the list, implicitly cutting one element at a time
1723 // We can't clear p->request *after* the calling mDNS_DeregisterService/unlink_and_free_service_instance
1724 // because by then we might have already freed p
1725 p->request = NULL;
1726 if (mDNS_DeregisterService(&mDNSStorage, &p->srs)) unlink_and_free_service_instance(p);
1727 // Don't touch service_instance *p after this -- it's likely to have been freed already
1728 }
1729 if (request->u.servicereg.txtdata)
1730 { freeL("service_info txtdata", request->u.servicereg.txtdata); request->u.servicereg.txtdata = NULL; }
1731 if (request->u.servicereg.autoname) UpdateDeviceInfoRecord(&mDNSStorage);
1732 }
1733
1734 mDNSlocal void udsserver_default_reg_domain_changed(const DNameListElem *const d, const mDNSBool add)
1735 {
1736 request_state *request;
1737
1738 #if APPLE_OSX_mDNSResponder
1739 machserver_automatic_registration_domain_changed(&d->name, add);
1740 #endif // APPLE_OSX_mDNSResponder
1741
1742 LogMsg("%s registration domain %##s", add ? "Adding" : "Removing", d->name.c);
1743 for (request = all_requests; request; request = request->next)
1744 {
1745 if (request->terminate != regservice_termination_callback) continue;
1746 if (!request->u.servicereg.default_domain) continue;
1747 if (!d->uid || SystemUID(request->uid) || request->uid == d->uid)
1748 {
1749 service_instance **ptr = &request->u.servicereg.instances;
1750 while (*ptr && !SameDomainName(&(*ptr)->domain, &d->name)) ptr = &(*ptr)->next;
1751 if (add)
1752 {
1753 // If we don't already have this domain in our list for this registration, add it now
1754 if (!*ptr) register_service_instance(request, &d->name);
1755 else debugf("udsserver_default_reg_domain_changed %##s already in list, not re-adding", &d->name);
1756 }
1757 else
1758 {
1759 // Normally we should not fail to find the specified instance
1760 // One case where this can happen is if a uDNS update fails for some reason,
1761 // and regservice_callback then calls unlink_and_free_service_instance and disposes of that instance.
1762 if (!*ptr)
1763 LogMsg("udsserver_default_reg_domain_changed domain %##s not found for service %#s type %s",
1764 &d->name, request->u.servicereg.name.c, request->u.servicereg.type_as_string);
1765 else
1766 {
1767 DNameListElem *p;
1768 for (p = AutoRegistrationDomains; p; p=p->next)
1769 if (!p->uid || SystemUID(request->uid) || request->uid == p->uid)
1770 if (SameDomainName(&d->name, &p->name)) break;
1771 if (p) debugf("udsserver_default_reg_domain_changed %##s still in list, not removing", &d->name);
1772 else
1773 {
1774 mStatus err;
1775 service_instance *si = *ptr;
1776 *ptr = si->next;
1777 if (si->clientnotified) SendServiceRemovalNotification(&si->srs); // Do this *before* clearing si->request backpointer
1778 // Now that we've cut this service_instance from the list, we MUST clear the si->request backpointer.
1779 // Otherwise what can happen is this: While our mDNS_DeregisterService is in the
1780 // process of completing asynchronously, the client cancels the entire operation, so
1781 // regservice_termination_callback then runs through the whole list deregistering each
1782 // instance, clearing the backpointers, and then disposing the parent request_state object.
1783 // However, because this service_instance isn't in the list any more, regservice_termination_callback
1784 // has no way to find it and clear its backpointer, and then when our mDNS_DeregisterService finally
1785 // completes later with a mStatus_MemFree message, it calls unlink_and_free_service_instance() with
1786 // a service_instance with a stale si->request backpointer pointing to memory that's already been freed.
1787 si->request = NULL;
1788 err = mDNS_DeregisterService(&mDNSStorage, &si->srs);
1789 if (err) { LogMsg("udsserver_default_reg_domain_changed err %d", err); unlink_and_free_service_instance(si); }
1790 }
1791 }
1792 }
1793 }
1794 }
1795 }
1796
1797 mDNSlocal mStatus handle_regservice_request(request_state *request)
1798 {
1799 char name[256]; // Lots of spare space for extra-long names that we'll auto-truncate down to 63 bytes
1800 char domain[MAX_ESCAPED_DOMAIN_NAME], host[MAX_ESCAPED_DOMAIN_NAME];
1801 char type_as_string[MAX_ESCAPED_DOMAIN_NAME];
1802 domainname d, srv;
1803 mStatus err;
1804
1805 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
1806 mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend);
1807 mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex);
1808 if (interfaceIndex && !InterfaceID)
1809 { LogMsg("ERROR: handle_regservice_request - Couldn't find interfaceIndex %d", interfaceIndex); return(mStatus_BadParamErr); }
1810
1811 if (get_string(&request->msgptr, request->msgend, name, sizeof(name)) < 0 ||
1812 get_string(&request->msgptr, request->msgend, type_as_string, MAX_ESCAPED_DOMAIN_NAME) < 0 ||
1813 get_string(&request->msgptr, request->msgend, domain, MAX_ESCAPED_DOMAIN_NAME) < 0 ||
1814 get_string(&request->msgptr, request->msgend, host, MAX_ESCAPED_DOMAIN_NAME) < 0)
1815 { LogMsg("ERROR: handle_regservice_request - Couldn't read name/regtype/domain"); return(mStatus_BadParamErr); }
1816
1817 request->u.servicereg.InterfaceID = InterfaceID;
1818 request->u.servicereg.instances = NULL;
1819 request->u.servicereg.txtlen = 0;
1820 request->u.servicereg.txtdata = NULL;
1821 mDNSPlatformStrCopy(request->u.servicereg.type_as_string, type_as_string);
1822
1823 if (request->msgptr + 2 > request->msgend) request->msgptr = NULL;
1824 else
1825 {
1826 request->u.servicereg.port.b[0] = *request->msgptr++;
1827 request->u.servicereg.port.b[1] = *request->msgptr++;
1828 }
1829
1830 request->u.servicereg.txtlen = get_uint16(&request->msgptr, request->msgend);
1831 if (request->u.servicereg.txtlen)
1832 {
1833 request->u.servicereg.txtdata = mallocL("service_info txtdata", request->u.servicereg.txtlen);
1834 if (!request->u.servicereg.txtdata) FatalError("ERROR: handle_regservice_request - malloc");
1835 mDNSPlatformMemCopy(request->u.servicereg.txtdata, get_rdata(&request->msgptr, request->msgend, request->u.servicereg.txtlen), request->u.servicereg.txtlen);
1836 }
1837 else request->u.servicereg.txtdata = NULL;
1838
1839 if (!request->msgptr) { LogMsg("%3d: DNSServiceRegister(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
1840
1841 // Check for sub-types after the service type
1842 request->u.servicereg.num_subtypes = ChopSubTypes(request->u.servicereg.type_as_string); // Note: Modifies regtype string to remove trailing subtypes
1843 if (request->u.servicereg.num_subtypes < 0)
1844 { LogMsg("ERROR: handle_regservice_request - ChopSubTypes failed %s", request->u.servicereg.type_as_string); return(mStatus_BadParamErr); }
1845
1846 // Don't try to construct "domainname t" until *after* ChopSubTypes has worked its magic
1847 if (!*request->u.servicereg.type_as_string || !MakeDomainNameFromDNSNameString(&request->u.servicereg.type, request->u.servicereg.type_as_string))
1848 { LogMsg("ERROR: handle_regservice_request - type_as_string bad %s", request->u.servicereg.type_as_string); return(mStatus_BadParamErr); }
1849
1850 if (!name[0])
1851 {
1852 request->u.servicereg.name = mDNSStorage.nicelabel;
1853 request->u.servicereg.autoname = mDNStrue;
1854 }
1855 else
1856 {
1857 // If the client is allowing AutoRename, then truncate name to legal length before converting it to a DomainLabel
1858 if ((flags & kDNSServiceFlagsNoAutoRename) == 0)
1859 {
1860 int newlen = TruncateUTF8ToLength((mDNSu8*)name, mDNSPlatformStrLen(name), MAX_DOMAIN_LABEL);
1861 name[newlen] = 0;
1862 }
1863 if (!MakeDomainLabelFromLiteralString(&request->u.servicereg.name, name))
1864 { LogMsg("ERROR: handle_regservice_request - name bad %s", name); return(mStatus_BadParamErr); }
1865 request->u.servicereg.autoname = mDNSfalse;
1866 }
1867
1868 if (*domain)
1869 {
1870 request->u.servicereg.default_domain = mDNSfalse;
1871 if (!MakeDomainNameFromDNSNameString(&d, domain))
1872 { LogMsg("ERROR: handle_regservice_request - domain bad %s", domain); return(mStatus_BadParamErr); }
1873 }
1874 else
1875 {
1876 request->u.servicereg.default_domain = mDNStrue;
1877 MakeDomainNameFromDNSNameString(&d, "local.");
1878 }
1879
1880 if (!ConstructServiceName(&srv, &request->u.servicereg.name, &request->u.servicereg.type, &d))
1881 {
1882 LogMsg("ERROR: handle_regservice_request - Couldn't ConstructServiceName from, “%#s” “%##s” “%##s”",
1883 request->u.servicereg.name.c, request->u.servicereg.type.c, d.c); return(mStatus_BadParamErr);
1884 }
1885
1886 if (!MakeDomainNameFromDNSNameString(&request->u.servicereg.host, host))
1887 { LogMsg("ERROR: handle_regservice_request - host bad %s", host); return(mStatus_BadParamErr); }
1888 request->u.servicereg.autorename = (flags & kDNSServiceFlagsNoAutoRename ) == 0;
1889 request->u.servicereg.allowremotequery = (flags & kDNSServiceFlagsAllowRemoteQuery) != 0;
1890
1891 // Some clients use mDNS for lightweight copy protection, registering a pseudo-service with
1892 // a port number of zero. When two instances of the protected client are allowed to run on one
1893 // machine, we don't want to see misleading "Bogus client" messages in syslog and the console.
1894 if (!mDNSIPPortIsZero(request->u.servicereg.port))
1895 {
1896 int count = CountExistingRegistrations(&srv, request->u.servicereg.port);
1897 if (count)
1898 LogMsg("Client application registered %d identical instances of service %##s port %u.",
1899 count+1, srv.c, mDNSVal16(request->u.servicereg.port));
1900 }
1901
1902 LogOperation("%3d: DNSServiceRegister(\"%s\", \"%s\", \"%s\", \"%s\", %u) START",
1903 request->sd, name, request->u.servicereg.type_as_string, domain, host, mDNSVal16(request->u.servicereg.port));
1904 err = register_service_instance(request, &d);
1905
1906 // Set request->terminate first, before adding additional service instances, because the
1907 // uds_validatelists uses the request->terminate function pointer to determine what kind
1908 // of request this is, and therefore what kind of list validation is required.
1909 if (!err)
1910 {
1911 request->terminate = regservice_termination_callback;
1912 if (request->u.servicereg.autoname) UpdateDeviceInfoRecord(&mDNSStorage);
1913 }
1914
1915 if (!err && !*domain)
1916 {
1917 DNameListElem *ptr;
1918 // note that we don't report errors for non-local, non-explicit domains
1919 for (ptr = AutoRegistrationDomains; ptr; ptr = ptr->next)
1920 if (!ptr->uid || SystemUID(request->uid) || request->uid == ptr->uid)
1921 register_service_instance(request, &ptr->name);
1922 }
1923
1924 return(err);
1925 }
1926
1927 // ***************************************************************************
1928 #if COMPILER_LIKES_PRAGMA_MARK
1929 #pragma mark -
1930 #pragma mark - DNSServiceBrowse
1931 #endif
1932
1933 mDNSlocal void FoundInstance(mDNS *const m, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord)
1934 {
1935 const DNSServiceFlags flags = AddRecord ? kDNSServiceFlagsAdd : 0;
1936 request_state *req = question->QuestionContext;
1937 reply_state *rep;
1938 (void)m; // Unused
1939
1940 if (answer->rrtype != kDNSType_PTR)
1941 { LogMsg("%3d: FoundInstance: Should not be called with rrtype %d (not a PTR record)", req->sd, answer->rrtype); return; }
1942
1943 if (GenerateNTDResponse(&answer->rdata->u.name, answer->InterfaceID, req, &rep, browse_reply_op, flags, mStatus_NoError) != mStatus_NoError)
1944 {
1945 LogMsg("%3d: FoundInstance: %##s PTR %##s received from network is not valid DNS-SD service pointer",
1946 req->sd, answer->name->c, answer->rdata->u.name.c);
1947 return;
1948 }
1949
1950 LogOperation("%3d: DNSServiceBrowse(%##s, %s) RESULT %s %d: %s",
1951 req->sd, question->qname.c, DNSTypeName(question->qtype), AddRecord ? "Add" : "Rmv",
1952 mDNSPlatformInterfaceIndexfromInterfaceID(m, answer->InterfaceID), RRDisplayString(m, answer));
1953
1954 append_reply(req, rep);
1955 }
1956
1957 mDNSlocal mStatus add_domain_to_browser(request_state *info, const domainname *d)
1958 {
1959 browser_t *b, *p;
1960 mStatus err;
1961
1962 for (p = info->u.browser.browsers; p; p = p->next)
1963 {
1964 if (SameDomainName(&p->domain, d))
1965 { debugf("add_domain_to_browser %##s already in list", d->c); return mStatus_AlreadyRegistered; }
1966 }
1967
1968 b = mallocL("browser_t", sizeof(*b));
1969 if (!b) return mStatus_NoMemoryErr;
1970 AssignDomainName(&b->domain, d);
1971 err = mDNS_StartBrowse(&mDNSStorage, &b->q,
1972 &info->u.browser.regtype, d, info->u.browser.interface_id, info->u.browser.ForceMCast, FoundInstance, info);
1973 if (err)
1974 {
1975 LogMsg("mDNS_StartBrowse returned %d for type %##s domain %##s", err, info->u.browser.regtype.c, d->c);
1976 freeL("browser_t/add_domain_to_browser", b);
1977 }
1978 else
1979 {
1980 b->next = info->u.browser.browsers;
1981 info->u.browser.browsers = b;
1982 }
1983 return err;
1984 }
1985
1986 mDNSlocal void browse_termination_callback(request_state *info)
1987 {
1988 while (info->u.browser.browsers)
1989 {
1990 browser_t *ptr = info->u.browser.browsers;
1991 info->u.browser.browsers = ptr->next;
1992 LogOperation("%3d: DNSServiceBrowse(%##s) STOP", info->sd, ptr->q.qname.c);
1993 mDNS_StopBrowse(&mDNSStorage, &ptr->q); // no need to error-check result
1994 freeL("browser_t/browse_termination_callback", ptr);
1995 }
1996 }
1997
1998 mDNSlocal void udsserver_automatic_browse_domain_changed(const DNameListElem *const d, const mDNSBool add)
1999 {
2000 request_state *request;
2001 debugf("udsserver_automatic_browse_domain_changed: %s default browse domain %##s", add ? "Adding" : "Removing", d->name.c);
2002
2003 #if APPLE_OSX_mDNSResponder
2004 machserver_automatic_browse_domain_changed(&d->name, add);
2005 #endif // APPLE_OSX_mDNSResponder
2006
2007 for (request = all_requests; request; request = request->next)
2008 {
2009 if (request->terminate != browse_termination_callback) continue; // Not a browse operation
2010 if (!request->u.browser.default_domain) continue; // Not an auto-browse operation
2011 if (!d->uid || SystemUID(request->uid) || request->uid == d->uid)
2012 {
2013 browser_t **ptr = &request->u.browser.browsers;
2014 while (*ptr && !SameDomainName(&(*ptr)->domain, &d->name)) ptr = &(*ptr)->next;
2015 if (add)
2016 {
2017 // If we don't already have this domain in our list for this browse operation, add it now
2018 if (!*ptr) add_domain_to_browser(request, &d->name);
2019 else debugf("udsserver_automatic_browse_domain_changed %##s already in list, not re-adding", &d->name);
2020 }
2021 else
2022 {
2023 if (!*ptr) LogMsg("udsserver_automatic_browse_domain_changed ERROR %##s not found", &d->name);
2024 else
2025 {
2026 DNameListElem *p;
2027 for (p = AutoBrowseDomains; p; p=p->next)
2028 if (!p->uid || SystemUID(request->uid) || request->uid == p->uid)
2029 if (SameDomainName(&d->name, &p->name)) break;
2030 if (p) debugf("udsserver_automatic_browse_domain_changed %##s still in list, not removing", &d->name);
2031 else
2032 {
2033 browser_t *remove = *ptr;
2034 *ptr = (*ptr)->next;
2035 mDNS_StopQueryWithRemoves(&mDNSStorage, &remove->q);
2036 freeL("browser_t/udsserver_automatic_browse_domain_changed", remove);
2037 }
2038 }
2039 }
2040 }
2041 }
2042 }
2043
2044 mDNSlocal void FreeARElemCallback(mDNS *const m, AuthRecord *const rr, mStatus result)
2045 {
2046 (void)m; // unused
2047 if (result == mStatus_MemFree) mDNSPlatformMemFree(rr->RecordContext);
2048 }
2049
2050 mDNSlocal void RegisterLocalOnlyDomainEnumPTR(mDNS *m, const domainname *d, int type)
2051 {
2052 // allocate/register legacy and non-legacy _browse PTR record
2053 mStatus err;
2054 ARListElem *ptr = mDNSPlatformMemAllocate(sizeof(*ptr));
2055
2056 LogOperation("Incrementing %s refcount for %##s",
2057 (type == mDNS_DomainTypeBrowse ) ? "browse domain " :
2058 (type == mDNS_DomainTypeRegistration ) ? "registration dom" :
2059 (type == mDNS_DomainTypeBrowseAutomatic) ? "automatic browse" : "?", d->c);
2060
2061 mDNS_SetupResourceRecord(&ptr->ar, mDNSNULL, mDNSInterface_LocalOnly, kDNSType_PTR, 7200, kDNSRecordTypeShared, FreeARElemCallback, ptr);
2062 MakeDomainNameFromDNSNameString(&ptr->ar.namestorage, mDNS_DomainTypeNames[type]);
2063 AppendDNSNameString (&ptr->ar.namestorage, "local");
2064 AssignDomainName(&ptr->ar.resrec.rdata->u.name, d);
2065 err = mDNS_Register(m, &ptr->ar);
2066 if (err)
2067 {
2068 LogMsg("SetSCPrefsBrowseDomain: mDNS_Register returned error %d", err);
2069 mDNSPlatformMemFree(ptr);
2070 }
2071 else
2072 {
2073 ptr->next = LocalDomainEnumRecords;
2074 LocalDomainEnumRecords = ptr;
2075 }
2076 }
2077
2078 mDNSlocal void DeregisterLocalOnlyDomainEnumPTR(mDNS *m, const domainname *d, int type)
2079 {
2080 ARListElem **ptr = &LocalDomainEnumRecords;
2081 domainname lhs; // left-hand side of PTR, for comparison
2082
2083 LogOperation("Decrementing %s refcount for %##s",
2084 (type == mDNS_DomainTypeBrowse ) ? "browse domain " :
2085 (type == mDNS_DomainTypeRegistration ) ? "registration dom" :
2086 (type == mDNS_DomainTypeBrowseAutomatic) ? "automatic browse" : "?", d->c);
2087
2088 MakeDomainNameFromDNSNameString(&lhs, mDNS_DomainTypeNames[type]);
2089 AppendDNSNameString (&lhs, "local");
2090
2091 while (*ptr)
2092 {
2093 if (SameDomainName(&(*ptr)->ar.resrec.rdata->u.name, d) && SameDomainName((*ptr)->ar.resrec.name, &lhs))
2094 {
2095 ARListElem *remove = *ptr;
2096 *ptr = (*ptr)->next;
2097 mDNS_Deregister(m, &remove->ar);
2098 return;
2099 }
2100 else ptr = &(*ptr)->next;
2101 }
2102 }
2103
2104 mDNSlocal void AddAutoBrowseDomain(const mDNSu32 uid, const domainname *const name)
2105 {
2106 DNameListElem *new = mDNSPlatformMemAllocate(sizeof(DNameListElem));
2107 if (!new) { LogMsg("ERROR: malloc"); return; }
2108 AssignDomainName(&new->name, name);
2109 new->uid = uid;
2110 new->next = AutoBrowseDomains;
2111 AutoBrowseDomains = new;
2112 udsserver_automatic_browse_domain_changed(new, mDNStrue);
2113 }
2114
2115 mDNSlocal void RmvAutoBrowseDomain(const mDNSu32 uid, const domainname *const name)
2116 {
2117 DNameListElem **p = &AutoBrowseDomains;
2118 while (*p && (!SameDomainName(&(*p)->name, name) || (*p)->uid != uid)) p = &(*p)->next;
2119 if (!*p) LogMsg("RmvAutoBrowseDomain: Got remove event for domain %##s not in list", name->c);
2120 else
2121 {
2122 DNameListElem *ptr = *p;
2123 *p = ptr->next;
2124 udsserver_automatic_browse_domain_changed(ptr, mDNSfalse);
2125 mDNSPlatformMemFree(ptr);
2126 }
2127 }
2128
2129 mDNSlocal void SetPrefsBrowseDomains(mDNS *m, DNameListElem *browseDomains, mDNSBool add)
2130 {
2131 DNameListElem *d;
2132 for (d = browseDomains; d; d = d->next)
2133 {
2134 if (add)
2135 {
2136 RegisterLocalOnlyDomainEnumPTR(m, &d->name, mDNS_DomainTypeBrowse);
2137 AddAutoBrowseDomain(d->uid, &d->name);
2138 }
2139 else
2140 {
2141 DeregisterLocalOnlyDomainEnumPTR(m, &d->name, mDNS_DomainTypeBrowse);
2142 RmvAutoBrowseDomain(d->uid, &d->name);
2143 }
2144 }
2145 }
2146
2147 mDNSlocal void UpdateDeviceInfoRecord(mDNS *const m)
2148 {
2149 int num_autoname = 0;
2150 request_state *req;
2151 for (req = all_requests; req; req = req->next)
2152 if (req->terminate == regservice_termination_callback && req->u.servicereg.autoname)
2153 num_autoname++;
2154
2155 // If DeviceInfo record is currently registered, see if we need to deregister it
2156 if (m->DeviceInfo.resrec.RecordType != kDNSRecordTypeUnregistered)
2157 if (num_autoname == 0 || !SameDomainLabelCS(m->DeviceInfo.resrec.name->c, m->nicelabel.c))
2158 {
2159 LogOperation("UpdateDeviceInfoRecord Deregister %##s", m->DeviceInfo.resrec.name);
2160 mDNS_Deregister(m, &m->DeviceInfo);
2161 }
2162
2163 // If DeviceInfo record is not currently registered, see if we need to register it
2164 if (m->DeviceInfo.resrec.RecordType == kDNSRecordTypeUnregistered)
2165 if (num_autoname > 0)
2166 {
2167 mDNSu8 len = m->HIHardware.c[0] < 255 - 6 ? m->HIHardware.c[0] : 255 - 6;
2168 mDNS_SetupResourceRecord(&m->DeviceInfo, mDNSNULL, mDNSNULL, kDNSType_TXT, kStandardTTL, kDNSRecordTypeAdvisory, mDNSNULL, mDNSNULL);
2169 ConstructServiceName(&m->DeviceInfo.namestorage, &m->nicelabel, &DeviceInfoName, &localdomain);
2170 mDNSPlatformMemCopy(m->DeviceInfo.resrec.rdata->u.data + 1, "model=", 6);
2171 mDNSPlatformMemCopy(m->DeviceInfo.resrec.rdata->u.data + 7, m->HIHardware.c + 1, len);
2172 m->DeviceInfo.resrec.rdata->u.data[0] = 6 + len; // "model=" plus the device string
2173 m->DeviceInfo.resrec.rdlength = 7 + len; // One extra for the length byte at the start of the string
2174 LogOperation("UpdateDeviceInfoRecord Register %##s", m->DeviceInfo.resrec.name);
2175 mDNS_Register(m, &m->DeviceInfo);
2176 }
2177 }
2178
2179 mDNSexport void udsserver_handle_configchange(mDNS *const m)
2180 {
2181 request_state *req;
2182 service_instance *ptr;
2183 DNameListElem *RegDomains;
2184 DNameListElem *BrowseDomains;
2185 DNameListElem *p;
2186
2187 UpdateDeviceInfoRecord(m);
2188
2189 // For autoname services, see if the default service name has changed, necessitating an automatic update
2190 for (req = all_requests; req; req = req->next)
2191 if (req->terminate == regservice_termination_callback)
2192 if (req->u.servicereg.autoname && !SameDomainLabelCS(req->u.servicereg.name.c, m->nicelabel.c))
2193 {
2194 req->u.servicereg.name = m->nicelabel;
2195 for (ptr = req->u.servicereg.instances; ptr; ptr = ptr->next)
2196 {
2197 ptr->renameonmemfree = 1;
2198 if (ptr->clientnotified) SendServiceRemovalNotification(&ptr->srs);
2199 if (mDNS_DeregisterService(m, &ptr->srs)) // If service was deregistered already
2200 regservice_callback(m, &ptr->srs, mStatus_MemFree); // we can re-register immediately
2201 }
2202 }
2203
2204 // Let the platform layer get the current DNS information
2205 mDNS_Lock(m);
2206 mDNSPlatformSetDNSConfig(m, mDNSfalse, mDNSfalse, mDNSNULL, &RegDomains, &BrowseDomains);
2207 mDNS_Unlock(m);
2208
2209 // Any automatic registration domains are also implicitly automatic browsing domains
2210 if (RegDomains) SetPrefsBrowseDomains(m, RegDomains, mDNStrue); // Add the new list first
2211 if (AutoRegistrationDomains) SetPrefsBrowseDomains(m, AutoRegistrationDomains, mDNSfalse); // Then clear the old list
2212
2213 // Add any new domains not already in our AutoRegistrationDomains list
2214 for (p=RegDomains; p; p=p->next)
2215 {
2216 DNameListElem **pp = &AutoRegistrationDomains;
2217 while (*pp && ((*pp)->uid != p->uid || !SameDomainName(&(*pp)->name, &p->name))) pp = &(*pp)->next;
2218 if (!*pp) // If not found in our existing list, this is a new default registration domain
2219 {
2220 RegisterLocalOnlyDomainEnumPTR(m, &p->name, mDNS_DomainTypeRegistration);
2221 udsserver_default_reg_domain_changed(p, mDNStrue);
2222 }
2223 else // else found same domainname in both old and new lists, so no change, just delete old copy
2224 {
2225 DNameListElem *del = *pp;
2226 *pp = (*pp)->next;
2227 mDNSPlatformMemFree(del);
2228 }
2229 }
2230
2231 // Delete any domains in our old AutoRegistrationDomains list that are now gone
2232 while (AutoRegistrationDomains)
2233 {
2234 DNameListElem *del = AutoRegistrationDomains;
2235 AutoRegistrationDomains = AutoRegistrationDomains->next; // Cut record from list FIRST,
2236 DeregisterLocalOnlyDomainEnumPTR(m, &del->name, mDNS_DomainTypeRegistration);
2237 udsserver_default_reg_domain_changed(del, mDNSfalse); // before calling udsserver_default_reg_domain_changed()
2238 mDNSPlatformMemFree(del);
2239 }
2240
2241 // Now we have our new updated automatic registration domain list
2242 AutoRegistrationDomains = RegDomains;
2243
2244 // Add new browse domains to internal list
2245 if (BrowseDomains) SetPrefsBrowseDomains(m, BrowseDomains, mDNStrue);
2246
2247 // Remove old browse domains from internal list
2248 if (SCPrefBrowseDomains)
2249 {
2250 SetPrefsBrowseDomains(m, SCPrefBrowseDomains, mDNSfalse);
2251 while (SCPrefBrowseDomains)
2252 {
2253 DNameListElem *fptr = SCPrefBrowseDomains;
2254 SCPrefBrowseDomains = SCPrefBrowseDomains->next;
2255 mDNSPlatformMemFree(fptr);
2256 }
2257 }
2258
2259 // Replace the old browse domains array with the new array
2260 SCPrefBrowseDomains = BrowseDomains;
2261 }
2262
2263 mDNSlocal void AutomaticBrowseDomainChange(mDNS *const m, DNSQuestion *q, const ResourceRecord *const answer, QC_result AddRecord)
2264 {
2265 (void)m; // unused;
2266 (void)q; // unused
2267
2268 LogOperation("AutomaticBrowseDomainChange: %s automatic browse domain %##s",
2269 AddRecord ? "Adding" : "Removing", answer->rdata->u.name.c);
2270
2271 if (AddRecord) AddAutoBrowseDomain(0, &answer->rdata->u.name);
2272 else RmvAutoBrowseDomain(0, &answer->rdata->u.name);
2273 }
2274
2275 mDNSlocal mStatus handle_browse_request(request_state *request)
2276 {
2277 char regtype[MAX_ESCAPED_DOMAIN_NAME], domain[MAX_ESCAPED_DOMAIN_NAME];
2278 domainname typedn, d, temp;
2279 mDNSs32 NumSubTypes;
2280 mStatus err = mStatus_NoError;
2281
2282 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
2283 mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend);
2284 mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex);
2285 if (interfaceIndex && !InterfaceID) return(mStatus_BadParamErr);
2286
2287 if (get_string(&request->msgptr, request->msgend, regtype, MAX_ESCAPED_DOMAIN_NAME) < 0 ||
2288 get_string(&request->msgptr, request->msgend, domain, MAX_ESCAPED_DOMAIN_NAME) < 0) return(mStatus_BadParamErr);
2289
2290 if (!request->msgptr) { LogMsg("%3d: DNSServiceBrowse(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
2291
2292 if (!domain || (domain[0] == '\0')) uDNS_RegisterSearchDomains(&mDNSStorage);
2293
2294 typedn.c[0] = 0;
2295 NumSubTypes = ChopSubTypes(regtype); // Note: Modifies regtype string to remove trailing subtypes
2296 if (NumSubTypes < 0 || NumSubTypes > 1) return(mStatus_BadParamErr);
2297 if (NumSubTypes == 1 && !AppendDNSNameString(&typedn, regtype + strlen(regtype) + 1)) return(mStatus_BadParamErr);
2298
2299 if (!regtype[0] || !AppendDNSNameString(&typedn, regtype)) return(mStatus_BadParamErr);
2300
2301 if (!MakeDomainNameFromDNSNameString(&temp, regtype)) return(mStatus_BadParamErr);
2302 // For over-long service types, we only allow domain "local"
2303 if (temp.c[0] > 15 && domain[0] == 0) mDNSPlatformStrCopy(domain, "local.");
2304
2305 // Set up browser info
2306 request->u.browser.ForceMCast = (flags & kDNSServiceFlagsForceMulticast) != 0;
2307 request->u.browser.interface_id = InterfaceID;
2308 AssignDomainName(&request->u.browser.regtype, &typedn);
2309 request->u.browser.default_domain = !domain[0];
2310 request->u.browser.browsers = NULL;
2311
2312 LogOperation("%3d: DNSServiceBrowse(\"%##s\", \"%s\") START", request->sd, request->u.browser.regtype.c, domain);
2313 if (domain[0])
2314 {
2315 if (!MakeDomainNameFromDNSNameString(&d, domain)) return(mStatus_BadParamErr);
2316 err = add_domain_to_browser(request, &d);
2317 }
2318
2319 else
2320 {
2321 DNameListElem *sdom;
2322 for (sdom = AutoBrowseDomains; sdom; sdom = sdom->next)
2323 if (!sdom->uid || SystemUID(request->uid) || request->uid == sdom->uid)
2324 {
2325 err = add_domain_to_browser(request, &sdom->name);
2326 if (err)
2327 {
2328 if (SameDomainName(&sdom->name, &localdomain)) break;
2329 else err = mStatus_NoError; // suppress errors for non-local "default" domains
2330 }
2331 }
2332 }
2333
2334 if (!err) request->terminate = browse_termination_callback;
2335
2336 return(err);
2337 }
2338
2339 // ***************************************************************************
2340 #if COMPILER_LIKES_PRAGMA_MARK
2341 #pragma mark -
2342 #pragma mark - DNSServiceResolve
2343 #endif
2344
2345 mDNSlocal void resolve_result_callback(mDNS *const m, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord)
2346 {
2347 size_t len = 0;
2348 char fullname[MAX_ESCAPED_DOMAIN_NAME], target[MAX_ESCAPED_DOMAIN_NAME];
2349 char *data;
2350 reply_state *rep;
2351 request_state *req = question->QuestionContext;
2352 (void)m; // Unused
2353
2354 LogOperation("%3d: DNSServiceResolve(%##s, %s) %s %s",
2355 req->sd, question->qname.c, DNSTypeName(question->qtype), AddRecord ? "ADD" : "RMV", RRDisplayString(m, answer));
2356
2357 // This code used to do this trick of just keeping a copy of the pointer to
2358 // the answer record in the cache, but the unicast query code doesn't currently
2359 // put its answer records in the cache, so for now we can't do this.
2360
2361 if (!AddRecord)
2362 {
2363 if (answer->rrtype == kDNSType_SRV && req->u.resolve.srv == answer) req->u.resolve.srv = mDNSNULL;
2364 if (answer->rrtype == kDNSType_TXT && req->u.resolve.txt == answer) req->u.resolve.txt = mDNSNULL;
2365 return;
2366 }
2367
2368 if (answer->rrtype == kDNSType_SRV) req->u.resolve.srv = answer;
2369 if (answer->rrtype == kDNSType_TXT) req->u.resolve.txt = answer;
2370
2371 if (!req->u.resolve.txt || !req->u.resolve.srv) return; // only deliver result to client if we have both answers
2372
2373 ConvertDomainNameToCString(answer->name, fullname);
2374 ConvertDomainNameToCString(&req->u.resolve.srv->rdata->u.srv.target, target);
2375
2376 // calculate reply length
2377 len += sizeof(DNSServiceFlags);
2378 len += sizeof(mDNSu32); // interface index
2379 len += sizeof(DNSServiceErrorType);
2380 len += strlen(fullname) + 1;
2381 len += strlen(target) + 1;
2382 len += 2 * sizeof(mDNSu16); // port, txtLen
2383 len += req->u.resolve.txt->rdlength;
2384
2385 // allocate/init reply header
2386 rep = create_reply(resolve_reply_op, len, req);
2387 rep->rhdr->flags = dnssd_htonl(0);
2388 rep->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(m, answer->InterfaceID));
2389 rep->rhdr->error = dnssd_htonl(kDNSServiceErr_NoError);
2390
2391 data = rep->sdata;
2392
2393 // write reply data to message
2394 put_string(fullname, &data);
2395 put_string(target, &data);
2396 *data++ = req->u.resolve.srv->rdata->u.srv.port.b[0];
2397 *data++ = req->u.resolve.srv->rdata->u.srv.port.b[1];
2398 put_uint16(req->u.resolve.txt->rdlength, &data);
2399 put_rdata(req->u.resolve.txt->rdlength, req->u.resolve.txt->rdata->u.data, &data);
2400
2401 append_reply(req, rep);
2402 }
2403
2404 mDNSlocal void resolve_termination_callback(request_state *request)
2405 {
2406 LogOperation("%3d: DNSServiceResolve(%##s) STOP", request->sd, request->u.resolve.qtxt.qname.c);
2407 mDNS_StopQuery(&mDNSStorage, &request->u.resolve.qtxt);
2408 mDNS_StopQuery(&mDNSStorage, &request->u.resolve.qsrv);
2409 }
2410
2411 mDNSlocal mStatus handle_resolve_request(request_state *request)
2412 {
2413 char name[256], regtype[MAX_ESCAPED_DOMAIN_NAME], domain[MAX_ESCAPED_DOMAIN_NAME];
2414 domainname fqdn;
2415 mStatus err;
2416
2417 // extract the data from the message
2418 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
2419 mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend);
2420 mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex);
2421 if (interfaceIndex && !InterfaceID)
2422 { LogMsg("ERROR: handle_resolve_request bad interfaceIndex %d", interfaceIndex); return(mStatus_BadParamErr); }
2423
2424 if (get_string(&request->msgptr, request->msgend, name, 256) < 0 ||
2425 get_string(&request->msgptr, request->msgend, regtype, MAX_ESCAPED_DOMAIN_NAME) < 0 ||
2426 get_string(&request->msgptr, request->msgend, domain, MAX_ESCAPED_DOMAIN_NAME) < 0)
2427 { LogMsg("ERROR: handle_resolve_request - Couldn't read name/regtype/domain"); return(mStatus_BadParamErr); }
2428
2429 if (!request->msgptr) { LogMsg("%3d: DNSServiceResolve(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
2430
2431 if (build_domainname_from_strings(&fqdn, name, regtype, domain) < 0)
2432 { LogMsg("ERROR: handle_resolve_request bad “%s” “%s” “%s”", name, regtype, domain); return(mStatus_BadParamErr); }
2433
2434 mDNSPlatformMemZero(&request->u.resolve, sizeof(request->u.resolve));
2435
2436 // format questions
2437 request->u.resolve.qsrv.InterfaceID = InterfaceID;
2438 request->u.resolve.qsrv.Target = zeroAddr;
2439 AssignDomainName(&request->u.resolve.qsrv.qname, &fqdn);
2440 request->u.resolve.qsrv.qtype = kDNSType_SRV;
2441 request->u.resolve.qsrv.qclass = kDNSClass_IN;
2442 request->u.resolve.qsrv.LongLived = (flags & kDNSServiceFlagsLongLivedQuery ) != 0;
2443 request->u.resolve.qsrv.ExpectUnique = mDNStrue;
2444 request->u.resolve.qsrv.ForceMCast = (flags & kDNSServiceFlagsForceMulticast ) != 0;
2445 request->u.resolve.qsrv.ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0;
2446 request->u.resolve.qsrv.QuestionCallback = resolve_result_callback;
2447 request->u.resolve.qsrv.QuestionContext = request;
2448
2449 request->u.resolve.qtxt.InterfaceID = InterfaceID;
2450 request->u.resolve.qtxt.Target = zeroAddr;
2451 AssignDomainName(&request->u.resolve.qtxt.qname, &fqdn);
2452 request->u.resolve.qtxt.qtype = kDNSType_TXT;
2453 request->u.resolve.qtxt.qclass = kDNSClass_IN;
2454 request->u.resolve.qtxt.LongLived = (flags & kDNSServiceFlagsLongLivedQuery ) != 0;
2455 request->u.resolve.qtxt.ExpectUnique = mDNStrue;
2456 request->u.resolve.qtxt.ForceMCast = (flags & kDNSServiceFlagsForceMulticast ) != 0;
2457 request->u.resolve.qtxt.ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0;
2458 request->u.resolve.qtxt.QuestionCallback = resolve_result_callback;
2459 request->u.resolve.qtxt.QuestionContext = request;
2460
2461 // ask the questions
2462 LogOperation("%3d: DNSServiceResolve(%##s) START", request->sd, request->u.resolve.qsrv.qname.c);
2463 err = mDNS_StartQuery(&mDNSStorage, &request->u.resolve.qsrv);
2464 if (!err)
2465 {
2466 err = mDNS_StartQuery(&mDNSStorage, &request->u.resolve.qtxt);
2467 if (err) mDNS_StopQuery(&mDNSStorage, &request->u.resolve.qsrv);
2468 }
2469
2470 if (!err) request->terminate = resolve_termination_callback;
2471
2472 return(err);
2473 }
2474
2475 // ***************************************************************************
2476 #if COMPILER_LIKES_PRAGMA_MARK
2477 #pragma mark -
2478 #pragma mark - DNSServiceQueryRecord
2479 #endif
2480
2481 // mDNS operation functions. Each operation has 3 associated functions - a request handler that parses
2482 // the client's request and makes the appropriate mDNSCore call, a result handler (passed as a callback
2483 // to the mDNSCore routine) that sends results back to the client, and a termination routine that aborts
2484 // the mDNSCore operation if the client dies or closes its socket.
2485
2486 // query and resolve calls have separate request handlers that parse the arguments from the client and
2487 // massage the name parameters appropriately, but the rest of the operations (making the query call,
2488 // delivering the result to the client, and termination) are identical.
2489
2490 // what gets called when a resolve is completed and we need to send the data back to the client
2491 mDNSlocal void queryrecord_result_callback(mDNS *const m, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord)
2492 {
2493 char name[MAX_ESCAPED_DOMAIN_NAME];
2494 request_state *req = question->QuestionContext;
2495 reply_state *rep;
2496 char *data;
2497 size_t len;
2498 DNSServiceErrorType error = kDNSServiceErr_NoError;
2499 (void)m; // Unused
2500
2501 LogOperation("%3d: %s(%##s, %s) %s %s", req->sd,
2502 req->hdr.op == query_request ? "DNSServiceQueryRecord" : "DNSServiceGetAddrInfo",
2503 question->qname.c, DNSTypeName(question->qtype), AddRecord ? "ADD" : "RMV", RRDisplayString(m, answer));
2504
2505 if (answer->RecordType == kDNSRecordTypePacketNegative)
2506 {
2507 error = kDNSServiceErr_NoSuchRecord;
2508 ConvertDomainNameToCString(&question->qname, name);
2509 AddRecord = mDNStrue;
2510 }
2511 else
2512 ConvertDomainNameToCString(answer->name, name);
2513
2514 len = sizeof(DNSServiceFlags); // calculate reply data length
2515 len += sizeof(mDNSu32); // interface index
2516 len += sizeof(DNSServiceErrorType);
2517 len += strlen(name) + 1;
2518 len += 3 * sizeof(mDNSu16); // type, class, rdlen
2519 len += answer->rdlength;
2520 len += sizeof(mDNSu32); // TTL
2521
2522 rep = create_reply(req->hdr.op == query_request ? query_reply_op : addrinfo_reply_op, len, req);
2523
2524 rep->rhdr->flags = dnssd_htonl(AddRecord ? kDNSServiceFlagsAdd : 0);
2525 rep->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(m, answer->InterfaceID));
2526 rep->rhdr->error = dnssd_htonl(error);
2527
2528 data = rep->sdata;
2529
2530 put_string(name, &data);
2531
2532 if (answer->RecordType == kDNSRecordTypePacketNegative)
2533 {
2534 put_uint16(question->qtype, &data);
2535 put_uint16(question->qclass, &data);
2536 put_uint16(0, &data);
2537 put_rdata(0, mDNSNULL, &data);
2538 put_uint32(0, &data);
2539 }
2540 else
2541 {
2542 put_uint16(answer->rrtype, &data);
2543 put_uint16(answer->rrclass, &data);
2544 put_uint16(answer->rdlength, &data);
2545 //put_rdata(answer->rdlength, answer->rdata->u.data, &data);
2546 if (!putRData(mDNSNULL, (mDNSu8 *)data, (mDNSu8 *)rep->rhdr + len, answer))
2547 LogMsg("queryrecord_result_callback putRData failed %d", (mDNSu8 *)rep->rhdr + len - (mDNSu8 *)data);
2548 data += answer->rdlength;
2549 put_uint32(AddRecord ? answer->rroriginalttl : 0, &data);
2550 }
2551
2552 append_reply(req, rep);
2553 }
2554
2555 mDNSlocal void queryrecord_termination_callback(request_state *request)
2556 {
2557 LogOperation("%3d: DNSServiceQueryRecord(%##s, %s) STOP",
2558 request->sd, request->u.queryrecord.q.qname.c, DNSTypeName(request->u.queryrecord.q.qtype));
2559 mDNS_StopQuery(&mDNSStorage, &request->u.queryrecord.q); // no need to error check
2560 }
2561
2562 mDNSlocal mStatus handle_queryrecord_request(request_state *request)
2563 {
2564 char name[256];
2565 mDNSu16 rrtype, rrclass;
2566 mStatus err;
2567
2568 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
2569 mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend);
2570 mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex);
2571 if (interfaceIndex && !InterfaceID) return(mStatus_BadParamErr);
2572
2573 if (get_string(&request->msgptr, request->msgend, name, 256) < 0) return(mStatus_BadParamErr);
2574 rrtype = get_uint16(&request->msgptr, request->msgend);
2575 rrclass = get_uint16(&request->msgptr, request->msgend);
2576
2577 if (!request->msgptr) { LogMsg("%3d: DNSServiceQueryRecord(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
2578
2579 mDNSPlatformMemZero(&request->u.queryrecord.q, sizeof(&request->u.queryrecord.q));
2580
2581 request->u.queryrecord.q.InterfaceID = InterfaceID;
2582 request->u.queryrecord.q.Target = zeroAddr;
2583 if (!MakeDomainNameFromDNSNameString(&request->u.queryrecord.q.qname, name)) return(mStatus_BadParamErr);
2584 request->u.queryrecord.q.qtype = rrtype;
2585 request->u.queryrecord.q.qclass = rrclass;
2586 request->u.queryrecord.q.LongLived = (flags & kDNSServiceFlagsLongLivedQuery ) != 0;
2587 request->u.queryrecord.q.ExpectUnique = mDNSfalse;
2588 request->u.queryrecord.q.ForceMCast = (flags & kDNSServiceFlagsForceMulticast ) != 0;
2589 request->u.queryrecord.q.ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0;
2590 request->u.queryrecord.q.QuestionCallback = queryrecord_result_callback;
2591 request->u.queryrecord.q.QuestionContext = request;
2592
2593 LogOperation("%3d: DNSServiceQueryRecord(%##s, %s, %X) START",
2594 request->sd, request->u.queryrecord.q.qname.c, DNSTypeName(request->u.queryrecord.q.qtype), flags);
2595 err = mDNS_StartQuery(&mDNSStorage, &request->u.queryrecord.q);
2596 if (err) LogMsg("ERROR: mDNS_StartQuery: %d", (int)err);
2597
2598 if (!err) request->terminate = queryrecord_termination_callback;
2599
2600 return(err);
2601 }
2602
2603 // ***************************************************************************
2604 #if COMPILER_LIKES_PRAGMA_MARK
2605 #pragma mark -
2606 #pragma mark - DNSServiceEnumerateDomains
2607 #endif
2608
2609 mDNSlocal reply_state *format_enumeration_reply(request_state *request,
2610 const char *domain, DNSServiceFlags flags, mDNSu32 ifi, DNSServiceErrorType err)
2611 {
2612 size_t len;
2613 reply_state *reply;
2614 char *data;
2615
2616 len = sizeof(DNSServiceFlags);
2617 len += sizeof(mDNSu32);
2618 len += sizeof(DNSServiceErrorType);
2619 len += strlen(domain) + 1;
2620
2621 reply = create_reply(enumeration_reply_op, len, request);
2622 reply->rhdr->flags = dnssd_htonl(flags);
2623 reply->rhdr->ifi = dnssd_htonl(ifi);
2624 reply->rhdr->error = dnssd_htonl(err);
2625 data = reply->sdata;
2626 put_string(domain, &data);
2627 return reply;
2628 }
2629
2630 mDNSlocal void enum_termination_callback(request_state *request)
2631 {
2632 mDNS_StopGetDomains(&mDNSStorage, &request->u.enumeration.q_all);
2633 mDNS_StopGetDomains(&mDNSStorage, &request->u.enumeration.q_default);
2634 }
2635
2636 mDNSlocal void enum_result_callback(mDNS *const m,
2637 DNSQuestion *const question, const ResourceRecord *const answer, QC_result AddRecord)
2638 {
2639 char domain[MAX_ESCAPED_DOMAIN_NAME];
2640 request_state *request = question->QuestionContext;
2641 DNSServiceFlags flags = 0;
2642 reply_state *reply;
2643 (void)m; // Unused
2644
2645 if (answer->rrtype != kDNSType_PTR) return;
2646
2647 // We only return add/remove events for the browse and registration lists
2648 // For the default browse and registration answers, we only give an "ADD" event
2649 if (question == &request->u.enumeration.q_default && !AddRecord) return;
2650
2651 if (AddRecord)
2652 {
2653 flags |= kDNSServiceFlagsAdd;
2654 if (question == &request->u.enumeration.q_default) flags |= kDNSServiceFlagsDefault;
2655 }
2656
2657 ConvertDomainNameToCString(&answer->rdata->u.name, domain);
2658 // Note that we do NOT propagate specific interface indexes to the client - for example, a domain we learn from
2659 // a machine's system preferences may be discovered on the LocalOnly interface, but should be browsed on the
2660 // network, so we just pass kDNSServiceInterfaceIndexAny
2661 reply = format_enumeration_reply(request, domain, flags, kDNSServiceInterfaceIndexAny, kDNSServiceErr_NoError);
2662 if (!reply) { LogMsg("ERROR: enum_result_callback, format_enumeration_reply"); return; }
2663 append_reply(request, reply);
2664 }
2665
2666 mDNSlocal mStatus handle_enum_request(request_state *request)
2667 {
2668 mStatus err;
2669 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
2670 DNSServiceFlags reg = flags & kDNSServiceFlagsRegistrationDomains;
2671 mDNS_DomainType t_all = reg ? mDNS_DomainTypeRegistration : mDNS_DomainTypeBrowse;
2672 mDNS_DomainType t_default = reg ? mDNS_DomainTypeRegistrationDefault : mDNS_DomainTypeBrowseDefault;
2673 mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend);
2674 mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex);
2675 if (interfaceIndex && !InterfaceID) return(mStatus_BadParamErr);
2676
2677 if (!request->msgptr) { LogMsg("%3d: DNSServiceEnumerateDomains(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
2678
2679 // allocate context structures
2680 uDNS_RegisterSearchDomains(&mDNSStorage);
2681
2682 // enumeration requires multiple questions, so we must link all the context pointers so that
2683 // necessary context can be reached from the callbacks
2684 request->u.enumeration.q_all .QuestionContext = request;
2685 request->u.enumeration.q_default.QuestionContext = request;
2686
2687 // if the caller hasn't specified an explicit interface, we use local-only to get the system-wide list.
2688 if (!InterfaceID) InterfaceID = mDNSInterface_LocalOnly;
2689
2690 // make the calls
2691 LogOperation("%3d: DNSServiceEnumerateDomains(%X=%s)", request->sd, flags,
2692 (flags & kDNSServiceFlagsBrowseDomains ) ? "kDNSServiceFlagsBrowseDomains" :
2693 (flags & kDNSServiceFlagsRegistrationDomains) ? "kDNSServiceFlagsRegistrationDomains" : "<<Unknown>>");
2694 err = mDNS_GetDomains(&mDNSStorage, &request->u.enumeration.q_all, t_all, NULL, InterfaceID, enum_result_callback, request);
2695 if (!err)
2696 {
2697 err = mDNS_GetDomains(&mDNSStorage, &request->u.enumeration.q_default, t_default, NULL, InterfaceID, enum_result_callback, request);
2698 if (err) mDNS_StopGetDomains(&mDNSStorage, &request->u.enumeration.q_all);
2699 }
2700 if (!err) request->terminate = enum_termination_callback;
2701
2702 return(err);
2703 }
2704
2705 // ***************************************************************************
2706 #if COMPILER_LIKES_PRAGMA_MARK
2707 #pragma mark -
2708 #pragma mark - DNSServiceReconfirmRecord & Misc
2709 #endif
2710
2711 mDNSlocal mStatus handle_reconfirm_request(request_state *request)
2712 {
2713 mStatus status = mStatus_BadParamErr;
2714 AuthRecord *rr = read_rr_from_ipc_msg(request, 0, 0);
2715 if (rr)
2716 {
2717 status = mDNS_ReconfirmByValue(&mDNSStorage, &rr->resrec);
2718 LogOperation(
2719 (status == mStatus_NoError) ?
2720 "%3d: DNSServiceReconfirmRecord(%s) interface %d initiated" :
2721 "%3d: DNSServiceReconfirmRecord(%s) interface %d failed: %d",
2722 request->sd, RRDisplayString(&mDNSStorage, &rr->resrec),
2723 mDNSPlatformInterfaceIndexfromInterfaceID(&mDNSStorage, rr->resrec.InterfaceID), status);
2724 freeL("AuthRecord/handle_reconfirm_request", rr);
2725 }
2726 return(status);
2727 }
2728
2729 mDNSlocal mStatus handle_setdomain_request(request_state *request)
2730 {
2731 char domainstr[MAX_ESCAPED_DOMAIN_NAME];
2732 domainname domain;
2733 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
2734 (void)flags; // Unused
2735 if (get_string(&request->msgptr, request->msgend, domainstr, MAX_ESCAPED_DOMAIN_NAME) < 0 ||
2736 !MakeDomainNameFromDNSNameString(&domain, domainstr))
2737 { LogMsg("%3d: DNSServiceSetDefaultDomainForUser(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
2738
2739 LogOperation("%3d: DNSServiceSetDefaultDomainForUser(%##s)", request->sd, domain.c);
2740 return(mStatus_NoError);
2741 }
2742
2743 typedef packedstruct
2744 {
2745 mStatus err;
2746 mDNSu32 len;
2747 mDNSu32 vers;
2748 } DaemonVersionReply;
2749
2750 mDNSlocal void handle_getproperty_request(request_state *request)
2751 {
2752 const mStatus BadParamErr = dnssd_htonl(mStatus_BadParamErr);
2753 char prop[256];
2754 if (get_string(&request->msgptr, request->msgend, prop, sizeof(prop)) >= 0)
2755 {
2756 LogOperation("%3d: DNSServiceGetProperty(%s)", request->sd, prop);
2757 if (!strcmp(prop, kDNSServiceProperty_DaemonVersion))
2758 {
2759 DaemonVersionReply x = { 0, dnssd_htonl(4), dnssd_htonl(_DNS_SD_H) };
2760 send_all(request->sd, (const char *)&x, sizeof(x));
2761 return;
2762 }
2763 }
2764
2765 // If we didn't recogize the requested property name, return BadParamErr
2766 send_all(request->sd, (const char *)&BadParamErr, sizeof(BadParamErr));
2767 }
2768
2769 // ***************************************************************************
2770 #if COMPILER_LIKES_PRAGMA_MARK
2771 #pragma mark -
2772 #pragma mark - DNSServiceNATPortMappingCreate
2773 #endif
2774
2775 #define DNSServiceProtocol(X) ((X) == NATOp_AddrRequest ? 0 : (X) == NATOp_MapUDP ? kDNSServiceProtocol_UDP : kDNSServiceProtocol_TCP)
2776
2777 mDNSlocal void port_mapping_termination_callback(request_state *request)
2778 {
2779 LogOperation("%3d: DNSServiceNATPortMappingCreate(%X, %u, %u, %d) STOP", request->sd,
2780 DNSServiceProtocol(request->u.pm.NATinfo.Protocol),
2781 mDNSVal16(request->u.pm.NATinfo.IntPort), mDNSVal16(request->u.pm.ReqExt), request->u.pm.NATinfo.NATLease);
2782 mDNS_StopNATOperation(&mDNSStorage, &request->u.pm.NATinfo);
2783 }
2784
2785 // Called via function pointer when we get a NAT-PMP address request or port mapping response
2786 mDNSlocal void port_mapping_create_request_callback(mDNS *m, NATTraversalInfo *n)
2787 {
2788 request_state *request = (request_state *)n->clientContext;
2789 reply_state *rep;
2790 int replyLen;
2791 char *data;
2792
2793 if (!request) { LogMsg("port_mapping_create_request_callback called with unknown request_state object"); return; }
2794
2795 // calculate reply data length
2796 replyLen = sizeof(DNSServiceFlags);
2797 replyLen += 3 * sizeof(mDNSu32); // if index + addr + ttl
2798 replyLen += sizeof(DNSServiceErrorType);
2799 replyLen += 2 * sizeof(mDNSu16); // Internal Port + External Port
2800 replyLen += sizeof(mDNSu8); // protocol
2801
2802 rep = create_reply(port_mapping_reply_op, replyLen, request);
2803
2804 rep->rhdr->flags = dnssd_htonl(0);
2805 rep->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(m, n->InterfaceID));
2806 rep->rhdr->error = dnssd_htonl(n->Result);
2807
2808 data = rep->sdata;
2809
2810 *data++ = request->u.pm.NATinfo.ExternalAddress.b[0];
2811 *data++ = request->u.pm.NATinfo.ExternalAddress.b[1];
2812 *data++ = request->u.pm.NATinfo.ExternalAddress.b[2];
2813 *data++ = request->u.pm.NATinfo.ExternalAddress.b[3];
2814 *data++ = DNSServiceProtocol(request->u.pm.NATinfo.Protocol);
2815 *data++ = request->u.pm.NATinfo.IntPort.b[0];
2816 *data++ = request->u.pm.NATinfo.IntPort.b[1];
2817 *data++ = request->u.pm.NATinfo.ExternalPort.b[0];
2818 *data++ = request->u.pm.NATinfo.ExternalPort.b[1];
2819 put_uint32(request->u.pm.NATinfo.Lifetime, &data);
2820
2821 LogOperation("%3d: DNSServiceNATPortMappingCreate(%X, %u, %u, %d) RESULT %.4a:%u TTL %u", request->sd,
2822 DNSServiceProtocol(request->u.pm.NATinfo.Protocol),
2823 mDNSVal16(request->u.pm.NATinfo.IntPort), mDNSVal16(request->u.pm.ReqExt), request->u.pm.NATinfo.NATLease,
2824 &request->u.pm.NATinfo.ExternalAddress, mDNSVal16(request->u.pm.NATinfo.ExternalPort), request->u.pm.NATinfo.Lifetime);
2825
2826 append_reply(request, rep);
2827 }
2828
2829 mDNSlocal mStatus handle_port_mapping_request(request_state *request)
2830 {
2831 mDNSu32 ttl = 0;
2832 mStatus err = mStatus_NoError;
2833
2834 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
2835 mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend);
2836 mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex);
2837 mDNSu8 protocol = get_uint32(&request->msgptr, request->msgend);
2838 (void)flags; // Unused
2839 if (interfaceIndex && !InterfaceID) return(mStatus_BadParamErr);
2840 if (request->msgptr + 8 > request->msgend) request->msgptr = NULL;
2841 else
2842 {
2843 request->u.pm.NATinfo.IntPort.b[0] = *request->msgptr++;
2844 request->u.pm.NATinfo.IntPort.b[1] = *request->msgptr++;
2845 request->u.pm.ReqExt.b[0] = *request->msgptr++;
2846 request->u.pm.ReqExt.b[1] = *request->msgptr++;
2847 ttl = get_uint32(&request->msgptr, request->msgend);
2848 }
2849
2850 if (!request->msgptr) { LogMsg("%3d: DNSServiceNATPortMappingCreate(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
2851
2852 if (protocol == 0) // If protocol == 0 (i.e. just request public address) then IntPort, ExtPort, ttl must be zero too
2853 {
2854 if (!mDNSIPPortIsZero(request->u.pm.NATinfo.IntPort) || !mDNSIPPortIsZero(request->u.pm.ReqExt) || ttl) return(mStatus_BadParamErr);
2855 }
2856 else
2857 {
2858 if (mDNSIPPortIsZero(request->u.pm.NATinfo.IntPort)) return(mStatus_BadParamErr);
2859 if (!(protocol & (kDNSServiceProtocol_UDP | kDNSServiceProtocol_TCP))) return(mStatus_BadParamErr);
2860 }
2861
2862 request->u.pm.NATinfo.Protocol = !protocol ? NATOp_AddrRequest : (protocol == kDNSServiceProtocol_UDP) ? NATOp_MapUDP : NATOp_MapTCP;
2863 // u.pm.NATinfo.IntPort = already set above
2864 request->u.pm.NATinfo.RequestedPort = request->u.pm.ReqExt;
2865 request->u.pm.NATinfo.NATLease = ttl;
2866 request->u.pm.NATinfo.clientCallback = port_mapping_create_request_callback;
2867 request->u.pm.NATinfo.clientContext = request;
2868
2869 LogOperation("%3d: DNSServiceNATPortMappingCreate(%X, %u, %u, %d) START", request->sd,
2870 protocol, mDNSVal16(request->u.pm.NATinfo.IntPort), mDNSVal16(request->u.pm.ReqExt), request->u.pm.NATinfo.NATLease);
2871 err = mDNS_StartNATOperation(&mDNSStorage, &request->u.pm.NATinfo);
2872 if (!err) request->terminate = port_mapping_termination_callback;
2873
2874 return(err);
2875 }
2876
2877 // ***************************************************************************
2878 #if COMPILER_LIKES_PRAGMA_MARK
2879 #pragma mark -
2880 #pragma mark - DNSServiceGetAddrInfo
2881 #endif
2882
2883 mDNSlocal void addrinfo_termination_callback(request_state *request)
2884 {
2885 if (request->u.addrinfo.q4.QuestionContext)
2886 {
2887 mDNS_StopQuery(&mDNSStorage, &request->u.addrinfo.q4);
2888 request->u.addrinfo.q4.QuestionContext = mDNSNULL;
2889 }
2890
2891 if (request->u.addrinfo.q6.QuestionContext)
2892 {
2893 mDNS_StopQuery(&mDNSStorage, &request->u.addrinfo.q6);
2894 request->u.addrinfo.q6.QuestionContext = mDNSNULL;
2895 }
2896 }
2897
2898 mDNSlocal mStatus handle_addrinfo_request(request_state *request)
2899 {
2900 char hostname[256];
2901 domainname d;
2902 mStatus err = 0;
2903
2904 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
2905 mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend);
2906 request->u.addrinfo.interface_id = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex);
2907 if (interfaceIndex && !request->u.addrinfo.interface_id) return(mStatus_BadParamErr);
2908 request->u.addrinfo.flags = flags;
2909 request->u.addrinfo.protocol = get_uint32(&request->msgptr, request->msgend);
2910 if (request->u.addrinfo.protocol > (kDNSServiceProtocol_IPv4|kDNSServiceProtocol_IPv6))
2911 return(mStatus_BadParamErr);
2912
2913 if (get_string(&request->msgptr, request->msgend, hostname, 256) < 0) return(mStatus_BadParamErr);
2914
2915 if (!request->msgptr) { LogMsg("%3d: DNSServiceGetAddrInfo(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
2916
2917 if (!MakeDomainNameFromDNSNameString(&d, hostname))
2918 { LogMsg("ERROR: handle_addrinfo_request: bad hostname: %s", hostname); return(mStatus_BadParamErr); }
2919
2920 if (!request->u.addrinfo.protocol)
2921 {
2922 NetworkInterfaceInfo *i;
2923 if (IsLocalDomain(&d))
2924 {
2925 for (i = mDNSStorage.HostInterfaces; i; i = i->next)
2926 {
2927 if ((i->ip.type == mDNSAddrType_IPv4) && !mDNSIPv4AddressIsZero(i->ip.ip.v4)) request->u.addrinfo.protocol |= kDNSServiceProtocol_IPv4;
2928 else if ((i->ip.type == mDNSAddrType_IPv6) && !mDNSIPv6AddressIsZero(i->ip.ip.v6)) request->u.addrinfo.protocol |= kDNSServiceProtocol_IPv6;
2929 }
2930 }
2931 else
2932 {
2933 for (i = mDNSStorage.HostInterfaces; i; i = i->next)
2934 {
2935 if ((i->ip.type == mDNSAddrType_IPv4) && !mDNSv4AddressIsLinkLocal(&i->ip.ip.v4)) request->u.addrinfo.protocol |= kDNSServiceProtocol_IPv4;
2936 else if ((i->ip.type == mDNSAddrType_IPv6) && !mDNSv4AddressIsLinkLocal(&i->ip.ip.v6)) request->u.addrinfo.protocol |= kDNSServiceProtocol_IPv6;
2937 }
2938 }
2939 }
2940
2941 if (request->u.addrinfo.protocol & kDNSServiceProtocol_IPv4)
2942 {
2943 request->u.addrinfo.q4.InterfaceID = request->u.addrinfo.interface_id;
2944 request->u.addrinfo.q4.Target = zeroAddr;
2945 request->u.addrinfo.q4.qname = d;
2946 request->u.addrinfo.q4.qtype = kDNSServiceType_A;
2947 request->u.addrinfo.q4.qclass = kDNSServiceClass_IN;
2948 request->u.addrinfo.q4.LongLived = (flags & kDNSServiceFlagsLongLivedQuery ) != 0;
2949 request->u.addrinfo.q4.ExpectUnique = mDNSfalse;
2950 request->u.addrinfo.q4.ForceMCast = (flags & kDNSServiceFlagsForceMulticast ) != 0;
2951 request->u.addrinfo.q4.ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0;
2952 request->u.addrinfo.q4.QuestionCallback = queryrecord_result_callback;
2953 request->u.addrinfo.q4.QuestionContext = request;
2954
2955 err = mDNS_StartQuery(&mDNSStorage, &request->u.addrinfo.q4);
2956 if (err != mStatus_NoError)
2957 {
2958 LogMsg("ERROR: mDNS_StartQuery: %d", (int)err);
2959 request->u.addrinfo.q4.QuestionContext = mDNSNULL;
2960 }
2961 }
2962
2963 if (!err && (request->u.addrinfo.protocol & kDNSServiceProtocol_IPv6))
2964 {
2965 request->u.addrinfo.q6.InterfaceID = request->u.addrinfo.interface_id;
2966 request->u.addrinfo.q6.Target = zeroAddr;
2967 request->u.addrinfo.q6.qname = d;
2968 request->u.addrinfo.q6.qtype = kDNSServiceType_AAAA;
2969 request->u.addrinfo.q6.qclass = kDNSServiceClass_IN;
2970 request->u.addrinfo.q6.LongLived = (flags & kDNSServiceFlagsLongLivedQuery ) != 0;
2971 request->u.addrinfo.q6.ExpectUnique = mDNSfalse;
2972 request->u.addrinfo.q6.ForceMCast = (flags & kDNSServiceFlagsForceMulticast ) != 0;
2973 request->u.addrinfo.q6.ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0;
2974 request->u.addrinfo.q6.QuestionCallback = queryrecord_result_callback;
2975 request->u.addrinfo.q6.QuestionContext = request;
2976
2977 err = mDNS_StartQuery(&mDNSStorage, &request->u.addrinfo.q6);
2978 if (err != mStatus_NoError)
2979 {
2980 LogMsg("ERROR: mDNS_StartQuery: %d", (int)err);
2981 request->u.addrinfo.q6.QuestionContext = mDNSNULL;
2982 }
2983 }
2984
2985 LogOperation("%3d: DNSServiceGetAddrInfo(%##s) START", request->sd, d.c);
2986
2987 if (!err) request->terminate = addrinfo_termination_callback;
2988
2989 return(err);
2990 }
2991
2992 // ***************************************************************************
2993 #if COMPILER_LIKES_PRAGMA_MARK
2994 #pragma mark -
2995 #pragma mark - Main Request Handler etc.
2996 #endif
2997
2998 mDNSlocal request_state *NewRequest(void)
2999 {
3000 request_state **p = &all_requests;
3001 while (*p) p=&(*p)->next;
3002 *p = mallocL("request_state", sizeof(request_state));
3003 if (!*p) FatalError("ERROR: malloc");
3004 mDNSPlatformMemZero(*p, sizeof(request_state));
3005 return(*p);
3006 }
3007
3008 // read_msg may be called any time when the transfer state (req->ts) is t_morecoming.
3009 // returns the current state of the request (morecoming, error, complete, terminated.)
3010 // if there is no data on the socket, the socket will be closed and t_terminated will be returned
3011 // *** NOTE return value is actually ignored -- should change return type to void ***
3012 mDNSlocal int read_msg(request_state *req)
3013 {
3014 mDNSu32 nleft;
3015 int nread;
3016
3017 if (req->ts == t_terminated || req->ts == t_error)
3018 {
3019 LogMsg("ERROR: read_msg called with transfer state terminated or error");
3020 req->ts = t_error;
3021 return t_error;
3022 }
3023
3024 if (req->ts == t_complete) // this must be death or something is wrong
3025 {
3026 char buf[4]; // dummy for death notification
3027 nread = recv(req->sd, buf, 4, 0);
3028 if (!nread) { req->ts = t_terminated; return t_terminated; }
3029 if (nread < 0) goto rerror;
3030 LogMsg("ERROR: read data from a completed request.");
3031 req->ts = t_error;
3032 return t_error;
3033 }
3034
3035 if (req->ts != t_morecoming)
3036 {
3037 LogMsg("ERROR: read_msg called with invalid transfer state (%d)", req->ts);
3038 req->ts = t_error;
3039 return t_error;
3040 }
3041
3042 if (req->hdr_bytes < sizeof(ipc_msg_hdr))
3043 {
3044 nleft = sizeof(ipc_msg_hdr) - req->hdr_bytes;
3045 nread = recv(req->sd, (char *)&req->hdr + req->hdr_bytes, nleft, 0);
3046 if (nread == 0) { req->ts = t_terminated; return t_terminated; }
3047 if (nread < 0) goto rerror;
3048 req->hdr_bytes += nread;
3049 if (req->hdr_bytes > sizeof(ipc_msg_hdr))
3050 {
3051 LogMsg("ERROR: read_msg - read too many header bytes");
3052 req->ts = t_error;
3053 return t_error;
3054 }
3055
3056 // only read data if header is complete
3057 if (req->hdr_bytes == sizeof(ipc_msg_hdr))
3058 {
3059 ConvertHeaderBytes(&req->hdr);
3060 if (req->hdr.version != VERSION)
3061 { LogMsg("ERROR: client version 0x%08X daemon version 0x%08X", req->hdr.version, VERSION); req->ts = t_error; return t_error; }
3062
3063 // Largest conceivable single request is a DNSServiceRegisterRecord() or DNSServiceAddRecord()
3064 // with 64kB of rdata. Adding 1005 byte for a maximal domain name, plus a safety margin
3065 // for other overhead, this means any message above 70kB is definitely bogus.
3066 if (req->hdr.datalen > 70000)
3067 {
3068 LogMsg("ERROR: read_msg - hdr.datalen %lu (%X) > 70000", req->hdr.datalen, req->hdr.datalen);
3069 req->ts = t_error;
3070 return t_error;
3071 }
3072 req->msgbuf = mallocL("request_state msgbuf", req->hdr.datalen + MSG_PAD_BYTES);
3073 if (!req->msgbuf) { my_perror("ERROR: malloc"); req->ts = t_error; return t_error; }
3074 req->msgptr = req->msgbuf;
3075 req->msgend = req->msgbuf + req->hdr.datalen;
3076 mDNSPlatformMemZero(req->msgbuf, req->hdr.datalen + MSG_PAD_BYTES);
3077 }
3078 }
3079
3080 // If our header is complete, but we're still needing more body data, then try to read it now
3081 // Note: For cancel_request req->hdr.datalen == 0, but there's no error return socket for cancel_request
3082 // Any time we need to get the error return socket we know we'll have at least one data byte
3083 // (even if only the one-byte empty C string placeholder for the old ctrl_path parameter)
3084 if (req->hdr_bytes == sizeof(ipc_msg_hdr) && req->data_bytes < req->hdr.datalen)
3085 {
3086 nleft = req->hdr.datalen - req->data_bytes;
3087 struct iovec vec = { req->msgbuf + req->data_bytes, nleft }; // Tell recvmsg where we want the bytes put
3088 struct msghdr msg;
3089 struct cmsghdr *cmsg;
3090 char cbuf[sizeof(struct cmsghdr) + sizeof(dnssd_sock_t)];
3091 msg.msg_name = 0;
3092 msg.msg_namelen = 0;
3093 msg.msg_iov = &vec;
3094 msg.msg_iovlen = 1;
3095 msg.msg_control = cbuf;
3096 msg.msg_controllen = sizeof(cbuf);
3097 msg.msg_flags = 0;
3098 nread = recvmsg(req->sd, &msg, 0);
3099 if (nread == 0) { req->ts = t_terminated; return t_terminated; }
3100 if (nread < 0) goto rerror;
3101 req->data_bytes += nread;
3102 if (req->data_bytes > req->hdr.datalen)
3103 {
3104 LogMsg("ERROR: read_msg - read too many data bytes");
3105 req->ts = t_error;
3106 return t_error;
3107 }
3108 cmsg = CMSG_FIRSTHDR(&msg);
3109 if (msg.msg_controllen == sizeof(cbuf) &&
3110 cmsg->cmsg_len == sizeof(cbuf) &&
3111 cmsg->cmsg_level == SOL_SOCKET &&
3112 cmsg->cmsg_type == SCM_RIGHTS)
3113 {
3114 req->errsd = *(dnssd_sock_t *)CMSG_DATA(cmsg);
3115 if (req->data_bytes < req->hdr.datalen)
3116 {
3117 LogMsg("%3d: Client sent error socket %d via SCM_RIGHTS with req->data_bytes %d < req->hdr.datalen %d",
3118 req->sd, req->errsd, req->data_bytes, req->hdr.datalen);
3119 req->ts = t_error;
3120 return t_error;
3121 }
3122 }
3123 }
3124
3125 // If our header and data are both complete, see if we need to make our separate error return socket
3126 if (req->hdr_bytes == sizeof(ipc_msg_hdr) && req->data_bytes == req->hdr.datalen)
3127 {
3128 if (req->terminate && req->hdr.op != cancel_request)
3129 {
3130 dnssd_sockaddr_t cliaddr;
3131 #if defined(USE_TCP_LOOPBACK)
3132 mDNSOpaque16 port;
3133 port.b[0] = req->msgptr[0];
3134 port.b[1] = req->msgptr[1];
3135 req->msgptr += 2;
3136 cliaddr.sin_family = AF_INET;
3137 cliaddr.sin_port = port.NotAnInteger;
3138 cliaddr.sin_addr.s_addr = inet_addr(MDNS_TCP_SERVERADDR);
3139 #else
3140 char ctrl_path[MAX_CTLPATH];
3141 get_string(&req->msgptr, req->msgend, ctrl_path, MAX_CTLPATH); // path is first element in message buffer
3142 mDNSPlatformMemZero(&cliaddr, sizeof(cliaddr));
3143 cliaddr.sun_family = AF_LOCAL;
3144 mDNSPlatformStrCopy(cliaddr.sun_path, ctrl_path);
3145 // If the error return path UDS name is empty string, that tells us
3146 // that this is a new version of the library that's going to pass us
3147 // the error return path socket via sendmsg/recvmsg
3148 if (ctrl_path[0] == 0)
3149 {
3150 if (req->errsd == req->sd)
3151 { LogMsg("%3d: request_callback: ERROR failed to get errsd via SCM_RIGHTS", req->sd); req->ts = t_error; return t_error; }
3152 goto got_errfd;
3153 }
3154 #endif
3155
3156 req->errsd = socket(AF_DNSSD, SOCK_STREAM, 0);
3157 if (!dnssd_SocketValid(req->errsd)) { my_perror("ERROR: socket"); req->ts = t_error; return t_error; }
3158
3159 if (connect(req->errsd, (struct sockaddr *)&cliaddr, sizeof(cliaddr)) < 0)
3160 {
3161 #if !defined(USE_TCP_LOOPBACK)
3162 struct stat sb;
3163 LogMsg("request_callback: Couldn't connect to error return path socket “%s” errno %d %s",
3164 cliaddr.sun_path, dnssd_errno(), dnssd_strerror(dnssd_errno()));
3165 if (stat(cliaddr.sun_path, &sb) < 0)
3166 LogMsg("request_callback: stat failed “%s” errno %d %s", cliaddr.sun_path, dnssd_errno(), dnssd_strerror(dnssd_errno()));
3167 else
3168 LogMsg("request_callback: file “%s” mode %o (octal) uid %d gid %d", cliaddr.sun_path, sb.st_mode, sb.st_uid, sb.st_gid);
3169 #endif
3170 req->ts = t_error;
3171 return t_error;
3172 }
3173
3174 got_errfd:
3175 LogOperation("%3d: Using separate error socket %d", req->sd, req->errsd);
3176 #if defined(_WIN32)
3177 if (ioctlsocket(req->errsd, FIONBIO, &opt) != 0)
3178 #else
3179 if (fcntl(req->errsd, F_SETFL, fcntl(req->errsd, F_GETFL, 0) | O_NONBLOCK) != 0)
3180 #endif
3181 { my_perror("ERROR: could not set control socket to non-blocking mode"); req->ts = t_error; return t_error; }
3182 }
3183
3184 req->ts = t_complete;
3185 }
3186
3187 return req->ts;
3188
3189 rerror:
3190 if (dnssd_errno() == dnssd_EWOULDBLOCK || dnssd_errno() == dnssd_EINTR) return t_morecoming;
3191 my_perror("ERROR: read_msg");
3192 req->ts = t_error;
3193 return t_error;
3194 }
3195
3196 #define RecordOrientedOp(X) \
3197 ((X) == reg_record_request || (X) == add_record_request || (X) == update_record_request || (X) == remove_record_request)
3198
3199 // The lightweight operations are the ones that don't need a dedicated request_state structure allocated for them
3200 #define LightweightOp(X) (RecordOrientedOp(X) || (X) == cancel_request)
3201
3202 mDNSlocal void request_callback(int fd, short filter, void *info)
3203 {
3204 mStatus err = 0;
3205 request_state *req = info;
3206 #if defined(_WIN32)
3207 u_long opt = 1;
3208 #endif
3209 mDNSs32 min_size = sizeof(DNSServiceFlags);
3210 (void)fd; // Unused
3211 (void)filter; // Unused
3212
3213 read_msg(req);
3214 if (req->ts == t_morecoming) return;
3215 if (req->ts == t_terminated || req->ts == t_error) { AbortUnlinkAndFree(req); return; }
3216
3217 if (req->hdr.version != VERSION)
3218 {
3219 LogMsg("ERROR: client incompatible with daemon (client version = %d, "
3220 "daemon version = %d)\n", req->hdr.version, VERSION);
3221 AbortUnlinkAndFree(req);
3222 return;
3223 }
3224
3225 switch(req->hdr.op) // Interface + other data
3226 {
3227 case connection_request: min_size = 0; break;
3228 case reg_service_request: min_size += sizeof(mDNSu32) + 4 /* name, type, domain, host */ + 4 /* port, textlen */; break;
3229 case add_record_request: min_size += 4 /* type, rdlen */ + 4 /* ttl */; break;
3230 case update_record_request: min_size += 2 /* rdlen */ + 4 /* ttl */; break;
3231 case remove_record_request: break;
3232 case browse_request: min_size += sizeof(mDNSu32) + 2 /* type, domain */; break;
3233 case resolve_request: min_size += sizeof(mDNSu32) + 3 /* type, type, domain */; break;
3234 case query_request: min_size += sizeof(mDNSu32) + 1 /* name */ + 4 /* type, class*/; break;
3235 case enumeration_request: min_size += sizeof(mDNSu32); break;
3236 case reg_record_request: min_size += sizeof(mDNSu32) + 1 /* name */ + 6 /* type, class, rdlen */ + 4 /* ttl */; break;
3237 case reconfirm_record_request: min_size += sizeof(mDNSu32) + 1 /* name */ + 6 /* type, class, rdlen */; break;
3238 case setdomain_request: min_size += 1 /* domain */; break;
3239 case getproperty_request: min_size = 2; break;
3240 case port_mapping_request: min_size += sizeof(mDNSu32) + 4 /* udp/tcp */ + 4 /* int/ext port */ + 4 /* ttl */; break;
3241 case addrinfo_request: min_size += sizeof(mDNSu32) + 4 /* v4/v6 */ + 1 /* hostname */; break;
3242 case cancel_request: min_size = 0; break;
3243 default: LogMsg("ERROR: validate_message - unsupported req type: %d", req->hdr.op); min_size = -1; break;
3244 }
3245
3246 if ((mDNSs32)req->data_bytes < min_size)
3247 { LogMsg("Invalid message %d bytes; min for %d is %d", req->data_bytes, req->hdr.op, min_size); AbortUnlinkAndFree(req); return; }
3248
3249 if (LightweightOp(req->hdr.op) && !req->terminate)
3250 { LogMsg("Reg/Add/Update/Remove %d require existing connection", req->hdr.op); AbortUnlinkAndFree(req); return; }
3251
3252 // check if client wants silent operation
3253 if (req->hdr.ipc_flags & IPC_FLAGS_NOREPLY) req->no_reply = 1;
3254
3255 // If req->terminate is already set, this means this operation is sharing an existing connection
3256 if (req->terminate && !LightweightOp(req->hdr.op))
3257 {
3258 request_state *newreq = NewRequest();
3259 newreq->primary = req;
3260 newreq->sd = req->sd;
3261 newreq->errsd = req->errsd;
3262 newreq->uid = req->uid;
3263 newreq->hdr = req->hdr;
3264 newreq->msgbuf = req->msgbuf;
3265 newreq->msgptr = req->msgptr;
3266 newreq->msgend = req->msgend;
3267 req = newreq;
3268 }
3269
3270 switch(req->hdr.op)
3271 {
3272 // These are all operations that have their own first-class request_state object
3273 case connection_request:
3274 LogOperation("%3d: DNSServiceCreateConnection START", req->sd);
3275 req->terminate = connection_termination;
3276 break;
3277 case resolve_request: err = handle_resolve_request (req); break;
3278 case query_request: err = handle_queryrecord_request (req); break;
3279 case browse_request: err = handle_browse_request (req); break;
3280 case reg_service_request: err = handle_regservice_request (req); break;
3281 case enumeration_request: err = handle_enum_request (req); break;
3282 case reconfirm_record_request: err = handle_reconfirm_request (req); break;
3283 case setdomain_request: err = handle_setdomain_request (req); break;
3284 case getproperty_request: handle_getproperty_request (req); break;
3285 case port_mapping_request: err = handle_port_mapping_request(req); break;
3286 case addrinfo_request: err = handle_addrinfo_request (req); break;
3287
3288 // These are all operations that work with an existing request_state object
3289 case reg_record_request: err = handle_regrecord_request (req); break;
3290 case add_record_request: err = handle_add_request (req); break;
3291 case update_record_request: err = handle_update_request (req); break;
3292 case remove_record_request: err = handle_removerecord_request(req); break;
3293 case cancel_request: handle_cancel_request (req); break;
3294 default: LogMsg("%3d: ERROR: Unsupported UDS req: %d", req->sd, req->hdr.op);
3295 }
3296
3297 // req->msgbuf may be NULL, e.g. for connection_request or remove_record_request
3298 if (req->msgbuf) freeL("request_state msgbuf", req->msgbuf);
3299
3300 // There's no return data for a cancel request (DNSServiceRefDeallocate returns no result)
3301 // For a DNSServiceGetProperty call, the handler already generated the response,
3302 // so no need to do it again here
3303 if (req->hdr.op != cancel_request && req->hdr.op != getproperty_request)
3304 {
3305 err = dnssd_htonl(err);
3306 send_all(req->errsd, (const char *)&err, sizeof(err));
3307 if (req->errsd != req->sd)
3308 {
3309 LogOperation("%3d: Closing error socket %d", req->sd, req->errsd);
3310 dnssd_close(req->errsd);
3311 req->errsd = req->sd;
3312 // Also need to reset the parent's errsd, if this is a subbordinate operation
3313 if (req->primary) req->primary->errsd = req->primary->sd;
3314 }
3315 }
3316
3317 // Reset ready to accept the next req on this pipe
3318 if (req->primary) req = req->primary;
3319 req->ts = t_morecoming;
3320 req->hdr_bytes = 0;
3321 req->data_bytes = 0;
3322 req->msgbuf = mDNSNULL;
3323 req->msgptr = mDNSNULL;
3324 req->msgend = 0;
3325 }
3326
3327 mDNSlocal void connect_callback(int fd, short filter, void *info)
3328 {
3329 dnssd_sockaddr_t cliaddr;
3330 dnssd_socklen_t len = (dnssd_socklen_t) sizeof(cliaddr);
3331 dnssd_sock_t sd = accept(listenfd, (struct sockaddr*) &cliaddr, &len);
3332 const unsigned long optval = 1;
3333
3334 (void)fd; // Unused
3335 (void)filter; // Unused
3336 (void)info; // Unused
3337
3338 if (!dnssd_SocketValid(sd))
3339 {
3340 if (dnssd_errno() != dnssd_EWOULDBLOCK) my_perror("ERROR: accept");
3341 return;
3342 }
3343
3344 #ifdef SO_NOSIGPIPE
3345 // Some environments (e.g. OS X) support turning off SIGPIPE for a socket
3346 if (setsockopt(sd, SOL_SOCKET, SO_NOSIGPIPE, &optval, sizeof(optval)) < 0)
3347 {
3348 my_perror("ERROR: setsockopt - SO_NOSIGPIPE - aborting client");
3349 dnssd_close(sd);
3350 return;
3351 }
3352 #endif
3353
3354 #if defined(_WIN32)
3355 if (ioctlsocket(sd, FIONBIO, &optval) != 0)
3356 #else
3357 if (fcntl(sd, F_SETFL, fcntl(sd, F_GETFL, 0) | O_NONBLOCK) != 0)
3358 #endif
3359 {
3360 my_perror("ERROR: fcntl(sd, F_SETFL, O_NONBLOCK) - aborting client");
3361 dnssd_close(sd);
3362 return;
3363 }
3364 else
3365 {
3366 request_state *request = NewRequest();
3367 request->ts = t_morecoming;
3368 request->sd = sd;
3369 request->errsd = sd;
3370 #if APPLE_OSX_mDNSResponder
3371 struct xucred x;
3372 socklen_t len = sizeof(x);
3373 if (getsockopt(sd, 0, LOCAL_PEERCRED, &x, &len) >= 0 && x.cr_version == XUCRED_VERSION) request->uid = x.cr_uid;
3374 else my_perror("ERROR: getsockopt, LOCAL_PEERCRED");
3375 debugf("LOCAL_PEERCRED %d %u %u %d", len, x.cr_version, x.cr_uid, x.cr_ngroups);
3376 #endif APPLE_OSX_mDNSResponder
3377 LogOperation("%3d: Adding FD for uid %u", request->sd, request->uid);
3378 udsSupportAddFDToEventLoop(sd, request_callback, request);
3379 }
3380 }
3381
3382 mDNSexport int udsserver_init(dnssd_sock_t skt)
3383 {
3384 dnssd_sockaddr_t laddr;
3385 int ret;
3386 #if defined(_WIN32)
3387 u_long opt = 1;
3388 #endif
3389
3390 LogOperation("udsserver_init");
3391
3392 // If a particular platform wants to opt out of having a PID file, define PID_FILE to be ""
3393 if (PID_FILE[0])
3394 {
3395 FILE *fp = fopen(PID_FILE, "w");
3396 if (fp != NULL)
3397 {
3398 fprintf(fp, "%d\n", getpid());
3399 fclose(fp);
3400 }
3401 }
3402
3403 if (dnssd_SocketValid(skt))
3404 listenfd = skt;
3405 else
3406 {
3407 listenfd = socket(AF_DNSSD, SOCK_STREAM, 0);
3408 if (!dnssd_SocketValid(listenfd))
3409 {
3410 my_perror("ERROR: socket(AF_DNSSD, SOCK_STREAM, 0); failed");
3411 goto error;
3412 }
3413
3414 mDNSPlatformMemZero(&laddr, sizeof(laddr));
3415
3416 #if defined(USE_TCP_LOOPBACK)
3417 {
3418 laddr.sin_family = AF_INET;
3419 laddr.sin_port = htons(MDNS_TCP_SERVERPORT);
3420 laddr.sin_addr.s_addr = inet_addr(MDNS_TCP_SERVERADDR);
3421 ret = bind(listenfd, (struct sockaddr *) &laddr, sizeof(laddr));
3422 if (ret < 0)
3423 {
3424 my_perror("ERROR: bind(listenfd, (struct sockaddr *) &laddr, sizeof(laddr)); failed");
3425 goto error;
3426 }
3427 }
3428 #else
3429 {
3430 mode_t mask = umask(0);
3431 unlink(MDNS_UDS_SERVERPATH); //OK if this fails
3432 laddr.sun_family = AF_LOCAL;
3433 #ifndef NOT_HAVE_SA_LEN
3434 // According to Stevens (section 3.2), there is no portable way to
3435 // determine whether sa_len is defined on a particular platform.
3436 laddr.sun_len = sizeof(struct sockaddr_un);
3437 #endif
3438 mDNSPlatformStrCopy(laddr.sun_path, MDNS_UDS_SERVERPATH);
3439 ret = bind(listenfd, (struct sockaddr *) &laddr, sizeof(laddr));
3440 umask(mask);
3441 if (ret < 0)
3442 {
3443 my_perror("ERROR: bind(listenfd, (struct sockaddr *) &laddr, sizeof(laddr)); failed");
3444 goto error;
3445 }
3446 }
3447 #endif
3448 }
3449
3450 #if defined(_WIN32)
3451 // SEH: do we even need to do this on windows?
3452 // This socket will be given to WSAEventSelect which will automatically set it to non-blocking
3453 if (ioctlsocket(listenfd, FIONBIO, &opt) != 0)
3454 #else
3455 if (fcntl(listenfd, F_SETFL, fcntl(listenfd, F_GETFL, 0) | O_NONBLOCK) != 0)
3456 #endif
3457 {
3458 my_perror("ERROR: could not set listen socket to non-blocking mode");
3459 goto error;
3460 }
3461
3462 if (listen(listenfd, LISTENQ) != 0)
3463 {
3464 my_perror("ERROR: could not listen on listen socket");
3465 goto error;
3466 }
3467
3468 if (mStatus_NoError != udsSupportAddFDToEventLoop(listenfd, connect_callback, (void *) NULL))
3469 {
3470 my_perror("ERROR: could not add listen socket to event loop");
3471 goto error;
3472 }
3473 else LogOperation("%3d: Listening for incoming Unix Domain Socket client requests", listenfd);
3474
3475 #if !defined(PLATFORM_NO_RLIMIT)
3476 {
3477 // Set maximum number of open file descriptors
3478 #define MIN_OPENFILES 10240
3479 struct rlimit maxfds, newfds;
3480
3481 // Due to bugs in OS X (<rdar://problem/2941095>, <rdar://problem/3342704>, <rdar://problem/3839173>)
3482 // you have to get and set rlimits once before getrlimit will return sensible values
3483 if (getrlimit(RLIMIT_NOFILE, &maxfds) < 0) { my_perror("ERROR: Unable to get file descriptor limit"); return 0; }
3484 if (setrlimit(RLIMIT_NOFILE, &maxfds) < 0) my_perror("ERROR: Unable to set maximum file descriptor limit");
3485
3486 if (getrlimit(RLIMIT_NOFILE, &maxfds) < 0) { my_perror("ERROR: Unable to get file descriptor limit"); return 0; }
3487 newfds.rlim_max = (maxfds.rlim_max > MIN_OPENFILES) ? maxfds.rlim_max : MIN_OPENFILES;
3488 newfds.rlim_cur = (maxfds.rlim_cur > MIN_OPENFILES) ? maxfds.rlim_cur : MIN_OPENFILES;
3489 if (newfds.rlim_max != maxfds.rlim_max || newfds.rlim_cur != maxfds.rlim_cur)
3490 if (setrlimit(RLIMIT_NOFILE, &newfds) < 0) my_perror("ERROR: Unable to set maximum file descriptor limit");
3491
3492 if (getrlimit(RLIMIT_NOFILE, &maxfds) < 0) { my_perror("ERROR: Unable to get file descriptor limit"); return 0; }
3493 debugf("maxfds.rlim_max %d", (long)maxfds.rlim_max);
3494 debugf("maxfds.rlim_cur %d", (long)maxfds.rlim_cur);
3495 }
3496 #endif
3497
3498 // We start a "LocalOnly" query looking for Automatic Browse Domain records.
3499 // When Domain Enumeration in uDNS.c finds an "lb" record from the network, it creates a
3500 // "LocalOnly" record, which results in our AutomaticBrowseDomainChange callback being invoked
3501 mDNS_GetDomains(&mDNSStorage, &mDNSStorage.AutomaticBrowseDomainQ, mDNS_DomainTypeBrowseAutomatic,
3502 mDNSNULL, mDNSInterface_LocalOnly, AutomaticBrowseDomainChange, mDNSNULL);
3503
3504 RegisterLocalOnlyDomainEnumPTR(&mDNSStorage, &localdomain, mDNS_DomainTypeRegistration); // Add "local" as recommended registration domain ("dns-sd -E")
3505 RegisterLocalOnlyDomainEnumPTR(&mDNSStorage, &localdomain, mDNS_DomainTypeBrowse); // Add "local" as recommended browsing domain ("dns-sd -F")
3506 AddAutoBrowseDomain(0, &localdomain); // Add "local" as automatic browsing domain
3507
3508 udsserver_handle_configchange(&mDNSStorage);
3509 return 0;
3510
3511 error:
3512
3513 my_perror("ERROR: udsserver_init");
3514 return -1;
3515 }
3516
3517 mDNSexport int udsserver_exit(dnssd_sock_t skt)
3518 {
3519 // If the launching environment created no listening socket,
3520 // that means we created it ourselves, so we should clean it up on exit
3521 if (!dnssd_SocketValid(skt))
3522 {
3523 dnssd_close(listenfd);
3524 #if !defined(USE_TCP_LOOPBACK)
3525 // Currently, we're unable to remove /var/run/mdnsd because we've changed to userid "nobody"
3526 // to give up unnecessary privilege, but we need to be root to remove this Unix Domain Socket.
3527 // It would be nice if we could find a solution to this problem
3528 if (unlink(MDNS_UDS_SERVERPATH))
3529 debugf("Unable to remove %s", MDNS_UDS_SERVERPATH);
3530 #endif
3531 }
3532
3533 if (PID_FILE[0]) unlink(PID_FILE);
3534
3535 return 0;
3536 }
3537
3538 mDNSlocal void LogClientInfo(mDNS *const m, request_state *req)
3539 {
3540 if (!req->terminate)
3541 LogMsgNoIdent("%3d: No operation yet on this socket", req->sd);
3542 else if (req->terminate == connection_termination)
3543 {
3544 registered_record_entry *p;
3545 LogMsgNoIdent("%3d: DNSServiceCreateConnection", req->sd);
3546 for (p = req->u.reg_recs; p; p=p->next)
3547 LogMsgNoIdent(" -> DNSServiceRegisterRecord %3d %s", p->key, ARDisplayString(m, p->rr));
3548 }
3549 else if (req->terminate == regservice_termination_callback)
3550 {
3551 service_instance *ptr;
3552 for (ptr = req->u.servicereg.instances; ptr; ptr = ptr->next)
3553 LogMsgNoIdent("%3d: DNSServiceRegister %##s %u", req->sd, ptr->srs.RR_SRV.resrec.name->c, SRS_PORT(&ptr->srs));
3554 }
3555 else if (req->terminate == browse_termination_callback)
3556 {
3557 browser_t *blist;
3558 for (blist = req->u.browser.browsers; blist; blist = blist->next)
3559 LogMsgNoIdent("%3d: DNSServiceBrowse %##s", req->sd, blist->q.qname.c);
3560 }
3561 else if (req->terminate == resolve_termination_callback)
3562 LogMsgNoIdent("%3d: DNSServiceResolve %##s", req->sd, req->u.resolve.qsrv.qname.c);
3563 else if (req->terminate == queryrecord_termination_callback)
3564 LogMsgNoIdent("%3d: DNSServiceQueryRecord %##s (%s)", req->sd, req->u.queryrecord.q.qname.c, DNSTypeName(req->u.queryrecord.q.qtype));
3565 else if (req->terminate == enum_termination_callback)
3566 LogMsgNoIdent("%3d: DNSServiceEnumerateDomains %##s", req->sd, req->u.enumeration.q_all.qname.c);
3567 else if (req->terminate == port_mapping_termination_callback)
3568 LogMsgNoIdent("%3d: DNSServiceNATPortMapping %.4a %s%s Int %d Req %d Ext %d Req TTL %d Granted TTL %d",
3569 req->sd,
3570 &req->u.pm.NATinfo.ExternalAddress,
3571 req->u.pm.NATinfo.Protocol & NATOp_MapTCP ? "TCP" : " ",
3572 req->u.pm.NATinfo.Protocol & NATOp_MapUDP ? "UDP" : " ",
3573 mDNSVal16(req->u.pm.NATinfo.IntPort),
3574 mDNSVal16(req->u.pm.ReqExt),
3575 mDNSVal16(req->u.pm.NATinfo.ExternalPort),
3576 req->u.pm.NATinfo.NATLease,
3577 req->u.pm.NATinfo.Lifetime);
3578 else if (req->terminate == addrinfo_termination_callback)
3579 LogMsgNoIdent("%3d: DNSServiceGetAddrInfo %s%s %##s", req->sd,
3580 req->u.addrinfo.protocol & kDNSServiceProtocol_IPv4 ? "v4" : " ",
3581 req->u.addrinfo.protocol & kDNSServiceProtocol_IPv6 ? "v6" : " ",
3582 req->u.addrinfo.q4.qname.c);
3583 else
3584 LogMsgNoIdent("%3d: Unrecognized operation %p", req->sd, req->terminate);
3585 }
3586
3587 mDNSexport void udsserver_info(mDNS *const m)
3588 {
3589 mDNSs32 now = mDNS_TimeNow(m);
3590 mDNSu32 CacheUsed = 0, CacheActive = 0;
3591 mDNSu32 slot;
3592 CacheGroup *cg;
3593 CacheRecord *cr;
3594
3595 LogMsgNoIdent("Timenow 0x%08lX (%ld)", (mDNSu32)now, now);
3596 LogMsgNoIdent("------------ Cache -------------");
3597
3598 LogMsgNoIdent("Slt Q TTL if U Type rdlen");
3599 for (slot = 0; slot < CACHE_HASH_SLOTS; slot++)
3600 for (cg = m->rrcache_hash[slot]; cg; cg=cg->next)
3601 {
3602 CacheUsed++; // Count one cache entity for the CacheGroup object
3603 for (cr = cg->members; cr; cr=cr->next)
3604 {
3605 mDNSs32 remain = cr->resrec.rroriginalttl - (now - cr->TimeRcvd) / mDNSPlatformOneSecond;
3606 NetworkInterfaceInfo *info = (NetworkInterfaceInfo *)cr->resrec.InterfaceID;
3607 CacheUsed++;
3608 if (cr->CRActiveQuestion) CacheActive++;
3609 LogMsgNoIdent("%3d %s%8ld %-6s%s %-6s%s",
3610 slot,
3611 cr->CRActiveQuestion ? "*" : " ",
3612 remain,
3613 info ? info->ifname : "-U-",
3614 (cr->resrec.RecordType == kDNSRecordTypePacketNegative) ? "-" :
3615 (cr->resrec.RecordType & kDNSRecordTypePacketUniqueMask) ? " " : "+",
3616 DNSTypeName(cr->resrec.rrtype),
3617 CRDisplayString(m, cr));
3618 usleep(1000); // Limit rate a little so we don't flood syslog too fast
3619 }
3620 }
3621
3622 if (m->rrcache_totalused != CacheUsed)
3623 LogMsgNoIdent("Cache use mismatch: rrcache_totalused is %lu, true count %lu", m->rrcache_totalused, CacheUsed);
3624 if (m->rrcache_active != CacheActive)
3625 LogMsgNoIdent("Cache use mismatch: rrcache_active is %lu, true count %lu", m->rrcache_active, CacheActive);
3626 LogMsgNoIdent("Cache currently contains %lu entities; %lu referenced by active questions", CacheUsed, CacheActive);
3627
3628 LogMsgNoIdent("--------- Auth Records ---------");
3629 if (!m->ResourceRecords) LogMsgNoIdent("<None>");
3630 else
3631 {
3632 AuthRecord *ar;
3633 for (ar = m->ResourceRecords; ar; ar=ar->next)
3634 LogMsgNoIdent("%s", ARDisplayString(m, ar));
3635 }
3636
3637 LogMsgNoIdent("---------- Questions -----------");
3638 if (!m->Questions) LogMsgNoIdent("<None>");
3639 else
3640 {
3641 DNSQuestion *q;
3642 CacheUsed = 0;
3643 CacheActive = 0;
3644 LogMsgNoIdent(" Int Next if T NumAns Type Name");
3645 for (q = m->Questions; q; q=q->next)
3646 {
3647 mDNSs32 i = q->ThisQInterval / mDNSPlatformOneSecond;
3648 mDNSs32 n = (q->LastQTime + q->ThisQInterval - now) / mDNSPlatformOneSecond;
3649 NetworkInterfaceInfo *info = (NetworkInterfaceInfo *)q->InterfaceID;
3650 CacheUsed++;
3651 if (q->ThisQInterval) CacheActive++;
3652 LogMsgNoIdent("%6d%6d %-6s%s %5d %-6s%##s%s",
3653 i, n,
3654 info ? info->ifname : mDNSOpaque16IsZero(q->TargetQID) ? "" : "-U-",
3655 mDNSOpaque16IsZero(q->TargetQID) ? " " : q->LongLived ? "L" : "O", // mDNS, long-lived, or one-shot query?
3656 q->CurrentAnswers,
3657 DNSTypeName(q->qtype), q->qname.c, q->DuplicateOf ? " (dup)" : "");
3658 usleep(1000); // Limit rate a little so we don't flood syslog too fast
3659 }
3660 LogMsgNoIdent("%lu question%s; %lu active", CacheUsed, CacheUsed > 1 ? "s" : "", CacheActive);
3661 }
3662
3663 LogMsgNoIdent("---- Active Client Requests ----");
3664 if (!all_requests) LogMsgNoIdent("<None>");
3665 else
3666 {
3667 request_state *req;
3668 for (req = all_requests; req; req=req->next)
3669 LogClientInfo(m, req);
3670 }
3671
3672 LogMsgNoIdent("-------- NAT Traversals --------");
3673 if (!m->NATTraversals) LogMsgNoIdent("<None>");
3674 else
3675 {
3676 NATTraversalInfo *nat;
3677 for (nat = m->NATTraversals; nat; nat=nat->next)
3678 {
3679 if (nat->Protocol)
3680 LogMsgNoIdent("%p %s Int %5d Ext %5d Err %d Retry %d Interval %d Expire %d",
3681 nat, nat->Protocol == NATOp_MapTCP ? "TCP" : "UDP",
3682 mDNSVal16(nat->IntPort), mDNSVal16(nat->ExternalPort), nat->Result,
3683 nat->retryPortMap ? (nat->retryPortMap - now) / mDNSPlatformOneSecond : 0,
3684 nat->retryInterval / mDNSPlatformOneSecond,
3685 nat->ExpiryTime ? (nat->ExpiryTime - now) / mDNSPlatformOneSecond : 0);
3686 else
3687 LogMsgNoIdent("%p Address Request Retry %d Interval %d", nat,
3688 (m->retryGetAddr - now) / mDNSPlatformOneSecond,
3689 m->retryIntervalGetAddr / mDNSPlatformOneSecond);
3690 }
3691 }
3692
3693 LogMsgNoIdent("--------- AuthInfoList ---------");
3694 if (!m->AuthInfoList) LogMsgNoIdent("<None>");
3695 else
3696 {
3697 DomainAuthInfo *a;
3698 for (a = m->AuthInfoList; a; a = a->next)
3699 LogMsgNoIdent("%##s %##s%s", a->domain.c, a->keyname.c, a->AutoTunnel ? " AutoTunnel" : "");
3700 }
3701
3702 #if APPLE_OSX_mDNSResponder
3703 LogMsgNoIdent("--------- TunnelClients ---------");
3704 if (!m->TunnelClients) LogMsgNoIdent("<None>");
3705 else
3706 {
3707 ClientTunnel *c;
3708 for (c = m->TunnelClients; c; c = c->next)
3709 LogMsgNoIdent("%##s local %.16a %.4a remote %.16a %.4a %5d interval %d", c->dstname.c, &c->loc_inner, &c->loc_outer, &c->rmt_inner, &c->rmt_outer, mDNSVal16(c->rmt_outer_port), c->q.ThisQInterval);
3710 }
3711 #endif
3712 }
3713
3714 #if APPLE_OSX_mDNSResponder && MACOSX_MDNS_MALLOC_DEBUGGING
3715 mDNSexport void uds_validatelists(void)
3716 {
3717 request_state *req;
3718 for (req = all_requests; req; req=req->next)
3719 {
3720 if (req->next == (request_state *)~0 || (req->sd < 0 && req->sd != -2))
3721 LogMemCorruption("UDS request list: %p is garbage (%d)", req, req->sd);
3722
3723 reply_state *rep;
3724 for (rep = req->replies; rep; rep=rep->next)
3725 if (rep->next == (reply_state *)~0)
3726 LogMemCorruption("UDS req->replies: %p is garbage", rep);
3727
3728 if (req->terminate == connection_termination)
3729 {
3730 registered_record_entry *p;
3731 for (p = req->u.reg_recs; p; p=p->next)
3732 if (p->next == (registered_record_entry *)~0)
3733 LogMemCorruption("UDS req->u.reg_recs: %p is garbage", p);
3734 }
3735 else if (req->terminate == regservice_termination_callback)
3736 {
3737 service_instance *p;
3738 for (p = req->u.servicereg.instances; p; p=p->next)
3739 if (p->next == (service_instance *)~0)
3740 LogMemCorruption("UDS req->u.servicereg.instances: %p is garbage", p);
3741 }
3742 else if (req->terminate == browse_termination_callback)
3743 {
3744 browser_t *p;
3745 for (p = req->u.browser.browsers; p; p=p->next)
3746 if (p->next == (browser_t *)~0)
3747 LogMemCorruption("UDS req->u.browser.browsers: %p is garbage", p);
3748 }
3749 }
3750
3751 DNameListElem *d;
3752 for (d = SCPrefBrowseDomains; d; d=d->next)
3753 if (d->next == (DNameListElem *)~0 || d->name.c[0] > 63)
3754 LogMemCorruption("SCPrefBrowseDomains: %p is garbage (%d)", d, d->name.c[0]);
3755
3756 ARListElem *b;
3757 for (b = LocalDomainEnumRecords; b; b=b->next)
3758 if (b->next == (ARListElem *)~0 || b->ar.resrec.name->c[0] > 63)
3759 LogMemCorruption("LocalDomainEnumRecords: %p is garbage (%d)", b, b->ar.resrec.name->c[0]);
3760
3761 for (d = AutoBrowseDomains; d; d=d->next)
3762 if (d->next == (DNameListElem *)~0 || d->name.c[0] > 63)
3763 LogMemCorruption("AutoBrowseDomains: %p is garbage (%d)", d, d->name.c[0]);
3764
3765 for (d = AutoRegistrationDomains; d; d=d->next)
3766 if (d->next == (DNameListElem *)~0 || d->name.c[0] > 63)
3767 LogMemCorruption("AutoRegistrationDomains: %p is garbage (%d)", d, d->name.c[0]);
3768 }
3769 #endif
3770
3771 mDNSlocal int send_msg(reply_state *rep)
3772 {
3773 ssize_t nwriten;
3774 if (!rep->msgbuf) { LogMsg("ERROR: send_msg called with NULL message buffer"); return(rep->ts = t_error); }
3775 if (rep->request->no_reply) { freeL("reply_state msgbuf (no_reply)", rep->msgbuf); return(rep->ts = t_complete); }
3776
3777 ConvertHeaderBytes(rep->mhdr);
3778 nwriten = send(rep->sd, rep->msgbuf + rep->nwriten, rep->len - rep->nwriten, 0);
3779 ConvertHeaderBytes(rep->mhdr);
3780
3781 if (nwriten < 0)
3782 {
3783 if (dnssd_errno() == dnssd_EINTR || dnssd_errno() == dnssd_EWOULDBLOCK) nwriten = 0;
3784 else
3785 {
3786 #if !defined(PLATFORM_NO_EPIPE)
3787 if (dnssd_errno() == EPIPE)
3788 return(rep->request->ts = rep->ts = t_terminated);
3789 else
3790 #endif
3791 {
3792 LogMsg("send_msg ERROR: failed to write %d bytes to fd %d errno %d %s",
3793 rep->len - rep->nwriten, rep->sd, dnssd_errno(), dnssd_strerror(dnssd_errno()));
3794 return(rep->ts = t_error);
3795 }
3796 }
3797 }
3798 rep->nwriten += nwriten;
3799 if (rep->nwriten == rep->len) { freeL("reply_state msgbuf (t_complete)", rep->msgbuf); rep->ts = t_complete; }
3800 return rep->ts;
3801 }
3802
3803 mDNSexport mDNSs32 udsserver_idle(mDNSs32 nextevent)
3804 {
3805 mDNSs32 now = mDNS_TimeNow(&mDNSStorage);
3806 request_state **req = &all_requests;
3807
3808 while (*req)
3809 {
3810 while ((*req)->replies) // Send queued replies
3811 {
3812 transfer_state result;
3813 if ((*req)->replies->next) (*req)->replies->rhdr->flags |= dnssd_htonl(kDNSServiceFlagsMoreComing);
3814 result = send_msg((*req)->replies); // Returns t_morecoming if buffer full because client is not reading
3815 if (result == t_complete)
3816 {
3817 reply_state *fptr = (*req)->replies;
3818 (*req)->replies = (*req)->replies->next;
3819 freeL("reply_state/udsserver_idle", fptr);
3820 (*req)->time_blocked = 0; // reset failure counter after successful send
3821 continue;
3822 }
3823 else if (result == t_terminated || result == t_error) abort_request(*req);
3824 break;
3825 }
3826
3827 if ((*req)->replies) // If we failed to send everything, check our time_blocked timer
3828 {
3829 if (!(*req)->time_blocked) (*req)->time_blocked = NonZeroTime(now);
3830 if (now - (*req)->time_blocked >= 60 * mDNSPlatformOneSecond)
3831 {
3832 LogMsg("Could not write data to client %d after %ld seconds - aborting connection",
3833 (*req)->sd, (now - (*req)->time_blocked) / mDNSPlatformOneSecond);
3834 LogClientInfo(&mDNSStorage, *req);
3835 abort_request(*req);
3836 }
3837 else if (nextevent - now > mDNSPlatformOneSecond) nextevent = now + mDNSPlatformOneSecond;
3838 }
3839
3840 if (!dnssd_SocketValid((*req)->sd)) // If this request is finished, unlink it from the list and free the memory
3841 {
3842 // Since we're already doing a list traversal, we unlink the request directly instead of using AbortUnlinkAndFree()
3843 request_state *tmp = *req;
3844 *req = tmp->next;
3845 freeL("request_state/udsserver_idle", tmp);
3846 }
3847 else
3848 req = &(*req)->next;
3849 }
3850 return nextevent;
3851 }
3852
3853 struct CompileTimeAssertionChecks_uds_daemon
3854 {
3855 // Check our structures are reasonable sizes. Including overly-large buffers, or embedding
3856 // other overly-large structures instead of having a pointer to them, can inadvertently
3857 // cause structure sizes (and therefore memory usage) to balloon unreasonably.
3858 char sizecheck_request_state [(sizeof(request_state) <= 1800) ? 1 : -1];
3859 char sizecheck_registered_record_entry[(sizeof(registered_record_entry) <= 30) ? 1 : -1];
3860 char sizecheck_service_instance [(sizeof(service_instance) <= 6000) ? 1 : -1];
3861 char sizecheck_browser_t [(sizeof(browser_t) <= 1000) ? 1 : -1];
3862 char sizecheck_reply_hdr [(sizeof(reply_hdr) <= 20) ? 1 : -1];
3863 char sizecheck_reply_state [(sizeof(reply_state) <= 40) ? 1 : -1];
3864 };