12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788 |
- #include <u.h>
- #include <libc.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*
- getpass(char *prompt)
- {
- char *p, line[4096];
- int n, nr;
- static int cons, consctl; // closing and reopening fails in ssh environment
- if(cons == 0){ // first time
- cons = open("/dev/cons", ORDWR);
- if(cons < 0){
- fprint(2, "couldn't open cons\n");
- exits("no cons");
- }
- consctl = open("/dev/consctl", OWRITE);
- if(consctl < 0){
- fprint(2, "couldn't set raw mode\n");
- exits("no 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 == 21){ /* 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;
- }
- }
- return nil; // NOT REACHED
- }
|