← Back to branch summary

~ubuntu-branches/ubuntu/vivid/freerdp/vivid

« back to all changes in this revision

Viewing changes to libfreerdp-utils/passphrase.c

  • Committer: Package Import Robot
  • Author(s): Iain Lane
  • Date: 2014-11-11 12:20:50 UTC
  • mfrom: (1.2.5)
  • mto: This revision was merged to the branch mainline in revision 24.
  • Revision ID: package-import@ubuntu.com-20141111122050-7z628f4ab38qxad5
Tags: upstream-1.1.0~git20140921.1.440916e+dfsg1
Import upstream version 1.1.0~git20140921.1.440916e+dfsg1

Show diffs side-by-side

added added

removed removed

Lines of Context:
libfreerdp-utils/passphrase.c
1
 
/**
2
 
 * FreeRDP: A Remote Desktop Protocol Client
3
 
 * Passphrase Handling Utils
4
 
 *
5
 
 * Copyright 2011 Shea Levy <shea@shealevy.com>
6
 
 *
7
 
 * Licensed under the Apache License, Version 2.0 (the "License");
8
 
 * you may not use this file except in compliance with the License.
9
 
 * You may obtain a copy of the License at
10
 
 *
11
 
 *     http://www.apache.org/licenses/LICENSE-2.0
12
 
 *
13
 
 * Unless required by applicable law or agreed to in writing, software
14
 
 * distributed under the License is distributed on an "AS IS" BASIS,
15
 
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16
 
 * See the License for the specific language governing permissions and
17
 
 * limitations under the License.
18
 
 */
19
 
 
20
 
#include <errno.h>
21
 
#include <freerdp/utils/passphrase.h>
22
 
#ifdef _WIN32
23
 
char* freerdp_passphrase_read(const char* prompt, char* buf, size_t bufsiz, int from_stdin)
24
 
{
25
 
        errno=ENOSYS;
26
 
        return NULL;
27
 
}
28
 
#else
29
 
#include <fcntl.h>
30
 
#include <stdio.h>
31
 
#include <string.h>
32
 
#include <sys/stat.h>
33
 
#include <termios.h>
34
 
#include <unistd.h>
35
 
#include <freerdp/utils/signal.h>
36
 
 
37
 
char* freerdp_passphrase_read(const char* prompt, char* buf, size_t bufsiz, int from_stdin)
38
 
{
39
 
        char read_char;
40
 
        char* buf_iter;
41
 
        char term_name[L_ctermid];
42
 
        int term_file, write_file;
43
 
        ssize_t nbytes;
44
 
        size_t read_bytes = 0;
45
 
 
46
 
        if (bufsiz == 0)
47
 
        {
48
 
                errno = EINVAL;
49
 
                return NULL;
50
 
        }
51
 
 
52
 
        ctermid(term_name);
53
 
        if(from_stdin || strcmp(term_name, "") == 0
54
 
                || (term_file = open(term_name, O_RDWR)) == -1)
55
 
        {
56
 
                write_file = STDERR_FILENO;
57
 
                terminal_fildes = STDIN_FILENO;
58
 
        }
59
 
        else
60
 
        {
61
 
                write_file = term_file;
62
 
                terminal_fildes = term_file;
63
 
        }
64
 
 
65
 
        if (tcgetattr(terminal_fildes, &orig_flags) != -1)
66
 
        {
67
 
                new_flags = orig_flags;
68
 
                new_flags.c_lflag &= ~ECHO;
69
 
                new_flags.c_lflag |= ECHONL;
70
 
                terminal_needs_reset = 1;
71
 
                if (tcsetattr(terminal_fildes, TCSAFLUSH, &new_flags) == -1)
72
 
                        terminal_needs_reset = 0;
73
 
        }
74
 
 
75
 
        if (write(write_file, prompt, strlen(prompt)) == (ssize_t) -1)
76
 
                goto error;
77
 
 
78
 
        buf_iter = buf;
79
 
        while ((nbytes = read(terminal_fildes, &read_char, sizeof read_char)) == (sizeof read_char))
80
 
        {
81
 
                if (read_char == '\n')
82
 
                        break;
83
 
                if (read_bytes < (bufsiz - (size_t) 1))
84
 
                {
85
 
                        read_bytes++;
86
 
                        *buf_iter = read_char;
87
 
                        buf_iter++;
88
 
                }
89
 
        }
90
 
        *buf_iter = '\0';
91
 
        buf_iter = NULL;
92
 
        read_char = '\0';
93
 
        if (nbytes == (ssize_t) -1)
94
 
                goto error;
95
 
 
96
 
        if (terminal_needs_reset)
97
 
        {
98
 
                if (tcsetattr(terminal_fildes, TCSAFLUSH, &orig_flags) == -1)   
99
 
                        goto error;
100
 
                terminal_needs_reset = 0;
101
 
        }
102
 
 
103
 
        if (terminal_fildes != STDIN_FILENO)
104
 
        {
105
 
                if (close(terminal_fildes) == -1)
106
 
                        goto error;
107
 
        }
108
 
 
109
 
        return buf;
110
 
 
111
 
        error:
112
 
        {
113
 
                int saved_errno = errno;
114
 
                buf_iter = NULL;
115
 
                read_char = '\0';
116
 
                if (terminal_needs_reset)
117
 
                        tcsetattr(terminal_fildes, TCSAFLUSH, &orig_flags);
118
 
                if (terminal_fildes != STDIN_FILENO)
119
 
                        close(terminal_fildes);
120
 
                errno = saved_errno;
121
 
                return NULL;
122
 
        }
123
 
}
124
 
#endif