2 Unix SMB/CIFS implementation.
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/>.
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"
35 #define DSGETDCNAME_CACHE_TTL 60*15
37 struct ip_service_name {
38 struct samba_sockaddr sa;
42 static NTSTATUS make_dc_info_from_cldap_reply(
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 "));
121 DEBUGADD(lvl,("\n"));
124 /****************************************************************
125 ****************************************************************/
127 static char *dsgetdcname_cache_key(TALLOC_CTX *mem_ctx, const char *domain)
133 return talloc_asprintf_strupper_m(mem_ctx, "DSGETDCNAME/DOMAIN/%s",
137 /****************************************************************
138 ****************************************************************/
140 static NTSTATUS dsgetdcname_cache_delete(TALLOC_CTX *mem_ctx,
141 const char *domain_name)
145 key = dsgetdcname_cache_key(mem_ctx, domain_name);
147 return NT_STATUS_NO_MEMORY;
150 if (!gencache_del(key)) {
151 return NT_STATUS_UNSUCCESSFUL;
157 /****************************************************************
158 ****************************************************************/
160 static NTSTATUS dsgetdcname_cache_store(TALLOC_CTX *mem_ctx,
161 const char *domain_name,
168 key = dsgetdcname_cache_key(mem_ctx, domain_name);
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,
185 struct samba_sockaddr *sa,
187 struct NETLOGON_SAM_LOGON_RESPONSE_EX *r)
190 enum ndr_err_code ndr_err;
192 char addr[INET6_ADDRSTRLEN];
194 print_sockaddr(addr, sizeof(addr), &sa->u.ss);
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,
210 if (!NT_STATUS_IS_OK(status)) {
213 if (r->client_site) {
214 sitename_store(r->domain_name, r->client_site);
218 status = dsgetdcname_cache_store(mem_ctx, r->dns_domain, blob);
219 if (!NT_STATUS_IS_OK(status)) {
222 if (r->client_site) {
223 sitename_store(r->dns_domain, r->client_site);
227 status = NT_STATUS_OK;
230 data_blob_free(&blob);
235 /****************************************************************
236 ****************************************************************/
238 static uint32_t get_cldap_reply_server_flags(struct netlogon_samlogon_response *r,
241 switch (nt_version & 0x0000001f) {
251 return r->data.nt5.server_type;
256 return r->data.nt5_ex.server_type;
265 return r->data.nt5_ex.server_type;
275 return r->data.nt5_ex.server_type;
279 return r->data.nt5_ex.server_type;
285 /****************************************************************
286 ****************************************************************/
288 static NTSTATUS dsgetdcname_cache_fetch(TALLOC_CTX *mem_ctx,
289 const char *domain_name,
290 const struct GUID *domain_guid,
292 struct netr_DsRGetDCNameInfo **info_p)
296 enum ndr_err_code ndr_err;
297 struct netr_DsRGetDCNameInfo *info;
298 struct NETLOGON_SAM_LOGON_RESPONSE_EX r;
301 key = dsgetdcname_cache_key(mem_ctx, domain_name);
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);
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,
327 if (!NT_STATUS_IS_OK(status)) {
331 if (DEBUGLEVEL >= 10) {
332 NDR_PRINT_DEBUG(netr_DsRGetDCNameInfo, info);
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;
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,
359 const char *site_name,
360 struct netr_DsRGetDCNameInfo **info)
364 status = dsgetdcname_cache_fetch(mem_ctx, domain_name, domain_guid,
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",
370 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
373 if (flags & DS_BACKGROUND_ONLY) {
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,
385 if (!NT_STATUS_IS_OK(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) {
414 if (return_type == (DS_RETURN_FLAT_NAME|DS_RETURN_DNS_NAME)) {
418 if (offered_type == (DS_IS_DNS_NAME|DS_IS_FLAT_NAME)) {
422 if (query_type == (DS_BACKGROUND_ONLY|DS_FORCE_REDISCOVERY)) {
427 if ((flags & DS_RETURN_DNS_NAME) && (!(flags & DS_IP_REQUIRED))) {
428 printf("gd: here5 \n");
435 /****************************************************************
436 ****************************************************************/
438 static NTSTATUS discover_dc_netbios(TALLOC_CTX *mem_ctx,
439 const char *domain_name,
441 struct ip_service_name **returned_dclist,
442 size_t *returned_count)
445 enum nbt_name_type name_type = NBT_NAME_LOGON;
446 struct samba_sockaddr *salist = NULL;
448 struct ip_service_name *dclist = NULL;
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,
463 if (!NT_STATUS_IS_OK(status)) {
464 DEBUG(10,("discover_dc_netbios: failed to find DC\n"));
468 dclist = talloc_zero_array(mem_ctx, struct ip_service_name, count);
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),
482 r->hostname = talloc_strdup(mem_ctx, addr);
486 return NT_STATUS_NO_MEMORY;
493 *returned_dclist = dclist;
494 *returned_count = count;
499 /****************************************************************
500 ****************************************************************/
502 static NTSTATUS discover_dc_dns(TALLOC_CTX *mem_ctx,
503 const char *domain_name,
504 const struct GUID *domain_guid,
506 const char *site_name,
507 struct ip_service_name **returned_dclist,
508 size_t *return_count)
512 struct dns_rr_srv *dcs = NULL;
514 struct ip_service_name *dclist = NULL;
515 size_t ret_count = 0;
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);
530 query = ads_dns_query_string_dcs(mem_ctx, domain_name);
534 return NT_STATUS_NO_MEMORY;
537 status = ads_dns_query_srv(
539 lp_get_async_dns_timeout(),
545 if (!NT_STATUS_IS_OK(status)) {
551 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
554 /* Check for integer wrap. */
555 if (numdcs + numdcs < numdcs) {
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,
570 return NT_STATUS_NO_MEMORY;
574 * First, copy the SRV record replies that
575 * have IP addresses returned with them.
578 for (i = 0; i < numdcs; i++) {
580 bool have_v4_addr = false;
581 bool have_v6_addr = false;
583 if (dcs[i].num_ips == 0) {
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)) {
600 dclist[ret_count].hostname =
601 talloc_strdup(dclist, dcs[i].hostname);
602 ok = sockaddr_storage_to_samba_sockaddr(
603 &dclist[ret_count].sa,
608 return NT_STATUS_INVALID_PARAMETER;
611 if (dcs[i].ss_s[j].ss_family == AF_INET) {
616 if (have_v4_addr && have_v6_addr) {
625 if (ret_count == 0) {
627 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
630 *returned_dclist = dclist;
631 *return_count = ret_count;
635 /****************************************************************
636 ****************************************************************/
638 static NTSTATUS make_domain_controller_info(TALLOC_CTX *mem_ctx,
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,
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);
656 if (!(dc_unc[0] == '\\' && dc_unc[1] == '\\')) {
657 info->dc_unc = talloc_asprintf(mem_ctx, "\\\\%s",
660 info->dc_unc = talloc_strdup(mem_ctx, dc_unc);
662 NT_STATUS_HAVE_NO_MEMORY(info->dc_unc);
666 if (!(dc_address[0] == '\\' && dc_address[1] == '\\')) {
667 info->dc_address = talloc_asprintf(mem_ctx, "\\\\%s",
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;
678 info->domain_guid = *domain_guid;
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;
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,
702 NT_STATUS_HAVE_NO_MEMORY(info->client_site_name);
710 /****************************************************************
711 ****************************************************************/
713 static void map_dc_and_domain_names(uint32_t flags,
715 const char *domain_name,
716 const char *dns_dc_name,
717 const char *dns_domain_name,
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;
732 case DS_RETURN_DNS_NAME:
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;
741 if (dc_name && domain_name &&
742 *dc_name && *domain_name) {
743 *hostname_p = dc_name;
744 *domain_p = domain_name;
750 /****************************************************************
751 ****************************************************************/
753 static NTSTATUS make_dc_info_from_cldap_reply(
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];
773 print_sockaddr(addr, sizeof(addr), &sa->u.ss);
775 dc_address_type = DS_ADDRESS_TYPE_INET;
777 if (r->sockaddr.pdc_ip) {
778 dc_address = r->sockaddr.pdc_ip;
779 dc_address_type = DS_ADDRESS_TYPE_INET;
781 dc_address = r->pdc_name;
782 dc_address_type = DS_ADDRESS_TYPE_NETBIOS;
786 map_dc_and_domain_names(flags,
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,
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;
840 /****************************************************************
841 ****************************************************************/
843 static NTSTATUS process_dc_dns(TALLOC_CTX *mem_ctx,
844 const char *domain_name,
846 struct ip_service_name *dclist,
848 struct netr_DsRGetDCNameInfo **info)
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;
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,
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)) {
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);
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,
908 struct ip_service_name *dclist,
910 struct netr_DsRGetDCNameInfo **info)
912 enum nbt_name_type name_type = NBT_NAME_LOGON;
915 const char *dc_name = NULL;
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;
933 nt_version |= map_ds_flags_to_nt_version(flags);
935 snprintf(my_acct_name,
936 sizeof(my_acct_name),
940 DEBUG(10,("process_dc_netbios\n"));
942 for (i=0; i<num_dcs; i++) {
945 generate_random_buffer((uint8_t *)&val, 2);
947 status = nbt_getdc(msg_ctx, 10, &dclist[i].sa.u.ss, domain_name,
948 NULL, my_acct_name, ACB_WSTRUST, nt_version,
949 mem_ctx, &nt_version, &dc_name, &r);
950 if (NT_STATUS_IS_OK(status)) {
952 namecache_store(dc_name,
959 if (name_status_find(domain_name,
965 struct NETLOGON_SAM_LOGON_RESPONSE_NT40 logon1;
967 r = talloc_zero(mem_ctx, struct netlogon_samlogon_response);
968 NT_STATUS_HAVE_NO_MEMORY(r);
972 nt_version = NETLOGON_NT_VERSION_1;
974 logon1.nt_version = nt_version;
975 logon1.pdc_name = tmp_dc_name;
976 logon1.domain_name = talloc_strdup_upper(mem_ctx, domain_name);
977 NT_STATUS_HAVE_NO_MEMORY(logon1.domain_name);
979 r->data.nt4 = logon1;
980 r->ntver = nt_version;
982 map_netlogon_samlogon_response(r);
984 namecache_store(tmp_dc_name,
993 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
997 status = make_dc_info_from_cldap_reply(mem_ctx, flags, &dclist[i].sa,
998 &r->data.nt5_ex, info);
999 if (NT_STATUS_IS_OK(status) && store_cache) {
1000 return store_cldap_reply(mem_ctx, flags, &dclist[i].sa,
1001 nt_version, &r->data.nt5_ex);
1007 /****************************************************************
1008 ****************************************************************/
1010 static NTSTATUS dsgetdcname_rediscover(TALLOC_CTX *mem_ctx,
1011 struct messaging_context *msg_ctx,
1012 const char *domain_name,
1013 const struct GUID *domain_guid,
1015 const char *site_name,
1016 struct netr_DsRGetDCNameInfo **info)
1019 struct ip_service_name *dclist = NULL;
1022 DEBUG(10,("dsgetdcname_rediscover\n"));
1024 if (flags & DS_IS_FLAT_NAME) {
1026 if (lp_disable_netbios()) {
1027 return NT_STATUS_NOT_SUPPORTED;
1030 status = discover_dc_netbios(mem_ctx, domain_name, flags,
1032 NT_STATUS_NOT_OK_RETURN(status);
1034 return process_dc_netbios(mem_ctx, msg_ctx, domain_name, flags,
1035 dclist, num_dcs, info);
1038 if (flags & DS_IS_DNS_NAME) {
1040 status = discover_dc_dns(mem_ctx, domain_name, domain_guid,
1041 flags, site_name, &dclist, &num_dcs);
1042 NT_STATUS_NOT_OK_RETURN(status);
1044 return process_dc_dns(mem_ctx, domain_name, flags,
1045 dclist, num_dcs, info);
1048 status = discover_dc_dns(mem_ctx, domain_name, domain_guid, flags,
1049 site_name, &dclist, &num_dcs);
1051 if (NT_STATUS_IS_OK(status) && num_dcs != 0) {
1053 status = process_dc_dns(mem_ctx, domain_name, flags, dclist,
1055 if (NT_STATUS_IS_OK(status)) {
1060 if (lp_disable_netbios()) {
1061 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
1064 status = discover_dc_netbios(mem_ctx, domain_name, flags, &dclist,
1066 NT_STATUS_NOT_OK_RETURN(status);
1068 return process_dc_netbios(mem_ctx, msg_ctx, domain_name, flags, dclist,
1072 static bool is_closest_site(struct netr_DsRGetDCNameInfo *info)
1074 if (info->dc_flags & DS_SERVER_CLOSEST) {
1078 if (!info->client_site_name) {
1082 if (!info->dc_site_name) {
1086 if (strcmp(info->client_site_name, info->dc_site_name) == 0) {
1093 /********************************************************************
1094 Internal dsgetdcname.
1095 ********************************************************************/
1097 static NTSTATUS dsgetdcname_internal(TALLOC_CTX *mem_ctx,
1098 struct messaging_context *msg_ctx,
1099 const char *domain_name,
1100 const struct GUID *domain_guid,
1101 const char *site_name,
1103 struct netr_DsRGetDCNameInfo **info)
1106 struct netr_DsRGetDCNameInfo *myinfo = NULL;
1108 struct netr_DsRGetDCNameInfo *first_info = NULL;
1110 DEBUG(10,("dsgetdcname_internal: domain_name: %s, "
1111 "domain_guid: %s, site_name: %s, flags: 0x%08x\n",
1113 domain_guid ? GUID_string(mem_ctx, domain_guid) : "(null)",
1114 site_name ? site_name : "(null)", flags));
1118 if (!check_allowed_required_flags(flags, site_name)) {
1119 DEBUG(0,("invalid flags specified\n"));
1120 return NT_STATUS_INVALID_PARAMETER;
1123 if (flags & DS_FORCE_REDISCOVERY) {
1127 status = dsgetdcname_cached(mem_ctx, msg_ctx, domain_name, domain_guid,
1128 flags, site_name, &myinfo);
1129 if (NT_STATUS_IS_OK(status)) {
1134 if (flags & DS_BACKGROUND_ONLY) {
1139 status = dsgetdcname_rediscover(mem_ctx, msg_ctx, domain_name,
1140 domain_guid, flags, site_name,
1144 if (!NT_STATUS_IS_OK(status)) {
1147 return NT_STATUS_OK;
1153 TALLOC_FREE(first_info);
1154 } else if (!is_closest_site(myinfo)) {
1156 first_info = myinfo;
1157 /* TODO: may use the next_closest_site here */
1158 site_name = myinfo->client_site_name;
1163 return NT_STATUS_OK;
1166 /********************************************************************
1169 This will be the only public function here.
1170 ********************************************************************/
1172 NTSTATUS dsgetdcname(TALLOC_CTX *mem_ctx,
1173 struct messaging_context *msg_ctx,
1174 const char *domain_name,
1175 const struct GUID *domain_guid,
1176 const char *site_name,
1178 struct netr_DsRGetDCNameInfo **info)
1181 const char *query_site = NULL;
1182 char *ptr_to_free = NULL;
1183 bool retry_query_with_null = false;
1185 if ((site_name == NULL) || (site_name[0] == '\0')) {
1186 ptr_to_free = sitename_fetch(mem_ctx, domain_name);
1187 if (ptr_to_free != NULL) {
1188 retry_query_with_null = true;
1190 query_site = ptr_to_free;
1192 query_site = site_name;
1195 status = dsgetdcname_internal(mem_ctx,
1203 TALLOC_FREE(ptr_to_free);
1205 if (!NT_STATUS_EQUAL(status, NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND)) {
1209 /* Should we try again with site_name == NULL ? */
1210 if (retry_query_with_null) {
1211 status = dsgetdcname_internal(mem_ctx,
1223 NTSTATUS dsgetonedcname(TALLOC_CTX *mem_ctx,
1224 struct messaging_context *msg_ctx,
1225 const char *domain_name,
1228 struct netr_DsRGetDCNameInfo **info)
1231 struct sockaddr_storage *addrs;
1232 unsigned int num_addrs, i;
1233 const char *hostname = strip_hostname(dcname);
1235 status = resolve_name_list(mem_ctx, hostname, 0x20,
1236 &addrs, &num_addrs);
1237 if (!NT_STATUS_IS_OK(status)) {
1241 for (i = 0; i < num_addrs; i++) {
1244 struct ip_service_name dclist;
1246 dclist.hostname = hostname;
1247 ok = sockaddr_storage_to_samba_sockaddr(&dclist.sa, &addrs[i]);
1250 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
1253 status = process_dc_dns(mem_ctx, domain_name, flags,
1255 if (NT_STATUS_IS_OK(status)) {
1257 return NT_STATUS_OK;
1260 if (lp_disable_netbios()) {
1264 status = process_dc_netbios(mem_ctx, msg_ctx, domain_name, flags,
1266 if (NT_STATUS_IS_OK(status)) {
1268 return NT_STATUS_OK;
1273 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;