Add -U, -N, -i, -A, -W to popt.
authorJelmer Vernooij <jelmer@samba.org>
Sat, 22 Mar 2003 23:48:02 +0000 (23:48 +0000)
committerJelmer Vernooij <jelmer@samba.org>
Sat, 22 Mar 2003 23:48:02 +0000 (23:48 +0000)
(This used to be commit 5a88d78f67fd7853d6f7d5042807afa56091d52c)

source3/include/popt_common.h [new file with mode: 0644]
source3/include/smb.h
source3/lib/popt_common.c

diff --git a/source3/include/popt_common.h b/source3/include/popt_common.h
new file mode 100644 (file)
index 0000000..9354e87
--- /dev/null
@@ -0,0 +1,47 @@
+/* 
+   Unix SMB/CIFS implementation.
+   Common popt arguments
+   Copyright (C) Jelmer Vernooij       2003
+   
+   This program is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 2 of the License, or
+   (at your option) any later version.
+   
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+   
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, write to the Free Software
+   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+*/
+
+#ifndef _POPT_COMMON_H
+#define _POPT_COMMON_H
+
+/* Common popt structures */
+extern struct poptOption popt_common_debug[];
+extern struct poptOption popt_common_configfile[];
+extern struct poptOption popt_common_socket_options[];
+extern struct poptOption popt_common_version[];
+extern struct poptOption popt_common_netbios_name[];
+extern struct poptOption popt_common_log_base[];
+extern struct poptOption popt_common_credentials[];
+extern struct poptOption popt_common_scope[];
+
+#define POPT_COMMON_SAMBA { NULL, 0, POPT_ARG_INCLUDE_TABLE, popt_common_version, 0, "Common samba options:", NULL },
+#define POPT_CREDENTIALS { NULL, 0, POPT_ARG_INCLUDE_TABLE, popt_common_credentials, 0, "Authentication options:", NULL },
+
+struct user_auth_info {
+       pstring username;
+       pstring password;
+       pstring workgroup;
+       BOOL got_pass;
+       BOOL use_kerberos;
+};
+
+extern struct user_auth_info cmdline_auth_info;
+
+#endif /* _POPT_COMMON_H */
index 0506c410f3b143fe86820c98e496e2ae4fb4b67d..4f37c38413c63213d68ae254cd87225784b2a677 100644 (file)
@@ -1711,14 +1711,7 @@ typedef struct {
 
 #define DEFAULT_TRUST_ACCOUNT_PASSWORD_LENGTH 14
 
-/* Common popt structures */
-
-extern struct poptOption popt_common_debug[];
-extern struct poptOption popt_common_configfile[];
-extern struct poptOption popt_common_socket_options[];
-extern struct poptOption popt_common_version[];
-extern struct poptOption popt_common_netbios_name[];
-extern struct poptOption popt_common_log_base[];
+#include "popt_common.h"
 
 /* Module support */
 typedef int (init_module_function) (void);
index 77c44f127a0459785446139c1058ad8d9fa7b4cb..03e39f5953e66b8a916dfab88585f578764669c2 100644 (file)
@@ -3,7 +3,7 @@
    Common popt routines
 
    Copyright (C) Tim Potter 2001,2002
-   Copyright (C) Jelmer Vernooij 2002
+   Copyright (C) Jelmer Vernooij 2002,2003
 
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
 #include "includes.h"
 
 /* Handle command line options:
- *             d,--debuglevel 
- *             s,--configfile 
- *             O,--socket-options 
- *             V,--version
- *             l,--log-base
- *             n,--netbios-name
+ *             -d,--debuglevel 
+ *             -s,--configfile 
+ *             -O,--socket-options 
+ *             -V,--version
+ *             -l,--log-base
+ *             -n,--netbios-name
+ *             -i,--scope
  */
 
 extern pstring user_socket_options;
 extern BOOL AllowDebugChange;
 
+struct user_auth_info cmdline_auth_info;
+
 static void popt_common_callback(poptContext con, 
                           enum poptCallbackReason reason,
                           const struct poptOption *opt,
@@ -93,13 +96,24 @@ static void popt_common_callback(poptContext con,
                        lp_set_logfile(logfile);
                }
                break;
+
+       case 'i':
+               if (arg) {
+                         set_global_scope(arg);
+               }
+               break;
        }
 }
 
 struct poptOption popt_common_debug[] = {
        { NULL, 0, POPT_ARG_CALLBACK, popt_common_callback },
-       { "debuglevel", 'd', POPT_ARG_STRING, NULL, 'd', "Set debug level", 
-         "DEBUGLEVEL" },
+       { "debuglevel", 'd', POPT_ARG_STRING, NULL, 'd', "Set debug level", "DEBUGLEVEL" },
+       { 0 }
+};
+
+struct poptOption popt_common_scope[] = {
+       { NULL, 0, POPT_ARG_CALLBACK, popt_common_callback }, 
+    { "scope", 'i', POPT_ARG_STRING, NULL, 'i', "Use this Netbios scope", "SCOPE" },  
        { 0 }
 };
 
@@ -132,3 +146,211 @@ struct poptOption popt_common_log_base[] = {
        { "log-basename", 'l', POPT_ARG_STRING, NULL, 'l', "Basename for log/debug files"},
        { 0 }
 };
+
+/****************************************************************************
+ * get a password from a a file or file descriptor
+ * exit on failure
+ * ****************************************************************************/
+static void get_password_file(struct user_auth_info *a)
+{
+       int fd = -1;
+       char *p;
+       BOOL close_it = False;
+       pstring spec;
+       char pass[128];
+
+       if ((p = getenv("PASSWD_FD")) != NULL) {
+               pstrcpy(spec, "descriptor ");
+               pstrcat(spec, p);
+               sscanf(p, "%d", &fd);
+               close_it = False;
+       } else if ((p = getenv("PASSWD_FILE")) != NULL) {
+               fd = sys_open(p, O_RDONLY, 0);
+               pstrcpy(spec, p);
+               if (fd < 0) {
+                       fprintf(stderr, "Error opening PASSWD_FILE %s: %s\n",
+                                       spec, strerror(errno));
+                       exit(1);
+               }
+               close_it = True;
+       }
+
+       for(p = pass, *p = '\0'; /* ensure that pass is null-terminated */
+               p && p - pass < sizeof(pass);) {
+               switch (read(fd, p, 1)) {
+               case 1:
+                       if (*p != '\n' && *p != '\0') {
+                               *++p = '\0'; /* advance p, and null-terminate pass */
+                               break;
+                       }
+               case 0:
+                       if (p - pass) {
+                               *p = '\0'; /* null-terminate it, just in case... */
+                               p = NULL; /* then force the loop condition to become false */
+                               break;
+                       } else {
+                               fprintf(stderr, "Error reading password from file %s: %s\n",
+                                               spec, "empty password\n");
+                               exit(1);
+                       }
+
+               default:
+                       fprintf(stderr, "Error reading password from file %s: %s\n",
+                                       spec, strerror(errno));
+                       exit(1);
+               }
+       }
+       pstrcpy(a->password, pass);
+       if (close_it)
+               close(fd);
+}
+
+static void get_credentials_file(const char *file, struct user_auth_info *info) 
+{
+       XFILE *auth;
+       fstring buf;
+       uint16 len = 0;
+       char *ptr, *val, *param;
+
+       if ((auth=x_fopen(file, O_RDONLY, 0)) == NULL)
+       {
+               /* fail if we can't open the credentials file */
+               d_printf("ERROR: Unable to open credentials file!\n");
+               exit(-1);
+       }
+
+       while (!x_feof(auth))
+       {
+               /* get a line from the file */
+               if (!x_fgets(buf, sizeof(buf), auth))
+                       continue;
+               len = strlen(buf);
+
+               if ((len) && (buf[len-1]=='\n'))
+               {
+                       buf[len-1] = '\0';
+                       len--;
+               }
+               if (len == 0)
+                       continue;
+
+               /* break up the line into parameter & value.
+                * will need to eat a little whitespace possibly */
+               param = buf;
+               if (!(ptr = strchr_m (buf, '=')))
+                       continue;
+
+               val = ptr+1;
+               *ptr = '\0';
+
+               /* eat leading white space */
+               while ((*val!='\0') && ((*val==' ') || (*val=='\t')))
+                       val++;
+
+               if (strwicmp("password", param) == 0)
+               {
+                       pstrcpy(info->password, val);
+                       info->got_pass = True;
+               }
+               else if (strwicmp("username", param) == 0)
+                       pstrcpy(info->username, val);
+               else if (strwicmp("domain", param) == 0)
+                       pstrcpy(info->workgroup,val);
+               memset(buf, 0, sizeof(buf));
+       }
+       x_fclose(auth);
+}
+
+/* Handle command line options:
+ *             -U,--user
+ *             -W,--workgroup
+ *             -A,--authentication-file
+ *             -k,--use-kerberos
+ *             -N,--no-pass
+ */
+
+
+static void popt_common_credentials_callback(poptContext con, 
+                                                                                        enum poptCallbackReason reason,
+                                                                                        const struct poptOption *opt,
+                                                                                        const char *arg, const void *data)
+{
+       char *p;
+
+       if (reason == POPT_CALLBACK_REASON_PRE) {
+               cmdline_auth_info.use_kerberos = False;
+               cmdline_auth_info.got_pass = False;
+               pstrcpy(cmdline_auth_info.username, "GUEST");   
+
+               if (getenv("LOGNAME"))pstrcpy(cmdline_auth_info.username,getenv("LOGNAME"));
+
+               if (getenv("USER")) {
+                       pstrcpy(cmdline_auth_info.username,getenv("USER"));
+
+                       if ((p = strchr_m(cmdline_auth_info.username,'%'))) {
+                               *p = 0;
+                               pstrcpy(cmdline_auth_info.password,p+1);
+                               cmdline_auth_info.got_pass = True;
+                               memset(strchr_m(getenv("USER"),'%')+1,'X',strlen(cmdline_auth_info.password));
+                       }
+               }
+
+               if (getenv("PASSWD")) {
+                       pstrcpy(cmdline_auth_info.password,getenv("PASSWD"));
+                       cmdline_auth_info.got_pass = True;
+               }
+
+               if (getenv("PASSWD_FD") || getenv("PASSWD_FILE")) {
+                       get_password_file(&cmdline_auth_info);
+                       cmdline_auth_info.got_pass = True;
+               }
+
+               return;
+       }
+
+       switch(opt->val) {
+       case 'U':
+               {
+                       char *lp;
+
+                       pstrcpy(cmdline_auth_info.username,arg);
+                       if ((lp=strchr_m(cmdline_auth_info.username,'%'))) {
+                               *lp = 0;
+                               pstrcpy(cmdline_auth_info.password,lp+1);
+                               cmdline_auth_info.got_pass = True;
+                               memset(strchr_m(arg,'%')+1,'X',strlen(cmdline_auth_info.password));
+                       }
+               }
+               break;
+
+       case 'A':
+               get_credentials_file(arg, &cmdline_auth_info);
+               break;
+
+               case 'W':
+                       pstrcpy(cmdline_auth_info.workgroup,arg);
+                       break;
+
+               case 'k':
+#ifndef HAVE_KRB5
+                       d_printf("No kerberos support compiled in\n");
+                       exit(1);
+#else
+                       cmdline_auth_info.got_pass = True;
+#endif
+
+                       break;
+               }
+       }
+
+
+
+       struct poptOption popt_common_credentials[] = {
+               { NULL, 0, POPT_ARG_CALLBACK|POPT_CBFLAG_PRE, popt_common_credentials_callback },
+               { "user", 'U', POPT_ARG_STRING, NULL, 'U', "Set the network username", "USERNAME" },
+               { "no-pass", 'N', POPT_ARG_VAL, &cmdline_auth_info.got_pass, TRUE, "Don't ask for a password" },
+               { "kerberos", 'k', POPT_ARG_VAL, &cmdline_auth_info.use_kerberos, TRUE, "Use kerberos (active directory) authentication" },
+               { "authentication-file", 'A', POPT_ARG_STRING, NULL, 'A', "Get the credentials from a file", "FILE" },
+               { "workgroup", 'W', POPT_ARG_STRING, NULL, 'W', "Set the workgroup name", "WORKGROUP" },
+               { 0 }
+       };