2 Unix SMB/Netbios implementation.
3 Network neighbourhood browser.
6 Copyright (C) Tim Potter 2000
8 This program is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 2 of the License, or
11 (at your option) any later version.
13 This program is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
18 You should have received a copy of the GNU General Public License
19 along with this program; if not, write to the Free Software
20 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
25 static BOOL use_bcast;
27 struct user_auth_info {
33 /* How low can we go? */
35 enum tree_level {LEV_WORKGROUP, LEV_SERVER, LEV_SHARE};
36 enum tree_level level = LEV_SHARE;
38 static void usage(void)
41 "Usage: smbtree [options]\n\
43 \t-d debuglevel set debug output level\n\
44 \t-U username user to autheticate as\n\
45 \t-W workgroup workgroup of user to authenticate as\n\
46 \t-D list only domains (workgroups) of tree\n\
47 \t-S list domains and servers of tree\n\
48 \t-b use bcast instead of using the master browser\n\
50 The username can be of the form username%%password or\n\
51 workgroup\\username%%password.\n\n\
55 /* Holds a list of workgroups or servers */
58 struct name_list *prev, *next;
59 pstring name, comment;
63 static struct name_list *workgroups, *servers, *shares;
65 static void free_name_list(struct name_list *list)
68 DLIST_REMOVE(list, list);
71 static void add_name(const char *machine_name, uint32 server_type,
72 const char *comment, void *state)
74 struct name_list **name_list = (struct name_list **)state;
75 struct name_list *new_name;
77 new_name = (struct name_list *)malloc(sizeof(struct name_list));
82 ZERO_STRUCTP(new_name);
84 pstrcpy(new_name->name, machine_name);
85 pstrcpy(new_name->comment, comment);
86 new_name->server_type = server_type;
88 DLIST_ADD(*name_list, new_name);
91 /* Return a cli_state pointing at the IPC$ share for the given workgroup */
93 static struct cli_state *get_ipc_connect(char *server,
94 struct user_auth_info *user_info)
96 struct nmb_name calling, called;
97 extern struct in_addr ipzero;
98 struct in_addr server_ip = ipzero;
99 struct cli_state *cli;
104 make_nmb_name(&called, myname, 0x0);
105 make_nmb_name(&calling, server, 0x20);
107 if (is_ipaddress(server))
108 if (!resolve_name(server, &server_ip, 0x20))
112 if (!(cli = cli_initialise(NULL))) {
113 DEBUG(4, ("Unable to initialise cli structure\n"));
117 if (!cli_connect(cli, server, &server_ip)) {
118 DEBUG(4, ("Unable to connect to %s\n", server));
122 if (!cli_session_request(cli, &calling, &called)) {
124 if (!strequal(called.name, "*SMBSERVER")) {
125 make_nmb_name(&called , "*SMBSERVER", 0x20);
128 DEBUG(4, ("Session request failed to %s\n", called.name));
132 if (!cli_negprot(cli)) {
133 DEBUG(4, ("Negprot failed\n"));
137 if (!cli_session_setup(cli, user_info->username, user_info->password,
138 strlen(user_info->password),
140 strlen(user_info->password), server) &&
141 /* try an anonymous login if it failed */
142 !cli_session_setup(cli, "", "", 1,"", 0, server)) {
143 DEBUG(4, ("Session setup failed\n"));
147 DEBUG(4,(" session setup ok\n"));
149 if (!cli_send_tconX(cli, "IPC$", "?????",
151 strlen(user_info->password)+1)) {
152 DEBUG(4, ("Tconx failed\n"));
158 /* Clean up after error */
161 if (cli && cli->initialised)
168 /* Return the IP address and workgroup of a master browser on the
171 static BOOL find_master_ip_bcast(pstring workgroup, struct in_addr *server_ip)
173 struct in_addr *ip_list;
176 /* Go looking for workgroups by broadcasting on the local network */
178 if (!name_resolve_bcast(MSBROWSE, 1, &ip_list, &count)) {
182 for (i = 0; i < count; i++) {
185 if (!name_status_find(0x1d, ip_list[i], name))
188 if (!find_master_ip(name, server_ip))
191 pstrcpy(workgroup, name);
193 DEBUG(4, ("found master browser %s, %s\n",
194 name, inet_ntoa(ip_list[i])));
202 /****************************************************************************
203 display tree of smb workgroups, servers and shares
204 ****************************************************************************/
205 static BOOL get_workgroups(struct user_auth_info *user_info)
207 struct cli_state *cli;
208 struct in_addr server_ip;
209 pstring master_workgroup;
211 /* Try to connect to a #1d name of our current workgroup. If that
212 doesn't work broadcast for a master browser and then jump off
215 pstrcpy(master_workgroup, lp_workgroup());
217 if (use_bcast || !find_master_ip(lp_workgroup(), &server_ip)) {
218 DEBUG(4, ("Unable to find master browser for workgroup %s\n",
220 if (!find_master_ip_bcast(master_workgroup, &server_ip)) {
221 DEBUG(4, ("Unable to find master browser by "
227 if (!(cli = get_ipc_connect(inet_ntoa(server_ip), user_info)))
230 if (!cli_NetServerEnum(cli, master_workgroup,
231 SV_TYPE_DOMAIN_ENUM, add_name, &workgroups))
237 /* Retrieve the list of servers for a given workgroup */
239 static BOOL get_servers(char *workgroup, struct user_auth_info *user_info)
241 struct cli_state *cli;
242 struct in_addr server_ip;
244 /* Open an IPC$ connection to the master browser for the workgroup */
246 if (!find_master_ip(workgroup, &server_ip)) {
247 DEBUG(4, ("Cannot find master browser for workgroup %s\n",
252 if (!(cli = get_ipc_connect(inet_ntoa(server_ip), user_info)))
255 if (!cli_NetServerEnum(cli, workgroup, SV_TYPE_ALL, add_name,
262 static BOOL get_shares(char *server_name, struct user_auth_info *user_info)
264 struct cli_state *cli;
266 if (!(cli = get_ipc_connect(server_name, user_info)))
269 if (!cli_RNetShareEnum(cli, add_name, &shares))
275 static BOOL print_tree(struct user_auth_info *user_info)
277 struct name_list *wg, *sv, *sh;
279 /* List workgroups */
281 if (!get_workgroups(user_info))
284 for (wg = workgroups; wg; wg = wg->next) {
286 printf("%s\n", wg->name);
290 free_name_list(servers);
293 if (level == LEV_WORKGROUP ||
294 !get_servers(wg->name, user_info))
297 for (sv = servers; sv; sv = sv->next) {
299 printf("\t\\\\%-15s\t\t%s\n",
300 sv->name, sv->comment);
304 free_name_list(shares);
307 if (level == LEV_SERVER ||
308 !get_shares(sv->name, user_info))
311 for (sh = shares; sh; sh = sh->next) {
312 printf("\t\t\\\\%s\\%-15s\t%s\n",
313 sv->name, sh->name, sh->comment);
321 /****************************************************************************
323 ****************************************************************************/
324 int main(int argc,char *argv[])
331 pstring servicesf = CONFIGFILE;
332 struct user_auth_info user_info;
333 BOOL got_pass = False;
335 /* Initialise samba stuff */
341 setup_logging(argv[0],True);
345 lp_load(servicesf,True,False,False);
348 if (getenv("USER")) {
349 pstrcpy(user_info.username, getenv("USER"));
351 if ((p=strchr(user_info.username, '%'))) {
353 pstrcpy(user_info.password, p+1);
355 memset(strchr(getenv("USER"), '%') + 1, 'X',
356 strlen(user_info.password));
360 pstrcpy(user_info.workgroup, lp_workgroup());
362 /* Parse command line args */
364 while ((opt = getopt(argc, argv, "U:hd:W:DSb")) != EOF) {
367 pstrcpy(user_info.username,optarg);
368 p = strchr(user_info.username,'%');
371 pstrcpy(user_info.password, p+1);
385 DEBUGLEVEL = atoi(optarg);
389 pstrcpy(user_info.workgroup, optarg);
393 level = LEV_WORKGROUP;
401 printf("Unknown option %c (%d)\n", (char)opt, opt);
415 char *pass = getpass("Password: ");
417 pstrcpy(user_info.password, pass);
422 /* Now do our stuff */
424 if (!print_tree(&user_info))