dsgetdcname: remove invalid assumption in discover_dc_dns().
[Samba.git] / source / libsmb / dsgetdcname.c
blob981994ca43fe2a26f109bba8b9f16f09be86488d
1 /*
2 Unix SMB/CIFS implementation.
4 dsgetdcname
6 Copyright (C) Gerald Carter 2006
7 Copyright (C) Guenther Deschner 2007-2008
9 This program is free software; you can redistribute it and/or modify
10 it under the terms of the GNU General Public License as published by
11 the Free Software Foundation; either version 3 of the License, or
12 (at your option) any later version.
14 This program is distributed in the hope that it will be useful,
15 but WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 GNU General Public License for more details.
19 You should have received a copy of the GNU General Public License
20 along with this program. If not, see <http://www.gnu.org/licenses/>.
23 #include "includes.h"
25 #define DSGETDCNAME_FMT "DSGETDCNAME/DOMAIN/%s"
26 /* 15 minutes */
27 #define DSGETDCNAME_CACHE_TTL 60*15
29 struct ip_service_name {
30 struct sockaddr_storage ss;
31 unsigned port;
32 const char *hostname;
35 /****************************************************************
36 ****************************************************************/
38 void debug_dsdcinfo_flags(int lvl, uint32_t flags)
40 DEBUG(lvl,("debug_dsdcinfo_flags: 0x%08x\n\t", flags));
42 if (flags & DS_FORCE_REDISCOVERY)
43 DEBUGADD(lvl,("DS_FORCE_REDISCOVERY "));
44 if (flags & 0x000000002)
45 DEBUGADD(lvl,("0x00000002 "));
46 if (flags & 0x000000004)
47 DEBUGADD(lvl,("0x00000004 "));
48 if (flags & 0x000000008)
49 DEBUGADD(lvl,("0x00000008 "));
50 if (flags & DS_DIRECTORY_SERVICE_REQUIRED)
51 DEBUGADD(lvl,("DS_DIRECTORY_SERVICE_REQUIRED "));
52 if (flags & DS_DIRECTORY_SERVICE_PREFERRED)
53 DEBUGADD(lvl,("DS_DIRECTORY_SERVICE_PREFERRED "));
54 if (flags & DS_GC_SERVER_REQUIRED)
55 DEBUGADD(lvl,("DS_GC_SERVER_REQUIRED "));
56 if (flags & DS_PDC_REQUIRED)
57 DEBUGADD(lvl,("DS_PDC_REQUIRED "));
58 if (flags & DS_BACKGROUND_ONLY)
59 DEBUGADD(lvl,("DS_BACKGROUND_ONLY "));
60 if (flags & DS_IP_REQUIRED)
61 DEBUGADD(lvl,("DS_IP_REQUIRED "));
62 if (flags & DS_KDC_REQUIRED)
63 DEBUGADD(lvl,("DS_KDC_REQUIRED "));
64 if (flags & DS_TIMESERV_REQUIRED)
65 DEBUGADD(lvl,("DS_TIMESERV_REQUIRED "));
66 if (flags & DS_WRITABLE_REQUIRED)
67 DEBUGADD(lvl,("DS_WRITABLE_REQUIRED "));
68 if (flags & DS_GOOD_TIMESERV_PREFERRED)
69 DEBUGADD(lvl,("DS_GOOD_TIMESERV_PREFERRED "));
70 if (flags & DS_AVOID_SELF)
71 DEBUGADD(lvl,("DS_AVOID_SELF "));
72 if (flags & DS_ONLY_LDAP_NEEDED)
73 DEBUGADD(lvl,("DS_ONLY_LDAP_NEEDED "));
74 if (flags & DS_IS_FLAT_NAME)
75 DEBUGADD(lvl,("DS_IS_FLAT_NAME "));
76 if (flags & DS_IS_DNS_NAME)
77 DEBUGADD(lvl,("DS_IS_DNS_NAME "));
78 if (flags & 0x00040000)
79 DEBUGADD(lvl,("0x00040000 "));
80 if (flags & 0x00080000)
81 DEBUGADD(lvl,("0x00080000 "));
82 if (flags & 0x00100000)
83 DEBUGADD(lvl,("0x00100000 "));
84 if (flags & 0x00200000)
85 DEBUGADD(lvl,("0x00200000 "));
86 if (flags & 0x00400000)
87 DEBUGADD(lvl,("0x00400000 "));
88 if (flags & 0x00800000)
89 DEBUGADD(lvl,("0x00800000 "));
90 if (flags & 0x01000000)
91 DEBUGADD(lvl,("0x01000000 "));
92 if (flags & 0x02000000)
93 DEBUGADD(lvl,("0x02000000 "));
94 if (flags & 0x04000000)
95 DEBUGADD(lvl,("0x04000000 "));
96 if (flags & 0x08000000)
97 DEBUGADD(lvl,("0x08000000 "));
98 if (flags & 0x10000000)
99 DEBUGADD(lvl,("0x10000000 "));
100 if (flags & 0x20000000)
101 DEBUGADD(lvl,("0x20000000 "));
102 if (flags & DS_RETURN_DNS_NAME)
103 DEBUGADD(lvl,("DS_RETURN_DNS_NAME "));
104 if (flags & DS_RETURN_FLAT_NAME)
105 DEBUGADD(lvl,("DS_RETURN_FLAT_NAME "));
106 if (flags)
107 DEBUGADD(lvl,("\n"));
110 /****************************************************************
111 ****************************************************************/
113 static char *dsgetdcname_cache_key(TALLOC_CTX *mem_ctx, const char *domain)
115 if (!mem_ctx || !domain) {
116 return NULL;
119 return talloc_asprintf_strupper_m(mem_ctx, DSGETDCNAME_FMT, domain);
122 /****************************************************************
123 ****************************************************************/
125 static NTSTATUS dsgetdcname_cache_delete(TALLOC_CTX *mem_ctx,
126 const char *domain_name)
128 char *key;
130 if (!gencache_init()) {
131 return NT_STATUS_INTERNAL_DB_ERROR;
134 key = dsgetdcname_cache_key(mem_ctx, domain_name);
135 if (!key) {
136 return NT_STATUS_NO_MEMORY;
139 if (!gencache_del(key)) {
140 return NT_STATUS_UNSUCCESSFUL;
143 return NT_STATUS_OK;
146 /****************************************************************
147 ****************************************************************/
149 static NTSTATUS dsgetdcname_cache_store(TALLOC_CTX *mem_ctx,
150 const char *domain_name,
151 struct netr_DsRGetDCNameInfo *info)
153 time_t expire_time;
154 char *key;
155 bool ret = false;
156 DATA_BLOB blob;
157 enum ndr_err_code ndr_err;
159 if (!gencache_init()) {
160 return NT_STATUS_INTERNAL_DB_ERROR;
163 key = dsgetdcname_cache_key(mem_ctx, domain_name);
164 if (!key) {
165 return NT_STATUS_NO_MEMORY;
168 expire_time = time(NULL) + DSGETDCNAME_CACHE_TTL;
170 ndr_err = ndr_push_struct_blob(&blob, mem_ctx, info,
171 (ndr_push_flags_fn_t)ndr_push_netr_DsRGetDCNameInfo);
172 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
173 return ndr_map_error2ntstatus(ndr_err);
176 if (gencache_lock_entry(key) != 0) {
177 data_blob_free(&blob);
178 return NT_STATUS_LOCK_NOT_GRANTED;
181 ret = gencache_set_data_blob(key, &blob, expire_time);
182 data_blob_free(&blob);
184 gencache_unlock_entry(key);
186 return ret ? NT_STATUS_OK : NT_STATUS_UNSUCCESSFUL;
189 /****************************************************************
190 ****************************************************************/
192 static NTSTATUS dsgetdcname_cache_refresh(TALLOC_CTX *mem_ctx,
193 const char *domain_name,
194 struct GUID *domain_guid,
195 uint32_t flags,
196 const char *site_name,
197 struct netr_DsRGetDCNameInfo *info)
199 uint32_t nt_version = NETLOGON_VERSION_1;
201 /* check if matching entry is older then 15 minutes, if yes, send
202 * CLDAP/MAILSLOT ping again and store the cached data */
204 if (ads_cldap_netlogon(mem_ctx, info->dc_unc,
205 info->domain_name, &nt_version, NULL)) {
207 dsgetdcname_cache_delete(mem_ctx, domain_name);
209 return dsgetdcname_cache_store(mem_ctx,
210 info->domain_name,
211 info);
214 return NT_STATUS_INVALID_NETWORK_RESPONSE;
217 /****************************************************************
218 ****************************************************************/
220 static uint32_t get_cldap_reply_server_flags(union nbt_cldap_netlogon *r,
221 uint32_t nt_version)
223 switch (nt_version & 0x000000ff) {
224 case 0:
225 case 1:
226 return 0;
227 case 2:
228 case 3:
229 return r->logon3.server_type;
230 case 4:
231 case 5:
232 case 6:
233 case 7:
234 return r->logon5.server_type;
235 case 8:
236 case 9:
237 case 10:
238 case 11:
239 case 12:
240 case 13:
241 case 14:
242 case 15:
243 return r->logon13.server_type;
244 default:
245 return r->logon29.server_type;
249 /****************************************************************
250 ****************************************************************/
252 #define RETURN_ON_FALSE(x) if (!x) return false;
254 static bool check_cldap_reply_required_flags(uint32_t ret_flags,
255 uint32_t req_flags)
257 if (ret_flags == 0) {
258 return true;
261 if (req_flags & DS_PDC_REQUIRED)
262 RETURN_ON_FALSE(ret_flags & NBT_SERVER_PDC);
264 if (req_flags & DS_GC_SERVER_REQUIRED)
265 RETURN_ON_FALSE(ret_flags & NBT_SERVER_GC);
267 if (req_flags & DS_ONLY_LDAP_NEEDED)
268 RETURN_ON_FALSE(ret_flags & NBT_SERVER_LDAP);
270 if ((req_flags & DS_DIRECTORY_SERVICE_REQUIRED) ||
271 (req_flags & DS_DIRECTORY_SERVICE_PREFERRED))
272 RETURN_ON_FALSE(ret_flags & NBT_SERVER_DS);
274 if (req_flags & DS_KDC_REQUIRED)
275 RETURN_ON_FALSE(ret_flags & NBT_SERVER_KDC);
277 if (req_flags & DS_TIMESERV_REQUIRED)
278 RETURN_ON_FALSE(ret_flags & NBT_SERVER_TIMESERV);
280 if (req_flags & DS_WRITABLE_REQUIRED)
281 RETURN_ON_FALSE(ret_flags & NBT_SERVER_WRITABLE);
283 return true;
286 /****************************************************************
287 ****************************************************************/
289 static NTSTATUS dsgetdcname_cache_fetch(TALLOC_CTX *mem_ctx,
290 const char *domain_name,
291 struct GUID *domain_guid,
292 uint32_t flags,
293 const char *site_name,
294 struct netr_DsRGetDCNameInfo **info_p,
295 bool *expired)
297 char *key;
298 DATA_BLOB blob;
299 enum ndr_err_code ndr_err;
300 struct netr_DsRGetDCNameInfo *info;
302 if (!gencache_init()) {
303 return NT_STATUS_INTERNAL_DB_ERROR;
306 key = dsgetdcname_cache_key(mem_ctx, domain_name);
307 if (!key) {
308 return NT_STATUS_NO_MEMORY;
311 if (!gencache_get_data_blob(key, &blob, expired)) {
312 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
315 info = TALLOC_ZERO_P(mem_ctx, struct netr_DsRGetDCNameInfo);
316 if (!info) {
317 return NT_STATUS_NO_MEMORY;
320 ndr_err = ndr_pull_struct_blob(&blob, mem_ctx, info,
321 (ndr_pull_flags_fn_t)ndr_pull_netr_DsRGetDCNameInfo);
323 data_blob_free(&blob);
324 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
325 dsgetdcname_cache_delete(mem_ctx, domain_name);
326 return ndr_map_error2ntstatus(ndr_err);
329 if (DEBUGLEVEL >= 10) {
330 NDR_PRINT_DEBUG(netr_DsRGetDCNameInfo, info);
333 /* check flags */
334 if (!check_cldap_reply_required_flags(info->dc_flags, flags)) {
335 DEBUG(10,("invalid flags\n"));
336 return NT_STATUS_INVALID_PARAMETER;
339 if ((flags & DS_IP_REQUIRED) &&
340 (info->dc_address_type != DS_ADDRESS_TYPE_INET)) {
341 return NT_STATUS_INVALID_PARAMETER_MIX;
344 *info_p = info;
346 return NT_STATUS_OK;
349 /****************************************************************
350 ****************************************************************/
352 static NTSTATUS dsgetdcname_cached(TALLOC_CTX *mem_ctx,
353 const char *domain_name,
354 struct GUID *domain_guid,
355 uint32_t flags,
356 const char *site_name,
357 struct netr_DsRGetDCNameInfo **info)
359 NTSTATUS status;
360 bool expired = false;
362 status = dsgetdcname_cache_fetch(mem_ctx, domain_name, domain_guid,
363 flags, site_name, info, &expired);
364 if (!NT_STATUS_IS_OK(status)) {
365 DEBUG(10,("dsgetdcname_cached: cache fetch failed with: %s\n",
366 nt_errstr(status)));
367 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
370 if (flags & DS_BACKGROUND_ONLY) {
371 return status;
374 if (expired) {
375 status = dsgetdcname_cache_refresh(mem_ctx, domain_name,
376 domain_guid, flags,
377 site_name, *info);
378 if (!NT_STATUS_IS_OK(status)) {
379 return status;
383 return status;
386 /****************************************************************
387 ****************************************************************/
389 static bool check_allowed_required_flags(uint32_t flags)
391 uint32_t return_type = flags & (DS_RETURN_FLAT_NAME|DS_RETURN_DNS_NAME);
392 uint32_t offered_type = flags & (DS_IS_FLAT_NAME|DS_IS_DNS_NAME);
393 uint32_t query_type = flags & (DS_BACKGROUND_ONLY|DS_FORCE_REDISCOVERY);
395 /* FIXME: check for DSGETDC_VALID_FLAGS and check for excluse bits
396 * (DS_PDC_REQUIRED, DS_KDC_REQUIRED, DS_GC_SERVER_REQUIRED) */
398 debug_dsdcinfo_flags(10, flags);
400 if (return_type == (DS_RETURN_FLAT_NAME|DS_RETURN_DNS_NAME)) {
401 return false;
404 if (offered_type == (DS_IS_DNS_NAME|DS_IS_FLAT_NAME)) {
405 return false;
408 if (query_type == (DS_BACKGROUND_ONLY|DS_FORCE_REDISCOVERY)) {
409 return false;
412 #if 0
413 if ((flags & DS_RETURN_DNS_NAME) && (!(flags & DS_IP_REQUIRED))) {
414 printf("gd: here5 \n");
415 return false;
417 #endif
418 return true;
421 /****************************************************************
422 ****************************************************************/
424 static NTSTATUS discover_dc_netbios(TALLOC_CTX *mem_ctx,
425 const char *domain_name,
426 uint32_t flags,
427 struct ip_service_name **returned_dclist,
428 int *returned_count)
430 NTSTATUS status;
431 enum nbt_name_type name_type = NBT_NAME_LOGON;
432 struct ip_service *iplist;
433 int i;
434 struct ip_service_name *dclist = NULL;
435 int count;
437 *returned_dclist = NULL;
438 *returned_count = 0;
440 if (lp_disable_netbios()) {
441 return NT_STATUS_NOT_SUPPORTED;
444 if (flags & DS_PDC_REQUIRED) {
445 name_type = NBT_NAME_PDC;
448 status = internal_resolve_name(domain_name, name_type, NULL,
449 &iplist, &count,
450 "lmhosts wins bcast");
451 if (!NT_STATUS_IS_OK(status)) {
452 DEBUG(10,("discover_dc_netbios: failed to find DC\n"));
453 return status;
456 dclist = TALLOC_ZERO_ARRAY(mem_ctx, struct ip_service_name, count);
457 if (!dclist) {
458 return NT_STATUS_NO_MEMORY;
461 for (i=0; i<count; i++) {
463 char addr[INET6_ADDRSTRLEN];
464 struct ip_service_name *r = &dclist[i];
466 print_sockaddr(addr, sizeof(addr),
467 &iplist[i].ss);
469 r->ss = iplist[i].ss;
470 r->port = iplist[i].port;
471 r->hostname = talloc_strdup(mem_ctx, addr);
472 if (!r->hostname) {
473 return NT_STATUS_NO_MEMORY;
478 *returned_dclist = dclist;
479 *returned_count = count;
481 return NT_STATUS_OK;
484 /****************************************************************
485 ****************************************************************/
487 static NTSTATUS discover_dc_dns(TALLOC_CTX *mem_ctx,
488 const char *domain_name,
489 struct GUID *domain_guid,
490 uint32_t flags,
491 const char *site_name,
492 struct ip_service_name **returned_dclist,
493 int *return_count)
495 int i, j;
496 NTSTATUS status;
497 struct dns_rr_srv *dcs = NULL;
498 int numdcs = 0;
499 int numaddrs = 0;
500 struct ip_service_name *dclist = NULL;
501 int count = 0;
503 if (flags & DS_PDC_REQUIRED) {
504 status = ads_dns_query_pdc(mem_ctx, domain_name,
505 &dcs, &numdcs);
506 } else if (flags & DS_GC_SERVER_REQUIRED) {
507 status = ads_dns_query_gcs(mem_ctx, domain_name, site_name,
508 &dcs, &numdcs);
509 } else if (flags & DS_KDC_REQUIRED) {
510 status = ads_dns_query_kdcs(mem_ctx, domain_name, site_name,
511 &dcs, &numdcs);
512 } else if (flags & DS_DIRECTORY_SERVICE_REQUIRED) {
513 status = ads_dns_query_dcs(mem_ctx, domain_name, site_name,
514 &dcs, &numdcs);
515 } else if (domain_guid) {
516 status = ads_dns_query_dcs_guid(mem_ctx, domain_name,
517 domain_guid, &dcs, &numdcs);
518 } else {
519 status = ads_dns_query_dcs(mem_ctx, domain_name, site_name,
520 &dcs, &numdcs);
523 if (!NT_STATUS_IS_OK(status)) {
524 return status;
527 if (numdcs == 0) {
528 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
531 for (i=0;i<numdcs;i++) {
532 numaddrs += MAX(dcs[i].num_ips,1);
535 dclist = TALLOC_ZERO_ARRAY(mem_ctx,
536 struct ip_service_name,
537 numaddrs);
538 if (!dclist) {
539 return NT_STATUS_NO_MEMORY;
542 /* now unroll the list of IP addresses */
544 *return_count = 0;
545 i = 0;
546 j = 0;
548 while ((i < numdcs) && (count < numaddrs)) {
550 struct ip_service_name *r = &dclist[count];
552 r->port = dcs[count].port;
553 r->hostname = dcs[count].hostname;
555 /* If we don't have an IP list for a name, lookup it up */
557 if (!dcs[i].ss_s) {
558 interpret_string_addr(&r->ss, dcs[i].hostname, 0);
559 i++;
560 j = 0;
561 } else {
562 /* use the IP addresses from the SRV sresponse */
564 if (j >= dcs[i].num_ips) {
565 i++;
566 j = 0;
567 continue;
570 r->ss = dcs[i].ss_s[j];
571 j++;
574 /* make sure it is a valid IP. I considered checking the
575 * negative connection cache, but this is the wrong place for
576 * it. Maybe only as a hac. After think about it, if all of
577 * the IP addresses retuend from DNS are dead, what hope does a
578 * netbios name lookup have? The standard reason for falling
579 * back to netbios lookups is that our DNS server doesn't know
580 * anything about the DC's -- jerry */
582 if (!is_zero_addr(&r->ss)) {
583 count++;
584 continue;
588 *returned_dclist = dclist;
589 *return_count = count;
591 if (count > 0) {
592 return NT_STATUS_OK;
595 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
598 /****************************************************************
599 ****************************************************************/
601 static NTSTATUS make_domain_controller_info(TALLOC_CTX *mem_ctx,
602 const char *dc_unc,
603 const char *dc_address,
604 uint32_t dc_address_type,
605 const struct GUID *domain_guid,
606 const char *domain_name,
607 const char *forest_name,
608 uint32_t flags,
609 const char *dc_site_name,
610 const char *client_site_name,
611 struct netr_DsRGetDCNameInfo **info_out)
613 struct netr_DsRGetDCNameInfo *info;
615 info = TALLOC_ZERO_P(mem_ctx, struct netr_DsRGetDCNameInfo);
616 NT_STATUS_HAVE_NO_MEMORY(info);
618 if (dc_unc) {
619 info->dc_unc = talloc_strdup(mem_ctx, dc_unc);
620 NT_STATUS_HAVE_NO_MEMORY(info->dc_unc);
623 if (dc_address) {
624 info->dc_address = talloc_strdup(mem_ctx, dc_address);
625 NT_STATUS_HAVE_NO_MEMORY(info->dc_address);
628 info->dc_address_type = dc_address_type;
630 if (domain_guid) {
631 info->domain_guid = *domain_guid;
634 if (domain_name) {
635 info->domain_name = talloc_strdup(mem_ctx, domain_name);
636 NT_STATUS_HAVE_NO_MEMORY(info->domain_name);
639 if (forest_name) {
640 info->forest_name = talloc_strdup(mem_ctx, forest_name);
641 NT_STATUS_HAVE_NO_MEMORY(info->forest_name);
642 flags |= DS_DNS_FOREST;
645 info->dc_flags = flags;
647 if (dc_site_name) {
648 info->dc_site_name = talloc_strdup(mem_ctx, dc_site_name);
649 NT_STATUS_HAVE_NO_MEMORY(info->dc_site_name);
652 if (client_site_name) {
653 info->client_site_name = talloc_strdup(mem_ctx,
654 client_site_name);
655 NT_STATUS_HAVE_NO_MEMORY(info->client_site_name);
658 *info_out = info;
660 return NT_STATUS_OK;
663 /****************************************************************
664 ****************************************************************/
666 static NTSTATUS make_dc_info_from_cldap_reply(TALLOC_CTX *mem_ctx,
667 uint32_t flags,
668 struct sockaddr_storage *ss,
669 uint32_t nt_version,
670 union nbt_cldap_netlogon *r,
671 struct netr_DsRGetDCNameInfo **info)
673 const char *dc_hostname, *dc_domain_name;
674 const char *dc_address = NULL;
675 const char *dc_forest = NULL;
676 uint32_t dc_address_type = 0;
677 uint32_t dc_flags = 0;
678 struct GUID *dc_domain_guid = NULL;
679 const char *dc_server_site = NULL;
680 const char *dc_client_site = NULL;
682 char addr[INET6_ADDRSTRLEN];
684 print_sockaddr(addr, sizeof(addr), ss);
686 dc_address = talloc_asprintf(mem_ctx, "\\\\%s", addr);
687 NT_STATUS_HAVE_NO_MEMORY(dc_address);
688 dc_address_type = DS_ADDRESS_TYPE_INET;
690 switch (nt_version & 0x000000ff) {
691 case 0:
692 return NT_STATUS_INVALID_PARAMETER;
693 case 1:
694 dc_hostname = r->logon1.pdc_name;
695 dc_domain_name = r->logon1.domain_name;
696 if (flags & DS_PDC_REQUIRED) {
697 dc_flags = NBT_SERVER_WRITABLE | NBT_SERVER_PDC;
699 break;
700 case 2:
701 case 3:
702 switch (flags & 0xf0000000) {
703 case DS_RETURN_FLAT_NAME:
704 dc_hostname = r->logon3.pdc_name;
705 dc_domain_name = r->logon3.domain_name;
706 break;
707 case DS_RETURN_DNS_NAME:
708 default:
709 dc_hostname = r->logon3.pdc_dns_name;
710 dc_domain_name = r->logon3.dns_domain;
711 dc_flags |= DS_DNS_DOMAIN | DS_DNS_CONTROLLER;
712 break;
715 dc_flags |= r->logon3.server_type;
716 dc_forest = r->logon3.forest;
717 dc_domain_guid = &r->logon3.domain_uuid;
719 break;
720 case 4:
721 case 5:
722 case 6:
723 case 7:
724 switch (flags & 0xf0000000) {
725 case DS_RETURN_FLAT_NAME:
726 dc_hostname = r->logon5.pdc_name;
727 dc_domain_name = r->logon5.domain;
728 break;
729 case DS_RETURN_DNS_NAME:
730 default:
731 dc_hostname = r->logon5.pdc_dns_name;
732 dc_domain_name = r->logon5.dns_domain;
733 dc_flags |= DS_DNS_DOMAIN | DS_DNS_CONTROLLER;
734 break;
737 dc_flags |= r->logon5.server_type;
738 dc_forest = r->logon5.forest;
739 dc_domain_guid = &r->logon5.domain_uuid;
740 dc_server_site = r->logon5.server_site;
741 dc_client_site = r->logon5.client_site;
743 break;
744 case 8:
745 case 9:
746 case 10:
747 case 11:
748 case 12:
749 case 13:
750 case 14:
751 case 15:
752 switch (flags & 0xf0000000) {
753 case DS_RETURN_FLAT_NAME:
754 dc_hostname = r->logon13.pdc_name;
755 dc_domain_name = r->logon13.domain;
756 break;
757 case DS_RETURN_DNS_NAME:
758 default:
759 dc_hostname = r->logon13.pdc_dns_name;
760 dc_domain_name = r->logon13.dns_domain;
761 dc_flags |= DS_DNS_DOMAIN | DS_DNS_CONTROLLER;
762 break;
765 dc_flags |= r->logon13.server_type;
766 dc_forest = r->logon13.forest;
767 dc_domain_guid = &r->logon13.domain_uuid;
768 dc_server_site = r->logon13.server_site;
769 dc_client_site = r->logon13.client_site;
771 break;
772 default:
773 switch (flags & 0xf0000000) {
774 case DS_RETURN_FLAT_NAME:
775 dc_hostname = r->logon29.pdc_name;
776 dc_domain_name = r->logon29.domain;
777 break;
778 case DS_RETURN_DNS_NAME:
779 default:
780 dc_hostname = r->logon29.pdc_dns_name;
781 dc_domain_name = r->logon29.dns_domain;
782 dc_flags |= DS_DNS_DOMAIN | DS_DNS_CONTROLLER;
783 break;
786 dc_flags |= r->logon29.server_type;
787 dc_forest = r->logon29.forest;
788 dc_domain_guid = &r->logon29.domain_uuid;
789 dc_server_site = r->logon29.server_site;
790 dc_client_site = r->logon29.client_site;
792 break;
795 return make_domain_controller_info(mem_ctx,
796 dc_hostname,
797 dc_address,
798 dc_address_type,
799 dc_domain_guid,
800 dc_domain_name,
801 dc_forest,
802 dc_flags,
803 dc_server_site,
804 dc_client_site,
805 info);
808 /****************************************************************
809 ****************************************************************/
811 static uint32_t map_ds_flags_to_nt_version(uint32_t flags)
813 uint32_t nt_version = 0;
815 if (flags & DS_PDC_REQUIRED) {
816 nt_version |= NETLOGON_VERSION_PDC;
819 if (flags & DS_GC_SERVER_REQUIRED) {
820 nt_version |= NETLOGON_VERSION_GC;
823 if (flags & DS_TRY_NEXTCLOSEST_SITE) {
824 nt_version |= NETLOGON_VERSION_WITH_CLOSEST_SITE;
827 if (flags & DS_IP_REQUIRED) {
828 nt_version |= NETLOGON_VERSION_IP;
831 return nt_version;
834 /****************************************************************
835 ****************************************************************/
837 static NTSTATUS process_dc_dns(TALLOC_CTX *mem_ctx,
838 const char *domain_name,
839 uint32_t flags,
840 struct ip_service_name *dclist,
841 int num_dcs,
842 struct netr_DsRGetDCNameInfo **info)
844 int i = 0;
845 bool valid_dc = false;
846 union nbt_cldap_netlogon *r = NULL;
847 uint32_t nt_version = NETLOGON_VERSION_5 |
848 NETLOGON_VERSION_5EX;
849 uint32_t ret_flags = 0;
851 nt_version |= map_ds_flags_to_nt_version(flags);
853 for (i=0; i<num_dcs; i++) {
855 DEBUG(10,("LDAP ping to %s\n", dclist[i].hostname));
857 if (ads_cldap_netlogon(mem_ctx, dclist[i].hostname,
858 domain_name,
859 &nt_version,
860 &r))
862 ret_flags = get_cldap_reply_server_flags(r, nt_version);
864 if (check_cldap_reply_required_flags(ret_flags, flags)) {
865 valid_dc = true;
866 break;
870 continue;
873 if (!valid_dc) {
874 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
877 return make_dc_info_from_cldap_reply(mem_ctx, flags, &dclist[i].ss,
878 nt_version, r, info);
881 /****************************************************************
882 ****************************************************************/
884 static struct event_context *ev_context(void)
886 static struct event_context *ctx;
888 if (!ctx && !(ctx = event_context_init(NULL))) {
889 smb_panic("Could not init event context");
891 return ctx;
894 /****************************************************************
895 ****************************************************************/
897 static struct messaging_context *msg_context(TALLOC_CTX *mem_ctx)
899 static struct messaging_context *ctx;
901 if (!ctx && !(ctx = messaging_init(mem_ctx, server_id_self(),
902 ev_context()))) {
903 smb_panic("Could not init messaging context");
905 return ctx;
908 /****************************************************************
909 ****************************************************************/
911 static NTSTATUS process_dc_netbios(TALLOC_CTX *mem_ctx,
912 const char *domain_name,
913 uint32_t flags,
914 struct ip_service_name *dclist,
915 int num_dcs,
916 struct netr_DsRGetDCNameInfo **info)
918 struct sockaddr_storage ss;
919 struct ip_service ip_list;
920 enum nbt_name_type name_type = NBT_NAME_LOGON;
922 int i;
923 const char *dc_name = NULL;
924 fstring tmp_dc_name;
925 struct messaging_context *msg_ctx = msg_context(mem_ctx);
926 union nbt_cldap_netlogon *reply = NULL;
927 uint32_t nt_version = NETLOGON_VERSION_1 |
928 NETLOGON_VERSION_5 |
929 NETLOGON_VERSION_5EX_WITH_IP;
931 if (flags & DS_PDC_REQUIRED) {
932 name_type = NBT_NAME_PDC;
935 nt_version |= map_ds_flags_to_nt_version(flags);
937 DEBUG(10,("process_dc_netbios\n"));
939 for (i=0; i<num_dcs; i++) {
941 ip_list.ss = dclist[i].ss;
942 ip_list.port = 0;
944 if (!interpret_string_addr(&ss, dclist[i].hostname, AI_NUMERICHOST)) {
945 return NT_STATUS_UNSUCCESSFUL;
948 if (send_getdc_request(mem_ctx, msg_ctx,
949 &dclist[i].ss, domain_name,
950 NULL, nt_version))
952 int k;
953 smb_msleep(300);
954 for (k=0; k<5; k++) {
955 if (receive_getdc_response(mem_ctx,
956 &dclist[i].ss,
957 domain_name,
958 &nt_version,
959 &dc_name,
960 &reply)) {
961 namecache_store(dc_name, NBT_NAME_SERVER, 1, &ip_list);
962 goto make_reply;
964 smb_msleep(1500);
968 if (name_status_find(domain_name,
969 name_type,
970 NBT_NAME_SERVER,
971 &dclist[i].ss,
972 tmp_dc_name))
974 struct nbt_cldap_netlogon_1 logon1;
976 reply = TALLOC_ZERO_P(mem_ctx, union nbt_cldap_netlogon);
977 NT_STATUS_HAVE_NO_MEMORY(reply);
979 ZERO_STRUCT(logon1);
981 nt_version = NETLOGON_VERSION_1;
983 logon1.nt_version = nt_version;
984 logon1.pdc_name = tmp_dc_name;
985 logon1.domain_name = talloc_strdup_upper(mem_ctx, domain_name);
986 NT_STATUS_HAVE_NO_MEMORY(logon1.domain_name);
988 reply->logon1 = logon1;
990 namecache_store(tmp_dc_name, NBT_NAME_SERVER, 1, &ip_list);
992 goto make_reply;
996 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
998 make_reply:
1000 return make_dc_info_from_cldap_reply(mem_ctx, flags, &dclist[i].ss,
1001 nt_version, reply, info);
1004 /****************************************************************
1005 ****************************************************************/
1007 static NTSTATUS dsgetdcname_rediscover(TALLOC_CTX *mem_ctx,
1008 const char *domain_name,
1009 struct GUID *domain_guid,
1010 uint32_t flags,
1011 const char *site_name,
1012 struct netr_DsRGetDCNameInfo **info)
1014 NTSTATUS status = NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
1015 struct ip_service_name *dclist = NULL;
1016 int num_dcs;
1018 DEBUG(10,("dsgetdcname_rediscover\n"));
1020 if (flags & DS_IS_FLAT_NAME) {
1022 status = discover_dc_netbios(mem_ctx, domain_name, flags,
1023 &dclist, &num_dcs);
1024 NT_STATUS_NOT_OK_RETURN(status);
1026 return process_dc_netbios(mem_ctx, domain_name, flags,
1027 dclist, num_dcs, info);
1030 if (flags & DS_IS_DNS_NAME) {
1032 status = discover_dc_dns(mem_ctx, domain_name, domain_guid,
1033 flags, site_name, &dclist, &num_dcs);
1034 NT_STATUS_NOT_OK_RETURN(status);
1036 return process_dc_dns(mem_ctx, domain_name, flags,
1037 dclist, num_dcs, info);
1040 status = discover_dc_dns(mem_ctx, domain_name, domain_guid, flags,
1041 site_name, &dclist, &num_dcs);
1043 if (NT_STATUS_IS_OK(status) && num_dcs != 0) {
1045 status = process_dc_dns(mem_ctx, domain_name, flags, dclist,
1046 num_dcs, info);
1047 if (NT_STATUS_IS_OK(status)) {
1048 return status;
1052 status = discover_dc_netbios(mem_ctx, domain_name, flags, &dclist,
1053 &num_dcs);
1054 NT_STATUS_NOT_OK_RETURN(status);
1056 return process_dc_netbios(mem_ctx, domain_name, flags, dclist,
1057 num_dcs, info);
1060 /********************************************************************
1061 dsgetdcname.
1063 This will be the only public function here.
1064 ********************************************************************/
1066 NTSTATUS dsgetdcname(TALLOC_CTX *mem_ctx,
1067 const char *domain_name,
1068 struct GUID *domain_guid,
1069 const char *site_name,
1070 uint32_t flags,
1071 struct netr_DsRGetDCNameInfo **info)
1073 NTSTATUS status = NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
1074 struct netr_DsRGetDCNameInfo *myinfo = NULL;
1076 DEBUG(10,("dsgetdcname: domain_name: %s, "
1077 "domain_guid: %s, site_name: %s, flags: 0x%08x\n",
1078 domain_name,
1079 domain_guid ? GUID_string(mem_ctx, domain_guid) : "(null)",
1080 site_name, flags));
1082 *info = NULL;
1084 if (!check_allowed_required_flags(flags)) {
1085 DEBUG(0,("invalid flags specified\n"));
1086 return NT_STATUS_INVALID_PARAMETER;
1089 if (flags & DS_FORCE_REDISCOVERY) {
1090 goto rediscover;
1093 status = dsgetdcname_cached(mem_ctx, domain_name, domain_guid,
1094 flags, site_name, &myinfo);
1095 if (NT_STATUS_IS_OK(status)) {
1096 *info = myinfo;
1097 return status;
1100 if (flags & DS_BACKGROUND_ONLY) {
1101 return status;
1104 rediscover:
1105 status = dsgetdcname_rediscover(mem_ctx, domain_name,
1106 domain_guid, flags, site_name,
1107 &myinfo);
1109 if (NT_STATUS_IS_OK(status)) {
1110 dsgetdcname_cache_store(mem_ctx, domain_name, myinfo);
1111 *info = myinfo;
1114 return status;
1117 /****************************************************************
1118 ****************************************************************/
1120 bool pull_mailslot_cldap_reply(TALLOC_CTX *mem_ctx,
1121 const DATA_BLOB *blob,
1122 union nbt_cldap_netlogon *r,
1123 uint32_t *nt_version)
1125 enum ndr_err_code ndr_err;
1126 uint32_t nt_version_query = ((*nt_version) & 0x000000ff);
1127 uint16_t command = 0;
1129 ndr_err = ndr_pull_struct_blob(blob, mem_ctx, &command,
1130 (ndr_pull_flags_fn_t)ndr_pull_uint16);
1131 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1132 return false;
1135 switch (command) {
1136 case 0x13: /* 19 */
1137 case 0x15: /* 21 */
1138 case 0x17: /* 23 */
1139 break;
1140 default:
1141 DEBUG(1,("got unexpected command: %d (0x%08x)\n",
1142 command, command));
1143 return false;
1146 ndr_err = ndr_pull_union_blob_all(blob, mem_ctx, r, nt_version_query,
1147 (ndr_pull_flags_fn_t)ndr_pull_nbt_cldap_netlogon);
1148 if (NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1149 goto done;
1152 /* when the caller requested just those nt_version bits that the server
1153 * was able to reply to, we are fine and all done. otherwise we need to
1154 * assume downgraded replies which are painfully parsed here - gd */
1156 if (nt_version_query & NETLOGON_VERSION_WITH_CLOSEST_SITE) {
1157 nt_version_query &= ~NETLOGON_VERSION_WITH_CLOSEST_SITE;
1159 ndr_err = ndr_pull_union_blob_all(blob, mem_ctx, r, nt_version_query,
1160 (ndr_pull_flags_fn_t)ndr_pull_nbt_cldap_netlogon);
1161 if (NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1162 goto done;
1164 if (nt_version_query & NETLOGON_VERSION_5EX_WITH_IP) {
1165 nt_version_query &= ~NETLOGON_VERSION_5EX_WITH_IP;
1167 ndr_err = ndr_pull_union_blob_all(blob, mem_ctx, r, nt_version_query,
1168 (ndr_pull_flags_fn_t)ndr_pull_nbt_cldap_netlogon);
1169 if (NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1170 goto done;
1172 if (nt_version_query & NETLOGON_VERSION_5EX) {
1173 nt_version_query &= ~NETLOGON_VERSION_5EX;
1175 ndr_err = ndr_pull_union_blob_all(blob, mem_ctx, r, nt_version_query,
1176 (ndr_pull_flags_fn_t)ndr_pull_nbt_cldap_netlogon);
1177 if (NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1178 goto done;
1180 if (nt_version_query & NETLOGON_VERSION_5) {
1181 nt_version_query &= ~NETLOGON_VERSION_5;
1183 ndr_err = ndr_pull_union_blob_all(blob, mem_ctx, r, nt_version_query,
1184 (ndr_pull_flags_fn_t)ndr_pull_nbt_cldap_netlogon);
1185 if (NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1186 goto done;
1189 return false;
1191 done:
1192 if (DEBUGLEVEL >= 10) {
1193 NDR_PRINT_UNION_DEBUG(nbt_cldap_netlogon, nt_version_query, r);
1196 *nt_version = nt_version_query;
1198 return true;