write(vnc_sock, buf, 10);
}
+void handle_control_pdu(STREAM s)
+{
+ uint16 action, userid;
+ uint32 control_id;
+
+ in_uint16_le(s, action);
+ in_uint16_le(s, userid);
+ in_uint32(s, control_id);
+
+ printf("Control PDU: action=%hu userid=%hu control_id=%u\n", action, userid, control_id);
+
+ switch (action) {
+ case RDP_CTL_COOPERATE:
+ printf("Cooperate\n");
+ rdp_send_control(RDP_CTL_COOPERATE);
+ break;
+ case RDP_CTL_REQUEST_CONTROL:
+ printf("Client requesting control; granting\n");
+ rdp_send_control(RDP_CTL_GRANT_CONTROL);
+ break;
+ default:
+ printf("Unhandled\n");
+ }
+}
+
+void
+handle_font2_pdu(STREAM s)
+{
+ printf("FONT2 PDU, responding with magic\n");
+
+ s = rdp_init_data(8);
+ out_uint16_le(s, 0);
+ out_uint16_le(s, 0);
+ out_uint16_le(s, 3);
+ out_uint16_le(s, 4);
+ s_mark_end(s);
+ rdp_send_data(s, 0x28);
+}
+
struct ServerInitialization {
unsigned short width;
unsigned short height;
unsigned int encoding;
};
+#define SBX 32
+#define SBY 32
+
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];
+ unsigned char smallblock[SBX * SBY * 3];
if (read(vnc_sock, &num_rect, 2) != 2)
error("short read on num_rect\n");
// 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) {
+ for (yt = 0; yt < (height + (SBY-1)) / SBY; ++yt) {
+ for (xt = 0; xt < (width + (SBX-1)) / SBX; ++xt) {
int x, y;
- int bw = width - xt * 64;
- int bh = height - yt * 64;
- if (bw > 64)
- bw = 64;
- if (bh > 64)
- bh = 64;
+ int bw = width - xt * SBX;
+ int bh = height - yt * SBY;
+ if (bw > SBX)
+ bw = SBX;
+ if (bh > SBY)
+ bh = SBY;
dst = smallblock;
for (y = 0; y < bh; ++y) {
- src = data + ((yt * 64 + (bh - 1 - y)) * width + (xt * 64)) * 4;
+ src = data + ((yt * SBY + (bh - 1 - y)) * width + (xt * SBX)) * 4;
for (x = 0; x < bw; ++x) {
*dst++ = *src++;
*dst++ = *src++;
}
}
- rdp_send_bitmap_update(ntohs(rect.x) + xt * 64, ntohs(rect.y) + yt * 64, bw, bh, smallblock);
+ rdp_send_bitmap_update(ntohs(rect.x) + xt * SBX, ntohs(rect.y) + yt * SBY, bw, bh, smallblock);
}
}
}
}
+int listen_on_vnc = 0;
+
int serve_client()
{
int vnc_sock = vnc_init();
- int listen_on_vnc = 0;
if (!mcs_recv_connect_initial())
error("MCS_CONNECT_INITIAL recv failed");
for ( ;; ) {
uint8 type, data_pdu_type;
+ uint16 id;
STREAM s;
fd_set readfs;
// 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:
- handle_input_pdu(s, vnc_sock);
- listen_on_vnc = 1;
+ switch (type) {
+ case RDP_PDU_DATA:
+ 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:
+ handle_input_pdu(s, vnc_sock);
+ listen_on_vnc = 1;
+ break;
+ case RDP_DATA_PDU_CONTROL:
+ handle_control_pdu(s);
+ break;
+ case RDP_DATA_PDU_SYNCHRONISE:
+ in_uint16_le(s, id);
+ printf("Synchronise, id=%u\n", id);
+ rdp_send_synchronise(id);
+ break;
+ case RDP_DATA_PDU_FONT2:
+ handle_font2_pdu(s);
+ break;
+ default:
+ printf("Unknown data PDU type %u\n", data_pdu_type);
+ };
+ break;
+ case RDP_PDU_CONFIRM_ACTIVE:
+ printf("Client confirms activity\n");
break;
default:
- printf("Unknown data PDU type %u\n", data_pdu_type);
- };
+ printf("Unknown RDP PDU type %u\n", type);
+ }
}
}
error("short read on vnc_sock\n");
if (read(vnc_sock, buf, 4) != 4)
error("short read on vnc_sock\n");
- if (read(vnc_sock, (buf[0] << 24) | (buf[1] << 16) | (buf[2] << 8) | buf[3], 4) != 4)
+ if (read(vnc_sock, buf, (buf[0] << 24) | (buf[1] << 16) | (buf[2] << 8) | buf[3]) != 4)
error("short read on vnc_sock\n");
default: