int main(void)
{
FILE *f;
char *pwd = strdup("password");
char *p;
p = getpassarg(pwd);
assert(p);
assert(strcmp(pwd, "********") == 0);
assert(strcmp(p, "password") == 0);
free(p);
free(pwd);
f = fopen("passarg.in", "w");
assert(f);
fputs("line1pwd\nline2pwd\n", f);
fclose(f);
f = freopen("passarg.in", "r", stdin);
assert(f);
p = getpassarg("-");
assert(p);
assert(strcmp(p, "line1pwd") == 0);
free(p);
unlink("passarg.in");
return 0;
}
LOGINREC *
get_login(int argc, char *argv[], OPTIONS *options)
{
LOGINREC *login;
int ch;
char *username = NULL, *password = NULL;
extern char *optarg;
assert(options && argv);
options->appname = tds_basename(argv[0]);
options->colsep = default_colsep; /* may be overridden by -t */
login = dblogin();
if (!login) {
fprintf(stderr, "%s: unable to allocate login structure\n", options->appname);
exit(1);
}
DBSETLAPP(login, options->appname);
options->servername = getenv("DSQUERY");
while ((ch = getopt(argc, argv, "U:P:R:S:dD:i:o:e:t:H:hqv")) != -1) {
switch (ch) {
case 'U':
username = strdup(optarg);
break;
case 'R':
parse_pivot_description(options, optarg);
break;
case 'P':
password = getpassarg(optarg);
break;
case 'S':
options->servername = strdup(optarg);
break;
case 'd':
case 'D':
options->database = strdup(optarg);
break;
case 'i':
options->input_filename = strdup(optarg);
break;
case 'o':
options->output_filename = strdup(optarg);
break;
case 'e':
options->error_filename = strdup(optarg);
break;
case 't':
unescape(optarg);
options->colsep = strdup(optarg);
break;
case 'h':
options->headers = stdout;
break;
case 'H':
strcpy(options->hostname, optarg);
break;
case 'q':
options->fquiet = 1;
break;
case 'v':
options->fverbose = 1;
break;
case '?':
default:
usage(options->appname);
exit(1);
}
}
if (username)
DBSETLUSER(login, username);
if( !options->hostname[0] ) {
if (-1 == gethostname(options->hostname, sizeof(options->hostname))) {
perror("unable to get hostname");
}
}
if( options->hostname[0] ) {
DBSETLHOST(login, options->hostname);
}
/* Look for a password if a username was provided, else assume domain login */
if (password) {
DBSETLPWD(login, password);
memset(password, 0, strlen(password));
} else if (username) {
char password[128];
readpassphrase("Password: ", password, sizeof(password), RPP_ECHO_OFF);
DBSETLPWD(login, password);
}
if (!options->servername) {
usage(options->appname);
exit(1);
}
return login;
}
