r13310: first round of server affinity patches for winbindd & net ads join
[amitay/samba.git] / source3 / libsmb / namequery_dc.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    Winbind daemon connection manager
5
6    Copyright (C) Tim Potter 2001
7    Copyright (C) Andrew Bartlett 2002
8    Copyright (C) Gerald Carter 2003
9    
10    This program is free software; you can redistribute it and/or modify
11    it under the terms of the GNU General Public License as published by
12    the Free Software Foundation; either version 2 of the License, or
13    (at your option) any later version.
14    
15    This program is distributed in the hope that it will be useful,
16    but WITHOUT ANY WARRANTY; without even the implied warranty of
17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18    GNU General Public License for more details.
19    
20    You should have received a copy of the GNU General Public License
21    along with this program; if not, write to the Free Software
22    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
23 */
24
25
26 #include "includes.h"
27
28 /**************************************************************************
29  Find the name and IP address for a server in he realm/domain
30  *************************************************************************/
31  
32 static BOOL ads_dc_name(const char *domain, const char *realm, struct in_addr *dc_ip, fstring srv_name)
33 {
34         ADS_STRUCT *ads;
35
36         if (!realm && strequal(domain, lp_workgroup()))
37                 realm = lp_realm();
38
39         ads = ads_init(realm, domain, NULL);
40         if (!ads)
41                 return False;
42
43         DEBUG(4,("ads_dc_name: domain=%s\n", domain));
44
45 #ifdef HAVE_ADS
46         /* we don't need to bind, just connect */
47         ads->auth.flags |= ADS_AUTH_NO_BIND;
48
49         ads_connect(ads);
50 #endif
51
52         if (!ads->config.realm) {
53                 ads_destroy(&ads);
54                 return False;
55         }
56
57         fstrcpy(srv_name, ads->config.ldap_server_name);
58         strupper_m(srv_name);
59         *dc_ip = ads->ldap_ip;
60         ads_destroy(&ads);
61         
62         DEBUG(4,("ads_dc_name: using server='%s' IP=%s\n",
63                  srv_name, inet_ntoa(*dc_ip)));
64         
65         return True;
66 }
67
68 /****************************************************************************
69  Utility function to return the name of a DC. The name is guaranteed to be 
70  valid since we have already done a name_status_find on it 
71  ***************************************************************************/
72
73 static BOOL rpc_dc_name(const char *domain, fstring srv_name, struct in_addr *ip_out)
74 {
75         struct ip_service *ip_list = NULL;
76         struct in_addr dc_ip, exclude_ip;
77         int count, i;
78         NTSTATUS result;
79         
80         zero_ip(&exclude_ip);
81
82         /* get a list of all domain controllers */
83         
84         if ( !get_sorted_dc_list(domain, &ip_list, &count, False) ) {
85                 DEBUG(3, ("Could not look up dc's for domain %s\n", domain));
86                 return False;
87         }
88
89         /* Remove the entry we've already failed with (should be the PDC). */
90
91         for (i = 0; i < count; i++) {
92                 if (is_zero_ip(ip_list[i].ip))
93                         continue;
94
95                 if (name_status_find(domain, 0x1c, 0x20, ip_list[i].ip, srv_name)) {
96                         result = check_negative_conn_cache( domain, srv_name );
97                         if ( NT_STATUS_IS_OK(result) ) {
98                                 dc_ip = ip_list[i].ip;
99                                 goto done;
100                         }
101                 }
102         }
103         
104
105         SAFE_FREE(ip_list);
106
107         /* No-one to talk to )-: */
108         return False;           /* Boo-hoo */
109         
110  done:
111         /* We have the netbios name and IP address of a domain controller.
112            Ideally we should sent a SAMLOGON request to determine whether
113            the DC is alive and kicking.  If we can catch a dead DC before
114            performing a cli_connect() we can avoid a 30-second timeout. */
115
116         DEBUG(3, ("rpc_dc_name: Returning DC %s (%s) for domain %s\n", srv_name,
117                   inet_ntoa(dc_ip), domain));
118
119         *ip_out = dc_ip;
120
121         SAFE_FREE(ip_list);
122
123         return True;
124 }
125
126 /**********************************************************************
127  wrapper around ads and rpc methods of finds DC's
128 **********************************************************************/
129
130 BOOL get_dc_name(const char *domain, const char *realm, fstring srv_name, struct in_addr *ip_out)
131 {
132         struct in_addr dc_ip;
133         BOOL ret;
134         BOOL our_domain = False;
135
136         zero_ip(&dc_ip);
137
138         ret = False;
139         
140         if ( strequal(lp_workgroup(), domain) || strequal(lp_realm(), realm) )
141                 our_domain = True;
142         
143         /* always try to obey what the admin specified in smb.conf 
144            (for the local domain) */
145         
146         if ( (our_domain && lp_security()==SEC_ADS) || realm ) {
147                 ret = ads_dc_name(domain, realm, &dc_ip, srv_name);
148         }
149         
150         if (!ret) {
151                 /* fall back on rpc methods if the ADS methods fail */
152                 ret = rpc_dc_name(domain, srv_name, &dc_ip);
153         }
154                 
155         *ip_out = dc_ip;
156
157         return ret;
158 }
159