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 struct user_auth_info {
31 /* How low can we go? */
33 enum tree_level {LEV_WORKGROUP, LEV_SERVER, LEV_SHARE};
34 enum tree_level level = LEV_SHARE;
36 static void usage(void)
39 "Usage: smbtree [options]\n\
41 \t-d debuglevel set debug output level\n\
42 \t-U username user to autheticate as\n\
43 \t-W workgroup workgroup of user to authenticate as\n\
44 \t-D list only domains (workgroups) of tree\n\
45 \t-S list domains and servers of tree\n\
47 The username can be of the form username%%password or\n\
48 workgroup\\username%%password.\n\n\
52 /* Holds a list of workgroups or servers */
55 struct name_list *prev, *next;
56 pstring name, comment;
60 static struct name_list *workgroups, *servers, *shares;
62 static void free_name_list(struct name_list *list)
65 DLIST_REMOVE(list, list);
68 static void add_name(const char *machine_name, uint32 server_type,
69 const char *comment, void *state)
71 struct name_list **name_list = (struct name_list **)state;
72 struct name_list *new_name;
74 new_name = (struct name_list *)malloc(sizeof(struct name_list));
79 ZERO_STRUCTP(new_name);
81 pstrcpy(new_name->name, machine_name);
82 pstrcpy(new_name->comment, comment);
83 new_name->server_type = server_type;
85 DLIST_ADD(*name_list, new_name);
88 /* Return a cli_state pointing at the IPC$ share for the given workgroup */
90 static struct cli_state *get_ipc_connect(char *server,
91 struct user_auth_info *user_info)
93 struct nmb_name calling, called;
94 extern struct in_addr ipzero;
95 struct in_addr server_ip = ipzero;
96 struct cli_state *cli;
101 make_nmb_name(&called, myname, 0x0);
102 make_nmb_name(&calling, server, 0x20);
104 if (is_ipaddress(server))
105 if (!resolve_name(server, &server_ip, 0x20))
109 if (!(cli = cli_initialise(NULL))) {
110 DEBUG(4, ("Unable to initialise cli structure\n"));
114 if (!cli_connect(cli, server, &server_ip)) {
115 DEBUG(4, ("Unable to connect to %s\n", server));
119 if (!cli_session_request(cli, &calling, &called)) {
121 if (!strequal(called.name, "*SMBSERVER")) {
122 make_nmb_name(&called , "*SMBSERVER", 0x20);
125 DEBUG(4, ("Session request failed to %s\n", called.name));
129 if (!cli_negprot(cli)) {
130 DEBUG(4, ("Negprot failed\n"));
134 if (!cli_session_setup(cli, user_info->username, user_info->password,
135 strlen(user_info->password),
137 strlen(user_info->password), server) &&
138 /* try an anonymous login if it failed */
139 !cli_session_setup(cli, "", "", 1,"", 0, server)) {
140 DEBUG(4, ("Session setup failed\n"));
144 DEBUG(4,(" session setup ok\n"));
146 if (!cli_send_tconX(cli, "IPC$", "?????",
148 strlen(user_info->password)+1)) {
149 DEBUG(4, ("Tconx failed\n"));
155 /* Clean up after error */
158 if (cli && cli->initialised)
165 /* Return the IP address and workgroup of a master browser on the
168 static BOOL find_master_ip_bcast(pstring workgroup, struct in_addr *server_ip)
170 struct in_addr *ip_list;
173 /* Go looking for workgroups by broadcasting on the local network */
175 if (!name_resolve_bcast(MSBROWSE, 1, &ip_list, &count)) {
179 for (i = 0; i < count; i++) {
182 if (!name_status_find(0x1d, ip_list[i], name))
185 if (!find_master_ip(name, server_ip))
188 pstrcpy(workgroup, name);
190 DEBUG(4, ("found master browser %s, %s\n",
191 name, inet_ntoa(ip_list[i])));
199 /****************************************************************************
200 display tree of smb workgroups, servers and shares
201 ****************************************************************************/
202 static BOOL get_workgroups(struct user_auth_info *user_info)
204 struct cli_state *cli;
205 struct in_addr server_ip;
206 pstring master_workgroup;
208 /* Try to connect to a #1d name of our current workgroup. If that
209 doesn't work broadcast for a master browser and then jump off
212 pstrcpy(master_workgroup, lp_workgroup());
214 if (!find_master_ip(lp_workgroup(), &server_ip)) {
215 DEBUG(4, ("Unable to find master browser for workgroup %s\n",
217 if (!find_master_ip_bcast(master_workgroup, &server_ip)) {
218 DEBUG(4, ("Unable to find master browser by "
224 if (!(cli = get_ipc_connect(inet_ntoa(server_ip), user_info)))
227 if (!cli_NetServerEnum(cli, master_workgroup,
228 SV_TYPE_DOMAIN_ENUM, add_name, &workgroups))
234 /* Retrieve the list of servers for a given workgroup */
236 static BOOL get_servers(char *workgroup, struct user_auth_info *user_info)
238 struct cli_state *cli;
239 struct in_addr server_ip;
241 /* Open an IPC$ connection to the master browser for the workgroup */
243 if (!find_master_ip(workgroup, &server_ip)) {
244 DEBUG(4, ("Cannot find master browser for workgroup %s\n",
249 if (!(cli = get_ipc_connect(inet_ntoa(server_ip), user_info)))
252 if (!cli_NetServerEnum(cli, workgroup, SV_TYPE_ALL, add_name,
259 static BOOL get_shares(char *server_name, struct user_auth_info *user_info)
261 struct cli_state *cli;
263 if (!(cli = get_ipc_connect(server_name, user_info)))
266 if (!cli_RNetShareEnum(cli, add_name, &shares))
272 static BOOL print_tree(struct user_auth_info *user_info)
274 struct name_list *wg, *sv, *sh;
276 /* List workgroups */
278 if (!get_workgroups(user_info))
281 for (wg = workgroups; wg; wg = wg->next) {
283 printf("%s\n", wg->name);
287 free_name_list(servers);
290 if (level == LEV_WORKGROUP ||
291 !get_servers(wg->name, user_info))
294 for (sv = servers; sv; sv = sv->next) {
296 printf("\t\\\\%-15s\t\t%s\n",
297 sv->name, sv->comment);
301 free_name_list(shares);
304 if (level == LEV_SERVER ||
305 !get_shares(sv->name, user_info))
308 for (sh = shares; sh; sh = sh->next) {
309 printf("\t\t\\\\%s\\%-15s\t%s\n",
310 sv->name, sh->name, sh->comment);
318 /****************************************************************************
320 ****************************************************************************/
321 int main(int argc,char *argv[])
328 pstring servicesf = CONFIGFILE;
329 struct user_auth_info user_info;
330 BOOL got_pass = False;
332 /* Initialise samba stuff */
338 setup_logging(argv[0],True);
342 lp_load(servicesf,True,False,False);
345 if (getenv("USER")) {
346 pstrcpy(user_info.username, getenv("USER"));
348 if ((p=strchr(user_info.username, '%'))) {
350 pstrcpy(user_info.password, p+1);
352 memset(strchr(getenv("USER"), '%') + 1, 'X',
353 strlen(user_info.password));
357 pstrcpy(user_info.workgroup, lp_workgroup());
359 /* Parse command line args */
361 while ((opt = getopt(argc, argv, "U:hd:W:DS")) != EOF) {
364 pstrcpy(user_info.username,optarg);
365 p = strchr(user_info.username,'%');
368 pstrcpy(user_info.password, p+1);
378 DEBUGLEVEL = atoi(optarg);
382 pstrcpy(user_info.workgroup, optarg);
386 level = LEV_WORKGROUP;
394 printf("Unknown option %c (%d)\n", (char)opt, opt);
408 char *pass = getpass("Password: ");
410 pstrcpy(user_info.password, pass);
415 /* Now do our stuff */
417 if (!print_tree(&user_info))