fbvnc.c

   1 /*
   2  * fbvnc - a small linux framebuffer vnc viewer
   3  *
   4  * Copyright (C) 2009-2010 Ali Gholami Rudi
   5  *
   6  * This program is free software; you can redistribute it and/or modify it
   7  * under the terms of the GNU General Public License, as published by the
   8  * Free Software Foundation.
   9  */
  10 #include <arpa/inet.h>
  11 #include <ctype.h>
  12 #include <errno.h>
  13 #include <fcntl.h>
  14 #include <netdb.h>
  15 #include <netinet/in.h>
  16 #include <poll.h>
  17 #include <pwd.h>
  18 #include <stdio.h>
  19 #include <stdlib.h>
  20 #include <string.h>
  21 #include <termios.h>
  22 #include <unistd.h>
  23 #include <sys/socket.h>
  24 #include <sys/stat.h>
  25 #include <sys/types.h>
  26 #include <linux/input.h>
  27 #include "draw.h"
  28 #include "vnc.h"
  29
  30 #define VNC_PORT                "5900"
  31
  32 #define MAXRES                  (1 << 21)
  33 #define MIN(a, b)               ((a) < (b) ? (a) : (b))
  34
  35 static int cols, rows;
  36 static int mr, mc;              /* mouse position */
  37
  38 static char buf[MAXRES];
  39
  40 static int vnc_connect(char *addr, char *port)
  41 {
  42         struct addrinfo hints, *addrinfo;
  43         int fd;
  44
  45         memset(&hints, 0, sizeof(hints));
  46         hints.ai_family = AF_UNSPEC;
  47         hints.ai_socktype = SOCK_STREAM;
  48         hints.ai_flags = AI_PASSIVE;
  49
  50         if (getaddrinfo(addr, port, &hints, &addrinfo))
  51                 return -1;
  52         fd = socket(addrinfo->ai_family, addrinfo->ai_socktype,
  53                         addrinfo->ai_protocol);
  54
  55         if (connect(fd, addrinfo->ai_addr, addrinfo->ai_addrlen) == -1) {
  56                 close(fd);
  57                 freeaddrinfo(addrinfo);
  58                 return -1;
  59         }
  60         freeaddrinfo(addrinfo);
  61         return fd;
  62 }
  63
  64 static int vnc_init(int fd)
  65 {
  66         char vncver[] = "RFB 003.003\n";
  67         struct vnc_client_init clientinit;
  68         struct vnc_server_init serverinit;
  69         struct vnc_client_pixelfmt pixfmt_cmd;
  70         int connstat = VNC_CONN_FAILED;
  71
  72         write(fd, vncver, 12);
  73         read(fd, vncver, 12);
  74
  75         read(fd, &connstat, sizeof(connstat));
  76
  77         if (ntohl(connstat) != VNC_CONN_NOAUTH)
  78                 return -1;
  79
  80         clientinit.shared = 1;
  81         write(fd, &clientinit, sizeof(clientinit));
  82         read(fd, &serverinit, sizeof(serverinit));
  83
  84         fb_init();
  85         cols = MIN(ntohs(serverinit.w), fb_cols());
  86         rows = MIN(ntohs(serverinit.h), fb_rows());
  87         mr = rows / 2;
  88         mc = cols / 2;
  89
  90         read(fd, buf, ntohl(serverinit.len));
  91         pixfmt_cmd.type = VNC_CLIENT_PIXFMT;
  92         pixfmt_cmd.format.bpp = 8;
  93         pixfmt_cmd.format.depth = 8;
  94         pixfmt_cmd.format.bigendian = 0;
  95         pixfmt_cmd.format.truecolor = 1;
  96
  97         pixfmt_cmd.format.rmax = htons(3);
  98         pixfmt_cmd.format.gmax = htons(7);
  99         pixfmt_cmd.format.bmax = htons(7);
 100         pixfmt_cmd.format.rshl = 0;
 101         pixfmt_cmd.format.gshl = 2;
 102         pixfmt_cmd.format.bshl = 5;
 103
 104         write(fd, &pixfmt_cmd, sizeof(pixfmt_cmd));
 105         return fd;
 106 }
 107
 108 static int vnc_free(void)
 109 {
 110         fb_free();
 111         return 0;
 112 }
 113
 114 static int vnc_refresh(int fd, int inc)
 115 {
 116         struct vnc_client_fbup fbup_req;
 117         fbup_req.type = VNC_CLIENT_FBUP;
 118         fbup_req.inc = inc;
 119         fbup_req.x = htons(0);
 120         fbup_req.y = htons(0);
 121         fbup_req.w = htons(cols);
 122         fbup_req.h = htons(rows);
 123         write(fd, &fbup_req, sizeof(fbup_req));
 124         return 0;
 125 }
 126
 127 static void drawfb(char *s, int x, int y, int w, int h)
 128 {
 129         fbval_t slice[1 << 14];
 130         int i, j;
 131         for (i = 0; i < h; i++) {
 132                 for (j = 0; j < w; j++) {
 133                         int c = *(unsigned char *) &s[i * w + j];
 134                         int r = (c & 0x3) << 6;
 135                         int g = ((c >> 2) & 0x7) << 5;
 136                         int b = ((c >> 5) & 0x7) << 5;
 137                         slice[j] = fb_color(r, g, b);
 138                 }
 139                 fb_set(y + i, x, slice, w);
 140         }
 141 }
 142
 143 static void xread(int fd, void *buf, int len)
 144 {
 145         int nr = 0;
 146         int n;
 147         while (nr < len && (n = read(fd, buf + nr, len - nr)) > 0)
 148                 nr += n;
 149         if (nr < len) {
 150                 printf("partial vnc read!\n");
 151                 exit(1);
 152         }
 153 }
 154
 155 static int vnc_event(int fd)
 156 {
 157         struct vnc_rect uprect;
 158         char msg[1 << 12];
 159         struct vnc_server_fbup *fbup = (void *) msg;
 160         struct vnc_server_cuttext *cuttext = (void *) msg;
 161         struct vnc_server_colormap *colormap = (void *) msg;
 162         int j;
 163         int n;
 164
 165         if (read(fd, msg, 1) != 1)
 166                 return -1;
 167         switch (msg[0]) {
 168         case VNC_SERVER_FBUP:
 169                 xread(fd, msg + 1, sizeof(*fbup) - 1);
 170                 n = ntohs(fbup->n);
 171                 for (j = 0; j < n; j++) {
 172                         int x, y, w, h;
 173                         xread(fd, &uprect, sizeof(uprect));
 174                         x = ntohs(uprect.x);
 175                         y = ntohs(uprect.y);
 176                         w = ntohs(uprect.w);
 177                         h = ntohs(uprect.h);
 178                         if (x >= cols || x + w > cols)
 179                                 return -1;
 180                         if (y >= rows || y + h > rows)
 181                                 return -1;
 182                         xread(fd, buf, w * h);
 183                         drawfb(buf, x, y, w, h);
 184                 }
 185                 break;
 186         case VNC_SERVER_BELL:
 187                 break;
 188         case VNC_SERVER_CUTTEXT:
 189                 xread(fd, msg + 1, sizeof(*cuttext) - 1);
 190                 xread(fd, buf, ntohl(cuttext->len));
 191                 break;
 192         case VNC_SERVER_COLORMAP:
 193                 xread(fd, msg + 1, sizeof(*colormap) - 1);
 194                 xread(fd, buf, ntohs(colormap->n) * 3 * 2);
 195                 break;
 196         default:
 197                 fprintf(stderr, "unknown vnc msg: %d\n", msg[0]);
 198                 return -1;
 199         }
 200         return 0;
 201 }
 202
 203 static int rat_event(int fd, int ratfd)
 204 {
 205         char ie[3];
 206         struct vnc_client_ratevent me = {VNC_CLIENT_RATEVENT};
 207         int mask = 0;
 208         if (read(ratfd, &ie, sizeof(ie)) != 3)
 209                 return -1;
 210         mc += ie[1];
 211         mr -= ie[2];
 212         mc = MAX(0, MIN(cols - 1, mc));
 213         mr = MAX(0, MIN(rows - 1, mr));
 214         if (ie[0] & 0x01)
 215                 mask |= VNC_BUTTON1_MASK;
 216         if (ie[0] & 0x04)
 217                 mask |= VNC_BUTTON2_MASK;
 218         if (ie[0] & 0x02)
 219                 mask |= VNC_BUTTON3_MASK;
 220         me.y = htons(mr);
 221         me.x = htons(mc);
 222         me.mask = mask;
 223         write(fd, &me, sizeof(me));
 224         return 0;
 225 }
 226
 227 static int press(int fd, int key, int down)
 228 {
 229         struct vnc_client_keyevent ke = {VNC_CLIENT_KEYEVENT};
 230         ke.key = htonl(key);
 231         ke.down = down;
 232         return write(fd, &ke, sizeof(ke));
 233 }
 234
 235 static int kbd_event(int fd, int kbdfd)
 236 {
 237         char key[1024];
 238         int i, nr;
 239
 240         if ((nr = read(kbdfd, key, sizeof(key))) <= 0 )
 241                 return -1;
 242         for (i = 0; i < nr; i++) {
 243                 int k = -1;
 244                 int mod[4];
 245                 int nmod = 0;
 246                 switch (key[i]) {
 247                 case 0x08:
 248                 case 0x7f:
 249                         k = 0xff08;
 250                         break;
 251                 case 0x09:
 252                         k = 0xff09;
 253                         break;
 254                 case 0x1b:
 255                         if (i + 2 < nr && key[i + 1] == '[') {
 256                                 if (key[i + 2] == 'A')
 257                                         k = 0xff52;
 258                                 if (key[i + 2] == 'B')
 259                                         k = 0xff54;
 260                                 if (key[i + 2] == 'C')
 261                                         k = 0xff53;
 262                                 if (key[i + 2] == 'D')
 263                                         k = 0xff51;
 264                                 if (key[i + 2] == 'H')
 265                                         k = 0xff50;
 266                                 if (k > 0) {
 267                                         i += 2;
 268                                         break;
 269                                 }
 270                         }
 271                         k = 0xff1b;
 272                         if (i + 1 < nr) {
 273                                 mod[nmod++] = 0xffe9;
 274                                 k = key[++i];
 275                         }
 276                         break;
 277                 case 0x0d:
 278                         k = 0xff0d;
 279                         break;
 280                 case 0x03:
 281                         return -1;
 282                 default:
 283                         k = (unsigned char) key[i];
 284                 }
 285                 if (isupper(k) || strchr(":\"<>?{}|+_()*&^%$#@!~", k))
 286                         mod[nmod++] = 0xffe1;
 287                 if (k >= 1 && k <= 26) {
 288                         k = 'a' + k - 1;
 289                         mod[nmod++] = 0xffe3;
 290                 }
 291                 if (k > 0) {
 292                         int j;
 293                         for (j = 0; j < nmod; j++)
 294                                 press(fd, mod[j], 1);
 295                         press(fd, k, 1);
 296                         press(fd, k, 0);
 297                         for (j = 0; j < nmod; j++)
 298                                 press(fd, mod[j], 0);
 299                 }
 300         }
 301         return 0;
 302 }
 303
 304 static void term_setup(struct termios *ti)
 305 {
 306         struct termios termios;
 307         char *hide = "\x1b[?25l";
 308         char *clear = "\x1b[2J\x1b[H";
 309         char *msg = "\t\t\t*** fbvnc ***\r\n";
 310
 311         write(STDIN_FILENO, hide, strlen(hide));
 312         write(STDOUT_FILENO, clear, strlen(clear));
 313         write(STDOUT_FILENO, msg, strlen(msg));
 314         tcgetattr (0, &termios);
 315         *ti = termios;
 316         cfmakeraw(&termios);
 317         tcsetattr(0, TCSANOW, &termios);
 318 }
 319
 320 static void term_cleanup(struct termios *ti)
 321 {
 322         char *show = "\x1b[?25h";
 323         tcsetattr(0, TCSANOW, ti);
 324         write(STDIN_FILENO, show, strlen(show));
 325 }
 326
 327 static void mainloop(int vnc_fd, int kbd_fd, int rat_fd)
 328 {
 329         struct pollfd ufds[3];
 330         int pending = 0;
 331         int update = 1;
 332         int err;
 333         ufds[0].fd = kbd_fd;
 334         ufds[0].events = POLLIN;
 335         ufds[1].fd = vnc_fd;
 336         ufds[1].events = POLLIN;
 337         ufds[2].fd = rat_fd;
 338         ufds[2].events = POLLIN;
 339         while (1) {
 340                 if (update && !pending) {
 341                         static int i;
 342                         if (vnc_refresh(vnc_fd, i++) == -1)
 343                                 break;
 344                         pending = 1;
 345                         update = 0;
 346                 }
 347                 err = poll(ufds, 3, 500);
 348                 if (err == -1 && errno != EINTR)
 349                         break;
 350                 if (!err)
 351                         continue;
 352                 if (ufds[0].revents & POLLIN) {
 353                         if (kbd_event(vnc_fd, kbd_fd) == -1)
 354                                 break;
 355                         update = 1;
 356                 }
 357                 if (ufds[1].revents & POLLIN) {
 358                         if (vnc_event(vnc_fd) == -1)
 359                                 break;
 360                         pending = 0;
 361                         update = 1;
 362                 }
 363                 if (ufds[2].revents & POLLIN) {
 364                         if (rat_event(vnc_fd, rat_fd) == -1)
 365                                 break;
 366                         update = 1;
 367                 }
 368         }
 369 }
 370
 371 int main(int argc, char * argv[])
 372 {
 373         char *port = VNC_PORT;
 374         char *host = "127.0.0.1";
 375         struct termios ti;
 376         int vnc_fd, rat_fd;
 377         if (argc >= 2)
 378                 host = argv[1];
 379         if (argc >= 3)
 380                 port = argv[2];
 381         if ((vnc_fd = vnc_connect(host, port)) == -1) {
 382                 fprintf(stderr, "could not connect!\n");
 383                 return 1;
 384         }
 385         if (vnc_init(vnc_fd) == -1) {
 386                 fprintf(stderr, "vnc init failed!\n");
 387                 return 1;
 388         }
 389         term_setup(&ti);
 390         rat_fd = open("/dev/input/mice", O_RDONLY);
 391
 392         mainloop(vnc_fd, 0, rat_fd);
 393
 394         term_cleanup(&ti);
 395         vnc_free();
 396         close(vnc_fd);
 397         close(rat_fd);
 398         return 0;
 399 }