source: vendor/current/libcli/nbt/tools/nmblookup.c

Last change on this file was 988, checked in by Silvan Scherrer, 9 years ago

Samba Server: update vendor to version 4.4.3
File size: 9.5 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#include <string.h>
36
37#define MAX_NETBIOSNAME_LEN 16
38
39/* command line options */
40static struct {
41 const char *broadcast_address;
42 const char *unicast_address;
43 bool find_master;
44 bool wins_lookup;
45 bool node_status;
46 bool root_port;
47 bool lookup_by_ip;
48 bool case_sensitive;
49} options;
50
51/*
52 clean any binary from a node name
53*/
54static const char *clean_name(TALLOC_CTX *mem_ctx, const char *name)
55{
56 char *ret = talloc_strdup(mem_ctx, name);
57 int i;
58 for (i=0;ret[i];i++) {
59 if (!isprint((unsigned char)ret[i])) ret[i] = '.';
60 }
61 return ret;
62}
63
64/*
65 turn a node status flags field into a string
66*/
67static char *node_status_flags(TALLOC_CTX *mem_ctx, uint16_t flags)
68{
69 char *ret;
70 const char *group = " ";
71 const char *type = "B";
72
73 if (flags & NBT_NM_GROUP) {
74 group = "<GROUP>";
75 }
76
77 switch (flags & NBT_NM_OWNER_TYPE) {
78 case NBT_NODE_B:
79 type = "B";
80 break;
81 case NBT_NODE_P:
82 type = "P";
83 break;
84 case NBT_NODE_M:
85 type = "M";
86 break;
87 case NBT_NODE_H:
88 type = "H";
89 break;
90 }
91
92 ret = talloc_asprintf(mem_ctx, "%s %s", group, type);
93
94 if (flags & NBT_NM_DEREGISTER) {
95 ret = talloc_asprintf_append_buffer(ret, " <DEREGISTERING>");
96 }
97 if (flags & NBT_NM_CONFLICT) {
98 ret = talloc_asprintf_append_buffer(ret, " <CONFLICT>");
99 }
100 if (flags & NBT_NM_ACTIVE) {
101 ret = talloc_asprintf_append_buffer(ret, " <ACTIVE>");
102 }
103 if (flags & NBT_NM_PERMANENT) {
104 ret = talloc_asprintf_append_buffer(ret, " <PERMANENT>");
105 }
106
107 return ret;
108}
109
110/* do a single node status */
111static bool do_node_status(struct nbt_name_socket *nbtsock,
112 const char *addr, uint16_t port)
113{
114 struct nbt_name_status io;
115 NTSTATUS status;
116
117 io.in.name.name = "*";
118 io.in.name.type = NBT_NAME_CLIENT;
119 io.in.name.scope = NULL;
120 io.in.dest_addr = addr;
121 io.in.dest_port = port;
122 io.in.timeout = 1;
123 io.in.retries = 2;
124
125 status = nbt_name_status(nbtsock, nbtsock, &io);
126 if (NT_STATUS_IS_OK(status)) {
127 int i;
128 printf("Node status reply from %s\n",
129 io.out.reply_from);
130 for (i=0;i<io.out.status.num_names;i++) {
131 d_printf("\t%-16s <%02x> %s\n",
132 clean_name(nbtsock, io.out.status.names[i].name),
133 io.out.status.names[i].type,
134 node_status_flags(nbtsock, io.out.status.names[i].nb_flags));
135 }
136 printf("\n\tMAC Address = %02X-%02X-%02X-%02X-%02X-%02X\n",
137 io.out.status.statistics.unit_id[0],
138 io.out.status.statistics.unit_id[1],
139 io.out.status.statistics.unit_id[2],
140 io.out.status.statistics.unit_id[3],
141 io.out.status.statistics.unit_id[4],
142 io.out.status.statistics.unit_id[5]);
143 return true;
144 }
145
146 return false;
147}
148
149/* do a single node query */
150static NTSTATUS do_node_query(struct nbt_name_socket *nbtsock,
151 const char *addr,
152 uint16_t port,
153 const char *node_name,
154 enum nbt_name_type node_type,
155 bool broadcast)
156{
157 struct nbt_name_query io;
158 NTSTATUS status;
159 int i;
160
161 io.in.name.name = node_name;
162 io.in.name.type = node_type;
163 io.in.name.scope = NULL;
164 io.in.dest_addr = addr;
165 io.in.dest_port = port;
166 io.in.broadcast = broadcast;
167 io.in.wins_lookup = options.wins_lookup;
168 io.in.timeout = 1;
169 io.in.retries = 2;
170
171 status = nbt_name_query(nbtsock, nbtsock, &io);
172 NT_STATUS_NOT_OK_RETURN(status);
173
174 for (i=0;i<io.out.num_addrs;i++) {
175 printf("%s %s<%02x>\n",
176 io.out.reply_addrs[i],
177 io.out.name.name,
178 io.out.name.type);
179 }
180 if (options.node_status && io.out.num_addrs > 0) {
181 do_node_status(nbtsock, io.out.reply_addrs[0], port);
182 }
183
184 return status;
185}
186
187
188static bool process_one(struct loadparm_context *lp_ctx, struct tevent_context *ev,
189 struct interface *ifaces, const char *name, int nbt_port)
190{
191 TALLOC_CTX *tmp_ctx = talloc_new(NULL);
192 enum nbt_name_type node_type = NBT_NAME_CLIENT;
193 char *node_name, *p;
194 struct socket_address *all_zero_addr;
195 struct nbt_name_socket *nbtsock;
196 NTSTATUS status = NT_STATUS_OK;
197 size_t nbt_len;
198 bool ret = true;
199
200 if (!options.case_sensitive) {
201 name = strupper_talloc(tmp_ctx, name);
202 }
203
204 if (options.find_master) {
205 node_type = NBT_NAME_MASTER;
206 if (*name == '-' || *name == '_') {
207 name = "\01\02__MSBROWSE__\02";
208 node_type = NBT_NAME_MS;
209 }
210 }
211
212 p = strchr(name, '#');
213 if (p) {
214 node_name = talloc_strndup(tmp_ctx, name, PTR_DIFF(p,name));
215 node_type = (enum nbt_name_type)strtol(p+1, NULL, 16);
216 } else {
217 node_name = talloc_strdup(tmp_ctx, name);
218 }
219
220 nbt_len = strlen(node_name);
221 if (nbt_len > MAX_NETBIOSNAME_LEN - 1) {
222 printf("The specified netbios name [%s] is too long.\n",
223 node_name);
224 talloc_free(tmp_ctx);
225 return false;
226 }
227
228 nbtsock = nbt_name_socket_init(tmp_ctx, ev);
229
230 if (options.root_port) {
231 all_zero_addr = socket_address_from_strings(tmp_ctx, nbtsock->sock->backend_name,
232 "0.0.0.0", NBT_NAME_SERVICE_PORT);
233
234 if (!all_zero_addr) {
235 talloc_free(tmp_ctx);
236 return false;
237 }
238
239 status = socket_listen(nbtsock->sock, all_zero_addr, 0, 0);
240 if (!NT_STATUS_IS_OK(status)) {
241 printf("Failed to bind to local port 137 - %s\n", nt_errstr(status));
242 talloc_free(tmp_ctx);
243 return false;
244 }
245 }
246
247 if (options.lookup_by_ip) {
248 ret = do_node_status(nbtsock, name, nbt_port);
249 talloc_free(tmp_ctx);
250 return ret;
251 }
252
253 if (options.broadcast_address) {
254 status = do_node_query(nbtsock, options.broadcast_address, nbt_port,
255 node_name, node_type, true);
256 } else if (options.unicast_address) {
257 status = do_node_query(nbtsock, options.unicast_address,
258 nbt_port, node_name, node_type, false);
259 } else {
260 int i, num_interfaces;
261
262 num_interfaces = iface_list_count(ifaces);
263 for (i=0;i<num_interfaces;i++) {
264 const char *bcast = iface_list_n_bcast(ifaces, i);
265 if (bcast == NULL) continue;
266 status = do_node_query(nbtsock, bcast, nbt_port,
267 node_name, node_type, true);
268 if (NT_STATUS_IS_OK(status)) break;
269 }
270 }
271
272 if (!NT_STATUS_IS_OK(status)) {
273 printf("Lookup failed - %s\n", nt_errstr(status));
274 ret = false;
275 }
276
277 talloc_free(tmp_ctx);
278 return ret;
279}
280
281/*
282 main program
283*/
284int main(int argc, const char *argv[])
285{
286 bool ret = true;
287 struct interface *ifaces;
288 struct tevent_context *ev;
289 poptContext pc;
290 int opt;
291 enum {
292 OPT_BROADCAST_ADDRESS = 1000,
293 OPT_UNICAST_ADDRESS,
294 OPT_FIND_MASTER,
295 OPT_WINS_LOOKUP,
296 OPT_NODE_STATUS,
297 OPT_ROOT_PORT,
298 OPT_LOOKUP_BY_IP,
299 OPT_CASE_SENSITIVE
300 };
301 struct poptOption long_options[] = {
302 POPT_AUTOHELP
303 { "broadcast", 'B', POPT_ARG_STRING, NULL, OPT_BROADCAST_ADDRESS,
304 "Specify address to use for broadcasts", "BROADCAST-ADDRESS" },
305
306 { "unicast", 'U', POPT_ARG_STRING, NULL, OPT_UNICAST_ADDRESS,
307 "Specify address to use for unicast", NULL },
308
309 { "master-browser", 'M', POPT_ARG_NONE, NULL, OPT_FIND_MASTER,
310 "Search for a master browser", NULL },
311
312 { "wins", 'W', POPT_ARG_NONE, NULL, OPT_WINS_LOOKUP,
313 "Do a WINS lookup", NULL },
314
315 { "status", 'S', POPT_ARG_NONE, NULL, OPT_NODE_STATUS,
316 "Lookup node status as well", NULL },
317
318 { "root-port", 'r', POPT_ARG_NONE, NULL, OPT_ROOT_PORT,
319 "Use root port 137 (Win95 only replies to this)", NULL },
320
321 { "lookup-by-ip", 'A', POPT_ARG_NONE, NULL, OPT_LOOKUP_BY_IP,
322 "Do a node status on <name> as an IP Address", NULL },
323
324 { "case-sensitive", 0, POPT_ARG_NONE, NULL, OPT_CASE_SENSITIVE,
325 "Don't uppercase the name before sending", NULL },
326
327 POPT_COMMON_SAMBA
328 { 0, 0, 0, 0 }
329 };
330
331 pc = poptGetContext("nmblookup", argc, argv, long_options,
332 POPT_CONTEXT_KEEP_FIRST);
333
334 poptSetOtherOptionHelp(pc, "<NODE> ...");
335
336 while ((opt = poptGetNextOpt(pc)) != -1) {
337 switch(opt) {
338 case OPT_BROADCAST_ADDRESS:
339 options.broadcast_address = poptGetOptArg(pc);
340 break;
341 case OPT_UNICAST_ADDRESS:
342 options.unicast_address = poptGetOptArg(pc);
343 break;
344 case OPT_FIND_MASTER:
345 options.find_master = true;
346 break;
347 case OPT_WINS_LOOKUP:
348 options.wins_lookup = true;
349 break;
350 case OPT_NODE_STATUS:
351 options.node_status = true;
352 break;
353 case OPT_ROOT_PORT:
354 options.root_port = true;
355 break;
356 case OPT_LOOKUP_BY_IP:
357 options.lookup_by_ip = true;
358 break;
359 case OPT_CASE_SENSITIVE:
360 options.case_sensitive = true;
361 break;
362 }
363 }
364
365 /* swallow argv[0] */
366 poptGetArg(pc);
367
368 if(!poptPeekArg(pc)) {
369 poptPrintUsage(pc, stderr, 0);
370 exit(1);
371 }
372
373 load_interface_list(NULL, cmdline_lp_ctx, &ifaces);
374
375 ev = s4_event_context_init(talloc_autofree_context());
376
377 while (poptPeekArg(pc)) {
378 const char *name = poptGetArg(pc);
379
380 ret &= process_one(cmdline_lp_ctx, ev, ifaces, name, lpcfg_nbt_port(cmdline_lp_ctx));
381 }
382
383 talloc_free(ev);
384
385 talloc_free(ifaces);
386
387 poptFreeContext(pc);
388
389 if (!ret) {
390 return 1;
391 }
392
393 return 0;
394}
Note: See TracBrowser for help on using the repository browser.