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.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
& 0x000000002)
59 DEBUGADD(lvl
,("0x00000002 "));
60 if (flags
& 0x000000004)
61 DEBUGADD(lvl
,("0x00000004 "));
62 if (flags
& 0x000000008)
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
& 0x00040000)
93 DEBUGADD(lvl
,("0x00040000 "));
94 if (flags
& 0x00080000)
95 DEBUGADD(lvl
,("0x00080000 "));
96 if (flags
& 0x00100000)
97 DEBUGADD(lvl
,("0x00100000 "));
98 if (flags
& 0x00200000)
99 DEBUGADD(lvl
,("0x00200000 "));
100 if (flags
& 0x00400000)
101 DEBUGADD(lvl
,("0x00400000 "));
102 if (flags
& 0x00800000)
103 DEBUGADD(lvl
,("0x00800000 "));
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 excluse 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;
516 size_t num_dns_lookups
= 0;
517 const char **dns_lookups
= NULL
;
518 size_t num_lookups_ret
= 0;
519 struct samba_sockaddr
*dns_addrs_ret
= NULL
;
520 char **dns_lookups_ret
= NULL
;
522 if (flags
& DS_PDC_REQUIRED
) {
523 status
= ads_dns_query_pdc(mem_ctx
,
527 } else if (flags
& DS_GC_SERVER_REQUIRED
) {
528 status
= ads_dns_query_gcs(mem_ctx
,
533 } else if (flags
& DS_KDC_REQUIRED
) {
534 status
= ads_dns_query_kdcs(mem_ctx
,
539 } else if (flags
& DS_DIRECTORY_SERVICE_REQUIRED
) {
540 status
= ads_dns_query_dcs(mem_ctx
,
545 } else if (domain_guid
) {
546 struct GUID_txt_buf buf
;
547 GUID_buf_string(domain_guid
, &buf
);
549 status
= ads_dns_query_dcs_guid(mem_ctx
,
555 status
= ads_dns_query_dcs(mem_ctx
,
562 if (!NT_STATUS_IS_OK(status
)) {
569 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND
;
573 * We're only returning one address per
574 * DC name, so just allocate size numdcs.
577 dclist
= talloc_zero_array(mem_ctx
,
578 struct ip_service_name
,
582 return NT_STATUS_NO_MEMORY
;
586 * First, copy the SRV record replies that
587 * have IP addresses returned with them.
590 for (i
= 0; i
< numdcs
; i
++) {
593 if (dcs
[i
].num_ips
== 0) {
595 * No addresses returned in the SRV
596 * reply, we must look this up via
603 dclist
[ret_count
].hostname
=
604 talloc_move(mem_ctx
, &dcs
[i
].hostname
);
607 * Pick the first IPv4 address,
608 * if none pick the first address.
610 * This is different from the previous
611 * code which picked a 'next ip' address
612 * each time, incrementing an index.
613 * Too complex to maintain :-(.
615 for (j
= 0; j
< dcs
[i
].num_ips
; j
++) {
616 if (dcs
[i
].ss_s
[j
].ss_family
== AF_INET
) {
618 ok
= sockaddr_storage_to_samba_sockaddr(
619 &dclist
[ret_count
].sa
,
624 return NT_STATUS_INVALID_PARAMETER
;
629 if (j
== dcs
[i
].num_ips
) {
630 /* No IPv4- use the first IPv6 addr. */
632 ok
= sockaddr_storage_to_samba_sockaddr(
633 &dclist
[ret_count
].sa
,
638 return NT_STATUS_INVALID_PARAMETER
;
645 * Now, create an array of hostnames we
646 * will asynchronously look up in DNS.
648 dns_lookups
= talloc_zero_array(mem_ctx
,
651 if (dns_lookups
== NULL
) {
654 return NT_STATUS_NO_MEMORY
;
658 for (i
= 0; i
< numdcs
; i
++) {
659 if (dcs
[i
].num_ips
!= 0) {
661 * We already got an address
662 * for this via SRV return.
666 dns_lookups
[num_dns_lookups
] =
667 talloc_strdup(mem_ctx
, dcs
[i
].hostname
);
668 if (dns_lookups
[num_dns_lookups
] == NULL
) {
671 TALLOC_FREE(dns_lookups
);
676 status
= dns_lookup_list_async(mem_ctx
,
682 if (!NT_STATUS_IS_OK(status
)) {
685 TALLOC_FREE(dns_lookups
);
690 * Remember, if we timed out or some
691 * addresses didn't look up then
692 * num_dns_lookups > num_lookups_ret,
693 * so there isn't a one to one correspondence.
695 * The order of the requests is preserved
696 * in the replies, but there may be requests
697 * for which there are no replies if we
700 * For example this means for looking up
706 * The replies might look like:
713 * But they will never be in the order:
719 * Also in dns_lookup_list_async()
720 * we arrange the requests/replies in
721 * the order IPv4 followed by IPv6, so in
722 * the replies, we can always pick the first
723 * reply - we know it will be IPv4 if there
724 * is an IPv4 for that name.
726 * Here ret_count is the index into the
727 * next entry in dclist we must fill (may
731 for (i
= 0; i
< num_lookups_ret
; i
++) {
732 dclist
[ret_count
].hostname
=
733 talloc_move(mem_ctx
, &dns_lookups_ret
[i
]);
734 dclist
[ret_count
].sa
= dns_addrs_ret
[i
];
736 * Is this a duplicate name return.
737 * Remember we can look up both A and
738 * AAAA records which can return multiple
739 * addresses and the order of names
743 if (strcmp(dclist
[ret_count
-1].hostname
,
744 dclist
[ret_count
].hostname
) == 0) {
746 * Same name. Keep the first address
747 * which is IPv4 by preference.
754 if (ret_count
== numdcs
) {
755 /* We've filled the return array. */
762 TALLOC_FREE(dns_lookups
);
763 *returned_dclist
= dclist
;
764 *return_count
= ret_count
;
770 TALLOC_FREE(dns_lookups
);
771 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND
;
774 /****************************************************************
775 ****************************************************************/
777 static NTSTATUS
make_domain_controller_info(TALLOC_CTX
*mem_ctx
,
779 const char *dc_address
,
780 uint32_t dc_address_type
,
781 const struct GUID
*domain_guid
,
782 const char *domain_name
,
783 const char *forest_name
,
785 const char *dc_site_name
,
786 const char *client_site_name
,
787 struct netr_DsRGetDCNameInfo
**info_out
)
789 struct netr_DsRGetDCNameInfo
*info
;
791 info
= talloc_zero(mem_ctx
, struct netr_DsRGetDCNameInfo
);
792 NT_STATUS_HAVE_NO_MEMORY(info
);
795 info
->dc_unc
= talloc_strdup(mem_ctx
, dc_unc
);
796 NT_STATUS_HAVE_NO_MEMORY(info
->dc_unc
);
800 if (!(dc_address
[0] == '\\' && dc_address
[1] == '\\')) {
801 info
->dc_address
= talloc_asprintf(mem_ctx
, "\\\\%s",
804 info
->dc_address
= talloc_strdup(mem_ctx
, dc_address
);
806 NT_STATUS_HAVE_NO_MEMORY(info
->dc_address
);
809 info
->dc_address_type
= dc_address_type
;
812 info
->domain_guid
= *domain_guid
;
816 info
->domain_name
= talloc_strdup(mem_ctx
, domain_name
);
817 NT_STATUS_HAVE_NO_MEMORY(info
->domain_name
);
820 if (forest_name
&& *forest_name
) {
821 info
->forest_name
= talloc_strdup(mem_ctx
, forest_name
);
822 NT_STATUS_HAVE_NO_MEMORY(info
->forest_name
);
823 flags
|= DS_DNS_FOREST_ROOT
;
826 info
->dc_flags
= flags
;
829 info
->dc_site_name
= talloc_strdup(mem_ctx
, dc_site_name
);
830 NT_STATUS_HAVE_NO_MEMORY(info
->dc_site_name
);
833 if (client_site_name
) {
834 info
->client_site_name
= talloc_strdup(mem_ctx
,
836 NT_STATUS_HAVE_NO_MEMORY(info
->client_site_name
);
844 /****************************************************************
845 ****************************************************************/
847 static void map_dc_and_domain_names(uint32_t flags
,
849 const char *domain_name
,
850 const char *dns_dc_name
,
851 const char *dns_domain_name
,
853 const char **hostname_p
,
854 const char **domain_p
)
856 switch (flags
& 0xf0000000) {
857 case DS_RETURN_FLAT_NAME
:
858 if (dc_name
&& domain_name
&&
859 *dc_name
&& *domain_name
) {
860 *hostname_p
= dc_name
;
861 *domain_p
= domain_name
;
866 case DS_RETURN_DNS_NAME
:
868 if (dns_dc_name
&& dns_domain_name
&&
869 *dns_dc_name
&& *dns_domain_name
) {
870 *hostname_p
= dns_dc_name
;
871 *domain_p
= dns_domain_name
;
872 *dc_flags
|= DS_DNS_DOMAIN
| DS_DNS_CONTROLLER
;
875 if (dc_name
&& domain_name
&&
876 *dc_name
&& *domain_name
) {
877 *hostname_p
= dc_name
;
878 *domain_p
= domain_name
;
884 /****************************************************************
885 ****************************************************************/
887 static NTSTATUS
make_dc_info_from_cldap_reply(
890 const struct samba_sockaddr
*sa
,
891 struct NETLOGON_SAM_LOGON_RESPONSE_EX
*r
,
892 struct netr_DsRGetDCNameInfo
**info
)
894 const char *dc_hostname
= NULL
;
895 const char *dc_domain_name
= NULL
;
896 const char *dc_address
= NULL
;
897 const char *dc_forest
= NULL
;
898 uint32_t dc_address_type
= 0;
899 uint32_t dc_flags
= 0;
900 struct GUID
*dc_domain_guid
= NULL
;
901 const char *dc_server_site
= NULL
;
902 const char *dc_client_site
= NULL
;
904 char addr
[INET6_ADDRSTRLEN
];
907 print_sockaddr(addr
, sizeof(addr
), &sa
->u
.ss
);
909 dc_address_type
= DS_ADDRESS_TYPE_INET
;
911 if (r
->sockaddr
.pdc_ip
) {
912 dc_address
= r
->sockaddr
.pdc_ip
;
913 dc_address_type
= DS_ADDRESS_TYPE_INET
;
915 dc_address
= r
->pdc_name
;
916 dc_address_type
= DS_ADDRESS_TYPE_NETBIOS
;
920 map_dc_and_domain_names(flags
,
929 dc_flags
|= r
->server_type
;
930 dc_forest
= r
->forest
;
931 dc_domain_guid
= &r
->domain_uuid
;
932 dc_server_site
= r
->server_site
;
933 dc_client_site
= r
->client_site
;
935 return make_domain_controller_info(mem_ctx
,
948 /****************************************************************
949 ****************************************************************/
951 static uint32_t map_ds_flags_to_nt_version(uint32_t flags
)
953 uint32_t nt_version
= 0;
955 if (flags
& DS_PDC_REQUIRED
) {
956 nt_version
|= NETLOGON_NT_VERSION_PDC
;
959 if (flags
& DS_GC_SERVER_REQUIRED
) {
960 nt_version
|= NETLOGON_NT_VERSION_GC
;
963 if (flags
& DS_TRY_NEXTCLOSEST_SITE
) {
964 nt_version
|= NETLOGON_NT_VERSION_WITH_CLOSEST_SITE
;
967 if (flags
& DS_IP_REQUIRED
) {
968 nt_version
|= NETLOGON_NT_VERSION_IP
;
974 /****************************************************************
975 ****************************************************************/
977 static NTSTATUS
process_dc_dns(TALLOC_CTX
*mem_ctx
,
978 const char *domain_name
,
980 struct ip_service_name
*dclist
,
982 struct netr_DsRGetDCNameInfo
**info
)
985 bool valid_dc
= false;
986 struct netlogon_samlogon_response
*r
= NULL
;
987 uint32_t nt_version
= NETLOGON_NT_VERSION_5
|
988 NETLOGON_NT_VERSION_5EX
;
989 uint32_t ret_flags
= 0;
992 nt_version
|= map_ds_flags_to_nt_version(flags
);
994 for (i
=0; i
<num_dcs
; i
++) {
995 char addr
[INET6_ADDRSTRLEN
];
997 print_sockaddr(addr
, sizeof(addr
), &dclist
[i
].sa
.u
.ss
);
999 DEBUG(10,("LDAP ping to %s (%s)\n", dclist
[i
].hostname
, addr
));
1001 if (ads_cldap_netlogon(mem_ctx
, &dclist
[i
].sa
.u
.ss
,
1006 nt_version
= r
->ntver
;
1007 ret_flags
= get_cldap_reply_server_flags(r
, nt_version
);
1009 if (check_cldap_reply_required_flags(ret_flags
, flags
)) {
1019 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND
;
1022 status
= make_dc_info_from_cldap_reply(mem_ctx
, flags
, &dclist
[i
].sa
,
1023 &r
->data
.nt5_ex
, info
);
1024 if (NT_STATUS_IS_OK(status
)) {
1025 return store_cldap_reply(mem_ctx
, flags
, &dclist
[i
].sa
,
1026 nt_version
, &r
->data
.nt5_ex
);
1032 /****************************************************************
1033 ****************************************************************/
1035 /****************************************************************
1036 ****************************************************************/
1038 static NTSTATUS
process_dc_netbios(TALLOC_CTX
*mem_ctx
,
1039 struct messaging_context
*msg_ctx
,
1040 const char *domain_name
,
1042 struct ip_service_name
*dclist
,
1044 struct netr_DsRGetDCNameInfo
**info
)
1046 enum nbt_name_type name_type
= NBT_NAME_LOGON
;
1049 const char *dc_name
= NULL
;
1050 fstring tmp_dc_name
;
1051 struct netlogon_samlogon_response
*r
= NULL
;
1052 bool store_cache
= false;
1053 uint32_t nt_version
= NETLOGON_NT_VERSION_1
|
1054 NETLOGON_NT_VERSION_5
|
1055 NETLOGON_NT_VERSION_5EX_WITH_IP
;
1056 size_t len
= strlen(lp_netbios_name());
1057 char my_acct_name
[len
+2];
1059 if (msg_ctx
== NULL
) {
1060 return NT_STATUS_INVALID_PARAMETER
;
1063 if (flags
& DS_PDC_REQUIRED
) {
1064 name_type
= NBT_NAME_PDC
;
1067 nt_version
|= map_ds_flags_to_nt_version(flags
);
1069 snprintf(my_acct_name
,
1070 sizeof(my_acct_name
),
1074 DEBUG(10,("process_dc_netbios\n"));
1076 for (i
=0; i
<num_dcs
; i
++) {
1079 generate_random_buffer((uint8_t *)&val
, 2);
1081 status
= nbt_getdc(msg_ctx
, 10, &dclist
[i
].sa
.u
.ss
, domain_name
,
1082 NULL
, my_acct_name
, ACB_WSTRUST
, nt_version
,
1083 mem_ctx
, &nt_version
, &dc_name
, &r
);
1084 if (NT_STATUS_IS_OK(status
)) {
1086 namecache_store(dc_name
,
1093 if (name_status_find(domain_name
,
1099 struct NETLOGON_SAM_LOGON_RESPONSE_NT40 logon1
;
1101 r
= talloc_zero(mem_ctx
, struct netlogon_samlogon_response
);
1102 NT_STATUS_HAVE_NO_MEMORY(r
);
1104 ZERO_STRUCT(logon1
);
1106 nt_version
= NETLOGON_NT_VERSION_1
;
1108 logon1
.nt_version
= nt_version
;
1109 logon1
.pdc_name
= tmp_dc_name
;
1110 logon1
.domain_name
= talloc_strdup_upper(mem_ctx
, domain_name
);
1111 NT_STATUS_HAVE_NO_MEMORY(logon1
.domain_name
);
1113 r
->data
.nt4
= logon1
;
1114 r
->ntver
= nt_version
;
1116 map_netlogon_samlogon_response(r
);
1118 namecache_store(tmp_dc_name
,
1127 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND
;
1131 status
= make_dc_info_from_cldap_reply(mem_ctx
, flags
, &dclist
[i
].sa
,
1132 &r
->data
.nt5_ex
, info
);
1133 if (NT_STATUS_IS_OK(status
) && store_cache
) {
1134 return store_cldap_reply(mem_ctx
, flags
, &dclist
[i
].sa
,
1135 nt_version
, &r
->data
.nt5_ex
);
1141 /****************************************************************
1142 ****************************************************************/
1144 static NTSTATUS
dsgetdcname_rediscover(TALLOC_CTX
*mem_ctx
,
1145 struct messaging_context
*msg_ctx
,
1146 const char *domain_name
,
1147 const struct GUID
*domain_guid
,
1149 const char *site_name
,
1150 struct netr_DsRGetDCNameInfo
**info
)
1153 struct ip_service_name
*dclist
= NULL
;
1156 DEBUG(10,("dsgetdcname_rediscover\n"));
1158 if (flags
& DS_IS_FLAT_NAME
) {
1160 if (lp_disable_netbios()) {
1161 return NT_STATUS_NOT_SUPPORTED
;
1164 status
= discover_dc_netbios(mem_ctx
, domain_name
, flags
,
1166 NT_STATUS_NOT_OK_RETURN(status
);
1168 return process_dc_netbios(mem_ctx
, msg_ctx
, domain_name
, flags
,
1169 dclist
, num_dcs
, info
);
1172 if (flags
& DS_IS_DNS_NAME
) {
1174 status
= discover_dc_dns(mem_ctx
, domain_name
, domain_guid
,
1175 flags
, site_name
, &dclist
, &num_dcs
);
1176 NT_STATUS_NOT_OK_RETURN(status
);
1178 return process_dc_dns(mem_ctx
, domain_name
, flags
,
1179 dclist
, num_dcs
, info
);
1182 status
= discover_dc_dns(mem_ctx
, domain_name
, domain_guid
, flags
,
1183 site_name
, &dclist
, &num_dcs
);
1185 if (NT_STATUS_IS_OK(status
) && num_dcs
!= 0) {
1187 status
= process_dc_dns(mem_ctx
, domain_name
, flags
, dclist
,
1189 if (NT_STATUS_IS_OK(status
)) {
1194 if (lp_disable_netbios()) {
1195 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND
;
1198 status
= discover_dc_netbios(mem_ctx
, domain_name
, flags
, &dclist
,
1200 NT_STATUS_NOT_OK_RETURN(status
);
1202 return process_dc_netbios(mem_ctx
, msg_ctx
, domain_name
, flags
, dclist
,
1206 static bool is_closest_site(struct netr_DsRGetDCNameInfo
*info
)
1208 if (info
->dc_flags
& DS_SERVER_CLOSEST
) {
1212 if (!info
->client_site_name
) {
1216 if (!info
->dc_site_name
) {
1220 if (strcmp(info
->client_site_name
, info
->dc_site_name
) == 0) {
1227 /********************************************************************
1228 Internal dsgetdcname.
1229 ********************************************************************/
1231 static NTSTATUS
dsgetdcname_internal(TALLOC_CTX
*mem_ctx
,
1232 struct messaging_context
*msg_ctx
,
1233 const char *domain_name
,
1234 const struct GUID
*domain_guid
,
1235 const char *site_name
,
1237 struct netr_DsRGetDCNameInfo
**info
)
1240 struct netr_DsRGetDCNameInfo
*myinfo
= NULL
;
1242 struct netr_DsRGetDCNameInfo
*first_info
= NULL
;
1244 DEBUG(10,("dsgetdcname_internal: domain_name: %s, "
1245 "domain_guid: %s, site_name: %s, flags: 0x%08x\n",
1247 domain_guid
? GUID_string(mem_ctx
, domain_guid
) : "(null)",
1248 site_name
? site_name
: "(null)", flags
));
1252 if (!check_allowed_required_flags(flags
, site_name
)) {
1253 DEBUG(0,("invalid flags specified\n"));
1254 return NT_STATUS_INVALID_PARAMETER
;
1257 if (flags
& DS_FORCE_REDISCOVERY
) {
1261 status
= dsgetdcname_cached(mem_ctx
, msg_ctx
, domain_name
, domain_guid
,
1262 flags
, site_name
, &myinfo
);
1263 if (NT_STATUS_IS_OK(status
)) {
1267 if (flags
& DS_BACKGROUND_ONLY
) {
1272 status
= dsgetdcname_rediscover(mem_ctx
, msg_ctx
, domain_name
,
1273 domain_guid
, flags
, site_name
,
1277 if (!NT_STATUS_IS_OK(status
)) {
1280 return NT_STATUS_OK
;
1286 TALLOC_FREE(first_info
);
1287 } else if (!is_closest_site(myinfo
)) {
1289 first_info
= myinfo
;
1290 /* TODO: may use the next_closest_site here */
1291 site_name
= myinfo
->client_site_name
;
1296 return NT_STATUS_OK
;
1299 /********************************************************************
1302 This will be the only public function here.
1303 ********************************************************************/
1305 NTSTATUS
dsgetdcname(TALLOC_CTX
*mem_ctx
,
1306 struct messaging_context
*msg_ctx
,
1307 const char *domain_name
,
1308 const struct GUID
*domain_guid
,
1309 const char *site_name
,
1311 struct netr_DsRGetDCNameInfo
**info
)
1314 const char *query_site
= NULL
;
1315 char *ptr_to_free
= NULL
;
1316 bool retry_query_with_null
= false;
1318 if ((site_name
== NULL
) || (site_name
[0] == '\0')) {
1319 ptr_to_free
= sitename_fetch(mem_ctx
, domain_name
);
1320 if (ptr_to_free
!= NULL
) {
1321 retry_query_with_null
= true;
1323 query_site
= ptr_to_free
;
1325 query_site
= site_name
;
1328 status
= dsgetdcname_internal(mem_ctx
,
1336 TALLOC_FREE(ptr_to_free
);
1338 if (!NT_STATUS_EQUAL(status
, NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND
)) {
1342 /* Should we try again with site_name == NULL ? */
1343 if (retry_query_with_null
) {
1344 status
= dsgetdcname_internal(mem_ctx
,