source: branches/samba-3.5.x/libcli/nbt/tools/nmblookup.c

Last change on this file was 414, checked in by Herwig Bauernfeind, 16 years ago

Samba 3.5.0: Initial import
File size: 9.3 KB
Line 
1/*
2 Unix SMB/CIFS implementation.
3
4 NBT client - used to lookup netbios names
5
6 Copyright (C) Andrew Tridgell 1994-2005
7 Copyright (C) Jelmer Vernooij 2003 (Conversion to popt)
8
9 This program is free software; you can redistribute it and/or modify
10 it under the terms of the GNU General Public License as published by
11 the Free Software Foundation; either version 3 of the License, or
12 (at your option) any later version.
13
14 This program is distributed in the hope that it will be useful,
15 but WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 GNU General Public License for more details.
18
19 You should have received a copy of the GNU General Public License
20 along with this program. If not, see <http://www.gnu.org/licenses/>.
21
22*/
23
24#include "includes.h"
25#include "lib/cmdline/popt_common.h"
26#include "lib/socket/socket.h"
27#include "lib/events/events.h"
28#include "system/network.h"
29#include "system/locale.h"
30#include "lib/socket/netif.h"
31#include "librpc/gen_ndr/nbt.h"
32#include "../libcli/nbt/libnbt.h"
33#include "param/param.h"
34
35/* command line options */
36static struct {
37 const char *broadcast_address;
38 const char *unicast_address;
39 bool find_master;
40 bool wins_lookup;
41 bool node_status;
42 bool root_port;
43 bool lookup_by_ip;
44 bool case_sensitive;
45} options;
46
47/*
48 clean any binary from a node name
49*/
50static const char *clean_name(TALLOC_CTX *mem_ctx, const char *name)
51{
52 char *ret = talloc_strdup(mem_ctx, name);
53 int i;
54 for (i=0;ret[i];i++) {
55 if (!isprint((unsigned char)ret[i])) ret[i] = '.';
56 }
57 return ret;
58}
59
60/*
61 turn a node status flags field into a string
62*/
63static char *node_status_flags(TALLOC_CTX *mem_ctx, uint16_t flags)
64{
65 char *ret;
66 const char *group = " ";
67 const char *type = "B";
68
69 if (flags & NBT_NM_GROUP) {
70 group = "<GROUP>";
71 }
72
73 switch (flags & NBT_NM_OWNER_TYPE) {
74 case NBT_NODE_B:
75 type = "B";
76 break;
77 case NBT_NODE_P:
78 type = "P";
79 break;
80 case NBT_NODE_M:
81 type = "M";
82 break;
83 case NBT_NODE_H:
84 type = "H";
85 break;
86 }
87
88 ret = talloc_asprintf(mem_ctx, "%s %s", group, type);
89
90 if (flags & NBT_NM_DEREGISTER) {
91 ret = talloc_asprintf_append_buffer(ret, " <DEREGISTERING>");
92 }
93 if (flags & NBT_NM_CONFLICT) {
94 ret = talloc_asprintf_append_buffer(ret, " <CONFLICT>");
95 }
96 if (flags & NBT_NM_ACTIVE) {
97 ret = talloc_asprintf_append_buffer(ret, " <ACTIVE>");
98 }
99 if (flags & NBT_NM_PERMANENT) {
100 ret = talloc_asprintf_append_buffer(ret, " <PERMANENT>");
101 }
102
103 return ret;
104}
105
106/* do a single node status */
107static bool do_node_status(struct nbt_name_socket *nbtsock,
108 const char *addr, uint16_t port)
109{
110 struct nbt_name_status io;
111 NTSTATUS status;
112
113 io.in.name.name = "*";
114 io.in.name.type = NBT_NAME_CLIENT;
115 io.in.name.scope = NULL;
116 io.in.dest_addr = addr;
117 io.in.dest_port = port;
118 io.in.timeout = 1;
119 io.in.retries = 2;
120
121 status = nbt_name_status(nbtsock, nbtsock, &io);
122 if (NT_STATUS_IS_OK(status)) {
123 int i;
124 printf("Node status reply from %s\n",
125 io.out.reply_from);
126 for (i=0;i<io.out.status.num_names;i++) {
127 d_printf("\t%-16s <%02x> %s\n",
128 clean_name(nbtsock, io.out.status.names[i].name),
129 io.out.status.names[i].type,
130 node_status_flags(nbtsock, io.out.status.names[i].nb_flags));
131 }
132 printf("\n\tMAC Address = %02X-%02X-%02X-%02X-%02X-%02X\n",
133 io.out.status.statistics.unit_id[0],
134 io.out.status.statistics.unit_id[1],
135 io.out.status.statistics.unit_id[2],
136 io.out.status.statistics.unit_id[3],
137 io.out.status.statistics.unit_id[4],
138 io.out.status.statistics.unit_id[5]);
139 return true;
140 }
141
142 return false;
143}
144
145/* do a single node query */
146static NTSTATUS do_node_query(struct nbt_name_socket *nbtsock,
147 const char *addr,
148 uint16_t port,
149 const char *node_name,
150 enum nbt_name_type node_type,
151 bool broadcast)
152{
153 struct nbt_name_query io;
154 NTSTATUS status;
155 int i;
156
157 io.in.name.name = node_name;
158 io.in.name.type = node_type;
159 io.in.name.scope = NULL;
160 io.in.dest_addr = addr;
161 io.in.dest_port = port;
162 io.in.broadcast = broadcast;
163 io.in.wins_lookup = options.wins_lookup;
164 io.in.timeout = 1;
165 io.in.retries = 2;
166
167 status = nbt_name_query(nbtsock, nbtsock, &io);
168 NT_STATUS_NOT_OK_RETURN(status);
169
170 for (i=0;i<io.out.num_addrs;i++) {
171 printf("%s %s<%02x>\n",
172 io.out.reply_addrs[i],
173 io.out.name.name,
174 io.out.name.type);
175 }
176 if (options.node_status && io.out.num_addrs > 0) {
177 do_node_status(nbtsock, io.out.reply_addrs[0], port);
178 }
179
180 return status;
181}
182
183
184static bool process_one(struct loadparm_context *lp_ctx, struct tevent_context *ev,
185 struct interface *ifaces, const char *name, int nbt_port)
186{
187 TALLOC_CTX *tmp_ctx = talloc_new(NULL);
188 enum nbt_name_type node_type = NBT_NAME_CLIENT;
189 char *node_name, *p;
190 struct socket_address *all_zero_addr;
191 struct nbt_name_socket *nbtsock;
192 NTSTATUS status = NT_STATUS_OK;
193 bool ret = true;
194
195 if (!options.case_sensitive) {
196 name = strupper_talloc(tmp_ctx, name);
197 }
198
199 if (options.find_master) {
200 node_type = NBT_NAME_MASTER;
201 if (*name == '-' || *name == '_') {
202 name = "\01\02__MSBROWSE__\02";
203 node_type = NBT_NAME_MS;
204 }
205 }
206
207 p = strchr(name, '#');
208 if (p) {
209 node_name = talloc_strndup(tmp_ctx, name, PTR_DIFF(p,name));
210 node_type = (enum nbt_name_type)strtol(p+1, NULL, 16);
211 } else {
212 node_name = talloc_strdup(tmp_ctx, name);
213 }
214
215 nbtsock = nbt_name_socket_init(tmp_ctx, ev, lp_iconv_convenience(lp_ctx));
216
217 if (options.root_port) {
218 all_zero_addr = socket_address_from_strings(tmp_ctx, nbtsock->sock->backend_name,
219 "0.0.0.0", NBT_NAME_SERVICE_PORT);
220
221 if (!all_zero_addr) {
222 talloc_free(tmp_ctx);
223 return false;
224 }
225
226 status = socket_listen(nbtsock->sock, all_zero_addr, 0, 0);
227 if (!NT_STATUS_IS_OK(status)) {
228 printf("Failed to bind to local port 137 - %s\n", nt_errstr(status));
229 talloc_free(tmp_ctx);
230 return false;
231 }
232 }
233
234 if (options.lookup_by_ip) {
235 ret = do_node_status(nbtsock, name, nbt_port);
236 talloc_free(tmp_ctx);
237 return ret;
238 }
239
240 if (options.broadcast_address) {
241 status = do_node_query(nbtsock, options.broadcast_address, nbt_port,
242 node_name, node_type, true);
243 } else if (options.unicast_address) {
244 status = do_node_query(nbtsock, options.unicast_address,
245 nbt_port, node_name, node_type, false);
246 } else {
247 int i, num_interfaces;
248
249 num_interfaces = iface_count(ifaces);
250 for (i=0;i<num_interfaces;i++) {
251 const char *bcast = iface_n_bcast(ifaces, i);
252 if (bcast == NULL) continue;
253 status = do_node_query(nbtsock, bcast, nbt_port,
254 node_name, node_type, true);
255 if (NT_STATUS_IS_OK(status)) break;
256 }
257 }
258
259 if (!NT_STATUS_IS_OK(status)) {
260 printf("Lookup failed - %s\n", nt_errstr(status));
261 ret = false;
262 }
263
264 talloc_free(tmp_ctx);
265 return ret;
266}
267
268/*
269 main program
270*/
271int main(int argc, const char *argv[])
272{
273 bool ret = true;
274 struct interface *ifaces;
275 struct tevent_context *ev;
276 poptContext pc;
277 int opt;
278 enum {
279 OPT_BROADCAST_ADDRESS = 1000,
280 OPT_UNICAST_ADDRESS,
281 OPT_FIND_MASTER,
282 OPT_WINS_LOOKUP,
283 OPT_NODE_STATUS,
284 OPT_ROOT_PORT,
285 OPT_LOOKUP_BY_IP,
286 OPT_CASE_SENSITIVE
287 };
288 struct poptOption long_options[] = {
289 POPT_AUTOHELP
290 { "broadcast", 'B', POPT_ARG_STRING, NULL, OPT_BROADCAST_ADDRESS,
291 "Specify address to use for broadcasts", "BROADCAST-ADDRESS" },
292
293 { "unicast", 'U', POPT_ARG_STRING, NULL, OPT_UNICAST_ADDRESS,
294 "Specify address to use for unicast", NULL },
295
296 { "master-browser", 'M', POPT_ARG_NONE, NULL, OPT_FIND_MASTER,
297 "Search for a master browser", NULL },
298
299 { "wins", 'W', POPT_ARG_NONE, NULL, OPT_WINS_LOOKUP,
300 "Do a WINS lookup", NULL },
301
302 { "status", 'S', POPT_ARG_NONE, NULL, OPT_NODE_STATUS,
303 "Lookup node status as well", NULL },
304
305 { "root-port", 'r', POPT_ARG_NONE, NULL, OPT_ROOT_PORT,
306 "Use root port 137 (Win95 only replies to this)", NULL },
307
308 { "lookup-by-ip", 'A', POPT_ARG_NONE, NULL, OPT_LOOKUP_BY_IP,
309 "Do a node status on <name> as an IP Address", NULL },
310
311 { "case-sensitive", 0, POPT_ARG_NONE, NULL, OPT_CASE_SENSITIVE,
312 "Don't uppercase the name before sending", NULL },
313
314 POPT_COMMON_SAMBA
315 { 0, 0, 0, 0 }
316 };
317
318 pc = poptGetContext("nmblookup", argc, argv, long_options,
319 POPT_CONTEXT_KEEP_FIRST);
320
321 poptSetOtherOptionHelp(pc, "<NODE> ...");
322
323 while ((opt = poptGetNextOpt(pc)) != -1) {
324 switch(opt) {
325 case OPT_BROADCAST_ADDRESS:
326 options.broadcast_address = poptGetOptArg(pc);
327 break;
328 case OPT_UNICAST_ADDRESS:
329 options.unicast_address = poptGetOptArg(pc);
330 break;
331 case OPT_FIND_MASTER:
332 options.find_master = true;
333 break;
334 case OPT_WINS_LOOKUP:
335 options.wins_lookup = true;
336 break;
337 case OPT_NODE_STATUS:
338 options.node_status = true;
339 break;
340 case OPT_ROOT_PORT:
341 options.root_port = true;
342 break;
343 case OPT_LOOKUP_BY_IP:
344 options.lookup_by_ip = true;
345 break;
346 case OPT_CASE_SENSITIVE:
347 options.case_sensitive = true;
348 break;
349 }
350 }
351
352 /* swallow argv[0] */
353 poptGetArg(pc);
354
355 if(!poptPeekArg(pc)) {
356 poptPrintUsage(pc, stderr, 0);
357 exit(1);
358 }
359
360 load_interfaces(NULL, lp_interfaces(cmdline_lp_ctx), &ifaces);
361
362 ev = s4_event_context_init(talloc_autofree_context());
363
364 while (poptPeekArg(pc)) {
365 const char *name = poptGetArg(pc);
366
367 ret &= process_one(cmdline_lp_ctx, ev, ifaces, name, lp_nbt_port(cmdline_lp_ctx));
368 }
369
370 talloc_free(ev);
371
372 talloc_free(ifaces);
373
374 poptFreeContext(pc);
375
376 if (!ret) {
377 return 1;
378 }
379
380 return 0;
381}
Note: See TracBrowser for help on using the repository browser.