Convert three include files from uint32/16/8 to _t types as well as the source that...
[sfrench/samba-autobuild/.git] / source3 / libsmb / namequery_dc.c
index ac1f9a54de3e45a446d5052463048ab78238b29d..eb34741defa34c775601384ea28e81fa2630df5a 100644 (file)
 
    Copyright (C) Tim Potter 2001
    Copyright (C) Andrew Bartlett 2002
-   
+   Copyright (C) Gerald Carter 2003
+
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
-   the Free Software Foundation; either version 2 of the License, or
+   the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
-   
+
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.         See the
    GNU General Public License for more details.
-   
+
    You should have received a copy of the GNU General Public License
-   along with this program; if not, write to the Free Software
-   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 
 #include "includes.h"
-
-
-#define FAILED_CONNECTION_CACHE_TIMEOUT 30 /* Seconds between attempts */
-
-struct failed_connection_cache {
-       fstring domain_name;
-       fstring controller;
-       time_t lookup_time;
-       NTSTATUS nt_status;
-       struct failed_connection_cache *prev, *next;
-};
-
-static struct failed_connection_cache *failed_connection_cache;
+#include "libads/sitename_cache.h"
+#include "ads.h"
+#include "../librpc/gen_ndr/nbt.h"
+#include "lib/param/loadparm.h"
 
 /**********************************************************************
- Check for a previously failed connection
+ Is this our primary domain ?
 **********************************************************************/
 
-static NTSTATUS check_negative_conn_cache( const char *domain, const char *server )
+#ifdef HAVE_KRB5
+static bool is_our_primary_domain(const char *domain)
 {
-       struct failed_connection_cache *fcc;
-       NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
-       
-       /* can't check if we don't have strings */
-       
-       if ( !domain || !server )
-               return NT_STATUS_OK;
-
-       for (fcc = failed_connection_cache; fcc; fcc = fcc->next) {
-               
-               /* 
-                * we have a match IFF the domain and server name matches
-                *   (a) the domain matches, 
-                *   (b) the IP address matches (if we have one)
-                *   (c) the server name (if specified) matches
-                */
-                
-               if ( !strequal(domain, fcc->domain_name) || !strequal(server, fcc->controller) )
-                       continue; /* no match; check the next entry */
-               
-               /* we have a match so see if it is still current */
-
-               if ((time(NULL) - fcc->lookup_time) > FAILED_CONNECTION_CACHE_TIMEOUT) 
-               {
-                       /* Cache entry has expired, delete it */
-
-                       DEBUG(10, ("check_negative_conn_cache: cache entry expired for %s, %s\n", 
-                               domain, server ));
-
-                       DLIST_REMOVE(failed_connection_cache, fcc);
-                       SAFE_FREE(fcc);
-
-                       return NT_STATUS_OK;
-               }
+       int role = lp_server_role();
 
-               /* The timeout hasn't expired yet so return false */
-
-               DEBUG(10, ("check_negative_conn_cache: returning negative entry for %s, %s\n", 
-                       domain, server ));
-
-               result = fcc->nt_status;
-               return result;
+       if ((role == ROLE_DOMAIN_MEMBER) && strequal(lp_workgroup(), domain)) {
+               return True;
+       } else if (strequal(get_global_sam_name(), domain)) {
+               return True;
        }
-
-       /* end of function means no cache entry */      
-       return NT_STATUS_OK;
+       return False;
 }
+#endif
 
-/**********************************************************************
- Add an entry to the failed conneciton cache
-**********************************************************************/
+/**************************************************************************
+ Find the name and IP address for a server in the realm/domain
+ *************************************************************************/
 
-void add_failed_connection_entry(const char *domain, const char *server, NTSTATUS result) 
+static bool ads_dc_name(const char *domain,
+                       const char *realm,
+                       struct sockaddr_storage *dc_ss,
+                       fstring srv_name)
 {
-       struct failed_connection_cache *fcc;
+       ADS_STRUCT *ads;
+       char *sitename;
+       int i;
+       char addr[INET6_ADDRSTRLEN];
 
-       SMB_ASSERT(!NT_STATUS_IS_OK(result));
+       if (!realm && strequal(domain, lp_workgroup())) {
+               realm = lp_realm();
+       }
 
-       /* Check we already aren't in the cache.  We always have to have 
-          a domain, but maybe not a specific DC name. */
+       sitename = sitename_fetch(talloc_tos(), realm);
 
-       for (fcc = failed_connection_cache; fcc; fcc = fcc->next) {
-               if ( strequal(fcc->domain_name, domain) && strequal(fcc->controller, server) ) 
-               {
-                       DEBUG(10, ("add_failed_connection_entry: domain %s (%s) already tried and failed\n",
-                                  domain, server ));
-                       return;
+       /* Try this 3 times then give up. */
+       for( i =0 ; i < 3; i++) {
+               ads = ads_init(realm, domain, NULL);
+               if (!ads) {
+                       TALLOC_FREE(sitename);
+                       return False;
                }
-       }
 
-       /* Create negative lookup cache entry for this domain and controller */
+               DEBUG(4,("ads_dc_name: domain=%s\n", domain));
 
-       if ( !(fcc = (struct failed_connection_cache *)malloc(sizeof(struct failed_connection_cache))) ) 
-       {
-               DEBUG(0, ("malloc failed in add_failed_connection_entry!\n"));
-               return;
-       }
-       
-       ZERO_STRUCTP(fcc);
-       
-       fstrcpy( fcc->domain_name, domain );
-       fstrcpy( fcc->controller, server );
-       fcc->lookup_time = time(NULL);
-       fcc->nt_status = result;
-       
-       DEBUG(10,("add_failed_connection_entry: added domain %s (%s) to failed conn cache\n",
-               domain, server ));
-       
-       DLIST_ADD(failed_connection_cache, fcc);
-}
+#ifdef HAVE_ADS
+               /* we don't need to bind, just connect */
+               ads->auth.flags |= ADS_AUTH_NO_BIND;
+               ads_connect(ads);
+#endif
 
-/****************************************************************************
-****************************************************************************/
-void flush_negative_conn_cache( void )
-{
-       struct failed_connection_cache *fcc;
-       
-       fcc = failed_connection_cache;
+               if (!ads->config.realm) {
+                       TALLOC_FREE(sitename);
+                       ads_destroy(&ads);
+                       return False;
+               }
 
-       while (fcc) {
-               struct failed_connection_cache *fcc_next;
+               /* Now we've found a server, see if our sitename
+                  has changed. If so, we need to re-do the DNS query
+                  to ensure we only find servers in our site. */
+
+               if (stored_sitename_changed(realm, sitename)) {
+                       TALLOC_FREE(sitename);
+                       sitename = sitename_fetch(talloc_tos(), realm);
+                       ads_destroy(&ads);
+                       /* Ensure we don't cache the DC we just connected to. */
+                       namecache_delete(realm, 0x1C);
+                       namecache_delete(domain, 0x1C);
+                       continue;
+               }
 
-               fcc_next = fcc->next;
-               DLIST_REMOVE(failed_connection_cache, fcc);
-               free(fcc);
+#ifdef HAVE_ADS
+               if (is_our_primary_domain(domain) && (ads->config.flags & NBT_SERVER_KDC)) {
+                       if (ads_closest_dc(ads)) {
+                               /* We're going to use this KDC for this realm/domain.
+                                  If we are using sites, then force the krb5 libs
+                                  to use this KDC. */
+
+                               create_local_private_krb5_conf_for_domain(realm,
+                                                                       domain,
+                                                                       sitename,
+                                                                       &ads->ldap.ss);
+                       } else {
+                               create_local_private_krb5_conf_for_domain(realm,
+                                                                       domain,
+                                                                       NULL,
+                                                                       &ads->ldap.ss);
+                       }
+               }
+#endif
+               break;
+       }
 
-               fcc = fcc_next;
+       if (i == 3) {
+               DEBUG(1,("ads_dc_name: sitename (now \"%s\") keeps changing ???\n",
+                       sitename ? sitename : ""));
+               TALLOC_FREE(sitename);
+               ads_destroy(&ads);
+               return False;
        }
 
+       TALLOC_FREE(sitename);
+
+       fstrcpy(srv_name, ads->config.ldap_server_name);
+       if (!strupper_m(srv_name)) {
+               ads_destroy(&ads);
+               return false;
+       }
+#ifdef HAVE_ADS
+       *dc_ss = ads->ldap.ss;
+#else
+       zero_sockaddr(dc_ss);
+#endif
+       ads_destroy(&ads);
+
+       print_sockaddr(addr, sizeof(addr), dc_ss);
+       DEBUG(4,("ads_dc_name: using server='%s' IP=%s\n",
+                srv_name, addr));
+
+       return True;
 }
 
 /****************************************************************************
- Utility function to return the name of a DC. The name is guaranteed to be 
- valid since we have already done a name_status_find on it 
+ Utility function to return the name of a DC. The name is guaranteed to be
+ valid since we have already done a name_status_find on it
  ***************************************************************************/
 
-BOOL rpc_dc_name(const char *domain, fstring srv_name, struct in_addr *ip_out)
+static bool rpc_dc_name(const char *domain,
+                       fstring srv_name,
+                       struct sockaddr_storage *ss_out)
 {
-       struct in_addr *ip_list = NULL, dc_ip, exclude_ip;
+       struct ip_service *ip_list = NULL;
+       struct sockaddr_storage dc_ss;
        int count, i;
-       BOOL list_ordered;
-       BOOL use_pdc_only;
        NTSTATUS result;
-       
-       zero_ip(&exclude_ip);
-
-       use_pdc_only = must_use_pdc(domain);
-       
-       /* Lookup domain controller name */
-          
-       if ( use_pdc_only && get_pdc_ip(domain, &dc_ip) ) 
-       {
-               DEBUG(10,("rpc_dc_name: Atempting to lookup PDC to avoid sam sync delays\n"));
-               
-               /* check the connection cache and perform the node status 
-                  lookup only if the IP is not found to be bad */
-
-               if (name_status_find(domain, 0x1c, 0x20, dc_ip, srv_name) ) {
-                       result = check_negative_conn_cache( domain, srv_name );
-                       if ( NT_STATUS_IS_OK(result) )
-                               goto done;
-               }
-               /* Didn't get name, remember not to talk to this DC. */
-               exclude_ip = dc_ip;
-       }
+       char addr[INET6_ADDRSTRLEN];
 
        /* get a list of all domain controllers */
-       
-       if (!get_dc_list( domain, &ip_list, &count, &list_ordered) ) {
+
+       if (!NT_STATUS_IS_OK(get_sorted_dc_list(domain, NULL, &ip_list, &count,
+                                               False))) {
                DEBUG(3, ("Could not look up dc's for domain %s\n", domain));
                return False;
        }
 
        /* Remove the entry we've already failed with (should be the PDC). */
 
-       if ( use_pdc_only ) {
-               for (i = 0; i < count; i++) {   
-                       if (ip_equal( exclude_ip, ip_list[i]))
-                               zero_ip(&ip_list[i]);
-               }
-       }
-
-       /* Pick a nice close server, but only if the list was not ordered */
-       
-       if (!list_ordered && (count > 1) ) {
-               qsort(ip_list, count, sizeof(struct in_addr), QSORT_CAST ip_compare);
-       }
-
        for (i = 0; i < count; i++) {
-               if (is_zero_ip(ip_list[i]))
+               if (is_zero_addr(&ip_list[i].ss))
                        continue;
 
-               if (name_status_find(domain, 0x1c, 0x20, ip_list[i], srv_name)) {
+               if (name_status_find(domain, 0x1c, 0x20, &ip_list[i].ss, srv_name)) {
                        result = check_negative_conn_cache( domain, srv_name );
                        if ( NT_STATUS_IS_OK(result) ) {
-                               dc_ip = ip_list[i];
+                               dc_ss = ip_list[i].ss;
                                goto done;
                        }
                }
        }
-       
 
        SAFE_FREE(ip_list);
 
        /* No-one to talk to )-: */
        return False;           /* Boo-hoo */
-       
+
  done:
        /* We have the netbios name and IP address of a domain controller.
           Ideally we should sent a SAMLOGON request to determine whether
           the DC is alive and kicking.  If we can catch a dead DC before
           performing a cli_connect() we can avoid a 30-second timeout. */
 
+       print_sockaddr(addr, sizeof(addr), &dc_ss);
        DEBUG(3, ("rpc_dc_name: Returning DC %s (%s) for domain %s\n", srv_name,
-                 inet_ntoa(dc_ip), domain));
-
-       *ip_out = dc_ip;
+                 addr, domain));
 
+       *ss_out = dc_ss;
        SAFE_FREE(ip_list);
 
        return True;
 }
+
+/**********************************************************************
+ wrapper around ads and rpc methods of finds DC's
+**********************************************************************/
+
+bool get_dc_name(const char *domain,
+               const char *realm,
+               fstring srv_name,
+               struct sockaddr_storage *ss_out)
+{
+       struct sockaddr_storage dc_ss;
+       bool ret;
+       bool our_domain = False;
+
+       zero_sockaddr(&dc_ss);
+
+       ret = False;
+
+       if ( strequal(lp_workgroup(), domain) || strequal(lp_realm(), realm) )
+               our_domain = True;
+
+       /* always try to obey what the admin specified in smb.conf
+          (for the local domain) */
+
+       if ( (our_domain && lp_security()==SEC_ADS) || realm ) {
+               ret = ads_dc_name(domain, realm, &dc_ss, srv_name);
+       }
+
+       if (!domain) {
+               /* if we have only the realm we can't do anything else */
+               return False;
+       }
+
+       if (!ret) {
+               /* fall back on rpc methods if the ADS methods fail */
+               ret = rpc_dc_name(domain, srv_name, &dc_ss);
+       }
+
+       *ss_out = dc_ss;
+
+       return ret;
+}