X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;ds=sidebyside;f=rdpsrv.c;h=b8bc56aa61001b44ecaf424ab9fb0d45eb69458a;hb=7df14c250a7cc36f89a8d348cd0af833a85d58dc;hp=577f4a726aef0e7f7076db91c0b79907e2d6c1dc;hpb=908d0bab155aee5757f902486dd1bbba859637f4;p=rdpsrv

diff --git a/rdpsrv.c b/rdpsrv.c
index 577f4a7..b8bc56a 100644
--- 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>
@@ -58,20 +59,173 @@ int create_server_socket()
 	return server_sock;
 }
 
+void handle_input_pdu(STREAM s)
+{
+	uint32 time;
+	uint16 message_type, device_flags, param1, param2;
+	uint16 num_events;
+	int i;
+
+	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, &param1, &param2);
+		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 (ignored)\n");
+			break;
+		case RDP_INPUT_MOUSE:
+			printf("- Type: Mouse\n");
+			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);
+			break;
+		}
+		printf("\n");
+	}
+}
+
+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
+
+	write(vnc_sock, buf, 5);
+	
+	// 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;
+}
 
 int serve_client()
 {
+	int vnc_sock = vnc_init();
+	
 	if (!mcs_recv_connect_initial())
 		error("MCS_CONNECT_INITIAL recv failed");
 	mcs_send_connect_response();
-	
+
 	for ( ;; ) {
-		short channel;
-		STREAM *s;
+		uint8 type, data_pdu_type;
+		STREAM s;
+
+		fd_set readfs;
+		FD_ZERO(&readfs);
+		FD_SET(tcp_get_socket(), &readfs);
+		FD_SET(vnc_sock, &readfs);
 		
-		/* receive ISO packets */
-		s = mcs_recv(&channel);
-		if (s != NULL)
-			printf("Packet on ch %u\n", channel);
+		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);
+				};
+			}
+		}
+
+		// 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 server message %x\n", buf[0]);
+				exit(1);
+			}
+		}
 	}
 }