We have picture!

[rdpsrv] / rdpsrv.c

diff --git a/rdpsrv.c b/rdpsrv.c
index b8bc56aa61001b44ecaf424ab9fb0d45eb69458a..1bf41d70a83821225294fec81bbf444e1527824f 100644 (file)
--- a/rdpsrv.c
+++ b/rdpsrv.c
@@ -91,9 +91,6 @@ void handle_input_pdu(STREAM s)
                        printf("- Device flags: %x\n", device_flags);
                        printf("- Position: (%u,%u)\n", param1, param2);
 
-                       // debug
-                       rdp_send_bitmap_update(param1, param2);
-                       
                        break;
                default:
                        printf("- Unknown type %x\n", message_type);
@@ -149,8 +146,11 @@ int vnc_init()
        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, 5);
+       write(vnc_sock, buf, 8);
        
        // request the entire framebuffer
        buf[0] = 3; // message type
@@ -160,9 +160,9 @@ int vnc_init()
        buf[4] = 0; // ypos
        buf[5] = 0;
        buf[6] = 640 >> 8; // width
-       buf[7] = 640 % 0xff;
+       buf[7] = 640 & 0xff;
        buf[8] = 480 >> 8; // height
-       buf[9] = 480 % 0xff;
+       buf[9] = 480 & 0xff;
 
        write(vnc_sock, buf, 10);
        
@@ -171,9 +171,85 @@ int vnc_init()
        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");
+
+               width = ntohs(rect.width);
+               height = ntohs(rect.height);
+               
+               printf("UPDATE: %ux%u at (%u,%u)\n", width, height,
+                       ntohs(rect.x), ntohs(rect.y));
+               
+               ptr = data = xmalloc(width * height * 4);
+               data_left = width * height * 4;
+
+               while (data_left > 0) {
+                       ret = read(vnc_sock, ptr, data_left);
+                       if (ret <= 0)
+                               error("error on data read\n");
+
+                       ptr += ret;
+                       data_left -= ret;
+               }
+
+               // push our 32-bit RGB data into small enough chunks
+               // (64x64) so we are sure we can send them without
+               // everything crashing and stuff :-)
+               for (yt = 0; yt < (height + 63) / 64; ++yt) {
+                       for (xt = 0; xt < (width + 63) / 64; ++xt) {
+                               int x, y;
+                               int bw = width - xt * 64;
+                               int bh = height - yt * 64;
+                               if (bw > 64)
+                                       bw = 64;
+                               if (bh > 64)
+                                       bh = 64;
+                       
+                               dst = smallblock;
+                               for (y = 0; y < bh; ++y) {
+                                       src = data + ((yt * 64 + (bh + 1 - y)) * width + (xt * 64)) * 4;
+                                       for (x = 0; x < bw; ++x) {
+                                               *dst++ = *src++;
+                                               *dst++ = *src++;
+                                               *dst++ = *src++;
+                                               ++src;
+                                       }
+                               }
+
+                               rdp_send_bitmap_update(ntohs(rect.x) + xt * 64, ntohs(rect.y) + yt * 64, bw, bh, smallblock);
+                       }
+               }
+               
+               xfree(data);
+       }
+}
+
 int serve_client()
 {
        int vnc_sock = vnc_init();
+       int listen_on_vnc = 0;
        
        if (!mcs_recv_connect_initial())
                error("MCS_CONNECT_INITIAL recv failed");
@@ -206,6 +282,7 @@ int serve_client()
                                case RDP_DATA_PDU_INPUT:
                                        printf("Input PDU\n");
                                        handle_input_pdu(s);
+                                       listen_on_vnc = 1;
                                        break;
                                default:
                                        printf("Unknown data PDU type %u\n", data_pdu_type);
@@ -214,16 +291,21 @@ int serve_client()
                }
 
                // activity on VNC socket?
-               if (FD_ISSET(vnc_sock, &readfs)) {
-                       char buf[256];
-                       read(vnc_sock, buf, 1);
+               if (FD_ISSET(vnc_sock, &readfs) && listen_on_vnc) {
+                       unsigned char buf[256];
+
+                       printf("Activity on VNC socket!\n");
+                       
+                       if (read(vnc_sock, buf, 2) != 2)
+                               error("short read on vnc_sock\n");
+
                        switch (buf[0]) {
                        case 0:
                                // frame buffer update!
-                               printf("Frame buffer update\n");
+                               handle_vnc_fbupdate(vnc_sock);
                                break;
                        default:
-                               printf("Unknown server message %x\n", buf[0]);
+                               printf("Unknown VNC server message %x\n", buf[0]);
                                exit(1);
                        }
                }