b81e37c0af4885461d1605f4deb0edd8b6aa913d
[samba.git] / source / utils / net.c
1 /* 
2    Samba Unix/Linux SMB client library 
3    Distributed SMB/CIFS Server Management Utility 
4    Copyright (C) 2001 Steve French  (sfrench@us.ibm.com)
5    Copyright (C) 2001 Jim McDonough (jmcd@us.ibm.com)
6    Copyright (C) 2001 Andrew Tridgell (tridge@samba.org)
7    Copyright (C) 2001 Andrew Bartlett (abartlet@samba.org)
8
9    Originally written by Steve and Jim. Largely rewritten by tridge in
10    November 2001.
11
12    Reworked again by abartlet in December 2001
13
14    This program is free software; you can redistribute it and/or modify
15    it under the terms of the GNU General Public License as published by
16    the Free Software Foundation; either version 2 of the License, or
17    (at your option) any later version.
18    
19    This program is distributed in the hope that it will be useful,
20    but WITHOUT ANY WARRANTY; without even the implied warranty of
21    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
22    GNU General Public License for more details.
23    
24    You should have received a copy of the GNU General Public License
25    along with this program; if not, write to the Free Software
26    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
27  
28 /*****************************************************/
29 /*                                                   */
30 /*   Distributed SMB/CIFS Server Management Utility  */
31 /*                                                   */
32 /*   The intent was to make the syntax similar       */
33 /*   to the NET utility (first developed in DOS      */
34 /*   with additional interesting & useful functions  */
35 /*   added in later SMB server network operating     */
36 /*   systems).                                       */
37 /*                                                   */
38 /*****************************************************/
39
40 #include "includes.h"
41 #include "../utils/net.h"
42
43 /***********************************************************************/
44 /* Beginning of internationalization section.  Translatable constants  */
45 /* should be kept in this area and referenced in the rest of the code. */
46 /*                                                                     */
47 /* No functions, outside of Samba or LSB (Linux Standards Base) should */
48 /* be used (if possible).                                              */
49 /***********************************************************************/
50
51 #define YES_STRING              "Yes"
52 #define NO_STRING               "No"
53
54 /************************************************************************************/
55 /*                       end of internationalization section                        */
56 /************************************************************************************/
57
58 /* Yes, these buggers are globals.... */
59 char *opt_requester_name = NULL;
60 char *opt_host = NULL; 
61 char *opt_password = NULL;
62 char *opt_user_name = NULL;
63 BOOL opt_user_specified = False;
64 char *opt_workgroup = NULL;
65 int opt_long_list_entries = 0;
66 int opt_reboot = 0;
67 int opt_force = 0;
68 int opt_port = 0;
69 int opt_maxusers = -1;
70 char *opt_comment = "";
71 int opt_flags = -1;
72 int opt_jobid = 0;
73 int opt_timeout = 0;
74 char *opt_target_workgroup = NULL;
75
76 BOOL opt_have_ip = False;
77 struct in_addr opt_dest_ip;
78
79 extern pstring global_myname;
80
81 /*
82   run a function from a function table. If not found then
83   call the specified usage function 
84 */
85 int net_run_function(int argc, const char **argv, struct functable *table, 
86                      int (*usage_fn)(int argc, const char **argv))
87 {
88         int i;
89         
90         if (argc < 1) {
91                 d_printf("\nUsage: \n");
92                 return usage_fn(argc, argv);
93         }
94         for (i=0; table[i].funcname; i++) {
95                 if (StrCaseCmp(argv[0], table[i].funcname) == 0)
96                         return table[i].fn(argc-1, argv+1);
97         }
98         d_printf("No command: %s\n", argv[0]);
99         return usage_fn(argc, argv);
100 }
101
102
103 /****************************************************************************
104 connect to \\server\ipc$  
105 ****************************************************************************/
106 NTSTATUS connect_to_ipc(struct cli_state **c, struct in_addr *server_ip,
107                                         const char *server_name)
108 {
109         NTSTATUS nt_status;
110
111         if (!opt_password) {
112                 char *pass = getpass("Password:");
113                 if (pass) {
114                         opt_password = strdup(pass);
115                 }
116         }
117         
118         nt_status = cli_full_connection(c, opt_requester_name, server_name, 
119                                         server_ip, opt_port,
120                                         "IPC$", "IPC",  
121                                         opt_user_name, opt_workgroup,
122                                         opt_password, strlen(opt_password));
123         
124         if (NT_STATUS_IS_OK(nt_status)) {
125                 return nt_status;
126         } else {
127                 DEBUG(1,("Cannot connect to server.  Error was %s\n", 
128                          nt_errstr(nt_status)));
129
130                 /* Display a nicer message depending on the result */
131
132                 if (NT_STATUS_V(nt_status) == 
133                     NT_STATUS_V(NT_STATUS_LOGON_FAILURE))
134                         d_printf("The username or password was not correct.\n");
135
136                 return nt_status;
137         }
138 }
139
140 /****************************************************************************
141 connect to \\server\ipc$ anonymously
142 ****************************************************************************/
143 NTSTATUS connect_to_ipc_anonymous(struct cli_state **c,
144                         struct in_addr *server_ip, const char *server_name)
145 {
146         NTSTATUS nt_status;
147
148         nt_status = cli_full_connection(c, opt_requester_name, server_name, 
149                                         server_ip, opt_port,
150                                         "IPC$", "IPC",  
151                                         "", "",
152                                         "", 0);
153         
154         if (NT_STATUS_IS_OK(nt_status)) {
155                 return nt_status;
156         } else {
157                 DEBUG(1,("Cannot connect to server (anonymously).  Error was %s\n", nt_errstr(nt_status)));
158                 return nt_status;
159         }
160 }
161
162 BOOL net_find_server(unsigned flags, struct in_addr *server_ip, char **server_name)
163 {
164
165         if (opt_host) {
166                 *server_name = strdup(opt_host);
167         }               
168
169         if (opt_have_ip) {
170                 *server_ip = opt_dest_ip;
171                 if (!*server_name) {
172                         *server_name = strdup(inet_ntoa(opt_dest_ip));
173                 }
174         } else if (*server_name) {
175                 /* resolve the IP address */
176                 if (!resolve_name(*server_name, server_ip, 0x20))  {
177                         DEBUG(1,("Unable to resolve server name\n"));
178                         return False;
179                 }
180         } else if (flags & NET_FLAGS_PDC) {
181                 struct in_addr *ip_list;
182                 int addr_count;
183                 if (get_dc_list(True /* PDC only*/, opt_target_workgroup, &ip_list, &addr_count)) {
184                         fstring dc_name;
185                         if (addr_count < 1) {
186                                 return False;
187                         }
188                         
189                         *server_ip = *ip_list;
190                         
191                         if (is_zero_ip(*server_ip))
192                                 return False;
193                         
194                         if (!lookup_dc_name(global_myname, opt_target_workgroup, server_ip, dc_name))
195                                 return False;
196                                 
197                         *server_name = strdup(dc_name);
198                 }
199                 
200         } else if (flags & NET_FLAGS_DMB) {
201                 struct in_addr msbrow_ip;
202                 /*  if (!resolve_name(MSBROWSE, &msbrow_ip, 1)) */
203                 if (!resolve_name(opt_target_workgroup, &msbrow_ip, 0x1B))  {
204                         DEBUG(1,("Unable to resolve domain browser via name lookup\n"));
205                         return False;
206                 } else {
207                         *server_ip = msbrow_ip;
208                 }
209                 *server_name = strdup(inet_ntoa(opt_dest_ip));
210         } else if (flags & NET_FLAGS_MASTER) {
211                 struct in_addr brow_ips;
212                 if (!resolve_name(opt_target_workgroup, &brow_ips, 0x1D))  {
213                                 /* go looking for workgroups */
214                         DEBUG(1,("Unable to resolve master browser via name lookup\n"));
215                         return False;
216                 } else {
217                         *server_ip = brow_ips;
218                 }
219                 *server_name = strdup(inet_ntoa(opt_dest_ip));
220         } else if (!(flags & NET_FLAGS_LOCALHOST_DEFAULT_INSANE)) {
221                 extern struct in_addr loopback_ip;
222                 *server_ip = loopback_ip;
223                 *server_name = strdup("127.0.0.1");
224         }
225
226         if (!server_name || !*server_name) {
227                 DEBUG(1,("no server to connect to\n"));
228                 return False;
229         }
230
231         return True;
232 }
233
234
235 BOOL net_find_dc(struct in_addr *server_ip, fstring server_name, char *domain_name)
236 {
237         struct in_addr *ip_list;
238         int addr_count;
239
240         if (get_dc_list(True /* PDC only*/, domain_name, &ip_list, &addr_count)) {
241                 fstring dc_name;
242                 if (addr_count < 1) {
243                         return False;
244                 }
245                         
246                 *server_ip = *ip_list;
247                 
248                 if (is_zero_ip(*server_ip))
249                         return False;
250                 
251                 if (!lookup_dc_name(global_myname, domain_name, server_ip, dc_name))
252                         return False;
253                         
254                 safe_strcpy(server_name, dc_name, FSTRING_LEN);
255                 return True;
256         } else
257                 return False;
258 }
259
260
261 struct cli_state *net_make_ipc_connection(unsigned flags)
262 {
263         char *server_name = NULL;
264         struct in_addr server_ip;
265         struct cli_state *cli = NULL;
266         NTSTATUS nt_status;
267
268         if (!net_find_server(flags, &server_ip, &server_name)) {
269                 d_printf("\nUnable to find a suitable server\n");
270                 return NULL;
271         }
272
273         if (flags & NET_FLAGS_ANONYMOUS) {
274                 nt_status = connect_to_ipc_anonymous(&cli, &server_ip, server_name);
275         } else {
276                 nt_status = connect_to_ipc(&cli, &server_ip, server_name);
277         }
278         SAFE_FREE(server_name);
279         return cli;
280 }
281
282
283
284 static int net_user(int argc, const char **argv)
285 {
286         if (net_ads_check() == 0)
287                 return net_ads_user(argc, argv);
288
289         /* if server is not specified, default to PDC? */
290         if (net_rpc_check(NET_FLAGS_PDC))
291                 return net_rpc_user(argc, argv);
292
293         return net_rap_user(argc, argv);
294 }
295
296
297 static int net_join(int argc, const char **argv)
298 {
299         if (net_ads_check() == 0) {
300                 if (net_ads_join(argc, argv) == 0)
301                         return 0;
302                 else
303                         d_printf("ADS join did not work, trying RPC...\n");
304         }
305         return net_rpc_join(argc, argv);
306 }
307
308 /* main function table */
309 static struct functable net_func[] = {
310         {"RPC", net_rpc},
311         {"RAP", net_rap},
312         {"ADS", net_ads},
313
314         /* eventually these should auto-choose the transport ... */
315         {"FILE", net_rap_file},
316         {"SHARE", net_rap_share},
317         {"SESSION", net_rap_session},
318         {"SERVER", net_rap_server},
319         {"DOMAIN", net_rap_domain},
320         {"PRINTQ", net_rap_printq},
321         {"USER", net_user},
322         {"GROUP", net_rap_group},
323         {"VALIDATE", net_rap_validate},
324         {"GROUPMEMBER", net_rap_groupmember},
325         {"ADMIN", net_rap_admin},
326         {"SERVICE", net_rap_service},   
327         {"PASSWORD", net_rap_password},
328         {"TIME", net_time},
329         {"LOOKUP", net_lookup},
330         {"JOIN", net_join},
331
332         {"HELP", net_help},
333         {NULL, NULL}
334 };
335
336
337 /****************************************************************************
338   main program
339 ****************************************************************************/
340  int main(int argc, const char **argv)
341 {
342         int opt,i;
343         char *p;
344         int rc = 0;
345         int argc_new = 0;
346         const char ** argv_new;
347         poptContext pc;
348         static char *servicesf = dyn_CONFIGFILE;
349         static int debuglevel = 0;
350
351         struct poptOption long_options[] = {
352                 {"help",        'h', POPT_ARG_NONE,   0, 'h'},
353                 {"workgroup",   'w', POPT_ARG_STRING, &opt_target_workgroup},
354                 {"myworkgroup", 'W', POPT_ARG_STRING, &opt_workgroup},
355                 {"user",        'U', POPT_ARG_STRING, &opt_user_name, 'U'},
356                 {"ipaddress",   'I', POPT_ARG_STRING, 0,'I'},
357                 {"port",        'p', POPT_ARG_INT,    &opt_port},
358                 {"myname",      'n', POPT_ARG_STRING, &opt_requester_name},
359                 {"conf",        's', POPT_ARG_STRING, &servicesf},
360                 {"debug",       'd', POPT_ARG_INT,    &debuglevel},
361                 {"debuglevel",  'd', POPT_ARG_INT,    &debuglevel},
362                 {"server",      'S', POPT_ARG_STRING, &opt_host},
363                 {"comment",     'C', POPT_ARG_STRING, &opt_comment},
364                 {"maxusers",    'M', POPT_ARG_INT,    &opt_maxusers},
365                 {"flags",       'F', POPT_ARG_INT,    &opt_flags},
366                 {"jobid",       'j', POPT_ARG_INT,    &opt_jobid},
367                 {"long",        'l', POPT_ARG_NONE,   &opt_long_list_entries},
368                 {"reboot",      'r', POPT_ARG_NONE,   &opt_reboot},
369                 {"force",       'f', POPT_ARG_NONE,   &opt_force},
370                 {"timeout",     't', POPT_ARG_INT,    &opt_timeout},
371                 { 0, 0, 0, 0}
372         };
373
374         zero_ip(&opt_dest_ip);
375
376         dbf = x_stderr;
377         
378         pc = poptGetContext(NULL, argc, (const char **) argv, long_options, 
379                             POPT_CONTEXT_KEEP_FIRST);
380         
381         while((opt = poptGetNextOpt(pc)) != -1) {
382                 switch (opt) {
383                 case 'h':
384                         net_help(argc, argv);
385                         exit(0);
386                         break;
387                 case 'I':
388                         opt_dest_ip = *interpret_addr2(poptGetOptArg(pc));
389                         if (is_zero_ip(opt_dest_ip))
390                                 d_printf("\nInvalid ip address specified\n");
391                         else
392                                 opt_have_ip = True;
393                         break;
394                 case 'U':
395                         opt_user_specified = True;
396                         opt_user_name = strdup(opt_user_name);
397                         p = strchr(opt_user_name,'%');
398                         if (p) {
399                                 *p = 0;
400                                 opt_password = p+1;
401                         }
402                         break;
403                 default:
404                         d_printf("\nInvalid option %c (%d)\n", (char)opt, opt);
405                         net_help(argc, argv);
406                 }
407         }
408
409         lp_load(servicesf,True,False,False);       
410
411         DEBUGLEVEL = debuglevel;
412
413         argv_new = (const char **)poptGetArgs(pc);
414
415         argc_new = argc;
416         for (i=0; i<argc; i++) {
417                 if (argv_new[i] == NULL) {
418                         argc_new = i;
419                         break;
420                 }
421         }
422          
423         if (!opt_requester_name) {
424                 static fstring myname;
425                 get_myname(myname);
426                 opt_requester_name = myname;
427         }
428
429         if (!opt_user_name && getenv("LOGNAME")) {
430                 opt_user_name = getenv("LOGNAME");
431         }
432
433         if (!opt_workgroup) {
434                 opt_workgroup = lp_workgroup();
435         }
436         
437         if (!opt_target_workgroup) {
438                 opt_target_workgroup = lp_workgroup();
439         }
440         
441         if (!*global_myname) {
442                 char *p2;
443
444                 fstrcpy(global_myname, myhostname());
445                 p2 = strchr_m(global_myname, '.');
446                 if (p2) 
447                         *p2 = 0;
448         }
449         
450         strupper(global_myname);
451
452         load_interfaces();
453
454         rc = net_run_function(argc_new-1, argv_new+1, net_func, net_help);
455         
456         DEBUG(2,("return code = %d\n", rc));
457         return rc;
458 }