146 lines
2.6 KiB
C
146 lines
2.6 KiB
C
#include "vnc.h"
|
|
#include <libsec.h>
|
|
#include <auth.h>
|
|
|
|
char *serveraddr;
|
|
|
|
enum
|
|
{
|
|
VerLen = 12
|
|
};
|
|
|
|
static char version[VerLen+1] = "RFB 003.003\n";
|
|
|
|
int
|
|
vncsrvhandshake(Vnc *v)
|
|
{
|
|
char msg[VerLen+1];
|
|
|
|
strecpy(msg, msg+sizeof msg, version);
|
|
if(verbose)
|
|
fprint(2, "server version: %s\n", msg);
|
|
vncwrbytes(v, msg, VerLen);
|
|
vncflush(v);
|
|
|
|
vncrdbytes(v, msg, VerLen);
|
|
if(verbose)
|
|
fprint(2, "client version: %s\n", msg);
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
vnchandshake(Vnc *v)
|
|
{
|
|
char msg[VerLen+1];
|
|
|
|
msg[VerLen] = 0;
|
|
vncrdbytes(v, msg, VerLen);
|
|
if(strncmp(msg, "RFB ", 4) != 0){
|
|
werrstr("bad rfb version \"%s\"", msg);
|
|
return -1;
|
|
}
|
|
if(verbose)
|
|
fprint(2, "server version: %s\n", msg);
|
|
strcpy(msg, version);
|
|
vncwrbytes(v, msg, VerLen);
|
|
vncflush(v);
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
vncauth(Vnc *v, char *keypattern)
|
|
{
|
|
char *reason;
|
|
uchar chal[VncChalLen];
|
|
ulong auth;
|
|
char *p, *server;
|
|
|
|
if(keypattern == nil)
|
|
keypattern = "";
|
|
auth = vncrdlong(v);
|
|
switch(auth){
|
|
default:
|
|
werrstr("unknown auth type 0x%lux", auth);
|
|
if(verbose)
|
|
fprint(2, "unknown auth type 0x%lux\n", auth);
|
|
return -1;
|
|
|
|
case AFailed:
|
|
reason = vncrdstring(v);
|
|
werrstr("%s", reason);
|
|
if(verbose)
|
|
fprint(2, "auth failed: %s\n", reason);
|
|
return -1;
|
|
|
|
case ANoAuth:
|
|
if(verbose)
|
|
fprint(2, "no auth needed\n");
|
|
break;
|
|
|
|
case AVncAuth:
|
|
vncrdbytes(v, chal, VncChalLen);
|
|
server = strdup(serveraddr);
|
|
p = strrchr(server, ':');
|
|
if(p)
|
|
*p = 0;
|
|
if(auth_respond(chal, VncChalLen, nil, 0, chal, VncChalLen, auth_getkey,
|
|
"proto=vnc role=client server=%s %s", server, keypattern) != VncChalLen){
|
|
free(server);
|
|
return -1;
|
|
}
|
|
free(server);
|
|
vncwrbytes(v, chal, VncChalLen);
|
|
vncflush(v);
|
|
|
|
auth = vncrdlong(v);
|
|
switch(auth){
|
|
default:
|
|
werrstr("unknown server response 0x%lux", auth);
|
|
return -1;
|
|
case VncAuthFailed:
|
|
werrstr("server says authentication failed");
|
|
return -1;
|
|
case VncAuthTooMany:
|
|
werrstr("server says too many tries");
|
|
return -1;
|
|
case VncAuthOK:
|
|
break;
|
|
}
|
|
break;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
vncsrvauth(Vnc *v)
|
|
{
|
|
Chalstate *c;
|
|
AuthInfo *ai;
|
|
|
|
if((c = auth_challenge("proto=vnc role=server user=%q", getuser()))==nil)
|
|
sysfatal("vncchal: %r");
|
|
if(c->nchal != VncChalLen)
|
|
sysfatal("vncchal got %d bytes wanted %d", c->nchal, VncChalLen);
|
|
vncwrlong(v, AVncAuth);
|
|
vncwrbytes(v, c->chal, VncChalLen);
|
|
vncflush(v);
|
|
|
|
vncrdbytes(v, c->chal, VncChalLen);
|
|
c->resp = c->chal;
|
|
c->nresp = VncChalLen;
|
|
ai = auth_response(c);
|
|
auth_freechal(c);
|
|
if(ai == nil){
|
|
fprint(2, "vnc auth failed: server factotum: %r\n");
|
|
vncwrlong(v, VncAuthFailed);
|
|
vncflush(v);
|
|
return -1;
|
|
}
|
|
auth_freeAI(ai);
|
|
vncwrlong(v, VncAuthOK);
|
|
vncflush(v);
|
|
|
|
return 0;
|
|
}
|
|
|