Understand server name _correctly_ this time (we hope :-P), initial "parsing" of...

[rdpsrv] / rdpsrv.c

diff --git a/rdpsrv.c b/rdpsrv.c
index b1b5ea78663c4e7ad06e87a9da73e1ec91d5e573..b8bc56aa61001b44ecaf424ab9fb0d45eb69458a 100644 (file)
--- a/rdpsrv.c
+++ b/rdpsrv.c
@@ -2,6 +2,7 @@
 #include <stdlib.h>
 #include <unistd.h>
 #include <errno.h>
+#include <sys/select.h>
 #include <sys/socket.h>
 #include <sys/ioctl.h>
 #include <arpa/inet.h>
@@ -137,10 +138,10 @@ int vnc_init()
 
        printf("Server is %u x %u\n", ntohs(si.width), ntohs(si.height));
 
-       if (read(vnc_sock, buf, ntohs(si.name_len)) != ntohs(si.name_len))
+       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'\n", ntohs(si.name_len), buf);
+       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
@@ -182,24 +183,49 @@ int serve_client()
                uint8 type, data_pdu_type;
                STREAM s;
 
-               while ((s = rdp_recv(&type)) != NULL) {
-                       if (type != RDP_PDU_DATA) {
-                               printf("Unknown RDP packet of type %u\n", type);
-                               continue;
+               fd_set readfs;
+               FD_ZERO(&readfs);
+               FD_SET(tcp_get_socket(), &readfs);
+               FD_SET(vnc_sock, &readfs);
+               
+               select(FD_SETSIZE, &readfs, NULL, NULL, NULL);
+
+               // activity on RDP socket?
+               if (FD_ISSET(tcp_get_socket(), &readfs)) {
+                       if ((s = rdp_recv(&type)) != NULL) {
+                               if (type != RDP_PDU_DATA) {
+                                       printf("Unknown RDP packet of type %u\n", type);
+                                       continue;
+                               }
+
+                               in_uint8s(s, 8);        /* shareid, pad, streamid, length */
+                               in_uint8(s, data_pdu_type);
+                               in_uint8s(s, 3);        /* compress_type, compress_len */
+
+                               switch (data_pdu_type) {
+                               case RDP_DATA_PDU_INPUT:
+                                       printf("Input PDU\n");
+                                       handle_input_pdu(s);
+                                       break;
+                               default:
+                                       printf("Unknown data PDU type %u\n", data_pdu_type);
+                               };
                        }
+               }
 
-                       in_uint8s(s, 8);        /* shareid, pad, streamid, length */
-                       in_uint8(s, data_pdu_type);
-                       in_uint8s(s, 3);        /* compress_type, compress_len */
-
-                       switch (data_pdu_type) {
-                       case RDP_DATA_PDU_INPUT:
-                               printf("Input PDU\n");
-                               handle_input_pdu(s);
+               // activity on VNC socket?
+               if (FD_ISSET(vnc_sock, &readfs)) {
+                       char buf[256];
+                       read(vnc_sock, buf, 1);
+                       switch (buf[0]) {
+                       case 0:
+                               // frame buffer update!
+                               printf("Frame buffer update\n");
                                break;
                        default:
-                               printf("Unknown data PDU type %u\n", data_pdu_type);
-                       };
+                               printf("Unknown server message %x\n", buf[0]);
+                               exit(1);
+                       }
                }
        }
 }