pticket.c
3.09 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
/* GNU mailutils - a suite of utilities for electronic mail
Copyright (C) 1999, 2000, 2001 Free Software Foundation, Inc.
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Library Public License as published by
the Free Software Foundation; either version 2, 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 Library General Public License for more details.
You should have received a copy of the GNU Library General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */
#ifdef HAVE_CONFIG_H
# include <config.h>
#endif
#include <errno.h>
#include <sys/types.h>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include <termios.h>
#include <mailutils/error.h>
#include <mailutils/refcount.h>
#include <mailutils/sys/ticket.h>
struct _prompt_ticket
{
struct _ticket base;
int ref;
mu_refcount_t refcount;
};
static void
echo_off(struct termios *stored_settings)
{
struct termios new_settings;
tcgetattr (0, stored_settings);
new_settings = *stored_settings;
new_settings.c_lflag &= (~ECHO);
tcsetattr (0, TCSANOW, &new_settings);
}
static void
echo_on(struct termios *stored_settings)
{
tcsetattr (0, TCSANOW, stored_settings);
}
static int
_prompt_ref (ticket_t ticket)
{
struct _prompt_ticket *prompt = (struct _prompt_ticket *)ticket;
return mu_refcount_inc (prompt->refcount);
}
static void
_prompt_destroy (ticket_t *pticket)
{
struct _prompt_ticket *prompt = (struct _prompt_ticket *)*pticket;
if (mu_refcount_dec (prompt->refcount) == 0)
{
mu_refcount_destroy (&prompt->refcount);
free (prompt);
}
}
static int
_prompt_pop (ticket_t ticket, const char *challenge, char **parg)
{
char arg[256];
struct termios stored_settings;
int echo = 1;
(void)ticket;
/* Being smart if we see "Passwd" and turning off echo. */
if (strstr (challenge, "ass") != NULL
|| strstr (challenge, "ASS") != NULL)
echo = 0;
fprintf (stdout, "%s", challenge);
fflush (stdout);
if (!echo)
echo_off (&stored_settings);
fgets (arg, sizeof (arg), stdin);
if (!echo)
{
echo_on (&stored_settings);
fputc ('\n', stdout);
fflush (stdout);
}
arg [strlen (arg) - 1] = '\0'; /* nuke the trailing line. */
*parg = strdup (arg);
return 0;
}
static struct _ticket_vtable _prompt_vtable =
{
_prompt_ref,
_prompt_destroy,
_prompt_pop,
};
int
ticket_prompt_create (ticket_t *pticket)
{
struct _prompt_ticket *prompt;
if (pticket == NULL)
return MU_ERROR_INVALID_PARAMETER;
prompt = calloc (1, sizeof *prompt);
if (prompt == NULL)
return MU_ERROR_NO_MEMORY;
mu_refcount_create (&prompt->refcount);
if (prompt->refcount == NULL)
{
free (prompt);
return MU_ERROR_NO_MEMORY;
}
*pticket = &prompt->base;
prompt->base.vtable = &_prompt_vtable;
return 0;
}