source3/utils/net_lookup.c

   1 /*
   2    Samba Unix/Linux SMB client library
   3    net lookup command
   4    Copyright (C) 2001 Andrew Tridgell (tridge@samba.org)
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 2 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program; if not, write to the Free Software
  18    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
  19
  20 #include "includes.h"
  21 #include "../utils/net.h"
  22
  23 int net_lookup_usage(int argc, const char **argv)
  24 {
  25         d_printf(
  26 "  net lookup host HOSTNAME <type>\n\tgives IP for a hostname\n\n"
  27 "  net lookup ldap [domain]\n\tgives IP of domain's ldap server\n\n"
  28 "  net lookup kdc [realm]\n\tgives IP of realm's kerberos KDC\n\n"
  29 "  net lookup dc [domain]\n\tgives IP of domains Domain Controllers\n\n"
  30 "  net lookup master [domain|wg]\n\tgive IP of master browser\n\n"
  31 );
  32         return -1;
  33 }
  34
  35 /* lookup a hostname giving an IP */
  36 static int net_lookup_host(int argc, const char **argv)
  37 {
  38         struct in_addr ip;
  39         int name_type = 0x20;
  40
  41         if (argc == 0) return net_lookup_usage(argc, argv);
  42         if (argc > 1) name_type = strtol(argv[1], NULL, 0);
  43
  44         if (!resolve_name(argv[0], &ip, name_type)) {
  45                 /* we deliberately use DEBUG() here to send it to stderr
  46                    so scripts aren't mucked up */
  47                 DEBUG(0,("Didn't find %s#%02x\n", argv[0], name_type));
  48                 return -1;
  49         }
  50
  51         d_printf("%s\n", inet_ntoa(ip));
  52         return 0;
  53 }
  54
  55 static void print_ldap_srvlist(char *srvlist)
  56 {
  57         char *cur, *next;
  58         struct in_addr ip;
  59         BOOL printit;
  60
  61         cur = srvlist;
  62         do {
  63                 next = strchr(cur,':');
  64                 if (next) *next++='\0';
  65                 printit = resolve_name(cur, &ip, 0x20);
  66                 cur=next;
  67                 next=cur ? strchr(cur,' ') :NULL;
  68                 if (next)
  69                         *next++='\0';
  70                 if (printit)
  71                         d_printf("%s:%s\n", inet_ntoa(ip), cur?cur:"");
  72                 cur = next;
  73         } while (next);
  74 }
  75
  76
  77 static int net_lookup_ldap(int argc, const char **argv)
  78 {
  79 #ifdef HAVE_LDAP
  80         char *srvlist;
  81         const char *domain;
  82         int rc, count;
  83         struct in_addr *addr;
  84         struct hostent *hostent;
  85
  86         if (argc > 0)
  87                 domain = argv[0];
  88         else
  89                 domain = opt_target_workgroup;
  90
  91         DEBUG(9, ("Lookup up ldap for domain %s\n", domain));
  92         rc = ldap_domain2hostlist(domain, &srvlist);
  93         if ((rc == LDAP_SUCCESS) && srvlist) {
  94                 print_ldap_srvlist(srvlist);
  95                 return 0;
  96         }
  97
  98         DEBUG(9, ("Looking up DC for domain %s\n", domain));
  99         if (!get_dc_list(True, domain, &addr, &count))
 100                 return -1;
 101
 102         hostent = gethostbyaddr((char *) &addr->s_addr, sizeof(addr->s_addr),
 103                                 AF_INET);
 104         if (!hostent)
 105                 return -1;
 106
 107         DEBUG(9, ("Found DC with DNS name %s\n", hostent->h_name));
 108         domain = strchr(hostent->h_name, '.');
 109         if (!domain)
 110                 return -1;
 111         domain++;
 112
 113         DEBUG(9, ("Looking up ldap for domain %s\n", domain));
 114         rc = ldap_domain2hostlist(domain, &srvlist);
 115         if ((rc == LDAP_SUCCESS) && srvlist) {
 116                 print_ldap_srvlist(srvlist);
 117                 return 0;
 118         }
 119         return -1;
 120 #endif
 121         DEBUG(1,("No LDAP support\n"));
 122         return -1;
 123 }
 124
 125 static int net_lookup_dc(int argc, const char **argv)
 126 {
 127         struct in_addr *ip_list;
 128         char *pdc_str = NULL;
 129         const char *domain=opt_target_workgroup;
 130         int count, i;
 131
 132         if (argc > 0)
 133                 domain=argv[0];
 134
 135         /* first get PDC */
 136         if (!get_dc_list(True, domain, &ip_list, &count))
 137                 return -1;
 138
 139         asprintf(&pdc_str, "%s", inet_ntoa(*ip_list));
 140         d_printf("%s\n", pdc_str);
 141
 142         if (!get_dc_list(False, domain, &ip_list, &count)) {
 143                 SAFE_FREE(pdc_str);
 144                 return 0;
 145         }
 146         for (i=0;i<count;i++) {
 147                 char *dc_str = inet_ntoa(ip_list[i]);
 148                 if (!strequal(pdc_str, dc_str))
 149                         d_printf("%s\n", dc_str);
 150         }
 151         SAFE_FREE(pdc_str);
 152         return 0;
 153 }
 154
 155 static int net_lookup_master(int argc, const char **argv)
 156 {
 157         struct in_addr master_ip;
 158         const char *domain=opt_target_workgroup;
 159
 160         if (argc > 0)
 161                 domain=argv[0];
 162
 163         if (!find_master_ip(domain, &master_ip))
 164                 return -1;
 165         d_printf("%s\n", inet_ntoa(master_ip));
 166         return 0;
 167 }
 168
 169 static int net_lookup_kdc(int argc, const char **argv)
 170 {
 171 #ifdef HAVE_KRB5
 172         krb5_error_code rc;
 173         krb5_context ctx;
 174         struct sockaddr_in *addrs;
 175         int num_kdcs,i;
 176         krb5_data realm;
 177         char **realms;
 178
 179         rc = krb5_init_context(&ctx);
 180         if (rc) {
 181                 DEBUG(1,("krb5_init_context failed (%s)\n",
 182                          error_message(rc)));
 183                 return -1;
 184         }
 185
 186         if (argc>0) {
 187                 realm.data = (krb5_pointer) argv[0];
 188                 realm.length = strlen(argv[0]);
 189         } else if (lp_realm() && *lp_realm()) {
 190                 realm.data = (krb5_pointer) lp_realm();
 191                 realm.length = strlen(realm.data);
 192         } else {
 193                 rc = krb5_get_host_realm(ctx, NULL, &realms);
 194                 if (rc) {
 195                         DEBUG(1,("krb5_gethost_realm failed (%s)\n",
 196                                  error_message(rc)));
 197                         return -1;
 198                 }
 199                 realm.data = (krb5_pointer) *realms;
 200                 realm.length = strlen(realm.data);
 201         }
 202
 203         rc = krb5_locate_kdc(ctx, &realm, &addrs, &num_kdcs, 0);
 204         if (rc) {
 205                 DEBUG(1, ("krb5_locate_kdc failed (%s)\n", error_message(rc)));
 206                 return -1;
 207         }
 208         for (i=0;i<num_kdcs;i++)
 209                 if (addrs[i].sin_family == AF_INET)
 210                         d_printf("%s:%hd\n", inet_ntoa(addrs[i].sin_addr),
 211                                  ntohs(addrs[i].sin_port));
 212         return 0;
 213
 214 #endif
 215         DEBUG(1, ("No kerberos support\n"));
 216         return -1;
 217 }
 218
 219
 220 /* lookup hosts or IP addresses using internal samba lookup fns */
 221 int net_lookup(int argc, const char **argv)
 222 {
 223         struct functable func[] = {
 224                 {"HOST", net_lookup_host},
 225                 {"LDAP", net_lookup_ldap},
 226                 {"DC", net_lookup_dc},
 227                 {"MASTER", net_lookup_master},
 228                 {"KDC", net_lookup_kdc},
 229                 {NULL, NULL}
 230         };
 231
 232         return net_run_function(argc, argv, func, net_lookup_usage);
 233 }