r5929: Use cli_credentials for the SMB functions as well.
[samba.git] / source4 / lib / cmdline / popt_common.c
1 /* 
2    Unix SMB/CIFS implementation.
3    Common popt routines
4
5    Copyright (C) Tim Potter 2001,2002
6    Copyright (C) Jelmer Vernooij 2002,2003
7
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.
12    
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.
17    
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.
21 */
22
23 #include "includes.h"
24 #include "version.h"
25 #include "dynconfig.h"
26 #include "system/filesys.h"
27 #include "system/passwd.h"
28 #include "lib/cmdline/popt_common.h"
29
30 /* Handle command line options:
31  *              -d,--debuglevel 
32  *              -s,--configfile 
33  *              -O,--socket-options 
34  *              -V,--version
35  *              -l,--log-base
36  *              -n,--netbios-name
37  *              -W,--workgroup
38  *              -i,--scope
39  */
40
41 enum {OPT_OPTION=1,OPT_LEAK_REPORT,OPT_LEAK_REPORT_FULL};
42
43 struct cli_credentials *cmdline_credentials = NULL;
44
45 static void popt_common_callback(poptContext con, 
46                            enum poptCallbackReason reason,
47                            const struct poptOption *opt,
48                            const char *arg, const void *data)
49 {
50         const char *pname;
51         
52         /* Find out basename of current program */
53         pname = strrchr_m(poptGetInvocationName(con),'/');
54
55         if (!pname)
56                 pname = poptGetInvocationName(con);
57         else 
58                 pname++;
59
60         if (reason == POPT_CALLBACK_REASON_PRE) {
61                 char *logfile = talloc_asprintf(NULL, "%s/log.%s", dyn_LOGFILEBASE, pname);
62                 lp_set_cmdline("log file", logfile);
63                 talloc_free(logfile);
64                 return;
65         }
66
67         switch(opt->val) {
68         case 'd':
69                 lp_set_cmdline("log level", arg);
70                 break;
71
72         case 'V':
73                 printf( "Version %s\n", SAMBA_VERSION_STRING );
74                 exit(0);
75                 break;
76
77         case 'O':
78                 if (arg) {
79                         lp_set_cmdline("socket options", arg);
80                 }
81                 break;
82
83         case 's':
84                 if (arg) {
85                         pstrcpy(dyn_CONFIGFILE, arg);
86                 }
87                 break;
88
89         case 'l':
90                 if (arg) {
91                         char *logfile = talloc_asprintf(NULL, "%s/log.%s", arg, pname);
92                         lp_set_cmdline("log file", logfile);
93                         talloc_free(logfile);
94                 }
95                 break;
96                 
97         case 'W':
98                 lp_set_cmdline("workgroup", arg);
99                 break;
100                 
101         case 'n':
102                 lp_set_cmdline("netbios name", arg);
103                 break;
104                 
105         case 'i':
106                 lp_set_cmdline("netbios scope", arg);
107                 break;
108
109         case 'm':
110                 lp_set_cmdline("max protocol", arg);
111                 break;
112
113         case 'R':
114                 lp_set_cmdline("name resolve order", arg);
115                 break;
116
117         case OPT_OPTION:
118                 if (!lp_set_option(arg)) {
119                         fprintf(stderr, "Error setting option '%s'\n", arg);
120                         exit(1);
121                 }
122                 break;
123
124         case OPT_LEAK_REPORT:
125                 talloc_enable_leak_report();
126                 break;
127
128         case OPT_LEAK_REPORT_FULL:
129                 talloc_enable_leak_report_full();
130                 break;
131         }
132 }
133
134 struct poptOption popt_common_connection[] = {
135         { NULL, 0, POPT_ARG_CALLBACK, popt_common_callback },
136         { "name-resolve", 'R', POPT_ARG_STRING, NULL, 'R', "Use these name resolution services only", "NAME-RESOLVE-ORDER" },
137         { "socket-options", 'O', POPT_ARG_STRING, NULL, 'O', "socket options to use", "SOCKETOPTIONS" },
138         { "netbiosname", 'n', POPT_ARG_STRING, NULL, 'n', "Primary netbios name", "NETBIOSNAME" },
139         { "workgroup", 'W', POPT_ARG_STRING, NULL, 'W', "Set the workgroup name", "WORKGROUP" },
140         { "scope", 'i', POPT_ARG_STRING, NULL, 'i', "Use this Netbios scope", "SCOPE" },
141         { "maxprotocol", 'm', POPT_ARG_STRING, NULL, 'm', "Set max protocol level", "MAXPROTOCOL" },
142         POPT_TABLEEND
143 };
144
145 struct poptOption popt_common_samba[] = {
146         { NULL, 0, POPT_ARG_CALLBACK|POPT_CBFLAG_PRE, popt_common_callback },
147         { "debuglevel",   'd', POPT_ARG_STRING, NULL, 'd', "Set debug level", "DEBUGLEVEL" },
148         { "configfile",   's', POPT_ARG_STRING, NULL, 's', "Use alternative configuration file", "CONFIGFILE" },
149         { "option",         0, POPT_ARG_STRING, NULL, OPT_OPTION, "Set smb.conf option from command line", "name=value" },
150         { "log-basename", 'l', POPT_ARG_STRING, NULL, 'l', "Basename for log/debug files", "LOGFILEBASE" },
151         { "leak-report",     0, POPT_ARG_NONE, NULL, OPT_LEAK_REPORT, "enable talloc leak reporting on exit", NULL },   
152         { "leak-report-full",0, POPT_ARG_NONE, NULL, OPT_LEAK_REPORT_FULL, "enable full talloc leak reporting on exit", NULL },
153         POPT_TABLEEND
154 };
155
156 struct poptOption popt_common_version[] = {
157         { NULL, 0, POPT_ARG_CALLBACK, popt_common_callback },
158         { "version", 'V', POPT_ARG_NONE, NULL, 'V', "Print version" },
159         POPT_TABLEEND
160 };
161
162 /* Handle command line options:
163  *              -U,--user
164  *              -A,--authentication-file
165  *              -k,--use-kerberos
166  *              -N,--no-pass
167  *              -S,--signing
168  *      -P --machine-pass
169  */
170
171
172 static BOOL dont_ask = False;
173
174 static void popt_common_credentials_callback(poptContext con, 
175                                                 enum poptCallbackReason reason,
176                                                 const struct poptOption *opt,
177                                                 const char *arg, const void *data)
178 {
179         if (reason == POPT_CALLBACK_REASON_PRE) {
180                 cmdline_credentials = talloc_zero(talloc_autofree_context(), struct cli_credentials);
181                 cli_credentials_guess(cmdline_credentials);
182
183                 return;
184         }
185         
186         if (reason == POPT_CALLBACK_REASON_POST) {
187                 if (!dont_ask) {
188                         cli_credentials_set_cmdline_callbacks(cmdline_credentials);
189                 }
190                 return;
191         }
192
193         switch(opt->val) {
194         case 'U':
195                 {
196                         char *lp;
197
198                         cli_credentials_parse_string(cmdline_credentials, arg, CRED_SPECIFIED);
199
200                         if ((lp=strchr_m(arg,'%'))) {
201                                 memset(lp,0,strlen(cmdline_credentials->password));
202                         }
203                 }
204                 break;
205
206         case 'A':
207                 cli_credentials_parse_file(cmdline_credentials, arg, CRED_SPECIFIED);
208                 break;
209
210         case 'S':
211                 lp_set_cmdline("client signing", arg);
212                 break;
213
214         case 'P':
215                 {
216                         char *opt_password = NULL;
217                         /* it is very useful to be able to make ads queries as the
218                            machine account for testing purposes and for domain leave */
219                         
220                         if (!secrets_init()) {
221                                 d_printf("ERROR: Unable to open secrets database\n");
222                                 exit(1);
223                         }
224                         
225                         opt_password = secrets_fetch_machine_password(lp_workgroup());
226                         
227                         if (!opt_password) {
228                                 d_printf("ERROR: Unable to fetch machine password\n");
229                                 exit(1);
230                         }
231                         cmdline_credentials->username = talloc_asprintf(cmdline_credentials, "%s$", lp_netbios_name());
232                         cmdline_credentials->username_obtained = CRED_SPECIFIED;
233                         cli_credentials_set_password(cmdline_credentials, opt_password, CRED_SPECIFIED);
234                         free(opt_password);
235
236                         cli_credentials_set_domain(cmdline_credentials, lp_workgroup(), CRED_SPECIFIED);
237                 }
238                 /* machine accounts only work with kerberos */
239
240         case 'k':
241 #ifndef HAVE_KRB5
242                 d_printf("No kerberos support compiled in\n");
243                 exit(1);
244 #else
245                 lp_set_cmdline("gensec:krb5", "True");
246                 lp_set_cmdline("gensec:ms_krb5", "True");
247 #endif
248                 break;
249
250
251         }
252 }
253
254
255
256 struct poptOption popt_common_credentials[] = {
257         { NULL, 0, POPT_ARG_CALLBACK|POPT_CBFLAG_PRE|POPT_CBFLAG_POST, popt_common_credentials_callback },
258         { "user", 'U', POPT_ARG_STRING, NULL, 'U', "Set the network username", "USERNAME" },
259         { "no-pass", 'N', POPT_ARG_NONE, &dont_ask, True, "Don't ask for a password" },
260         { "kerberos", 'k', POPT_ARG_NONE, NULL, 'k', "Use kerberos (active directory) authentication" },
261         { "authentication-file", 'A', POPT_ARG_STRING, NULL, 'A', "Get the credentials from a file", "FILE" },
262         { "signing", 'S', POPT_ARG_STRING, NULL, 'S', "Set the client signing state", "on|off|required" },
263         { "machine-pass", 'P', POPT_ARG_NONE, NULL, 'P', "Use stored machine account password (implies -k)" },
264         POPT_TABLEEND
265 };