source3/lib/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    Copyright (C) James Peach 2006
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 3 of the License, or
  12    (at your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21 */
  22
  23 #include "includes.h"
  24 #include "popt_common.h"
  25 #include "lib/param/param.h"
  26
  27 /* Handle command line options:
  28  *              -d,--debuglevel
  29  *              -s,--configfile
  30  *              -O,--socket-options
  31  *              -V,--version
  32  *              -l,--log-base
  33  *              -n,--netbios-name
  34  *              -W,--workgroup
  35  *              -i,--scope
  36  */
  37
  38 enum {OPT_OPTION=1};
  39
  40 extern bool override_logfile;
  41
  42 static void set_logfile(poptContext con, const char * arg)
  43 {
  44
  45         char lfile[PATH_MAX];
  46         const char *pname;
  47         int ret;
  48
  49         /* Find out basename of current program */
  50         pname = strrchr_m(poptGetInvocationName(con), '/');
  51         if (pname == NULL) {
  52                 pname = poptGetInvocationName(con);
  53         } else {
  54                 pname++;
  55         }
  56
  57         ret = snprintf(lfile, sizeof(lfile), "%s/log.%s", arg, pname);
  58         if (ret >= sizeof(lfile)) {
  59                 return;
  60         }
  61         lp_set_logfile(lfile);
  62 }
  63
  64 static bool PrintSambaVersionString;
  65
  66 static void popt_s3_talloc_log_fn(const char *message)
  67 {
  68         DEBUG(0,("%s", message));
  69 }
  70
  71 static void popt_common_callback(poptContext con,
  72                            enum poptCallbackReason reason,
  73                            const struct poptOption *opt,
  74                            const char *arg, const void *data)
  75 {
  76         TALLOC_CTX *mem_ctx = talloc_new(NULL);
  77         if (mem_ctx == NULL) {
  78                 exit(1);
  79         }
  80
  81         if (reason == POPT_CALLBACK_REASON_PRE) {
  82                 set_logfile(con, get_dyn_LOGFILEBASE());
  83                 talloc_set_log_fn(popt_s3_talloc_log_fn);
  84                 talloc_set_abort_fn(smb_panic);
  85                 talloc_free(mem_ctx);
  86                 return;
  87         }
  88
  89         if (reason == POPT_CALLBACK_REASON_POST) {
  90
  91                 if (PrintSambaVersionString) {
  92                         printf( "Version %s\n", samba_version_string());
  93                         talloc_free(mem_ctx);
  94                         exit(0);
  95                 }
  96
  97                 if (is_default_dyn_CONFIGFILE()) {
  98                         if (getenv("SMB_CONF_PATH")) {
  99                                 set_dyn_CONFIGFILE(getenv("SMB_CONF_PATH"));
 100                         }
 101                 }
 102
 103                 if (override_logfile) {
 104                         const struct loadparm_substitution *lp_sub =
 105                                 loadparm_s3_global_substitution();
 106                         char *logfile = lp_logfile(mem_ctx, lp_sub);
 107                         if (logfile == NULL) {
 108                                 talloc_free(mem_ctx);
 109                                 exit(1);
 110                         }
 111                         setup_logging(logfile, DEBUG_FILE);
 112                 }
 113
 114                 /* Further 'every Samba program must do this' hooks here. */
 115                 talloc_free(mem_ctx);
 116                 return;
 117         }
 118
 119         switch(opt->val) {
 120         case OPT_OPTION:
 121         {
 122                 struct loadparm_context *lp_ctx;
 123                 bool ok;
 124
 125                 lp_ctx = loadparm_init_s3(mem_ctx, loadparm_s3_helpers());
 126                 if (lp_ctx == NULL) {
 127                         fprintf(stderr, "loadparm_init_s3() failed!\n");
 128                         talloc_free(mem_ctx);
 129                         exit(1);
 130                 }
 131
 132                 ok = lpcfg_set_option(lp_ctx, arg);
 133                 if (!ok) {
 134                         fprintf(stderr, "Error setting option '%s'\n", arg);
 135                         talloc_free(mem_ctx);
 136                         exit(1);
 137                 }
 138                 break;
 139         }
 140         case 'd':
 141                 if (arg) {
 142                         lp_set_cmdline("log level", arg);
 143                 }
 144                 break;
 145
 146         case 'V':
 147                 PrintSambaVersionString = True;
 148                 break;
 149
 150         case 'O':
 151                 if (arg) {
 152                         lp_set_cmdline("socket options", arg);
 153                 }
 154                 break;
 155
 156         case 's':
 157                 if (arg) {
 158                         set_dyn_CONFIGFILE(arg);
 159                 }
 160                 break;
 161
 162         case 'n':
 163                 if (arg) {
 164                         lp_set_cmdline("netbios name", arg);
 165                 }
 166                 break;
 167
 168         case 'l':
 169                 if (arg) {
 170                         set_logfile(con, arg);
 171                         override_logfile = True;
 172                         set_dyn_LOGFILEBASE(arg);
 173                 }
 174                 break;
 175
 176         case 'i':
 177                 if (arg) {
 178                         lp_set_cmdline("netbios scope", arg);
 179                 }
 180                 break;
 181
 182         case 'W':
 183                 if (arg) {
 184                         lp_set_cmdline("workgroup", arg);
 185                 }
 186                 break;
 187         }
 188
 189         talloc_free(mem_ctx);
 190 }
 191
 192 struct poptOption popt_common_connection[] = {
 193         {
 194                 .argInfo    = POPT_ARG_CALLBACK,
 195                 .arg        = (void *)popt_common_callback,
 196         },
 197         {
 198                 .longName   = "socket-options",
 199                 .shortName  = 'O',
 200                 .argInfo    = POPT_ARG_STRING,
 201                 .val        = 'O',
 202                 .descrip    = "socket options to use",
 203                 .argDescrip = "SOCKETOPTIONS",
 204         },
 205         {
 206                 .longName   = "netbiosname",
 207                 .shortName  = 'n',
 208                 .argInfo    = POPT_ARG_STRING,
 209                 .val        = 'n',
 210                 .descrip    = "Primary netbios name",
 211                 .argDescrip = "NETBIOSNAME"
 212         },
 213         {
 214                 .longName   = "workgroup",
 215                 .shortName  = 'W',
 216                 .argInfo    = POPT_ARG_STRING,
 217                 .val        = 'W',
 218                 .descrip    = "Set the workgroup name",
 219                 .argDescrip = "WORKGROUP"
 220         },
 221         {
 222                 .longName   = "scope",
 223                 .shortName  = 'i',
 224                 .argInfo    = POPT_ARG_STRING,
 225                 .val        = 'i',
 226                 .descrip    = "Use this Netbios scope",
 227                 .argDescrip = "SCOPE"
 228         },
 229         POPT_TABLEEND
 230 };
 231
 232 struct poptOption popt_common_samba[] = {
 233         {
 234                 .argInfo    = POPT_ARG_CALLBACK|POPT_CBFLAG_PRE|POPT_CBFLAG_POST,
 235                 .arg        = (void *)popt_common_callback,
 236         },
 237         {
 238                 .longName   = "debuglevel",
 239                 .shortName  = 'd',
 240                 .argInfo    = POPT_ARG_STRING,
 241                 .val        = 'd',
 242                 .descrip    = "Set debug level",
 243                 .argDescrip = "DEBUGLEVEL",
 244         },
 245         {
 246                 .longName   = "configfile",
 247                 .shortName  = 's',
 248                 .argInfo    = POPT_ARG_STRING,
 249                 .val        = 's',
 250                 .descrip    = "Use alternate configuration file",
 251                 .argDescrip = "CONFIGFILE",
 252         },
 253         {
 254                 .longName   = "log-basename",
 255                 .shortName  = 'l',
 256                 .argInfo    = POPT_ARG_STRING,
 257                 .val        = 'l',
 258                 .descrip    = "Base name for log files",
 259                 .argDescrip = "LOGFILEBASE",
 260         },
 261         {
 262                 .longName   = "version",
 263                 .shortName  = 'V',
 264                 .argInfo    = POPT_ARG_NONE,
 265                 .val        = 'V',
 266                 .descrip    = "Print version",
 267         },
 268         {
 269                 .longName   = "option",
 270                 .argInfo    = POPT_ARG_STRING,
 271                 .val        = OPT_OPTION,
 272                 .descrip    = "Set smb.conf option from command line",
 273                 .argDescrip = "name=value",
 274         },
 275         POPT_TABLEEND
 276 };
 277
 278 struct poptOption popt_common_configfile[] = {
 279         {
 280                 .argInfo    = POPT_ARG_CALLBACK|POPT_CBFLAG_PRE|POPT_CBFLAG_POST,
 281                 .arg        = (void *)popt_common_callback,
 282         },
 283         {
 284                 .longName   = "configfile",
 285                 .argInfo    = POPT_ARG_STRING,
 286                 .val        = 's',
 287                 .descrip    = "Use alternate configuration file",
 288                 .argDescrip = "CONFIGFILE",
 289         },
 290         POPT_TABLEEND
 291 };
 292
 293 struct poptOption popt_common_version[] = {
 294         {
 295                 .argInfo    = POPT_ARG_CALLBACK|POPT_CBFLAG_POST,
 296                 .arg        = (void *)popt_common_callback
 297         },
 298         {
 299                 .longName   = "version",
 300                 .shortName  = 'V',
 301                 .argInfo    = POPT_ARG_NONE,
 302                 .val        = 'V',
 303                 .descrip    = "Print version",
 304         },
 305         POPT_TABLEEND
 306 };
 307
 308 struct poptOption popt_common_debuglevel[] = {
 309         {
 310                 .argInfo    = POPT_ARG_CALLBACK,
 311                 .arg        = (void *)popt_common_callback,
 312         },
 313         {
 314                 .longName   = "debuglevel",
 315                 .shortName  = 'd',
 316                 .argInfo    = POPT_ARG_STRING,
 317                 .val        = 'd',
 318                 .descrip    = "Set debug level",
 319                 .argDescrip = "DEBUGLEVEL",
 320         },
 321         POPT_TABLEEND
 322 };
 323
 324 struct poptOption popt_common_option[] = {
 325         {
 326                 .argInfo    = POPT_ARG_CALLBACK|POPT_CBFLAG_POST,
 327                 .arg        = (void *)popt_common_callback,
 328         },
 329         {
 330                 .longName   = "option",
 331                 .argInfo    = POPT_ARG_STRING,
 332                 .val        = OPT_OPTION,
 333                 .descrip    = "Set smb.conf option from command line",
 334                 .argDescrip = "name=value",
 335         },
 336         POPT_TABLEEND
 337 };