?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Fix convert_ss2service() to filter out zero addresses.
[Samba.git] / source3 / libsmb / namequery.c
blobb68448e16aedcd83cca86a71b59e785df9004188
1 /*
2 Unix SMB/CIFS implementation.
3 name query routines
4 Copyright (C) Andrew Tridgell 1994-1998
5 Copyright (C) Jeremy Allison 2007.
6
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3 of the License, or
10 (at your option) any later version.
11
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
16
17 You should have received a copy of the GNU General Public License
18 along with this program. If not, see <http://www.gnu.org/licenses/>.
19 */
20
21 #include "includes.h"
22
23 /* nmbd.c sets this to True. */
24 bool global_in_nmbd = False;
25
26 /****************************
27 * SERVER AFFINITY ROUTINES *
28 ****************************/
29
30 /* Server affinity is the concept of preferring the last domain
31 controller with whom you had a successful conversation */
32
33 /****************************************************************************
34 ****************************************************************************/
35 #define SAFKEY_FMT "SAF/DOMAIN/%s"
36 #define SAF_TTL 900
37 #define SAFJOINKEY_FMT "SAFJOIN/DOMAIN/%s"
38 #define SAFJOIN_TTL 3600
39
40 static char *saf_key(const char *domain)
41 {
42 char *keystr;
43
44 asprintf_strupper_m(&keystr, SAFKEY_FMT, domain);
45
46 return keystr;
47 }
48
49 static char *saf_join_key(const char *domain)
50 {
51 char *keystr;
52
53 asprintf_strupper_m(&keystr, SAFJOINKEY_FMT, domain);
54
55 return keystr;
56 }
57
58 /****************************************************************************
59 ****************************************************************************/
60
61 bool saf_store( const char *domain, const char *servername )
62 {
63 char *key;
64 time_t expire;
65 bool ret = False;
66
67 if ( !domain || !servername ) {
68 DEBUG(2,("saf_store: "
69 "Refusing to store empty domain or servername!\n"));
70 return False;
71 }
72
73 if ( (strlen(domain) == 0) || (strlen(servername) == 0) ) {
74 DEBUG(0,("saf_store: "
75 "refusing to store 0 length domain or servername!\n"));
76 return False;
77 }
78
79 key = saf_key( domain );
80 expire = time( NULL ) + lp_parm_int(-1, "saf","ttl", SAF_TTL);
81
82 DEBUG(10,("saf_store: domain = [%s], server = [%s], expire = [%u]\n",
83 domain, servername, (unsigned int)expire ));
84
85 ret = gencache_set( key, servername, expire );
86
87 SAFE_FREE( key );
88
89 return ret;
90 }
91
92 bool saf_join_store( const char *domain, const char *servername )
93 {
94 char *key;
95 time_t expire;
96 bool ret = False;
97
98 if ( !domain || !servername ) {
99 DEBUG(2,("saf_join_store: Refusing to store empty domain or servername!\n"));
100 return False;
101 }
102
103 if ( (strlen(domain) == 0) || (strlen(servername) == 0) ) {
104 DEBUG(0,("saf_join_store: refusing to store 0 length domain or servername!\n"));
105 return False;
106 }
107
108 key = saf_join_key( domain );
109 expire = time( NULL ) + lp_parm_int(-1, "saf","join ttl", SAFJOIN_TTL);
110
111 DEBUG(10,("saf_join_store: domain = [%s], server = [%s], expire = [%u]\n",
112 domain, servername, (unsigned int)expire ));
113
114 ret = gencache_set( key, servername, expire );
115
116 SAFE_FREE( key );
117
118 return ret;
119 }
120
121 bool saf_delete( const char *domain )
122 {
123 char *key;
124 bool ret = False;
125
126 if ( !domain ) {
127 DEBUG(2,("saf_delete: Refusing to delete empty domain\n"));
128 return False;
129 }
130
131 key = saf_join_key(domain);
132 ret = gencache_del(key);
133 SAFE_FREE(key);
134
135 if (ret) {
136 DEBUG(10,("saf_delete[join]: domain = [%s]\n", domain ));
137 }
138
139 key = saf_key(domain);
140 ret = gencache_del(key);
141 SAFE_FREE(key);
142
143 if (ret) {
144 DEBUG(10,("saf_delete: domain = [%s]\n", domain ));
145 }
146
147 return ret;
148 }
149
150 /****************************************************************************
151 ****************************************************************************/
152
153 char *saf_fetch( const char *domain )
154 {
155 char *server = NULL;
156 time_t timeout;
157 bool ret = False;
158 char *key = NULL;
159
160 if ( !domain || strlen(domain) == 0) {
161 DEBUG(2,("saf_fetch: Empty domain name!\n"));
162 return NULL;
163 }
164
165 key = saf_join_key( domain );
166
167 ret = gencache_get( key, &server, &timeout );
168
169 SAFE_FREE( key );
170
171 if ( ret ) {
172 DEBUG(5,("saf_fetch[join]: Returning \"%s\" for \"%s\" domain\n",
173 server, domain ));
174 return server;
175 }
176
177 key = saf_key( domain );
178
179 ret = gencache_get( key, &server, &timeout );
180
181 SAFE_FREE( key );
182
183 if ( !ret ) {
184 DEBUG(5,("saf_fetch: failed to find server for \"%s\" domain\n",
185 domain ));
186 } else {
187 DEBUG(5,("saf_fetch: Returning \"%s\" for \"%s\" domain\n",
188 server, domain ));
189 }
190
191 return server;
192 }
193
194 /****************************************************************************
195 Generate a random trn_id.
196 ****************************************************************************/
197
198 static int generate_trn_id(void)
199 {
200 uint16 id;
201
202 generate_random_buffer((uint8 *)&id, sizeof(id));
203
204 return id % (unsigned)0x7FFF;
205 }
206
207 /****************************************************************************
208 Parse a node status response into an array of structures.
209 ****************************************************************************/
210
211 static NODE_STATUS_STRUCT *parse_node_status(char *p,
212 int *num_names,
213 struct node_status_extra *extra)
214 {
215 NODE_STATUS_STRUCT *ret;
216 int i;
217
218 *num_names = CVAL(p,0);
219
220 if (*num_names == 0)
221 return NULL;
222
223 ret = SMB_MALLOC_ARRAY(NODE_STATUS_STRUCT,*num_names);
224 if (!ret)
225 return NULL;
226
227 p++;
228 for (i=0;i< *num_names;i++) {
229 StrnCpy(ret[i].name,p,15);
230 trim_char(ret[i].name,'\0',' ');
231 ret[i].type = CVAL(p,15);
232 ret[i].flags = p[16];
233 p += 18;
234 DEBUG(10, ("%s#%02x: flags = 0x%02x\n", ret[i].name,
235 ret[i].type, ret[i].flags));
236 }
237 /*
238 * Also, pick up the MAC address ...
239 */
240 if (extra) {
241 memcpy(&extra->mac_addr, p, 6); /* Fill in the mac addr */
242 }
243 return ret;
244 }
245
246
247 /****************************************************************************
248 Do a NBT node status query on an open socket and return an array of
249 structures holding the returned names or NULL if the query failed.
250 **************************************************************************/
251
252 NODE_STATUS_STRUCT *node_status_query(int fd,
253 struct nmb_name *name,
254 const struct sockaddr_storage *to_ss,
255 int *num_names,
256 struct node_status_extra *extra)
257 {
258 bool found=False;
259 int retries = 2;
260 int retry_time = 2000;
261 struct timeval tval;
262 struct packet_struct p;
263 struct packet_struct *p2;
264 struct nmb_packet *nmb = &p.packet.nmb;
265 NODE_STATUS_STRUCT *ret;
266
267 ZERO_STRUCT(p);
268
269 if (to_ss->ss_family != AF_INET) {
270 /* Can't do node status to IPv6 */
271 return NULL;
272 }
273 nmb->header.name_trn_id = generate_trn_id();
274 nmb->header.opcode = 0;
275 nmb->header.response = false;
276 nmb->header.nm_flags.bcast = false;
277 nmb->header.nm_flags.recursion_available = false;
278 nmb->header.nm_flags.recursion_desired = false;
279 nmb->header.nm_flags.trunc = false;
280 nmb->header.nm_flags.authoritative = false;
281 nmb->header.rcode = 0;
282 nmb->header.qdcount = 1;
283 nmb->header.ancount = 0;
284 nmb->header.nscount = 0;
285 nmb->header.arcount = 0;
286 nmb->question.question_name = *name;
287 nmb->question.question_type = 0x21;
288 nmb->question.question_class = 0x1;
289
290 p.ip = ((const struct sockaddr_in *)to_ss)->sin_addr;
291 p.port = NMB_PORT;
292 p.recv_fd = -1;
293 p.send_fd = fd;
294 p.timestamp = time(NULL);
295 p.packet_type = NMB_PACKET;
296
297 GetTimeOfDay(&tval);
298
299 if (!send_packet(&p))
300 return NULL;
301
302 retries--;
303
304 while (1) {
305 struct timeval tval2;
306 GetTimeOfDay(&tval2);
307 if (TvalDiff(&tval,&tval2) > retry_time) {
308 if (!retries)
309 break;
310 if (!found && !send_packet(&p))
311 return NULL;
312 GetTimeOfDay(&tval);
313 retries--;
314 }
315
316 if ((p2=receive_nmb_packet(fd,90,nmb->header.name_trn_id))) {
317 struct nmb_packet *nmb2 = &p2->packet.nmb;
318 debug_nmb_packet(p2);
319
320 if (nmb2->header.opcode != 0 ||
321 nmb2->header.nm_flags.bcast ||
322 nmb2->header.rcode ||
323 !nmb2->header.ancount ||
324 nmb2->answers->rr_type != 0x21) {
325 /* XXXX what do we do with this? could be a
326 redirect, but we'll discard it for the
327 moment */
328 free_packet(p2);
329 continue;
330 }
331
332 ret = parse_node_status(&nmb2->answers->rdata[0],
333 num_names, extra);
334 free_packet(p2);
335 return ret;
336 }
337 }
338
339 return NULL;
340 }
341
342 /****************************************************************************
343 Find the first type XX name in a node status reply - used for finding
344 a servers name given its IP. Return the matched name in *name.
345 **************************************************************************/
346
347 bool name_status_find(const char *q_name,
348 int q_type,
349 int type,
350 const struct sockaddr_storage *to_ss,
351 fstring name)
352 {
353 char addr[INET6_ADDRSTRLEN];
354 struct sockaddr_storage ss;
355 NODE_STATUS_STRUCT *status = NULL;
356 struct nmb_name nname;
357 int count, i;
358 int sock;
359 bool result = false;
360
361 if (lp_disable_netbios()) {
362 DEBUG(5,("name_status_find(%s#%02x): netbios is disabled\n",
363 q_name, q_type));
364 return False;
365 }
366
367 print_sockaddr(addr, sizeof(addr), to_ss);
368
369 DEBUG(10, ("name_status_find: looking up %s#%02x at %s\n", q_name,
370 q_type, addr));
371
372 /* Check the cache first. */
373
374 if (namecache_status_fetch(q_name, q_type, type, to_ss, name)) {
375 return True;
376 }
377
378 if (to_ss->ss_family != AF_INET) {
379 /* Can't do node status to IPv6 */
380 return false;
381 }
382
383 if (!interpret_string_addr(&ss, lp_socket_address(),
384 AI_NUMERICHOST|AI_PASSIVE)) {
385 zero_sockaddr(&ss);
386 }
387
388 sock = open_socket_in(SOCK_DGRAM, 0, 3, &ss, True);
389 if (sock == -1)
390 goto done;
391
392 /* W2K PDC's seem not to respond to '*'#0. JRA */
393 make_nmb_name(&nname, q_name, q_type);
394 status = node_status_query(sock, &nname, to_ss, &count, NULL);
395 close(sock);
396 if (!status)
397 goto done;
398
399 for (i=0;i<count;i++) {
400 /* Find first one of the requested type that's not a GROUP. */
401 if (status[i].type == type && ! (status[i].flags & 0x80))
402 break;
403 }
404 if (i == count)
405 goto done;
406
407 pull_ascii_nstring(name, sizeof(fstring), status[i].name);
408
409 /* Store the result in the cache. */
410 /* but don't store an entry for 0x1c names here. Here we have
411 a single host and DOMAIN<0x1c> names should be a list of hosts */
412
413 if ( q_type != 0x1c ) {
414 namecache_status_store(q_name, q_type, type, to_ss, name);
415 }
416
417 result = true;
418
419 done:
420 SAFE_FREE(status);
421
422 DEBUG(10, ("name_status_find: name %sfound", result ? "" : "not "));
423
424 if (result)
425 DEBUGADD(10, (", name %s ip address is %s", name, addr));
426
427 DEBUG(10, ("\n"));
428
429 return result;
430 }
431
432 /*
433 comparison function used by sort_addr_list
434 */
435
436 static int addr_compare(const struct sockaddr *ss1,
437 const struct sockaddr *ss2)
438 {
439 int max_bits1=0, max_bits2=0;
440 int num_interfaces = iface_count();
441 int i;
442
443 /* Sort IPv4 addresses first. */
444 if (ss1->sa_family != ss2->sa_family) {
445 if (ss2->sa_family == AF_INET) {
446 return 1;
447 } else {
448 return -1;
449 }
450 }
451
452 /* Here we know both addresses are of the same
453 * family. */
454
455 for (i=0;i<num_interfaces;i++) {
456 const struct sockaddr_storage *pss = iface_n_bcast(i);
457 unsigned char *p_ss1 = NULL;
458 unsigned char *p_ss2 = NULL;
459 unsigned char *p_if = NULL;
460 size_t len = 0;
461 int bits1, bits2;
462
463 if (pss->ss_family != ss1->sa_family) {
464 /* Ignore interfaces of the wrong type. */
465 continue;
466 }
467 if (pss->ss_family == AF_INET) {
468 p_if = (unsigned char *)
469 &((const struct sockaddr_in *)pss)->sin_addr;
470 p_ss1 = (unsigned char *)
471 &((const struct sockaddr_in *)ss1)->sin_addr;
472 p_ss2 = (unsigned char *)
473 &((const struct sockaddr_in *)ss2)->sin_addr;
474 len = 4;
475 }
476 #if defined(HAVE_IPV6)
477 if (pss->ss_family == AF_INET6) {
478 p_if = (unsigned char *)
479 &((const struct sockaddr_in6 *)pss)->sin6_addr;
480 p_ss1 = (unsigned char *)
481 &((const struct sockaddr_in6 *)ss1)->sin6_addr;
482 p_ss2 = (unsigned char *)
483 &((const struct sockaddr_in6 *)ss2)->sin6_addr;
484 len = 16;
485 }
486 #endif
487 if (!p_ss1 || !p_ss2 || !p_if || len == 0) {
488 continue;
489 }
490 bits1 = matching_len_bits(p_ss1, p_if, len);
491 bits2 = matching_len_bits(p_ss2, p_if, len);
492 max_bits1 = MAX(bits1, max_bits1);
493 max_bits2 = MAX(bits2, max_bits2);
494 }
495
496 /* Bias towards directly reachable IPs */
497 if (iface_local(ss1)) {
498 if (ss1->sa_family == AF_INET) {
499 max_bits1 += 32;
500 } else {
501 max_bits1 += 128;
502 }
503 }
504 if (iface_local(ss2)) {
505 if (ss2->sa_family == AF_INET) {
506 max_bits2 += 32;
507 } else {
508 max_bits2 += 128;
509 }
510 }
511 return max_bits2 - max_bits1;
512 }
513
514 /*******************************************************************
515 compare 2 ldap IPs by nearness to our interfaces - used in qsort
516 *******************************************************************/
517
518 int ip_service_compare(struct ip_service *ss1, struct ip_service *ss2)
519 {
520 int result;
521
522 if ((result = addr_compare((struct sockaddr *)&ss1->ss, (struct sockaddr *)&ss2->ss)) != 0) {
523 return result;
524 }
525
526 if (ss1->port > ss2->port) {
527 return 1;
528 }
529
530 if (ss1->port < ss2->port) {
531 return -1;
532 }
533
534 return 0;
535 }
536
537 /*
538 sort an IP list so that names that are close to one of our interfaces
539 are at the top. This prevents the problem where a WINS server returns an IP
540 that is not reachable from our subnet as the first match
541 */
542
543 static void sort_addr_list(struct sockaddr_storage *sslist, int count)
544 {
545 if (count <= 1) {
546 return;
547 }
548
549 qsort(sslist, count, sizeof(struct sockaddr_storage),
550 QSORT_CAST addr_compare);
551 }
552
553 static void sort_service_list(struct ip_service *servlist, int count)
554 {
555 if (count <= 1) {
556 return;
557 }
558
559 qsort(servlist, count, sizeof(struct ip_service),
560 QSORT_CAST ip_service_compare);
561 }
562
563 /**********************************************************************
564 Remove any duplicate address/port pairs in the list
565 *********************************************************************/
566
567 static int remove_duplicate_addrs2(struct ip_service *iplist, int count )
568 {
569 int i, j;
570
571 DEBUG(10,("remove_duplicate_addrs2: "
572 "looking for duplicate address/port pairs\n"));
573
574 /* One loop to set duplicates to a zero addr. */
575 for ( i=0; i<count; i++ ) {
576 if ( is_zero_addr((struct sockaddr *)&iplist[i].ss)) {
577 continue;
578 }
579
580 for ( j=i+1; j<count; j++ ) {
581 if (sockaddr_equal((struct sockaddr *)&iplist[i].ss, (struct sockaddr *)&iplist[j].ss) &&
582 iplist[i].port == iplist[j].port) {
583 zero_sockaddr(&iplist[j].ss);
584 }
585 }
586 }
587
588 /* Now remove any addresses set to zero above. */
589 for (i = 0; i < count; i++) {
590 while (i < count &&
591 is_zero_addr((struct sockaddr *)&iplist[i].ss)) {
592 if (count-i-1>0) {
593 memmove(&iplist[i],
594 &iplist[i+1],
595 (count-i-1)*sizeof(struct ip_service));
596 }
597 count--;
598 }
599 }
600
601 return count;
602 }
603
604 static bool prioritize_ipv4_list(struct ip_service *iplist, int count)
605 {
606 TALLOC_CTX *frame = talloc_stackframe();
607 struct ip_service *iplist_new = TALLOC_ARRAY(frame, struct ip_service, count);
608 int i, j;
609
610 if (iplist_new == NULL) {
611 TALLOC_FREE(frame);
612 return false;
613 }
614
615 j = 0;
616
617 /* Copy IPv4 first. */
618 for (i = 0; i < count; i++) {
619 if (iplist[i].ss.ss_family == AF_INET) {
620 iplist_new[j++] = iplist[i];
621 }
622 }
623
624 /* Copy IPv6. */
625 for (i = 0; i < count; i++) {
626 if (iplist[i].ss.ss_family != AF_INET) {
627 iplist_new[j++] = iplist[i];
628 }
629 }
630
631 memcpy(iplist, iplist_new, sizeof(struct ip_service)*count);
632 TALLOC_FREE(frame);
633 return true;
634 }
635
636 /****************************************************************************
637 Do a netbios name query to find someones IP.
638 Returns an array of IP addresses or NULL if none.
639 *count will be set to the number of addresses returned.
640 *timed_out is set if we failed by timing out
641 ****************************************************************************/
642
643 struct sockaddr_storage *name_query(int fd,
644 const char *name,
645 int name_type,
646 bool bcast,
647 bool recurse,
648 const struct sockaddr_storage *to_ss,
649 int *count,
650 int *flags,
651 bool *timed_out)
652 {
653 bool found=false;
654 int i, retries = 3;
655 int retry_time = bcast?250:2000;
656 struct timeval tval;
657 struct packet_struct p;
658 struct packet_struct *p2;
659 struct nmb_packet *nmb = &p.packet.nmb;
660 struct sockaddr_storage *ss_list = NULL;
661
662 if (lp_disable_netbios()) {
663 DEBUG(5,("name_query(%s#%02x): netbios is disabled\n",
664 name, name_type));
665 return NULL;
666 }
667
668 if (to_ss->ss_family != AF_INET) {
669 return NULL;
670 }
671
672 if (timed_out) {
673 *timed_out = false;
674 }
675
676 memset((char *)&p,'\0',sizeof(p));
677 (*count) = 0;
678 (*flags) = 0;
679
680 nmb->header.name_trn_id = generate_trn_id();
681 nmb->header.opcode = 0;
682 nmb->header.response = false;
683 nmb->header.nm_flags.bcast = bcast;
684 nmb->header.nm_flags.recursion_available = false;
685 nmb->header.nm_flags.recursion_desired = recurse;
686 nmb->header.nm_flags.trunc = false;
687 nmb->header.nm_flags.authoritative = false;
688 nmb->header.rcode = 0;
689 nmb->header.qdcount = 1;
690 nmb->header.ancount = 0;
691 nmb->header.nscount = 0;
692 nmb->header.arcount = 0;
693
694 make_nmb_name(&nmb->question.question_name,name,name_type);
695
696 nmb->question.question_type = 0x20;
697 nmb->question.question_class = 0x1;
698
699 p.ip = ((struct sockaddr_in *)to_ss)->sin_addr;
700 p.port = NMB_PORT;
701 p.recv_fd = -1;
702 p.send_fd = fd;
703 p.timestamp = time(NULL);
704 p.packet_type = NMB_PACKET;
705
706 GetTimeOfDay(&tval);
707
708 if (!send_packet(&p))
709 return NULL;
710
711 retries--;
712
713 while (1) {
714 struct timeval tval2;
715
716 GetTimeOfDay(&tval2);
717 if (TvalDiff(&tval,&tval2) > retry_time) {
718 if (!retries)
719 break;
720 if (!found && !send_packet(&p))
721 return NULL;
722 GetTimeOfDay(&tval);
723 retries--;
724 }
725
726 if ((p2=receive_nmb_packet(fd,90,nmb->header.name_trn_id))) {
727 struct nmb_packet *nmb2 = &p2->packet.nmb;
728 debug_nmb_packet(p2);
729
730 /* If we get a Negative Name Query Response from a WINS
731 * server, we should report it and give up.
732 */
733 if( 0 == nmb2->header.opcode /* A query response */
734 && !(bcast) /* from a WINS server */
735 && nmb2->header.rcode /* Error returned */
736 ) {
737
738 if( DEBUGLVL( 3 ) ) {
739 /* Only executed if DEBUGLEVEL >= 3 */
740 dbgtext( "Negative name query "
741 "response, rcode 0x%02x: ",
742 nmb2->header.rcode );
743 switch( nmb2->header.rcode ) {
744 case 0x01:
745 dbgtext( "Request "
746 "was invalidly formatted.\n" );
747 break;
748 case 0x02:
749 dbgtext( "Problem with NBNS, "
750 "cannot process name.\n");
751 break;
752 case 0x03:
753 dbgtext( "The name requested "
754 "does not exist.\n" );
755 break;
756 case 0x04:
757 dbgtext( "Unsupported request "
758 "error.\n" );
759 break;
760 case 0x05:
761 dbgtext( "Query refused "
762 "error.\n" );
763 break;
764 default:
765 dbgtext( "Unrecognized error "
766 "code.\n" );
767 break;
768 }
769 }
770 free_packet(p2);
771 return( NULL );
772 }
773
774 if (nmb2->header.opcode != 0 ||
775 nmb2->header.nm_flags.bcast ||
776 nmb2->header.rcode ||
777 !nmb2->header.ancount) {
778 /*
779 * XXXX what do we do with this? Could be a
780 * redirect, but we'll discard it for the
781 * moment.
782 */
783 free_packet(p2);
784 continue;
785 }
786
787 ss_list = SMB_REALLOC_ARRAY(ss_list,
788 struct sockaddr_storage,
789 (*count) +
790 nmb2->answers->rdlength/6);
791
792 if (!ss_list) {
793 DEBUG(0,("name_query: Realloc failed.\n"));
794 free_packet(p2);
795 return NULL;
796 }
797
798 DEBUG(2,("Got a positive name query response "
799 "from %s ( ",
800 inet_ntoa(p2->ip)));
801
802 for (i=0;i<nmb2->answers->rdlength/6;i++) {
803 struct in_addr ip;
804 putip((char *)&ip,&nmb2->answers->rdata[2+i*6]);
805 in_addr_to_sockaddr_storage(&ss_list[(*count)],
806 ip);
807 DEBUGADD(2,("%s ",inet_ntoa(ip)));
808 (*count)++;
809 }
810 DEBUGADD(2,(")\n"));
811
812 found=true;
813 retries=0;
814 /* We add the flags back ... */
815 if (nmb2->header.response)
816 (*flags) |= NM_FLAGS_RS;
817 if (nmb2->header.nm_flags.authoritative)
818 (*flags) |= NM_FLAGS_AA;
819 if (nmb2->header.nm_flags.trunc)
820 (*flags) |= NM_FLAGS_TC;
821 if (nmb2->header.nm_flags.recursion_desired)
822 (*flags) |= NM_FLAGS_RD;
823 if (nmb2->header.nm_flags.recursion_available)
824 (*flags) |= NM_FLAGS_RA;
825 if (nmb2->header.nm_flags.bcast)
826 (*flags) |= NM_FLAGS_B;
827 free_packet(p2);
828 /*
829 * If we're doing a unicast lookup we only
830 * expect one reply. Don't wait the full 2
831 * seconds if we got one. JRA.
832 */
833 if(!bcast && found)
834 break;
835 }
836 }
837
838 /* only set timed_out if we didn't fund what we where looking for*/
839
840 if ( !found && timed_out ) {
841 *timed_out = true;
842 }
843
844 /* sort the ip list so we choose close servers first if possible */
845 sort_addr_list(ss_list, *count);
846
847 return ss_list;
848 }
849
850 /********************************************************
851 Convert an array if struct sockaddr_storage to struct ip_service
852 return false on failure. Port is set to PORT_NONE;
853 pcount is [in/out] - it is the length of ss_list on input,
854 and the length of return_iplist on output as we remove any
855 zero addresses from ss_list.
856 *********************************************************/
857
858 static bool convert_ss2service(struct ip_service **return_iplist,
859 const struct sockaddr_storage *ss_list,
860 int *pcount)
861 {
862 int i;
863 int orig_count = *pcount;
864 int real_count = 0;
865
866 if (orig_count==0 || !ss_list )
867 return False;
868
869 /* Filter out zero addrs. */
870 for ( i=0; i<orig_count; i++ ) {
871 if (is_zero_addr((struct sockaddr *)&ss_list[i])) {
872 continue;
873 }
874 real_count++;
875 }
876 if (real_count==0) {
877 return false;
878 }
879
880 /* copy the ip address; port will be PORT_NONE */
881 if ((*return_iplist = SMB_MALLOC_ARRAY(struct ip_service, real_count)) ==
882 NULL) {
883 DEBUG(0,("convert_ip2service: malloc failed "
884 "for %d enetries!\n", real_count ));
885 return False;
886 }
887
888 for ( i=0, real_count = 0; i<orig_count; i++ ) {
889 if (is_zero_addr((struct sockaddr *)&ss_list[i])) {
890 continue;
891 }
892 (*return_iplist)[real_count].ss = ss_list[i];
893 (*return_iplist)[real_count].port = PORT_NONE;
894 real_count++;
895 }
896
897 *pcount = real_count;
898 return true;
899 }
900
901 /********************************************************
902 Resolve via "bcast" method.
903 *********************************************************/
904
905 NTSTATUS name_resolve_bcast(const char *name,
906 int name_type,
907 struct ip_service **return_iplist,
908 int *return_count)
909 {
910 int sock, i;
911 int num_interfaces = iface_count();
912 struct sockaddr_storage *ss_list;
913 struct sockaddr_storage ss;
914 NTSTATUS status;
915
916 if (lp_disable_netbios()) {
917 DEBUG(5,("name_resolve_bcast(%s#%02x): netbios is disabled\n",
918 name, name_type));
919 return NT_STATUS_INVALID_PARAMETER;
920 }
921
922 *return_iplist = NULL;
923 *return_count = 0;
924
925 /*
926 * "bcast" means do a broadcast lookup on all the local interfaces.
927 */
928
929 DEBUG(3,("name_resolve_bcast: Attempting broadcast lookup "
930 "for name %s<0x%x>\n", name, name_type));
931
932 if (!interpret_string_addr(&ss, lp_socket_address(),
933 AI_NUMERICHOST|AI_PASSIVE)) {
934 zero_sockaddr(&ss);
935 }
936
937 sock = open_socket_in( SOCK_DGRAM, 0, 3, &ss, true );
938 if (sock == -1) {
939 return NT_STATUS_UNSUCCESSFUL;
940 }
941
942 set_socket_options(sock,"SO_BROADCAST");
943 /*
944 * Lookup the name on all the interfaces, return on
945 * the first successful match.
946 */
947 for( i = num_interfaces-1; i >= 0; i--) {
948 const struct sockaddr_storage *pss = iface_n_bcast(i);
949 int flags;
950
951 /* Done this way to fix compiler error on IRIX 5.x */
952 if (!pss) {
953 continue;
954 }
955 ss_list = name_query(sock, name, name_type, true,
956 true, pss, return_count, &flags, NULL);
957 if (ss_list) {
958 goto success;
959 }
960 }
961
962 /* failed - no response */
963
964 close(sock);
965 return NT_STATUS_UNSUCCESSFUL;
966
967 success:
968
969 status = NT_STATUS_OK;
970 if (!convert_ss2service(return_iplist, ss_list, return_count) )
971 status = NT_STATUS_INVALID_PARAMETER;
972
973 SAFE_FREE(ss_list);
974 close(sock);
975 return status;
976 }
977
978 /********************************************************
979 Resolve via "wins" method.
980 *********************************************************/
981
982 NTSTATUS resolve_wins(const char *name,
983 int name_type,
984 struct ip_service **return_iplist,
985 int *return_count)
986 {
987 int sock, t, i;
988 char **wins_tags;
989 struct sockaddr_storage src_ss, *ss_list = NULL;
990 struct in_addr src_ip;
991 NTSTATUS status;
992
993 if (lp_disable_netbios()) {
994 DEBUG(5,("resolve_wins(%s#%02x): netbios is disabled\n",
995 name, name_type));
996 return NT_STATUS_INVALID_PARAMETER;
997 }
998
999 *return_iplist = NULL;
1000 *return_count = 0;
1001
1002 DEBUG(3,("resolve_wins: Attempting wins lookup for name %s<0x%x>\n",
1003 name, name_type));
1004
1005 if (wins_srv_count() < 1) {
1006 DEBUG(3,("resolve_wins: WINS server resolution selected "
1007 "and no WINS servers listed.\n"));
1008 return NT_STATUS_INVALID_PARAMETER;
1009 }
1010
1011 /* we try a lookup on each of the WINS tags in turn */
1012 wins_tags = wins_srv_tags();
1013
1014 if (!wins_tags) {
1015 /* huh? no tags?? give up in disgust */
1016 return NT_STATUS_INVALID_PARAMETER;
1017 }
1018
1019 /* the address we will be sending from */
1020 if (!interpret_string_addr(&src_ss, lp_socket_address(),
1021 AI_NUMERICHOST|AI_PASSIVE)) {
1022 zero_sockaddr(&src_ss);
1023 }
1024
1025 if (src_ss.ss_family != AF_INET) {
1026 char addr[INET6_ADDRSTRLEN];
1027 print_sockaddr(addr, sizeof(addr), &src_ss);
1028 DEBUG(3,("resolve_wins: cannot receive WINS replies "
1029 "on IPv6 address %s\n",
1030 addr));
1031 wins_srv_tags_free(wins_tags);
1032 return NT_STATUS_INVALID_PARAMETER;
1033 }
1034
1035 src_ip = ((struct sockaddr_in *)&src_ss)->sin_addr;
1036
1037 /* in the worst case we will try every wins server with every
1038 tag! */
1039 for (t=0; wins_tags && wins_tags[t]; t++) {
1040 int srv_count = wins_srv_count_tag(wins_tags[t]);
1041 for (i=0; i<srv_count; i++) {
1042 struct sockaddr_storage wins_ss;
1043 struct in_addr wins_ip;
1044 int flags;
1045 bool timed_out;
1046
1047 wins_ip = wins_srv_ip_tag(wins_tags[t], src_ip);
1048
1049 if (global_in_nmbd && ismyip_v4(wins_ip)) {
1050 /* yikes! we'll loop forever */
1051 continue;
1052 }
1053
1054 /* skip any that have been unresponsive lately */
1055 if (wins_srv_is_dead(wins_ip, src_ip)) {
1056 continue;
1057 }
1058
1059 DEBUG(3,("resolve_wins: using WINS server %s "
1060 "and tag '%s'\n",
1061 inet_ntoa(wins_ip), wins_tags[t]));
1062
1063 sock = open_socket_in(SOCK_DGRAM, 0, 3, &src_ss, true);
1064 if (sock == -1) {
1065 continue;
1066 }
1067
1068 in_addr_to_sockaddr_storage(&wins_ss, wins_ip);
1069 ss_list = name_query(sock,
1070 name,
1071 name_type,
1072 false,
1073 true,
1074 &wins_ss,
1075 return_count,
1076 &flags,
1077 &timed_out);
1078
1079 /* exit loop if we got a list of addresses */
1080
1081 if (ss_list)
1082 goto success;
1083
1084 close(sock);
1085
1086 if (timed_out) {
1087 /* Timed out wating for WINS server to respond.
1088 * Mark it dead. */
1089 wins_srv_died(wins_ip, src_ip);
1090 } else {
1091 /* The name definately isn't in this
1092 group of WINS servers.
1093 goto the next group */
1094 break;
1095 }
1096 }
1097 }
1098
1099 wins_srv_tags_free(wins_tags);
1100 return NT_STATUS_NO_LOGON_SERVERS;
1101
1102 success:
1103
1104 status = NT_STATUS_OK;
1105 if (!convert_ss2service(return_iplist, ss_list, return_count))
1106 status = NT_STATUS_INVALID_PARAMETER;
1107
1108 SAFE_FREE(ss_list);
1109 wins_srv_tags_free(wins_tags);
1110 close(sock);
1111
1112 return status;
1113 }
1114
1115 /********************************************************
1116 Resolve via "lmhosts" method.
1117 *********************************************************/
1118
1119 static NTSTATUS resolve_lmhosts(const char *name, int name_type,
1120 struct ip_service **return_iplist,
1121 int *return_count)
1122 {
1123 /*
1124 * "lmhosts" means parse the local lmhosts file.
1125 */
1126
1127 XFILE *fp;
1128 char *lmhost_name = NULL;
1129 int name_type2;
1130 struct sockaddr_storage return_ss;
1131 NTSTATUS status = NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
1132 TALLOC_CTX *ctx = NULL;
1133
1134 *return_iplist = NULL;
1135 *return_count = 0;
1136
1137 DEBUG(3,("resolve_lmhosts: "
1138 "Attempting lmhosts lookup for name %s<0x%x>\n",
1139 name, name_type));
1140
1141 fp = startlmhosts(get_dyn_LMHOSTSFILE());
1142
1143 if ( fp == NULL )
1144 return NT_STATUS_NO_SUCH_FILE;
1145
1146 ctx = talloc_init("resolve_lmhosts");
1147 if (!ctx) {
1148 endlmhosts(fp);
1149 return NT_STATUS_NO_MEMORY;
1150 }
1151
1152 while (getlmhostsent(ctx, fp, &lmhost_name, &name_type2, &return_ss)) {
1153
1154 if (!strequal(name, lmhost_name)) {
1155 TALLOC_FREE(lmhost_name);
1156 continue;
1157 }
1158
1159 if ((name_type2 != -1) && (name_type != name_type2)) {
1160 TALLOC_FREE(lmhost_name);
1161 continue;
1162 }
1163
1164 *return_iplist = SMB_REALLOC_ARRAY((*return_iplist),
1165 struct ip_service,
1166 (*return_count)+1);
1167
1168 if ((*return_iplist) == NULL) {
1169 TALLOC_FREE(ctx);
1170 endlmhosts(fp);
1171 DEBUG(3,("resolve_lmhosts: malloc fail !\n"));
1172 return NT_STATUS_NO_MEMORY;
1173 }
1174
1175 (*return_iplist)[*return_count].ss = return_ss;
1176 (*return_iplist)[*return_count].port = PORT_NONE;
1177 *return_count += 1;
1178
1179 /* we found something */
1180 status = NT_STATUS_OK;
1181
1182 /* Multiple names only for DC lookup */
1183 if (name_type != 0x1c)
1184 break;
1185 }
1186
1187 TALLOC_FREE(ctx);
1188 endlmhosts(fp);
1189 return status;
1190 }
1191
1192
1193 /********************************************************
1194 Resolve via "hosts" method.
1195 *********************************************************/
1196
1197 static NTSTATUS resolve_hosts(const char *name, int name_type,
1198 struct ip_service **return_iplist,
1199 int *return_count)
1200 {
1201 /*
1202 * "host" means do a localhost, or dns lookup.
1203 */
1204 struct addrinfo hints;
1205 struct addrinfo *ailist = NULL;
1206 struct addrinfo *res = NULL;
1207 int ret = -1;
1208 int i = 0;
1209
1210 if ( name_type != 0x20 && name_type != 0x0) {
1211 DEBUG(5, ("resolve_hosts: not appropriate "
1212 "for name type <0x%x>\n",
1213 name_type));
1214 return NT_STATUS_INVALID_PARAMETER;
1215 }
1216
1217 *return_iplist = NULL;
1218 *return_count = 0;
1219
1220 DEBUG(3,("resolve_hosts: Attempting host lookup for name %s<0x%x>\n",
1221 name, name_type));
1222
1223 ZERO_STRUCT(hints);
1224 /* By default make sure it supports TCP. */
1225 hints.ai_socktype = SOCK_STREAM;
1226 hints.ai_flags = AI_ADDRCONFIG;
1227
1228 #if !defined(HAVE_IPV6)
1229 /* Unless we have IPv6, we really only want IPv4 addresses back. */
1230 hints.ai_family = AF_INET;
1231 #endif
1232
1233 ret = getaddrinfo(name,
1234 NULL,
1235 &hints,
1236 &ailist);
1237 if (ret) {
1238 DEBUG(3,("resolve_hosts: getaddrinfo failed for name %s [%s]\n",
1239 name,
1240 gai_strerror(ret) ));
1241 }
1242
1243 for (res = ailist; res; res = res->ai_next) {
1244 struct sockaddr_storage ss;
1245
1246 if (!res->ai_addr || res->ai_addrlen == 0) {
1247 continue;
1248 }
1249
1250 ZERO_STRUCT(ss);
1251 memcpy(&ss, res->ai_addr, res->ai_addrlen);
1252
1253 *return_count += 1;
1254
1255 *return_iplist = SMB_REALLOC_ARRAY(*return_iplist,
1256 struct ip_service,
1257 *return_count);
1258 if (!*return_iplist) {
1259 DEBUG(3,("resolve_hosts: malloc fail !\n"));
1260 freeaddrinfo(ailist);
1261 return NT_STATUS_NO_MEMORY;
1262 }
1263 (*return_iplist)[i].ss = ss;
1264 (*return_iplist)[i].port = PORT_NONE;
1265 i++;
1266 }
1267 if (ailist) {
1268 freeaddrinfo(ailist);
1269 }
1270 if (*return_count) {
1271 return NT_STATUS_OK;
1272 }
1273 return NT_STATUS_UNSUCCESSFUL;
1274 }
1275
1276 /********************************************************
1277 Resolve via "ADS" method.
1278 *********************************************************/
1279
1280 static NTSTATUS resolve_ads(const char *name,
1281 int name_type,
1282 const char *sitename,
1283 struct ip_service **return_iplist,
1284 int *return_count)
1285 {
1286 int i, j;
1287 NTSTATUS status;
1288 TALLOC_CTX *ctx;
1289 struct dns_rr_srv *dcs = NULL;
1290 int numdcs = 0;
1291 int numaddrs = 0;
1292
1293 if ((name_type != 0x1c) && (name_type != KDC_NAME_TYPE) &&
1294 (name_type != 0x1b)) {
1295 return NT_STATUS_INVALID_PARAMETER;
1296 }
1297
1298 if ( (ctx = talloc_init("resolve_ads")) == NULL ) {
1299 DEBUG(0,("resolve_ads: talloc_init() failed!\n"));
1300 return NT_STATUS_NO_MEMORY;
1301 }
1302
1303 /* The DNS code needs fixing to find IPv6 addresses... JRA. */
1304
1305 switch (name_type) {
1306 case 0x1b:
1307 DEBUG(5,("resolve_ads: Attempting to resolve "
1308 "PDC for %s using DNS\n", name));
1309 status = ads_dns_query_pdc(ctx, name, &dcs, &numdcs);
1310 break;
1311
1312 case 0x1c:
1313 DEBUG(5,("resolve_ads: Attempting to resolve "
1314 "DCs for %s using DNS\n", name));
1315 status = ads_dns_query_dcs(ctx, name, sitename, &dcs,
1316 &numdcs);
1317 break;
1318 case KDC_NAME_TYPE:
1319 DEBUG(5,("resolve_ads: Attempting to resolve "
1320 "KDCs for %s using DNS\n", name));
1321 status = ads_dns_query_kdcs(ctx, name, sitename, &dcs,
1322 &numdcs);
1323 break;
1324 default:
1325 status = NT_STATUS_INVALID_PARAMETER;
1326 break;
1327 }
1328
1329 if ( !NT_STATUS_IS_OK( status ) ) {
1330 talloc_destroy(ctx);
1331 return status;
1332 }
1333
1334 for (i=0;i<numdcs;i++) {
1335 numaddrs += MAX(dcs[i].num_ips,1);
1336 }
1337
1338 if ((*return_iplist = SMB_MALLOC_ARRAY(struct ip_service, numaddrs)) ==
1339 NULL ) {
1340 DEBUG(0,("resolve_ads: malloc failed for %d entries\n",
1341 numaddrs ));
1342 talloc_destroy(ctx);
1343 return NT_STATUS_NO_MEMORY;
1344 }
1345
1346 /* now unroll the list of IP addresses */
1347
1348 *return_count = 0;
1349 i = 0;
1350 j = 0;
1351 while ( i < numdcs && (*return_count<numaddrs) ) {
1352 struct ip_service *r = &(*return_iplist)[*return_count];
1353
1354 r->port = dcs[i].port;
1355
1356 /* If we don't have an IP list for a name, lookup it up */
1357
1358 if (!dcs[i].ss_s) {
1359 interpret_string_addr(&r->ss, dcs[i].hostname, 0);
1360 i++;
1361 j = 0;
1362 } else {
1363 /* use the IP addresses from the SRV sresponse */
1364
1365 if ( j >= dcs[i].num_ips ) {
1366 i++;
1367 j = 0;
1368 continue;
1369 }
1370
1371 r->ss = dcs[i].ss_s[j];
1372 j++;
1373 }
1374
1375 /* make sure it is a valid IP. I considered checking the
1376 * negative connection cache, but this is the wrong place
1377 * for it. Maybe only as a hack. After think about it, if
1378 * all of the IP addresses returned from DNS are dead, what
1379 * hope does a netbios name lookup have ? The standard reason
1380 * for falling back to netbios lookups is that our DNS server
1381 * doesn't know anything about the DC's -- jerry */
1382
1383 if (!is_zero_addr((struct sockaddr *)&r->ss)) {
1384 (*return_count)++;
1385 }
1386 }
1387
1388 talloc_destroy(ctx);
1389 return NT_STATUS_OK;
1390 }
1391
1392 /*******************************************************************
1393 Internal interface to resolve a name into an IP address.
1394 Use this function if the string is either an IP address, DNS
1395 or host name or NetBIOS name. This uses the name switch in the
1396 smb.conf to determine the order of name resolution.
1397
1398 Added support for ip addr/port to support ADS ldap servers.
1399 the only place we currently care about the port is in the
1400 resolve_hosts() when looking up DC's via SRV RR entries in DNS
1401 **********************************************************************/
1402
1403 NTSTATUS internal_resolve_name(const char *name,
1404 int name_type,
1405 const char *sitename,
1406 struct ip_service **return_iplist,
1407 int *return_count,
1408 const char *resolve_order)
1409 {
1410 char *tok;
1411 const char *ptr;
1412 NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
1413 int i;
1414 TALLOC_CTX *frame = NULL;
1415
1416 *return_iplist = NULL;
1417 *return_count = 0;
1418
1419 DEBUG(10, ("internal_resolve_name: looking up %s#%x (sitename %s)\n",
1420 name, name_type, sitename ? sitename : "(null)"));
1421
1422 if (is_ipaddress(name)) {
1423 if ((*return_iplist = SMB_MALLOC_P(struct ip_service)) ==
1424 NULL) {
1425 DEBUG(0,("internal_resolve_name: malloc fail !\n"));
1426 return NT_STATUS_NO_MEMORY;
1427 }
1428
1429 /* ignore the port here */
1430 (*return_iplist)->port = PORT_NONE;
1431
1432 /* if it's in the form of an IP address then get the lib to interpret it */
1433 if (!interpret_string_addr(&(*return_iplist)->ss,
1434 name, AI_NUMERICHOST)) {
1435 DEBUG(1,("internal_resolve_name: interpret_string_addr "
1436 "failed on %s\n",
1437 name));
1438 SAFE_FREE(*return_iplist);
1439 return NT_STATUS_INVALID_PARAMETER;
1440 }
1441 *return_count = 1;
1442 return NT_STATUS_OK;
1443 }
1444
1445 /* Check name cache */
1446
1447 if (namecache_fetch(name, name_type, return_iplist, return_count)) {
1448 /* This could be a negative response */
1449 if (*return_count > 0) {
1450 return NT_STATUS_OK;
1451 } else {
1452 return NT_STATUS_UNSUCCESSFUL;
1453 }
1454 }
1455
1456 /* set the name resolution order */
1457
1458 if (strcmp( resolve_order, "NULL") == 0) {
1459 DEBUG(8,("internal_resolve_name: all lookups disabled\n"));
1460 return NT_STATUS_INVALID_PARAMETER;
1461 }
1462
1463 if (!resolve_order[0]) {
1464 ptr = "host";
1465 } else {
1466 ptr = resolve_order;
1467 }
1468
1469 /* iterate through the name resolution backends */
1470
1471 frame = talloc_stackframe();
1472 while (next_token_talloc(frame, &ptr, &tok, LIST_SEP)) {
1473 if((strequal(tok, "host") || strequal(tok, "hosts"))) {
1474 status = resolve_hosts(name, name_type, return_iplist,
1475 return_count);
1476 if (NT_STATUS_IS_OK(status)) {
1477 goto done;
1478 }
1479 } else if(strequal( tok, "kdc")) {
1480 /* deal with KDC_NAME_TYPE names here.
1481 * This will result in a SRV record lookup */
1482 status = resolve_ads(name, KDC_NAME_TYPE, sitename,
1483 return_iplist, return_count);
1484 if (NT_STATUS_IS_OK(status)) {
1485 /* Ensure we don't namecache
1486 * this with the KDC port. */
1487 name_type = KDC_NAME_TYPE;
1488 goto done;
1489 }
1490 } else if(strequal( tok, "ads")) {
1491 /* deal with 0x1c and 0x1b names here.
1492 * This will result in a SRV record lookup */
1493 status = resolve_ads(name, name_type, sitename,
1494 return_iplist, return_count);
1495 if (NT_STATUS_IS_OK(status)) {
1496 goto done;
1497 }
1498 } else if(strequal( tok, "lmhosts")) {
1499 status = resolve_lmhosts(name, name_type,
1500 return_iplist, return_count);
1501 if (NT_STATUS_IS_OK(status)) {
1502 goto done;
1503 }
1504 } else if(strequal( tok, "wins")) {
1505 /* don't resolve 1D via WINS */
1506 if (name_type != 0x1D) {
1507 status = resolve_wins(name, name_type,
1508 return_iplist,
1509 return_count);
1510 if (NT_STATUS_IS_OK(status)) {
1511 goto done;
1512 }
1513 }
1514 } else if(strequal( tok, "bcast")) {
1515 status = name_resolve_bcast(name, name_type,
1516 return_iplist,
1517 return_count);
1518 if (NT_STATUS_IS_OK(status)) {
1519 goto done;
1520 }
1521 } else {
1522 DEBUG(0,("resolve_name: unknown name switch type %s\n",
1523 tok));
1524 }
1525 }
1526
1527 /* All of the resolve_* functions above have returned false. */
1528
1529 TALLOC_FREE(frame);
1530 SAFE_FREE(*return_iplist);
1531 *return_count = 0;
1532
1533 return NT_STATUS_UNSUCCESSFUL;
1534
1535 done:
1536
1537 /* Remove duplicate entries. Some queries, notably #1c (domain
1538 controllers) return the PDC in iplist[0] and then all domain
1539 controllers including the PDC in iplist[1..n]. Iterating over
1540 the iplist when the PDC is down will cause two sets of timeouts. */
1541
1542 if ( *return_count ) {
1543 *return_count = remove_duplicate_addrs2(*return_iplist,
1544 *return_count );
1545 }
1546
1547 /* Save in name cache */
1548 if ( DEBUGLEVEL >= 100 ) {
1549 for (i = 0; i < *return_count && DEBUGLEVEL == 100; i++) {
1550 char addr[INET6_ADDRSTRLEN];
1551 print_sockaddr(addr, sizeof(addr),
1552 &(*return_iplist)[i].ss);
1553 DEBUG(100, ("Storing name %s of type %d (%s:%d)\n",
1554 name,
1555 name_type,
1556 addr,
1557 (*return_iplist)[i].port));
1558 }
1559 }
1560
1561 namecache_store(name, name_type, *return_count, *return_iplist);
1562
1563 /* Display some debugging info */
1564
1565 if ( DEBUGLEVEL >= 10 ) {
1566 DEBUG(10, ("internal_resolve_name: returning %d addresses: ",
1567 *return_count));
1568
1569 for (i = 0; i < *return_count; i++) {
1570 char addr[INET6_ADDRSTRLEN];
1571 print_sockaddr(addr, sizeof(addr),
1572 &(*return_iplist)[i].ss);
1573 DEBUGADD(10, ("%s:%d ",
1574 addr,
1575 (*return_iplist)[i].port));
1576 }
1577 DEBUG(10, ("\n"));
1578 }
1579
1580 TALLOC_FREE(frame);
1581 return status;
1582 }
1583
1584 /********************************************************
1585 Internal interface to resolve a name into one IP address.
1586 Use this function if the string is either an IP address, DNS
1587 or host name or NetBIOS name. This uses the name switch in the
1588 smb.conf to determine the order of name resolution.
1589 *********************************************************/
1590
1591 bool resolve_name(const char *name,
1592 struct sockaddr_storage *return_ss,
1593 int name_type,
1594 bool prefer_ipv4)
1595 {
1596 struct ip_service *ss_list = NULL;
1597 char *sitename = NULL;
1598 int count = 0;
1599
1600 if (is_ipaddress(name)) {
1601 return interpret_string_addr(return_ss, name, AI_NUMERICHOST);
1602 }
1603
1604 sitename = sitename_fetch(lp_realm()); /* wild guess */
1605
1606 if (NT_STATUS_IS_OK(internal_resolve_name(name, name_type, sitename,
1607 &ss_list, &count,
1608 lp_name_resolve_order()))) {
1609 int i;
1610
1611 if (prefer_ipv4) {
1612 for (i=0; i<count; i++) {
1613 if (!is_zero_addr((struct sockaddr *)&ss_list[i].ss) &&
1614 !is_broadcast_addr((struct sockaddr *)&ss_list[i].ss) &&
1615 (ss_list[i].ss.ss_family == AF_INET)) {
1616 *return_ss = ss_list[i].ss;
1617 SAFE_FREE(ss_list);
1618 SAFE_FREE(sitename);
1619 return True;
1620 }
1621 }
1622 }
1623
1624 /* only return valid addresses for TCP connections */
1625 for (i=0; i<count; i++) {
1626 if (!is_zero_addr((struct sockaddr *)&ss_list[i].ss) &&
1627 !is_broadcast_addr((struct sockaddr *)&ss_list[i].ss)) {
1628 *return_ss = ss_list[i].ss;
1629 SAFE_FREE(ss_list);
1630 SAFE_FREE(sitename);
1631 return True;
1632 }
1633 }
1634 }
1635
1636 SAFE_FREE(ss_list);
1637 SAFE_FREE(sitename);
1638 return False;
1639 }
1640
1641 /********************************************************
1642 Internal interface to resolve a name into a list of IP addresses.
1643 Use this function if the string is either an IP address, DNS
1644 or host name or NetBIOS name. This uses the name switch in the
1645 smb.conf to determine the order of name resolution.
1646 *********************************************************/
1647
1648 NTSTATUS resolve_name_list(TALLOC_CTX *ctx,
1649 const char *name,
1650 int name_type,
1651 struct sockaddr_storage **return_ss_arr,
1652 unsigned int *p_num_entries)
1653 {
1654 struct ip_service *ss_list = NULL;
1655 char *sitename = NULL;
1656 int count = 0;
1657 int i;
1658 unsigned int num_entries;
1659 NTSTATUS status;
1660
1661 *p_num_entries = 0;
1662 *return_ss_arr = NULL;
1663
1664 if (is_ipaddress(name)) {
1665 *return_ss_arr = TALLOC_P(ctx, struct sockaddr_storage);
1666 if (!*return_ss_arr) {
1667 return NT_STATUS_NO_MEMORY;
1668 }
1669 if (!interpret_string_addr(*return_ss_arr, name, AI_NUMERICHOST)) {
1670 TALLOC_FREE(*return_ss_arr);
1671 return NT_STATUS_BAD_NETWORK_NAME;
1672 }
1673 *p_num_entries = 1;
1674 return NT_STATUS_OK;
1675 }
1676
1677 sitename = sitename_fetch(lp_realm()); /* wild guess */
1678
1679 status = internal_resolve_name(name, name_type, sitename,
1680 &ss_list, &count,
1681 lp_name_resolve_order());
1682 SAFE_FREE(sitename);
1683
1684 if (!NT_STATUS_IS_OK(status)) {
1685 return status;
1686 }
1687
1688 /* only return valid addresses for TCP connections */
1689 for (i=0, num_entries = 0; i<count; i++) {
1690 if (!is_zero_addr((struct sockaddr *)&ss_list[i].ss) &&
1691 !is_broadcast_addr((struct sockaddr *)&ss_list[i].ss)) {
1692 num_entries++;
1693 }
1694 }
1695 if (num_entries == 0) {
1696 SAFE_FREE(ss_list);
1697 return NT_STATUS_BAD_NETWORK_NAME;
1698 }
1699
1700 *return_ss_arr = TALLOC_ARRAY(ctx,
1701 struct sockaddr_storage,
1702 num_entries);
1703 if (!(*return_ss_arr)) {
1704 SAFE_FREE(ss_list);
1705 return NT_STATUS_NO_MEMORY;
1706 }
1707
1708 for (i=0, num_entries = 0; i<count; i++) {
1709 if (!is_zero_addr((struct sockaddr *)&ss_list[i].ss) &&
1710 !is_broadcast_addr((struct sockaddr *)&ss_list[i].ss)) {
1711 (*return_ss_arr)[num_entries++] = ss_list[i].ss;
1712 }
1713 }
1714
1715 status = NT_STATUS_OK;
1716 *p_num_entries = num_entries;
1717
1718 SAFE_FREE(ss_list);
1719 return NT_STATUS_OK;
1720 }
1721
1722 /********************************************************
1723 Find the IP address of the master browser or DMB for a workgroup.
1724 *********************************************************/
1725
1726 bool find_master_ip(const char *group, struct sockaddr_storage *master_ss)
1727 {
1728 struct ip_service *ip_list = NULL;
1729 int count = 0;
1730 NTSTATUS status;
1731
1732 if (lp_disable_netbios()) {
1733 DEBUG(5,("find_master_ip(%s): netbios is disabled\n", group));
1734 return false;
1735 }
1736
1737 status = internal_resolve_name(group, 0x1D, NULL, &ip_list, &count,
1738 lp_name_resolve_order());
1739 if (NT_STATUS_IS_OK(status)) {
1740 *master_ss = ip_list[0].ss;
1741 SAFE_FREE(ip_list);
1742 return true;
1743 }
1744
1745 status = internal_resolve_name(group, 0x1B, NULL, &ip_list, &count,
1746 lp_name_resolve_order());
1747 if (NT_STATUS_IS_OK(status)) {
1748 *master_ss = ip_list[0].ss;
1749 SAFE_FREE(ip_list);
1750 return true;
1751 }
1752
1753 SAFE_FREE(ip_list);
1754 return false;
1755 }
1756
1757 /********************************************************
1758 Get the IP address list of the primary domain controller
1759 for a domain.
1760 *********************************************************/
1761
1762 bool get_pdc_ip(const char *domain, struct sockaddr_storage *pss)
1763 {
1764 struct ip_service *ip_list = NULL;
1765 int count = 0;
1766 NTSTATUS status = NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
1767
1768 /* Look up #1B name */
1769
1770 if (lp_security() == SEC_ADS) {
1771 status = internal_resolve_name(domain, 0x1b, NULL, &ip_list,
1772 &count, "ads");
1773 }
1774
1775 if (!NT_STATUS_IS_OK(status) || count == 0) {
1776 status = internal_resolve_name(domain, 0x1b, NULL, &ip_list,
1777 &count,
1778 lp_name_resolve_order());
1779 if (!NT_STATUS_IS_OK(status)) {
1780 return false;
1781 }
1782 }
1783
1784 /* if we get more than 1 IP back we have to assume it is a
1785 multi-homed PDC and not a mess up */
1786
1787 if ( count > 1 ) {
1788 DEBUG(6,("get_pdc_ip: PDC has %d IP addresses!\n", count));
1789 sort_service_list(ip_list, count);
1790 }
1791
1792 *pss = ip_list[0].ss;
1793 SAFE_FREE(ip_list);
1794 return true;
1795 }
1796
1797 /* Private enum type for lookups. */
1798
1799 enum dc_lookup_type { DC_NORMAL_LOOKUP, DC_ADS_ONLY, DC_KDC_ONLY };
1800
1801 /********************************************************
1802 Get the IP address list of the domain controllers for
1803 a domain.
1804 *********************************************************/
1805
1806 static NTSTATUS get_dc_list(const char *domain,
1807 const char *sitename,
1808 struct ip_service **ip_list,
1809 int *count,
1810 enum dc_lookup_type lookup_type,
1811 bool *ordered)
1812 {
1813 char *resolve_order = NULL;
1814 char *saf_servername = NULL;
1815 char *pserver = NULL;
1816 const char *p;
1817 char *port_str = NULL;
1818 int port;
1819 char *name;
1820 int num_addresses = 0;
1821 int local_count, i, j;
1822 struct ip_service *return_iplist = NULL;
1823 struct ip_service *auto_ip_list = NULL;
1824 bool done_auto_lookup = false;
1825 int auto_count = 0;
1826 NTSTATUS status;
1827 TALLOC_CTX *ctx = talloc_init("get_dc_list");
1828
1829 *ip_list = NULL;
1830 *count = 0;
1831
1832 if (!ctx) {
1833 return NT_STATUS_NO_MEMORY;
1834 }
1835
1836 *ordered = False;
1837
1838 /* if we are restricted to solely using DNS for looking
1839 up a domain controller, make sure that host lookups
1840 are enabled for the 'name resolve order'. If host lookups
1841 are disabled and ads_only is True, then set the string to
1842 NULL. */
1843
1844 resolve_order = talloc_strdup(ctx, lp_name_resolve_order());
1845 if (!resolve_order) {
1846 status = NT_STATUS_NO_MEMORY;
1847 goto out;
1848 }
1849 strlower_m(resolve_order);
1850 if (lookup_type == DC_ADS_ONLY) {
1851 if (strstr( resolve_order, "host")) {
1852 resolve_order = talloc_strdup(ctx, "ads");
1853
1854 /* DNS SRV lookups used by the ads resolver
1855 are already sorted by priority and weight */
1856 *ordered = true;
1857 } else {
1858 resolve_order = talloc_strdup(ctx, "NULL");
1859 }
1860 } else if (lookup_type == DC_KDC_ONLY) {
1861 /* DNS SRV lookups used by the ads/kdc resolver
1862 are already sorted by priority and weight */
1863 *ordered = true;
1864 resolve_order = talloc_strdup(ctx, "kdc");
1865 }
1866 if (!resolve_order) {
1867 status = NT_STATUS_NO_MEMORY;
1868 goto out;
1869 }
1870
1871 /* fetch the server we have affinity for. Add the
1872 'password server' list to a search for our domain controllers */
1873
1874 saf_servername = saf_fetch( domain);
1875
1876 if (strequal(domain, lp_workgroup()) || strequal(domain, lp_realm())) {
1877 pserver = talloc_asprintf(NULL, "%s, %s",
1878 saf_servername ? saf_servername : "",
1879 lp_passwordserver());
1880 } else {
1881 pserver = talloc_asprintf(NULL, "%s, *",
1882 saf_servername ? saf_servername : "");
1883 }
1884
1885 SAFE_FREE(saf_servername);
1886 if (!pserver) {
1887 status = NT_STATUS_NO_MEMORY;
1888 goto out;
1889 }
1890
1891 /* if we are starting from scratch, just lookup DOMAIN<0x1c> */
1892
1893 if (!*pserver ) {
1894 DEBUG(10,("get_dc_list: no preferred domain controllers.\n"));
1895 status = internal_resolve_name(domain, 0x1C, sitename, ip_list,
1896 count, resolve_order);
1897 goto out;
1898 }
1899
1900 DEBUG(3,("get_dc_list: preferred server list: \"%s\"\n", pserver ));
1901
1902 /*
1903 * if '*' appears in the "password server" list then add
1904 * an auto lookup to the list of manually configured
1905 * DC's. If any DC is listed by name, then the list should be
1906 * considered to be ordered
1907 */
1908
1909 p = pserver;
1910 while (next_token_talloc(ctx, &p, &name, LIST_SEP)) {
1911 if (!done_auto_lookup && strequal(name, "*")) {
1912 status = internal_resolve_name(domain, 0x1C, sitename,
1913 &auto_ip_list,
1914 &auto_count,
1915 resolve_order);
1916 if (NT_STATUS_IS_OK(status)) {
1917 num_addresses += auto_count;
1918 }
1919 done_auto_lookup = true;
1920 DEBUG(8,("Adding %d DC's from auto lookup\n",
1921 auto_count));
1922 } else {
1923 num_addresses++;
1924 }
1925 }
1926
1927 /* if we have no addresses and haven't done the auto lookup, then
1928 just return the list of DC's. Or maybe we just failed. */
1929
1930 if ((num_addresses == 0)) {
1931 if (done_auto_lookup) {
1932 DEBUG(4,("get_dc_list: no servers found\n"));
1933 status = NT_STATUS_NO_LOGON_SERVERS;
1934 goto out;
1935 }
1936 status = internal_resolve_name(domain, 0x1C, sitename, ip_list,
1937 count, resolve_order);
1938 goto out;
1939 }
1940
1941 if ((return_iplist = SMB_MALLOC_ARRAY(struct ip_service,
1942 num_addresses)) == NULL) {
1943 DEBUG(3,("get_dc_list: malloc fail !\n"));
1944 status = NT_STATUS_NO_MEMORY;
1945 goto out;
1946 }
1947
1948 p = pserver;
1949 local_count = 0;
1950
1951 /* fill in the return list now with real IP's */
1952
1953 while ((local_count<num_addresses) &&
1954 next_token_talloc(ctx, &p, &name, LIST_SEP)) {
1955 struct sockaddr_storage name_ss;
1956
1957 /* copy any addersses from the auto lookup */
1958
1959 if (strequal(name, "*")) {
1960 for (j=0; j<auto_count; j++) {
1961 char addr[INET6_ADDRSTRLEN];
1962 print_sockaddr(addr,
1963 sizeof(addr),
1964 &auto_ip_list[j].ss);
1965 /* Check for and don't copy any
1966 * known bad DC IP's. */
1967 if(!NT_STATUS_IS_OK(check_negative_conn_cache(
1968 domain,
1969 addr))) {
1970 DEBUG(5,("get_dc_list: "
1971 "negative entry %s removed "
1972 "from DC list\n",
1973 addr));
1974 continue;
1975 }
1976 return_iplist[local_count].ss =
1977 auto_ip_list[j].ss;
1978 return_iplist[local_count].port =
1979 auto_ip_list[j].port;
1980 local_count++;
1981 }
1982 continue;
1983 }
1984
1985 /* added support for address:port syntax for ads
1986 * (not that I think anyone will ever run the LDAP
1987 * server in an AD domain on something other than
1988 * port 389 */
1989
1990 port = (lp_security() == SEC_ADS) ? LDAP_PORT : PORT_NONE;
1991 if ((port_str=strchr(name, ':')) != NULL) {
1992 *port_str = '\0';
1993 port_str++;
1994 port = atoi(port_str);
1995 }
1996
1997 /* explicit lookup; resolve_name() will
1998 * handle names & IP addresses */
1999 if (resolve_name( name, &name_ss, 0x20, true )) {
2000 char addr[INET6_ADDRSTRLEN];
2001 print_sockaddr(addr,
2002 sizeof(addr),
2003 &name_ss);
2004
2005 /* Check for and don't copy any known bad DC IP's. */
2006 if( !NT_STATUS_IS_OK(check_negative_conn_cache(domain,
2007 addr)) ) {
2008 DEBUG(5,("get_dc_list: negative entry %s "
2009 "removed from DC list\n",
2010 name ));
2011 continue;
2012 }
2013
2014 return_iplist[local_count].ss = name_ss;
2015 return_iplist[local_count].port = port;
2016 local_count++;
2017 *ordered = true;
2018 }
2019 }
2020
2021 /* need to remove duplicates in the list if we have any
2022 explicit password servers */
2023
2024 if (local_count) {
2025 local_count = remove_duplicate_addrs2(return_iplist,
2026 local_count );
2027 }
2028
2029 /* For DC's we always prioritize IPv4 due to W2K3 not
2030 * supporting LDAP, KRB5 or CLDAP over IPv6. */
2031
2032 if (local_count && return_iplist) {
2033 prioritize_ipv4_list(return_iplist, local_count);
2034 }
2035
2036 if ( DEBUGLEVEL >= 4 ) {
2037 DEBUG(4,("get_dc_list: returning %d ip addresses "
2038 "in an %sordered list\n",
2039 local_count,
2040 *ordered ? "":"un"));
2041 DEBUG(4,("get_dc_list: "));
2042 for ( i=0; i<local_count; i++ ) {
2043 char addr[INET6_ADDRSTRLEN];
2044 print_sockaddr(addr,
2045 sizeof(addr),
2046 &return_iplist[i].ss);
2047 DEBUGADD(4,("%s:%d ", addr, return_iplist[i].port ));
2048 }
2049 DEBUGADD(4,("\n"));
2050 }
2051
2052 *ip_list = return_iplist;
2053 *count = local_count;
2054
2055 status = ( *count != 0 ? NT_STATUS_OK : NT_STATUS_NO_LOGON_SERVERS );
2056
2057 out:
2058
2059 if (!NT_STATUS_IS_OK(status)) {
2060 SAFE_FREE(return_iplist);
2061 *ip_list = NULL;
2062 *count = 0;
2063 }
2064
2065 SAFE_FREE(auto_ip_list);
2066 TALLOC_FREE(ctx);
2067 return status;
2068 }
2069
2070 /*********************************************************************
2071 Small wrapper function to get the DC list and sort it if neccessary.
2072 *********************************************************************/
2073
2074 NTSTATUS get_sorted_dc_list( const char *domain,
2075 const char *sitename,
2076 struct ip_service **ip_list,
2077 int *count,
2078 bool ads_only )
2079 {
2080 bool ordered = false;
2081 NTSTATUS status;
2082 enum dc_lookup_type lookup_type = DC_NORMAL_LOOKUP;
2083
2084 *ip_list = NULL;
2085 *count = 0;
2086
2087 DEBUG(8,("get_sorted_dc_list: attempting lookup "
2088 "for name %s (sitename %s) using [%s]\n",
2089 domain,
2090 sitename ? sitename : "NULL",
2091 (ads_only ? "ads" : lp_name_resolve_order())));
2092
2093 if (ads_only) {
2094 lookup_type = DC_ADS_ONLY;
2095 }
2096
2097 status = get_dc_list(domain, sitename, ip_list,
2098 count, lookup_type, &ordered);
2099 if (NT_STATUS_EQUAL(status, NT_STATUS_NO_LOGON_SERVERS)
2100 && sitename) {
2101 DEBUG(3,("get_sorted_dc_list: no server for name %s available"
2102 " in site %s, fallback to all servers\n",
2103 domain, sitename));
2104 status = get_dc_list(domain, NULL, ip_list,
2105 count, lookup_type, &ordered);
2106 }
2107
2108 if (!NT_STATUS_IS_OK(status)) {
2109 SAFE_FREE(*ip_list);
2110 *count = 0;
2111 return status;
2112 }
2113
2114 /* only sort if we don't already have an ordered list */
2115 if (!ordered) {
2116 sort_service_list(*ip_list, *count);
2117 }
2118
2119 return NT_STATUS_OK;
2120 }
2121
2122 /*********************************************************************
2123 Get the KDC list - re-use all the logic in get_dc_list.
2124 *********************************************************************/
2125
2126 NTSTATUS get_kdc_list( const char *realm,
2127 const char *sitename,
2128 struct ip_service **ip_list,
2129 int *count)
2130 {
2131 bool ordered;
2132 NTSTATUS status;
2133
2134 *count = 0;
2135 *ip_list = NULL;
2136
2137 status = get_dc_list(realm, sitename, ip_list,
2138 count, DC_KDC_ONLY, &ordered);
2139
2140 if (!NT_STATUS_IS_OK(status)) {
2141 SAFE_FREE(*ip_list);
2142 *count = 0;
2143 return status;
2144 }
2145
2146 /* only sort if we don't already have an ordered list */
2147 if ( !ordered ) {
2148 sort_service_list(*ip_list, *count);
2149 }
2150
2151 return NT_STATUS_OK;
2152 }
Samba GIT mirror