1b31b32ff07d2cac8bdaa4c0c195018e1b9ecedd
[samba.git] / 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 3 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, see <http://www.gnu.org/licenses/>.  */
18
19 #include "includes.h"
20 #include "utils/net.h"
21 #include "libads/sitename_cache.h"
22 #include "libads/dns.h"
23 #include "../librpc/gen_ndr/ndr_netlogon.h"
24
25 int net_lookup_usage(struct net_context *c, int argc, const char **argv)
26 {
27         d_printf(_(
28 "  net lookup [host] HOSTNAME[#<type>]\n\tgives IP for a hostname\n\n"
29 "  net lookup ldap [domain]\n\tgives IP of domain's ldap server\n\n"
30 "  net lookup kdc [realm]\n\tgives IP of realm's kerberos KDC\n\n"
31 "  net lookup pdc [domain|realm]\n\tgives IP of realm's kerberos KDC\n\n"
32 "  net lookup dc [domain]\n\tgives IP of domains Domain Controllers\n\n"
33 "  net lookup master [domain|wg]\n\tgive IP of master browser\n\n"
34 "  net lookup name [name]\n\tLookup name's sid and type\n\n"
35 "  net lookup sid [sid]\n\tGive sid's name and type\n\n"
36 "  net lookup dsgetdcname [name] [flags] [sitename]\n\n"
37 ));
38         return -1;
39 }
40
41 /* lookup a hostname giving an IP */
42 static int net_lookup_host(struct net_context *c, int argc, const char **argv)
43 {
44         struct sockaddr_storage ss;
45         int name_type = 0x20;
46         char addr[INET6_ADDRSTRLEN];
47         const char *name = argv[0];
48         char *p;
49
50         if (argc == 0)
51                 return net_lookup_usage(c, argc, argv);
52
53         p = strchr_m(name,'#');
54         if (p) {
55                 *p = '\0';
56                 sscanf(++p,"%x",&name_type);
57         }
58
59         if (!resolve_name(name, &ss, name_type, false)) {
60                 /* we deliberately use DEBUG() here to send it to stderr
61                    so scripts aren't mucked up */
62                 DEBUG(0,("Didn't find %s#%02x\n", name, name_type));
63                 return -1;
64         }
65
66         print_sockaddr(addr, sizeof(addr), &ss);
67         d_printf("%s\n", addr);
68         return 0;
69 }
70
71 #ifdef HAVE_ADS
72 static void print_ldap_srvlist(struct dns_rr_srv *dclist, int numdcs )
73 {
74         struct sockaddr_storage ss;
75         int i;
76
77         for ( i=0; i<numdcs; i++ ) {
78                 if (resolve_name(dclist[i].hostname, &ss, 0x20, true) ) {
79                         char addr[INET6_ADDRSTRLEN];
80                         print_sockaddr(addr, sizeof(addr), &ss);
81 #ifdef HAVE_IPV6
82                         if (ss.ss_family == AF_INET6) {
83                                 d_printf("[%s]:%d\n", addr, dclist[i].port);
84                         }
85 #endif
86                         if (ss.ss_family == AF_INET) {
87                                 d_printf("%s:%d\n", addr, dclist[i].port);
88                         }
89                 }
90         }
91 }
92 #endif
93
94 static int net_lookup_ldap(struct net_context *c, int argc, const char **argv)
95 {
96 #ifdef HAVE_ADS
97         const char *domain;
98         struct sockaddr_storage ss;
99         struct dns_rr_srv *dcs = NULL;
100         int numdcs = 0;
101         char *sitename;
102         TALLOC_CTX *ctx;
103         NTSTATUS status;
104         int ret;
105         char h_name[MAX_DNS_NAME_LENGTH];
106
107         if (argc > 0)
108                 domain = argv[0];
109         else
110                 domain = c->opt_target_workgroup;
111
112         sitename = sitename_fetch(domain);
113
114         if ( (ctx = talloc_init("net_lookup_ldap")) == NULL ) {
115                 d_fprintf(stderr,"net_lookup_ldap: talloc_init() %s!\n",
116                           _("failed"));
117                 SAFE_FREE(sitename);
118                 return -1;
119         }
120
121         DEBUG(9, ("Lookup up ldap for domain %s\n", domain));
122
123         status = ads_dns_query_dcs( ctx, domain, sitename, &dcs, &numdcs );
124         if ( NT_STATUS_IS_OK(status) && numdcs ) {
125                 print_ldap_srvlist(dcs, numdcs);
126                 TALLOC_FREE( ctx );
127                 SAFE_FREE(sitename);
128                 return 0;
129         }
130
131         DEBUG(9, ("Looking up PDC for domain %s\n", domain));
132         if (!get_pdc_ip(domain, &ss)) {
133                 TALLOC_FREE( ctx );
134                 SAFE_FREE(sitename);
135                 return -1;
136         }
137
138         ret = sys_getnameinfo((struct sockaddr *)&ss,
139                         sizeof(struct sockaddr_storage),
140                         h_name, sizeof(h_name),
141                         NULL, 0,
142                         NI_NAMEREQD);
143
144         if (ret) {
145                 TALLOC_FREE( ctx );
146                 SAFE_FREE(sitename);
147                 return -1;
148         }
149
150         DEBUG(9, ("Found PDC with DNS name %s\n", h_name));
151         domain = strchr(h_name, '.');
152         if (!domain) {
153                 TALLOC_FREE( ctx );
154                 SAFE_FREE(sitename);
155                 return -1;
156         }
157         domain++;
158
159         DEBUG(9, ("Looking up ldap for domain %s\n", domain));
160
161         status = ads_dns_query_dcs( ctx, domain, sitename, &dcs, &numdcs );
162         if ( NT_STATUS_IS_OK(status) && numdcs ) {
163                 print_ldap_srvlist(dcs, numdcs);
164                 TALLOC_FREE( ctx );
165                 SAFE_FREE(sitename);
166                 return 0;
167         }
168
169         TALLOC_FREE( ctx );
170         SAFE_FREE(sitename);
171
172         return -1;
173 #endif
174         DEBUG(1,("No ADS support\n"));
175         return -1;
176 }
177
178 static int net_lookup_dc(struct net_context *c, int argc, const char **argv)
179 {
180         struct ip_service *ip_list;
181         struct sockaddr_storage ss;
182         char *pdc_str = NULL;
183         const char *domain = NULL;
184         char *sitename = NULL;
185         int count, i;
186         char addr[INET6_ADDRSTRLEN];
187         bool sec_ads = (lp_security() == SEC_ADS);
188
189         if (sec_ads) {
190                 domain = lp_realm();
191         } else {
192                 domain = c->opt_target_workgroup;
193         }
194
195         if (argc > 0)
196                 domain=argv[0];
197
198         /* first get PDC */
199         if (!get_pdc_ip(domain, &ss))
200                 return -1;
201
202         print_sockaddr(addr, sizeof(addr), &ss);
203         if (asprintf(&pdc_str, "%s", addr) == -1) {
204                 return -1;
205         }
206         d_printf("%s\n", pdc_str);
207
208         sitename = sitename_fetch(domain);
209         if (!NT_STATUS_IS_OK(get_sorted_dc_list(domain, sitename,
210                                         &ip_list, &count, sec_ads))) {
211                 SAFE_FREE(pdc_str);
212                 SAFE_FREE(sitename);
213                 return 0;
214         }
215         SAFE_FREE(sitename);
216         for (i=0;i<count;i++) {
217                 print_sockaddr(addr, sizeof(addr), &ip_list[i].ss);
218                 if (!strequal(pdc_str, addr))
219                         d_printf("%s\n", addr);
220         }
221         SAFE_FREE(pdc_str);
222         return 0;
223 }
224
225 static int net_lookup_pdc(struct net_context *c, int argc, const char **argv)
226 {
227         struct sockaddr_storage ss;
228         char *pdc_str = NULL;
229         const char *domain;
230         char addr[INET6_ADDRSTRLEN];
231
232         if (lp_security() == SEC_ADS) {
233                 domain = lp_realm();
234         } else {
235                 domain = c->opt_target_workgroup;
236         }
237
238         if (argc > 0)
239                 domain=argv[0];
240
241         /* first get PDC */
242         if (!get_pdc_ip(domain, &ss))
243                 return -1;
244
245         print_sockaddr(addr, sizeof(addr), &ss);
246         if (asprintf(&pdc_str, "%s", addr) == -1) {
247                 return -1;
248         }
249         d_printf("%s\n", pdc_str);
250         SAFE_FREE(pdc_str);
251         return 0;
252 }
253
254
255 static int net_lookup_master(struct net_context *c, int argc, const char **argv)
256 {
257         struct sockaddr_storage master_ss;
258         const char *domain = c->opt_target_workgroup;
259         char addr[INET6_ADDRSTRLEN];
260
261         if (argc > 0)
262                 domain=argv[0];
263
264         if (!find_master_ip(domain, &master_ss))
265                 return -1;
266         print_sockaddr(addr, sizeof(addr), &master_ss);
267         d_printf("%s\n", addr);
268         return 0;
269 }
270
271 static int net_lookup_kdc(struct net_context *c, int argc, const char **argv)
272 {
273 #ifdef HAVE_KRB5
274         krb5_error_code rc;
275         krb5_context ctx;
276         struct sockaddr_in *addrs;
277         int num_kdcs,i;
278         krb5_data realm;
279         char **realms;
280
281         initialize_krb5_error_table();
282         rc = krb5_init_context(&ctx);
283         if (rc) {
284                 DEBUG(1,("krb5_init_context failed (%s)\n",
285                          error_message(rc)));
286                 return -1;
287         }
288
289         if (argc>0) {
290                 realm.data = CONST_DISCARD(char *, argv[0]);
291                 realm.length = strlen(argv[0]);
292         } else if (lp_realm() && *lp_realm()) {
293                 realm.data = lp_realm();
294                 realm.length = strlen((const char *)realm.data);
295         } else {
296                 rc = krb5_get_host_realm(ctx, NULL, &realms);
297                 if (rc) {
298                         DEBUG(1,("krb5_gethost_realm failed (%s)\n",
299                                  error_message(rc)));
300                         return -1;
301                 }
302                 realm.data = (char *) *realms;
303                 realm.length = strlen((const char *)realm.data);
304         }
305
306         rc = smb_krb5_locate_kdc(ctx, &realm, (struct sockaddr **)(void *)&addrs, &num_kdcs, 0);
307         if (rc) {
308                 DEBUG(1, ("smb_krb5_locate_kdc failed (%s)\n", error_message(rc)));
309                 return -1;
310         }
311         for (i=0;i<num_kdcs;i++)
312                 if (addrs[i].sin_family == AF_INET)
313                         d_printf("%s:%hd\n", inet_ntoa(addrs[i].sin_addr),
314                                  ntohs(addrs[i].sin_port));
315         return 0;
316
317 #endif
318         DEBUG(1, ("No kerberos support\n"));
319         return -1;
320 }
321
322 static int net_lookup_name(struct net_context *c, int argc, const char **argv)
323 {
324         const char *dom, *name;
325         struct dom_sid sid;
326         enum lsa_SidType type;
327
328         if (argc != 1) {
329                 d_printf("%s\n%s",
330                          _("Usage:"),
331                          _(" net lookup name <name>\n"));
332                 return -1;
333         }
334
335         if (!lookup_name(talloc_tos(), argv[0], LOOKUP_NAME_ALL,
336                          &dom, &name, &sid, &type)) {
337                 d_printf(_("Could not lookup name %s\n"), argv[0]);
338                 return -1;
339         }
340
341         d_printf("%s %d (%s) %s\\%s\n", sid_string_tos(&sid),
342                  type, sid_type_lookup(type), dom, name);
343         return 0;
344 }
345
346 static int net_lookup_sid(struct net_context *c, int argc, const char **argv)
347 {
348         const char *dom, *name;
349         struct dom_sid sid;
350         enum lsa_SidType type;
351
352         if (argc != 1) {
353                 d_printf("%s\n%s",
354                          _("Usage:"),
355                          _(" net lookup sid <sid>\n"));
356                 return -1;
357         }
358
359         if (!string_to_sid(&sid, argv[0])) {
360                 d_printf(_("Could not convert %s to SID\n"), argv[0]);
361                 return -1;
362         }
363
364         if (!lookup_sid(talloc_tos(), &sid,
365                         &dom, &name, &type)) {
366                 d_printf(_("Could not lookup name %s\n"), argv[0]);
367                 return -1;
368         }
369
370         d_printf("%s %d (%s) %s\\%s\n", sid_string_tos(&sid),
371                  type, sid_type_lookup(type), dom, name);
372         return 0;
373 }
374
375 static int net_lookup_dsgetdcname(struct net_context *c, int argc, const char **argv)
376 {
377         NTSTATUS status;
378         const char *domain_name = NULL;
379         const char *site_name = NULL;
380         uint32_t flags = 0;
381         struct netr_DsRGetDCNameInfo *info = NULL;
382         TALLOC_CTX *mem_ctx;
383         char *s = NULL;
384
385         if (argc < 1 || argc > 3) {
386                 d_printf("%s\n%s",
387                          _("Usage:"),
388                          _(" net lookup dsgetdcname "
389                            "<name> <flags> <sitename>\n"));
390                 return -1;
391         }
392
393         mem_ctx = talloc_init("net_lookup_dsgetdcname");
394         if (!mem_ctx) {
395                 return -1;
396         }
397
398         domain_name = argv[0];
399
400         if (argc >= 2)
401                 sscanf(argv[1], "%x", &flags);
402
403         if (!flags) {
404                 flags |= DS_DIRECTORY_SERVICE_REQUIRED;
405         }
406
407         if (argc == 3) {
408                 site_name = argv[2];
409         }
410
411         status = dsgetdcname(mem_ctx, NULL, domain_name, NULL, site_name,
412                              flags, &info);
413         if (!NT_STATUS_IS_OK(status)) {
414                 d_printf(_("failed with: %s\n"), nt_errstr(status));
415                 TALLOC_FREE(mem_ctx);
416                 return -1;
417         }
418
419         s = NDR_PRINT_STRUCT_STRING(mem_ctx, netr_DsRGetDCNameInfo, info);
420         printf("%s\n", s);
421         TALLOC_FREE(s);
422
423         TALLOC_FREE(mem_ctx);
424         return 0;
425 }
426
427
428 /* lookup hosts or IP addresses using internal samba lookup fns */
429 int net_lookup(struct net_context *c, int argc, const char **argv)
430 {
431         int i;
432
433         struct functable table[] = {
434                 {"HOST", net_lookup_host},
435                 {"LDAP", net_lookup_ldap},
436                 {"DC", net_lookup_dc},
437                 {"PDC", net_lookup_pdc},
438                 {"MASTER", net_lookup_master},
439                 {"KDC", net_lookup_kdc},
440                 {"NAME", net_lookup_name},
441                 {"SID", net_lookup_sid},
442                 {"DSGETDCNAME", net_lookup_dsgetdcname},
443                 {NULL, NULL}
444         };
445
446         if (argc < 1) {
447                 d_printf(_("\nUsage: \n"));
448                 return net_lookup_usage(c, argc, argv);
449         }
450         for (i=0; table[i].funcname; i++) {
451                 if (StrCaseCmp(argv[0], table[i].funcname) == 0)
452                         return table[i].fn(c, argc-1, argv+1);
453         }
454
455         /* Default to lookup a hostname so 'net lookup foo#1b' can be
456            used instead of 'net lookup host foo#1b'.  The host syntax
457            is a bit confusing as non #00 names can't really be
458            considered hosts as such. */
459
460         return net_lookup_host(c, argc, argv);
461 }