s3:idmap_ad: add support for ADS_AUTH_SASL_{STARTTLS,LDAPS}
[Samba.git] / source3 / libsmb / dsgetdcname.c
blob0fcf23a280ee98933d83e73393b34a1c2b91820b
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"
24 #include "libsmb/dsgetdcname.h"
25 #include "libsmb/namequery.h"
26 #include "libads/sitename_cache.h"
27 #include "../librpc/gen_ndr/ndr_netlogon.h"
28 #include "libads/cldap.h"
29 #include "lib/addns/dnsquery_srv.h"
30 #include "libsmb/clidgram.h"
31 #include "lib/gencache.h"
32 #include "lib/util/util_net.h"
34 /* 15 minutes */
35 #define DSGETDCNAME_CACHE_TTL 60*15
37 struct ip_service_name {
38 struct samba_sockaddr sa;
39 const char *hostname;
42 static NTSTATUS make_dc_info_from_cldap_reply(
43 TALLOC_CTX *mem_ctx,
44 uint32_t flags,
45 const struct samba_sockaddr *sa,
46 struct NETLOGON_SAM_LOGON_RESPONSE_EX *r,
47 struct netr_DsRGetDCNameInfo **info);
49 /****************************************************************
50 ****************************************************************/
52 static void debug_dsdcinfo_flags(int lvl, uint32_t flags)
54 DEBUG(lvl,("debug_dsdcinfo_flags: 0x%08x\n\t", flags));
56 if (flags & DS_FORCE_REDISCOVERY)
57 DEBUGADD(lvl,("DS_FORCE_REDISCOVERY "));
58 if (flags & 0x00000002)
59 DEBUGADD(lvl,("0x00000002 "));
60 if (flags & 0x00000004)
61 DEBUGADD(lvl,("0x00000004 "));
62 if (flags & 0x00000008)
63 DEBUGADD(lvl,("0x00000008 "));
64 if (flags & DS_DIRECTORY_SERVICE_REQUIRED)
65 DEBUGADD(lvl,("DS_DIRECTORY_SERVICE_REQUIRED "));
66 if (flags & DS_DIRECTORY_SERVICE_PREFERRED)
67 DEBUGADD(lvl,("DS_DIRECTORY_SERVICE_PREFERRED "));
68 if (flags & DS_GC_SERVER_REQUIRED)
69 DEBUGADD(lvl,("DS_GC_SERVER_REQUIRED "));
70 if (flags & DS_PDC_REQUIRED)
71 DEBUGADD(lvl,("DS_PDC_REQUIRED "));
72 if (flags & DS_BACKGROUND_ONLY)
73 DEBUGADD(lvl,("DS_BACKGROUND_ONLY "));
74 if (flags & DS_IP_REQUIRED)
75 DEBUGADD(lvl,("DS_IP_REQUIRED "));
76 if (flags & DS_KDC_REQUIRED)
77 DEBUGADD(lvl,("DS_KDC_REQUIRED "));
78 if (flags & DS_TIMESERV_REQUIRED)
79 DEBUGADD(lvl,("DS_TIMESERV_REQUIRED "));
80 if (flags & DS_WRITABLE_REQUIRED)
81 DEBUGADD(lvl,("DS_WRITABLE_REQUIRED "));
82 if (flags & DS_GOOD_TIMESERV_PREFERRED)
83 DEBUGADD(lvl,("DS_GOOD_TIMESERV_PREFERRED "));
84 if (flags & DS_AVOID_SELF)
85 DEBUGADD(lvl,("DS_AVOID_SELF "));
86 if (flags & DS_ONLY_LDAP_NEEDED)
87 DEBUGADD(lvl,("DS_ONLY_LDAP_NEEDED "));
88 if (flags & DS_IS_FLAT_NAME)
89 DEBUGADD(lvl,("DS_IS_FLAT_NAME "));
90 if (flags & DS_IS_DNS_NAME)
91 DEBUGADD(lvl,("DS_IS_DNS_NAME "));
92 if (flags & DS_TRY_NEXTCLOSEST_SITE)
93 DEBUGADD(lvl,("DS_TRY_NEXTCLOSEST_SITE "));
94 if (flags & DS_DIRECTORY_SERVICE_6_REQUIRED)
95 DEBUGADD(lvl,("DS_DIRECTORY_SERVICE_6_REQUIRED "));
96 if (flags & DS_WEB_SERVICE_REQUIRED)
97 DEBUGADD(lvl,("DS_WEB_SERVICE_REQUIRED "));
98 if (flags & DS_DIRECTORY_SERVICE_8_REQUIRED)
99 DEBUGADD(lvl,("DS_DIRECTORY_SERVICE_8_REQUIRED "));
100 if (flags & DS_DIRECTORY_SERVICE_9_REQUIRED)
101 DEBUGADD(lvl,("DS_DIRECTORY_SERVICE_9_REQUIRED "));
102 if (flags & DS_DIRECTORY_SERVICE_10_REQUIRED)
103 DEBUGADD(lvl,("DS_DIRECTORY_SERVICE_10_REQUIRED "));
104 if (flags & 0x01000000)
105 DEBUGADD(lvl,("0x01000000 "));
106 if (flags & 0x02000000)
107 DEBUGADD(lvl,("0x02000000 "));
108 if (flags & 0x04000000)
109 DEBUGADD(lvl,("0x04000000 "));
110 if (flags & 0x08000000)
111 DEBUGADD(lvl,("0x08000000 "));
112 if (flags & 0x10000000)
113 DEBUGADD(lvl,("0x10000000 "));
114 if (flags & 0x20000000)
115 DEBUGADD(lvl,("0x20000000 "));
116 if (flags & DS_RETURN_DNS_NAME)
117 DEBUGADD(lvl,("DS_RETURN_DNS_NAME "));
118 if (flags & DS_RETURN_FLAT_NAME)
119 DEBUGADD(lvl,("DS_RETURN_FLAT_NAME "));
120 if (flags)
121 DEBUGADD(lvl,("\n"));
124 /****************************************************************
125 ****************************************************************/
127 static char *dsgetdcname_cache_key(TALLOC_CTX *mem_ctx, const char *domain)
129 if (!domain) {
130 return NULL;
133 return talloc_asprintf_strupper_m(mem_ctx, "DSGETDCNAME/DOMAIN/%s",
134 domain);
137 /****************************************************************
138 ****************************************************************/
140 static NTSTATUS dsgetdcname_cache_delete(TALLOC_CTX *mem_ctx,
141 const char *domain_name)
143 char *key;
145 key = dsgetdcname_cache_key(mem_ctx, domain_name);
146 if (!key) {
147 return NT_STATUS_NO_MEMORY;
150 if (!gencache_del(key)) {
151 return NT_STATUS_UNSUCCESSFUL;
154 return NT_STATUS_OK;
157 /****************************************************************
158 ****************************************************************/
160 static NTSTATUS dsgetdcname_cache_store(TALLOC_CTX *mem_ctx,
161 const char *domain_name,
162 DATA_BLOB blob)
164 time_t expire_time;
165 char *key;
166 bool ret = false;
168 key = dsgetdcname_cache_key(mem_ctx, domain_name);
169 if (!key) {
170 return NT_STATUS_NO_MEMORY;
173 expire_time = time(NULL) + DSGETDCNAME_CACHE_TTL;
175 ret = gencache_set_data_blob(key, blob, expire_time);
177 return ret ? NT_STATUS_OK : NT_STATUS_UNSUCCESSFUL;
180 /****************************************************************
181 ****************************************************************/
183 static NTSTATUS store_cldap_reply(TALLOC_CTX *mem_ctx,
184 uint32_t flags,
185 struct samba_sockaddr *sa,
186 uint32_t nt_version,
187 struct NETLOGON_SAM_LOGON_RESPONSE_EX *r)
189 DATA_BLOB blob;
190 enum ndr_err_code ndr_err;
191 NTSTATUS status;
192 char addr[INET6_ADDRSTRLEN];
194 print_sockaddr(addr, sizeof(addr), &sa->u.ss);
196 /* FIXME */
197 r->sockaddr_size = 0x10; /* the w32 winsock addr size */
198 r->sockaddr.sockaddr_family = 2; /* AF_INET */
199 r->sockaddr.pdc_ip = talloc_strdup(mem_ctx, addr);
201 ndr_err = ndr_push_struct_blob(&blob, mem_ctx, r,
202 (ndr_push_flags_fn_t)ndr_push_NETLOGON_SAM_LOGON_RESPONSE_EX);
203 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
204 return ndr_map_error2ntstatus(ndr_err);
207 if (r->domain_name) {
208 status = dsgetdcname_cache_store(mem_ctx, r->domain_name,
209 blob);
210 if (!NT_STATUS_IS_OK(status)) {
211 goto done;
213 if (r->client_site) {
214 sitename_store(r->domain_name, r->client_site);
217 if (r->dns_domain) {
218 status = dsgetdcname_cache_store(mem_ctx, r->dns_domain, blob);
219 if (!NT_STATUS_IS_OK(status)) {
220 goto done;
222 if (r->client_site) {
223 sitename_store(r->dns_domain, r->client_site);
227 status = NT_STATUS_OK;
229 done:
230 data_blob_free(&blob);
232 return status;
235 /****************************************************************
236 ****************************************************************/
238 static uint32_t get_cldap_reply_server_flags(struct netlogon_samlogon_response *r,
239 uint32_t nt_version)
241 switch (nt_version & 0x0000001f) {
242 case 0:
243 case 1:
244 case 16:
245 case 17:
246 return 0;
247 case 2:
248 case 3:
249 case 18:
250 case 19:
251 return r->data.nt5.server_type;
252 case 4:
253 case 5:
254 case 6:
255 case 7:
256 return r->data.nt5_ex.server_type;
257 case 8:
258 case 9:
259 case 10:
260 case 11:
261 case 12:
262 case 13:
263 case 14:
264 case 15:
265 return r->data.nt5_ex.server_type;
266 case 20:
267 case 21:
268 case 22:
269 case 23:
270 case 24:
271 case 25:
272 case 26:
273 case 27:
274 case 28:
275 return r->data.nt5_ex.server_type;
276 case 29:
277 case 30:
278 case 31:
279 return r->data.nt5_ex.server_type;
280 default:
281 return 0;
285 /****************************************************************
286 ****************************************************************/
288 static NTSTATUS dsgetdcname_cache_fetch(TALLOC_CTX *mem_ctx,
289 const char *domain_name,
290 const struct GUID *domain_guid,
291 uint32_t flags,
292 struct netr_DsRGetDCNameInfo **info_p)
294 char *key;
295 DATA_BLOB blob;
296 enum ndr_err_code ndr_err;
297 struct netr_DsRGetDCNameInfo *info;
298 struct NETLOGON_SAM_LOGON_RESPONSE_EX r;
299 NTSTATUS status;
301 key = dsgetdcname_cache_key(mem_ctx, domain_name);
302 if (!key) {
303 return NT_STATUS_NO_MEMORY;
306 if (!gencache_get_data_blob(key, NULL, &blob, NULL, NULL)) {
307 return NT_STATUS_NOT_FOUND;
310 info = talloc_zero(mem_ctx, struct netr_DsRGetDCNameInfo);
311 if (!info) {
312 data_blob_free(&blob);
313 return NT_STATUS_NO_MEMORY;
316 ndr_err = ndr_pull_struct_blob(&blob, mem_ctx, &r,
317 (ndr_pull_flags_fn_t)ndr_pull_NETLOGON_SAM_LOGON_RESPONSE_EX);
319 data_blob_free(&blob);
320 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
321 dsgetdcname_cache_delete(mem_ctx, domain_name);
322 return ndr_map_error2ntstatus(ndr_err);
325 status = make_dc_info_from_cldap_reply(mem_ctx, flags, NULL,
326 &r, &info);
327 if (!NT_STATUS_IS_OK(status)) {
328 return status;
331 if (DEBUGLEVEL >= 10) {
332 NDR_PRINT_DEBUG(netr_DsRGetDCNameInfo, info);
335 /* check flags */
336 if (!check_cldap_reply_required_flags(info->dc_flags, flags)) {
337 DEBUG(10,("invalid flags\n"));
338 return NT_STATUS_INVALID_PARAMETER;
341 if ((flags & DS_IP_REQUIRED) &&
342 (info->dc_address_type != DS_ADDRESS_TYPE_INET)) {
343 return NT_STATUS_INVALID_PARAMETER_MIX;
346 *info_p = info;
348 return NT_STATUS_OK;
351 /****************************************************************
352 ****************************************************************/
354 static NTSTATUS dsgetdcname_cached(TALLOC_CTX *mem_ctx,
355 struct messaging_context *msg_ctx,
356 const char *domain_name,
357 const struct GUID *domain_guid,
358 uint32_t flags,
359 const char *site_name,
360 struct netr_DsRGetDCNameInfo **info)
362 NTSTATUS status;
364 status = dsgetdcname_cache_fetch(mem_ctx, domain_name, domain_guid,
365 flags, info);
366 if (!NT_STATUS_IS_OK(status)
367 && !NT_STATUS_EQUAL(status, NT_STATUS_NOT_FOUND)) {
368 DEBUG(10,("dsgetdcname_cached: cache fetch failed with: %s\n",
369 nt_errstr(status)));
370 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
373 if (flags & DS_BACKGROUND_ONLY) {
374 return status;
377 if (NT_STATUS_EQUAL(status, NT_STATUS_NOT_FOUND)) {
378 struct netr_DsRGetDCNameInfo *dc_info;
380 status = dsgetdcname(mem_ctx, msg_ctx, domain_name,
381 domain_guid, site_name,
382 flags | DS_FORCE_REDISCOVERY,
383 &dc_info);
385 if (!NT_STATUS_IS_OK(status)) {
386 return status;
389 *info = dc_info;
392 return status;
395 /****************************************************************
396 ****************************************************************/
398 static bool check_allowed_required_flags(uint32_t flags,
399 const char *site_name)
401 uint32_t return_type = flags & (DS_RETURN_FLAT_NAME|DS_RETURN_DNS_NAME);
402 uint32_t offered_type = flags & (DS_IS_FLAT_NAME|DS_IS_DNS_NAME);
403 uint32_t query_type = flags & (DS_BACKGROUND_ONLY|DS_FORCE_REDISCOVERY);
405 /* FIXME: check for DSGETDC_VALID_FLAGS and check for exclusive bits
406 * (DS_PDC_REQUIRED, DS_KDC_REQUIRED, DS_GC_SERVER_REQUIRED) */
408 debug_dsdcinfo_flags(10, flags);
410 if ((flags & DS_TRY_NEXTCLOSEST_SITE) && site_name) {
411 return false;
414 if (return_type == (DS_RETURN_FLAT_NAME|DS_RETURN_DNS_NAME)) {
415 return false;
418 if (offered_type == (DS_IS_DNS_NAME|DS_IS_FLAT_NAME)) {
419 return false;
422 if (query_type == (DS_BACKGROUND_ONLY|DS_FORCE_REDISCOVERY)) {
423 return false;
426 #if 0
427 if ((flags & DS_RETURN_DNS_NAME) && (!(flags & DS_IP_REQUIRED))) {
428 printf("gd: here5 \n");
429 return false;
431 #endif
432 return true;
435 /****************************************************************
436 ****************************************************************/
438 static NTSTATUS discover_dc_netbios(TALLOC_CTX *mem_ctx,
439 const char *domain_name,
440 uint32_t flags,
441 struct ip_service_name **returned_dclist,
442 size_t *returned_count)
444 NTSTATUS status;
445 enum nbt_name_type name_type = NBT_NAME_LOGON;
446 struct samba_sockaddr *salist = NULL;
447 size_t i;
448 struct ip_service_name *dclist = NULL;
449 size_t count = 0;
450 static const char *resolve_order[] = { "lmhosts", "wins", "bcast", NULL };
452 if (flags & DS_PDC_REQUIRED) {
453 name_type = NBT_NAME_PDC;
456 status = internal_resolve_name(mem_ctx,
457 domain_name,
458 name_type,
459 NULL,
460 &salist,
461 &count,
462 resolve_order);
463 if (!NT_STATUS_IS_OK(status)) {
464 DEBUG(10,("discover_dc_netbios: failed to find DC\n"));
465 return status;
468 dclist = talloc_zero_array(mem_ctx, struct ip_service_name, count);
469 if (!dclist) {
470 TALLOC_FREE(salist);
471 return NT_STATUS_NO_MEMORY;
474 for (i=0; i<count; i++) {
475 char addr[INET6_ADDRSTRLEN];
476 struct ip_service_name *r = &dclist[i];
478 print_sockaddr(addr, sizeof(addr),
479 &salist[i].u.ss);
481 r->sa = salist[i];
482 r->hostname = talloc_strdup(mem_ctx, addr);
483 if (!r->hostname) {
484 TALLOC_FREE(salist);
485 TALLOC_FREE(dclist);
486 return NT_STATUS_NO_MEMORY;
491 TALLOC_FREE(salist);
493 *returned_dclist = dclist;
494 *returned_count = count;
496 return NT_STATUS_OK;
499 /****************************************************************
500 ****************************************************************/
502 static NTSTATUS discover_dc_dns(TALLOC_CTX *mem_ctx,
503 const char *domain_name,
504 const struct GUID *domain_guid,
505 uint32_t flags,
506 const char *site_name,
507 struct ip_service_name **returned_dclist,
508 size_t *return_count)
510 size_t i;
511 NTSTATUS status;
512 struct dns_rr_srv *dcs = NULL;
513 size_t numdcs = 0;
514 struct ip_service_name *dclist = NULL;
515 size_t ret_count = 0;
516 char *query = NULL;
518 if (flags & DS_PDC_REQUIRED) {
519 query = ads_dns_query_string_pdc(mem_ctx, domain_name);
520 } else if (flags & DS_GC_SERVER_REQUIRED) {
521 query = ads_dns_query_string_gcs(mem_ctx, domain_name);
522 } else if (flags & DS_KDC_REQUIRED) {
523 query = ads_dns_query_string_kdcs(mem_ctx, domain_name);
524 } else if (flags & DS_DIRECTORY_SERVICE_REQUIRED) {
525 query = ads_dns_query_string_dcs(mem_ctx, domain_name);
526 } else if (domain_guid) {
527 query = ads_dns_query_string_dcs_guid(
528 mem_ctx, domain_guid, domain_name);
529 } else {
530 query = ads_dns_query_string_dcs(mem_ctx, domain_name);
533 if (query == NULL) {
534 return NT_STATUS_NO_MEMORY;
537 status = ads_dns_query_srv(
538 mem_ctx,
539 lp_get_async_dns_timeout(),
540 site_name,
541 query,
542 &dcs,
543 &numdcs);
544 TALLOC_FREE(query);
545 if (!NT_STATUS_IS_OK(status)) {
546 return status;
549 if (numdcs == 0) {
550 TALLOC_FREE(dcs);
551 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
554 /* Check for integer wrap. */
555 if (numdcs + numdcs < numdcs) {
556 TALLOC_FREE(dcs);
557 return NT_STATUS_INVALID_PARAMETER;
561 * We're only returning up to 2 addresses per
562 * DC name, so just allocate size numdcs x 2.
565 dclist = talloc_zero_array(mem_ctx,
566 struct ip_service_name,
567 numdcs * 2);
568 if (!dclist) {
569 TALLOC_FREE(dcs);
570 return NT_STATUS_NO_MEMORY;
574 * First, copy the SRV record replies that
575 * have IP addresses returned with them.
577 ret_count = 0;
578 for (i = 0; i < numdcs; i++) {
579 size_t j;
580 bool have_v4_addr = false;
581 bool have_v6_addr = false;
583 if (dcs[i].num_ips == 0) {
584 continue;
588 * Pick up to 1 address from each address
589 * family (IPv4, IPv6).
591 * This is different from the previous
592 * code which picked a 'next ip' address
593 * each time, incrementing an index.
594 * Too complex to maintain :-(.
596 for (j = 0; j < dcs[i].num_ips; j++) {
597 if ((dcs[i].ss_s[j].ss_family == AF_INET && !have_v4_addr) ||
598 (dcs[i].ss_s[j].ss_family == AF_INET6 && !have_v6_addr)) {
599 bool ok;
600 dclist[ret_count].hostname =
601 talloc_strdup(dclist, dcs[i].hostname);
602 ok = sockaddr_storage_to_samba_sockaddr(
603 &dclist[ret_count].sa,
604 &dcs[i].ss_s[j]);
605 if (!ok) {
606 TALLOC_FREE(dcs);
607 TALLOC_FREE(dclist);
608 return NT_STATUS_INVALID_PARAMETER;
610 ret_count++;
611 if (dcs[i].ss_s[j].ss_family == AF_INET) {
612 have_v4_addr = true;
613 } else {
614 have_v6_addr = true;
616 if (have_v4_addr && have_v6_addr) {
617 break;
623 TALLOC_FREE(dcs);
625 if (ret_count == 0) {
626 TALLOC_FREE(dclist);
627 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
630 *returned_dclist = dclist;
631 *return_count = ret_count;
632 return NT_STATUS_OK;
635 /****************************************************************
636 ****************************************************************/
638 static NTSTATUS make_domain_controller_info(TALLOC_CTX *mem_ctx,
639 const char *dc_unc,
640 const char *dc_address,
641 uint32_t dc_address_type,
642 const struct GUID *domain_guid,
643 const char *domain_name,
644 const char *forest_name,
645 uint32_t flags,
646 const char *dc_site_name,
647 const char *client_site_name,
648 struct netr_DsRGetDCNameInfo **info_out)
650 struct netr_DsRGetDCNameInfo *info;
652 info = talloc_zero(mem_ctx, struct netr_DsRGetDCNameInfo);
653 NT_STATUS_HAVE_NO_MEMORY(info);
655 if (dc_unc) {
656 if (!(dc_unc[0] == '\\' && dc_unc[1] == '\\')) {
657 info->dc_unc = talloc_asprintf(mem_ctx, "\\\\%s",
658 dc_unc);
659 } else {
660 info->dc_unc = talloc_strdup(mem_ctx, dc_unc);
662 NT_STATUS_HAVE_NO_MEMORY(info->dc_unc);
665 if (dc_address) {
666 if (!(dc_address[0] == '\\' && dc_address[1] == '\\')) {
667 info->dc_address = talloc_asprintf(mem_ctx, "\\\\%s",
668 dc_address);
669 } else {
670 info->dc_address = talloc_strdup(mem_ctx, dc_address);
672 NT_STATUS_HAVE_NO_MEMORY(info->dc_address);
675 info->dc_address_type = dc_address_type;
677 if (domain_guid) {
678 info->domain_guid = *domain_guid;
681 if (domain_name) {
682 info->domain_name = talloc_strdup(mem_ctx, domain_name);
683 NT_STATUS_HAVE_NO_MEMORY(info->domain_name);
686 if (forest_name && *forest_name) {
687 info->forest_name = talloc_strdup(mem_ctx, forest_name);
688 NT_STATUS_HAVE_NO_MEMORY(info->forest_name);
689 flags |= DS_DNS_FOREST_ROOT;
692 info->dc_flags = flags;
694 if (dc_site_name) {
695 info->dc_site_name = talloc_strdup(mem_ctx, dc_site_name);
696 NT_STATUS_HAVE_NO_MEMORY(info->dc_site_name);
699 if (client_site_name) {
700 info->client_site_name = talloc_strdup(mem_ctx,
701 client_site_name);
702 NT_STATUS_HAVE_NO_MEMORY(info->client_site_name);
705 *info_out = info;
707 return NT_STATUS_OK;
710 /****************************************************************
711 ****************************************************************/
713 static void map_dc_and_domain_names(uint32_t flags,
714 const char *dc_name,
715 const char *domain_name,
716 const char *dns_dc_name,
717 const char *dns_domain_name,
718 uint32_t *dc_flags,
719 const char **hostname_p,
720 const char **domain_p)
722 switch (flags & 0xf0000000) {
723 case DS_RETURN_FLAT_NAME:
724 if (dc_name && domain_name &&
725 *dc_name && *domain_name) {
726 *hostname_p = dc_name;
727 *domain_p = domain_name;
728 break;
731 FALL_THROUGH;
732 case DS_RETURN_DNS_NAME:
733 default:
734 if (dns_dc_name && dns_domain_name &&
735 *dns_dc_name && *dns_domain_name) {
736 *hostname_p = dns_dc_name;
737 *domain_p = dns_domain_name;
738 *dc_flags |= DS_DNS_DOMAIN | DS_DNS_CONTROLLER;
739 break;
741 if (dc_name && domain_name &&
742 *dc_name && *domain_name) {
743 *hostname_p = dc_name;
744 *domain_p = domain_name;
745 break;
750 /****************************************************************
751 ****************************************************************/
753 static NTSTATUS make_dc_info_from_cldap_reply(
754 TALLOC_CTX *mem_ctx,
755 uint32_t flags,
756 const struct samba_sockaddr *sa,
757 struct NETLOGON_SAM_LOGON_RESPONSE_EX *r,
758 struct netr_DsRGetDCNameInfo **info)
760 const char *dc_hostname = NULL;
761 const char *dc_domain_name = NULL;
762 const char *dc_address = NULL;
763 const char *dc_forest = NULL;
764 uint32_t dc_address_type = 0;
765 uint32_t dc_flags = 0;
766 struct GUID *dc_domain_guid = NULL;
767 const char *dc_server_site = NULL;
768 const char *dc_client_site = NULL;
770 char addr[INET6_ADDRSTRLEN];
772 if (sa != NULL) {
773 print_sockaddr(addr, sizeof(addr), &sa->u.ss);
774 dc_address = addr;
775 dc_address_type = DS_ADDRESS_TYPE_INET;
776 } else {
777 if (r->sockaddr.pdc_ip) {
778 dc_address = r->sockaddr.pdc_ip;
779 dc_address_type = DS_ADDRESS_TYPE_INET;
780 } else {
781 dc_address = r->pdc_name;
782 dc_address_type = DS_ADDRESS_TYPE_NETBIOS;
786 map_dc_and_domain_names(flags,
787 r->pdc_name,
788 r->domain_name,
789 r->pdc_dns_name,
790 r->dns_domain,
791 &dc_flags,
792 &dc_hostname,
793 &dc_domain_name);
795 dc_flags |= r->server_type;
796 dc_forest = r->forest;
797 dc_domain_guid = &r->domain_uuid;
798 dc_server_site = r->server_site;
799 dc_client_site = r->client_site;
801 return make_domain_controller_info(mem_ctx,
802 dc_hostname,
803 dc_address,
804 dc_address_type,
805 dc_domain_guid,
806 dc_domain_name,
807 dc_forest,
808 dc_flags,
809 dc_server_site,
810 dc_client_site,
811 info);
814 /****************************************************************
815 ****************************************************************/
817 static uint32_t map_ds_flags_to_nt_version(uint32_t flags)
819 uint32_t nt_version = 0;
821 if (flags & DS_PDC_REQUIRED) {
822 nt_version |= NETLOGON_NT_VERSION_PDC;
825 if (flags & DS_GC_SERVER_REQUIRED) {
826 nt_version |= NETLOGON_NT_VERSION_GC;
829 if (flags & DS_TRY_NEXTCLOSEST_SITE) {
830 nt_version |= NETLOGON_NT_VERSION_WITH_CLOSEST_SITE;
833 if (flags & DS_IP_REQUIRED) {
834 nt_version |= NETLOGON_NT_VERSION_IP;
837 return nt_version;
840 /****************************************************************
841 ****************************************************************/
843 static NTSTATUS process_dc_dns(TALLOC_CTX *mem_ctx,
844 const char *domain_name,
845 uint32_t flags,
846 struct ip_service_name *dclist,
847 size_t num_dcs,
848 struct netr_DsRGetDCNameInfo **info)
850 size_t i = 0;
851 bool valid_dc = false;
852 struct netlogon_samlogon_response *r = NULL;
853 uint32_t nt_version = NETLOGON_NT_VERSION_5 |
854 NETLOGON_NT_VERSION_5EX;
855 uint32_t ret_flags = 0;
856 NTSTATUS status;
858 nt_version |= map_ds_flags_to_nt_version(flags);
860 for (i=0; i<num_dcs; i++) {
861 char addr[INET6_ADDRSTRLEN];
863 print_sockaddr(addr, sizeof(addr), &dclist[i].sa.u.ss);
865 DEBUG(10,("LDAP ping to %s (%s)\n", dclist[i].hostname, addr));
867 if (ads_cldap_netlogon(mem_ctx, &dclist[i].sa.u.ss,
868 domain_name,
869 nt_version,
870 &r))
872 nt_version = r->ntver;
873 ret_flags = get_cldap_reply_server_flags(r, nt_version);
875 if (check_cldap_reply_required_flags(ret_flags, flags)) {
876 valid_dc = true;
877 break;
881 continue;
884 if (!valid_dc) {
885 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
888 status = make_dc_info_from_cldap_reply(mem_ctx, flags, &dclist[i].sa,
889 &r->data.nt5_ex, info);
890 if (NT_STATUS_IS_OK(status)) {
891 return store_cldap_reply(mem_ctx, flags, &dclist[i].sa,
892 nt_version, &r->data.nt5_ex);
895 return status;
898 /****************************************************************
899 ****************************************************************/
901 /****************************************************************
902 ****************************************************************/
904 static NTSTATUS process_dc_netbios(TALLOC_CTX *mem_ctx,
905 struct messaging_context *msg_ctx,
906 const char *domain_name,
907 uint32_t flags,
908 struct ip_service_name *dclist,
909 size_t num_dcs,
910 struct netr_DsRGetDCNameInfo **info)
912 enum nbt_name_type name_type = NBT_NAME_LOGON;
913 NTSTATUS status;
914 size_t i;
915 const char *dc_name = NULL;
916 fstring tmp_dc_name;
917 struct netlogon_samlogon_response *r = NULL;
918 bool store_cache = false;
919 uint32_t nt_version = NETLOGON_NT_VERSION_1 |
920 NETLOGON_NT_VERSION_5 |
921 NETLOGON_NT_VERSION_5EX_WITH_IP;
922 size_t len = strlen(lp_netbios_name());
923 char my_acct_name[len+2];
925 if (msg_ctx == NULL) {
926 return NT_STATUS_INVALID_PARAMETER;
929 if (flags & DS_PDC_REQUIRED) {
930 name_type = NBT_NAME_PDC;
934 * It's 2024 we always want an AD style response!
936 nt_version |= NETLOGON_NT_VERSION_AVOID_NT4EMUL;
938 nt_version |= map_ds_flags_to_nt_version(flags);
940 snprintf(my_acct_name,
941 sizeof(my_acct_name),
942 "%s$",
943 lp_netbios_name());
945 DEBUG(10,("process_dc_netbios\n"));
947 for (i=0; i<num_dcs; i++) {
948 uint16_t val;
950 generate_random_buffer((uint8_t *)&val, 2);
952 status = nbt_getdc(msg_ctx, 10, &dclist[i].sa.u.ss, domain_name,
953 NULL, my_acct_name, ACB_WSTRUST, nt_version,
954 mem_ctx, &nt_version, &dc_name, &r);
955 if (NT_STATUS_IS_OK(status)) {
956 store_cache = true;
957 namecache_store(dc_name,
958 NBT_NAME_SERVER,
960 &dclist[i].sa);
961 goto make_reply;
964 if (name_status_find(domain_name,
965 name_type,
966 NBT_NAME_SERVER,
967 &dclist[i].sa.u.ss,
968 tmp_dc_name))
970 struct NETLOGON_SAM_LOGON_RESPONSE_NT40 logon1;
972 r = talloc_zero(mem_ctx, struct netlogon_samlogon_response);
973 NT_STATUS_HAVE_NO_MEMORY(r);
975 ZERO_STRUCT(logon1);
977 nt_version = NETLOGON_NT_VERSION_1;
979 logon1.nt_version = nt_version;
980 logon1.pdc_name = tmp_dc_name;
981 logon1.domain_name = talloc_strdup_upper(mem_ctx, domain_name);
982 NT_STATUS_HAVE_NO_MEMORY(logon1.domain_name);
984 r->data.nt4 = logon1;
985 r->ntver = nt_version;
987 map_netlogon_samlogon_response(r);
989 namecache_store(tmp_dc_name,
990 NBT_NAME_SERVER,
992 &dclist[i].sa);
994 goto make_reply;
998 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
1000 make_reply:
1002 status = make_dc_info_from_cldap_reply(mem_ctx, flags, &dclist[i].sa,
1003 &r->data.nt5_ex, info);
1004 if (NT_STATUS_IS_OK(status) && store_cache) {
1005 return store_cldap_reply(mem_ctx, flags, &dclist[i].sa,
1006 nt_version, &r->data.nt5_ex);
1009 return status;
1012 /****************************************************************
1013 ****************************************************************/
1015 static NTSTATUS dsgetdcname_rediscover(TALLOC_CTX *mem_ctx,
1016 struct messaging_context *msg_ctx,
1017 const char *domain_name,
1018 const struct GUID *domain_guid,
1019 uint32_t flags,
1020 const char *site_name,
1021 struct netr_DsRGetDCNameInfo **info)
1023 NTSTATUS status;
1024 struct ip_service_name *dclist = NULL;
1025 size_t num_dcs = 0;
1027 DEBUG(10,("dsgetdcname_rediscover\n"));
1029 if (flags & DS_IS_FLAT_NAME) {
1031 if (lp_disable_netbios()) {
1032 return NT_STATUS_NOT_SUPPORTED;
1035 status = discover_dc_netbios(mem_ctx, domain_name, flags,
1036 &dclist, &num_dcs);
1037 NT_STATUS_NOT_OK_RETURN(status);
1039 return process_dc_netbios(mem_ctx, msg_ctx, domain_name, flags,
1040 dclist, num_dcs, info);
1043 if (flags & DS_IS_DNS_NAME) {
1045 status = discover_dc_dns(mem_ctx, domain_name, domain_guid,
1046 flags, site_name, &dclist, &num_dcs);
1047 NT_STATUS_NOT_OK_RETURN(status);
1049 return process_dc_dns(mem_ctx, domain_name, flags,
1050 dclist, num_dcs, info);
1053 status = discover_dc_dns(mem_ctx, domain_name, domain_guid, flags,
1054 site_name, &dclist, &num_dcs);
1056 if (NT_STATUS_IS_OK(status) && num_dcs != 0) {
1058 status = process_dc_dns(mem_ctx, domain_name, flags, dclist,
1059 num_dcs, info);
1060 if (NT_STATUS_IS_OK(status)) {
1061 return status;
1065 if (lp_disable_netbios()) {
1066 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
1069 status = discover_dc_netbios(mem_ctx, domain_name, flags, &dclist,
1070 &num_dcs);
1071 NT_STATUS_NOT_OK_RETURN(status);
1073 return process_dc_netbios(mem_ctx, msg_ctx, domain_name, flags, dclist,
1074 num_dcs, info);
1077 static bool is_closest_site(struct netr_DsRGetDCNameInfo *info)
1079 if (info->dc_flags & DS_SERVER_CLOSEST) {
1080 return true;
1083 if (!info->client_site_name) {
1084 return true;
1087 if (!info->dc_site_name) {
1088 return false;
1091 if (strcmp(info->client_site_name, info->dc_site_name) == 0) {
1092 return true;
1095 return false;
1098 /********************************************************************
1099 Internal dsgetdcname.
1100 ********************************************************************/
1102 static NTSTATUS dsgetdcname_internal(TALLOC_CTX *mem_ctx,
1103 struct messaging_context *msg_ctx,
1104 const char *domain_name,
1105 const struct GUID *domain_guid,
1106 const char *site_name,
1107 uint32_t flags,
1108 struct netr_DsRGetDCNameInfo **info)
1110 NTSTATUS status;
1111 struct netr_DsRGetDCNameInfo *myinfo = NULL;
1112 bool first = true;
1113 struct netr_DsRGetDCNameInfo *first_info = NULL;
1115 DEBUG(10,("dsgetdcname_internal: domain_name: %s, "
1116 "domain_guid: %s, site_name: %s, flags: 0x%08x\n",
1117 domain_name,
1118 domain_guid ? GUID_string(mem_ctx, domain_guid) : "(null)",
1119 site_name ? site_name : "(null)", flags));
1121 *info = NULL;
1123 if (!check_allowed_required_flags(flags, site_name)) {
1124 DEBUG(0,("invalid flags specified\n"));
1125 return NT_STATUS_INVALID_PARAMETER;
1128 if (flags & DS_FORCE_REDISCOVERY) {
1129 goto rediscover;
1132 status = dsgetdcname_cached(mem_ctx, msg_ctx, domain_name, domain_guid,
1133 flags, site_name, &myinfo);
1134 if (NT_STATUS_IS_OK(status)) {
1135 *info = myinfo;
1136 goto done;
1139 if (flags & DS_BACKGROUND_ONLY) {
1140 goto done;
1143 rediscover:
1144 status = dsgetdcname_rediscover(mem_ctx, msg_ctx, domain_name,
1145 domain_guid, flags, site_name,
1146 &myinfo);
1148 done:
1149 if (!NT_STATUS_IS_OK(status)) {
1150 if (!first) {
1151 *info = first_info;
1152 return NT_STATUS_OK;
1154 return status;
1157 if (!first) {
1158 TALLOC_FREE(first_info);
1159 } else if (!is_closest_site(myinfo)) {
1160 first = false;
1161 first_info = myinfo;
1162 /* TODO: may use the next_closest_site here */
1163 site_name = myinfo->client_site_name;
1164 goto rediscover;
1167 *info = myinfo;
1168 return NT_STATUS_OK;
1171 /********************************************************************
1172 dsgetdcname.
1174 This will be the only public function here.
1175 ********************************************************************/
1177 NTSTATUS dsgetdcname(TALLOC_CTX *mem_ctx,
1178 struct messaging_context *msg_ctx,
1179 const char *domain_name,
1180 const struct GUID *domain_guid,
1181 const char *site_name,
1182 uint32_t flags,
1183 struct netr_DsRGetDCNameInfo **info)
1185 NTSTATUS status;
1186 const char *query_site = NULL;
1187 char *ptr_to_free = NULL;
1188 bool retry_query_with_null = false;
1190 if ((site_name == NULL) || (site_name[0] == '\0')) {
1191 ptr_to_free = sitename_fetch(mem_ctx, domain_name);
1192 if (ptr_to_free != NULL) {
1193 retry_query_with_null = true;
1195 query_site = ptr_to_free;
1196 } else {
1197 query_site = site_name;
1200 status = dsgetdcname_internal(mem_ctx,
1201 msg_ctx,
1202 domain_name,
1203 domain_guid,
1204 query_site,
1205 flags,
1206 info);
1208 TALLOC_FREE(ptr_to_free);
1210 if (!NT_STATUS_EQUAL(status, NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND)) {
1211 return status;
1214 /* Should we try again with site_name == NULL ? */
1215 if (retry_query_with_null) {
1216 status = dsgetdcname_internal(mem_ctx,
1217 msg_ctx,
1218 domain_name,
1219 domain_guid,
1220 NULL,
1221 flags,
1222 info);
1225 return status;
1228 NTSTATUS dsgetonedcname(TALLOC_CTX *mem_ctx,
1229 struct messaging_context *msg_ctx,
1230 const char *domain_name,
1231 const char *dcname,
1232 uint32_t flags,
1233 struct netr_DsRGetDCNameInfo **info)
1235 NTSTATUS status;
1236 struct sockaddr_storage *addrs;
1237 unsigned int num_addrs, i;
1238 const char *hostname = strip_hostname(dcname);
1240 status = resolve_name_list(mem_ctx, hostname, 0x20,
1241 &addrs, &num_addrs);
1242 if (!NT_STATUS_IS_OK(status)) {
1243 return status;
1246 for (i = 0; i < num_addrs; i++) {
1248 bool ok;
1249 struct ip_service_name dclist;
1251 dclist.hostname = hostname;
1252 ok = sockaddr_storage_to_samba_sockaddr(&dclist.sa, &addrs[i]);
1253 if (!ok) {
1254 TALLOC_FREE(addrs);
1255 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
1258 status = process_dc_dns(mem_ctx, domain_name, flags,
1259 &dclist, 1, info);
1260 if (NT_STATUS_IS_OK(status)) {
1261 TALLOC_FREE(addrs);
1262 return NT_STATUS_OK;
1265 if (lp_disable_netbios()) {
1266 continue;
1269 status = process_dc_netbios(mem_ctx, msg_ctx, domain_name, flags,
1270 &dclist, 1, info);
1271 if (NT_STATUS_IS_OK(status)) {
1272 TALLOC_FREE(addrs);
1273 return NT_STATUS_OK;
1277 TALLOC_FREE(addrs);
1278 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;