dsgetdcname: map additional flags to nt_version.
[Samba.git] / source / libsmb / dsgetdcname.c
bloba278c4ef508abdf5afaefc8ce72ecc365fca0061
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 (!(flags & DS_IP_REQUIRED)) {
556 count++;
557 continue;
560 /* If we don't have an IP list for a name, lookup it up */
562 if (!dcs[i].ss_s) {
563 interpret_string_addr(&r->ss, dcs[i].hostname, 0);
564 i++;
565 j = 0;
566 } else {
567 /* use the IP addresses from the SRV sresponse */
569 if (j >= dcs[i].num_ips) {
570 i++;
571 j = 0;
572 continue;
575 r->ss = dcs[i].ss_s[j];
576 j++;
579 /* make sure it is a valid IP. I considered checking the
580 * negative connection cache, but this is the wrong place for
581 * it. Maybe only as a hac. After think about it, if all of
582 * the IP addresses retuend from DNS are dead, what hope does a
583 * netbios name lookup have? The standard reason for falling
584 * back to netbios lookups is that our DNS server doesn't know
585 * anything about the DC's -- jerry */
587 if (!is_zero_addr(&r->ss)) {
588 count++;
589 continue;
593 *returned_dclist = dclist;
594 *return_count = count;
596 if (count > 0) {
597 return NT_STATUS_OK;
600 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
603 /****************************************************************
604 ****************************************************************/
606 static NTSTATUS make_domain_controller_info(TALLOC_CTX *mem_ctx,
607 const char *dc_unc,
608 const char *dc_address,
609 uint32_t dc_address_type,
610 const struct GUID *domain_guid,
611 const char *domain_name,
612 const char *forest_name,
613 uint32_t flags,
614 const char *dc_site_name,
615 const char *client_site_name,
616 struct netr_DsRGetDCNameInfo **info_out)
618 struct netr_DsRGetDCNameInfo *info;
620 info = TALLOC_ZERO_P(mem_ctx, struct netr_DsRGetDCNameInfo);
621 NT_STATUS_HAVE_NO_MEMORY(info);
623 if (dc_unc) {
624 info->dc_unc = talloc_strdup(mem_ctx, dc_unc);
625 NT_STATUS_HAVE_NO_MEMORY(info->dc_unc);
628 if (dc_address) {
629 info->dc_address = talloc_strdup(mem_ctx, dc_address);
630 NT_STATUS_HAVE_NO_MEMORY(info->dc_address);
633 info->dc_address_type = dc_address_type;
635 if (domain_guid) {
636 info->domain_guid = *domain_guid;
639 if (domain_name) {
640 info->domain_name = talloc_strdup(mem_ctx, domain_name);
641 NT_STATUS_HAVE_NO_MEMORY(info->domain_name);
644 if (forest_name) {
645 info->forest_name = talloc_strdup(mem_ctx, forest_name);
646 NT_STATUS_HAVE_NO_MEMORY(info->forest_name);
647 flags |= DS_DNS_FOREST;
650 info->dc_flags = flags;
652 if (dc_site_name) {
653 info->dc_site_name = talloc_strdup(mem_ctx, dc_site_name);
654 NT_STATUS_HAVE_NO_MEMORY(info->dc_site_name);
657 if (client_site_name) {
658 info->client_site_name = talloc_strdup(mem_ctx,
659 client_site_name);
660 NT_STATUS_HAVE_NO_MEMORY(info->client_site_name);
663 *info_out = info;
665 return NT_STATUS_OK;
668 /****************************************************************
669 ****************************************************************/
671 static NTSTATUS make_dc_info_from_cldap_reply(TALLOC_CTX *mem_ctx,
672 uint32_t flags,
673 struct sockaddr_storage *ss,
674 uint32_t nt_version,
675 union nbt_cldap_netlogon *r,
676 struct netr_DsRGetDCNameInfo **info)
678 const char *dc_hostname, *dc_domain_name;
679 const char *dc_address = NULL;
680 const char *dc_forest = NULL;
681 uint32_t dc_address_type = 0;
682 uint32_t dc_flags = 0;
683 struct GUID *dc_domain_guid = NULL;
684 const char *dc_server_site = NULL;
685 const char *dc_client_site = NULL;
687 char addr[INET6_ADDRSTRLEN];
689 print_sockaddr(addr, sizeof(addr), ss);
691 dc_address = talloc_asprintf(mem_ctx, "\\\\%s", addr);
692 NT_STATUS_HAVE_NO_MEMORY(dc_address);
693 dc_address_type = DS_ADDRESS_TYPE_INET;
695 switch (nt_version & 0x000000ff) {
696 case 0:
697 return NT_STATUS_INVALID_PARAMETER;
698 case 1:
699 dc_hostname = r->logon1.pdc_name;
700 dc_domain_name = r->logon1.domain_name;
701 if (flags & DS_PDC_REQUIRED) {
702 dc_flags = NBT_SERVER_WRITABLE | NBT_SERVER_PDC;
704 break;
705 case 2:
706 case 3:
707 switch (flags & 0xf0000000) {
708 case DS_RETURN_FLAT_NAME:
709 dc_hostname = r->logon3.pdc_name;
710 dc_domain_name = r->logon3.domain_name;
711 break;
712 case DS_RETURN_DNS_NAME:
713 default:
714 dc_hostname = r->logon3.pdc_dns_name;
715 dc_domain_name = r->logon3.dns_domain;
716 dc_flags |= DS_DNS_DOMAIN | DS_DNS_CONTROLLER;
717 break;
720 dc_flags |= r->logon3.server_type;
721 dc_forest = r->logon3.forest;
722 dc_domain_guid = &r->logon3.domain_uuid;
724 break;
725 case 4:
726 case 5:
727 case 6:
728 case 7:
729 switch (flags & 0xf0000000) {
730 case DS_RETURN_FLAT_NAME:
731 dc_hostname = r->logon5.pdc_name;
732 dc_domain_name = r->logon5.domain;
733 break;
734 case DS_RETURN_DNS_NAME:
735 default:
736 dc_hostname = r->logon5.pdc_dns_name;
737 dc_domain_name = r->logon5.dns_domain;
738 dc_flags |= DS_DNS_DOMAIN | DS_DNS_CONTROLLER;
739 break;
742 dc_flags |= r->logon5.server_type;
743 dc_forest = r->logon5.forest;
744 dc_domain_guid = &r->logon5.domain_uuid;
745 dc_server_site = r->logon5.server_site;
746 dc_client_site = r->logon5.client_site;
748 break;
749 case 8:
750 case 9:
751 case 10:
752 case 11:
753 case 12:
754 case 13:
755 case 14:
756 case 15:
757 switch (flags & 0xf0000000) {
758 case DS_RETURN_FLAT_NAME:
759 dc_hostname = r->logon13.pdc_name;
760 dc_domain_name = r->logon13.domain;
761 break;
762 case DS_RETURN_DNS_NAME:
763 default:
764 dc_hostname = r->logon13.pdc_dns_name;
765 dc_domain_name = r->logon13.dns_domain;
766 dc_flags |= DS_DNS_DOMAIN | DS_DNS_CONTROLLER;
767 break;
770 dc_flags |= r->logon13.server_type;
771 dc_forest = r->logon13.forest;
772 dc_domain_guid = &r->logon13.domain_uuid;
773 dc_server_site = r->logon13.server_site;
774 dc_client_site = r->logon13.client_site;
776 break;
777 default:
778 switch (flags & 0xf0000000) {
779 case DS_RETURN_FLAT_NAME:
780 dc_hostname = r->logon29.pdc_name;
781 dc_domain_name = r->logon29.domain;
782 break;
783 case DS_RETURN_DNS_NAME:
784 default:
785 dc_hostname = r->logon29.pdc_dns_name;
786 dc_domain_name = r->logon29.dns_domain;
787 dc_flags |= DS_DNS_DOMAIN | DS_DNS_CONTROLLER;
788 break;
791 dc_flags |= r->logon29.server_type;
792 dc_forest = r->logon29.forest;
793 dc_domain_guid = &r->logon29.domain_uuid;
794 dc_server_site = r->logon29.server_site;
795 dc_client_site = r->logon29.client_site;
797 break;
800 return make_domain_controller_info(mem_ctx,
801 dc_hostname,
802 dc_address,
803 dc_address_type,
804 dc_domain_guid,
805 dc_domain_name,
806 dc_forest,
807 dc_flags,
808 dc_server_site,
809 dc_client_site,
810 info);
813 /****************************************************************
814 ****************************************************************/
816 static uint32_t map_ds_flags_to_nt_version(uint32_t flags)
818 uint32_t nt_version = 0;
820 if (flags & DS_PDC_REQUIRED) {
821 nt_version |= NETLOGON_VERSION_PDC;
824 if (flags & DS_GC_SERVER_REQUIRED) {
825 nt_version |= NETLOGON_VERSION_GC;
828 if (flags & DS_TRY_NEXTCLOSEST_SITE) {
829 nt_version |= NETLOGON_VERSION_WITH_CLOSEST_SITE;
832 if (flags & DS_IP_REQUIRED) {
833 nt_version |= NETLOGON_VERSION_IP;
836 return nt_version;
839 /****************************************************************
840 ****************************************************************/
842 static NTSTATUS process_dc_dns(TALLOC_CTX *mem_ctx,
843 const char *domain_name,
844 uint32_t flags,
845 struct ip_service_name *dclist,
846 int num_dcs,
847 struct netr_DsRGetDCNameInfo **info)
849 int i = 0;
850 bool valid_dc = false;
851 union nbt_cldap_netlogon *r = NULL;
852 uint32_t nt_version = NETLOGON_VERSION_5 |
853 NETLOGON_VERSION_5EX;
854 uint32_t ret_flags = 0;
856 nt_version |= map_ds_flags_to_nt_version(flags);
858 for (i=0; i<num_dcs; i++) {
860 DEBUG(10,("LDAP ping to %s\n", dclist[i].hostname));
862 if (ads_cldap_netlogon(mem_ctx, dclist[i].hostname,
863 domain_name,
864 &nt_version,
865 &r))
867 ret_flags = get_cldap_reply_server_flags(r, nt_version);
869 if (check_cldap_reply_required_flags(ret_flags, flags)) {
870 valid_dc = true;
871 break;
875 continue;
878 if (!valid_dc) {
879 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
882 return make_dc_info_from_cldap_reply(mem_ctx, flags, &dclist[i].ss,
883 nt_version, r, info);
886 /****************************************************************
887 ****************************************************************/
889 static struct event_context *ev_context(void)
891 static struct event_context *ctx;
893 if (!ctx && !(ctx = event_context_init(NULL))) {
894 smb_panic("Could not init event context");
896 return ctx;
899 /****************************************************************
900 ****************************************************************/
902 static struct messaging_context *msg_context(TALLOC_CTX *mem_ctx)
904 static struct messaging_context *ctx;
906 if (!ctx && !(ctx = messaging_init(mem_ctx, server_id_self(),
907 ev_context()))) {
908 smb_panic("Could not init messaging context");
910 return ctx;
913 /****************************************************************
914 ****************************************************************/
916 static NTSTATUS process_dc_netbios(TALLOC_CTX *mem_ctx,
917 const char *domain_name,
918 uint32_t flags,
919 struct ip_service_name *dclist,
920 int num_dcs,
921 struct netr_DsRGetDCNameInfo **info)
923 struct sockaddr_storage ss;
924 struct ip_service ip_list;
925 enum nbt_name_type name_type = NBT_NAME_LOGON;
927 int i;
928 const char *dc_hostname, *dc_domain_name;
929 const char *dc_address;
930 uint32_t dc_address_type;
931 uint32_t dc_flags = 0;
932 const char *dc_name = NULL;
933 const char *dc_forest = NULL;
934 const char *dc_server_site = NULL;
935 const char *dc_client_site = NULL;
936 struct GUID *dc_domain_guid = NULL;
937 fstring tmp_dc_name;
938 struct messaging_context *msg_ctx = msg_context(mem_ctx);
939 struct nbt_ntlogon_packet *reply = NULL;
940 uint32_t nt_version = NETLOGON_VERSION_1 |
941 NETLOGON_VERSION_5 |
942 NETLOGON_VERSION_5EX_WITH_IP;
944 if (flags & DS_PDC_REQUIRED) {
945 name_type = NBT_NAME_PDC;
948 nt_version |= map_ds_flags_to_nt_version(flags);
950 DEBUG(10,("process_dc_netbios\n"));
952 for (i=0; i<num_dcs; i++) {
954 ip_list.ss = dclist[i].ss;
955 ip_list.port = 0;
957 if (!interpret_string_addr(&ss, dclist[i].hostname, AI_NUMERICHOST)) {
958 return NT_STATUS_UNSUCCESSFUL;
961 if (send_getdc_request(mem_ctx, msg_ctx,
962 &dclist[i].ss, domain_name,
963 NULL, nt_version))
965 int k;
966 smb_msleep(100);
967 for (k=0; k<5; k++) {
968 if (receive_getdc_response(mem_ctx,
969 &dclist[i].ss,
970 domain_name,
971 &dc_name,
972 &reply)) {
973 namecache_store(dc_name, NBT_NAME_SERVER, 1, &ip_list);
974 dc_hostname = dc_name;
975 dc_domain_name = talloc_strdup_upper(mem_ctx, domain_name);
976 NT_STATUS_HAVE_NO_MEMORY(dc_domain_name);
977 goto make_reply;
979 smb_msleep(500);
983 if (name_status_find(domain_name,
984 name_type,
985 NBT_NAME_SERVER,
986 &dclist[i].ss,
987 tmp_dc_name))
989 dc_hostname = tmp_dc_name;
990 dc_domain_name = talloc_strdup_upper(mem_ctx, domain_name);
991 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 if (reply && reply->command == NTLOGON_RESPONSE_FROM_PDC2) {
1002 dc_flags |= reply->req.reply2.server_type;
1003 dc_forest = reply->req.reply2.forest;
1004 dc_server_site = reply->req.reply2.server_site;
1005 dc_client_site = reply->req.reply2.client_site;
1007 dc_domain_guid = &reply->req.reply2.domain_uuid;
1009 if (flags & DS_RETURN_DNS_NAME) {
1010 dc_domain_name = reply->req.reply2.dns_domain;
1011 dc_hostname = reply->req.reply2.pdc_dns_name;
1012 dc_flags |= DS_DNS_DOMAIN | DS_DNS_CONTROLLER;
1013 } else if (flags & DS_RETURN_FLAT_NAME) {
1014 dc_domain_name = reply->req.reply2.domain;
1015 dc_hostname = reply->req.reply2.pdc_name;
1019 if (flags & DS_IP_REQUIRED) {
1020 char addr[INET6_ADDRSTRLEN];
1021 print_sockaddr(addr, sizeof(addr), &dclist[i].ss);
1022 dc_address = talloc_asprintf(mem_ctx, "\\\\%s", addr);
1023 dc_address_type = DS_ADDRESS_TYPE_INET;
1024 } else {
1025 dc_address = talloc_asprintf(mem_ctx, "\\\\%s", dc_hostname);
1026 dc_address_type = DS_ADDRESS_TYPE_NETBIOS;
1029 if (flags & DS_PDC_REQUIRED) {
1030 dc_flags |= NBT_SERVER_PDC | NBT_SERVER_WRITABLE;
1033 if (dc_forest) {
1034 dc_flags |= DS_DNS_FOREST;
1037 return make_domain_controller_info(mem_ctx,
1038 dc_hostname,
1039 dc_address,
1040 dc_address_type,
1041 dc_domain_guid,
1042 dc_domain_name,
1043 dc_forest,
1044 dc_flags,
1045 dc_server_site,
1046 dc_client_site,
1047 info);
1050 /****************************************************************
1051 ****************************************************************/
1053 static NTSTATUS dsgetdcname_rediscover(TALLOC_CTX *mem_ctx,
1054 const char *domain_name,
1055 struct GUID *domain_guid,
1056 uint32_t flags,
1057 const char *site_name,
1058 struct netr_DsRGetDCNameInfo **info)
1060 NTSTATUS status = NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
1061 struct ip_service_name *dclist = NULL;
1062 int num_dcs;
1064 DEBUG(10,("dsgetdcname_rediscover\n"));
1066 if (flags & DS_IS_FLAT_NAME) {
1068 status = discover_dc_netbios(mem_ctx, domain_name, flags,
1069 &dclist, &num_dcs);
1070 NT_STATUS_NOT_OK_RETURN(status);
1072 return process_dc_netbios(mem_ctx, domain_name, flags,
1073 dclist, num_dcs, info);
1076 if (flags & DS_IS_DNS_NAME) {
1078 status = discover_dc_dns(mem_ctx, domain_name, domain_guid,
1079 flags, site_name, &dclist, &num_dcs);
1080 NT_STATUS_NOT_OK_RETURN(status);
1082 return process_dc_dns(mem_ctx, domain_name, flags,
1083 dclist, num_dcs, info);
1086 status = discover_dc_dns(mem_ctx, domain_name, domain_guid, flags,
1087 site_name, &dclist, &num_dcs);
1089 if (NT_STATUS_IS_OK(status) && num_dcs != 0) {
1091 status = process_dc_dns(mem_ctx, domain_name, flags, dclist,
1092 num_dcs, info);
1093 if (NT_STATUS_IS_OK(status)) {
1094 return status;
1098 status = discover_dc_netbios(mem_ctx, domain_name, flags, &dclist,
1099 &num_dcs);
1100 NT_STATUS_NOT_OK_RETURN(status);
1102 return process_dc_netbios(mem_ctx, domain_name, flags, dclist,
1103 num_dcs, info);
1106 /********************************************************************
1107 dsgetdcname.
1109 This will be the only public function here.
1110 ********************************************************************/
1112 NTSTATUS dsgetdcname(TALLOC_CTX *mem_ctx,
1113 const char *domain_name,
1114 struct GUID *domain_guid,
1115 const char *site_name,
1116 uint32_t flags,
1117 struct netr_DsRGetDCNameInfo **info)
1119 NTSTATUS status = NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
1120 struct netr_DsRGetDCNameInfo *myinfo = NULL;
1122 DEBUG(10,("dsgetdcname: domain_name: %s, "
1123 "domain_guid: %s, site_name: %s, flags: 0x%08x\n",
1124 domain_name,
1125 domain_guid ? GUID_string(mem_ctx, domain_guid) : "(null)",
1126 site_name, flags));
1128 *info = NULL;
1130 if (!check_allowed_required_flags(flags)) {
1131 DEBUG(0,("invalid flags specified\n"));
1132 return NT_STATUS_INVALID_PARAMETER;
1135 if (flags & DS_FORCE_REDISCOVERY) {
1136 goto rediscover;
1139 status = dsgetdcname_cached(mem_ctx, domain_name, domain_guid,
1140 flags, site_name, &myinfo);
1141 if (NT_STATUS_IS_OK(status)) {
1142 *info = myinfo;
1143 return status;
1146 if (flags & DS_BACKGROUND_ONLY) {
1147 return status;
1150 rediscover:
1151 status = dsgetdcname_rediscover(mem_ctx, domain_name,
1152 domain_guid, flags, site_name,
1153 &myinfo);
1155 if (NT_STATUS_IS_OK(status)) {
1156 dsgetdcname_cache_store(mem_ctx, domain_name, myinfo);
1157 *info = myinfo;
1160 return status;
1163 /****************************************************************
1164 ****************************************************************/
1166 bool pull_mailslot_cldap_reply(TALLOC_CTX *mem_ctx,
1167 const DATA_BLOB *blob,
1168 union nbt_cldap_netlogon *r,
1169 uint32_t *nt_version)
1171 enum ndr_err_code ndr_err;
1172 uint32_t nt_version_query = ((*nt_version) & 0x000000ff);
1173 uint16_t command = 0;
1175 ndr_err = ndr_pull_struct_blob(blob, mem_ctx, &command,
1176 (ndr_pull_flags_fn_t)ndr_pull_uint16);
1177 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1178 return false;
1181 switch (command) {
1182 case 0x13: /* 19 */
1183 case 0x15: /* 21 */
1184 case 0x17: /* 23 */
1185 break;
1186 default:
1187 DEBUG(1,("got unexpected command: %d (0x%08x)\n",
1188 command, command));
1189 return false;
1192 ndr_err = ndr_pull_union_blob_all(blob, mem_ctx, r, nt_version_query,
1193 (ndr_pull_flags_fn_t)ndr_pull_nbt_cldap_netlogon);
1194 if (NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1195 goto done;
1198 /* when the caller requested just those nt_version bits that the server
1199 * was able to reply to, we are fine and all done. otherwise we need to
1200 * assume downgraded replies which are painfully parsed here - gd */
1202 if (nt_version_query & NETLOGON_VERSION_WITH_CLOSEST_SITE) {
1203 nt_version_query &= ~NETLOGON_VERSION_WITH_CLOSEST_SITE;
1205 ndr_err = ndr_pull_union_blob_all(blob, mem_ctx, r, nt_version_query,
1206 (ndr_pull_flags_fn_t)ndr_pull_nbt_cldap_netlogon);
1207 if (NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1208 goto done;
1210 if (nt_version_query & NETLOGON_VERSION_5EX_WITH_IP) {
1211 nt_version_query &= ~NETLOGON_VERSION_5EX_WITH_IP;
1213 ndr_err = ndr_pull_union_blob_all(blob, mem_ctx, r, nt_version_query,
1214 (ndr_pull_flags_fn_t)ndr_pull_nbt_cldap_netlogon);
1215 if (NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1216 goto done;
1218 if (nt_version_query & NETLOGON_VERSION_5EX) {
1219 nt_version_query &= ~NETLOGON_VERSION_5EX;
1221 ndr_err = ndr_pull_union_blob_all(blob, mem_ctx, r, nt_version_query,
1222 (ndr_pull_flags_fn_t)ndr_pull_nbt_cldap_netlogon);
1223 if (NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1224 goto done;
1226 if (nt_version_query & NETLOGON_VERSION_5) {
1227 nt_version_query &= ~NETLOGON_VERSION_5;
1229 ndr_err = ndr_pull_union_blob_all(blob, mem_ctx, r, nt_version_query,
1230 (ndr_pull_flags_fn_t)ndr_pull_nbt_cldap_netlogon);
1231 if (NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1232 goto done;
1235 return false;
1237 done:
1238 if (DEBUGLEVEL >= 10) {
1239 NDR_PRINT_UNION_DEBUG(nbt_cldap_netlogon, nt_version_query, r);
1242 *nt_version = nt_version_query;
1244 return true;