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