2 Unix SMB/CIFS implementation.
3 NBT client - used to lookup netbios names
4 Copyright (C) Andrew Tridgell 1994-1998
5 Copyright (C) Jelmer Vernooij 2003 (Conversion to popt)
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 2 of the License, or
10 (at your option) any later version.
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with this program; if not, write to the Free Software
19 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
27 extern BOOL AllowDebugChange
;
29 static BOOL give_flags
= False
;
30 static BOOL use_bcast
= True
;
31 static BOOL got_bcast
= False
;
32 static struct in_addr bcast_addr
;
33 static BOOL recursion_desired
= False
;
34 static BOOL translate_addresses
= False
;
35 static int ServerFD
= -1;
36 static int RootPort
= False
;
37 static BOOL find_status
=False
;
39 /****************************************************************************
40 open the socket communication
41 **************************************************************************/
42 static BOOL
open_sockets(void)
44 ServerFD
= open_socket_in( SOCK_DGRAM
,
47 interpret_addr(lp_socket_address()), True
);
52 set_socket_options( ServerFD
, "SO_BROADCAST" );
54 DEBUG(3, ("Socket opened.\n"));
58 /****************************************************************************
59 turn a node status flags field into a string
60 ****************************************************************************/
61 static char *node_status_flags(unsigned char flags
)
66 fstrcat(ret
, (flags
& 0x80) ? "<GROUP> " : " ");
67 if ((flags
& 0x60) == 0x00) fstrcat(ret
,"B ");
68 if ((flags
& 0x60) == 0x20) fstrcat(ret
,"P ");
69 if ((flags
& 0x60) == 0x40) fstrcat(ret
,"M ");
70 if ((flags
& 0x60) == 0x60) fstrcat(ret
,"H ");
71 if (flags
& 0x10) fstrcat(ret
,"<DEREGISTERING> ");
72 if (flags
& 0x08) fstrcat(ret
,"<CONFLICT> ");
73 if (flags
& 0x04) fstrcat(ret
,"<ACTIVE> ");
74 if (flags
& 0x02) fstrcat(ret
,"<PERMANENT> ");
79 /****************************************************************************
80 turn the NMB Query flags into a string
81 ****************************************************************************/
82 static char *query_flags(int flags
)
87 if (flags
& NM_FLAGS_RS
) fstrcat(ret1
, "Response ");
88 if (flags
& NM_FLAGS_AA
) fstrcat(ret1
, "Authoritative ");
89 if (flags
& NM_FLAGS_TC
) fstrcat(ret1
, "Truncated ");
90 if (flags
& NM_FLAGS_RD
) fstrcat(ret1
, "Recursion_Desired ");
91 if (flags
& NM_FLAGS_RA
) fstrcat(ret1
, "Recursion_Available ");
92 if (flags
& NM_FLAGS_B
) fstrcat(ret1
, "Broadcast ");
97 /****************************************************************************
98 do a node status query
99 ****************************************************************************/
100 static void do_node_status(int fd
, const char *name
, int type
, struct in_addr ip
)
102 struct nmb_name nname
;
104 struct node_status
*status
;
105 struct node_status_extra extra
;
108 d_printf("Looking up status of %s\n",inet_ntoa(ip
));
109 make_nmb_name(&nname
, name
, type
);
110 status
= node_status_query(fd
,&nname
,ip
, &count
, &extra
);
112 for (i
=0;i
<count
;i
++) {
113 pull_ascii_fstring(cleanname
, status
[i
].name
);
114 for (j
=0;cleanname
[j
];j
++) {
115 if (!isprint((int)cleanname
[j
])) cleanname
[j
] = '.';
117 d_printf("\t%-15s <%02x> - %s\n",
118 cleanname
,status
[i
].type
,
119 node_status_flags(status
[i
].flags
));
123 d_printf("\n\tMAC Address = %02X-%02X-%02X-%02X-%02X-%02X\n",
124 extra
.mac_addr
[0], extra
.mac_addr
[1], extra
.mac_addr
[2],
125 extra
.mac_addr
[3], extra
.mac_addr
[4], extra
.mac_addr
[5]);
130 /****************************************************************************
132 ****************************************************************************/
133 static BOOL
query_one(const char *lookup
, unsigned int lookup_type
)
135 int j
, count
, flags
= 0;
136 struct in_addr
*ip_list
=NULL
;
139 d_printf("querying %s on %s\n", lookup
, inet_ntoa(bcast_addr
));
140 ip_list
= name_query(ServerFD
,lookup
,lookup_type
,use_bcast
,
141 use_bcast
?True
:recursion_desired
,
142 bcast_addr
,&count
, &flags
, NULL
);
144 struct in_addr
*bcast
;
145 for (j
=iface_count() - 1;
148 bcast
= iface_n_bcast(j
);
149 d_printf("querying %s on %s\n",
150 lookup
, inet_ntoa(*bcast
));
151 ip_list
= name_query(ServerFD
,lookup
,lookup_type
,
153 use_bcast
?True
:recursion_desired
,
154 *bcast
,&count
, &flags
, NULL
);
158 if (!ip_list
) return False
;
161 d_printf("Flags: %s\n", query_flags(flags
));
163 for (j
=0;j
<count
;j
++) {
164 if (translate_addresses
) {
165 struct hostent
*host
= gethostbyaddr((char *)&ip_list
[j
], sizeof(ip_list
[j
]), AF_INET
);
167 d_printf("%s, ", host
-> h_name
);
170 d_printf("%s %s<%02x>\n",inet_ntoa(ip_list
[j
]),lookup
, lookup_type
);
173 /* We can only do find_status if the ip address returned
174 was valid - ie. name_query returned true.
177 do_node_status(ServerFD
, lookup
, lookup_type
, ip_list
[0]);
182 return (ip_list
!= NULL
);
186 /****************************************************************************
188 ****************************************************************************/
189 int main(int argc
,char *argv
[])
192 unsigned int lookup_type
= 0x0;
194 static BOOL find_master
=False
;
195 static BOOL lookup_by_ip
= False
;
198 struct poptOption long_options
[] = {
200 { "broadcast", 'B', POPT_ARG_STRING
, NULL
, 'B', "Specify address to use for broadcasts", "BROADCAST-ADDRESS" },
201 { "flags", 'f', POPT_ARG_VAL
, &give_flags
, True
, "List the NMB flags returned" },
202 { "unicast", 'U', POPT_ARG_STRING
, NULL
, 'U', "Specify address to use for unicast" },
203 { "master-browser", 'M', POPT_ARG_VAL
, &find_master
, True
, "Search for a master browser" },
204 { "recursion", 'R', POPT_ARG_VAL
, &recursion_desired
, True
, "Set recursion desired in package" },
205 { "status", 'S', POPT_ARG_VAL
, &find_status
, True
, "Lookup node status as well" },
206 { "translate", 'T', POPT_ARG_NONE
, NULL
, 'T', "Translate IP addresses into names" },
207 { "root-port", 'r', POPT_ARG_VAL
, &RootPort
, True
, "Use root port 137 (Win95 only replies to this)" },
208 { "lookup-by-ip", 'A', POPT_ARG_VAL
, &lookup_by_ip
, True
, "Do a node status on <name> as an IP Address" },
210 POPT_COMMON_CONNECTION
216 setup_logging(argv
[0],True
);
218 pc
= poptGetContext("nmblookup", argc
, (const char **)argv
, long_options
,
219 POPT_CONTEXT_KEEP_FIRST
);
221 poptSetOtherOptionHelp(pc
, "<NODE> ...");
223 while ((opt
= poptGetNextOpt(pc
)) != -1) {
226 bcast_addr
= *interpret_addr2(poptGetOptArg(pc
));
231 bcast_addr
= *interpret_addr2(poptGetOptArg(pc
));
236 translate_addresses
= !translate_addresses
;
241 poptGetArg(pc
); /* Remove argv[0] */
243 if(!poptPeekArg(pc
)) {
244 poptPrintUsage(pc
, stderr
, 0);
248 if (!lp_load(dyn_CONFIGFILE
,True
,False
,False
)) {
249 fprintf(stderr
, "Can't load %s - run testparm to debug it\n", dyn_CONFIGFILE
);
253 if (!open_sockets()) return(1);
255 while(poptPeekArg(pc
))
260 fstrcpy(lookup
,poptGetArg(pc
));
264 ip
= *interpret_addr2(lookup
);
266 do_node_status(ServerFD
, lookup
, lookup_type
, ip
);
271 if (*lookup
== '-') {
272 fstrcpy(lookup
,"\01\02__MSBROWSE__\02");
279 p
= strchr_m(lookup
,'#');
282 sscanf(++p
,"%x",&lookup_type
);
285 if (!query_one(lookup
, lookup_type
)) {
286 d_printf( "name_query failed to find name %s", lookup
);
287 if( 0 != lookup_type
)
288 d_printf( "#%02x", lookup_type
);