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
|
#include <u.h>
#include <libc.h>
#include <mp.h>
#include <libsec.h>
#include "SConn.h"
#include "secstore.h"
void *
emalloc(ulong n)
{
void *p = malloc(n);
if(p == nil)
sysfatal("emalloc");
memset(p, 0, n);
return p;
}
void *
erealloc(void *p, ulong n)
{
if ((p = realloc(p, n)) == nil)
sysfatal("erealloc");
return p;
}
char *
estrdup(char *s)
{
if ((s = strdup(s)) == nil)
sysfatal("estrdup");
return s;
}
char*
getpassm(char *prompt)
{
char *p, line[4096];
int n, nr;
static int cons, consctl; /* closing & reopening fails in ssh environment */
if(cons == 0){ /* first time? */
cons = open("/dev/cons", ORDWR);
if(cons < 0)
sysfatal("couldn't open cons");
consctl = open("/dev/consctl", OWRITE);
if(consctl < 0)
sysfatal("couldn't set raw mode via consctl");
}
fprint(consctl, "rawon");
fprint(cons, "%s", prompt);
nr = 0;
p = line;
for(;;){
n = read(cons, p, 1);
if(n < 0){
fprint(consctl, "rawoff");
fprint(cons, "\n");
return nil;
}
if(n == 0 || *p == '\n' || *p == '\r' || *p == 0x7f){
*p = '\0';
fprint(consctl, "rawoff");
fprint(cons, "\n");
p = strdup(line);
memset(line, 0, nr);
return p;
}
if(*p == '\b'){
if(nr > 0){
nr--;
p--;
}
}else if(*p == ('u' & 037)){ /* cntrl-u */
fprint(cons, "\n%s", prompt);
nr = 0;
p = line;
}else{
nr++;
p++;
}
if(nr+1 == sizeof line){
fprint(cons, "line too long; try again\n%s", prompt);
nr = 0;
p = line;
}
}
}
static char *
illegal(char *f)
{
syslog(0, LOG, "illegal name: %s", f);
return nil;
}
char *
validatefile(char *f)
{
char *p;
if(f == nil || *f == '\0')
return nil;
if(strcmp(f, "..") == 0 || strlen(f) >= 250)
return illegal(f);
for(p = f; *p; p++)
if(*p < 040 || *p == '/')
return illegal(f);
return f;
}
|
