5
0
mirror of git://git.proxmox.com/git/vncterm.git synced 2024-12-22 21:33:49 +03:00
vncterm/vncterm.c
Dietmar Maurer c9e46b9a7d do not create a process group with setsid
So that we can kill the whole task when started with fork_worker().
2011-09-13 10:55:40 +02:00

2453 lines
54 KiB
C

/*
Copyright (C) 2007-2011 Proxmox Server Solutions GmbH
Copyright: vzdump is under GNU GPL, the GNU General Public License.
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; version 2 dated June, 1991.
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 General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
02111-1307, USA.
Author: Dietmar Maurer <dietmar@proxmox.com>
*/
#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <rfb/rfb.h>
#include <rfb/keysym.h>
#include <pty.h> /* for openpty and forkpty */
#include <string.h>
#include <errno.h>
#include <sys/ioctl.h>
#include <sys/wait.h>
#include <signal.h>
#include <locale.h>
#include "vncterm.h"
#include "glyphs.h"
#include <gnutls/gnutls.h>
#include <gnutls/x509.h>
/* define this for debugging */
//#define DEBUG
char *auth_path = "/";
char *auth_perm = "Sys.Console";
int use_x509 = 1;
static char *
urlencode(char *buf, const char *value)
{
static const char *hexchar = "0123456789abcdef";
char *p = buf;
int i;
int l = strlen(value);
for (i = 0; i < l; i++) {
char c = value[i];
if (('a' <= c && c <= 'z') ||
('A' <= c && c <= 'Z') ||
('0' <= c && c <= '9')) {
*p++ = c;
} else if (c == 32) {
*p++ = '+';
} else {
*p++ = '%';
*p++ = hexchar[c >> 4];
*p++ = hexchar[c & 15];
}
}
*p = 0;
return p;
}
static int
pve_auth_verify(const char *clientip, const char *username, const char *passwd)
{
struct sockaddr_in server;
int sfd = socket(AF_INET, SOCK_STREAM, 0);
if (sfd == -1) {
perror("pve_auth_verify: socket failed");
return -1;
}
struct hostent *he;
if ((he = gethostbyname("localhost")) == NULL) {
fprintf(stderr, "pve_auth_verify: error resolving hostname\n");
goto err;
}
memcpy(&server.sin_addr, he->h_addr_list[0], he->h_length);
server.sin_family = AF_INET;
server.sin_port = htons(85);
if (connect(sfd, (struct sockaddr *)&server, sizeof(server))) {
perror("pve_auth_verify: error connecting to server");
goto err;
}
char buf[8192];
char form[8192];
char *p = form;
p = urlencode(p, "username");
*p++ = '=';
p = urlencode(p, username);
*p++ = '&';
p = urlencode(p, "password");
*p++ = '=';
p = urlencode(p, passwd);
*p++ = '&';
p = urlencode(p, "path");
*p++ = '=';
p = urlencode(p, auth_path);
*p++ = '&';
p = urlencode(p, "privs");
*p++ = '=';
p = urlencode(p, auth_perm);
sprintf(buf, "POST /api2/json/access/ticket HTTP/1.1\n"
"Host: localhost:85\n"
"Connection: close\n"
"PVEClientIP: %s\n"
"Content-Type: application/x-www-form-urlencoded\n"
"Content-Length: %zd\n\n%s\n", clientip, strlen(form), form);
ssize_t len = strlen(buf);
ssize_t sb = send(sfd, buf, len, 0);
if (sb < 0) {
perror("pve_auth_verify: send failed");
goto err;
}
if (sb != len) {
fprintf(stderr, "pve_auth_verify: partial send error\n");
goto err;
}
len = recv(sfd, buf, sizeof(buf) - 1, 0);
if (len < 0) {
perror("pve_auth_verify: recv failed");
goto err;
}
buf[len] = 0;
//printf("DATA:%s\n", buf);
shutdown(sfd, SHUT_RDWR);
return strncmp(buf, "HTTP/1.1 200 OK", 15);
err:
shutdown(sfd, SHUT_RDWR);
return -1;
}
#ifdef DEBUG
static void vnc_debug_gnutls_log(int level, const char* str) {
fprintf(stderr, "%d %s", level, str);
}
#endif
#define DH_BITS 1024
static gnutls_dh_params_t dh_params;
typedef struct {
gnutls_session_t session;
} tls_client_t;
static ssize_t
vnc_tls_push(
gnutls_transport_ptr_t transport,
const void *data,
size_t len)
{
rfbClientPtr cl = (rfbClientPtr)transport;
int n;
retry:
n = send(cl->sock, data, len, 0);
if (n < 0) {
if (errno == EINTR)
goto retry;
return -1;
}
return n;
}
static ssize_t
vnc_tls_pull(
gnutls_transport_ptr_t transport,
void *data,
size_t len)
{
rfbClientPtr cl = (rfbClientPtr)transport;
int n;
retry:
n = recv(cl->sock, data, len, 0);
if (n < 0) {
if (errno == EINTR)
goto retry;
return -1;
}
return n;
}
ssize_t vnc_tls_read(rfbClientPtr cl, void *buf, size_t count)
{
tls_client_t *sd = (tls_client_t *)cl->clientData;
int ret = gnutls_read(sd->session, buf, count);
if (ret < 0) {
if (ret == GNUTLS_E_AGAIN)
errno = EAGAIN;
else
errno = EIO;
ret = -1;
}
return ret;
}
ssize_t vnc_tls_write(rfbClientPtr cl, void *buf, size_t count)
{
tls_client_t *sd = (tls_client_t *)cl->clientData;
int ret = gnutls_write(sd->session, buf, count);
if (ret < 0) {
if (ret == GNUTLS_E_AGAIN)
errno = EAGAIN;
else
errno = EIO;
ret = -1;
}
return ret;
}
static gnutls_anon_server_credentials
tls_initialize_anon_cred(void)
{
gnutls_anon_server_credentials anon_cred;
int ret;
if ((ret = gnutls_anon_allocate_server_credentials(&anon_cred)) < 0) {
rfbLog("can't allocate credentials: %s\n", gnutls_strerror(ret));
return NULL;
}
gnutls_anon_set_server_dh_params(anon_cred, dh_params);
return anon_cred;
}
static gnutls_certificate_credentials_t
tls_initialize_x509_cred(void)
{
gnutls_certificate_credentials_t x509_cred;
int ret;
/* Paths to x509 certs/keys */
char *x509cacert = "/etc/pve/pve-root-ca.pem";
char *x509cert = "/etc/pve/local/pve-ssl.pem";
char *x509key = "/etc/pve/local/pve-ssl.key";
if ((ret = gnutls_certificate_allocate_credentials(&x509_cred)) < 0) {
rfbLog("can't allocate credentials: %s\n", gnutls_strerror(ret));
return NULL;
}
if ((ret = gnutls_certificate_set_x509_trust_file
(x509_cred, x509cacert, GNUTLS_X509_FMT_PEM)) < 0) {
rfbLog("can't load CA certificate: %s\n", gnutls_strerror(ret));
gnutls_certificate_free_credentials(x509_cred);
return NULL;
}
if ((ret = gnutls_certificate_set_x509_key_file
(x509_cred, x509cert, x509key, GNUTLS_X509_FMT_PEM)) < 0) {
rfbLog("can't load certificate & key: %s\n", gnutls_strerror(ret));
gnutls_certificate_free_credentials(x509_cred);
return NULL;
}
gnutls_certificate_set_dh_params (x509_cred, dh_params);
return x509_cred;
}
/* rfb tls security handler */
#define rfbSecTypeVencrypt 19
#define rfbVencryptTlsPlain 259
#define rfbVencryptX509Plain 262
void rfbEncodeU32(char *buf, uint32_t value)
{
buf[0] = (value >> 24) & 0xFF;
buf[1] = (value >> 16) & 0xFF;
buf[2] = (value >> 8) & 0xFF;
buf[3] = value & 0xFF;
}
uint32_t rfbDecodeU32(char *data, size_t offset)
{
return ((data[offset] << 24) | (data[offset + 1] << 16) |
(data[offset + 2] << 8) | data[offset + 3]);
}
static void
vencrypt_subauth_plain(rfbClientPtr cl)
{
const char *err = NULL;
char buf[4096];
int n;
char clientip[INET6_ADDRSTRLEN];
clientip[0] = 0;
struct sockaddr_in client;
socklen_t addrlen = sizeof(client);
if (getpeername(cl->sock, &client, &addrlen) == 0) {
inet_ntop(client.sin_family, &client.sin_addr,
clientip, sizeof(clientip));
}
if ((n = rfbReadExact(cl, buf, 8)) <= 0) {
err = n ? "read failed" : "client gone";
goto err;
}
uint32_t ulen = rfbDecodeU32(buf, 0);
uint32_t pwlen = rfbDecodeU32(buf, 4);
if (!ulen) {
err = "No User name.";
goto err;
}
if (ulen >= 255) {
err = "User name too long.";
goto err;
}
if (!pwlen) {
err = "Password too short";
goto err;
}
if (pwlen >= 511) {
err = "Password too long.";
goto err;
}
if ((n = rfbReadExact(cl, buf, ulen)) <= 0) {
err = n ? "read failed" : "client gone";
goto err;
}
buf[ulen] = 0;
char *username = buf;
char *passwd = buf + ulen + 1;
if ((n = rfbReadExact(cl, passwd, pwlen)) <= 0) {
err = n ? "read failed" : "client gone";
goto err;
}
passwd[pwlen] = 0;
rfbLog("VencryptPlain: username: %s pw: %s\n", username, passwd);
if (pve_auth_verify(clientip, username, passwd) == 0) {
rfbEncodeU32(buf, 0); /* Accept auth completion */
rfbWriteExact(cl, buf, 4);
cl->state = RFB_INITIALISATION;
return;
}
err = "Authentication failed";
err:
rfbLog("VencryptPlain: %s\n", err ? err : "no reason specified");
if (err) {
rfbEncodeU32(buf, 1); /* Reject auth */
rfbWriteExact(cl, buf, 4);
if (cl->protocolMinorVersion >= 8) {
int elen = strlen(err);
rfbEncodeU32(buf, elen);
rfbWriteExact(cl, buf, 4);
rfbWriteExact(cl, err, elen);
}
}
rfbCloseClient(cl);
return;
}
static void
rfbVncAuthVencrypt(rfbClientPtr cl)
{
int ret;
/* Send VeNCrypt version 0.2 */
char buf[256];
buf[0] = 0;
buf[1] = 2;
if (rfbWriteExact(cl, buf, 2) < 0) {
rfbLogPerror("rfbVncAuthVencrypt: write");
rfbCloseClient(cl);
return;
}
int n = rfbReadExact(cl, buf, 2);
if (n <= 0) {
if (n == 0)
rfbLog("rfbVncAuthVencrypt: client gone\n");
else
rfbLogPerror("rfbVncAuthVencrypt: read");
rfbCloseClient(cl);
return;
}
if (buf[0] != 0 || buf[1] != 2) {
rfbLog("Unsupported VeNCrypt protocol %d.%d\n",
(int)buf[0], (int)buf[1]);
buf[0] = 1; /* Reject version */
rfbWriteExact(cl, buf, 1);
rfbCloseClient(cl);
return;
}
/* Sending allowed auth */
int req_auth = use_x509 ? rfbVencryptX509Plain : rfbVencryptTlsPlain;
buf[0] = 0; /* Accept version */
buf[1] = 1; /* number of sub auths */
rfbEncodeU32(buf+2, req_auth);
if (rfbWriteExact(cl, buf, 6) < 0) {
rfbLogPerror("rfbVncAuthVencrypt: write");
rfbCloseClient(cl);
return;
}
n = rfbReadExact(cl, buf, 4);
if (n <= 0) {
if (n == 0)
rfbLog("rfbVncAuthVencrypt: client gone\n");
else
rfbLogPerror("rfbVncAuthVencrypt: read");
rfbCloseClient(cl);
return;
}
int auth = rfbDecodeU32(buf, 0);
if (auth != req_auth) {
buf[0] = 1; /* Reject auth*/
rfbWriteExact(cl, buf, 1);
rfbCloseClient(cl);
return;
}
buf[0] = 1; /* Accept auth */
if (rfbWriteExact(cl, buf, 1) < 0) {
rfbLogPerror("rfbVncAuthVencrypt: write");
rfbCloseClient(cl);
return;
}
tls_client_t *sd = calloc(1, sizeof(tls_client_t));
if (sd->session == NULL) {
if (gnutls_init(&sd->session, GNUTLS_SERVER) < 0) {
rfbLog("gnutls_init failed\n");
rfbCloseClient(cl);
return;
}
if ((ret = gnutls_set_default_priority(sd->session)) < 0) {
rfbLog("gnutls_set_default_priority failed: %s\n", gnutls_strerror(ret));
sd->session = NULL;
rfbCloseClient(cl);
return;
}
/* optimize for speed */
static const int cipher_priority_performance[] = {
GNUTLS_CIPHER_ARCFOUR_128,
GNUTLS_CIPHER_AES_128_CBC,
GNUTLS_CIPHER_3DES_CBC, 0
};
if ((ret = gnutls_cipher_set_priority(sd->session, cipher_priority_performance)) < 0) {
rfbLog("gnutls_cipher_set_priority failed: %s\n", gnutls_strerror(ret));
sd->session = NULL;
rfbCloseClient(cl);
return;
}
static const int kx_anon[] = {GNUTLS_KX_ANON_DH, 0};
static const int kx_x509[] = {GNUTLS_KX_DHE_DSS, GNUTLS_KX_RSA, GNUTLS_KX_DHE_RSA, GNUTLS_KX_SRP, 0};
if ((ret = gnutls_kx_set_priority(sd->session, use_x509 ? kx_x509 : kx_anon)) < 0) {
rfbLog("gnutls_kx_set_priority failed: %s\n", gnutls_strerror(ret));
sd->session = NULL;
rfbCloseClient(cl);
return;
}
static const int cert_type_priority[] = { GNUTLS_CRT_X509, 0 };
if ((ret = gnutls_certificate_type_set_priority(sd->session, cert_type_priority)) < 0) {
rfbLog("gnutls_certificate_type_set_priority failed: %s\n",
gnutls_strerror(ret));
sd->session = NULL;
rfbCloseClient(cl);
return;
}
static const int protocol_priority[]= { GNUTLS_TLS1_1, GNUTLS_TLS1_0, GNUTLS_SSL3, 0 };
if ((ret = gnutls_protocol_set_priority(sd->session, protocol_priority)) < 0) {
rfbLog("gnutls_protocol_set_priority failed: %s\n",
gnutls_strerror(ret));
sd->session = NULL;
rfbCloseClient(cl);
return;
}
if (use_x509) {
gnutls_certificate_server_credentials x509_cred;
if (!(x509_cred = tls_initialize_x509_cred())) {
sd->session = NULL;
rfbCloseClient(cl);
return;
}
if (gnutls_credentials_set(sd->session, GNUTLS_CRD_CERTIFICATE, x509_cred) < 0) {
sd->session = NULL;
gnutls_certificate_free_credentials(x509_cred);
rfbCloseClient(cl);
return;
}
} else {
gnutls_anon_server_credentials anon_cred;
if (!(anon_cred = tls_initialize_anon_cred())) {
sd->session = NULL;
rfbCloseClient(cl);
return;
}
if ((ret = gnutls_credentials_set(sd->session, GNUTLS_CRD_ANON, anon_cred)) < 0) {
rfbLog("gnutls_credentials_set failed: %s\n", gnutls_strerror(ret));
gnutls_anon_free_server_credentials(anon_cred);
sd->session = NULL;
rfbCloseClient(cl);
return;
}
}
gnutls_transport_set_ptr(sd->session, (gnutls_transport_ptr_t)cl);
gnutls_transport_set_push_function(sd->session, vnc_tls_push);
gnutls_transport_set_pull_function(sd->session, vnc_tls_pull);
}
retry:
if ((ret = gnutls_handshake(sd->session)) < 0) {
if (!gnutls_error_is_fatal(ret)) {
usleep(100000);
goto retry;
}
rfbLog("rfbVncAuthVencrypt: handshake failed\n");
rfbCloseClient(cl);
return;
}
/* set up TLS read/write hooks */
cl->clientData = sd;
cl->sock_read_fn = &vnc_tls_read;
cl->sock_write_fn = &vnc_tls_write;
vencrypt_subauth_plain(cl);
}
static rfbSecurityHandler VncSecurityHandlerVencrypt = {
rfbSecTypeVencrypt,
rfbVncAuthVencrypt,
NULL
};
#define TERM "xterm"
#define TERMIDCODE "[?1;2c" // vt100 ID
#define CHECK_ARGC(argc,argv,i) if (i >= argc-1) { \
fprintf (stderr, "ERROR: not enough arguments for: %s\n", argv[i]); \
print_usage (NULL); \
exit(1); \
}
/* these colours are from linux kernel drivers/char/vt.c */
static int idle_timeout = 1;
unsigned char color_table[] = { 0, 4, 2, 6, 1, 5, 3, 7,
8,12,10,14, 9,13,11,15 };
/* the default colour table, for VGA+ colour systems */
int default_red[] = {0x00,0xaa,0x00,0xaa,0x00,0xaa,0x00,0xaa,
0x55,0xff,0x55,0xff,0x55,0xff,0x55,0xff};
int default_grn[] = {0x00,0x00,0xaa,0x55,0x00,0x00,0xaa,0xaa,
0x55,0x55,0xff,0xff,0x55,0x55,0xff,0xff};
int default_blu[] = {0x00,0x00,0x00,0x00,0xaa,0xaa,0xaa,0xaa,
0x55,0x55,0x55,0x55,0xff,0xff,0xff,0xff};
static void
print_usage (const char *msg)
{
if (msg) { fprintf (stderr, "ERROR: %s\n", msg); }
fprintf (stderr, "USAGE: vncterm [vncopts] [-c command [args]]\n");
}
/* Convert UCS2 to UTF8 sequence, trailing zero */
static int
ucs2_to_utf8 (unicode c, char *out)
{
if (c < 0x80) {
out[0] = c; // 0*******
out[1] = 0;
return 1;
} else if (c < 0x800) {
out[0] = 0xc0 | (c >> 6); // 110***** 10******
out[1] = 0x80 | (c & 0x3f);
out[2] = 0;
return 2;
} else {
out[0] = 0xe0 | (c >> 12); // 1110**** 10****** 10******
out[1] = 0x80 | ((c >> 6) & 0x3f);
out[2] = 0x80 | (c & 0x3f);
out[3] = 0;
return 3;
}
return 0;
}
static void
rfb_draw_char (rfbScreenInfoPtr rfbScreen, int x, int y,
unsigned short c, rfbPixel col)
{
if (c > vt_font_size) {
rfbLog ("undefined font glyph %d\n", c);
return;
}
int i,j;
unsigned char *data= vt_font_data + c*16;
unsigned char d=*data;
int rowstride=rfbScreen->paddedWidthInBytes;
char *colour=(char*)&col;
for(j = 0; j < 16; j++) {
for(i = 0; i < 8; i++) {
if ((i&7) == 0) {
d=*data;
data++;
}
if (d&0x80)
*(rfbScreen->frameBuffer+(y+j)*rowstride+(x+i)) = *colour;
d<<=1;
}
}
}
static void
draw_char_at (vncTerm *vt, int x, int y, unicode ch, TextAttributes attrib)
{
if (x < 0 || y < 0 || x >= vt->width || y >= vt->height) { return; }
int rx = x*8;
int ry = y*16;
int rxe = x*8+8;
int rye = y*16+16;
int fg, bg;
if (attrib.invers) {
bg = attrib.fgcol;
fg = attrib.bgcol;
} else {
bg = attrib.bgcol;
fg = attrib.fgcol;
}
int ec = vt_fontmap[ch];
rfbFillRect (vt->screen, rx, ry, rxe, rye, bg);
if (attrib.bold) {
fg += 8;
}
// unsuported attributes = (attrib.blink || attrib.unvisible)
rfb_draw_char (vt->screen, rx, ry, ec, fg);
if (attrib.uline) {
rfbDrawLine (vt->screen, rx, ry + 14, rxe, ry + 14, fg);
}
rfbMarkRectAsModified (vt->screen, rx, ry, rxe, rye);
}
static void
vncterm_update_xy (vncTerm *vt, int x, int y)
{
if (x < 0 || y < 0 || x >= vt->width || y >= vt->height) { return; }
int y1 = (vt->y_base + y) % vt->total_height;
int y2 = y1 - vt->y_displ;
if (y2 < 0) {
y2 += vt->total_height;
}
if (y2 < vt->height) {
TextCell *c = &vt->cells[y1 * vt->width + x];
draw_char_at (vt, x, y2, c->ch, c->attrib);
}
}
static void
vncterm_clear_xy (vncTerm *vt, int x, int y)
{
if (x < 0 || y < 0 || x >= vt->width || y >= vt->height) { return; }
int y1 = (vt->y_base + y) % vt->total_height;
int y2 = y1 - vt->y_displ;
if (y2 < 0) {
y2 += vt->total_height;
}
if (y2 < vt->height) {
TextCell *c = &vt->cells[y1 * vt->width + x];
c->ch = ' ';
c->attrib = vt->default_attrib;
c->attrib.fgcol = vt->cur_attrib.fgcol;
c->attrib.bgcol = vt->cur_attrib.bgcol;
draw_char_at (vt, x, y, c->ch, c->attrib);
}
}
static void
vncterm_show_cursor (vncTerm *vt, int show)
{
int x = vt->cx;
if (x >= vt->width) {
x = vt->width - 1;
}
int y1 = (vt->y_base + vt->cy) % vt->total_height;
int y = y1 - vt->y_displ;
if (y < 0) {
y += vt->total_height;
}
if (y < vt->height) {
TextCell *c = &vt->cells[y1 * vt->width + x];
if (show) {
TextAttributes attrib = vt->default_attrib;
attrib.invers = !(attrib.invers); /* invert fg and bg */
draw_char_at (vt, x, y, c->ch, attrib);
} else {
draw_char_at (vt, x, y, c->ch, c->attrib);
}
}
}
static void
vncterm_refresh (vncTerm *vt)
{
int x, y, y1;
rfbFillRect (vt->screen, 0, 0, vt->maxx, vt->maxy, vt->default_attrib.bgcol);
y1 = vt->y_displ;
for(y = 0; y < vt->height; y++) {
TextCell *c = vt->cells + y1 * vt->width;
for(x = 0; x < vt->width; x++) {
draw_char_at (vt, x, y, c->ch, c->attrib);
c++;
}
if (++y1 == vt->total_height)
y1 = 0;
}
rfbMarkRectAsModified (vt->screen, 0, 0, vt->maxx, vt->maxy);
vncterm_show_cursor (vt, 1);
}
static void
vncterm_scroll_down (vncTerm *vt, int top, int bottom, int lines)
{
if ((top + lines) >= bottom) {
lines = bottom - top -1;
}
if (top < 0 || bottom > vt->height || top >= bottom || lines < 1) {
return;
}
int h = lines * 16;
int y0 = top*16;
int y1 = y0 + h;
int y2 = bottom*16;
int rowstride = vt->screen->paddedWidthInBytes;
int rows = (bottom - top - lines)*16;
char *in = vt->screen->frameBuffer+y0*rowstride;
char *out = vt->screen->frameBuffer+y1*rowstride;
memmove(out,in, rowstride*rows);
memset(vt->screen->frameBuffer+y0*rowstride, 0, h*rowstride);
rfbMarkRectAsModified (vt->screen, 0, y0, vt->screen->width, y2);
int i;
for(i = bottom - top - lines - 1; i >= 0; i--) {
int src = ((vt->y_base + top + i) % vt->total_height)*vt->width;
int dst = ((vt->y_base + top + lines + i) % vt->total_height)*vt->width;
memmove(vt->cells + dst, vt->cells + src, vt->width*sizeof (TextCell));
}
for (i = 0; i < lines; i++) {
int j;
TextCell *c = vt->cells + ((vt->y_base + top + i) % vt->total_height)*vt->width;
for(j = 0; j < vt->width; j++) {
c->attrib = vt->default_attrib;
c->ch = ' ';
c++;
}
}
}
static void
vncterm_scroll_up (vncTerm *vt, int top, int bottom, int lines, int moveattr)
{
if ((top + lines) >= bottom) {
lines = bottom - top - 1;
}
if (top < 0 || bottom > vt->height || top >= bottom || lines < 1) {
return;
}
int h = lines * 16;
int y0 = top*16;
int y1 = (top + lines)*16;
int y2 = bottom*16;
int rowstride = vt->screen->paddedWidthInBytes;
int rows = (bottom - top - lines)*16;
char *in = vt->screen->frameBuffer+y1*rowstride;
char *out = vt->screen->frameBuffer+y0*rowstride;
memmove(out,in, rowstride*rows);
memset(vt->screen->frameBuffer+(y2-h)*rowstride, 0, h*rowstride);
rfbMarkRectAsModified (vt->screen, 0, y0, vt->screen->width, y2);
if (!moveattr) return;
// move attributes
int i;
for(i = 0; i < (bottom - top - lines); i++) {
int dst = ((vt->y_base + top + i) % vt->total_height)*vt->width;
int src = ((vt->y_base + top + lines + i) % vt->total_height)*vt->width;
memmove(vt->cells + dst, vt->cells + src, vt->width*sizeof (TextCell));
}
for (i = 1; i <= lines; i++) {
int j;
TextCell *c = vt->cells + ((vt->y_base + bottom - i) % vt->total_height)*vt->width;
for(j = 0; j < vt->width; j++) {
c->attrib = vt->default_attrib;
c->ch = ' ';
c++;
}
}
}
static void
vncterm_virtual_scroll (vncTerm *vt, int lines)
{
if (vt->altbuf || lines == 0) return;
if (lines < 0) {
lines = -lines;
int i = vt->scroll_height;
if (i > vt->total_height - vt->height)
i = vt->total_height - vt->height;
int y1 = vt->y_base - i;
if (y1 < 0)
y1 += vt->total_height;
for(i = 0; i < lines; i++) {
if (vt->y_displ == y1) break;
if (--vt->y_displ < 0) {
vt->y_displ = vt->total_height - 1;
}
}
} else {
int i;
for(i = 0; i < lines; i++) {
if (vt->y_displ == vt->y_base) break;
if (++vt->y_displ == vt->total_height) {
vt->y_displ = 0;
}
}
}
vncterm_refresh (vt);
}
static void
vncterm_respond_esc (vncTerm *vt, const char *esc)
{
int len = strlen (esc);
int i;
if (vt->ibuf_count < (IBUFSIZE - 1 - len)) {
vt->ibuf[vt->ibuf_count++] = 27;
for (i = 0; i < len; i++) {
vt->ibuf[vt->ibuf_count++] = esc[i];
}
}
}
static void
vncterm_put_lf (vncTerm *vt)
{
if (vt->cy + 1 == vt->region_bottom) {
if (vt->altbuf || vt->region_top != 0 || vt->region_bottom != vt->height) {
vncterm_scroll_up (vt, vt->region_top, vt->region_bottom, 1, 1);
return;
}
if (vt->y_displ == vt->y_base) {
vncterm_scroll_up (vt, vt->region_top, vt->region_bottom, 1, 0);
}
if (vt->y_displ == vt->y_base) {
if (++vt->y_displ == vt->total_height) {
vt->y_displ = 0;
}
}
if (++vt->y_base == vt->total_height) {
vt->y_base = 0;
}
if (vt->scroll_height < vt->total_height) {
vt->scroll_height++;
}
int y1 = (vt->y_base + vt->height - 1) % vt->total_height;
TextCell *c = &vt->cells[y1 * vt->width];
int x;
for (x = 0; x < vt->width; x++) {
c->ch = ' ';
c->attrib = vt->default_attrib;
c++;
}
// fprintf (stderr, "BASE: %d DISPLAY %d\n", vt->y_base, vt->y_displ);
} else if (vt->cy < vt->height - 1) {
vt->cy += 1;
}
}
static void
vncterm_csi_m (vncTerm *vt)
{
int i;
for (i = 0; i < vt->esc_count; i++) {
switch (vt->esc_buf[i]) {
case 0: /* reset all console attributes to default */
vt->cur_attrib = vt->default_attrib;
break;
case 1:
vt->cur_attrib.bold = 1;
break;
case 4:
vt->cur_attrib.uline = 1;
break;
case 5:
vt->cur_attrib.blink = 1;
break;
case 7:
vt->cur_attrib.invers = 1;
break;
case 8:
vt->cur_attrib.unvisible = 1;
break;
case 10:
vt->cur_enc = LAT1_MAP;
// fixme: dispaly controls = 0 ?
// fixme: toggle meta = 0 ?
break;
case 11:
vt->cur_enc = IBMPC_MAP;
// fixme: dispaly controls = 1 ?
// fixme: toggle meta = 0 ?
break;
case 12:
vt->cur_enc = IBMPC_MAP;
// fixme: dispaly controls = 1 ?
// fixme: toggle meta = 1 ?
break;
case 22:
vt->cur_attrib.bold = 0;
break;
case 24:
vt->cur_attrib.uline = 0;
break;
case 25:
vt->cur_attrib.blink = 0;
break;
case 27:
vt->cur_attrib.invers = 0;
break;
case 28:
vt->cur_attrib.unvisible = 0;
break;
case 30:
case 31:
case 32:
case 33:
case 34:
case 35:
case 36:
case 37:
/* set foreground color */
vt->cur_attrib.fgcol = color_table [vt->esc_buf[i] - 30];
break;
case 38:
/* reset color to default, enable underline */
vt->cur_attrib.fgcol = vt->default_attrib.fgcol;
vt->cur_attrib.uline = 1;
break;
case 39:
/* reset color to default, disable underline */
vt->cur_attrib.fgcol = vt->default_attrib.fgcol;
vt->cur_attrib.uline = 0;
break;
case 40:
case 41:
case 42:
case 43:
case 44:
case 45:
case 46:
case 47:
/* set background color */
vt->cur_attrib.bgcol = color_table [vt->esc_buf[i] - 40];
break;
case 49:
/* reset background color */
vt->cur_attrib.bgcol = vt->default_attrib.bgcol;
break;
default:
fprintf (stderr, "unhandled ESC[%d m code\n",vt->esc_buf[i]);
//fixme: implement
}
}
}
static void
vncterm_save_cursor (vncTerm *vt)
{
vt->cx_saved = vt->cx;
vt->cy_saved = vt->cy;
vt->cur_attrib_saved = vt->cur_attrib;
vt->charset_saved = vt->charset;
vt->g0enc_saved = vt->g0enc;
vt->g1enc_saved = vt->g1enc;
vt->cur_enc_saved = vt->cur_enc;
}
static void
vncterm_restore_cursor (vncTerm *vt)
{
vt->cx = vt->cx_saved;
vt->cy = vt->cy_saved;
vt->cur_attrib = vt->cur_attrib_saved;
vt->charset = vt->charset_saved;
vt->g0enc = vt->g0enc_saved;
vt->g1enc = vt->g1enc_saved;
vt->cur_enc = vt->cur_enc_saved;
}
static void
vncterm_set_alternate_buffer (vncTerm *vt, int on_off)
{
int x, y;
vt->y_displ = vt->y_base;
if (on_off) {
if (vt->altbuf) return;
vt->altbuf = 1;
/* alternate buffer & cursor */
vncterm_save_cursor (vt);
/* save screen to altcels */
for (y = 0; y < vt->height; y++) {
int y1 = (vt->y_base + y) % vt->total_height;
for (x = 0; x < vt->width; x++) {
vt->altcells[y*vt->width + x] = vt->cells[y1*vt->width + x];
}
}
/* clear screen */
for (y = 0; y <= vt->height; y++) {
for (x = 0; x < vt->width; x++) {
vncterm_clear_xy (vt, x, y);
}
}
} else {
if (vt->altbuf == 0) return;
vt->altbuf = 0;
/* restore saved data */
for (y = 0; y < vt->height; y++) {
int y1 = (vt->y_base + y) % vt->total_height;
for (x = 0; x < vt->width; x++) {
vt->cells[y1*vt->width + x] = vt->altcells[y*vt->width + x];
}
}
vncterm_restore_cursor (vt);
}
vncterm_refresh (vt);
}
static void
vncterm_set_mode (vncTerm *vt, int on_off)
{
int i;
for (i = 0; i <= vt->esc_count; i++) {
if (vt->esc_ques) { /* DEC private modes set/reset */
switch(vt->esc_buf[i]) {
case 10: /* X11 mouse reporting on/off */
case 1000:
vt->report_mouse = on_off;
break;
case 1049: /* start/end special app mode (smcup/rmcup) */
vncterm_set_alternate_buffer (vt, on_off);
break;
case 25: /* Cursor on/off */
case 9: /* X10 mouse reporting on/off */
case 6: /* Origin relative/absolute */
case 1: /* Cursor keys in appl mode*/
case 5: /* Inverted screen on/off */
case 7: /* Autowrap on/off */
case 8: /* Autorepeat on/off */
break;
}
} else { /* ANSI modes set/reset */
/* fixme: implement me */
}
}
}
static void
vncterm_gotoxy (vncTerm *vt, int x, int y)
{
/* verify all boundaries */
if (x < 0) {
x = 0;
}
if (x >= vt->width) {
x = vt->width - 1;
}
vt->cx = x;
if (y < 0) {
y = 0;
}
if (y >= vt->height) {
y = vt->height - 1;
}
vt->cy = y;
}
enum { ESnormal, ESesc, ESsquare, ESgetpars, ESgotpars, ESfunckey,
EShash, ESsetG0, ESsetG1, ESpercent, ESignore, ESnonstd,
ESpalette, ESidquery, ESosc1, ESosc2};
static void
vncterm_putchar (vncTerm *vt, unicode ch)
{
int x, y, i, c;
#ifdef DEBUG
if (!vt->tty_state)
fprintf (stderr, "CHAR:%2d: %4x '%c' (cur_enc %d) %d %d\n", vt->tty_state, ch, ch, vt->cur_enc, vt->cx, vt->cy);
#endif
switch(vt->tty_state) {
case ESesc:
vt->tty_state = ESnormal;
switch (ch) {
case '[':
vt->tty_state = ESsquare;
break;
case ']':
vt->tty_state = ESnonstd;
break;
case '%':
vt->tty_state = ESpercent;
break;
case '7':
vncterm_save_cursor (vt);
break;
case '8':
vncterm_restore_cursor (vt);
break;
case '(':
vt->tty_state = ESsetG0; // SET G0
break;
case ')':
vt->tty_state = ESsetG1; // SET G1
break;
case 'M':
/* cursor up (ri) */
if (vt->cy == vt->region_top)
vncterm_scroll_down (vt, vt->region_top, vt->region_bottom, 1);
else if (vt->cy > 0) {
vt->cy--;
}
break;
case '>':
/* numeric keypad - ignored */
break;
case '=':
/* appl. keypad - ignored */
break;
default:
#ifdef DEBUG
fprintf(stderr, "got unhandled ESC%c %d\n", ch, ch);
#endif
break;
}
break;
case ESnonstd: /* Operating System Controls */
vt->tty_state = ESnormal;
switch (ch) {
case 'P': /* palette escape sequence */
for(i = 0; i < MAX_ESC_PARAMS; i++) {
vt->esc_buf[i] = 0;
}
vt->esc_count = 0;
vt->tty_state = ESpalette;
break;
case 'R': /* reset palette */
// fixme: reset_palette(vc);
break;
case '0':
case '1':
case '2':
case '4':
vt->osc_cmd = ch;
vt->osc_textbuf[0] = 0;
vt->tty_state = ESosc1;
break;
default:
#ifdef DEBUG
fprintf (stderr, "unhandled OSC %c\n", ch);
#endif
vt->tty_state = ESnormal;
break;
}
break;
case ESosc1:
vt->tty_state = ESnormal;
if (ch == ';') {
vt->tty_state = ESosc2;
} else {
#ifdef DEBUG
fprintf (stderr, "got illegal OSC sequence\n");
#endif
}
break;
case ESosc2:
if (ch != 0x9c && ch != 7) {
int i = 0;
while (vt->osc_textbuf[i]) i++;
vt->osc_textbuf[i++] = ch;
vt->osc_textbuf[i] = 0;
} else {
#ifdef DEBUG
fprintf (stderr, "OSC:%c:%s\n", vt->osc_cmd, vt->osc_textbuf);
#endif
vt->tty_state = ESnormal;
}
break;
case ESpalette:
if ((ch >= '0' && ch <= '9') || (ch >= 'A' && ch <= 'F')
|| (ch >= 'a' && ch <= 'f')) {
vt->esc_buf[vt->esc_count++] = (ch > '9' ? (ch & 0xDF) - 'A' + 10 : ch - '0');
if (vt->esc_count == 7) {
// fixme: this does not work - please test
rfbColourMap *cmap =&vt->screen->colourMap;
int i = color_table[vt->esc_buf[0]] * 3, j = 1;
cmap->data.bytes[i] = 16 * vt->esc_buf[j++];
cmap->data.bytes[i++] += vt->esc_buf[j++];
cmap->data.bytes[i] = 16 * vt->esc_buf[j++];
cmap->data.bytes[i++] += vt->esc_buf[j++];
cmap->data.bytes[i] = 16 * vt->esc_buf[j++];
cmap->data.bytes[i] += vt->esc_buf[j];
//set_palette(vc); ?
vt->tty_state = ESnormal;
}
} else
vt->tty_state = ESnormal;
break;
case ESsquare:
for(i = 0; i < MAX_ESC_PARAMS; i++) {
vt->esc_buf[i] = 0;
}
vt->esc_count = 0;
vt->esc_has_par = 0;
vt->tty_state = ESgetpars;
if (ch == '>') {
vt->tty_state = ESidquery;
break;
}
if ((vt->esc_ques = (ch == '?'))) {
break;
}
case ESgetpars:
if (ch >= '0' && ch <= '9') {
vt->esc_has_par = 1;
if (vt->esc_count < MAX_ESC_PARAMS) {
vt->esc_buf[vt->esc_count] = vt->esc_buf[vt->esc_count] * 10 + ch - '0';
}
break;
} else if (ch == ';') {
vt->esc_count++;
break;
} else {
if (vt->esc_has_par) {
vt->esc_count++;
}
vt->tty_state = ESgotpars;
}
case ESgotpars:
vt->tty_state = ESnormal;
#ifdef DEBUG
char *qes = vt->esc_ques ? "?" : "";
if (vt->esc_count == 0) {
fprintf(stderr, "ESC[%s%c\n", qes, ch);
} else if (vt->esc_count == 1) {
fprintf(stderr, "ESC[%s%d%c\n", qes, vt->esc_buf[0], ch);
} else {
int i;
fprintf(stderr, "ESC[%s%d", qes, vt->esc_buf[0]);
for (i = 1; i < vt->esc_count; i++) {
fprintf(stderr, ";%d", vt->esc_buf[i]);
}
fprintf (stderr, "%c\n", ch);
}
#endif
switch (ch) {
case 'h':
vncterm_set_mode (vt, 1);
break;
case 'l':
vncterm_set_mode (vt, 0);
break;
case 'm':
if (!vt->esc_count) {
vt->esc_count++; // default parameter 0
}
vncterm_csi_m (vt);
break;
case 'n':
/* report cursor position */
/* TODO: send ESC[row;colR */
break;
case 'A':
/* move cursor up */
if (vt->esc_buf[0] == 0) {
vt->esc_buf[0] = 1;
}
vt->cy -= vt->esc_buf[0];
if (vt->cy < 0) {
vt->cy = 0;
}
break;
case 'B':
case 'e':
/* move cursor down */
if (vt->esc_buf[0] == 0) {
vt->esc_buf[0] = 1;
}
vt->cy += vt->esc_buf[0];
if (vt->cy >= vt->height) {
vt->cy = vt->height - 1;
}
break;
case 'C':
case 'a':
/* move cursor right */
if (vt->esc_buf[0] == 0) {
vt->esc_buf[0] = 1;
}
vt->cx += vt->esc_buf[0];
if (vt->cx >= vt->width) {
vt->cx = vt->width - 1;
}
break;
case 'D':
/* move cursor left */
if (vt->esc_buf[0] == 0) {
vt->esc_buf[0] = 1;
}
vt->cx -= vt->esc_buf[0];
if (vt->cx < 0) {
vt->cx = 0;
}
break;
case 'G':
case '`':
/* move cursor to column */
vncterm_gotoxy (vt, vt->esc_buf[0] - 1, vt->cy);
break;
case 'd':
/* move cursor to row */
vncterm_gotoxy (vt, vt->cx , vt->esc_buf[0] - 1);
break;
case 'f':
case 'H':
/* move cursor to row, column */
vncterm_gotoxy (vt, vt->esc_buf[1] - 1, vt->esc_buf[0] - 1);
break;
case 'J':
switch (vt->esc_buf[0]) {
case 0:
/* clear to end of screen */
for (y = vt->cy; y < vt->height; y++) {
for (x = 0; x < vt->width; x++) {
if (y == vt->cy && x < vt->cx) {
continue;
}
vncterm_clear_xy (vt, x, y);
}
}
break;
case 1:
/* clear from beginning of screen */
for (y = 0; y <= vt->cy; y++) {
for (x = 0; x < vt->width; x++) {
if (y == vt->cy && x > vt->cx) {
break;
}
vncterm_clear_xy (vt, x, y);
}
}
break;
case 2:
/* clear entire screen */
for (y = 0; y <= vt->height; y++) {
for (x = 0; x < vt->width; x++) {
vncterm_clear_xy (vt, x, y);
}
}
break;
}
break;
case 'K':
switch (vt->esc_buf[0]) {
case 0:
/* clear to eol */
for(x = vt->cx; x < vt->width; x++) {
vncterm_clear_xy (vt, x, vt->cy);
}
break;
case 1:
/* clear from beginning of line */
for (x = 0; x <= vt->cx; x++) {
vncterm_clear_xy (vt, x, vt->cy);
}
break;
case 2:
/* clear entire line */
for(x = 0; x < vt->width; x++) {
vncterm_clear_xy (vt, x, vt->cy);
}
break;
}
break;
case 'L':
/* insert line */
c = vt->esc_buf[0];
if (c > vt->height - vt->cy)
c = vt->height - vt->cy;
else if (!c)
c = 1;
vncterm_scroll_down (vt, vt->cy, vt->region_bottom, c);
break;
case 'M':
/* delete line */
c = vt->esc_buf[0];
if (c > vt->height - vt->cy)
c = vt->height - vt->cy;
else if (!c)
c = 1;
vncterm_scroll_up (vt, vt->cy, vt->region_bottom, c, 1);
break;
case 'T':
/* scroll down */
c = vt->esc_buf[0];
if (!c) c = 1;
vncterm_scroll_down (vt, vt->region_top, vt->region_bottom, c);
break;
case 'S':
/* scroll up */
c = vt->esc_buf[0];
if (!c) c = 1;
vncterm_scroll_up (vt, vt->region_top, vt->region_bottom, c, 1);
break;
case 'P':
/* delete c character */
c = vt->esc_buf[0];
if (c > vt->width - vt->cx)
c = vt->width - vt->cx;
else if (!c)
c = 1;
for (x = vt->cx; x < vt->width - c; x++) {
int y1 = (vt->y_base + vt->cy) % vt->total_height;
TextCell *dst = &vt->cells[y1 * vt->width + x];
TextCell *src = dst + c;
*dst = *src;
vncterm_update_xy (vt, x + c, vt->cy);
src->ch = ' ';
src->attrib = vt->default_attrib;
vncterm_update_xy (vt, x, vt->cy);
}
break;
case 's':
/* save cursor position */
vncterm_save_cursor (vt);
break;
case 'u':
/* restore cursor position */
vncterm_restore_cursor (vt);
break;
case 'X':
/* erase c characters */
c = vt->esc_buf[0];
if (!c) c = 1;
if (c > (vt->width - vt->cx)) c = vt->width - vt->cx;
for(i = 0; i < c; i++) {
vncterm_clear_xy (vt, vt->cx + i, vt->cy);
}
break;
case '@':
/* insert c character */
c = vt->esc_buf[0];
if (c > (vt->width - vt->cx)) {
c = vt->width - vt->cx;
}
if (!c) c = 1;
for (x = vt->width - c; x >= vt->cx; x--) {
int y1 = (vt->y_base + vt->cy) % vt->total_height;
TextCell *src = &vt->cells[y1 * vt->width + x];
TextCell *dst = src + c;
*dst = *src;
vncterm_update_xy (vt, x + c, vt->cy);
src->ch = ' ';
src->attrib = vt->cur_attrib;
vncterm_update_xy (vt, x, vt->cy);
}
break;
case 'r':
/* set region */
if (!vt->esc_buf[0])
vt->esc_buf[0]++;
if (!vt->esc_buf[1])
vt->esc_buf[1] = vt->height;
/* Minimum allowed region is 2 lines */
if (vt->esc_buf[0] < vt->esc_buf[1] &&
vt->esc_buf[1] <= vt->height) {
vt->region_top = vt->esc_buf[0] - 1;
vt->region_bottom = vt->esc_buf[1];
vt->cx = 0;
vt->cy = vt->region_top;
#ifdef DEBUG
fprintf (stderr, "set region %d %d\n", vt->region_top, vt->region_bottom);
#endif
}
break;
default:
#ifdef DEBUG
if (vt->esc_count == 0) {
fprintf(stderr, "unhandled escape ESC[%s%c\n", qes, ch);
} else if (vt->esc_count == 1) {
fprintf(stderr, "unhandled escape ESC[%s%d%c\n", qes, vt->esc_buf[0], ch);
} else {
int i;
fprintf(stderr, "unhandled escape ESC[%s%d", qes, vt->esc_buf[0]);
for (i = 1; i < vt->esc_count; i++) {
fprintf(stderr, ";%d", vt->esc_buf[i]);
}
fprintf (stderr, "%c\n", ch);
}
#endif
break;
}
vt->esc_ques = 0;
break;
case ESsetG0: // Set G0
vt->tty_state = ESnormal;
if (ch == '0')
vt->g0enc = GRAF_MAP;
else if (ch == 'B')
vt->g0enc = LAT1_MAP;
else if (ch == 'U')
vt->g0enc = IBMPC_MAP;
else if (ch == 'K')
vt->g0enc = USER_MAP;
if (vt->charset == 0)
vt->cur_enc = vt->g0enc;
break;
case ESsetG1: // Set G1
vt->tty_state = ESnormal;
if (ch == '0')
vt->g1enc = GRAF_MAP;
else if (ch == 'B')
vt->g1enc = LAT1_MAP;
else if (ch == 'U')
vt->g1enc = IBMPC_MAP;
else if (ch == 'K')
vt->g1enc = USER_MAP;
if (vt->charset == 1)
vt->cur_enc = vt->g1enc;
break;
case ESidquery: // vt100 query id
vt->tty_state = ESnormal;
if (ch == 'c') {
#ifdef DEBUG
fprintf (stderr, "ESC[>c Query term ID\n");
#endif
vncterm_respond_esc (vt, TERMIDCODE);
}
break;
case ESpercent:
vt->tty_state = ESnormal;
switch (ch) {
case '@': /* defined in ISO 2022 */
vt->utf8 = 0;
break;
case 'G': /* prelim official escape code */
case '8': /* retained for compatibility */
vt->utf8 = 1;
break;
}
break;
default: // ESnormal
vt->tty_state = ESnormal;
switch(ch) {
case 0:
break;
case 7: /* alert aka. bell */
rfbSendBell(vt->screen);
break;
case 8: /* backspace */
if (vt->cx > 0)
vt->cx--;
break;
case 9: /* tabspace */
if (vt->cx + (8 - (vt->cx % 8)) > vt->width) {
vt->cx = 0;
vncterm_put_lf (vt);
} else {
vt->cx = vt->cx + (8 - (vt->cx % 8));
}
break;
case 10: /* LF,*/
case 11: /* VT */
case 12: /* FF */
vncterm_put_lf (vt);
break;
case 13: /* carriage return */
vt->cx = 0;
break;
case 14:
/* SI (shift in), select character set 1 */
vt->charset = 1;
vt->cur_enc = vt->g1enc;
/* fixme: display controls = 1 */
break;
case 15:
/* SO (shift out), select character set 0 */
vt->charset = 0;
vt->cur_enc = vt->g0enc;
/* fixme: display controls = 0 */
break;
case 27: /* esc */
vt->tty_state = ESesc;
break;
case 127: /* delete */
/* ignore */
break;
case 128+27: /* csi */
vt->tty_state = ESsquare;
break;
default:
if (vt->cx >= vt->width) {
/* line wrap */
vt->cx = 0;
vncterm_put_lf (vt);
}
int y1 = (vt->y_base + vt->cy) % vt->total_height;
TextCell *c = &vt->cells[y1*vt->width + vt->cx];
c->attrib = vt->cur_attrib;
c->ch = ch;
vncterm_update_xy (vt, vt->cx, vt->cy);
vt->cx++;
break;
}
break;
}
}
static int
vncterm_puts (vncTerm *vt, const char *buf, int len)
{
unicode tc;
vncterm_show_cursor (vt, 0);
while (len) {
unsigned char c = *buf;
len--;
buf++;
if (vt->tty_state != ESnormal) {
// never translate escape sequence
tc = c;
} else if (vt->utf8 && !vt->cur_enc) {
if(c & 0x80) { // utf8 multi-byte sequence
if (vt->utf_count > 0 && (c & 0xc0) == 0x80) {
// inside UTF8 sequence
vt->utf_char = (vt->utf_char << 6) | (c & 0x3f);
vt->utf_count--;
if (vt->utf_count == 0) {
tc = vt->utf_char;
} else {
continue;
}
} else {
// first char of a UTF8 sequence
if ((c & 0xe0) == 0xc0) {
vt->utf_count = 1;
vt->utf_char = (c & 0x1f);
} else if ((c & 0xf0) == 0xe0) {
vt->utf_count = 2;
vt->utf_char = (c & 0x0f);
} else if ((c & 0xf8) == 0xf0) {
vt->utf_count = 3;
vt->utf_char = (c & 0x07);
} else if ((c & 0xfc) == 0xf8) {
vt->utf_count = 4;
vt->utf_char = (c & 0x03);
} else if ((c & 0xfe) == 0xfc) {
vt->utf_count = 5;
vt->utf_char = (c & 0x01);
} else
vt->utf_count = 0;
continue;
}
} else {
// utf8 single byte
tc = c;
vt->utf_count = 0;
}
} else {
// never translate controls
if (c >= 32 && c != 127 && c != (128+27)) {
tc = translations[vt->cur_enc][c & 0x0ff];
} else {
tc = c;
}
}
vncterm_putchar (vt, tc);
}
vncterm_show_cursor (vt, 1);
return len;
}
void
vncterm_kbd_event (rfbBool down, rfbKeySym keySym, rfbClientPtr cl)
{
vncTerm *vt =(vncTerm *)cl->screen->screenData;
static int control = 0;
static int shift = 0;
char *esc = NULL;
//fprintf (stderr, "KEYEVENT:%d: %08x\n", down == 0, keySym);fflush (stderr);
if (down) {
//fprintf (stderr, "KEYPRESS: %d\n", keySym);fflush (stderr);
if (keySym == XK_Shift_L || keySym == XK_Shift_R) {
shift = 1;
} if (keySym == XK_Control_L || keySym == XK_Control_R) {
control = 1;
} else if (vt->ibuf_count < (IBUFSIZE - 32)) {
if (control) {
if(keySym >= 'a' && keySym <= 'z')
keySym -= 'a' -1;
else if (keySym >= 'A' && keySym <= 'Z')
keySym -= 'A'-1;
else
keySym=0xffff;
} else {
switch (keySym) {
case XK_Escape:
keySym=27; break;
case XK_Return:
keySym='\r'; break;
case XK_BackSpace:
keySym=8; break;
case XK_Tab:
keySym='\t'; break;
case XK_Delete: /* kdch1 */
case XK_KP_Delete:
esc = "[3~";break;
case XK_Home: /* khome */
case XK_KP_Home:
esc = "OH";break;
case XK_End:
case XK_KP_End: /* kend */
esc = "OF";break;
case XK_Insert: /* kich1 */
case XK_KP_Insert:
esc = "[2~";break;
case XK_Up:
case XK_KP_Up: /* kcuu1 */
esc = "OA";break;
case XK_Down: /* kcud1 */
case XK_KP_Down:
esc = "OB";break;
case XK_Right:
case XK_KP_Right: /* kcuf1 */
esc = "OC";break;
case XK_Left:
case XK_KP_Left: /* kcub1 */
esc = "OD";break;
case XK_Page_Up:
if (shift) {
vncterm_virtual_scroll (vt, -vt->height/2);
return;
}
esc = "[5~";break;
case XK_Page_Down:
if (shift) {
vncterm_virtual_scroll (vt, vt->height/2);
return;
}
esc = "[6~";break;
case XK_F1:
esc = "OP";break;
case XK_F2:
esc = "OQ";break;
case XK_F3:
esc = "OR";break;
case XK_F4:
esc = "OS";break;
case XK_F5:
esc = "[15~";break;
case XK_F6:
esc = "[17~";break;
case XK_F7:
esc = "[18~";break;
case XK_F8:
esc = "[19~";break;
case XK_F9:
esc = "[20~";break;
case XK_F10:
esc = "[21~";break;
case XK_F11:
esc = "[23~";break;
case XK_F12:
esc = "[24~";break;
default:
break;
}
}
#ifdef DEBUG
fprintf (stderr, "KEYPRESS OUT:%s: %d\n", esc, keySym); fflush (stderr);
#endif
if (vt->y_displ != vt->y_base) {
vt->y_displ = vt->y_base;
vncterm_refresh (vt);
}
if (esc) {
vncterm_respond_esc (vt, esc);
} else if(keySym<0x100) {
if (vt->utf8) {
int len = ucs2_to_utf8 (keySym & 0x0fff, &vt->ibuf[vt->ibuf_count]);
vt->ibuf_count += len;
} else {
vt->ibuf[vt->ibuf_count++] = (char)keySym;
}
}
}
} else {
if (keySym == XK_Shift_L || keySym == XK_Shift_R) {
shift = 0;
} else if (keySym == XK_Control_L || keySym == XK_Control_R) {
control = 0;
}
}
}
void
vncterm_set_xcut_text (char* str, int len, struct _rfbClientRec* cl)
{
vncTerm *vt =(vncTerm *)cl->screen->screenData;
// seems str is Latin-1 encoded
if (vt->selection) free (vt->selection);
vt->selection = (unicode *)malloc (len*sizeof (unicode));
int i;
for (i = 0; i < len; i++) {
vt->selection[i] = str[i] & 0xff;
}
vt->selection_len = len;
}
static void
mouse_report (vncTerm *vt, int butt, int mrx, int mry)
{
char buf[8];
sprintf (buf, "[M%c%c%c", (char)(' ' + butt), (char)('!' + mrx),
(char)('!' + mry));
vncterm_respond_esc (vt, buf);
}
void
vncterm_toggle_marked_cell (vncTerm *vt, int pos)
{
int x= (pos%vt->width)*8;
int y= (pos/vt->width)*16;
int i,j;
rfbScreenInfoPtr s=vt->screen;
char *b = s->frameBuffer+y*s->width+x;
for (j=0; j < 16; j++) {
for(i=0; i < 8; i++) {
b[j*s->width+i] ^= 0x0f;
rfbMarkRectAsModified (s, x, y, x+8, y+16);
}
}
}
void
vncterm_pointer_event (int buttonMask, int x, int y, rfbClientPtr cl)
{
vncTerm *vt =(vncTerm *)cl->screen->screenData;
static int button2_released = 1;
static int last_mask = 0;
static int sel_start_pos = 0;
static int sel_end_pos = 0;
int i;
int cx = x/8;
int cy = y/16;
if (cx < 0) cx = 0;
if (cx >= vt->width) cx = vt->width - 1;
if (cy < 0) cy = 0;
if (cy >= vt->height) cy = vt->height - 1;
if (vt->report_mouse && buttonMask != last_mask) {
last_mask = buttonMask;
if (buttonMask & 1) {
mouse_report (vt, 0, cx, cy);
}
if (buttonMask & 2) {
mouse_report (vt, 1, cx, cy);
}
if (buttonMask & 4) {
mouse_report (vt, 2, cx, cy);
}
if (!buttonMask) {
mouse_report (vt, 3, cx, cy);
}
}
if (buttonMask & 2) {
if(button2_released && vt->selection) {
int i;
for(i = 0; i < vt->selection_len; i++) {
if (vt->ibuf_count < IBUFSIZE - 6) { // uft8 is max 6 characters wide
if (vt->utf8) {
vt->ibuf_count += ucs2_to_utf8 (vt->selection[i], &vt->ibuf[vt->ibuf_count]);
} else {
vt->ibuf[vt->ibuf_count++] = vt->selection[i];
}
}
}
if (vt->y_displ != vt->y_base) {
vt->y_displ = vt->y_base;
vncterm_refresh (vt);
}
}
button2_released = 0;
} else {
button2_released = 1;
}
if (buttonMask & 1) {
int pos = cy*vt->width + cx;
// code borrowed from libvncserver (VNConsole.c)
if (!vt->mark_active) {
vt->mark_active = 1;
sel_start_pos = sel_end_pos = pos;
vncterm_toggle_marked_cell (vt, pos);
} else {
if (pos != sel_end_pos) {
if (pos > sel_end_pos) {
cx = sel_end_pos; cy=pos;
} else {
cx=pos; cy=sel_end_pos;
}
if (cx < sel_start_pos) {
if (cy < sel_start_pos) cy--;
} else {
cx++;
}
while (cx <= cy) {
vncterm_toggle_marked_cell (vt, cx);
cx++;
}
sel_end_pos = pos;
}
}
} else if (vt->mark_active) {
vt->mark_active = 0;
if (sel_start_pos > sel_end_pos) {
int tmp = sel_start_pos - 1;
sel_start_pos = sel_end_pos;
sel_end_pos = tmp;
}
int len = sel_end_pos - sel_start_pos + 1;
if (vt->selection) free (vt->selection);
vt->selection = (unicode *)malloc (len*sizeof (unicode));
vt->selection_len = len;
char *sel_latin1 = (char *)malloc (len + 1);
for (i = 0; i < len; i++) {
int pos = sel_start_pos + i;
int x = pos % vt->width;
int y1 = ((pos / vt->width) + vt->y_displ) % vt->total_height;
TextCell *c = &vt->cells[y1*vt->width + x];
vt->selection[i] = c->ch;
sel_latin1[i] = (char)c->ch;
c++;
}
sel_latin1[len] = 0;
rfbGotXCutText (vt->screen, sel_latin1, len);
free (sel_latin1);
while (sel_start_pos <= sel_end_pos) {
vncterm_toggle_marked_cell (vt, sel_start_pos++);
}
}
rfbDefaultPtrAddEvent (buttonMask, x, y, cl);
}
static int client_count = 0;
static int client_connected = 0;
static int last_client = 1;
static time_t last_time = 0;
void
client_gone (rfbClientPtr client)
{
client_count--;
last_time = time (NULL);
if (client_count <= 0) {
last_client = 1;
}
}
/* libvncserver callback for when a new client connects */
enum rfbNewClientAction
new_client (rfbClientPtr client)
{
client->clientGoneHook = client_gone;
client_count++;
last_time = time (NULL);
last_client = 0;
client_connected = 1;
return RFB_CLIENT_ACCEPT;
}
vncTerm *
create_vncterm (int argc, char** argv, int maxx, int maxy)
{
int i;
rfbScreenInfoPtr screen = rfbGetScreen (&argc, argv, maxx, maxy, 8, 1, 1);
screen->frameBuffer=(char*)calloc(maxx*maxy, 1);
vncTerm *vt = (vncTerm *)calloc (sizeof(vncTerm), 1);
rfbColourMap *cmap =&screen->colourMap;
cmap->data.bytes = malloc (16*3);
for(i=0;i<16;i++) {
cmap->data.bytes[i*3 + 0] = default_red[color_table[i]];
cmap->data.bytes[i*3 + 1] = default_grn[color_table[i]];
cmap->data.bytes[i*3 + 2] = default_blu[color_table[i]];
}
cmap->count = 16;
cmap->is16 = FALSE;
screen->serverFormat.trueColour = FALSE;
screen->kbdAddEvent = vncterm_kbd_event;
screen->setXCutText = vncterm_set_xcut_text;
screen->ptrAddEvent = vncterm_pointer_event;
screen->desktopName = "VNC Command Terminal";
screen->newClientHook = new_client;
vt->maxx = screen->width;
vt->maxy = screen->height;
vt->width = vt->maxx / 8;
vt->height = vt->maxy / 16;
vt->total_height = vt->height * 20;
vt->scroll_height = 0;
vt->y_base = 0;
vt->y_displ = 0;
vt->region_top = 0;
vt->region_bottom = vt->height;
vt->g0enc = LAT1_MAP;
vt->g1enc = GRAF_MAP;
vt->cur_enc = vt->g0enc;
vt->charset = 0;
/* default text attributes */
vt->default_attrib.bold = 0;
vt->default_attrib.uline = 0;
vt->default_attrib.blink = 0;
vt->default_attrib.invers = 0;
vt->default_attrib.unvisible = 0;
vt->default_attrib.fgcol = 7;
vt->default_attrib.bgcol = 0;
vt->cur_attrib = vt->default_attrib;
vt->cells = (TextCell *)calloc (sizeof (TextCell), vt->width*vt->total_height);
for (i = 0; i < vt->width*vt->total_height; i++) {
vt->cells[i].ch = ' ';
vt->cells[i].attrib = vt->default_attrib;
}
vt->altcells = (TextCell *)calloc (sizeof (TextCell), vt->width*vt->height);
vt->screen = screen;
screen->screenData = (void*)vt;
//screen->autoPort = 1;
rfbRegisterSecurityHandler(&VncSecurityHandlerVencrypt);
rfbInitServer(screen);
return vt;
}
int
main (int argc, char** argv)
{
int i;
char **cmdargv = NULL;
char *command = "/bin/bash"; // execute normal shell as default
int pid;
int master;
char ptyname[1024];
fd_set fs, fs1;
struct timeval tv, tv1;
time_t elapsed, cur_time;
struct winsize dimensions;
if (gnutls_global_init () < 0) {
fprintf(stderr, "gnutls_global_init failed\n");
exit(-1);
}
if (gnutls_dh_params_init (&dh_params) < 0) {
fprintf(stderr, "gnutls_dh_params_init failed\n");
exit(-1);
}
if (gnutls_dh_params_generate2 (dh_params, DH_BITS) < 0) {
fprintf(stderr, "gnutls_dh_params_init failed\n");
exit(-1);
}
for (i = 1; i < argc; i++) {
if (!strcmp (argv[i], "-c")) {
command = argv[i+1];
cmdargv = &argv[i+1];
argc = i;
argv[i] = NULL;
break;
}
}
for (i = 1; i < argc; i++) {
if (!strcmp (argv[i], "-timeout")) {
CHECK_ARGC (argc, argv, i);
idle_timeout = atoi(argv[i+1]);
rfbPurgeArguments(&argc, &i, 2, argv);
}
if (!strcmp (argv[i], "-authpath")) {
CHECK_ARGC (argc, argv, i);
auth_path = argv[i+1];
rfbPurgeArguments(&argc, &i, 2, argv);
}
if (!strcmp (argv[i], "-perm")) {
CHECK_ARGC (argc, argv, i);
auth_perm = argv[i+1];
rfbPurgeArguments(&argc, &i, 2, argv);
}
}
#ifdef DEBUG
rfbLogEnable (1);
gnutls_global_set_log_level(10);
gnutls_global_set_log_function(vnc_debug_gnutls_log);
#else
rfbLogEnable (0);
#endif
vncTerm *vt = create_vncterm (argc, argv, 745, 400);
setlocale(LC_ALL, ""); // set from environment
char *ctype = setlocale (LC_CTYPE, NULL); // query LC_CTYPE
// fixme: ist there a standard way to detect utf8 mode ?
if (strcasestr (ctype, ".utf-8")||strcasestr (ctype, ".utf8")) {
vt->utf8 = 1;
}
dimensions.ws_col = vt->width;
dimensions.ws_row = vt->height;
setenv ("TERM", TERM, 1);
pid = forkpty (&master, ptyname, NULL, &dimensions);
if(!pid) {
// install default signal handlers
signal (SIGQUIT, SIG_DFL);
signal (SIGTERM, SIG_DFL);
signal (SIGINT, SIG_DFL);
if (cmdargv) {
execvp (command, cmdargv);
} else {
execlp (command, command, NULL);
}
perror ("Error: exec failed\n");
exit (-1); // should not be reached
} else if (pid == -1) {
perror ("Error: fork failed\n");
exit (-1);
}
FD_ZERO (&fs);
FD_SET (master, &fs);
tv.tv_sec = 0;
tv.tv_usec = 5000; /* 5 ms */
last_time = time (NULL);
int count = 0;
while (1) {
count ++;
tv1 = tv;
fs1 = fs;
cur_time = time (NULL);
elapsed = cur_time - last_time;
//printf ("Elapsed %ld\n", elapsed);
if (last_client) {
if (client_connected) {
if (idle_timeout && (elapsed >= idle_timeout)) {
break;
}
} else {
// wait at least 20 seconds for initial connect
if (idle_timeout && (elapsed >= (idle_timeout > 20 ? idle_timeout : 20))) {
break;
}
}
} else {
// exit after 30 minutes idle time
if (elapsed >= 30*60) {
break;
}
}
rfbProcessEvents (vt->screen, 40000); /* 40 ms */
if (vt->ibuf_count > 0) {
//printf ("DEBUG: WRITE %d %d\n", vt->ibuf[0], vt->ibuf_count);
write (master, vt->ibuf, vt->ibuf_count);
vt->ibuf_count = 0;
last_time = time (NULL);
}
if (!vt->mark_active) {
int num_fds = select (master+1, &fs1, NULL, NULL, &tv1);
if (num_fds >= 0) {
if (FD_ISSET (master, &fs1)) {
char buffer[1024];
int c;
while ((c = read (master, buffer, 1024)) == -1) {
if (errno != EAGAIN) break;
}
if (c == -1) break;
vncterm_puts (vt, buffer, c);
}
} else {
break;
}
}
}
kill (pid, 9);
int status;
waitpid(pid, &status, 0);
exit (0);
}