+
+int map_scancode_to_vnc_key(int scancode)
+{
+ switch (scancode)
+ {
+ case SCANCODE_CHAR_1 ... SCANCODE_CHAR_9:
+ return scancode + '1' - SCANCODE_CHAR_1;
+ case SCANCODE_CHAR_0:
+ return '0';
+ case SCANCODE_CHAR_BACKSPACE:
+ return 0xff08;
+ case SCANCODE_CHAR_TAB:
+ return 0xff09;
+ case SCANCODE_CHAR_Q ... SCANCODE_CHAR_P:
+ return "QWERTYUIOP"[scancode - SCANCODE_CHAR_Q];
+ case SCANCODE_CHAR_A ... SCANCODE_CHAR_L:
+ return "ASDFGHJKL"[scancode - SCANCODE_CHAR_A];
+ case SCANCODE_CHAR_Z ... SCANCODE_CHAR_M:
+ return "ZXCVBNM"[scancode - SCANCODE_CHAR_Z];
+ default:
+ printf("Unknown scancode %x, ignoring\n", scancode);
+ return -1;
+ }
+}
+
+void handle_input_pdu(STREAM s, int vnc_sock)
+{
+ uint32 time;
+ uint16 message_type, device_flags, param1, param2;
+ uint16 num_events;
+ int i;
+ char buf[256];
+ static int mouse1_down = 0, mouse2_down = 0;
+ int vnc_key;
+
+ in_uint16_le(s, num_events); // number of events
+ in_uint8s(s, 2); // pad
+
+ for (i = 0; i < num_events; ++i) {
+ rdp_recv_input(s, &time, &message_type, &device_flags, ¶m1, ¶m2);
+ printf("Input event at time %u\n", time);
+
+ switch (message_type) {
+ case RDP_INPUT_SYNCHRONIZE:
+ printf("- Type: Synchronize (ignored)\n");
+ break;
+ case RDP_INPUT_CODEPOINT:
+ printf("- Type: Codepoint (ignored)\n");
+ break;
+ case RDP_INPUT_VIRTKEY:
+ printf("- Type: Virtual key (ignored)\n");
+ break;
+ case RDP_INPUT_SCANCODE:
+ printf("- Type: Scancode\n");
+ printf("- Device flags: %x\n", device_flags);
+ printf("- Key: %x\n", param1);
+
+ vnc_key = map_scancode_to_vnc_key(param1);
+
+ if (vnc_key != -1) {
+ buf[0] = 4; // message type
+ buf[1] = (device_flags & KBD_FLAG_DOWN) ? 1 : 0; // down flag
+ buf[2] = 0; // padding
+ buf[3] = 0;
+ buf[4] = vnc_key >> 24; // keysym
+ buf[5] = (vnc_key >> 16) & 0xff;
+ buf[6] = (vnc_key >> 8) & 0xff;
+ buf[7] = vnc_key & 0xff;
+ write(vnc_sock, buf, 8);
+ }
+
+ break;
+ case RDP_INPUT_MOUSE:
+ printf("- Type: Mouse\n");
+ printf("- Device flags: %x\n", device_flags);
+ printf("- Position: (%u,%u)\n", param1, param2);
+
+ if (device_flags & MOUSE_FLAG_BUTTON1)
+ mouse1_down = (device_flags & MOUSE_FLAG_DOWN) ? 0x01 : 0;
+ if (device_flags & MOUSE_FLAG_BUTTON2)
+ mouse2_down = (device_flags & MOUSE_FLAG_DOWN) ? 0x02 : 0;
+
+ printf("button mask = %x\n", mouse1_down | mouse2_down);
+
+ buf[0] = 5; // message type
+ buf[1] = mouse1_down | mouse2_down; // button mask
+ buf[2] = param1 >> 8;
+ buf[3] = param1 & 0xff;
+ buf[4] = param2 >> 8;
+ buf[5] = param2 & 0xff;
+ write(vnc_sock, buf, 6);
+
+ break;
+ default:
+ printf("- Unknown type %x\n", message_type);
+ break;
+ }
+ printf("\n");
+ }
+
+ // re-request the entire framebuffer
+ buf[0] = 3; // message type
+ buf[1] = 1; // incremental
+ buf[2] = 0; // xpos
+ buf[3] = 0;
+ buf[4] = 0; // ypos
+ buf[5] = 0;
+ buf[6] = 640 >> 8; // width
+ buf[7] = 640 & 0xff;
+ buf[8] = 480 >> 8; // height
+ buf[9] = 480 & 0xff;
+
+ write(vnc_sock, buf, 10);
+}
+
+struct ServerInitialization {
+ unsigned short width;
+ unsigned short height;
+ unsigned char pixelformat[16]; // FIXME
+ unsigned int name_len;
+};
+
+int vnc_init()
+{
+ char buf[256];
+ int vnc_sock = tcp_connect("127.0.0.1", 5901);
+ struct ServerInitialization si;
+
+ // get handshake
+ if (read(vnc_sock, buf, 12) != 12)
+ error("short read on handshake\n");
+ write(vnc_sock, "RFB 003.003\n", 12);
+
+ // get auth
+ if (read(vnc_sock, buf, 4) != 4)
+ error("short read on auth\n");
+
+ if (buf[3] != 1)
+ error("auth needed or connection failed\n");
+
+ // send shared flag
+ buf[0] = 1;
+ write(vnc_sock, buf, 1);
+
+ // read the server initialization
+ if (read(vnc_sock, &si, sizeof(struct ServerInitialization)) != sizeof(struct ServerInitialization))
+ error("short read on SI");
+
+ printf("Server is %u x %u\n", ntohs(si.width), ntohs(si.height));
+
+ if (read(vnc_sock, buf, ntohl(si.name_len)) != ntohl(si.name_len))
+ error("short read on server name\n");
+
+// printf("Server name is '%*s' (%u bytes)\n", ntohl(si.name_len), buf, ntohl(si.name_len));
+
+ // we can only accept raw encoding
+ buf[0] = 2; // message type
+ buf[1] = 0; // padding
+ buf[2] = 0; // number of encodings
+ buf[3] = 1;
+ buf[4] = 0; // raw encoding
+ buf[5] = 0;
+ buf[6] = 0;
+ buf[7] = 0;
+
+ write(vnc_sock, buf, 8);
+
+ // request the entire framebuffer
+ buf[0] = 3; // message type
+ buf[1] = 0; // incremental
+ buf[2] = 0; // xpos
+ buf[3] = 0;
+ buf[4] = 0; // ypos
+ buf[5] = 0;
+ buf[6] = 640 >> 8; // width
+ buf[7] = 640 & 0xff;
+ buf[8] = 480 >> 8; // height
+ buf[9] = 480 & 0xff;
+
+ write(vnc_sock, buf, 10);
+
+ printf("Connected to VNC!\n");
+
+ return vnc_sock;
+}
+
+struct vnc_rectangle {
+ unsigned short x, y, width, height;
+ unsigned int encoding;
+};
+
+void handle_vnc_fbupdate(int vnc_sock)
+{
+ struct vnc_rectangle rect;
+ unsigned char *data, *ptr, *src, *dst;
+ unsigned short num_rect;
+ int ret, data_left, i, xt, yt;
+ unsigned char smallblock[64 * 64 * 3];
+
+ if (read(vnc_sock, &num_rect, 2) != 2)
+ error("short read on num_rect\n");
+
+ for (i = 0; i < ntohs(num_rect); ++i) {
+ int width, height;
+
+ if (read(vnc_sock, &rect, sizeof(struct vnc_rectangle)) != sizeof(struct vnc_rectangle))
+ error("short read on vnc_rectangle\n");
+
+ if (ntohl(rect.encoding) != 0)
+ error("unsupported encoding\n");