+/*
+
+ 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;
+
+ setsid ();
+
+ 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);
+}