?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
dsgetdcname: add map_ds_flags_to_nt_version.
[Samba.git] / source / libsmb / dsgetdcname.c
blobc7eb8005bc3a81c5485cdfde80b8d79db51b343b
1 /*
2 Unix SMB/CIFS implementation.
3
4 dsgetdcname
5
6 Copyright (C) Gerald Carter 2006
7 Copyright (C) Guenther Deschner 2007-2008
8
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.
13
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.
18
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/>.
21 */
22
23 #include "includes.h"
24
25 #define DSGETDCNAME_FMT "DSGETDCNAME/DOMAIN/%s"
26 /* 15 minutes */
27 #define DSGETDCNAME_CACHE_TTL 60*15
28
29 struct ip_service_name {
30 struct sockaddr_storage ss;
31 unsigned port;
32 const char *hostname;
33 };
34
35 /****************************************************************
36 ****************************************************************/
37
38 void debug_dsdcinfo_flags(int lvl, uint32_t flags)
39 {
40 DEBUG(lvl,("debug_dsdcinfo_flags: 0x%08x\n\t", flags));
41
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"));
108 }
109
110 /****************************************************************
111 ****************************************************************/
112
113 static char *dsgetdcname_cache_key(TALLOC_CTX *mem_ctx, const char *domain)
114 {
115 if (!mem_ctx || !domain) {
116 return NULL;
117 }
118
119 return talloc_asprintf_strupper_m(mem_ctx, DSGETDCNAME_FMT, domain);
120 }
121
122 /****************************************************************
123 ****************************************************************/
124
125 static NTSTATUS dsgetdcname_cache_delete(TALLOC_CTX *mem_ctx,
126 const char *domain_name)
127 {
128 char *key;
129
130 if (!gencache_init()) {
131 return NT_STATUS_INTERNAL_DB_ERROR;
132 }
133
134 key = dsgetdcname_cache_key(mem_ctx, domain_name);
135 if (!key) {
136 return NT_STATUS_NO_MEMORY;
137 }
138
139 if (!gencache_del(key)) {
140 return NT_STATUS_UNSUCCESSFUL;
141 }
142
143 return NT_STATUS_OK;
144 }
145
146 /****************************************************************
147 ****************************************************************/
148
149 static NTSTATUS dsgetdcname_cache_store(TALLOC_CTX *mem_ctx,
150 const char *domain_name,
151 struct netr_DsRGetDCNameInfo *info)
152 {
153 time_t expire_time;
154 char *key;
155 bool ret = false;
156 DATA_BLOB blob;
157 enum ndr_err_code ndr_err;
158
159 if (!gencache_init()) {
160 return NT_STATUS_INTERNAL_DB_ERROR;
161 }
162
163 key = dsgetdcname_cache_key(mem_ctx, domain_name);
164 if (!key) {
165 return NT_STATUS_NO_MEMORY;
166 }
167
168 expire_time = time(NULL) + DSGETDCNAME_CACHE_TTL;
169
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);
174 }
175
176 if (gencache_lock_entry(key) != 0) {
177 data_blob_free(&blob);
178 return NT_STATUS_LOCK_NOT_GRANTED;
179 }
180
181 ret = gencache_set_data_blob(key, &blob, expire_time);
182 data_blob_free(&blob);
183
184 gencache_unlock_entry(key);
185
186 return ret ? NT_STATUS_OK : NT_STATUS_UNSUCCESSFUL;
187 }
188
189 /****************************************************************
190 ****************************************************************/
191
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)
198 {
199 struct nbt_cldap_netlogon_5 r;
200
201 /* check if matching entry is older then 15 minutes, if yes, send
202 * CLDAP/MAILSLOT ping again and store the cached data */
203
204 ZERO_STRUCT(r);
205
206 if (ads_cldap_netlogon(mem_ctx, info->dc_unc,
207 info->domain_name, &r)) {
208
209 dsgetdcname_cache_delete(mem_ctx, domain_name);
210
211 return dsgetdcname_cache_store(mem_ctx,
212 info->domain_name,
213 info);
214 }
215
216 return NT_STATUS_INVALID_NETWORK_RESPONSE;
217 }
218
219 /****************************************************************
220 ****************************************************************/
221
222 #define RETURN_ON_FALSE(x) if (!x) return false;
223
224 static bool check_cldap_reply_required_flags(uint32_t ret_flags,
225 uint32_t req_flags)
226 {
227 if (req_flags & DS_PDC_REQUIRED)
228 RETURN_ON_FALSE(ret_flags & NBT_SERVER_PDC);
229
230 if (req_flags & DS_GC_SERVER_REQUIRED)
231 RETURN_ON_FALSE(ret_flags & NBT_SERVER_GC);
232
233 if (req_flags & DS_ONLY_LDAP_NEEDED)
234 RETURN_ON_FALSE(ret_flags & NBT_SERVER_LDAP);
235
236 if ((req_flags & DS_DIRECTORY_SERVICE_REQUIRED) ||
237 (req_flags & DS_DIRECTORY_SERVICE_PREFERRED))
238 RETURN_ON_FALSE(ret_flags & NBT_SERVER_DS);
239
240 if (req_flags & DS_KDC_REQUIRED)
241 RETURN_ON_FALSE(ret_flags & NBT_SERVER_KDC);
242
243 if (req_flags & DS_TIMESERV_REQUIRED)
244 RETURN_ON_FALSE(ret_flags & NBT_SERVER_TIMESERV);
245
246 if (req_flags & DS_WRITABLE_REQUIRED)
247 RETURN_ON_FALSE(ret_flags & NBT_SERVER_WRITABLE);
248
249 return true;
250 }
251
252 /****************************************************************
253 ****************************************************************/
254
255 static NTSTATUS dsgetdcname_cache_fetch(TALLOC_CTX *mem_ctx,
256 const char *domain_name,
257 struct GUID *domain_guid,
258 uint32_t flags,
259 const char *site_name,
260 struct netr_DsRGetDCNameInfo **info_p,
261 bool *expired)
262 {
263 char *key;
264 DATA_BLOB blob;
265 enum ndr_err_code ndr_err;
266 struct netr_DsRGetDCNameInfo *info;
267
268 if (!gencache_init()) {
269 return NT_STATUS_INTERNAL_DB_ERROR;
270 }
271
272 key = dsgetdcname_cache_key(mem_ctx, domain_name);
273 if (!key) {
274 return NT_STATUS_NO_MEMORY;
275 }
276
277 if (!gencache_get_data_blob(key, &blob, expired)) {
278 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
279 }
280
281 info = TALLOC_ZERO_P(mem_ctx, struct netr_DsRGetDCNameInfo);
282 if (!info) {
283 return NT_STATUS_NO_MEMORY;
284 }
285
286 ndr_err = ndr_pull_struct_blob(&blob, mem_ctx, info,
287 (ndr_pull_flags_fn_t)ndr_pull_netr_DsRGetDCNameInfo);
288
289 data_blob_free(&blob);
290 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
291 dsgetdcname_cache_delete(mem_ctx, domain_name);
292 return ndr_map_error2ntstatus(ndr_err);
293 }
294
295 if (DEBUGLEVEL >= 10) {
296 NDR_PRINT_DEBUG(netr_DsRGetDCNameInfo, info);
297 }
298
299 /* check flags */
300 if (!check_cldap_reply_required_flags(info->dc_flags, flags)) {
301 DEBUG(10,("invalid flags\n"));
302 return NT_STATUS_INVALID_PARAMETER;
303 }
304
305 if ((flags & DS_IP_REQUIRED) &&
306 (info->dc_address_type != DS_ADDRESS_TYPE_INET)) {
307 return NT_STATUS_INVALID_PARAMETER_MIX;
308 }
309
310 *info_p = info;
311
312 return NT_STATUS_OK;
313 }
314
315 /****************************************************************
316 ****************************************************************/
317
318 static NTSTATUS dsgetdcname_cached(TALLOC_CTX *mem_ctx,
319 const char *domain_name,
320 struct GUID *domain_guid,
321 uint32_t flags,
322 const char *site_name,
323 struct netr_DsRGetDCNameInfo **info)
324 {
325 NTSTATUS status;
326 bool expired = false;
327
328 status = dsgetdcname_cache_fetch(mem_ctx, domain_name, domain_guid,
329 flags, site_name, info, &expired);
330 if (!NT_STATUS_IS_OK(status)) {
331 DEBUG(10,("dsgetdcname_cached: cache fetch failed with: %s\n",
332 nt_errstr(status)));
333 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
334 }
335
336 if (flags & DS_BACKGROUND_ONLY) {
337 return status;
338 }
339
340 if (expired) {
341 status = dsgetdcname_cache_refresh(mem_ctx, domain_name,
342 domain_guid, flags,
343 site_name, *info);
344 if (!NT_STATUS_IS_OK(status)) {
345 return status;
346 }
347 }
348
349 return status;
350 }
351
352 /****************************************************************
353 ****************************************************************/
354
355 static bool check_allowed_required_flags(uint32_t flags)
356 {
357 uint32_t return_type = flags & (DS_RETURN_FLAT_NAME|DS_RETURN_DNS_NAME);
358 uint32_t offered_type = flags & (DS_IS_FLAT_NAME|DS_IS_DNS_NAME);
359 uint32_t query_type = flags & (DS_BACKGROUND_ONLY|DS_FORCE_REDISCOVERY);
360
361 /* FIXME: check for DSGETDC_VALID_FLAGS and check for excluse bits
362 * (DS_PDC_REQUIRED, DS_KDC_REQUIRED, DS_GC_SERVER_REQUIRED) */
363
364 debug_dsdcinfo_flags(10, flags);
365
366 if (return_type == (DS_RETURN_FLAT_NAME|DS_RETURN_DNS_NAME)) {
367 return false;
368 }
369
370 if (offered_type == (DS_IS_DNS_NAME|DS_IS_FLAT_NAME)) {
371 return false;
372 }
373
374 if (query_type == (DS_BACKGROUND_ONLY|DS_FORCE_REDISCOVERY)) {
375 return false;
376 }
377
378 #if 0
379 if ((flags & DS_RETURN_DNS_NAME) && (!(flags & DS_IP_REQUIRED))) {
380 printf("gd: here5 \n");
381 return false;
382 }
383 #endif
384 return true;
385 }
386
387 /****************************************************************
388 ****************************************************************/
389
390 static NTSTATUS discover_dc_netbios(TALLOC_CTX *mem_ctx,
391 const char *domain_name,
392 uint32_t flags,
393 struct ip_service_name **returned_dclist,
394 int *returned_count)
395 {
396 NTSTATUS status;
397 enum nbt_name_type name_type = NBT_NAME_LOGON;
398 struct ip_service *iplist;
399 int i;
400 struct ip_service_name *dclist = NULL;
401 int count;
402
403 *returned_dclist = NULL;
404 *returned_count = 0;
405
406 if (lp_disable_netbios()) {
407 return NT_STATUS_NOT_SUPPORTED;
408 }
409
410 if (flags & DS_PDC_REQUIRED) {
411 name_type = NBT_NAME_PDC;
412 }
413
414 status = internal_resolve_name(domain_name, name_type, NULL,
415 &iplist, &count,
416 "lmhosts wins bcast");
417 if (!NT_STATUS_IS_OK(status)) {
418 DEBUG(10,("discover_dc_netbios: failed to find DC\n"));
419 return status;
420 }
421
422 dclist = TALLOC_ZERO_ARRAY(mem_ctx, struct ip_service_name, count);
423 if (!dclist) {
424 return NT_STATUS_NO_MEMORY;
425 }
426
427 for (i=0; i<count; i++) {
428
429 char addr[INET6_ADDRSTRLEN];
430 struct ip_service_name *r = &dclist[i];
431
432 print_sockaddr(addr, sizeof(addr),
433 &iplist[i].ss);
434
435 r->ss = iplist[i].ss;
436 r->port = iplist[i].port;
437 r->hostname = talloc_strdup(mem_ctx, addr);
438 if (!r->hostname) {
439 return NT_STATUS_NO_MEMORY;
440 }
441
442 }
443
444 *returned_dclist = dclist;
445 *returned_count = count;
446
447 return NT_STATUS_OK;
448 }
449
450 /****************************************************************
451 ****************************************************************/
452
453 static NTSTATUS discover_dc_dns(TALLOC_CTX *mem_ctx,
454 const char *domain_name,
455 struct GUID *domain_guid,
456 uint32_t flags,
457 const char *site_name,
458 struct ip_service_name **returned_dclist,
459 int *return_count)
460 {
461 int i, j;
462 NTSTATUS status;
463 struct dns_rr_srv *dcs = NULL;
464 int numdcs = 0;
465 int numaddrs = 0;
466 struct ip_service_name *dclist = NULL;
467 int count = 0;
468
469 if (flags & DS_PDC_REQUIRED) {
470 status = ads_dns_query_pdc(mem_ctx, domain_name,
471 &dcs, &numdcs);
472 } else if (flags & DS_GC_SERVER_REQUIRED) {
473 status = ads_dns_query_gcs(mem_ctx, domain_name, site_name,
474 &dcs, &numdcs);
475 } else if (flags & DS_KDC_REQUIRED) {
476 status = ads_dns_query_kdcs(mem_ctx, domain_name, site_name,
477 &dcs, &numdcs);
478 } else if (flags & DS_DIRECTORY_SERVICE_REQUIRED) {
479 status = ads_dns_query_dcs(mem_ctx, domain_name, site_name,
480 &dcs, &numdcs);
481 } else if (domain_guid) {
482 status = ads_dns_query_dcs_guid(mem_ctx, domain_name,
483 domain_guid, &dcs, &numdcs);
484 } else {
485 status = ads_dns_query_dcs(mem_ctx, domain_name, site_name,
486 &dcs, &numdcs);
487 }
488
489 if (!NT_STATUS_IS_OK(status)) {
490 return status;
491 }
492
493 if (numdcs == 0) {
494 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
495 }
496
497 for (i=0;i<numdcs;i++) {
498 numaddrs += MAX(dcs[i].num_ips,1);
499 }
500
501 dclist = TALLOC_ZERO_ARRAY(mem_ctx,
502 struct ip_service_name,
503 numaddrs);
504 if (!dclist) {
505 return NT_STATUS_NO_MEMORY;
506 }
507
508 /* now unroll the list of IP addresses */
509
510 *return_count = 0;
511 i = 0;
512 j = 0;
513
514 while ((i < numdcs) && (count < numaddrs)) {
515
516 struct ip_service_name *r = &dclist[count];
517
518 r->port = dcs[count].port;
519 r->hostname = dcs[count].hostname;
520
521 if (!(flags & DS_IP_REQUIRED)) {
522 count++;
523 continue;
524 }
525
526 /* If we don't have an IP list for a name, lookup it up */
527
528 if (!dcs[i].ss_s) {
529 interpret_string_addr(&r->ss, dcs[i].hostname, 0);
530 i++;
531 j = 0;
532 } else {
533 /* use the IP addresses from the SRV sresponse */
534
535 if (j >= dcs[i].num_ips) {
536 i++;
537 j = 0;
538 continue;
539 }
540
541 r->ss = dcs[i].ss_s[j];
542 j++;
543 }
544
545 /* make sure it is a valid IP. I considered checking the
546 * negative connection cache, but this is the wrong place for
547 * it. Maybe only as a hac. After think about it, if all of
548 * the IP addresses retuend from DNS are dead, what hope does a
549 * netbios name lookup have? The standard reason for falling
550 * back to netbios lookups is that our DNS server doesn't know
551 * anything about the DC's -- jerry */
552
553 if (!is_zero_addr(&r->ss)) {
554 count++;
555 continue;
556 }
557 }
558
559 *returned_dclist = dclist;
560 *return_count = count;
561
562 if (count > 0) {
563 return NT_STATUS_OK;
564 }
565
566 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
567 }
568
569 /****************************************************************
570 ****************************************************************/
571
572 static NTSTATUS make_domain_controller_info(TALLOC_CTX *mem_ctx,
573 const char *dc_unc,
574 const char *dc_address,
575 uint32_t dc_address_type,
576 const struct GUID *domain_guid,
577 const char *domain_name,
578 const char *forest_name,
579 uint32_t flags,
580 const char *dc_site_name,
581 const char *client_site_name,
582 struct netr_DsRGetDCNameInfo **info_out)
583 {
584 struct netr_DsRGetDCNameInfo *info;
585
586 info = TALLOC_ZERO_P(mem_ctx, struct netr_DsRGetDCNameInfo);
587 NT_STATUS_HAVE_NO_MEMORY(info);
588
589 if (dc_unc) {
590 info->dc_unc = talloc_strdup(mem_ctx, dc_unc);
591 NT_STATUS_HAVE_NO_MEMORY(info->dc_unc);
592 }
593
594 if (dc_address) {
595 info->dc_address = talloc_strdup(mem_ctx, dc_address);
596 NT_STATUS_HAVE_NO_MEMORY(info->dc_address);
597 }
598
599 info->dc_address_type = dc_address_type;
600
601 if (domain_guid) {
602 info->domain_guid = *domain_guid;
603 }
604
605 if (domain_name) {
606 info->domain_name = talloc_strdup(mem_ctx, domain_name);
607 NT_STATUS_HAVE_NO_MEMORY(info->domain_name);
608 }
609
610 if (forest_name) {
611 info->forest_name = talloc_strdup(mem_ctx, forest_name);
612 NT_STATUS_HAVE_NO_MEMORY(info->forest_name);
613 }
614
615 info->dc_flags = flags;
616
617 if (dc_site_name) {
618 info->dc_site_name = talloc_strdup(mem_ctx, dc_site_name);
619 NT_STATUS_HAVE_NO_MEMORY(info->dc_site_name);
620 }
621
622 if (client_site_name) {
623 info->client_site_name = talloc_strdup(mem_ctx,
624 client_site_name);
625 NT_STATUS_HAVE_NO_MEMORY(info->client_site_name);
626 }
627
628 *info_out = info;
629
630 return NT_STATUS_OK;
631 }
632
633 /****************************************************************
634 ****************************************************************/
635
636 static uint32_t map_ds_flags_to_nt_version(uint32_t flags)
637 {
638 uint32_t nt_version = 0;
639
640 if (flags & DS_PDC_REQUIRED) {
641 nt_version |= NETLOGON_VERSION_PDC;
642 }
643
644 if (flags & DS_GC_SERVER_REQUIRED) {
645 nt_version |= NETLOGON_VERSION_GC;
646 }
647
648 if (flags & DS_TRY_NEXTCLOSEST_SITE) {
649 nt_version |= NETLOGON_VERSION_WITH_CLOSEST_SITE;
650 }
651
652 if (flags & DS_IP_REQUIRED) {
653 nt_version |= NETLOGON_VERSION_IP;
654 }
655
656 return nt_version;
657 }
658
659 /****************************************************************
660 ****************************************************************/
661
662 static NTSTATUS process_dc_dns(TALLOC_CTX *mem_ctx,
663 const char *domain_name,
664 uint32_t flags,
665 struct ip_service_name *dclist,
666 int num_dcs,
667 struct netr_DsRGetDCNameInfo **info)
668 {
669 int i = 0;
670 bool valid_dc = false;
671 struct nbt_cldap_netlogon_5 r;
672 const char *dc_hostname, *dc_domain_name;
673 const char *dc_address;
674 uint32_t dc_address_type;
675 uint32_t dc_flags;
676
677 for (i=0; i<num_dcs; i++) {
678
679 ZERO_STRUCT(r);
680
681 DEBUG(10,("LDAP ping to %s\n", dclist[i].hostname));
682
683 if ((ads_cldap_netlogon(mem_ctx, dclist[i].hostname,
684 domain_name, &r)) &&
685 (check_cldap_reply_required_flags(r.server_type, flags))) {
686 valid_dc = true;
687 break;
688 }
689
690 continue;
691 }
692
693 if (!valid_dc) {
694 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
695 }
696
697 dc_flags = r.server_type;
698
699 if (flags & DS_RETURN_FLAT_NAME) {
700 if (!strlen(r.pdc_name) || !strlen(r.domain)) {
701 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
702 }
703 dc_hostname = r.pdc_name;
704 dc_domain_name = r.domain;
705 } else if (flags & DS_RETURN_DNS_NAME) {
706 if (!strlen(r.pdc_dns_name) || !strlen(r.dns_domain)) {
707 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
708 }
709 dc_hostname = r.pdc_dns_name;
710 dc_domain_name = r.dns_domain;
711 dc_flags |= DS_DNS_DOMAIN | DS_DNS_CONTROLLER;
712 } else {
713 /* FIXME */
714 dc_hostname = r.pdc_dns_name;
715 dc_domain_name = r.dns_domain;
716 dc_flags |= DS_DNS_DOMAIN | DS_DNS_CONTROLLER;
717 }
718
719 if (flags & DS_IP_REQUIRED) {
720 char addr[INET6_ADDRSTRLEN];
721 print_sockaddr(addr, sizeof(addr), &dclist[i].ss);
722 dc_address = talloc_asprintf(mem_ctx, "\\\\%s",
723 addr);
724 dc_address_type = DS_ADDRESS_TYPE_INET;
725 } else {
726 dc_address = talloc_asprintf(mem_ctx, "\\\\%s",
727 r.pdc_name);
728 dc_address_type = DS_ADDRESS_TYPE_NETBIOS;
729 }
730 NT_STATUS_HAVE_NO_MEMORY(dc_address);
731
732 if (r.forest) {
733 dc_flags |= DS_DNS_FOREST;
734 }
735
736 return make_domain_controller_info(mem_ctx,
737 dc_hostname,
738 dc_address,
739 dc_address_type,
740 &r.domain_uuid,
741 dc_domain_name,
742 r.forest,
743 dc_flags,
744 r.server_site,
745 r.client_site,
746 info);
747
748 }
749
750 /****************************************************************
751 ****************************************************************/
752
753 static struct event_context *ev_context(void)
754 {
755 static struct event_context *ctx;
756
757 if (!ctx && !(ctx = event_context_init(NULL))) {
758 smb_panic("Could not init event context");
759 }
760 return ctx;
761 }
762
763 /****************************************************************
764 ****************************************************************/
765
766 static struct messaging_context *msg_context(TALLOC_CTX *mem_ctx)
767 {
768 static struct messaging_context *ctx;
769
770 if (!ctx && !(ctx = messaging_init(mem_ctx, server_id_self(),
771 ev_context()))) {
772 smb_panic("Could not init messaging context");
773 }
774 return ctx;
775 }
776
777 /****************************************************************
778 ****************************************************************/
779
780 static NTSTATUS process_dc_netbios(TALLOC_CTX *mem_ctx,
781 const char *domain_name,
782 uint32_t flags,
783 struct ip_service_name *dclist,
784 int num_dcs,
785 struct netr_DsRGetDCNameInfo **info)
786 {
787 struct sockaddr_storage ss;
788 struct ip_service ip_list;
789 enum nbt_name_type name_type = NBT_NAME_LOGON;
790
791 int i;
792 const char *dc_hostname, *dc_domain_name;
793 const char *dc_address;
794 uint32_t dc_address_type;
795 uint32_t dc_flags = 0;
796 const char *dc_name = NULL;
797 const char *dc_forest = NULL;
798 const char *dc_server_site = NULL;
799 const char *dc_client_site = NULL;
800 struct GUID *dc_domain_guid = NULL;
801 fstring tmp_dc_name;
802 struct messaging_context *msg_ctx = msg_context(mem_ctx);
803 struct nbt_ntlogon_packet *reply = NULL;
804 uint32_t nt_version = NETLOGON_VERSION_1 |
805 NETLOGON_VERSION_5 |
806 NETLOGON_VERSION_5EX_WITH_IP;
807
808 if (flags & DS_PDC_REQUIRED) {
809 name_type = NBT_NAME_PDC;
810 }
811
812 nt_version |= map_ds_flags_to_nt_version(flags);
813
814 DEBUG(10,("process_dc_netbios\n"));
815
816 for (i=0; i<num_dcs; i++) {
817
818 ip_list.ss = dclist[i].ss;
819 ip_list.port = 0;
820
821 if (!interpret_string_addr(&ss, dclist[i].hostname, AI_NUMERICHOST)) {
822 return NT_STATUS_UNSUCCESSFUL;
823 }
824
825 if (send_getdc_request(mem_ctx, msg_ctx,
826 &dclist[i].ss, domain_name,
827 NULL, nt_version))
828 {
829 int k;
830 smb_msleep(100);
831 for (k=0; k<5; k++) {
832 if (receive_getdc_response(mem_ctx,
833 &dclist[i].ss,
834 domain_name,
835 &dc_name,
836 &reply)) {
837 namecache_store(dc_name, NBT_NAME_SERVER, 1, &ip_list);
838 dc_hostname = dc_name;
839 dc_domain_name = talloc_strdup_upper(mem_ctx, domain_name);
840 NT_STATUS_HAVE_NO_MEMORY(dc_domain_name);
841 goto make_reply;
842 }
843 smb_msleep(500);
844 }
845 }
846
847 if (name_status_find(domain_name,
848 name_type,
849 NBT_NAME_SERVER,
850 &dclist[i].ss,
851 tmp_dc_name))
852 {
853 dc_hostname = tmp_dc_name;
854 dc_domain_name = talloc_strdup_upper(mem_ctx, domain_name);
855 namecache_store(tmp_dc_name, NBT_NAME_SERVER, 1, &ip_list);
856 goto make_reply;
857 }
858 }
859
860 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
861
862 make_reply:
863
864 if (reply && reply->command == NTLOGON_RESPONSE_FROM_PDC2) {
865
866 dc_flags |= reply->req.reply2.server_type;
867 dc_forest = reply->req.reply2.forest;
868 dc_server_site = reply->req.reply2.server_site;
869 dc_client_site = reply->req.reply2.client_site;
870
871 dc_domain_guid = &reply->req.reply2.domain_uuid;
872
873 if (flags & DS_RETURN_DNS_NAME) {
874 dc_domain_name = reply->req.reply2.dns_domain;
875 dc_hostname = reply->req.reply2.pdc_dns_name;
876 dc_flags |= DS_DNS_DOMAIN | DS_DNS_CONTROLLER;
877 } else if (flags & DS_RETURN_FLAT_NAME) {
878 dc_domain_name = reply->req.reply2.domain;
879 dc_hostname = reply->req.reply2.pdc_name;
880 }
881 }
882
883 if (flags & DS_IP_REQUIRED) {
884 char addr[INET6_ADDRSTRLEN];
885 print_sockaddr(addr, sizeof(addr), &dclist[i].ss);
886 dc_address = talloc_asprintf(mem_ctx, "\\\\%s", addr);
887 dc_address_type = DS_ADDRESS_TYPE_INET;
888 } else {
889 dc_address = talloc_asprintf(mem_ctx, "\\\\%s", dc_hostname);
890 dc_address_type = DS_ADDRESS_TYPE_NETBIOS;
891 }
892
893 if (flags & DS_PDC_REQUIRED) {
894 dc_flags |= NBT_SERVER_PDC | NBT_SERVER_WRITABLE;
895 }
896
897 if (dc_forest) {
898 dc_flags |= DS_DNS_FOREST;
899 }
900
901 return make_domain_controller_info(mem_ctx,
902 dc_hostname,
903 dc_address,
904 dc_address_type,
905 dc_domain_guid,
906 dc_domain_name,
907 dc_forest,
908 dc_flags,
909 dc_server_site,
910 dc_client_site,
911 info);
912 }
913
914 /****************************************************************
915 ****************************************************************/
916
917 static NTSTATUS dsgetdcname_rediscover(TALLOC_CTX *mem_ctx,
918 const char *domain_name,
919 struct GUID *domain_guid,
920 uint32_t flags,
921 const char *site_name,
922 struct netr_DsRGetDCNameInfo **info)
923 {
924 NTSTATUS status = NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
925 struct ip_service_name *dclist = NULL;
926 int num_dcs;
927
928 DEBUG(10,("dsgetdcname_rediscover\n"));
929
930 if (flags & DS_IS_FLAT_NAME) {
931
932 status = discover_dc_netbios(mem_ctx, domain_name, flags,
933 &dclist, &num_dcs);
934 NT_STATUS_NOT_OK_RETURN(status);
935
936 return process_dc_netbios(mem_ctx, domain_name, flags,
937 dclist, num_dcs, info);
938 }
939
940 if (flags & DS_IS_DNS_NAME) {
941
942 status = discover_dc_dns(mem_ctx, domain_name, domain_guid,
943 flags, site_name, &dclist, &num_dcs);
944 NT_STATUS_NOT_OK_RETURN(status);
945
946 return process_dc_dns(mem_ctx, domain_name, flags,
947 dclist, num_dcs, info);
948 }
949
950 status = discover_dc_dns(mem_ctx, domain_name, domain_guid, flags,
951 site_name, &dclist, &num_dcs);
952
953 if (NT_STATUS_IS_OK(status) && num_dcs != 0) {
954
955 status = process_dc_dns(mem_ctx, domain_name, flags, dclist,
956 num_dcs, info);
957 if (NT_STATUS_IS_OK(status)) {
958 return status;
959 }
960 }
961
962 status = discover_dc_netbios(mem_ctx, domain_name, flags, &dclist,
963 &num_dcs);
964 NT_STATUS_NOT_OK_RETURN(status);
965
966 return process_dc_netbios(mem_ctx, domain_name, flags, dclist,
967 num_dcs, info);
968 }
969
970 /********************************************************************
971 dsgetdcname.
972
973 This will be the only public function here.
974 ********************************************************************/
975
976 NTSTATUS dsgetdcname(TALLOC_CTX *mem_ctx,
977 const char *domain_name,
978 struct GUID *domain_guid,
979 const char *site_name,
980 uint32_t flags,
981 struct netr_DsRGetDCNameInfo **info)
982 {
983 NTSTATUS status = NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
984 struct netr_DsRGetDCNameInfo *myinfo = NULL;
985
986 DEBUG(10,("dsgetdcname: domain_name: %s, "
987 "domain_guid: %s, site_name: %s, flags: 0x%08x\n",
988 domain_name,
989 domain_guid ? GUID_string(mem_ctx, domain_guid) : "(null)",
990 site_name, flags));
991
992 *info = NULL;
993
994 if (!check_allowed_required_flags(flags)) {
995 DEBUG(0,("invalid flags specified\n"));
996 return NT_STATUS_INVALID_PARAMETER;
997 }
998
999 if (flags & DS_FORCE_REDISCOVERY) {
1000 goto rediscover;
1001 }
1002
1003 status = dsgetdcname_cached(mem_ctx, domain_name, domain_guid,
1004 flags, site_name, &myinfo);
1005 if (NT_STATUS_IS_OK(status)) {
1006 *info = myinfo;
1007 return status;
1008 }
1009
1010 if (flags & DS_BACKGROUND_ONLY) {
1011 return status;
1012 }
1013
1014 rediscover:
1015 status = dsgetdcname_rediscover(mem_ctx, domain_name,
1016 domain_guid, flags, site_name,
1017 &myinfo);
1018
1019 if (NT_STATUS_IS_OK(status)) {
1020 dsgetdcname_cache_store(mem_ctx, domain_name, myinfo);
1021 *info = myinfo;
1022 }
1023
1024 return status;
1025 }
Samba GIT mirror