s3-tevent: only include ../lib/util/tevent wrappers where needed.
[bbaumbach/samba-autobuild/.git] / source3 / libsmb / namequery.c
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 #include "../lib/util/tevent_ntstatus.h"
23 #include "libads/sitename_cache.h"
24 #include "libads/dns.h"
25 #include "../libcli/netlogon/netlogon.h"
26 #include "lib/async_req/async_sock.h"
27 #include "libsmb/nmblib.h"
28
29 /* nmbd.c sets this to True. */
30 bool global_in_nmbd = False;
31
32 /****************************
33  * SERVER AFFINITY ROUTINES *
34  ****************************/
35
36  /* Server affinity is the concept of preferring the last domain
37     controller with whom you had a successful conversation */
38
39 /****************************************************************************
40 ****************************************************************************/
41 #define SAFKEY_FMT      "SAF/DOMAIN/%s"
42 #define SAF_TTL         900
43 #define SAFJOINKEY_FMT  "SAFJOIN/DOMAIN/%s"
44 #define SAFJOIN_TTL     3600
45
46 static char *saf_key(const char *domain)
47 {
48         char *keystr;
49
50         asprintf_strupper_m(&keystr, SAFKEY_FMT, domain);
51
52         return keystr;
53 }
54
55 static char *saf_join_key(const char *domain)
56 {
57         char *keystr;
58
59         asprintf_strupper_m(&keystr, SAFJOINKEY_FMT, domain);
60
61         return keystr;
62 }
63
64 /****************************************************************************
65 ****************************************************************************/
66
67 bool saf_store( const char *domain, const char *servername )
68 {
69         char *key;
70         time_t expire;
71         bool ret = False;
72
73         if ( !domain || !servername ) {
74                 DEBUG(2,("saf_store: "
75                         "Refusing to store empty domain or servername!\n"));
76                 return False;
77         }
78
79         if ( (strlen(domain) == 0) || (strlen(servername) == 0) ) {
80                 DEBUG(0,("saf_store: "
81                         "refusing to store 0 length domain or servername!\n"));
82                 return False;
83         }
84
85         key = saf_key( domain );
86         expire = time( NULL ) + lp_parm_int(-1, "saf","ttl", SAF_TTL);
87
88         DEBUG(10,("saf_store: domain = [%s], server = [%s], expire = [%u]\n",
89                 domain, servername, (unsigned int)expire ));
90
91         ret = gencache_set( key, servername, expire );
92
93         SAFE_FREE( key );
94
95         return ret;
96 }
97
98 bool saf_join_store( const char *domain, const char *servername )
99 {
100         char *key;
101         time_t expire;
102         bool ret = False;
103
104         if ( !domain || !servername ) {
105                 DEBUG(2,("saf_join_store: Refusing to store empty domain or servername!\n"));
106                 return False;
107         }
108
109         if ( (strlen(domain) == 0) || (strlen(servername) == 0) ) {
110                 DEBUG(0,("saf_join_store: refusing to store 0 length domain or servername!\n"));
111                 return False;
112         }
113
114         key = saf_join_key( domain );
115         expire = time( NULL ) + lp_parm_int(-1, "saf","join ttl", SAFJOIN_TTL);
116
117         DEBUG(10,("saf_join_store: domain = [%s], server = [%s], expire = [%u]\n",
118                 domain, servername, (unsigned int)expire ));
119
120         ret = gencache_set( key, servername, expire );
121
122         SAFE_FREE( key );
123
124         return ret;
125 }
126
127 bool saf_delete( const char *domain )
128 {
129         char *key;
130         bool ret = False;
131
132         if ( !domain ) {
133                 DEBUG(2,("saf_delete: Refusing to delete empty domain\n"));
134                 return False;
135         }
136
137         key = saf_join_key(domain);
138         ret = gencache_del(key);
139         SAFE_FREE(key);
140
141         if (ret) {
142                 DEBUG(10,("saf_delete[join]: domain = [%s]\n", domain ));
143         }
144
145         key = saf_key(domain);
146         ret = gencache_del(key);
147         SAFE_FREE(key);
148
149         if (ret) {
150                 DEBUG(10,("saf_delete: domain = [%s]\n", domain ));
151         }
152
153         return ret;
154 }
155
156 /****************************************************************************
157 ****************************************************************************/
158
159 char *saf_fetch( const char *domain )
160 {
161         char *server = NULL;
162         time_t timeout;
163         bool ret = False;
164         char *key = NULL;
165
166         if ( !domain || strlen(domain) == 0) {
167                 DEBUG(2,("saf_fetch: Empty domain name!\n"));
168                 return NULL;
169         }
170
171         key = saf_join_key( domain );
172
173         ret = gencache_get( key, &server, &timeout );
174
175         SAFE_FREE( key );
176
177         if ( ret ) {
178                 DEBUG(5,("saf_fetch[join]: Returning \"%s\" for \"%s\" domain\n",
179                         server, domain ));
180                 return server;
181         }
182
183         key = saf_key( domain );
184
185         ret = gencache_get( key, &server, &timeout );
186
187         SAFE_FREE( key );
188
189         if ( !ret ) {
190                 DEBUG(5,("saf_fetch: failed to find server for \"%s\" domain\n",
191                                         domain ));
192         } else {
193                 DEBUG(5,("saf_fetch: Returning \"%s\" for \"%s\" domain\n",
194                         server, domain ));
195         }
196
197         return server;
198 }
199
200 /****************************************************************************
201  Generate a random trn_id.
202 ****************************************************************************/
203
204 static int generate_trn_id(void)
205 {
206         uint16 id;
207
208         generate_random_buffer((uint8 *)&id, sizeof(id));
209
210         return id % (unsigned)0x7FFF;
211 }
212
213 /****************************************************************************
214  Parse a node status response into an array of structures.
215 ****************************************************************************/
216
217 static struct node_status *parse_node_status(TALLOC_CTX *mem_ctx, char *p,
218                                 int *num_names,
219                                 struct node_status_extra *extra)
220 {
221         struct node_status *ret;
222         int i;
223
224         *num_names = CVAL(p,0);
225
226         if (*num_names == 0)
227                 return NULL;
228
229         ret = TALLOC_ARRAY(mem_ctx, struct node_status,*num_names);
230         if (!ret)
231                 return NULL;
232
233         p++;
234         for (i=0;i< *num_names;i++) {
235                 StrnCpy(ret[i].name,p,15);
236                 trim_char(ret[i].name,'\0',' ');
237                 ret[i].type = CVAL(p,15);
238                 ret[i].flags = p[16];
239                 p += 18;
240                 DEBUG(10, ("%s#%02x: flags = 0x%02x\n", ret[i].name,
241                            ret[i].type, ret[i].flags));
242         }
243         /*
244          * Also, pick up the MAC address ...
245          */
246         if (extra) {
247                 memcpy(&extra->mac_addr, p, 6); /* Fill in the mac addr */
248         }
249         return ret;
250 }
251
252 struct sock_packet_read_state {
253         struct tevent_context *ev;
254         enum packet_type type;
255         int trn_id;
256
257         struct nb_packet_reader *reader;
258         struct tevent_req *reader_req;
259
260         int sock;
261         struct tevent_req *socket_req;
262         uint8_t buf[1024];
263         struct sockaddr_storage addr;
264         socklen_t addr_len;
265
266         bool (*validator)(struct packet_struct *p,
267                           void *private_data);
268         void *private_data;
269
270         struct packet_struct *packet;
271 };
272
273 static int sock_packet_read_state_destructor(struct sock_packet_read_state *s);
274 static void sock_packet_read_got_packet(struct tevent_req *subreq);
275 static void sock_packet_read_got_socket(struct tevent_req *subreq);
276
277 static struct tevent_req *sock_packet_read_send(
278         TALLOC_CTX *mem_ctx,
279         struct tevent_context *ev,
280         int sock, /* dgram socket */
281         struct nb_packet_reader *reader,
282         enum packet_type type,
283         int trn_id,
284         bool (*validator)(struct packet_struct *p, void *private_data),
285         void *private_data)
286 {
287         struct tevent_req *req;
288         struct sock_packet_read_state *state;
289
290         req = tevent_req_create(mem_ctx, &state,
291                                 struct sock_packet_read_state);
292         if (req == NULL) {
293                 return NULL;
294         }
295         talloc_set_destructor(state, sock_packet_read_state_destructor);
296         state->ev = ev;
297         state->reader = reader;
298         state->sock = sock;
299         state->type = type;
300         state->trn_id = trn_id;
301         state->validator = validator;
302         state->private_data = private_data;
303
304         if (reader != NULL) {
305                 state->reader_req = nb_packet_read_send(state, ev, reader);
306                 if (tevent_req_nomem(state->reader_req, req)) {
307                         return tevent_req_post(req, ev);
308                 }
309                 tevent_req_set_callback(
310                         state->reader_req, sock_packet_read_got_packet, req);
311         }
312
313         state->addr_len = sizeof(state->addr);
314         state->socket_req = recvfrom_send(state, ev, sock,
315                                           state->buf, sizeof(state->buf), 0,
316                                           &state->addr, &state->addr_len);
317         if (tevent_req_nomem(state->socket_req, req)) {
318                 return tevent_req_post(req, ev);
319         }
320         tevent_req_set_callback(state->socket_req, sock_packet_read_got_socket,
321                                 req);
322
323         return req;
324 }
325
326 static int sock_packet_read_state_destructor(struct sock_packet_read_state *s)
327 {
328         if (s->packet != NULL) {
329                 free_packet(s->packet);
330                 s->packet = NULL;
331         }
332         return 0;
333 }
334
335 static void sock_packet_read_got_packet(struct tevent_req *subreq)
336 {
337         struct tevent_req *req = tevent_req_callback_data(
338                 subreq, struct tevent_req);
339         struct sock_packet_read_state *state = tevent_req_data(
340                 req, struct sock_packet_read_state);
341         NTSTATUS status;
342
343         status = nb_packet_read_recv(subreq, &state->packet);
344
345         TALLOC_FREE(state->reader_req);
346
347         if (!NT_STATUS_IS_OK(status)) {
348                 if (state->socket_req != NULL) {
349                         /*
350                          * Still waiting for socket
351                          */
352                         return;
353                 }
354                 /*
355                  * Both socket and packet reader failed
356                  */
357                 tevent_req_nterror(req, status);
358                 return;
359         }
360
361         if ((state->validator != NULL) &&
362             !state->validator(state->packet, state->private_data)) {
363                 DEBUG(10, ("validator failed\n"));
364
365                 free_packet(state->packet);
366                 state->packet = NULL;
367
368                 state->reader_req = nb_packet_read_send(state, state->ev,
369                                                         state->reader);
370                 if (tevent_req_nomem(state->reader_req, req)) {
371                         return;
372                 }
373                 tevent_req_set_callback(
374                         state->reader_req, sock_packet_read_got_packet, req);
375                 return;
376         }
377
378         TALLOC_FREE(state->socket_req);
379         tevent_req_done(req);
380 }
381
382 static void sock_packet_read_got_socket(struct tevent_req *subreq)
383 {
384         struct tevent_req *req = tevent_req_callback_data(
385                 subreq, struct tevent_req);
386         struct sock_packet_read_state *state = tevent_req_data(
387                 req, struct sock_packet_read_state);
388         struct sockaddr_in *in_addr;
389         ssize_t received;
390         int err;
391
392         received = recvfrom_recv(subreq, &err);
393
394         TALLOC_FREE(state->socket_req);
395
396         if (received == -1) {
397                 if (state->reader_req != NULL) {
398                         /*
399                          * Still waiting for reader
400                          */
401                         return;
402                 }
403                 /*
404                  * Both socket and reader failed
405                  */
406                 tevent_req_nterror(req, map_nt_error_from_unix(err));
407                 return;
408         }
409         if (state->addr.ss_family != AF_INET) {
410                 goto retry;
411         }
412         in_addr = (struct sockaddr_in *)(void *)&state->addr;
413
414         state->packet = parse_packet((char *)state->buf, received, state->type,
415                                      in_addr->sin_addr, in_addr->sin_port);
416         if (state->packet == NULL) {
417                 DEBUG(10, ("parse_packet failed\n"));
418                 goto retry;
419         }
420         if ((state->trn_id != -1) &&
421             (state->trn_id != packet_trn_id(state->packet))) {
422                 DEBUG(10, ("Expected transaction id %d, got %d\n",
423                            state->trn_id, packet_trn_id(state->packet)));
424                 goto retry;
425         }
426
427         if ((state->validator != NULL) &&
428             !state->validator(state->packet, state->private_data)) {
429                 DEBUG(10, ("validator failed\n"));
430                 goto retry;
431         }
432
433         tevent_req_done(req);
434         return;
435
436 retry:
437         if (state->packet != NULL) {
438                 free_packet(state->packet);
439                 state->packet = NULL;
440         }
441         state->socket_req = recvfrom_send(state, state->ev, state->sock,
442                                           state->buf, sizeof(state->buf), 0,
443                                           &state->addr, &state->addr_len);
444         if (tevent_req_nomem(state->socket_req, req)) {
445                 return;
446         }
447         tevent_req_set_callback(state->socket_req, sock_packet_read_got_socket,
448                                 req);
449 }
450
451 static NTSTATUS sock_packet_read_recv(struct tevent_req *req,
452                                       struct packet_struct **ppacket)
453 {
454         struct sock_packet_read_state *state = tevent_req_data(
455                 req, struct sock_packet_read_state);
456         NTSTATUS status;
457
458         if (tevent_req_is_nterror(req, &status)) {
459                 return status;
460         }
461         *ppacket = state->packet;
462         state->packet = NULL;
463         return NT_STATUS_OK;
464 }
465
466 struct nb_trans_state {
467         struct tevent_context *ev;
468         int sock;
469         struct nb_packet_reader *reader;
470
471         const struct sockaddr_storage *dst_addr;
472         uint8_t *buf;
473         size_t buflen;
474         enum packet_type type;
475         int trn_id;
476
477         bool (*validator)(struct packet_struct *p,
478                           void *private_data);
479         void *private_data;
480
481         struct packet_struct *packet;
482 };
483
484 static int nb_trans_state_destructor(struct nb_trans_state *s);
485 static void nb_trans_got_reader(struct tevent_req *subreq);
486 static void nb_trans_done(struct tevent_req *subreq);
487 static void nb_trans_sent(struct tevent_req *subreq);
488 static void nb_trans_send_next(struct tevent_req *subreq);
489
490 static struct tevent_req *nb_trans_send(
491         TALLOC_CTX *mem_ctx,
492         struct tevent_context *ev,
493         const struct sockaddr_storage *my_addr,
494         const struct sockaddr_storage *dst_addr,
495         bool bcast,
496         uint8_t *buf, size_t buflen,
497         enum packet_type type, int trn_id,
498         bool (*validator)(struct packet_struct *p,
499                           void *private_data),
500         void *private_data)
501 {
502         struct tevent_req *req, *subreq;
503         struct nb_trans_state *state;
504
505         req = tevent_req_create(mem_ctx, &state, struct nb_trans_state);
506         if (req == NULL) {
507                 return NULL;
508         }
509         talloc_set_destructor(state, nb_trans_state_destructor);
510         state->ev = ev;
511         state->dst_addr = dst_addr;
512         state->buf = buf;
513         state->buflen = buflen;
514         state->type = type;
515         state->trn_id = trn_id;
516         state->validator = validator;
517         state->private_data = private_data;
518
519         state->sock = open_socket_in(SOCK_DGRAM, 0, 3, my_addr, True);
520         if (state->sock == -1) {
521                 tevent_req_nterror(req, map_nt_error_from_unix(errno));
522                 DEBUG(10, ("open_socket_in failed: %s\n", strerror(errno)));
523                 return tevent_req_post(req, ev);
524         }
525
526         if (bcast) {
527                 set_socket_options(state->sock,"SO_BROADCAST");
528         }
529
530         subreq = nb_packet_reader_send(state, ev, type, state->trn_id, NULL);
531         if (tevent_req_nomem(subreq, req)) {
532                 return tevent_req_post(req, ev);
533         }
534         tevent_req_set_callback(subreq, nb_trans_got_reader, req);
535         return req;
536 }
537
538 static int nb_trans_state_destructor(struct nb_trans_state *s)
539 {
540         if (s->sock != -1) {
541                 close(s->sock);
542                 s->sock = -1;
543         }
544         if (s->packet != NULL) {
545                 free_packet(s->packet);
546                 s->packet = NULL;
547         }
548         return 0;
549 }
550
551 static void nb_trans_got_reader(struct tevent_req *subreq)
552 {
553         struct tevent_req *req = tevent_req_callback_data(
554                 subreq, struct tevent_req);
555         struct nb_trans_state *state = tevent_req_data(
556                 req, struct nb_trans_state);
557         NTSTATUS status;
558
559         status = nb_packet_reader_recv(subreq, state, &state->reader);
560         TALLOC_FREE(subreq);
561
562         if (!NT_STATUS_IS_OK(status)) {
563                 DEBUG(10, ("nmbd not around\n"));
564                 state->reader = NULL;
565         }
566
567         subreq = sock_packet_read_send(
568                 state, state->ev, state->sock,
569                 state->reader, state->type, state->trn_id,
570                 state->validator, state->private_data);
571         if (tevent_req_nomem(subreq, req)) {
572                 return;
573         }
574         tevent_req_set_callback(subreq, nb_trans_done, req);
575
576         subreq = sendto_send(state, state->ev, state->sock,
577                              state->buf, state->buflen, 0, state->dst_addr);
578         if (tevent_req_nomem(subreq, req)) {
579                 return;
580         }
581         tevent_req_set_callback(subreq, nb_trans_sent, req);
582 }
583
584 static void nb_trans_sent(struct tevent_req *subreq)
585 {
586         struct tevent_req *req = tevent_req_callback_data(
587                 subreq, struct tevent_req);
588         struct nb_trans_state *state = tevent_req_data(
589                 req, struct nb_trans_state);
590         ssize_t sent;
591         int err;
592
593         sent = sendto_recv(subreq, &err);
594         TALLOC_FREE(subreq);
595         if (sent == -1) {
596                 DEBUG(10, ("sendto failed: %s\n", strerror(err)));
597                 tevent_req_nterror(req, map_nt_error_from_unix(err));
598                 return;
599         }
600         subreq = tevent_wakeup_send(state, state->ev,
601                                     timeval_current_ofs(1, 0));
602         if (tevent_req_nomem(subreq, req)) {
603                 return;
604         }
605         tevent_req_set_callback(subreq, nb_trans_send_next, req);
606 }
607
608 static void nb_trans_send_next(struct tevent_req *subreq)
609 {
610         struct tevent_req *req = tevent_req_callback_data(
611                 subreq, struct tevent_req);
612         struct nb_trans_state *state = tevent_req_data(
613                 req, struct nb_trans_state);
614         bool ret;
615
616         ret = tevent_wakeup_recv(subreq);
617         TALLOC_FREE(subreq);
618         if (!ret) {
619                 tevent_req_nterror(req, NT_STATUS_INTERNAL_ERROR);
620                 return;
621         }
622         subreq = sendto_send(state, state->ev, state->sock,
623                              state->buf, state->buflen, 0, state->dst_addr);
624         if (tevent_req_nomem(subreq, req)) {
625                 return;
626         }
627         tevent_req_set_callback(subreq, nb_trans_sent, req);
628 }
629
630 static void nb_trans_done(struct tevent_req *subreq)
631 {
632         struct tevent_req *req = tevent_req_callback_data(
633                 subreq, struct tevent_req);
634         struct nb_trans_state *state = tevent_req_data(
635                 req, struct nb_trans_state);
636         NTSTATUS status;
637
638         status = sock_packet_read_recv(subreq, &state->packet);
639         TALLOC_FREE(subreq);
640         if (tevent_req_nterror(req, status)) {
641                 return;
642         }
643         tevent_req_done(req);
644 }
645
646 static NTSTATUS nb_trans_recv(struct tevent_req *req,
647                               struct packet_struct **ppacket)
648 {
649         struct nb_trans_state *state = tevent_req_data(
650                 req, struct nb_trans_state);
651         NTSTATUS status;
652
653         if (tevent_req_is_nterror(req, &status)) {
654                 return status;
655         }
656         *ppacket = state->packet;
657         state->packet = NULL;
658         return NT_STATUS_OK;
659 }
660
661 /****************************************************************************
662  Do a NBT node status query on an open socket and return an array of
663  structures holding the returned names or NULL if the query failed.
664 **************************************************************************/
665
666 struct node_status_query_state {
667         struct sockaddr_storage my_addr;
668         struct sockaddr_storage addr;
669         uint8_t buf[1024];
670         ssize_t buflen;
671         struct packet_struct *packet;
672 };
673
674 static int node_status_query_state_destructor(
675         struct node_status_query_state *s);
676 static bool node_status_query_validator(struct packet_struct *p,
677                                         void *private_data);
678 static void node_status_query_done(struct tevent_req *subreq);
679
680 struct tevent_req *node_status_query_send(TALLOC_CTX *mem_ctx,
681                                           struct tevent_context *ev,
682                                           struct nmb_name *name,
683                                           const struct sockaddr_storage *addr)
684 {
685         struct tevent_req *req, *subreq;
686         struct node_status_query_state *state;
687         struct packet_struct p;
688         struct nmb_packet *nmb = &p.packet.nmb;
689         struct sockaddr_in *in_addr;
690
691         req = tevent_req_create(mem_ctx, &state,
692                                 struct node_status_query_state);
693         if (req == NULL) {
694                 return NULL;
695         }
696         talloc_set_destructor(state, node_status_query_state_destructor);
697
698         if (addr->ss_family != AF_INET) {
699                 /* Can't do node status to IPv6 */
700                 tevent_req_nterror(req, NT_STATUS_INVALID_ADDRESS);
701                 return tevent_req_post(req, ev);
702         }
703
704         state->addr = *addr;
705         in_addr = (struct sockaddr_in *)(void *)&state->addr;
706         in_addr->sin_port = htons(NMB_PORT);
707
708         if (!interpret_string_addr(&state->my_addr, lp_socket_address(),
709                                    AI_NUMERICHOST|AI_PASSIVE)) {
710                 zero_sockaddr(&state->my_addr);
711         }
712
713         ZERO_STRUCT(p);
714         nmb->header.name_trn_id = generate_trn_id();
715         nmb->header.opcode = 0;
716         nmb->header.response = false;
717         nmb->header.nm_flags.bcast = false;
718         nmb->header.nm_flags.recursion_available = false;
719         nmb->header.nm_flags.recursion_desired = false;
720         nmb->header.nm_flags.trunc = false;
721         nmb->header.nm_flags.authoritative = false;
722         nmb->header.rcode = 0;
723         nmb->header.qdcount = 1;
724         nmb->header.ancount = 0;
725         nmb->header.nscount = 0;
726         nmb->header.arcount = 0;
727         nmb->question.question_name = *name;
728         nmb->question.question_type = 0x21;
729         nmb->question.question_class = 0x1;
730
731         state->buflen = build_packet((char *)state->buf, sizeof(state->buf),
732                                      &p);
733         if (state->buflen == 0) {
734                 tevent_req_nterror(req, NT_STATUS_INTERNAL_ERROR);
735                 DEBUG(10, ("build_packet failed\n"));
736                 return tevent_req_post(req, ev);
737         }
738
739         subreq = nb_trans_send(state, ev, &state->my_addr, &state->addr, false,
740                                state->buf, state->buflen,
741                                NMB_PACKET, nmb->header.name_trn_id,
742                                node_status_query_validator, NULL);
743         if (tevent_req_nomem(subreq, req)) {
744                 DEBUG(10, ("nb_trans_send failed\n"));
745                 return tevent_req_post(req, ev);
746         }
747         if (!tevent_req_set_endtime(req, ev, timeval_current_ofs(10, 0))) {
748                 return tevent_req_post(req, ev);
749         }
750         tevent_req_set_callback(subreq, node_status_query_done, req);
751         return req;
752 }
753
754 static bool node_status_query_validator(struct packet_struct *p,
755                                         void *private_data)
756 {
757         struct nmb_packet *nmb = &p->packet.nmb;
758         debug_nmb_packet(p);
759
760         if (nmb->header.opcode != 0 ||
761             nmb->header.nm_flags.bcast ||
762             nmb->header.rcode ||
763             !nmb->header.ancount ||
764             nmb->answers->rr_type != 0x21) {
765                 /*
766                  * XXXX what do we do with this? could be a redirect,
767                  * but we'll discard it for the moment
768                  */
769                 return false;
770         }
771         return true;
772 }
773
774 static int node_status_query_state_destructor(
775         struct node_status_query_state *s)
776 {
777         if (s->packet != NULL) {
778                 free_packet(s->packet);
779                 s->packet = NULL;
780         }
781         return 0;
782 }
783
784 static void node_status_query_done(struct tevent_req *subreq)
785 {
786         struct tevent_req *req = tevent_req_callback_data(
787                 subreq, struct tevent_req);
788         struct node_status_query_state *state = tevent_req_data(
789                 req, struct node_status_query_state);
790         NTSTATUS status;
791
792         status = nb_trans_recv(subreq, &state->packet);
793         TALLOC_FREE(subreq);
794         if (tevent_req_nterror(req, status)) {
795                 return;
796         }
797         tevent_req_done(req);
798 }
799
800 NTSTATUS node_status_query_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
801                                 struct node_status **pnode_status,
802                                 int *pnum_names,
803                                 struct node_status_extra *extra)
804 {
805         struct node_status_query_state *state = tevent_req_data(
806                 req, struct node_status_query_state);
807         struct node_status *node_status;
808         int num_names;
809         NTSTATUS status;
810
811         if (tevent_req_is_nterror(req, &status)) {
812                 return status;
813         }
814         node_status = parse_node_status(
815                 mem_ctx, &state->packet->packet.nmb.answers->rdata[0],
816                 &num_names, extra);
817         if (node_status == NULL) {
818                 return NT_STATUS_NO_MEMORY;
819         }
820         *pnode_status = node_status;
821         *pnum_names = num_names;
822         return NT_STATUS_OK;
823 }
824
825 NTSTATUS node_status_query(TALLOC_CTX *mem_ctx, struct nmb_name *name,
826                            const struct sockaddr_storage *addr,
827                            struct node_status **pnode_status,
828                            int *pnum_names,
829                            struct node_status_extra *extra)
830 {
831         TALLOC_CTX *frame = talloc_stackframe();
832         struct tevent_context *ev;
833         struct tevent_req *req;
834         NTSTATUS status = NT_STATUS_NO_MEMORY;
835
836         ev = tevent_context_init(frame);
837         if (ev == NULL) {
838                 goto fail;
839         }
840         req = node_status_query_send(ev, ev, name, addr);
841         if (req == NULL) {
842                 goto fail;
843         }
844         if (!tevent_req_poll_ntstatus(req, ev, &status)) {
845                 goto fail;
846         }
847         status = node_status_query_recv(req, mem_ctx, pnode_status,
848                                         pnum_names, extra);
849  fail:
850         TALLOC_FREE(frame);
851         return status;
852 }
853
854 /****************************************************************************
855  Find the first type XX name in a node status reply - used for finding
856  a servers name given its IP. Return the matched name in *name.
857 **************************************************************************/
858
859 bool name_status_find(const char *q_name,
860                         int q_type,
861                         int type,
862                         const struct sockaddr_storage *to_ss,
863                         fstring name)
864 {
865         char addr[INET6_ADDRSTRLEN];
866         struct sockaddr_storage ss;
867         struct node_status *addrs = NULL;
868         struct nmb_name nname;
869         int count, i;
870         bool result = false;
871         NTSTATUS status;
872
873         if (lp_disable_netbios()) {
874                 DEBUG(5,("name_status_find(%s#%02x): netbios is disabled\n",
875                                         q_name, q_type));
876                 return False;
877         }
878
879         print_sockaddr(addr, sizeof(addr), to_ss);
880
881         DEBUG(10, ("name_status_find: looking up %s#%02x at %s\n", q_name,
882                    q_type, addr));
883
884         /* Check the cache first. */
885
886         if (namecache_status_fetch(q_name, q_type, type, to_ss, name)) {
887                 return True;
888         }
889
890         if (to_ss->ss_family != AF_INET) {
891                 /* Can't do node status to IPv6 */
892                 return false;
893         }
894
895         if (!interpret_string_addr(&ss, lp_socket_address(),
896                                 AI_NUMERICHOST|AI_PASSIVE)) {
897                 zero_sockaddr(&ss);
898         }
899
900         /* W2K PDC's seem not to respond to '*'#0. JRA */
901         make_nmb_name(&nname, q_name, q_type);
902         status = node_status_query(talloc_tos(), &nname, to_ss,
903                                    &addrs, &count, NULL);
904         if (!NT_STATUS_IS_OK(status)) {
905                 goto done;
906         }
907
908         for (i=0;i<count;i++) {
909                 /* Find first one of the requested type that's not a GROUP. */
910                 if (addrs[i].type == type && ! (addrs[i].flags & 0x80))
911                         break;
912         }
913         if (i == count)
914                 goto done;
915
916         pull_ascii_nstring(name, sizeof(fstring), addrs[i].name);
917
918         /* Store the result in the cache. */
919         /* but don't store an entry for 0x1c names here.  Here we have
920            a single host and DOMAIN<0x1c> names should be a list of hosts */
921
922         if ( q_type != 0x1c ) {
923                 namecache_status_store(q_name, q_type, type, to_ss, name);
924         }
925
926         result = true;
927
928  done:
929         TALLOC_FREE(addrs);
930
931         DEBUG(10, ("name_status_find: name %sfound", result ? "" : "not "));
932
933         if (result)
934                 DEBUGADD(10, (", name %s ip address is %s", name, addr));
935
936         DEBUG(10, ("\n"));
937
938         return result;
939 }
940
941 /*
942   comparison function used by sort_addr_list
943 */
944
945 static int addr_compare(const struct sockaddr_storage *ss1,
946                         const struct sockaddr_storage *ss2)
947 {
948         int max_bits1=0, max_bits2=0;
949         int num_interfaces = iface_count();
950         int i;
951
952         /* Sort IPv4 addresses first. */
953         if (ss1->ss_family != ss2->ss_family) {
954                 if (ss2->ss_family == AF_INET) {
955                         return 1;
956                 } else {
957                         return -1;
958                 }
959         }
960
961         /* Here we know both addresses are of the same
962          * family. */
963
964         for (i=0;i<num_interfaces;i++) {
965                 const struct sockaddr_storage *pss = iface_n_bcast(i);
966                 unsigned char *p_ss1 = NULL;
967                 unsigned char *p_ss2 = NULL;
968                 unsigned char *p_if = NULL;
969                 size_t len = 0;
970                 int bits1, bits2;
971
972                 if (pss->ss_family != ss1->ss_family) {
973                         /* Ignore interfaces of the wrong type. */
974                         continue;
975                 }
976                 if (pss->ss_family == AF_INET) {
977                         p_if = (unsigned char *)
978                                 &((const struct sockaddr_in *)pss)->sin_addr;
979                         p_ss1 = (unsigned char *)
980                                 &((const struct sockaddr_in *)ss1)->sin_addr;
981                         p_ss2 = (unsigned char *)
982                                 &((const struct sockaddr_in *)ss2)->sin_addr;
983                         len = 4;
984                 }
985 #if defined(HAVE_IPV6)
986                 if (pss->ss_family == AF_INET6) {
987                         p_if = (unsigned char *)
988                                 &((const struct sockaddr_in6 *)pss)->sin6_addr;
989                         p_ss1 = (unsigned char *)
990                                 &((const struct sockaddr_in6 *)ss1)->sin6_addr;
991                         p_ss2 = (unsigned char *)
992                                 &((const struct sockaddr_in6 *)ss2)->sin6_addr;
993                         len = 16;
994                 }
995 #endif
996                 if (!p_ss1 || !p_ss2 || !p_if || len == 0) {
997                         continue;
998                 }
999                 bits1 = matching_len_bits(p_ss1, p_if, len);
1000                 bits2 = matching_len_bits(p_ss2, p_if, len);
1001                 max_bits1 = MAX(bits1, max_bits1);
1002                 max_bits2 = MAX(bits2, max_bits2);
1003         }
1004
1005         /* Bias towards directly reachable IPs */
1006         if (iface_local((struct sockaddr *)ss1)) {
1007                 if (ss1->ss_family == AF_INET) {
1008                         max_bits1 += 32;
1009                 } else {
1010                         max_bits1 += 128;
1011                 }
1012         }
1013         if (iface_local((struct sockaddr *)ss2)) {
1014                 if (ss2->ss_family == AF_INET) {
1015                         max_bits2 += 32;
1016                 } else {
1017                         max_bits2 += 128;
1018                 }
1019         }
1020         return max_bits2 - max_bits1;
1021 }
1022
1023 /*******************************************************************
1024  compare 2 ldap IPs by nearness to our interfaces - used in qsort
1025 *******************************************************************/
1026
1027 int ip_service_compare(struct ip_service *ss1, struct ip_service *ss2)
1028 {
1029         int result;
1030
1031         if ((result = addr_compare(&ss1->ss, &ss2->ss)) != 0) {
1032                 return result;
1033         }
1034
1035         if (ss1->port > ss2->port) {
1036                 return 1;
1037         }
1038
1039         if (ss1->port < ss2->port) {
1040                 return -1;
1041         }
1042
1043         return 0;
1044 }
1045
1046 /*
1047   sort an IP list so that names that are close to one of our interfaces
1048   are at the top. This prevents the problem where a WINS server returns an IP
1049   that is not reachable from our subnet as the first match
1050 */
1051
1052 static void sort_addr_list(struct sockaddr_storage *sslist, int count)
1053 {
1054         if (count <= 1) {
1055                 return;
1056         }
1057
1058         TYPESAFE_QSORT(sslist, count, addr_compare);
1059 }
1060
1061 static void sort_service_list(struct ip_service *servlist, int count)
1062 {
1063         if (count <= 1) {
1064                 return;
1065         }
1066
1067         TYPESAFE_QSORT(servlist, count, ip_service_compare);
1068 }
1069
1070 /**********************************************************************
1071  Remove any duplicate address/port pairs in the list
1072  *********************************************************************/
1073
1074 static int remove_duplicate_addrs2(struct ip_service *iplist, int count )
1075 {
1076         int i, j;
1077
1078         DEBUG(10,("remove_duplicate_addrs2: "
1079                         "looking for duplicate address/port pairs\n"));
1080
1081         /* one loop to remove duplicates */
1082         for ( i=0; i<count; i++ ) {
1083                 if ( is_zero_addr(&iplist[i].ss)) {
1084                         continue;
1085                 }
1086
1087                 for ( j=i+1; j<count; j++ ) {
1088                         if (sockaddr_equal((struct sockaddr *)&iplist[i].ss, (struct sockaddr *)&iplist[j].ss) &&
1089                                         iplist[i].port == iplist[j].port) {
1090                                 zero_sockaddr(&iplist[j].ss);
1091                         }
1092                 }
1093         }
1094
1095         /* one loop to clean up any holes we left */
1096         /* first ip should never be a zero_ip() */
1097         for (i = 0; i<count; ) {
1098                 if (is_zero_addr(&iplist[i].ss) ) {
1099                         if (i != count-1) {
1100                                 memmove(&iplist[i], &iplist[i+1],
1101                                         (count - i - 1)*sizeof(iplist[i]));
1102                         }
1103                         count--;
1104                         continue;
1105                 }
1106                 i++;
1107         }
1108
1109         return count;
1110 }
1111
1112 static bool prioritize_ipv4_list(struct ip_service *iplist, int count)
1113 {
1114         TALLOC_CTX *frame = talloc_stackframe();
1115         struct ip_service *iplist_new = TALLOC_ARRAY(frame, struct ip_service, count);
1116         int i, j;
1117
1118         if (iplist_new == NULL) {
1119                 TALLOC_FREE(frame);
1120                 return false;
1121         }
1122
1123         j = 0;
1124
1125         /* Copy IPv4 first. */
1126         for (i = 0; i < count; i++) {
1127                 if (iplist[i].ss.ss_family == AF_INET) {
1128                         iplist_new[j++] = iplist[i];
1129                 }
1130         }
1131
1132         /* Copy IPv6. */
1133         for (i = 0; i < count; i++) {
1134                 if (iplist[i].ss.ss_family != AF_INET) {
1135                         iplist_new[j++] = iplist[i];
1136                 }
1137         }
1138
1139         memcpy(iplist, iplist_new, sizeof(struct ip_service)*count);
1140         TALLOC_FREE(frame);
1141         return true;
1142 }
1143
1144 /****************************************************************************
1145  Do a netbios name query to find someones IP.
1146  Returns an array of IP addresses or NULL if none.
1147  *count will be set to the number of addresses returned.
1148  *timed_out is set if we failed by timing out
1149 ****************************************************************************/
1150
1151 struct name_query_state {
1152         struct sockaddr_storage my_addr;
1153         struct sockaddr_storage addr;
1154         bool bcast;
1155
1156
1157         uint8_t buf[1024];
1158         ssize_t buflen;
1159
1160         NTSTATUS validate_error;
1161         uint8_t flags;
1162
1163         struct sockaddr_storage *addrs;
1164         int num_addrs;
1165 };
1166
1167 static bool name_query_validator(struct packet_struct *p, void *private_data);
1168 static void name_query_done(struct tevent_req *subreq);
1169
1170 struct tevent_req *name_query_send(TALLOC_CTX *mem_ctx,
1171                                    struct tevent_context *ev,
1172                                    const char *name, int name_type,
1173                                    bool bcast, bool recurse,
1174                                    const struct sockaddr_storage *addr)
1175 {
1176         struct tevent_req *req, *subreq;
1177         struct name_query_state *state;
1178         struct packet_struct p;
1179         struct nmb_packet *nmb = &p.packet.nmb;
1180         struct sockaddr_in *in_addr;
1181
1182         req = tevent_req_create(mem_ctx, &state, struct name_query_state);
1183         if (req == NULL) {
1184                 return NULL;
1185         }
1186         state->bcast = bcast;
1187
1188         if (addr->ss_family != AF_INET) {
1189                 /* Can't do node status to IPv6 */
1190                 tevent_req_nterror(req, NT_STATUS_INVALID_ADDRESS);
1191                 return tevent_req_post(req, ev);
1192         }
1193
1194         if (lp_disable_netbios()) {
1195                 DEBUG(5,("name_query(%s#%02x): netbios is disabled\n",
1196                                         name, name_type));
1197                 tevent_req_nterror(req, NT_STATUS_NOT_SUPPORTED);
1198                 return tevent_req_post(req, ev);
1199         }
1200
1201         state->addr = *addr;
1202         in_addr = (struct sockaddr_in *)(void *)&state->addr;
1203         in_addr->sin_port = htons(NMB_PORT);
1204
1205         if (!interpret_string_addr(&state->my_addr, lp_socket_address(),
1206                                    AI_NUMERICHOST|AI_PASSIVE)) {
1207                 zero_sockaddr(&state->my_addr);
1208         }
1209
1210         ZERO_STRUCT(p);
1211         nmb->header.name_trn_id = generate_trn_id();
1212         nmb->header.opcode = 0;
1213         nmb->header.response = false;
1214         nmb->header.nm_flags.bcast = bcast;
1215         nmb->header.nm_flags.recursion_available = false;
1216         nmb->header.nm_flags.recursion_desired = recurse;
1217         nmb->header.nm_flags.trunc = false;
1218         nmb->header.nm_flags.authoritative = false;
1219         nmb->header.rcode = 0;
1220         nmb->header.qdcount = 1;
1221         nmb->header.ancount = 0;
1222         nmb->header.nscount = 0;
1223         nmb->header.arcount = 0;
1224
1225         make_nmb_name(&nmb->question.question_name,name,name_type);
1226
1227         nmb->question.question_type = 0x20;
1228         nmb->question.question_class = 0x1;
1229
1230         state->buflen = build_packet((char *)state->buf, sizeof(state->buf),
1231                                      &p);
1232         if (state->buflen == 0) {
1233                 tevent_req_nterror(req, NT_STATUS_INTERNAL_ERROR);
1234                 DEBUG(10, ("build_packet failed\n"));
1235                 return tevent_req_post(req, ev);
1236         }
1237
1238         subreq = nb_trans_send(state, ev, &state->my_addr, &state->addr, bcast,
1239                                state->buf, state->buflen,
1240                                NMB_PACKET, nmb->header.name_trn_id,
1241                                name_query_validator, state);
1242         if (tevent_req_nomem(subreq, req)) {
1243                 DEBUG(10, ("nb_trans_send failed\n"));
1244                 return tevent_req_post(req, ev);
1245         }
1246         tevent_req_set_callback(subreq, name_query_done, req);
1247         return req;
1248 }
1249
1250 static bool name_query_validator(struct packet_struct *p, void *private_data)
1251 {
1252         struct name_query_state *state = talloc_get_type_abort(
1253                 private_data, struct name_query_state);
1254         struct nmb_packet *nmb = &p->packet.nmb;
1255         struct sockaddr_storage *tmp_addrs;
1256         int i;
1257
1258         debug_nmb_packet(p);
1259
1260         /*
1261          * If we get a Negative Name Query Response from a WINS
1262          * server, we should report it and give up.
1263          */
1264         if( 0 == nmb->header.opcode     /* A query response   */
1265             && !state->bcast            /* from a WINS server */
1266             && nmb->header.rcode        /* Error returned     */
1267                 ) {
1268
1269                 if( DEBUGLVL( 3 ) ) {
1270                         /* Only executed if DEBUGLEVEL >= 3 */
1271                         dbgtext( "Negative name query "
1272                                  "response, rcode 0x%02x: ",
1273                                  nmb->header.rcode );
1274                         switch( nmb->header.rcode ) {
1275                         case 0x01:
1276                                 dbgtext("Request was invalidly formatted.\n");
1277                                 break;
1278                         case 0x02:
1279                                 dbgtext("Problem with NBNS, cannot process "
1280                                         "name.\n");
1281                                 break;
1282                         case 0x03:
1283                                 dbgtext("The name requested does not "
1284                                         "exist.\n");
1285                                 break;
1286                         case 0x04:
1287                                 dbgtext("Unsupported request error.\n");
1288                                 break;
1289                         case 0x05:
1290                                 dbgtext("Query refused error.\n");
1291                                 break;
1292                         default:
1293                                 dbgtext("Unrecognized error code.\n" );
1294                                 break;
1295                         }
1296                 }
1297
1298                 /*
1299                  * We accept this packet as valid, but tell the upper
1300                  * layers that it's a negative response.
1301                  */
1302                 state->validate_error = NT_STATUS_NOT_FOUND;
1303                 return true;
1304         }
1305
1306         if (nmb->header.opcode != 0 ||
1307             nmb->header.nm_flags.bcast ||
1308             nmb->header.rcode ||
1309             !nmb->header.ancount) {
1310                 /*
1311                  * XXXX what do we do with this? Could be a redirect,
1312                  * but we'll discard it for the moment.
1313                  */
1314                 return false;
1315         }
1316
1317         tmp_addrs = TALLOC_REALLOC_ARRAY(
1318                 state, state->addrs, struct sockaddr_storage,
1319                 state->num_addrs + nmb->answers->rdlength/6);
1320         if (tmp_addrs == NULL) {
1321                 state->validate_error = NT_STATUS_NO_MEMORY;
1322                 return true;
1323         }
1324         state->addrs = tmp_addrs;
1325
1326         DEBUG(2,("Got a positive name query response "
1327                  "from %s ( ", inet_ntoa(p->ip)));
1328
1329         for (i=0; i<nmb->answers->rdlength/6; i++) {
1330                 struct in_addr ip;
1331                 putip((char *)&ip,&nmb->answers->rdata[2+i*6]);
1332                 in_addr_to_sockaddr_storage(
1333                         &state->addrs[state->num_addrs], ip);
1334                 DEBUGADD(2,("%s ",inet_ntoa(ip)));
1335                 state->num_addrs += 1;
1336         }
1337         DEBUGADD(2,(")\n"));
1338
1339         /* We add the flags back ... */
1340         if (nmb->header.response)
1341                 state->flags |= NM_FLAGS_RS;
1342         if (nmb->header.nm_flags.authoritative)
1343                 state->flags |= NM_FLAGS_AA;
1344         if (nmb->header.nm_flags.trunc)
1345                 state->flags |= NM_FLAGS_TC;
1346         if (nmb->header.nm_flags.recursion_desired)
1347                 state->flags |= NM_FLAGS_RD;
1348         if (nmb->header.nm_flags.recursion_available)
1349                 state->flags |= NM_FLAGS_RA;
1350         if (nmb->header.nm_flags.bcast)
1351                 state->flags |= NM_FLAGS_B;
1352
1353         if (state->bcast) {
1354                 /*
1355                  * We have to collect all entries coming in from
1356                  * broadcast queries
1357                  */
1358                 return false;
1359         }
1360         /*
1361          * WINS responses are accepted when they are received
1362          */
1363         return true;
1364 }
1365
1366 static void name_query_done(struct tevent_req *subreq)
1367 {
1368         struct tevent_req *req = tevent_req_callback_data(
1369                 subreq, struct tevent_req);
1370         struct name_query_state *state = tevent_req_data(
1371                 req, struct name_query_state);
1372         NTSTATUS status;
1373         struct packet_struct *p = NULL;
1374
1375         status = nb_trans_recv(subreq, &p);
1376         TALLOC_FREE(subreq);
1377         if (tevent_req_nterror(req, status)) {
1378                 return;
1379         }
1380         if (!NT_STATUS_IS_OK(state->validate_error)) {
1381                 tevent_req_nterror(req, state->validate_error);
1382                 return;
1383         }
1384         if (p != NULL) {
1385                 /*
1386                  * Free the packet here, we've collected the response in the
1387                  * validator
1388                  */
1389                 free_packet(p);
1390         }
1391         tevent_req_done(req);
1392 }
1393
1394 NTSTATUS name_query_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
1395                          struct sockaddr_storage **addrs, int *num_addrs,
1396                          uint8_t *flags)
1397 {
1398         struct name_query_state *state = tevent_req_data(
1399                 req, struct name_query_state);
1400         NTSTATUS status;
1401
1402         if (tevent_req_is_nterror(req, &status)
1403             && !NT_STATUS_EQUAL(status, NT_STATUS_IO_TIMEOUT)) {
1404                 return status;
1405         }
1406         if (state->num_addrs == 0) {
1407                 return NT_STATUS_NOT_FOUND;
1408         }
1409         *addrs = talloc_move(mem_ctx, &state->addrs);
1410         sort_addr_list(*addrs, state->num_addrs);
1411         *num_addrs = state->num_addrs;
1412         if (flags != NULL) {
1413                 *flags = state->flags;
1414         }
1415         return NT_STATUS_OK;
1416 }
1417
1418 NTSTATUS name_query(const char *name, int name_type,
1419                     bool bcast, bool recurse,
1420                     const struct sockaddr_storage *to_ss,
1421                     TALLOC_CTX *mem_ctx,
1422                     struct sockaddr_storage **addrs,
1423                     int *num_addrs, uint8_t *flags)
1424 {
1425         TALLOC_CTX *frame = talloc_stackframe();
1426         struct tevent_context *ev;
1427         struct tevent_req *req;
1428         struct timeval timeout;
1429         NTSTATUS status = NT_STATUS_NO_MEMORY;
1430
1431         ev = tevent_context_init(frame);
1432         if (ev == NULL) {
1433                 goto fail;
1434         }
1435         req = name_query_send(ev, ev, name, name_type, bcast, recurse, to_ss);
1436         if (req == NULL) {
1437                 goto fail;
1438         }
1439         if (bcast) {
1440                 timeout = timeval_current_ofs(0, 250000);
1441         } else {
1442                 timeout = timeval_current_ofs(2, 0);
1443         }
1444         if (!tevent_req_set_endtime(req, ev, timeout)) {
1445                 goto fail;
1446         }
1447         if (!tevent_req_poll_ntstatus(req, ev, &status)) {
1448                 goto fail;
1449         }
1450         status = name_query_recv(req, mem_ctx, addrs, num_addrs, flags);
1451  fail:
1452         TALLOC_FREE(frame);
1453         return status;
1454 }
1455
1456 /********************************************************
1457  convert an array if struct sockaddr_storage to struct ip_service
1458  return false on failure.  Port is set to PORT_NONE;
1459 *********************************************************/
1460
1461 static bool convert_ss2service(struct ip_service **return_iplist,
1462                 const struct sockaddr_storage *ss_list,
1463                 int count)
1464 {
1465         int i;
1466
1467         if ( count==0 || !ss_list )
1468                 return False;
1469
1470         /* copy the ip address; port will be PORT_NONE */
1471         if ((*return_iplist = SMB_MALLOC_ARRAY(struct ip_service, count)) ==
1472                         NULL) {
1473                 DEBUG(0,("convert_ip2service: malloc failed "
1474                         "for %d enetries!\n", count ));
1475                 return False;
1476         }
1477
1478         for ( i=0; i<count; i++ ) {
1479                 (*return_iplist)[i].ss   = ss_list[i];
1480                 (*return_iplist)[i].port = PORT_NONE;
1481         }
1482
1483         return true;
1484 }
1485
1486 /********************************************************
1487  Resolve via "bcast" method.
1488 *********************************************************/
1489
1490 NTSTATUS name_resolve_bcast(const char *name,
1491                         int name_type,
1492                         struct ip_service **return_iplist,
1493                         int *return_count)
1494 {
1495         int i;
1496         int num_interfaces = iface_count();
1497         struct sockaddr_storage *ss_list;
1498         NTSTATUS status = NT_STATUS_NOT_FOUND;
1499
1500         if (lp_disable_netbios()) {
1501                 DEBUG(5,("name_resolve_bcast(%s#%02x): netbios is disabled\n",
1502                                         name, name_type));
1503                 return NT_STATUS_INVALID_PARAMETER;
1504         }
1505
1506         *return_iplist = NULL;
1507         *return_count = 0;
1508
1509         /*
1510          * "bcast" means do a broadcast lookup on all the local interfaces.
1511          */
1512
1513         DEBUG(3,("name_resolve_bcast: Attempting broadcast lookup "
1514                 "for name %s<0x%x>\n", name, name_type));
1515
1516         /*
1517          * Lookup the name on all the interfaces, return on
1518          * the first successful match.
1519          */
1520         for( i = num_interfaces-1; i >= 0; i--) {
1521                 const struct sockaddr_storage *pss = iface_n_bcast(i);
1522
1523                 /* Done this way to fix compiler error on IRIX 5.x */
1524                 if (!pss) {
1525                         continue;
1526                 }
1527                 status = name_query(name, name_type, true, true, pss,
1528                                     talloc_tos(), &ss_list, return_count,
1529                                     NULL);
1530                 if (NT_STATUS_IS_OK(status)) {
1531                         goto success;
1532                 }
1533         }
1534
1535         /* failed - no response */
1536
1537         return status;
1538
1539 success:
1540
1541         if (!convert_ss2service(return_iplist, ss_list, *return_count) )
1542                 status = NT_STATUS_NO_MEMORY;
1543
1544         TALLOC_FREE(ss_list);
1545         return status;
1546 }
1547
1548 /********************************************************
1549  Resolve via "wins" method.
1550 *********************************************************/
1551
1552 NTSTATUS resolve_wins(const char *name,
1553                 int name_type,
1554                 struct ip_service **return_iplist,
1555                 int *return_count)
1556 {
1557         int t, i;
1558         char **wins_tags;
1559         struct sockaddr_storage src_ss, *ss_list = NULL;
1560         struct in_addr src_ip;
1561         NTSTATUS status;
1562
1563         if (lp_disable_netbios()) {
1564                 DEBUG(5,("resolve_wins(%s#%02x): netbios is disabled\n",
1565                                         name, name_type));
1566                 return NT_STATUS_INVALID_PARAMETER;
1567         }
1568
1569         *return_iplist = NULL;
1570         *return_count = 0;
1571
1572         DEBUG(3,("resolve_wins: Attempting wins lookup for name %s<0x%x>\n",
1573                                 name, name_type));
1574
1575         if (wins_srv_count() < 1) {
1576                 DEBUG(3,("resolve_wins: WINS server resolution selected "
1577                         "and no WINS servers listed.\n"));
1578                 return NT_STATUS_INVALID_PARAMETER;
1579         }
1580
1581         /* we try a lookup on each of the WINS tags in turn */
1582         wins_tags = wins_srv_tags();
1583
1584         if (!wins_tags) {
1585                 /* huh? no tags?? give up in disgust */
1586                 return NT_STATUS_INVALID_PARAMETER;
1587         }
1588
1589         /* the address we will be sending from */
1590         if (!interpret_string_addr(&src_ss, lp_socket_address(),
1591                                 AI_NUMERICHOST|AI_PASSIVE)) {
1592                 zero_sockaddr(&src_ss);
1593         }
1594
1595         if (src_ss.ss_family != AF_INET) {
1596                 char addr[INET6_ADDRSTRLEN];
1597                 print_sockaddr(addr, sizeof(addr), &src_ss);
1598                 DEBUG(3,("resolve_wins: cannot receive WINS replies "
1599                         "on IPv6 address %s\n",
1600                         addr));
1601                 wins_srv_tags_free(wins_tags);
1602                 return NT_STATUS_INVALID_PARAMETER;
1603         }
1604
1605         src_ip = ((struct sockaddr_in *)&src_ss)->sin_addr;
1606
1607         /* in the worst case we will try every wins server with every
1608            tag! */
1609         for (t=0; wins_tags && wins_tags[t]; t++) {
1610                 int srv_count = wins_srv_count_tag(wins_tags[t]);
1611                 for (i=0; i<srv_count; i++) {
1612                         struct sockaddr_storage wins_ss;
1613                         struct in_addr wins_ip;
1614
1615                         wins_ip = wins_srv_ip_tag(wins_tags[t], src_ip);
1616
1617                         if (global_in_nmbd && ismyip_v4(wins_ip)) {
1618                                 /* yikes! we'll loop forever */
1619                                 continue;
1620                         }
1621
1622                         /* skip any that have been unresponsive lately */
1623                         if (wins_srv_is_dead(wins_ip, src_ip)) {
1624                                 continue;
1625                         }
1626
1627                         DEBUG(3,("resolve_wins: using WINS server %s "
1628                                 "and tag '%s'\n",
1629                                 inet_ntoa(wins_ip), wins_tags[t]));
1630
1631                         in_addr_to_sockaddr_storage(&wins_ss, wins_ip);
1632                         status = name_query(name,
1633                                                 name_type,
1634                                                 false,
1635                                                 true,
1636                                                 &wins_ss,
1637                                                 talloc_tos(),
1638                                                 &ss_list,
1639                                                 return_count,
1640                                                 NULL);
1641
1642                         /* exit loop if we got a list of addresses */
1643
1644                         if (NT_STATUS_IS_OK(status)) {
1645                                 goto success;
1646                         }
1647
1648                         if (NT_STATUS_EQUAL(status,
1649                                             NT_STATUS_IO_TIMEOUT)) {
1650                                 /* Timed out waiting for WINS server to
1651                                  * respond.
1652                                  * Mark it dead. */
1653                                 wins_srv_died(wins_ip, src_ip);
1654                         } else {
1655                                 /* The name definitely isn't in this
1656                                    group of WINS servers.
1657                                    goto the next group  */
1658                                 break;
1659                         }
1660                 }
1661         }
1662
1663         wins_srv_tags_free(wins_tags);
1664         return NT_STATUS_NO_LOGON_SERVERS;
1665
1666 success:
1667
1668         status = NT_STATUS_OK;
1669         if (!convert_ss2service(return_iplist, ss_list, *return_count))
1670                 status = NT_STATUS_INVALID_PARAMETER;
1671
1672         TALLOC_FREE(ss_list);
1673         wins_srv_tags_free(wins_tags);
1674
1675         return status;
1676 }
1677
1678 /********************************************************
1679  Resolve via "lmhosts" method.
1680 *********************************************************/
1681
1682 static NTSTATUS resolve_lmhosts(const char *name, int name_type,
1683                                 struct ip_service **return_iplist,
1684                                 int *return_count)
1685 {
1686         /*
1687          * "lmhosts" means parse the local lmhosts file.
1688          */
1689         struct sockaddr_storage *ss_list;
1690         NTSTATUS status = NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
1691         TALLOC_CTX *ctx = NULL;
1692
1693         *return_iplist = NULL;
1694         *return_count = 0;
1695
1696         DEBUG(3,("resolve_lmhosts: "
1697                 "Attempting lmhosts lookup for name %s<0x%x>\n",
1698                 name, name_type));
1699
1700         ctx = talloc_init("resolve_lmhosts");
1701         if (!ctx) {
1702                 return NT_STATUS_NO_MEMORY;
1703         }
1704
1705         status = resolve_lmhosts_file_as_sockaddr(get_dyn_LMHOSTSFILE(), 
1706                                                   name, name_type, 
1707                                                   ctx, 
1708                                                   &ss_list, 
1709                                                   return_count);
1710         if (NT_STATUS_IS_OK(status)) {
1711                 if (convert_ss2service(return_iplist, 
1712                                        ss_list,
1713                                        *return_count)) {
1714                         talloc_free(ctx);
1715                         return NT_STATUS_OK;
1716                 } else {
1717                         talloc_free(ctx);
1718                         return NT_STATUS_NO_MEMORY;
1719                 }
1720         }
1721         talloc_free(ctx);
1722         return status;
1723 }
1724
1725
1726 /********************************************************
1727  Resolve via "hosts" method.
1728 *********************************************************/
1729
1730 static NTSTATUS resolve_hosts(const char *name, int name_type,
1731                               struct ip_service **return_iplist,
1732                               int *return_count)
1733 {
1734         /*
1735          * "host" means do a localhost, or dns lookup.
1736          */
1737         struct addrinfo hints;
1738         struct addrinfo *ailist = NULL;
1739         struct addrinfo *res = NULL;
1740         int ret = -1;
1741         int i = 0;
1742         const char *dns_hosts_file;
1743
1744         if ( name_type != 0x20 && name_type != 0x0) {
1745                 DEBUG(5, ("resolve_hosts: not appropriate "
1746                         "for name type <0x%x>\n",
1747                         name_type));
1748                 return NT_STATUS_INVALID_PARAMETER;
1749         }
1750
1751         *return_iplist = NULL;
1752         *return_count = 0;
1753
1754         DEBUG(3,("resolve_hosts: Attempting host lookup for name %s<0x%x>\n",
1755                                 name, name_type));
1756
1757         ZERO_STRUCT(hints);
1758         /* By default make sure it supports TCP. */
1759         hints.ai_socktype = SOCK_STREAM;
1760         hints.ai_flags = AI_ADDRCONFIG;
1761
1762 #if !defined(HAVE_IPV6)
1763         /* Unless we have IPv6, we really only want IPv4 addresses back. */
1764         hints.ai_family = AF_INET;
1765 #endif
1766
1767         dns_hosts_file = lp_parm_const_string(-1, "resolv", "host file", NULL);
1768         if (dns_hosts_file) {
1769                 struct sockaddr_storage *ss_list;
1770                 NTSTATUS status;
1771                 TALLOC_CTX *ctx = talloc_stackframe();
1772                 if (!ctx) {
1773                         return NT_STATUS_NO_MEMORY;
1774                 }
1775
1776                 status = resolve_dns_hosts_file_as_sockaddr(dns_hosts_file, name, false,
1777                                                             ctx, &ss_list, return_count);
1778                 if (NT_STATUS_IS_OK(status)) {
1779                         if (convert_ss2service(return_iplist,
1780                                                ss_list,
1781                                                *return_count)) {
1782                                 talloc_free(ctx);
1783                                 return NT_STATUS_OK;
1784                         } else {
1785                                 talloc_free(ctx);
1786                                 return NT_STATUS_NO_MEMORY;
1787                         }
1788                 }
1789                 talloc_free(ctx);
1790                 return NT_STATUS_UNSUCCESSFUL;
1791         }
1792
1793         ret = getaddrinfo(name,
1794                         NULL,
1795                         &hints,
1796                         &ailist);
1797         if (ret) {
1798                 DEBUG(3,("resolve_hosts: getaddrinfo failed for name %s [%s]\n",
1799                         name,
1800                         gai_strerror(ret) ));
1801         }
1802
1803         for (res = ailist; res; res = res->ai_next) {
1804                 struct sockaddr_storage ss;
1805
1806                 if (!res->ai_addr || res->ai_addrlen == 0) {
1807                         continue;
1808                 }
1809
1810                 ZERO_STRUCT(ss);
1811                 memcpy(&ss, res->ai_addr, res->ai_addrlen);
1812
1813                 *return_count += 1;
1814
1815                 *return_iplist = SMB_REALLOC_ARRAY(*return_iplist,
1816                                                 struct ip_service,
1817                                                 *return_count);
1818                 if (!*return_iplist) {
1819                         DEBUG(3,("resolve_hosts: malloc fail !\n"));
1820                         freeaddrinfo(ailist);
1821                         return NT_STATUS_NO_MEMORY;
1822                 }
1823                 (*return_iplist)[i].ss = ss;
1824                 (*return_iplist)[i].port = PORT_NONE;
1825                 i++;
1826         }
1827         if (ailist) {
1828                 freeaddrinfo(ailist);
1829         }
1830         if (*return_count) {
1831                 return NT_STATUS_OK;
1832         }
1833         return NT_STATUS_UNSUCCESSFUL;
1834 }
1835
1836 /********************************************************
1837  Resolve via "ADS" method.
1838 *********************************************************/
1839
1840 /* Special name type used to cause a _kerberos DNS lookup. */
1841 #define KDC_NAME_TYPE 0xDCDC
1842
1843 static NTSTATUS resolve_ads(const char *name,
1844                             int name_type,
1845                             const char *sitename,
1846                             struct ip_service **return_iplist,
1847                             int *return_count)
1848 {
1849         int                     i, j;
1850         NTSTATUS                status;
1851         TALLOC_CTX              *ctx;
1852         struct dns_rr_srv       *dcs = NULL;
1853         int                     numdcs = 0;
1854         int                     numaddrs = 0;
1855
1856         if ((name_type != 0x1c) && (name_type != KDC_NAME_TYPE) &&
1857             (name_type != 0x1b)) {
1858                 return NT_STATUS_INVALID_PARAMETER;
1859         }
1860
1861         if ( (ctx = talloc_init("resolve_ads")) == NULL ) {
1862                 DEBUG(0,("resolve_ads: talloc_init() failed!\n"));
1863                 return NT_STATUS_NO_MEMORY;
1864         }
1865
1866         /* The DNS code needs fixing to find IPv6 addresses... JRA. */
1867
1868         switch (name_type) {
1869                 case 0x1b:
1870                         DEBUG(5,("resolve_ads: Attempting to resolve "
1871                                  "PDC for %s using DNS\n", name));
1872                         status = ads_dns_query_pdc(ctx, name, &dcs, &numdcs);
1873                         break;
1874
1875                 case 0x1c:
1876                         DEBUG(5,("resolve_ads: Attempting to resolve "
1877                                  "DCs for %s using DNS\n", name));
1878                         status = ads_dns_query_dcs(ctx, name, sitename, &dcs,
1879                                                    &numdcs);
1880                         break;
1881                 case KDC_NAME_TYPE:
1882                         DEBUG(5,("resolve_ads: Attempting to resolve "
1883                                  "KDCs for %s using DNS\n", name));
1884                         status = ads_dns_query_kdcs(ctx, name, sitename, &dcs,
1885                                                     &numdcs);
1886                         break;
1887                 default:
1888                         status = NT_STATUS_INVALID_PARAMETER;
1889                         break;
1890         }
1891
1892         if ( !NT_STATUS_IS_OK( status ) ) {
1893                 talloc_destroy(ctx);
1894                 return status;
1895         }
1896
1897         for (i=0;i<numdcs;i++) {
1898                 numaddrs += MAX(dcs[i].num_ips,1);
1899         }
1900
1901         if ((*return_iplist = SMB_MALLOC_ARRAY(struct ip_service, numaddrs)) ==
1902                         NULL ) {
1903                 DEBUG(0,("resolve_ads: malloc failed for %d entries\n",
1904                                         numaddrs ));
1905                 talloc_destroy(ctx);
1906                 return NT_STATUS_NO_MEMORY;
1907         }
1908
1909         /* now unroll the list of IP addresses */
1910
1911         *return_count = 0;
1912         i = 0;
1913         j = 0;
1914         while ( i < numdcs && (*return_count<numaddrs) ) {
1915                 struct ip_service *r = &(*return_iplist)[*return_count];
1916
1917                 r->port = dcs[i].port;
1918
1919                 /* If we don't have an IP list for a name, lookup it up */
1920
1921                 if (!dcs[i].ss_s) {
1922                         interpret_string_addr(&r->ss, dcs[i].hostname, 0);
1923                         i++;
1924                         j = 0;
1925                 } else {
1926                         /* use the IP addresses from the SRV sresponse */
1927
1928                         if ( j >= dcs[i].num_ips ) {
1929                                 i++;
1930                                 j = 0;
1931                                 continue;
1932                         }
1933
1934                         r->ss = dcs[i].ss_s[j];
1935                         j++;
1936                 }
1937
1938                 /* make sure it is a valid IP.  I considered checking the
1939                  * negative connection cache, but this is the wrong place
1940                  * for it. Maybe only as a hack. After think about it, if
1941                  * all of the IP addresses returned from DNS are dead, what
1942                  * hope does a netbios name lookup have ? The standard reason
1943                  * for falling back to netbios lookups is that our DNS server
1944                  * doesn't know anything about the DC's   -- jerry */
1945
1946                 if (!is_zero_addr(&r->ss)) {
1947                         (*return_count)++;
1948                 }
1949         }
1950
1951         talloc_destroy(ctx);
1952         return NT_STATUS_OK;
1953 }
1954
1955 /*******************************************************************
1956  Internal interface to resolve a name into an IP address.
1957  Use this function if the string is either an IP address, DNS
1958  or host name or NetBIOS name. This uses the name switch in the
1959  smb.conf to determine the order of name resolution.
1960
1961  Added support for ip addr/port to support ADS ldap servers.
1962  the only place we currently care about the port is in the
1963  resolve_hosts() when looking up DC's via SRV RR entries in DNS
1964 **********************************************************************/
1965
1966 NTSTATUS internal_resolve_name(const char *name,
1967                                 int name_type,
1968                                 const char *sitename,
1969                                 struct ip_service **return_iplist,
1970                                 int *return_count,
1971                                 const char *resolve_order)
1972 {
1973         char *tok;
1974         const char *ptr;
1975         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
1976         int i;
1977         TALLOC_CTX *frame = NULL;
1978
1979         *return_iplist = NULL;
1980         *return_count = 0;
1981
1982         DEBUG(10, ("internal_resolve_name: looking up %s#%x (sitename %s)\n",
1983                         name, name_type, sitename ? sitename : "(null)"));
1984
1985         if (is_ipaddress(name)) {
1986                 if ((*return_iplist = SMB_MALLOC_P(struct ip_service)) ==
1987                                 NULL) {
1988                         DEBUG(0,("internal_resolve_name: malloc fail !\n"));
1989                         return NT_STATUS_NO_MEMORY;
1990                 }
1991
1992                 /* ignore the port here */
1993                 (*return_iplist)->port = PORT_NONE;
1994
1995                 /* if it's in the form of an IP address then get the lib to interpret it */
1996                 if (!interpret_string_addr(&(*return_iplist)->ss,
1997                                         name, AI_NUMERICHOST)) {
1998                         DEBUG(1,("internal_resolve_name: interpret_string_addr "
1999                                 "failed on %s\n",
2000                                 name));
2001                         SAFE_FREE(*return_iplist);
2002                         return NT_STATUS_INVALID_PARAMETER;
2003                 }
2004                 *return_count = 1;
2005                 return NT_STATUS_OK;
2006         }
2007
2008         /* Check name cache */
2009
2010         if (namecache_fetch(name, name_type, return_iplist, return_count)) {
2011                 /* This could be a negative response */
2012                 if (*return_count > 0) {
2013                         return NT_STATUS_OK;
2014                 } else {
2015                         return NT_STATUS_UNSUCCESSFUL;
2016                 }
2017         }
2018
2019         /* set the name resolution order */
2020
2021         if (strcmp( resolve_order, "NULL") == 0) {
2022                 DEBUG(8,("internal_resolve_name: all lookups disabled\n"));
2023                 return NT_STATUS_INVALID_PARAMETER;
2024         }
2025
2026         if (!resolve_order[0]) {
2027                 ptr = "host";
2028         } else {
2029                 ptr = resolve_order;
2030         }
2031
2032         /* iterate through the name resolution backends */
2033
2034         frame = talloc_stackframe();
2035         while (next_token_talloc(frame, &ptr, &tok, LIST_SEP)) {
2036                 if((strequal(tok, "host") || strequal(tok, "hosts"))) {
2037                         status = resolve_hosts(name, name_type, return_iplist,
2038                                                return_count);
2039                         if (NT_STATUS_IS_OK(status)) {
2040                                 goto done;
2041                         }
2042                 } else if(strequal( tok, "kdc")) {
2043                         /* deal with KDC_NAME_TYPE names here.
2044                          * This will result in a SRV record lookup */
2045                         status = resolve_ads(name, KDC_NAME_TYPE, sitename,
2046                                              return_iplist, return_count);
2047                         if (NT_STATUS_IS_OK(status)) {
2048                                 /* Ensure we don't namecache
2049                                  * this with the KDC port. */
2050                                 name_type = KDC_NAME_TYPE;
2051                                 goto done;
2052                         }
2053                 } else if(strequal( tok, "ads")) {
2054                         /* deal with 0x1c and 0x1b names here.
2055                          * This will result in a SRV record lookup */
2056                         status = resolve_ads(name, name_type, sitename,
2057                                              return_iplist, return_count);
2058                         if (NT_STATUS_IS_OK(status)) {
2059                                 goto done;
2060                         }
2061                 } else if(strequal( tok, "lmhosts")) {
2062                         status = resolve_lmhosts(name, name_type,
2063                                                  return_iplist, return_count);
2064                         if (NT_STATUS_IS_OK(status)) {
2065                                 goto done;
2066                         }
2067                 } else if(strequal( tok, "wins")) {
2068                         /* don't resolve 1D via WINS */
2069                         if (name_type != 0x1D) {
2070                                 status = resolve_wins(name, name_type,
2071                                                       return_iplist,
2072                                                       return_count);
2073                                 if (NT_STATUS_IS_OK(status)) {
2074                                         goto done;
2075                                 }
2076                         }
2077                 } else if(strequal( tok, "bcast")) {
2078                         status = name_resolve_bcast(name, name_type,
2079                                                     return_iplist,
2080                                                     return_count);
2081                         if (NT_STATUS_IS_OK(status)) {
2082                                 goto done;
2083                         }
2084                 } else {
2085                         DEBUG(0,("resolve_name: unknown name switch type %s\n",
2086                                 tok));
2087                 }
2088         }
2089
2090         /* All of the resolve_* functions above have returned false. */
2091
2092         TALLOC_FREE(frame);
2093         SAFE_FREE(*return_iplist);
2094         *return_count = 0;
2095
2096         return NT_STATUS_UNSUCCESSFUL;
2097
2098   done:
2099
2100         /* Remove duplicate entries.  Some queries, notably #1c (domain
2101         controllers) return the PDC in iplist[0] and then all domain
2102         controllers including the PDC in iplist[1..n].  Iterating over
2103         the iplist when the PDC is down will cause two sets of timeouts. */
2104
2105         if ( *return_count ) {
2106                 *return_count = remove_duplicate_addrs2(*return_iplist,
2107                                         *return_count );
2108         }
2109
2110         /* Save in name cache */
2111         if ( DEBUGLEVEL >= 100 ) {
2112                 for (i = 0; i < *return_count && DEBUGLEVEL == 100; i++) {
2113                         char addr[INET6_ADDRSTRLEN];
2114                         print_sockaddr(addr, sizeof(addr),
2115                                         &(*return_iplist)[i].ss);
2116                         DEBUG(100, ("Storing name %s of type %d (%s:%d)\n",
2117                                         name,
2118                                         name_type,
2119                                         addr,
2120                                         (*return_iplist)[i].port));
2121                 }
2122         }
2123
2124         namecache_store(name, name_type, *return_count, *return_iplist);
2125
2126         /* Display some debugging info */
2127
2128         if ( DEBUGLEVEL >= 10 ) {
2129                 DEBUG(10, ("internal_resolve_name: returning %d addresses: ",
2130                                         *return_count));
2131
2132                 for (i = 0; i < *return_count; i++) {
2133                         char addr[INET6_ADDRSTRLEN];
2134                         print_sockaddr(addr, sizeof(addr),
2135                                         &(*return_iplist)[i].ss);
2136                         DEBUGADD(10, ("%s:%d ",
2137                                         addr,
2138                                         (*return_iplist)[i].port));
2139                 }
2140                 DEBUG(10, ("\n"));
2141         }
2142
2143         TALLOC_FREE(frame);
2144         return status;
2145 }
2146
2147 /********************************************************
2148  Internal interface to resolve a name into one IP address.
2149  Use this function if the string is either an IP address, DNS
2150  or host name or NetBIOS name. This uses the name switch in the
2151  smb.conf to determine the order of name resolution.
2152 *********************************************************/
2153
2154 bool resolve_name(const char *name,
2155                 struct sockaddr_storage *return_ss,
2156                 int name_type,
2157                 bool prefer_ipv4)
2158 {
2159         struct ip_service *ss_list = NULL;
2160         char *sitename = NULL;
2161         int count = 0;
2162
2163         if (is_ipaddress(name)) {
2164                 return interpret_string_addr(return_ss, name, AI_NUMERICHOST);
2165         }
2166
2167         sitename = sitename_fetch(lp_realm()); /* wild guess */
2168
2169         if (NT_STATUS_IS_OK(internal_resolve_name(name, name_type, sitename,
2170                                                   &ss_list, &count,
2171                                                   lp_name_resolve_order()))) {
2172                 int i;
2173
2174                 if (prefer_ipv4) {
2175                         for (i=0; i<count; i++) {
2176                                 if (!is_zero_addr(&ss_list[i].ss) &&
2177                                                 !is_broadcast_addr((struct sockaddr *)&ss_list[i].ss) &&
2178                                                 (ss_list[i].ss.ss_family == AF_INET)) {
2179                                         *return_ss = ss_list[i].ss;
2180                                         SAFE_FREE(ss_list);
2181                                         SAFE_FREE(sitename);
2182                                         return True;
2183                                 }
2184                         }
2185                 }
2186
2187                 /* only return valid addresses for TCP connections */
2188                 for (i=0; i<count; i++) {
2189                         if (!is_zero_addr(&ss_list[i].ss) &&
2190                                         !is_broadcast_addr((struct sockaddr *)&ss_list[i].ss)) {
2191                                 *return_ss = ss_list[i].ss;
2192                                 SAFE_FREE(ss_list);
2193                                 SAFE_FREE(sitename);
2194                                 return True;
2195                         }
2196                 }
2197         }
2198
2199         SAFE_FREE(ss_list);
2200         SAFE_FREE(sitename);
2201         return False;
2202 }
2203
2204 /********************************************************
2205  Internal interface to resolve a name into a list of IP addresses.
2206  Use this function if the string is either an IP address, DNS
2207  or host name or NetBIOS name. This uses the name switch in the
2208  smb.conf to determine the order of name resolution.
2209 *********************************************************/
2210
2211 NTSTATUS resolve_name_list(TALLOC_CTX *ctx,
2212                 const char *name,
2213                 int name_type,
2214                 struct sockaddr_storage **return_ss_arr,
2215                 unsigned int *p_num_entries)
2216 {
2217         struct ip_service *ss_list = NULL;
2218         char *sitename = NULL;
2219         int count = 0;
2220         int i;
2221         unsigned int num_entries;
2222         NTSTATUS status;
2223
2224         *p_num_entries = 0;
2225         *return_ss_arr = NULL;
2226
2227         if (is_ipaddress(name)) {
2228                 *return_ss_arr = TALLOC_P(ctx, struct sockaddr_storage);
2229                 if (!*return_ss_arr) {
2230                         return NT_STATUS_NO_MEMORY;
2231                 }
2232                 if (!interpret_string_addr(*return_ss_arr, name, AI_NUMERICHOST)) {
2233                         TALLOC_FREE(*return_ss_arr);
2234                         return NT_STATUS_BAD_NETWORK_NAME;
2235                 }
2236                 *p_num_entries = 1;
2237                 return NT_STATUS_OK;
2238         }
2239
2240         sitename = sitename_fetch(lp_realm()); /* wild guess */
2241
2242         status = internal_resolve_name(name, name_type, sitename,
2243                                                   &ss_list, &count,
2244                                                   lp_name_resolve_order());
2245         SAFE_FREE(sitename);
2246
2247         if (!NT_STATUS_IS_OK(status)) {
2248                 return status;
2249         }
2250
2251         /* only return valid addresses for TCP connections */
2252         for (i=0, num_entries = 0; i<count; i++) {
2253                 if (!is_zero_addr(&ss_list[i].ss) &&
2254                                 !is_broadcast_addr((struct sockaddr *)&ss_list[i].ss)) {
2255                         num_entries++;
2256                 }
2257         }
2258         if (num_entries == 0) {
2259                 SAFE_FREE(ss_list);
2260                 return NT_STATUS_BAD_NETWORK_NAME;
2261         }
2262
2263         *return_ss_arr = TALLOC_ARRAY(ctx,
2264                                 struct sockaddr_storage,
2265                                 num_entries);
2266         if (!(*return_ss_arr)) {
2267                 SAFE_FREE(ss_list);
2268                 return NT_STATUS_NO_MEMORY;
2269         }
2270
2271         for (i=0, num_entries = 0; i<count; i++) {
2272                 if (!is_zero_addr(&ss_list[i].ss) &&
2273                                 !is_broadcast_addr((struct sockaddr *)&ss_list[i].ss)) {
2274                         (*return_ss_arr)[num_entries++] = ss_list[i].ss;
2275                 }
2276         }
2277
2278         status = NT_STATUS_OK;
2279         *p_num_entries = num_entries;
2280
2281         SAFE_FREE(ss_list);
2282         return NT_STATUS_OK;
2283 }
2284
2285 /********************************************************
2286  Find the IP address of the master browser or DMB for a workgroup.
2287 *********************************************************/
2288
2289 bool find_master_ip(const char *group, struct sockaddr_storage *master_ss)
2290 {
2291         struct ip_service *ip_list = NULL;
2292         int count = 0;
2293         NTSTATUS status;
2294
2295         if (lp_disable_netbios()) {
2296                 DEBUG(5,("find_master_ip(%s): netbios is disabled\n", group));
2297                 return false;
2298         }
2299
2300         status = internal_resolve_name(group, 0x1D, NULL, &ip_list, &count,
2301                                        lp_name_resolve_order());
2302         if (NT_STATUS_IS_OK(status)) {
2303                 *master_ss = ip_list[0].ss;
2304                 SAFE_FREE(ip_list);
2305                 return true;
2306         }
2307
2308         status = internal_resolve_name(group, 0x1B, NULL, &ip_list, &count,
2309                                        lp_name_resolve_order());
2310         if (NT_STATUS_IS_OK(status)) {
2311                 *master_ss = ip_list[0].ss;
2312                 SAFE_FREE(ip_list);
2313                 return true;
2314         }
2315
2316         SAFE_FREE(ip_list);
2317         return false;
2318 }
2319
2320 /********************************************************
2321  Get the IP address list of the primary domain controller
2322  for a domain.
2323 *********************************************************/
2324
2325 bool get_pdc_ip(const char *domain, struct sockaddr_storage *pss)
2326 {
2327         struct ip_service *ip_list = NULL;
2328         int count = 0;
2329         NTSTATUS status = NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND;
2330
2331         /* Look up #1B name */
2332
2333         if (lp_security() == SEC_ADS) {
2334                 status = internal_resolve_name(domain, 0x1b, NULL, &ip_list,
2335                                                &count, "ads");
2336         }
2337
2338         if (!NT_STATUS_IS_OK(status) || count == 0) {
2339                 status = internal_resolve_name(domain, 0x1b, NULL, &ip_list,
2340                                                &count,
2341                                                lp_name_resolve_order());
2342                 if (!NT_STATUS_IS_OK(status)) {
2343                         return false;
2344                 }
2345         }
2346
2347         /* if we get more than 1 IP back we have to assume it is a
2348            multi-homed PDC and not a mess up */
2349
2350         if ( count > 1 ) {
2351                 DEBUG(6,("get_pdc_ip: PDC has %d IP addresses!\n", count));
2352                 sort_service_list(ip_list, count);
2353         }
2354
2355         *pss = ip_list[0].ss;
2356         SAFE_FREE(ip_list);
2357         return true;
2358 }
2359
2360 /* Private enum type for lookups. */
2361
2362 enum dc_lookup_type { DC_NORMAL_LOOKUP, DC_ADS_ONLY, DC_KDC_ONLY };
2363
2364 /********************************************************
2365  Get the IP address list of the domain controllers for
2366  a domain.
2367 *********************************************************/
2368
2369 static NTSTATUS get_dc_list(const char *domain,
2370                         const char *sitename,
2371                         struct ip_service **ip_list,
2372                         int *count,
2373                         enum dc_lookup_type lookup_type,
2374                         bool *ordered)
2375 {
2376         char *resolve_order = NULL;
2377         char *saf_servername = NULL;
2378         char *pserver = NULL;
2379         const char *p;
2380         char *port_str = NULL;
2381         int port;
2382         char *name;
2383         int num_addresses = 0;
2384         int  local_count, i, j;
2385         struct ip_service *return_iplist = NULL;
2386         struct ip_service *auto_ip_list = NULL;
2387         bool done_auto_lookup = false;
2388         int auto_count = 0;
2389         NTSTATUS status;
2390         TALLOC_CTX *ctx = talloc_init("get_dc_list");
2391
2392         *ip_list = NULL;
2393         *count = 0;
2394
2395         if (!ctx) {
2396                 return NT_STATUS_NO_MEMORY;
2397         }
2398
2399         *ordered = False;
2400
2401         /* if we are restricted to solely using DNS for looking
2402            up a domain controller, make sure that host lookups
2403            are enabled for the 'name resolve order'.  If host lookups
2404            are disabled and ads_only is True, then set the string to
2405            NULL. */
2406
2407         resolve_order = talloc_strdup(ctx, lp_name_resolve_order());
2408         if (!resolve_order) {
2409                 status = NT_STATUS_NO_MEMORY;
2410                 goto out;
2411         }
2412         strlower_m(resolve_order);
2413         if (lookup_type == DC_ADS_ONLY)  {
2414                 if (strstr( resolve_order, "host")) {
2415                         resolve_order = talloc_strdup(ctx, "ads");
2416
2417                         /* DNS SRV lookups used by the ads resolver
2418                            are already sorted by priority and weight */
2419                         *ordered = true;
2420                 } else {
2421                         resolve_order = talloc_strdup(ctx, "NULL");
2422                 }
2423         } else if (lookup_type == DC_KDC_ONLY) {
2424                 /* DNS SRV lookups used by the ads/kdc resolver
2425                    are already sorted by priority and weight */
2426                 *ordered = true;
2427                 resolve_order = talloc_strdup(ctx, "kdc");
2428         }
2429         if (!resolve_order) {
2430                 status = NT_STATUS_NO_MEMORY;
2431                 goto out;
2432         }
2433
2434         /* fetch the server we have affinity for.  Add the
2435            'password server' list to a search for our domain controllers */
2436
2437         saf_servername = saf_fetch( domain);
2438
2439         if (strequal(domain, lp_workgroup()) || strequal(domain, lp_realm())) {
2440                 pserver = talloc_asprintf(ctx, "%s, %s",
2441                         saf_servername ? saf_servername : "",
2442                         lp_passwordserver());
2443         } else {
2444                 pserver = talloc_asprintf(ctx, "%s, *",
2445                         saf_servername ? saf_servername : "");
2446         }
2447
2448         SAFE_FREE(saf_servername);
2449         if (!pserver) {
2450                 status = NT_STATUS_NO_MEMORY;
2451                 goto out;
2452         }
2453
2454         /* if we are starting from scratch, just lookup DOMAIN<0x1c> */
2455
2456         if (!*pserver ) {
2457                 DEBUG(10,("get_dc_list: no preferred domain controllers.\n"));
2458                 status = internal_resolve_name(domain, 0x1C, sitename, ip_list,
2459                                              count, resolve_order);
2460                 goto out;
2461         }
2462
2463         DEBUG(3,("get_dc_list: preferred server list: \"%s\"\n", pserver ));
2464
2465         /*
2466          * if '*' appears in the "password server" list then add
2467          * an auto lookup to the list of manually configured
2468          * DC's.  If any DC is listed by name, then the list should be
2469          * considered to be ordered
2470          */
2471
2472         p = pserver;
2473         while (next_token_talloc(ctx, &p, &name, LIST_SEP)) {
2474                 if (!done_auto_lookup && strequal(name, "*")) {
2475                         status = internal_resolve_name(domain, 0x1C, sitename,
2476                                                        &auto_ip_list,
2477                                                        &auto_count,
2478                                                        resolve_order);
2479                         if (NT_STATUS_IS_OK(status)) {
2480                                 num_addresses += auto_count;
2481                         }
2482                         done_auto_lookup = true;
2483                         DEBUG(8,("Adding %d DC's from auto lookup\n",
2484                                                 auto_count));
2485                 } else  {
2486                         num_addresses++;
2487                 }
2488         }
2489
2490         /* if we have no addresses and haven't done the auto lookup, then
2491            just return the list of DC's.  Or maybe we just failed. */
2492
2493         if ((num_addresses == 0)) {
2494                 if (done_auto_lookup) {
2495                         DEBUG(4,("get_dc_list: no servers found\n"));
2496                         status = NT_STATUS_NO_LOGON_SERVERS;
2497                         goto out;
2498                 }
2499                 status = internal_resolve_name(domain, 0x1C, sitename, ip_list,
2500                                              count, resolve_order);
2501                 goto out;
2502         }
2503
2504         if ((return_iplist = SMB_MALLOC_ARRAY(struct ip_service,
2505                                         num_addresses)) == NULL) {
2506                 DEBUG(3,("get_dc_list: malloc fail !\n"));
2507                 status = NT_STATUS_NO_MEMORY;
2508                 goto out;
2509         }
2510
2511         p = pserver;
2512         local_count = 0;
2513
2514         /* fill in the return list now with real IP's */
2515
2516         while ((local_count<num_addresses) &&
2517                         next_token_talloc(ctx, &p, &name, LIST_SEP)) {
2518                 struct sockaddr_storage name_ss;
2519
2520                 /* copy any addersses from the auto lookup */
2521
2522                 if (strequal(name, "*")) {
2523                         for (j=0; j<auto_count; j++) {
2524                                 char addr[INET6_ADDRSTRLEN];
2525                                 print_sockaddr(addr,
2526                                                 sizeof(addr),
2527                                                 &auto_ip_list[j].ss);
2528                                 /* Check for and don't copy any
2529                                  * known bad DC IP's. */
2530                                 if(!NT_STATUS_IS_OK(check_negative_conn_cache(
2531                                                 domain,
2532                                                 addr))) {
2533                                         DEBUG(5,("get_dc_list: "
2534                                                 "negative entry %s removed "
2535                                                 "from DC list\n",
2536                                                 addr));
2537                                         continue;
2538                                 }
2539                                 return_iplist[local_count].ss =
2540                                         auto_ip_list[j].ss;
2541                                 return_iplist[local_count].port =
2542                                         auto_ip_list[j].port;
2543                                 local_count++;
2544                         }
2545                         continue;
2546                 }
2547
2548                 /* added support for address:port syntax for ads
2549                  * (not that I think anyone will ever run the LDAP
2550                  * server in an AD domain on something other than
2551                  * port 389 */
2552
2553                 port = (lp_security() == SEC_ADS) ? LDAP_PORT : PORT_NONE;
2554                 if ((port_str=strchr(name, ':')) != NULL) {
2555                         *port_str = '\0';
2556                         port_str++;
2557                         port = atoi(port_str);
2558                 }
2559
2560                 /* explicit lookup; resolve_name() will
2561                  * handle names & IP addresses */
2562                 if (resolve_name( name, &name_ss, 0x20, true )) {
2563                         char addr[INET6_ADDRSTRLEN];
2564                         print_sockaddr(addr,
2565                                         sizeof(addr),
2566                                         &name_ss);
2567
2568                         /* Check for and don't copy any known bad DC IP's. */
2569                         if( !NT_STATUS_IS_OK(check_negative_conn_cache(domain,
2570                                                         addr)) ) {
2571                                 DEBUG(5,("get_dc_list: negative entry %s "
2572                                         "removed from DC list\n",
2573                                         name ));
2574                                 continue;
2575                         }
2576
2577                         return_iplist[local_count].ss = name_ss;
2578                         return_iplist[local_count].port = port;
2579                         local_count++;
2580                         *ordered = true;
2581                 }
2582         }
2583
2584         /* need to remove duplicates in the list if we have any
2585            explicit password servers */
2586
2587         if (local_count) {
2588                 local_count = remove_duplicate_addrs2(return_iplist,
2589                                 local_count );
2590         }
2591
2592         /* For DC's we always prioritize IPv4 due to W2K3 not
2593          * supporting LDAP, KRB5 or CLDAP over IPv6. */
2594
2595         if (local_count && return_iplist) {
2596                 prioritize_ipv4_list(return_iplist, local_count);
2597         }
2598
2599         if ( DEBUGLEVEL >= 4 ) {
2600                 DEBUG(4,("get_dc_list: returning %d ip addresses "
2601                                 "in an %sordered list\n",
2602                                 local_count,
2603                                 *ordered ? "":"un"));
2604                 DEBUG(4,("get_dc_list: "));
2605                 for ( i=0; i<local_count; i++ ) {
2606                         char addr[INET6_ADDRSTRLEN];
2607                         print_sockaddr(addr,
2608                                         sizeof(addr),
2609                                         &return_iplist[i].ss);
2610                         DEBUGADD(4,("%s:%d ", addr, return_iplist[i].port ));
2611                 }
2612                 DEBUGADD(4,("\n"));
2613         }
2614
2615         *ip_list = return_iplist;
2616         *count = local_count;
2617
2618         status = ( *count != 0 ? NT_STATUS_OK : NT_STATUS_NO_LOGON_SERVERS );
2619
2620   out:
2621
2622         if (!NT_STATUS_IS_OK(status)) {
2623                 SAFE_FREE(return_iplist);
2624                 *ip_list = NULL;
2625                 *count = 0;
2626         }
2627
2628         SAFE_FREE(auto_ip_list);
2629         TALLOC_FREE(ctx);
2630         return status;
2631 }
2632
2633 /*********************************************************************
2634  Small wrapper function to get the DC list and sort it if neccessary.
2635 *********************************************************************/
2636
2637 NTSTATUS get_sorted_dc_list( const char *domain,
2638                         const char *sitename,
2639                         struct ip_service **ip_list,
2640                         int *count,
2641                         bool ads_only )
2642 {
2643         bool ordered = false;
2644         NTSTATUS status;
2645         enum dc_lookup_type lookup_type = DC_NORMAL_LOOKUP;
2646
2647         *ip_list = NULL;
2648         *count = 0;
2649
2650         DEBUG(8,("get_sorted_dc_list: attempting lookup "
2651                 "for name %s (sitename %s) using [%s]\n",
2652                 domain,
2653                 sitename ? sitename : "NULL",
2654                 (ads_only ? "ads" : lp_name_resolve_order())));
2655
2656         if (ads_only) {
2657                 lookup_type = DC_ADS_ONLY;
2658         }
2659
2660         status = get_dc_list(domain, sitename, ip_list,
2661                         count, lookup_type, &ordered);
2662         if (NT_STATUS_EQUAL(status, NT_STATUS_NO_LOGON_SERVERS)
2663             && sitename) {
2664                 DEBUG(3,("get_sorted_dc_list: no server for name %s available"
2665                          " in site %s, fallback to all servers\n",
2666                          domain, sitename));
2667                 status = get_dc_list(domain, NULL, ip_list,
2668                                      count, lookup_type, &ordered);
2669         }
2670
2671         if (!NT_STATUS_IS_OK(status)) {
2672                 SAFE_FREE(*ip_list);
2673                 *count = 0;
2674                 return status;
2675         }
2676
2677         /* only sort if we don't already have an ordered list */
2678         if (!ordered) {
2679                 sort_service_list(*ip_list, *count);
2680         }
2681
2682         return NT_STATUS_OK;
2683 }
2684
2685 /*********************************************************************
2686  Get the KDC list - re-use all the logic in get_dc_list.
2687 *********************************************************************/
2688
2689 NTSTATUS get_kdc_list( const char *realm,
2690                         const char *sitename,
2691                         struct ip_service **ip_list,
2692                         int *count)
2693 {
2694         bool ordered;
2695         NTSTATUS status;
2696
2697         *count = 0;
2698         *ip_list = NULL;
2699
2700         status = get_dc_list(realm, sitename, ip_list,
2701                         count, DC_KDC_ONLY, &ordered);
2702
2703         if (!NT_STATUS_IS_OK(status)) {
2704                 SAFE_FREE(*ip_list);
2705                 *count = 0;
2706                 return status;
2707         }
2708
2709         /* only sort if we don't already have an ordered list */
2710         if ( !ordered ) {
2711                 sort_service_list(*ip_list, *count);
2712         }
2713
2714         return NT_STATUS_OK;
2715 }