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,2005
   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 3 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, see <http://www.gnu.org/licenses/>.
  20 */
  21
  22 #include "includes.h"
  23 #include "version.h"
  24 #include "lib/cmdline/popt_common.h"
  25 #include "param/param.h"
  26 #include "dynconfig/dynconfig.h"
  27
  28 /* Handle command line options:
  29  *              -d,--debuglevel
  30  *              -s,--configfile
  31  *              -O,--socket-options
  32  *              -V,--version
  33  *              -l,--log-base
  34  *              -n,--netbios-name
  35  *              -W,--workgroup
  36  *              --realm
  37  *              -i,--scope
  38  */
  39
  40 enum {OPT_OPTION=1,OPT_LEAK_REPORT,OPT_LEAK_REPORT_FULL,OPT_DEBUG_STDERR};
  41
  42 struct cli_credentials *cmdline_credentials = NULL;
  43 struct loadparm_context *cmdline_lp_ctx = NULL;
  44
  45 static void popt_version_callback(poptContext con,
  46                            enum poptCallbackReason reason,
  47                            const struct poptOption *opt,
  48                            const char *arg, const void *data)
  49 {
  50         switch(opt->val) {
  51         case 'V':
  52                 printf("Version %s\n", SAMBA_VERSION_STRING );
  53                 exit(0);
  54         }
  55 }
  56
  57 static void popt_samba_callback(poptContext con,
  58                            enum poptCallbackReason reason,
  59                            const struct poptOption *opt,
  60                            const char *arg, const void *data)
  61 {
  62         const char *pname;
  63
  64         if (reason == POPT_CALLBACK_REASON_POST) {
  65                 if (lp_configfile(cmdline_lp_ctx) == NULL) {
  66                         if (getenv("SMB_CONF_PATH"))
  67                                 lp_load(cmdline_lp_ctx, getenv("SMB_CONF_PATH"));
  68                         else
  69                                 lp_load(cmdline_lp_ctx, dyn_CONFIGFILE);
  70                 }
  71                 /* Hook any 'every Samba program must do this, after
  72                  * the smb.conf is setup' functions here */
  73                 return;
  74         }
  75
  76         /* Find out basename of current program */
  77         pname = strrchr_m(poptGetInvocationName(con),'/');
  78
  79         if (!pname)
  80                 pname = poptGetInvocationName(con);
  81         else
  82                 pname++;
  83
  84         if (reason == POPT_CALLBACK_REASON_PRE) {
  85                 if (global_loadparm != NULL) {
  86                         cmdline_lp_ctx = global_loadparm;
  87                 } else {
  88                         cmdline_lp_ctx = global_loadparm = loadparm_init(talloc_autofree_context());
  89                 }
  90
  91                 /* Hook for 'almost the first thing to do in a samba program' here */
  92                 /* setup for panics */
  93                 fault_setup(poptGetInvocationName(con));
  94
  95                 /* and logging */
  96                 setup_logging(pname, DEBUG_STDOUT);
  97
  98                 return;
  99         }
 100
 101         switch(opt->val) {
 102
 103         case OPT_LEAK_REPORT:
 104                 talloc_enable_leak_report();
 105                 break;
 106
 107         case OPT_LEAK_REPORT_FULL:
 108                 talloc_enable_leak_report_full();
 109                 break;
 110
 111         case OPT_OPTION:
 112                 if (!lp_set_option(cmdline_lp_ctx, arg)) {
 113                         fprintf(stderr, "Error setting option '%s'\n", arg);
 114                         exit(1);
 115                 }
 116                 break;
 117
 118         case 'd':
 119                 lp_set_cmdline(cmdline_lp_ctx, "log level", arg);
 120                 break;
 121
 122         case OPT_DEBUG_STDERR:
 123                 setup_logging(pname, DEBUG_STDERR);
 124                 break;
 125
 126         case 's':
 127                 if (arg) {
 128                         lp_load(cmdline_lp_ctx, arg);
 129                 }
 130                 break;
 131
 132         case 'l':
 133                 if (arg) {
 134                         char *new_logfile = talloc_asprintf(NULL, "%s/log.%s", arg, pname);
 135                         lp_set_cmdline(cmdline_lp_ctx, "log file", new_logfile);
 136                         talloc_free(new_logfile);
 137                 }
 138                 break;
 139
 140
 141         }
 142
 143 }
 144
 145
 146 static void popt_common_callback(poptContext con,
 147                            enum poptCallbackReason reason,
 148                            const struct poptOption *opt,
 149                            const char *arg, const void *data)
 150 {
 151         struct loadparm_context *lp_ctx = cmdline_lp_ctx;
 152
 153         switch(opt->val) {
 154         case 'O':
 155                 if (arg) {
 156                         lp_set_cmdline(lp_ctx, "socket options", arg);
 157                 }
 158                 break;
 159
 160         case 'W':
 161                 lp_set_cmdline(lp_ctx, "workgroup", arg);
 162                 break;
 163
 164         case 'r':
 165                 lp_set_cmdline(lp_ctx, "realm", arg);
 166                 break;
 167
 168         case 'n':
 169                 lp_set_cmdline(lp_ctx, "netbios name", arg);
 170                 break;
 171
 172         case 'i':
 173                 lp_set_cmdline(lp_ctx, "netbios scope", arg);
 174                 break;
 175
 176         case 'm':
 177                 lp_set_cmdline(lp_ctx, "client max protocol", arg);
 178                 break;
 179
 180         case 'R':
 181                 lp_set_cmdline(lp_ctx, "name resolve order", arg);
 182                 break;
 183
 184         case 'S':
 185                 lp_set_cmdline(lp_ctx, "client signing", arg);
 186                 break;
 187
 188         }
 189 }
 190
 191 struct poptOption popt_common_connection[] = {
 192         { NULL, 0, POPT_ARG_CALLBACK, (void *)popt_common_callback },
 193         { "name-resolve", 'R', POPT_ARG_STRING, NULL, 'R', "Use these name resolution services only", "NAME-RESOLVE-ORDER" },
 194         { "socket-options", 'O', POPT_ARG_STRING, NULL, 'O', "socket options to use", "SOCKETOPTIONS" },
 195         { "netbiosname", 'n', POPT_ARG_STRING, NULL, 'n', "Primary netbios name", "NETBIOSNAME" },
 196         { "signing", 'S', POPT_ARG_STRING, NULL, 'S', "Set the client signing state", "on|off|required" },
 197         { "workgroup", 'W', POPT_ARG_STRING, NULL, 'W', "Set the workgroup name", "WORKGROUP" },
 198         { "realm", 0, POPT_ARG_STRING, NULL, 'r', "Set the realm name", "REALM" },
 199         { "scope", 'i', POPT_ARG_STRING, NULL, 'i', "Use this Netbios scope", "SCOPE" },
 200         { "maxprotocol", 'm', POPT_ARG_STRING, NULL, 'm', "Set max protocol level", "MAXPROTOCOL" },
 201         { NULL }
 202 };
 203
 204 struct poptOption popt_common_samba[] = {
 205         { NULL, 0, POPT_ARG_CALLBACK|POPT_CBFLAG_PRE|POPT_CBFLAG_POST, (void *)popt_samba_callback },
 206         { "debuglevel",   'd', POPT_ARG_STRING, NULL, 'd', "Set debug level", "DEBUGLEVEL" },
 207         { "debug-stderr", 0, POPT_ARG_NONE, NULL, OPT_DEBUG_STDERR, "Send debug output to STDERR", NULL },
 208         { "configfile",   's', POPT_ARG_STRING, NULL, 's', "Use alternative configuration file", "CONFIGFILE" },
 209         { "option",         0, POPT_ARG_STRING, NULL, OPT_OPTION, "Set smb.conf option from command line", "name=value" },
 210         { "log-basename", 'l', POPT_ARG_STRING, NULL, 'l', "Basename for log/debug files", "LOGFILEBASE" },
 211         { "leak-report",     0, POPT_ARG_NONE, NULL, OPT_LEAK_REPORT, "enable talloc leak reporting on exit", NULL },
 212         { "leak-report-full",0, POPT_ARG_NONE, NULL, OPT_LEAK_REPORT_FULL, "enable full talloc leak reporting on exit", NULL },
 213         { NULL }
 214 };
 215
 216 struct poptOption popt_common_version[] = {
 217         { NULL, 0, POPT_ARG_CALLBACK, (void *)popt_version_callback },
 218         { "version", 'V', POPT_ARG_NONE, NULL, 'V', "Print version" },
 219         { NULL }
 220 };
 221