?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
cldap: move out cldap object to fix the build.
[Samba.git] / source / libsmb / dsgetdcname.c
blob0db457a45af8366b4cb1a25589baaeaa8ab00ffd
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 static NTSTATUS make_dc_info_from_cldap_reply(TALLOC_CTX *mem_ctx,
36 uint32_t flags,
37 struct sockaddr_storage *ss,
38 uint32_t nt_version,
39 union nbt_cldap_netlogon *r,
40 struct netr_DsRGetDCNameInfo **info);
41
42 /****************************************************************
43 ****************************************************************/
44
45 void debug_dsdcinfo_flags(int lvl, uint32_t flags)
46 {
47 DEBUG(lvl,("debug_dsdcinfo_flags: 0x%08x\n\t", flags));
48
49 if (flags & DS_FORCE_REDISCOVERY)
50 DEBUGADD(lvl,("DS_FORCE_REDISCOVERY "));
51 if (flags & 0x000000002)
52 DEBUGADD(lvl,("0x00000002 "));
53 if (flags & 0x000000004)
54 DEBUGADD(lvl,("0x00000004 "));
55 if (flags & 0x000000008)
56 DEBUGADD(lvl,("0x00000008 "));
57 if (flags & DS_DIRECTORY_SERVICE_REQUIRED)
58 DEBUGADD(lvl,("DS_DIRECTORY_SERVICE_REQUIRED "));
59 if (flags & DS_DIRECTORY_SERVICE_PREFERRED)
60 DEBUGADD(lvl,("DS_DIRECTORY_SERVICE_PREFERRED "));
61 if (flags & DS_GC_SERVER_REQUIRED)
62 DEBUGADD(lvl,("DS_GC_SERVER_REQUIRED "));
63 if (flags & DS_PDC_REQUIRED)
64 DEBUGADD(lvl,("DS_PDC_REQUIRED "));
65 if (flags & DS_BACKGROUND_ONLY)
66 DEBUGADD(lvl,("DS_BACKGROUND_ONLY "));
67 if (flags & DS_IP_REQUIRED)
68 DEBUGADD(lvl,("DS_IP_REQUIRED "));
69 if (flags & DS_KDC_REQUIRED)
70 DEBUGADD(lvl,("DS_KDC_REQUIRED "));
71 if (flags & DS_TIMESERV_REQUIRED)
72 DEBUGADD(lvl,("DS_TIMESERV_REQUIRED "));
73 if (flags & DS_WRITABLE_REQUIRED)
74 DEBUGADD(lvl,("DS_WRITABLE_REQUIRED "));
75 if (flags & DS_GOOD_TIMESERV_PREFERRED)
76 DEBUGADD(lvl,("DS_GOOD_TIMESERV_PREFERRED "));
77 if (flags & DS_AVOID_SELF)
78 DEBUGADD(lvl,("DS_AVOID_SELF "));
79 if (flags & DS_ONLY_LDAP_NEEDED)
80 DEBUGADD(lvl,("DS_ONLY_LDAP_NEEDED "));
81 if (flags & DS_IS_FLAT_NAME)
82 DEBUGADD(lvl,("DS_IS_FLAT_NAME "));
83 if (flags & DS_IS_DNS_NAME)
84 DEBUGADD(lvl,("DS_IS_DNS_NAME "));
85 if (flags & 0x00040000)
86 DEBUGADD(lvl,("0x00040000 "));
87 if (flags & 0x00080000)
88 DEBUGADD(lvl,("0x00080000 "));
89 if (flags & 0x00100000)
90 DEBUGADD(lvl,("0x00100000 "));
91 if (flags & 0x00200000)
92 DEBUGADD(lvl,("0x00200000 "));
93 if (flags & 0x00400000)
94 DEBUGADD(lvl,("0x00400000 "));
95 if (flags & 0x00800000)
96 DEBUGADD(lvl,("0x00800000 "));
97 if (flags & 0x01000000)
98 DEBUGADD(lvl,("0x01000000 "));
99 if (flags & 0x02000000)
100 DEBUGADD(lvl,("0x02000000 "));
101 if (flags & 0x04000000)
102 DEBUGADD(lvl,("0x04000000 "));
103 if (flags & 0x08000000)
104 DEBUGADD(lvl,("0x08000000 "));
105 if (flags & 0x10000000)
106 DEBUGADD(lvl,("0x10000000 "));
107 if (flags & 0x20000000)
108 DEBUGADD(lvl,("0x20000000 "));
109 if (flags & DS_RETURN_DNS_NAME)
110 DEBUGADD(lvl,("DS_RETURN_DNS_NAME "));
111 if (flags & DS_RETURN_FLAT_NAME)
112 DEBUGADD(lvl,("DS_RETURN_FLAT_NAME "));
113 if (flags)
114 DEBUGADD(lvl,("\n"));
115 }
116
117 /****************************************************************
118 ****************************************************************/
119
120 static char *dsgetdcname_cache_key(TALLOC_CTX *mem_ctx, const char *domain)
121 {
122 if (!mem_ctx || !domain) {
123 return NULL;
124 }
125
126 return talloc_asprintf_strupper_m(mem_ctx, DSGETDCNAME_FMT, domain);
127 }
128
129 /****************************************************************
130 ****************************************************************/
131
132 static NTSTATUS dsgetdcname_cache_delete(TALLOC_CTX *mem_ctx,
133 const char *domain_name)
134 {
135 char *key;
136
137 if (!gencache_init()) {
138 return NT_STATUS_INTERNAL_DB_ERROR;
139 }
140
141 key = dsgetdcname_cache_key(mem_ctx, domain_name);
142 if (!key) {
143 return NT_STATUS_NO_MEMORY;
144 }
145
146 if (!gencache_del(key)) {
147 return NT_STATUS_UNSUCCESSFUL;
148 }
149
150 return NT_STATUS_OK;
151 }
152
153 /****************************************************************
154 ****************************************************************/
155
156 static NTSTATUS dsgetdcname_cache_store(TALLOC_CTX *mem_ctx,
157 const char *domain_name,
158 const DATA_BLOB *blob)
159 {
160 time_t expire_time;
161 char *key;
162 bool ret = false;
163
164 if (!gencache_init()) {
165 return NT_STATUS_INTERNAL_DB_ERROR;
166 }
167
168 key = dsgetdcname_cache_key(mem_ctx, domain_name);
169 if (!key) {
170 return NT_STATUS_NO_MEMORY;
171 }
172
173 expire_time = time(NULL) + DSGETDCNAME_CACHE_TTL;
174
175 if (gencache_lock_entry(key) != 0) {
176 return NT_STATUS_LOCK_NOT_GRANTED;
177 }
178
179 ret = gencache_set_data_blob(key, blob, expire_time);
180
181 gencache_unlock_entry(key);
182
183 return ret ? NT_STATUS_OK : NT_STATUS_UNSUCCESSFUL;
184 }
185
186 /****************************************************************
187 ****************************************************************/
188
189 #define SET_STRING(x) \
190 talloc_strdup(mem_ctx, x); \
191 NT_STATUS_HAVE_NO_MEMORY(x);
192
193 static NTSTATUS map_logon29_from_cldap_reply(TALLOC_CTX *mem_ctx,
194 uint32_t flags,
195 struct sockaddr_storage *ss,
196 uint32_t nt_version,
197 union nbt_cldap_netlogon *r,
198 struct nbt_cldap_netlogon_29 *p)
199 {
200 char addr[INET6_ADDRSTRLEN];
201
202 ZERO_STRUCTP(p);
203
204 print_sockaddr(addr, sizeof(addr), ss);
205 p->dc_sock_addr_size = 0x10;
206 p->dc_sock_addr.sa_family = 2;
207 p->dc_sock_addr.pdc_ip = talloc_strdup(mem_ctx, addr);
208
209 switch (nt_version & 0x000000ff) {
210 case 0:
211 return NT_STATUS_INVALID_PARAMETER;
212 case 1:
213 p->pdc_name = SET_STRING(r->logon1.pdc_name);
214 p->domain = SET_STRING(r->logon1.domain_name);
215
216 if (flags & DS_PDC_REQUIRED) {
217 p->server_type = NBT_SERVER_WRITABLE |
218 NBT_SERVER_PDC;
219 }
220 break;
221 case 2:
222 case 3:
223 p->pdc_name = SET_STRING(r->logon3.pdc_name);
224 p->domain = SET_STRING(r->logon3.domain_name);
225 p->pdc_dns_name = SET_STRING(r->logon3.pdc_dns_name);
226 p->dns_domain = SET_STRING(r->logon3.dns_domain);
227 p->server_type = r->logon3.server_type;
228 p->forest = SET_STRING(r->logon3.forest);
229 p->domain_uuid = r->logon3.domain_uuid;
230
231 break;
232 case 4:
233 case 5:
234 case 6:
235 case 7:
236 p->pdc_name = SET_STRING(r->logon5.pdc_name);
237 p->domain = SET_STRING(r->logon5.domain);
238 p->pdc_dns_name = SET_STRING(r->logon5.pdc_dns_name);
239 p->dns_domain = SET_STRING(r->logon5.dns_domain);
240 p->server_type = r->logon5.server_type;
241 p->forest = SET_STRING(r->logon5.forest);
242 p->domain_uuid = r->logon5.domain_uuid;
243 p->server_site = SET_STRING(r->logon5.server_site);
244 p->client_site = SET_STRING(r->logon5.client_site);
245
246 break;
247 case 8:
248 case 9:
249 case 10:
250 case 11:
251 case 12:
252 case 13:
253 case 14:
254 case 15:
255 p->pdc_name = SET_STRING(r->logon13.pdc_name);
256 p->domain = SET_STRING(r->logon13.domain);
257 p->pdc_dns_name = SET_STRING(r->logon13.pdc_dns_name);
258 p->dns_domain = SET_STRING(r->logon13.dns_domain);
259 p->server_type = r->logon13.server_type;
260 p->forest = SET_STRING(r->logon13.forest);
261 p->domain_uuid = r->logon13.domain_uuid;
262 p->server_site = SET_STRING(r->logon13.server_site);
263 p->client_site = SET_STRING(r->logon13.client_site);
264
265 break;
266 default:
267 p->pdc_name = SET_STRING(r->logon29.pdc_name);
268 p->domain = SET_STRING(r->logon29.domain);
269 p->pdc_dns_name = SET_STRING(r->logon29.pdc_dns_name);
270 p->dns_domain = SET_STRING(r->logon29.dns_domain);
271 p->server_type = r->logon29.server_type;
272 p->forest = SET_STRING(r->logon29.forest);
273 p->domain_uuid = r->logon29.domain_uuid;
274 p->server_site = SET_STRING(r->logon29.server_site);
275 p->client_site = SET_STRING(r->logon29.client_site);
276 p->next_closest_site = SET_STRING(r->logon29.next_closest_site);
277
278 break;
279 }
280
281 return NT_STATUS_OK;
282 }
283
284 /****************************************************************
285 ****************************************************************/
286
287 static NTSTATUS store_cldap_reply(TALLOC_CTX *mem_ctx,
288 uint32_t flags,
289 struct sockaddr_storage *ss,
290 uint32_t nt_version,
291 union nbt_cldap_netlogon *r)
292 {
293 DATA_BLOB blob;
294 enum ndr_err_code ndr_err;
295 NTSTATUS status;
296 struct nbt_cldap_netlogon_29 logon29;
297
298 status = map_logon29_from_cldap_reply(mem_ctx, flags, ss,
299 nt_version, r, &logon29);
300 if (!NT_STATUS_IS_OK(status)) {
301 return status;
302 }
303
304 ndr_err = ndr_push_struct_blob(&blob, mem_ctx, &logon29,
305 (ndr_push_flags_fn_t)ndr_push_nbt_cldap_netlogon_29);
306 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
307 return ndr_map_error2ntstatus(ndr_err);
308 }
309
310 if (logon29.domain) {
311 status = dsgetdcname_cache_store(mem_ctx, logon29.domain, &blob);
312 if (!NT_STATUS_IS_OK(status)) {
313 goto done;
314 }
315 }
316 if (logon29.dns_domain) {
317 status = dsgetdcname_cache_store(mem_ctx, logon29.dns_domain, &blob);
318 if (!NT_STATUS_IS_OK(status)) {
319 goto done;
320 }
321 }
322
323 done:
324 data_blob_free(&blob);
325
326 return status;
327 }
328
329 /****************************************************************
330 ****************************************************************/
331
332 static NTSTATUS dsgetdcname_cache_refresh(TALLOC_CTX *mem_ctx,
333 const char *domain_name,
334 struct GUID *domain_guid,
335 uint32_t flags,
336 const char *site_name,
337 struct netr_DsRGetDCNameInfo *info)
338 {
339 struct netr_DsRGetDCNameInfo *dc_info;
340
341 return dsgetdcname(mem_ctx,
342 domain_name,
343 domain_guid,
344 site_name,
345 flags | DS_FORCE_REDISCOVERY,
346 &dc_info);
347 }
348
349 /****************************************************************
350 ****************************************************************/
351
352 static uint32_t get_cldap_reply_server_flags(union nbt_cldap_netlogon *r,
353 uint32_t nt_version)
354 {
355 switch (nt_version & 0x000000ff) {
356 case 0:
357 case 1:
358 return 0;
359 case 2:
360 case 3:
361 return r->logon3.server_type;
362 case 4:
363 case 5:
364 case 6:
365 case 7:
366 return r->logon5.server_type;
367 case 8:
368 case 9:
369 case 10:
370 case 11:
371 case 12:
372 case 13:
373 case 14:
374 case 15:
375 return r->logon13.server_type;
376 default:
377 return r->logon29.server_type;
378 }
379 }
380
381 /****************************************************************
382 ****************************************************************/
383
384 #define RETURN_ON_FALSE(x) if (!x) return false;
385
386 static bool check_cldap_reply_required_flags(uint32_t ret_flags,
387 uint32_t req_flags)
388 {
389 if (ret_flags == 0) {
390 return true;
391 }
392
393 if (req_flags & DS_PDC_REQUIRED)
394 RETURN_ON_FALSE(ret_flags & NBT_SERVER_PDC);
395
396 if (req_flags & DS_GC_SERVER_REQUIRED)
397 RETURN_ON_FALSE(ret_flags & NBT_SERVER_GC);
398
399 if (req_flags & DS_ONLY_LDAP_NEEDED)
400 RETURN_ON_FALSE(ret_flags & NBT_SERVER_LDAP);
401
402 if ((req_flags & DS_DIRECTORY_SERVICE_REQUIRED) ||
403 (req_flags & DS_DIRECTORY_SERVICE_PREFERRED))
404 RETURN_ON_FALSE(ret_flags & NBT_SERVER_DS);
405
406 if (req_flags & DS_KDC_REQUIRED)
407 RETURN_ON_FALSE(ret_flags & NBT_SERVER_KDC);
408
409 if (req_flags & DS_TIMESERV_REQUIRED)
410 RETURN_ON_FALSE(ret_flags & NBT_SERVER_TIMESERV);
411
412 if (req_flags & DS_WRITABLE_REQUIRED)
413 RETURN_ON_FALSE(ret_flags & NBT_SERVER_WRITABLE);
414
415 return true;
416 }
417
418 /****************************************************************
419 ****************************************************************/
420
421 static NTSTATUS dsgetdcname_cache_fetch(TALLOC_CTX *mem_ctx,
422 const char *domain_name,
423 struct GUID *domain_guid,
424 uint32_t flags,
425 const char *site_name,
426 struct netr_DsRGetDCNameInfo **info_p,
427 bool *expired)
428 {
429 char *key;
430 DATA_BLOB blob;
431 enum ndr_err_code ndr_err;
432 struct netr_DsRGetDCNameInfo *info;
433 union nbt_cldap_netlogon p;
434 struct nbt_cldap_netlogon_29 r;
435 NTSTATUS status;
436
437 if (!gencache_init()) {
438 return NT_STATUS_INTERNAL_DB_ERROR;
439 }
440
441 key = dsgetdcname_cache_key(mem_ctx, domain_name);
442 if (!key) {
443 return NT_STATUS_NO_MEMORY;
444 }
445
446 if (!gencache_get_data_blob(key, &blob, expired)) {
447 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
448 }
449
450 info = TALLOC_ZERO_P(mem_ctx, struct netr_DsRGetDCNameInfo);
451 if (!info) {
452 return NT_STATUS_NO_MEMORY;
453 }
454
455 ndr_err = ndr_pull_struct_blob(&blob, mem_ctx, &r,
456 (ndr_pull_flags_fn_t)ndr_pull_nbt_cldap_netlogon_29);
457
458 data_blob_free(&blob);
459 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
460 dsgetdcname_cache_delete(mem_ctx, domain_name);
461 return ndr_map_error2ntstatus(ndr_err);
462 }
463
464 p.logon29 = r;
465
466 status = make_dc_info_from_cldap_reply(mem_ctx, flags, NULL,
467 NETLOGON_VERSION_WITH_CLOSEST_SITE,
468 &p, &info);
469 if (!NT_STATUS_IS_OK(status)) {
470 return status;
471 }
472
473 if (DEBUGLEVEL >= 10) {
474 NDR_PRINT_DEBUG(netr_DsRGetDCNameInfo, info);
475 }
476
477 /* check flags */
478 if (!check_cldap_reply_required_flags(info->dc_flags, flags)) {
479 DEBUG(10,("invalid flags\n"));
480 return NT_STATUS_INVALID_PARAMETER;
481 }
482
483 if ((flags & DS_IP_REQUIRED) &&
484 (info->dc_address_type != DS_ADDRESS_TYPE_INET)) {
485 return NT_STATUS_INVALID_PARAMETER_MIX;
486 }
487
488 *info_p = info;
489
490 return NT_STATUS_OK;
491 }
492
493 /****************************************************************
494 ****************************************************************/
495
496 static NTSTATUS dsgetdcname_cached(TALLOC_CTX *mem_ctx,
497 const char *domain_name,
498 struct GUID *domain_guid,
499 uint32_t flags,
500 const char *site_name,
501 struct netr_DsRGetDCNameInfo **info)
502 {
503 NTSTATUS status;
504 bool expired = false;
505
506 status = dsgetdcname_cache_fetch(mem_ctx, domain_name, domain_guid,
507 flags, site_name, info, &expired);
508 if (!NT_STATUS_IS_OK(status)) {
509 DEBUG(10,("dsgetdcname_cached: cache fetch failed with: %s\n",
510 nt_errstr(status)));
511 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
512 }
513
514 if (flags & DS_BACKGROUND_ONLY) {
515 return status;
516 }
517
518 if (expired) {
519 status = dsgetdcname_cache_refresh(mem_ctx, domain_name,
520 domain_guid, flags,
521 site_name, *info);
522 if (!NT_STATUS_IS_OK(status)) {
523 return status;
524 }
525 }
526
527 return status;
528 }
529
530 /****************************************************************
531 ****************************************************************/
532
533 static bool check_allowed_required_flags(uint32_t flags)
534 {
535 uint32_t return_type = flags & (DS_RETURN_FLAT_NAME|DS_RETURN_DNS_NAME);
536 uint32_t offered_type = flags & (DS_IS_FLAT_NAME|DS_IS_DNS_NAME);
537 uint32_t query_type = flags & (DS_BACKGROUND_ONLY|DS_FORCE_REDISCOVERY);
538
539 /* FIXME: check for DSGETDC_VALID_FLAGS and check for excluse bits
540 * (DS_PDC_REQUIRED, DS_KDC_REQUIRED, DS_GC_SERVER_REQUIRED) */
541
542 debug_dsdcinfo_flags(10, flags);
543
544 if (return_type == (DS_RETURN_FLAT_NAME|DS_RETURN_DNS_NAME)) {
545 return false;
546 }
547
548 if (offered_type == (DS_IS_DNS_NAME|DS_IS_FLAT_NAME)) {
549 return false;
550 }
551
552 if (query_type == (DS_BACKGROUND_ONLY|DS_FORCE_REDISCOVERY)) {
553 return false;
554 }
555
556 #if 0
557 if ((flags & DS_RETURN_DNS_NAME) && (!(flags & DS_IP_REQUIRED))) {
558 printf("gd: here5 \n");
559 return false;
560 }
561 #endif
562 return true;
563 }
564
565 /****************************************************************
566 ****************************************************************/
567
568 static NTSTATUS discover_dc_netbios(TALLOC_CTX *mem_ctx,
569 const char *domain_name,
570 uint32_t flags,
571 struct ip_service_name **returned_dclist,
572 int *returned_count)
573 {
574 NTSTATUS status;
575 enum nbt_name_type name_type = NBT_NAME_LOGON;
576 struct ip_service *iplist;
577 int i;
578 struct ip_service_name *dclist = NULL;
579 int count;
580
581 *returned_dclist = NULL;
582 *returned_count = 0;
583
584 if (lp_disable_netbios()) {
585 return NT_STATUS_NOT_SUPPORTED;
586 }
587
588 if (flags & DS_PDC_REQUIRED) {
589 name_type = NBT_NAME_PDC;
590 }
591
592 status = internal_resolve_name(domain_name, name_type, NULL,
593 &iplist, &count,
594 "lmhosts wins bcast");
595 if (!NT_STATUS_IS_OK(status)) {
596 DEBUG(10,("discover_dc_netbios: failed to find DC\n"));
597 return status;
598 }
599
600 dclist = TALLOC_ZERO_ARRAY(mem_ctx, struct ip_service_name, count);
601 if (!dclist) {
602 return NT_STATUS_NO_MEMORY;
603 }
604
605 for (i=0; i<count; i++) {
606
607 char addr[INET6_ADDRSTRLEN];
608 struct ip_service_name *r = &dclist[i];
609
610 print_sockaddr(addr, sizeof(addr),
611 &iplist[i].ss);
612
613 r->ss = iplist[i].ss;
614 r->port = iplist[i].port;
615 r->hostname = talloc_strdup(mem_ctx, addr);
616 if (!r->hostname) {
617 return NT_STATUS_NO_MEMORY;
618 }
619
620 }
621
622 *returned_dclist = dclist;
623 *returned_count = count;
624
625 return NT_STATUS_OK;
626 }
627
628 /****************************************************************
629 ****************************************************************/
630
631 static NTSTATUS discover_dc_dns(TALLOC_CTX *mem_ctx,
632 const char *domain_name,
633 struct GUID *domain_guid,
634 uint32_t flags,
635 const char *site_name,
636 struct ip_service_name **returned_dclist,
637 int *return_count)
638 {
639 int i, j;
640 NTSTATUS status;
641 struct dns_rr_srv *dcs = NULL;
642 int numdcs = 0;
643 int numaddrs = 0;
644 struct ip_service_name *dclist = NULL;
645 int count = 0;
646
647 if (flags & DS_PDC_REQUIRED) {
648 status = ads_dns_query_pdc(mem_ctx, domain_name,
649 &dcs, &numdcs);
650 } else if (flags & DS_GC_SERVER_REQUIRED) {
651 status = ads_dns_query_gcs(mem_ctx, domain_name, site_name,
652 &dcs, &numdcs);
653 } else if (flags & DS_KDC_REQUIRED) {
654 status = ads_dns_query_kdcs(mem_ctx, domain_name, site_name,
655 &dcs, &numdcs);
656 } else if (flags & DS_DIRECTORY_SERVICE_REQUIRED) {
657 status = ads_dns_query_dcs(mem_ctx, domain_name, site_name,
658 &dcs, &numdcs);
659 } else if (domain_guid) {
660 status = ads_dns_query_dcs_guid(mem_ctx, domain_name,
661 domain_guid, &dcs, &numdcs);
662 } else {
663 status = ads_dns_query_dcs(mem_ctx, domain_name, site_name,
664 &dcs, &numdcs);
665 }
666
667 if (!NT_STATUS_IS_OK(status)) {
668 return status;
669 }
670
671 if (numdcs == 0) {
672 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
673 }
674
675 for (i=0;i<numdcs;i++) {
676 numaddrs += MAX(dcs[i].num_ips,1);
677 }
678
679 dclist = TALLOC_ZERO_ARRAY(mem_ctx,
680 struct ip_service_name,
681 numaddrs);
682 if (!dclist) {
683 return NT_STATUS_NO_MEMORY;
684 }
685
686 /* now unroll the list of IP addresses */
687
688 *return_count = 0;
689 i = 0;
690 j = 0;
691
692 while ((i < numdcs) && (count < numaddrs)) {
693
694 struct ip_service_name *r = &dclist[count];
695
696 r->port = dcs[count].port;
697 r->hostname = dcs[count].hostname;
698
699 /* If we don't have an IP list for a name, lookup it up */
700
701 if (!dcs[i].ss_s) {
702 interpret_string_addr(&r->ss, dcs[i].hostname, 0);
703 i++;
704 j = 0;
705 } else {
706 /* use the IP addresses from the SRV sresponse */
707
708 if (j >= dcs[i].num_ips) {
709 i++;
710 j = 0;
711 continue;
712 }
713
714 r->ss = dcs[i].ss_s[j];
715 j++;
716 }
717
718 /* make sure it is a valid IP. I considered checking the
719 * negative connection cache, but this is the wrong place for
720 * it. Maybe only as a hac. After think about it, if all of
721 * the IP addresses retuend from DNS are dead, what hope does a
722 * netbios name lookup have? The standard reason for falling
723 * back to netbios lookups is that our DNS server doesn't know
724 * anything about the DC's -- jerry */
725
726 if (!is_zero_addr(&r->ss)) {
727 count++;
728 continue;
729 }
730 }
731
732 *returned_dclist = dclist;
733 *return_count = count;
734
735 if (count > 0) {
736 return NT_STATUS_OK;
737 }
738
739 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
740 }
741
742 /****************************************************************
743 ****************************************************************/
744
745 static NTSTATUS make_domain_controller_info(TALLOC_CTX *mem_ctx,
746 const char *dc_unc,
747 const char *dc_address,
748 uint32_t dc_address_type,
749 const struct GUID *domain_guid,
750 const char *domain_name,
751 const char *forest_name,
752 uint32_t flags,
753 const char *dc_site_name,
754 const char *client_site_name,
755 struct netr_DsRGetDCNameInfo **info_out)
756 {
757 struct netr_DsRGetDCNameInfo *info;
758
759 info = TALLOC_ZERO_P(mem_ctx, struct netr_DsRGetDCNameInfo);
760 NT_STATUS_HAVE_NO_MEMORY(info);
761
762 if (dc_unc) {
763 info->dc_unc = talloc_strdup(mem_ctx, dc_unc);
764 NT_STATUS_HAVE_NO_MEMORY(info->dc_unc);
765 }
766
767 if (dc_address) {
768 if (!(dc_address[0] == '\\' && dc_address[1] == '\\')) {
769 info->dc_address = talloc_asprintf(mem_ctx, "\\\\%s",
770 dc_address);
771 } else {
772 info->dc_address = talloc_strdup(mem_ctx, dc_address);
773 }
774 NT_STATUS_HAVE_NO_MEMORY(info->dc_address);
775 }
776
777 info->dc_address_type = dc_address_type;
778
779 if (domain_guid) {
780 info->domain_guid = *domain_guid;
781 }
782
783 if (domain_name) {
784 info->domain_name = talloc_strdup(mem_ctx, domain_name);
785 NT_STATUS_HAVE_NO_MEMORY(info->domain_name);
786 }
787
788 if (forest_name) {
789 info->forest_name = talloc_strdup(mem_ctx, forest_name);
790 NT_STATUS_HAVE_NO_MEMORY(info->forest_name);
791 flags |= DS_DNS_FOREST;
792 }
793
794 info->dc_flags = flags;
795
796 if (dc_site_name) {
797 info->dc_site_name = talloc_strdup(mem_ctx, dc_site_name);
798 NT_STATUS_HAVE_NO_MEMORY(info->dc_site_name);
799 }
800
801 if (client_site_name) {
802 info->client_site_name = talloc_strdup(mem_ctx,
803 client_site_name);
804 NT_STATUS_HAVE_NO_MEMORY(info->client_site_name);
805 }
806
807 *info_out = info;
808
809 return NT_STATUS_OK;
810 }
811
812 /****************************************************************
813 ****************************************************************/
814
815 static NTSTATUS make_dc_info_from_cldap_reply(TALLOC_CTX *mem_ctx,
816 uint32_t flags,
817 struct sockaddr_storage *ss,
818 uint32_t nt_version,
819 union nbt_cldap_netlogon *r,
820 struct netr_DsRGetDCNameInfo **info)
821 {
822 const char *dc_hostname, *dc_domain_name;
823 const char *dc_address = NULL;
824 const char *dc_forest = NULL;
825 uint32_t dc_address_type = 0;
826 uint32_t dc_flags = 0;
827 struct GUID *dc_domain_guid = NULL;
828 const char *dc_server_site = NULL;
829 const char *dc_client_site = NULL;
830
831 char addr[INET6_ADDRSTRLEN];
832
833 if (ss) {
834 print_sockaddr(addr, sizeof(addr), ss);
835 dc_address = addr;
836 dc_address_type = DS_ADDRESS_TYPE_INET;
837 }
838
839 switch (nt_version & 0x000000ff) {
840 case 0:
841 return NT_STATUS_INVALID_PARAMETER;
842 case 1:
843 if (!ss) {
844 dc_address = r->logon1.pdc_name;
845 dc_address_type = DS_ADDRESS_TYPE_NETBIOS;
846 }
847
848 dc_hostname = r->logon1.pdc_name;
849 dc_domain_name = r->logon1.domain_name;
850 if (flags & DS_PDC_REQUIRED) {
851 dc_flags = NBT_SERVER_WRITABLE | NBT_SERVER_PDC;
852 }
853 break;
854 case 2:
855 case 3:
856 if (!ss) {
857 dc_address = r->logon3.pdc_ip;
858 dc_address_type = DS_ADDRESS_TYPE_INET;
859 }
860
861 switch (flags & 0xf0000000) {
862 case DS_RETURN_FLAT_NAME:
863 dc_hostname = r->logon3.pdc_name;
864 dc_domain_name = r->logon3.domain_name;
865 break;
866 case DS_RETURN_DNS_NAME:
867 default:
868 dc_hostname = r->logon3.pdc_dns_name;
869 dc_domain_name = r->logon3.dns_domain;
870 dc_flags |= DS_DNS_DOMAIN | DS_DNS_CONTROLLER;
871 break;
872 }
873
874 dc_flags |= r->logon3.server_type;
875 dc_forest = r->logon3.forest;
876 dc_domain_guid = &r->logon3.domain_uuid;
877
878 break;
879 case 4:
880 case 5:
881 case 6:
882 case 7:
883 if (!ss) {
884 dc_address = r->logon5.pdc_name;
885 dc_address_type = DS_ADDRESS_TYPE_NETBIOS;
886 }
887
888 switch (flags & 0xf0000000) {
889 case DS_RETURN_FLAT_NAME:
890 dc_hostname = r->logon5.pdc_name;
891 dc_domain_name = r->logon5.domain;
892 break;
893 case DS_RETURN_DNS_NAME:
894 default:
895 dc_hostname = r->logon5.pdc_dns_name;
896 dc_domain_name = r->logon5.dns_domain;
897 dc_flags |= DS_DNS_DOMAIN | DS_DNS_CONTROLLER;
898 break;
899 }
900
901 dc_flags |= r->logon5.server_type;
902 dc_forest = r->logon5.forest;
903 dc_domain_guid = &r->logon5.domain_uuid;
904 dc_server_site = r->logon5.server_site;
905 dc_client_site = r->logon5.client_site;
906
907 break;
908 case 8:
909 case 9:
910 case 10:
911 case 11:
912 case 12:
913 case 13:
914 case 14:
915 case 15:
916 if (!ss) {
917 dc_address = r->logon13.dc_sock_addr.pdc_ip;
918 dc_address_type = DS_ADDRESS_TYPE_INET;
919 }
920
921 switch (flags & 0xf0000000) {
922 case DS_RETURN_FLAT_NAME:
923 dc_hostname = r->logon13.pdc_name;
924 dc_domain_name = r->logon13.domain;
925 break;
926 case DS_RETURN_DNS_NAME:
927 default:
928 dc_hostname = r->logon13.pdc_dns_name;
929 dc_domain_name = r->logon13.dns_domain;
930 dc_flags |= DS_DNS_DOMAIN | DS_DNS_CONTROLLER;
931 break;
932 }
933
934 dc_flags |= r->logon13.server_type;
935 dc_forest = r->logon13.forest;
936 dc_domain_guid = &r->logon13.domain_uuid;
937 dc_server_site = r->logon13.server_site;
938 dc_client_site = r->logon13.client_site;
939
940 break;
941 default:
942 if (!ss) {
943 dc_address = r->logon29.dc_sock_addr.pdc_ip;
944 dc_address_type = DS_ADDRESS_TYPE_INET;
945 }
946
947 switch (flags & 0xf0000000) {
948 case DS_RETURN_FLAT_NAME:
949 dc_hostname = r->logon29.pdc_name;
950 dc_domain_name = r->logon29.domain;
951 break;
952 case DS_RETURN_DNS_NAME:
953 default:
954 dc_hostname = r->logon29.pdc_dns_name;
955 dc_domain_name = r->logon29.dns_domain;
956 dc_flags |= DS_DNS_DOMAIN | DS_DNS_CONTROLLER;
957 break;
958 }
959
960 dc_flags |= r->logon29.server_type;
961 dc_forest = r->logon29.forest;
962 dc_domain_guid = &r->logon29.domain_uuid;
963 dc_server_site = r->logon29.server_site;
964 dc_client_site = r->logon29.client_site;
965
966 break;
967 }
968
969 return make_domain_controller_info(mem_ctx,
970 dc_hostname,
971 dc_address,
972 dc_address_type,
973 dc_domain_guid,
974 dc_domain_name,
975 dc_forest,
976 dc_flags,
977 dc_server_site,
978 dc_client_site,
979 info);
980 }
981
982 /****************************************************************
983 ****************************************************************/
984
985 static uint32_t map_ds_flags_to_nt_version(uint32_t flags)
986 {
987 uint32_t nt_version = 0;
988
989 if (flags & DS_PDC_REQUIRED) {
990 nt_version |= NETLOGON_VERSION_PDC;
991 }
992
993 if (flags & DS_GC_SERVER_REQUIRED) {
994 nt_version |= NETLOGON_VERSION_GC;
995 }
996
997 if (flags & DS_TRY_NEXTCLOSEST_SITE) {
998 nt_version |= NETLOGON_VERSION_WITH_CLOSEST_SITE;
999 }
1000
1001 if (flags & DS_IP_REQUIRED) {
1002 nt_version |= NETLOGON_VERSION_IP;
1003 }
1004
1005 return nt_version;
1006 }
1007
1008 /****************************************************************
1009 ****************************************************************/
1010
1011 static NTSTATUS process_dc_dns(TALLOC_CTX *mem_ctx,
1012 const char *domain_name,
1013 uint32_t flags,
1014 struct ip_service_name *dclist,
1015 int num_dcs,
1016 struct netr_DsRGetDCNameInfo **info)
1017 {
1018 int i = 0;
1019 bool valid_dc = false;
1020 union nbt_cldap_netlogon *r = NULL;
1021 uint32_t nt_version = NETLOGON_VERSION_5 |
1022 NETLOGON_VERSION_5EX;
1023 uint32_t ret_flags = 0;
1024 NTSTATUS status;
1025
1026 nt_version |= map_ds_flags_to_nt_version(flags);
1027
1028 for (i=0; i<num_dcs; i++) {
1029
1030 DEBUG(10,("LDAP ping to %s\n", dclist[i].hostname));
1031
1032 if (ads_cldap_netlogon(mem_ctx, dclist[i].hostname,
1033 domain_name,
1034 &nt_version,
1035 &r))
1036 {
1037 ret_flags = get_cldap_reply_server_flags(r, nt_version);
1038
1039 if (check_cldap_reply_required_flags(ret_flags, flags)) {
1040 valid_dc = true;
1041 break;
1042 }
1043 }
1044
1045 continue;
1046 }
1047
1048 if (!valid_dc) {
1049 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
1050 }
1051
1052 status = make_dc_info_from_cldap_reply(mem_ctx, flags, &dclist[i].ss,
1053 nt_version, r, info);
1054 if (NT_STATUS_IS_OK(status)) {
1055 return store_cldap_reply(mem_ctx, flags, &dclist[i].ss,
1056 nt_version, r);
1057 }
1058
1059 return status;
1060 }
1061
1062 /****************************************************************
1063 ****************************************************************/
1064
1065 static struct event_context *ev_context(void)
1066 {
1067 static struct event_context *ctx;
1068
1069 if (!ctx && !(ctx = event_context_init(NULL))) {
1070 smb_panic("Could not init event context");
1071 }
1072 return ctx;
1073 }
1074
1075 /****************************************************************
1076 ****************************************************************/
1077
1078 static struct messaging_context *msg_context(TALLOC_CTX *mem_ctx)
1079 {
1080 static struct messaging_context *ctx;
1081
1082 if (!ctx && !(ctx = messaging_init(mem_ctx, server_id_self(),
1083 ev_context()))) {
1084 smb_panic("Could not init messaging context");
1085 }
1086 return ctx;
1087 }
1088
1089 /****************************************************************
1090 ****************************************************************/
1091
1092 static NTSTATUS process_dc_netbios(TALLOC_CTX *mem_ctx,
1093 const char *domain_name,
1094 uint32_t flags,
1095 struct ip_service_name *dclist,
1096 int num_dcs,
1097 struct netr_DsRGetDCNameInfo **info)
1098 {
1099 struct sockaddr_storage ss;
1100 struct ip_service ip_list;
1101 enum nbt_name_type name_type = NBT_NAME_LOGON;
1102 NTSTATUS status;
1103 int i;
1104 const char *dc_name = NULL;
1105 fstring tmp_dc_name;
1106 struct messaging_context *msg_ctx = msg_context(mem_ctx);
1107 union nbt_cldap_netlogon *r = NULL;
1108 uint32_t nt_version = NETLOGON_VERSION_1 |
1109 NETLOGON_VERSION_5 |
1110 NETLOGON_VERSION_5EX_WITH_IP;
1111
1112 if (flags & DS_PDC_REQUIRED) {
1113 name_type = NBT_NAME_PDC;
1114 }
1115
1116 nt_version |= map_ds_flags_to_nt_version(flags);
1117
1118 DEBUG(10,("process_dc_netbios\n"));
1119
1120 for (i=0; i<num_dcs; i++) {
1121
1122 ip_list.ss = dclist[i].ss;
1123 ip_list.port = 0;
1124
1125 if (!interpret_string_addr(&ss, dclist[i].hostname, AI_NUMERICHOST)) {
1126 return NT_STATUS_UNSUCCESSFUL;
1127 }
1128
1129 if (send_getdc_request(mem_ctx, msg_ctx,
1130 &dclist[i].ss, domain_name,
1131 NULL, nt_version))
1132 {
1133 int k;
1134 smb_msleep(300);
1135 for (k=0; k<5; k++) {
1136 if (receive_getdc_response(mem_ctx,
1137 &dclist[i].ss,
1138 domain_name,
1139 &nt_version,
1140 &dc_name,
1141 &r)) {
1142 namecache_store(dc_name, NBT_NAME_SERVER, 1, &ip_list);
1143 goto make_reply;
1144 }
1145 smb_msleep(1500);
1146 }
1147 }
1148
1149 if (name_status_find(domain_name,
1150 name_type,
1151 NBT_NAME_SERVER,
1152 &dclist[i].ss,
1153 tmp_dc_name))
1154 {
1155 struct nbt_cldap_netlogon_1 logon1;
1156
1157 r = TALLOC_ZERO_P(mem_ctx, union nbt_cldap_netlogon);
1158 NT_STATUS_HAVE_NO_MEMORY(r);
1159
1160 ZERO_STRUCT(logon1);
1161
1162 nt_version = NETLOGON_VERSION_1;
1163
1164 logon1.nt_version = nt_version;
1165 logon1.pdc_name = tmp_dc_name;
1166 logon1.domain_name = talloc_strdup_upper(mem_ctx, domain_name);
1167 NT_STATUS_HAVE_NO_MEMORY(logon1.domain_name);
1168
1169 r->logon1 = logon1;
1170
1171 namecache_store(tmp_dc_name, NBT_NAME_SERVER, 1, &ip_list);
1172
1173 goto make_reply;
1174 }
1175 }
1176
1177 return NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
1178
1179 make_reply:
1180
1181 status = make_dc_info_from_cldap_reply(mem_ctx, flags, &dclist[i].ss,
1182 nt_version, r, info);
1183 if (NT_STATUS_IS_OK(status)) {
1184 return store_cldap_reply(mem_ctx, flags, &dclist[i].ss,
1185 nt_version, r);
1186 }
1187
1188 return status;
1189 }
1190
1191 /****************************************************************
1192 ****************************************************************/
1193
1194 static NTSTATUS dsgetdcname_rediscover(TALLOC_CTX *mem_ctx,
1195 const char *domain_name,
1196 struct GUID *domain_guid,
1197 uint32_t flags,
1198 const char *site_name,
1199 struct netr_DsRGetDCNameInfo **info)
1200 {
1201 NTSTATUS status = NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
1202 struct ip_service_name *dclist = NULL;
1203 int num_dcs;
1204
1205 DEBUG(10,("dsgetdcname_rediscover\n"));
1206
1207 if (flags & DS_IS_FLAT_NAME) {
1208
1209 status = discover_dc_netbios(mem_ctx, domain_name, flags,
1210 &dclist, &num_dcs);
1211 NT_STATUS_NOT_OK_RETURN(status);
1212
1213 return process_dc_netbios(mem_ctx, domain_name, flags,
1214 dclist, num_dcs, info);
1215 }
1216
1217 if (flags & DS_IS_DNS_NAME) {
1218
1219 status = discover_dc_dns(mem_ctx, domain_name, domain_guid,
1220 flags, site_name, &dclist, &num_dcs);
1221 NT_STATUS_NOT_OK_RETURN(status);
1222
1223 return process_dc_dns(mem_ctx, domain_name, flags,
1224 dclist, num_dcs, info);
1225 }
1226
1227 status = discover_dc_dns(mem_ctx, domain_name, domain_guid, flags,
1228 site_name, &dclist, &num_dcs);
1229
1230 if (NT_STATUS_IS_OK(status) && num_dcs != 0) {
1231
1232 status = process_dc_dns(mem_ctx, domain_name, flags, dclist,
1233 num_dcs, info);
1234 if (NT_STATUS_IS_OK(status)) {
1235 return status;
1236 }
1237 }
1238
1239 status = discover_dc_netbios(mem_ctx, domain_name, flags, &dclist,
1240 &num_dcs);
1241 NT_STATUS_NOT_OK_RETURN(status);
1242
1243 return process_dc_netbios(mem_ctx, domain_name, flags, dclist,
1244 num_dcs, info);
1245 }
1246
1247 /********************************************************************
1248 dsgetdcname.
1249
1250 This will be the only public function here.
1251 ********************************************************************/
1252
1253 NTSTATUS dsgetdcname(TALLOC_CTX *mem_ctx,
1254 const char *domain_name,
1255 struct GUID *domain_guid,
1256 const char *site_name,
1257 uint32_t flags,
1258 struct netr_DsRGetDCNameInfo **info)
1259 {
1260 NTSTATUS status = NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
1261 struct netr_DsRGetDCNameInfo *myinfo = NULL;
1262
1263 DEBUG(10,("dsgetdcname: domain_name: %s, "
1264 "domain_guid: %s, site_name: %s, flags: 0x%08x\n",
1265 domain_name,
1266 domain_guid ? GUID_string(mem_ctx, domain_guid) : "(null)",
1267 site_name, flags));
1268
1269 *info = NULL;
1270
1271 if (!check_allowed_required_flags(flags)) {
1272 DEBUG(0,("invalid flags specified\n"));
1273 return NT_STATUS_INVALID_PARAMETER;
1274 }
1275
1276 if (flags & DS_FORCE_REDISCOVERY) {
1277 goto rediscover;
1278 }
1279
1280 status = dsgetdcname_cached(mem_ctx, domain_name, domain_guid,
1281 flags, site_name, &myinfo);
1282 if (NT_STATUS_IS_OK(status)) {
1283 *info = myinfo;
1284 return status;
1285 }
1286
1287 if (flags & DS_BACKGROUND_ONLY) {
1288 return status;
1289 }
1290
1291 rediscover:
1292 status = dsgetdcname_rediscover(mem_ctx, domain_name,
1293 domain_guid, flags, site_name,
1294 &myinfo);
1295
1296 if (NT_STATUS_IS_OK(status)) {
1297 *info = myinfo;
1298 }
1299
1300 return status;
1301 }
Samba GIT mirror