mirror of
https://github.com/samba-team/samba.git
synced 2025-12-09 00:23:50 +03:00
Add -U, -N, -i, -A, -W to popt.
This commit is contained in:
47
source/include/popt_common.h
Normal file
47
source/include/popt_common.h
Normal file
@@ -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 */
|
||||
@@ -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);
|
||||
|
||||
@@ -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
|
||||
@@ -23,17 +23,20 @@
|
||||
#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 }
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user