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