/[rdesktop]/sourceforge.net/trunk/rdesktop/rdp.c

This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!

Diff of /sourceforge.net/trunk/rdesktop/rdp.c

Parent Directory | Revision Log | View Patch Patch

-revision 9 by matty,
Tue Jul 25 12:34:29 2000 UTC
+revision 1238 by matthewc,
Wed Jun 14 08:59:03 2006 UTC
 Line 1
- /*
+ /* -*- c-basic-offset: 8 -*-
     rdesktop: A Remote Desktop Protocol client.
     Protocol services - RDP layer
-    Copyright (C) Matthew Chapman 1999-2000
+    Copyright (C) Matthew Chapman 1999-2005
     This program is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
     the Free Software Foundation; either version 2 of the License, or
     (at your option) any later version.
     This program is distributed in the hope that it will be useful,
     but WITHOUT ANY WARRANTY; without even the implied warranty of
     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     GNU General Public License for more details.
     You should have received a copy of the GNU General Public License
     along with this program; if not, write to the Free Software
     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  */
- #include "includes.h"
+ #include <time.h>
+ #include <errno.h>
- /* Establish a connection up to the RDP layer */
+ #include <unistd.h>
- HCONN rdp_connect(char *server, int width, int height)
+ #include "rdesktop.h"
- {
-         HCONN conn;
+ #ifdef HAVE_ICONV
+ #ifdef HAVE_ICONV_H
-         if ((conn = mcs_connect(server)) == NULL)
+ #include <iconv.h>
-                 return NULL;
+ #endif
-         rdp_establish_key(conn);
-         mcs_recv(conn, False); /* Server's licensing certificate */
-         rdp_send_cert(conn);
-         mcs_recv(conn, False);
-         mcs_recv(conn, False);
-         return conn;
- }
+ #ifndef ICONV_CONST
+ #define ICONV_CONST ""
+ #endif
+ #endif
- /* Work this out later. This is useless anyway when encryption is off. */
+ extern uint16 g_mcs_userid;
- uint8 precanned_key_packet[] = {
+ extern char g_username[64];
-x48,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x33,0x00,0x00,0x00,0x00,0x00,
+ extern char g_codepage[16];
-x00,0x00,0x20,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6c,0x86,
+ extern BOOL g_bitmap_compression;
-xf7,0x99,0xef,0x60,0xc4,0x49,0x52,0xd0,0xd8,0xea,0xb5,0x4f,0x58,0x19,
+ extern BOOL g_orders;
-x52,0x2a,0x93,0x83,0x57,0x4f,0x4e,0x04,0xde,0x96,0x51,0xab,0x13,0x20,
+ extern BOOL g_encryption;
-xd8,0xe5,0x00,0x00,0x00,0x00,0x00,0x00
+ extern BOOL g_desktop_save;
- };
+ extern BOOL g_polygon_ellipse_orders;
+ extern BOOL g_use_rdp5;
+ extern uint16 g_server_rdp_version;
+ extern uint32 g_rdp5_performanceflags;
+ extern int g_server_depth;
+ extern int g_width;
+ extern int g_height;
+ extern BOOL g_bitmap_cache;
+ extern BOOL g_bitmap_cache_persist_enable;
+ extern BOOL g_numlock_sync;
+ uint8 *g_next_packet;
+ uint32 g_rdp_shareid;
+ extern RDPCOMP g_mppc_dict;
+ /* Session Directory support */
+ extern BOOL g_redirect;
+ extern char g_redirect_server[64];
+ extern char g_redirect_domain[16];
+ extern char g_redirect_password[64];
+ extern char g_redirect_username[64];
+ extern char g_redirect_cookie[128];
+ extern uint32 g_redirect_flags;
+ /* END Session Directory support */
- uint8 precanned_key_packet_e1[] = {
+ #if WITH_DEBUG
-x01,0x00,0x00,0x00,0x48,0x00,0x00,0x00,0x7c,0xbd,0x8b,0x8f,0x16,0x2b,0xa1,0x00,
+ static uint32 g_packetno;
-xc6,0xfb,0x8a,0x39,0xf5,0x33,0xed,0x36,0x14,0x55,0x17,0x8c,0x3a,0xde,0x5e,0xdf,
+ #endif
-xcb,0x41,0x4c,0xc7,0x89,0x7d,0xe3,0xe9,0x34,0x08,0xda,0xdc,0x08,0x77,0x98,0xda,
-x65,0xae,0x27,0x74,0xf1,0x79,0xd0,0x28,0x54,0x64,0x86,0x7f,0x02,0xe0,0x71,0x51,
-x56,0x4e,0xca,0x72,0x94,0x62,0x49,0x27,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00
- };
- uint8 precanned_key_packet_e2[] = {
+ #ifdef HAVE_ICONV
-x48,0x00,0x00,0x00,0x8a,0xe4,0x9f,0x8a,0xd5,0x04,0x02,0xfd,0x09,0x1f,0xff,0x53,
+ static BOOL g_iconv_works = True;
-xe0,0xb2,0x72,0x8b,0x19,0xba,0x22,0xe4,0x2a,0x7b,0xeb,0x79,0xa8,0x83,0x31,0x6f,
+ #endif
-x5c,0xcc,0x37,0x9c,0xe8,0x73,0x64,0x64,0xd3,0xab,0xaa,0x9f,0xbe,0x49,0x27,0xfc,
-x95,0xf3,0x6e,0xf8,0xb1,0x01,0x7c,0xba,0xa9,0xc5,0x35,0x9c,0x8f,0x74,0x3a,0x9f,
-xd4,0x26,0x4d,0x39,0x90,0xbe,0xf4,0xfb,0x72,0x9e,0x54,0x18
- };
- /* Create an RC4 key and transfer it to the server */
+ /* Receive an RDP packet */
- void rdp_establish_key(HCONN conn)
+ static STREAM
- {
+ rdp_recv(uint8 * type)
-         mcs_init_data(conn);
+ {
-         memcpy(conn->out.data + conn->out.offset, precanned_key_packet,
+         static STREAM rdp_s;
-                sizeof(precanned_key_packet));
+         uint16 length, pdu_type;
-         conn->out.offset += sizeof(precanned_key_packet);
+         uint8 rdpver;
-         MARK_END(conn->out);
-         mcs_send_data(conn, MCS_GLOBAL_CHANNEL, True);
+         if ((rdp_s == NULL) || (g_next_packet >= rdp_s->end) || (g_next_packet == NULL))
- }
+         {
+                 rdp_s = sec_recv(&rdpver);
- /* Create an RC4 key and transfer it to the server */
+                 if (rdp_s == NULL)
- void rdp_establish_key_e1(HCONN conn)
+                         return NULL;
- {
+                 if (rdpver == 0xff)
-         mcs_init_data(conn);
+                 {
-         memcpy(conn->out.data + conn->out.offset, precanned_key_packet_e1,
+                         g_next_packet = rdp_s->end;
-                sizeof(precanned_key_packet_e1));
+                         *type = 0;
-         conn->out.offset += sizeof(precanned_key_packet_e1);
+                         return rdp_s;
-         MARK_END(conn->out);
+                 }
-         mcs_send_data(conn, MCS_GLOBAL_CHANNEL, True);
+                 else if (rdpver != 3)
- }
+                 {
+                         /* rdp5_process should move g_next_packet ok */
- /* Create an RC4 key and transfer it to the server */
+                         rdp5_process(rdp_s);
- void rdp_establish_key_e2(HCONN conn)
+                         *type = 0;
- {
+                         return rdp_s;
-         mcs_init_data(conn);
+                 }
-         memcpy(conn->out.data + conn->out.offset, precanned_key_packet_e2,
-                sizeof(precanned_key_packet_e2));
-         conn->out.offset += sizeof(precanned_key_packet_e2);
-         MARK_END(conn->out);
-         mcs_send_data(conn, MCS_GLOBAL_CHANNEL, True);
- }
- /* Horrible horrible certificate stuff. Work out later. */
- uint8 precanned_cert_packet[] = { // 4c8
-x80,0x00,0x00,0x00,0x12,0x02,0xb4,0x04,0x01,0x00,0x00,
-x00,0x00,0x00,0x01,0x02,0x9d,0xa3,0x7a,0x93,0x34,0x7b,0x28,0x37,0x24,0xa0,0x1f,
-x61,0x26,0xfd,0x96,0x3a,0x92,0x83,0xf3,0xe9,0x6a,0x2e,0x81,0x7c,0x2c,0xe4,0x72,//
-x01,0x18,0xe9,0xa1,0x0f,0x00,0x00,0x48,0x00,0x84,0x23,0x90,0xe6,0xd3,0xf8,0x20,
-xdb,0xa8,0x1b,0xb2,0xd0,0x78,0x2c,0x35,0xde,0xe3,0x0e,0x63,0x40,0xca,0xac,0x71,
-xc9,0x17,0x49,0x05,0x25,0xeb,0x9b,0xd0,0xa6,0x5c,0x90,0x3e,0x9d,0x4b,0x27,0x01,
-x79,0x1c,0x22,0xfb,0x3c,0x2c,0xb9,0x9f,0xf5,0x21,0xf3,0xee,0xd5,0x4d,0x47,0x1c,
-x85,0xbe,0x83,0x93,0xe8,0xed,0x8c,0x5c,0x82,0x00,0x00,0x00,0x00,0x00,0x00,0x00,
-x00,0x01,0x00,0x10,0x04,0x30,0x82,0x04,0x0c,0x06,0x09,0x2a,0x86,0x48,0x86,0xf7,
-x0d,0x01,0x07,0x02,0xa0,0x82,0x03,0xfd,0x30,0x82,0x03,0xf9,0x02,0x01,0x01,0x31,
-x00,0x30,0x0b,0x06,0x09,0x2a,0x86,0x48,0x86,0xf7,0x0d,0x01,0x07,0x01,0xa0,0x82,
-x03,0xe1,0x30,0x82,0x01,0x77,0x30,0x82,0x01,0x25,0xa0,0x03,0x02,0x01,0x02,0x02,
-x08,0x01,0xbf,0x06,0x84,0x9d,0xdb,0x2d,0xe0,0x30,0x0d,0x06,0x09,0x2a,0x86,0x48,
-x86,0xf7,0x0d,0x01,0x01,0x01,0x05,0x00,0x30,0x38,0x31,0x36,0x30,0x11,0x06,0x03,
-x55,0x04,0x03,0x1e,0x0a,0x00,0x4e,0x00,0x54,0x00,0x54,0x00,0x53,0x00,0x45,0x30,
-x21,0x06,0x03,0x55,0x04,0x07,0x1e,0x1a,0x00,0x4d,0x00,0x69,0x00,0x63,0x00,0x72,
-x00,0x6f,0x00,0x73,0x00,0x6f,0x00,0x66,0x00,0x74,0x00,0x2e,0x00,0x63,0x00,0x6f,
-x00,0x6d,0x30,0x1e,0x17,0x0d,0x39,0x39,0x30,0x39,0x32,0x34,0x31,0x32,0x30,0x32,
-x30,0x34,0x5a,0x17,0x0d,0x34,0x39,0x30,0x39,0x32,0x34,0x31,0x32,0x30,0x32,0x30,
-x34,0x5a,0x30,0x38,0x31,0x36,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x1e,0x0a,0x00,
-x4e,0x00,0x54,0x00,0x54,0x00,0x53,0x00,0x45,0x30,0x21,0x06,0x03,0x55,0x04,0x07,
-x1e,0x1a,0x00,0x4d,0x00,0x69,0x00,0x63,0x00,0x72,0x00,0x6f,0x00,0x73,0x00,0x6f,
-x00,0x66,0x00,0x74,0x00,0x2e,0x00,0x63,0x00,0x6f,0x00,0x6d,0x30,0x5c,0x30,0x0d,
-x06,0x09,0x2a,0x86,0x48,0x86,0xf7,0x0d,0x01,0x01,0x01,0x05,0x00,0x03,0x4b,0x00,
-x30,0x48,0x02,0x41,0x00,0x91,0xb2,0x16,0x1c,0xae,0x4f,0x7f,0x7c,0xaf,0x57,0x2b,
-x23,0x4c,0x0c,0x25,0x3c,0x4f,0x66,0x9d,0x25,0xc3,0x4f,0x29,0xee,0x8b,0xda,0x4e,
-x95,0xe7,0x3b,0xaa,0xc0,0xa7,0xba,0xaf,0x99,0x8c,0x47,0x24,0x8b,0x09,0x77,0xbc,
-x2c,0xf4,0xe7,0x1a,0x07,0x58,0x7b,0x11,0x37,0x2a,0xa8,0x90,0xc3,0x50,0x92,0x80,
-x15,0xc5,0xda,0x51,0x8b,0x02,0x03,0x01,0x00,0x01,0xa3,0x13,0x30,0x11,0x30,0x0f,
-x06,0x03,0x55,0x1d,0x13,0x04,0x08,0x30,0x06,0x01,0x01,0xff,0x02,0x01,0x00,0x30,
-x09,0x06,0x05,0x2b,0x0e,0x03,0x02,0x1d,0x05,0x00,0x03,0x41,0x00,0x14,0x04,0x67,
-x28,0xc8,0xd3,0x1f,0x13,0x14,0x2e,0x2c,0x93,0x09,0x25,0xbb,0xbe,0x86,0x6a,0xd3,
-x47,0x6f,0x44,0x16,0x7b,0x94,0x8c,0xb2,0xa2,0xd5,0xf7,0x4f,0xb1,0x8f,0x7f,0xde,
-x0b,0x88,0x34,0x4a,0x1d,0xdc,0xa1,0xfd,0x26,0xbd,0x43,0xbb,0x38,0xf1,0x87,0x34,
-xbb,0xe9,0x3b,0xfa,0x7f,0x1e,0xff,0xe1,0x10,0x7e,0xee,0x6e,0xd8,0x30,0x82,0x02,
-x62,0x30,0x82,0x02,0x10,0xa0,0x03,0x02,0x01,0x02,0x02,0x05,0x01,0x00,0x00,0x00,
-x01,0x30,0x09,0x06,0x05,0x2b,0x0e,0x03,0x02,0x1d,0x05,0x00,0x30,0x38,0x31,0x36,
-x30,0x11,0x06,0x03,0x55,0x04,0x03,0x1e,0x0a,0x00,0x4e,0x00,0x54,0x00,0x54,0x00,
-x53,0x00,0x45,0x30,0x21,0x06,0x03,0x55,0x04,0x07,0x1e,0x1a,0x00,0x4d,0x00,0x69,
-x00,0x63,0x00,0x72,0x00,0x6f,0x00,0x73,0x00,0x6f,0x00,0x66,0x00,0x74,0x00,0x2e,
-x00,0x63,0x00,0x6f,0x00,0x6d,0x30,0x1e,0x17,0x0d,0x39,0x39,0x30,0x39,0x32,0x34,
-x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x17,0x0d,0x34,0x39,0x30,0x39,0x32,0x34,0x30,
-x30,0x30,0x30,0x30,0x30,0x5a,0x30,0x79,0x31,0x77,0x30,0x17,0x06,0x03,0x55,0x04,
-x03,0x1e,0x10,0x00,0x52,0x00,0x45,0x00,0x53,0x00,0x37,0x00,0x2d,0x00,0x4e,0x00,
-x45,0x00,0x57,0x30,0x17,0x06,0x03,0x55,0x04,0x07,0x1e,0x10,0x00,0x7a,0x00,0x32,
-x00,0x32,0x00,0x33,0x00,0x32,0x00,0x32,0x00,0x30,0x00,0x33,0x30,0x43,0x06,0x03,
-x55,0x04,0x05,0x1e,0x3c,0x00,0x31,0x00,0x42,0x00,0x63,0x00,0x4b,0x00,0x65,0x00,
-x57,0x00,0x50,0x00,0x6c,0x00,0x37,0x00,0x58,0x00,0x47,0x00,0x61,0x00,0x73,0x00,
-x38,0x00,0x4a,0x00,0x79,0x00,0x50,0x00,0x34,0x00,0x30,0x00,0x7a,0x00,0x49,0x00,
-x6d,0x00,0x6e,0x00,0x6f,0x00,0x51,0x00,0x5a,0x00,0x59,0x00,0x3d,0x00,0x0d,0x00,
-x0a,0x30,0x5c,0x30,0x0d,0x06,0x09,0x2a,0x86,0x48,0x86,0xf7,0x0d,0x01,0x01,0x01,
-x05,0x00,0x03,0x4b,0x00,0x30,0x48,0x02,0x41,0x00,0x91,0xb2,0x16,0x1c,0xae,0x4f,
-x7f,0x7c,0xaf,0x57,0x2b,0x23,0x4c,0x0c,0x25,0x3c,0x4f,0x66,0x9d,0x25,0xc3,0x4f,
-x29,0xee,0x8b,0xda,0x4e,0x95,0xe7,0x3b,0xaa,0xc0,0xa7,0xba,0xaf,0x99,0x8c,0x47,
-x24,0x8b,0x09,0x77,0xbc,0x2c,0xf4,0xe7,0x1a,0x07,0x58,0x7b,0x11,0x37,0x2a,0xa8,
-x90,0xc3,0x50,0x92,0x80,0x15,0xc5,0xda,0x51,0x8b,0x02,0x03,0x01,0x00,0x01,0xa3,
-x81,0xc3,0x30,0x81,0xc0,0x30,0x14,0x06,0x09,0x2b,0x06,0x01,0x04,0x01,0x82,0x37,
-x12,0x04,0x01,0x01,0xff,0x04,0x04,0x01,0x00,0x01,0x00,0x30,0x3c,0x06,0x09,0x2b,
-x06,0x01,0x04,0x01,0x82,0x37,0x12,0x02,0x01,0x01,0xff,0x04,0x2c,0x4d,0x00,0x69,
-x00,0x63,0x00,0x72,0x00,0x6f,0x00,0x73,0x00,0x6f,0x00,0x66,0x00,0x74,0x00,0x20,
-x00,0x43,0x00,0x6f,0x00,0x72,0x00,0x70,0x00,0x6f,0x00,0x72,0x00,0x61,0x00,0x74,
-x00,0x69,0x00,0x6f,0x00,0x6e,0x00,0x00,0x00,0x30,0x4c,0x06,0x09,0x2b,0x06,0x01,
-x04,0x01,0x82,0x37,0x12,0x05,0x01,0x01,0xff,0x04,0x3c,0x00,0x10,0x00,0x00,0x01,
-x00,0x00,0x00,0x02,0x00,0x00,0x00,0x09,0x04,0x00,0x00,0x18,0x00,0x18,0x00,0x30,
-x00,0x01,0x00,0x32,0x00,0x33,0x00,0x36,0x00,0x2d,0x00,0x34,0x00,0x2e,0x00,0x30,
-x00,0x30,0x00,0x2d,0x00,0x45,0x00,0x58,0x00,0x00,0x00,0x04,0x00,0x00,0x00,0x00,
-x00,0x00,0x00,0x00,0x00,0x00,0x00,0x30,0x1c,0x06,0x03,0x55,0x1d,0x23,0x01,0x01,
-xff,0x04,0x12,0x30,0x10,0xa1,0x07,0x81,0x05,0x4e,0x54,0x54,0x53,0x45,0x82,0x05,
-x01,0x00,0x00,0x00,0x01,0x30,0x09,0x06,0x05,0x2b,0x0e,0x03,0x02,0x1d,0x05,0x00,
-x03,0x41,0x00,0x7b,0x1d,0xfd,0x24,0xea,0xf2,0xe8,0x17,0xdd,0x88,0x7e,0xfd,0xee,
-x28,0x61,0x7a,0x02,0xc3,0x73,0xcf,0x32,0x0f,0x7c,0x66,0x87,0x31,0xa7,0xbe,0x1b,
-x31,0xe2,0x20,0xa5,0x76,0x91,0x68,0x97,0x53,0x9e,0x80,0xcd,0x2b,0xd0,0x8e,0x8b,
-x7f,0x89,0x1b,0x62,0xa8,0xf8,0xee,0x5e,0x56,0xbd,0x9c,0x6b,0x80,0x06,0x54,0xd3,
-xf0,0xbf,0xb2,0x31,0x00,0x01,0x00,0x14,0x00,0xc7,0x32,0xf2,0x5b,0x98,0x0e,0x04,
-x49,0xa0,0x27,0x7e,0xf5,0xf6,0x0f,0xda,0x08,0x1d,0xe9,0x79,0xd1,0x31,0xc6,0x50,
-x90,0x4a,0xd3,0x1f,0x1d,0xf0,0x65,0x0d,0xb6,0x1f,0xaf,0xc9,0x1d
- };
- /* Send license certificate and related data to the server */
+                 g_next_packet = rdp_s->p;
- void rdp_send_cert(HCONN conn)
+         }
- {
+         else
-         mcs_init_data(conn);
+         {
-         prs_io_uint8s(&conn->out, precanned_cert_packet, sizeof(precanned_cert_packet));
+                 rdp_s->p = g_next_packet;
-         MARK_END(conn->out);
+         }
-         mcs_send_data(conn, MCS_GLOBAL_CHANNEL, True);
- }
- /* Initialise RDP transport packet */
+         in_uint16_le(rdp_s, length);
- void rdp_init(HCONN conn)
+         /* 32k packets are really 8, keepalive fix */
- {
+         if (length == 0x8000)
-         mcs_init_data(conn);
+         {
-         PUSH_LAYER(conn->out, rdp_offset, 6);
+                 g_next_packet += 8;
- }
+                 *type = 0;
+                 return rdp_s;
+         }
+         in_uint16_le(rdp_s, pdu_type);
+         in_uint8s(rdp_s, 2);    /* userid */
+         *type = pdu_type & 0xf;
- /* Transmit RDP transport packet */
+ #if WITH_DEBUG
- void rdp_send(HCONN conn, uint16 pdu_type)
+         DEBUG(("RDP packet #%d, (type %x)\n", ++g_packetno, *type));
- {
+         hexdump(g_next_packet, length);
-         RDP_HEADER hdr;
+ #endif /*  */
-         int length;
-         POP_LAYER(conn->out, rdp_offset);
+         g_next_packet += length;
-         length = conn->out.end - conn->out.offset;
+         return rdp_s;
-         rdp_make_header(&hdr, length, pdu_type, conn->mcs_userid);
-         rdp_io_header(&conn->out, &hdr);
-         mcs_send_data(conn, MCS_GLOBAL_CHANNEL, True);
  }
- /* Initialise RDP transport data packet */
+ /* Initialise an RDP data packet */
- void rdp_init_data(HCONN conn)
+ static STREAM
+ rdp_init_data(int maxlen)
  {
-         mcs_init_data(conn);
+         STREAM s;
-         PUSH_LAYER(conn->out, rdp_offset, 18);
- }
- /* Transmit RDP transport data packet */
+         s = sec_init(g_encryption ? SEC_ENCRYPT : 0, maxlen + 18);
- void rdp_send_data(HCONN conn, uint16 data_pdu_type)
+         s_push_layer(s, rdp_hdr, 18);
- {
-         RDP_HEADER hdr;
-         RDP_DATA_HEADER datahdr;
-         int length = conn->out.end - conn->out.offset;
-         POP_LAYER(conn->out, rdp_offset);
+         return s;
-         length = conn->out.end - conn->out.offset;
-         rdp_make_header(&hdr, length, RDP_PDU_DATA, conn->mcs_userid);
-         rdp_io_header(&conn->out, &hdr);
-         rdp_make_data_header(&datahdr, 0x103ea, length, data_pdu_type);
-         rdp_io_data_header(&conn->out, &datahdr);
-         mcs_send_data(conn, MCS_GLOBAL_CHANNEL, True);
  }
- void rdp_send_confirm_active(HCONN conn, uint32 shareid, int width, int height)
+ /* Send an RDP data packet */
+ static void
+ rdp_send_data(STREAM s, uint8 data_pdu_type)
  {
-         RDP_ACTIVE_PDU active;
+         uint16 length;
-         rdp_init(conn);
-         rdp_make_active_pdu(&active, shareid, conn->mcs_userid, width, height);
-         rdp_io_active_pdu(&conn->out, &active, RDP_PDU_CONFIRM_ACTIVE);
-         MARK_END(conn->out);
-         rdp_send(conn, RDP_PDU_CONFIRM_ACTIVE);
- }
- void rdp_send_synchronize(HCONN conn)
+         s_pop_layer(s, rdp_hdr);
- {
+         length = s->end - s->p;
-         RDP_SYNCHRONISE_PDU sync;
-         rdp_init_data(conn);
+         out_uint16_le(s, length);
-         rdp_make_synchronise_pdu(&sync, 1002);
+         out_uint16_le(s, (RDP_PDU_DATA | 0x10));
-         rdp_io_synchronise_pdu(&conn->out, &sync);
+         out_uint16_le(s, (g_mcs_userid + 1001));
-         MARK_END(conn->out);
-         rdp_send_data(conn, RDP_DATA_PDU_SYNCHRONIZE);
- }
- void rdp_send_control(HCONN conn, uint16 action)
+         out_uint32_le(s, g_rdp_shareid);
- {
+         out_uint8(s, 0);        /* pad */
-         RDP_CONTROL_PDU control;
+         out_uint8(s, 1);        /* streamid */
+         out_uint16_le(s, (length - 14));
+         out_uint8(s, data_pdu_type);
+         out_uint8(s, 0);        /* compress_type */
+         out_uint16(s, 0);       /* compress_len */
-         rdp_init_data(conn);
+         sec_send(s, g_encryption ? SEC_ENCRYPT : 0);
-         rdp_make_control_pdu(&control, action);
-         rdp_io_control_pdu(&conn->out, &control);
-         MARK_END(conn->out);
-         rdp_send_data(conn, RDP_DATA_PDU_CONTROL);
  }
- void rdp_send_fonts(HCONN conn, uint16 seqno)
+ /* Output a string in Unicode */
+ void
+ rdp_out_unistr(STREAM s, char *string, int len)
  {
-         RDP_FONT_PDU fonts;
+ #ifdef HAVE_ICONV
+         size_t ibl = strlen(string), obl = len + 2;
+         static iconv_t iconv_h = (iconv_t) - 1;
+         char *pin = string, *pout = (char *) s->p;
-         rdp_init_data(conn);
+         memset(pout, 0, len + 4);
-         rdp_make_font_pdu(&fonts, seqno);
-         rdp_io_font_pdu(&conn->out, &fonts);
-         MARK_END(conn->out);
-         rdp_send_data(conn, RDP_DATA_PDU_FONT2);
- }
- void rdp_send_input(HCONN conn, uint16 message_type, uint16 device_flags,
+         if (g_iconv_works)
-                                 uint16 param1, uint16 param2)
+         {
- {
+                 if (iconv_h == (iconv_t) - 1)
-         RDP_INPUT_PDU input;
+                 {
+                         size_t i = 1, o = 4;
+                         if ((iconv_h = iconv_open(WINDOWS_CODEPAGE, g_codepage)) == (iconv_t) - 1)
+                         {
+                                 warning("rdp_out_unistr: iconv_open[%s -> %s] fail %d\n",
+                                         g_codepage, WINDOWS_CODEPAGE, (int) iconv_h);
-         rdp_init_data(conn);
+                                 g_iconv_works = False;
-         rdp_make_input_pdu(&input, message_type, device_flags, param1, param2);
+                                 rdp_out_unistr(s, string, len);
-         rdp_io_input_pdu(&conn->out, &input);
+                                 return;
-         MARK_END(conn->out);
+                         }
-         rdp_send_data(conn, RDP_DATA_PDU_INPUT);
+                         if (iconv(iconv_h, (ICONV_CONST char **) &pin, &i, &pout, &o) ==
- }
+                             (size_t) - 1)
+                         {
+                                 iconv_close(iconv_h);
+                                 iconv_h = (iconv_t) - 1;
+                                 warning("rdp_out_unistr: iconv(1) fail, errno %d\n", errno);
+                                 g_iconv_works = False;
+                                 rdp_out_unistr(s, string, len);
+                                 return;
+                         }
+                         pin = string;
+                         pout = (char *) s->p;
+                 }
- BOOL rdp_recv_pdu(HCONN conn, uint8 *type)
+                 if (iconv(iconv_h, (ICONV_CONST char **) &pin, &ibl, &pout, &obl) == (size_t) - 1)
- {
+                 {
-         RDP_HEADER hdr;
+                         iconv_close(iconv_h);
+                         iconv_h = (iconv_t) - 1;
+                         warning("rdp_out_unistr: iconv(2) fail, errno %d\n", errno);
+                         g_iconv_works = False;
+                         rdp_out_unistr(s, string, len);
+                         return;
+                 }
-         conn->in.offset = conn->in.rdp_offset;
+                 s->p += len + 2;
-         if (conn->in.offset >= conn->in.end)
-         {
-                 if (!mcs_recv(conn, False))
-                         return False;
          }
+         else
-         if (!rdp_io_header(&conn->in, &hdr))
-                 return False;
-         conn->in.rdp_offset += hdr.length;
-         *type = hdr.pdu_type & 0xf;
- #if DUMP
-         fprintf(stderr, "RDP packet (type %x):\n", *type);
-         dump_data(conn->in.data+conn->in.offset, conn->in.rdp_offset-conn->in.offset);
  #endif
+         {
+                 int i = 0, j = 0;
-         return True;
+                 len += 2;
- }
- /* Disconnect from the RDP layer */
+                 while (i < len)
- void rdp_disconnect(HCONN conn)
+                 {
- {
+                         s->p[i++] = string[j++];
-         mcs_disconnect(conn);
+                         s->p[i++] = 0;
- }
+                 }
- /* Construct an RDP header */
+                 s->p += len;
- void rdp_make_header(RDP_HEADER *hdr, uint16 length, uint16 pdu_type,
+         }
-                      uint16 userid)
- {
-         hdr->length = length;
-         hdr->pdu_type = pdu_type | 0x10; /* Version 1 */
-         hdr->userid = userid + 1001;
  }
- /* Parse an RDP header */
+ /* Input a string in Unicode
- BOOL rdp_io_header(STREAM s, RDP_HEADER *hdr)
+  *
+  * Returns str_len of string
+  */
+ int
+ rdp_in_unistr(STREAM s, char *string, int uni_len)
  {
-         BOOL res = True;
+ #ifdef HAVE_ICONV
+         size_t ibl = uni_len, obl = uni_len;
-         res = res ? lsb_io_uint16(s, &hdr->length  ) : False;
+         char *pin = (char *) s->p, *pout = string;
-         res = res ? lsb_io_uint16(s, &hdr->pdu_type) : False;
+         static iconv_t iconv_h = (iconv_t) - 1;
-         if ((hdr->pdu_type & 0xf) != RDP_PDU_DEACTIVATE)
-                 res = res ? lsb_io_uint16(s, &hdr->userid  ) : False;
-         return res;
+         if (g_iconv_works)
- }
+         {
+                 if (iconv_h == (iconv_t) - 1)
- /* Construct a data header */
+                 {
- void rdp_make_data_header(RDP_DATA_HEADER *hdr, uint32 shareid,
+                         if ((iconv_h = iconv_open(g_codepage, WINDOWS_CODEPAGE)) == (iconv_t) - 1)
-                           uint16 length, uint16 data_pdu_type)
+                         {
- {
+                                 warning("rdp_in_unistr: iconv_open[%s -> %s] fail %d\n",
-         hdr->shareid = shareid;
+                                         WINDOWS_CODEPAGE, g_codepage, (int) iconv_h);
-         hdr->pad = 0;
-         hdr->streamid = 1;
-         hdr->length = length - 14;
-         hdr->data_pdu_type = data_pdu_type;
-         hdr->compress_type = 0;
-         hdr->compress_len = 0;
- }
- /* Parse a data header */
+                                 g_iconv_works = False;
- BOOL rdp_io_data_header(STREAM s, RDP_DATA_HEADER *hdr)
+                                 return rdp_in_unistr(s, string, uni_len);
- {
+                         }
-         BOOL res = True;
+                 }
-         res = res ? lsb_io_uint32(s, &hdr->shareid      ) : False;
+                 if (iconv(iconv_h, (ICONV_CONST char **) &pin, &ibl, &pout, &obl) == (size_t) - 1)
-         res = res ? prs_io_uint8 (s, &hdr->pad          ) : False;
+                 {
-         res = res ? prs_io_uint8 (s, &hdr->streamid     ) : False;
+                         iconv_close(iconv_h);
-         res = res ? lsb_io_uint16(s, &hdr->length       ) : False;
+                         iconv_h = (iconv_t) - 1;
-         res = res ? prs_io_uint8 (s, &hdr->data_pdu_type) : False;
+                         warning("rdp_in_unistr: iconv fail, errno %d\n", errno);
-         res = res ? prs_io_uint8 (s, &hdr->compress_type) : False;
-         res = res ? lsb_io_uint16(s, &hdr->compress_len ) : False;
-         return res;
+                         g_iconv_works = False;
- }
+                         return rdp_in_unistr(s, string, uni_len);
+                 }
- BOOL rdp_io_present(STREAM s, uint32 *present, uint8 flags, int size)
+                 /* we must update the location of the current STREAM for future reads of s->p */
- {
+                 s->p += uni_len;
-         uint8 bits;
-         int i;
-         if (flags & RDP_ORDER_SMALL)
+                 return pout - string;
-         {
-                 size--;
          }
+         else
-         if (flags & RDP_ORDER_TINY)
+ #endif
          {
-                 if (size < 2)
+                 int i = 0;
-                         return False;
-                 size -= 2;
+                 while (i < uni_len / 2)
-         }
+                 {
+                         in_uint8a(s, &string[i++], 1);
+                         in_uint8s(s, 1);
+                 }
-         *present = 0;
+                 return i - 1;
-         for (i = 0; i < size; i++)
-         {
-                 prs_io_uint8(s, &bits);
-                 *present |= bits << (i * 8);
          }
-         return True;
  }
- BOOL rdp_io_coord(STREAM s, uint16 *coord, BOOL delta)
- {
-         uint8 change;
-         BOOL res;
-         if (delta)
+ /* Parse a logon info packet */
-         {
+ static void
-                 res = prs_io_uint8(s, &change);
+ rdp_send_logon_info(uint32 flags, char *domain, char *user,
-                 *coord += (char)change;
+                     char *password, char *program, char *directory)
+ {
+         char *ipaddr = tcp_get_address();
+         int len_domain = 2 * strlen(domain);
+         int len_user = 2 * strlen(user);
+         int len_password = 2 * strlen(password);
+         int len_program = 2 * strlen(program);
+         int len_directory = 2 * strlen(directory);
+         int len_ip = 2 * strlen(ipaddr);
+         int len_dll = 2 * strlen("C:\\WINNT\\System32\\mstscax.dll");
+         int packetlen = 0;
+         uint32 sec_flags = g_encryption ? (SEC_LOGON_INFO | SEC_ENCRYPT) : SEC_LOGON_INFO;
+         STREAM s;
+         time_t t = time(NULL);
+         time_t tzone;
+         if (!g_use_rdp5 || 1 == g_server_rdp_version)
+         {
+                 DEBUG_RDP5(("Sending RDP4-style Logon packet\n"));
+                 s = sec_init(sec_flags, 18 + len_domain + len_user + len_password
+                              + len_program + len_directory + 10);
+                 out_uint32(s, 0);
+                 out_uint32_le(s, flags);
+                 out_uint16_le(s, len_domain);
+                 out_uint16_le(s, len_user);
+                 out_uint16_le(s, len_password);
+                 out_uint16_le(s, len_program);
+                 out_uint16_le(s, len_directory);
+                 rdp_out_unistr(s, domain, len_domain);
+                 rdp_out_unistr(s, user, len_user);
+                 rdp_out_unistr(s, password, len_password);
+                 rdp_out_unistr(s, program, len_program);
+                 rdp_out_unistr(s, directory, len_directory);
          }
          else
          {
-                 res = lsb_io_uint16(s, coord);
-         }
-         return res;
- }
- BOOL rdp_io_colour(STREAM s, uint8 *colour)
+                 flags |= RDP_LOGON_BLOB;
- {
+                 DEBUG_RDP5(("Sending RDP5-style Logon packet\n"));
-         BOOL res;
+                 packetlen = 4 + /* Unknown uint32 */
++     /* flags */
++     /* len_domain */
++     /* len_user */
+                         (flags & RDP_LOGON_AUTO ? 2 : 0) +      /* len_password */
+                         (flags & RDP_LOGON_BLOB ? 2 : 0) +      /* Length of BLOB */
++     /* len_program */
++     /* len_directory */
+                         (0 < len_domain ? len_domain : 2) +     /* domain */
+                         len_user + (flags & RDP_LOGON_AUTO ? len_password : 0) + 0 +    /* We have no 512 byte BLOB. Perhaps we must? */
+                         (flags & RDP_LOGON_BLOB && !(flags & RDP_LOGON_AUTO) ? 2 : 0) + /* After the BLOB is a unknown int16. If there is a BLOB, that is. */
+                         (0 < len_program ? len_program : 2) + (0 < len_directory ? len_directory : 2) + 2 +     /* Unknown (2) */
++     /* Client ip length */
+                         len_ip +        /* Client ip */
++     /* DLL string length */
+                         len_dll +       /* DLL string */
++     /* Unknown */
++     /* Unknown */
++    /* Time zone #0 */
++     /* Unknown */
++    /* Time zone #1 */
+;     /* Unknown */
+                 s = sec_init(sec_flags, packetlen);
+                 DEBUG_RDP5(("Called sec_init with packetlen %d\n", packetlen));
+                 out_uint32(s, 0);       /* Unknown */
+                 out_uint32_le(s, flags);
+                 out_uint16_le(s, len_domain);
+                 out_uint16_le(s, len_user);
+                 if (flags & RDP_LOGON_AUTO)
+                 {
+                         out_uint16_le(s, len_password);
-         res = prs_io_uint8(s, colour);
+                 }
-         s->offset += 2;
+                 if (flags & RDP_LOGON_BLOB && !(flags & RDP_LOGON_AUTO))
+                 {
+                         out_uint16_le(s, 0);
+                 }
+                 out_uint16_le(s, len_program);
+                 out_uint16_le(s, len_directory);
+                 if (0 < len_domain)
+                         rdp_out_unistr(s, domain, len_domain);
+                 else
+                         out_uint16_le(s, 0);
+                 rdp_out_unistr(s, user, len_user);
+                 if (flags & RDP_LOGON_AUTO)
+                 {
+                         rdp_out_unistr(s, password, len_password);
+                 }
+                 if (flags & RDP_LOGON_BLOB && !(flags & RDP_LOGON_AUTO))
+                 {
+                         out_uint16_le(s, 0);
+                 }
+                 if (0 < len_program)
+                 {
+                         rdp_out_unistr(s, program, len_program);
-         return res;
+                 }
- }
+                 else
+                 {
+                         out_uint16_le(s, 0);
+                 }
+                 if (0 < len_directory)
+                 {
+                         rdp_out_unistr(s, directory, len_directory);
+                 }
+                 else
+                 {
+                         out_uint16_le(s, 0);
+                 }
+                 out_uint16_le(s, 2);
+                 out_uint16_le(s, len_ip + 2);   /* Length of client ip */
+                 rdp_out_unistr(s, ipaddr, len_ip);
+                 out_uint16_le(s, len_dll + 2);
+                 rdp_out_unistr(s, "C:\\WINNT\\System32\\mstscax.dll", len_dll);
- BOOL rdp_io_colourmap(STREAM s, COLOURMAP *colours)
+                 tzone = (mktime(gmtime(&t)) - mktime(localtime(&t))) / 60;
- {
+                 out_uint32_le(s, tzone);
-         int datasize;
-         lsb_io_uint16(s, &colours->ncolours);
+                 rdp_out_unistr(s, "GTB, normaltid", 2 * strlen("GTB, normaltid"));
-         datasize = colours->ncolours * 3;
+                 out_uint8s(s, 62 - 2 * strlen("GTB, normaltid"));
-         if (datasize > sizeof(colours->colours))
+                 out_uint32_le(s, 0x0a0000);
-                 return False;
+                 out_uint32_le(s, 0x050000);
+                 out_uint32_le(s, 3);
+                 out_uint32_le(s, 0);
+                 out_uint32_le(s, 0);
-         memcpy(colours->colours, s->data + s->offset, datasize);
+                 rdp_out_unistr(s, "GTB, sommartid", 2 * strlen("GTB, sommartid"));
-         s->offset += datasize;
+                 out_uint8s(s, 62 - 2 * strlen("GTB, sommartid"));
-         return True;
- }
- BOOL rdp_io_bounds(STREAM s, BOUNDS *bounds)
+                 out_uint32_le(s, 0x30000);
- {
+                 out_uint32_le(s, 0x050000);
-         uint8 present;
+                 out_uint32_le(s, 2);
+                 out_uint32(s, 0);
+                 out_uint32_le(s, 0xffffffc4);
+                 out_uint32_le(s, 0xfffffffe);
+                 out_uint32_le(s, g_rdp5_performanceflags);
+                 out_uint32(s, 0);
-         prs_io_uint8(s, &present);
-         if (present & 1)
+         }
-                 rdp_io_coord(s, &bounds->left, False);
+         s_mark_end(s);
-         else if (present & 16)
+         sec_send(s, sec_flags);
-                 rdp_io_coord(s, &bounds->left, True);
-         if (present & 2)
-                 rdp_io_coord(s, &bounds->top, False);
-         else if (present & 32)
-                 rdp_io_coord(s, &bounds->top, True);
-         if (present & 4)
-                 rdp_io_coord(s, &bounds->right, False);
-         else if (present & 64)
-                 rdp_io_coord(s, &bounds->right, True);
-         if (present & 8)
-                 rdp_io_coord(s, &bounds->bottom, False);
-         else if (present & 128)
-                 rdp_io_coord(s, &bounds->bottom, True);
-         return True;
  }
- BOOL rdp_io_pen(STREAM s, PEN *pen, uint32 present)
+ /* Send a control PDU */
+ static void
+ rdp_send_control(uint16 action)
  {
-         BOOL res = True;
+         STREAM s;
-         if (present & 1)
-                 res = res ? prs_io_uint8(s, &pen->style) : False;
-         if (present & 2)
+         s = rdp_init_data(8);
-                 res = res ? prs_io_uint8(s, &pen->width) : False;
-         if (present & 4)
+         out_uint16_le(s, action);
-                 res = res ? rdp_io_colour(s, &pen->colour) : False;
+         out_uint16(s, 0);       /* userid */
+         out_uint32(s, 0);       /* control id */
-         return res;
+         s_mark_end(s);
+         rdp_send_data(s, RDP_DATA_PDU_CONTROL);
  }
- BOOL rdp_io_brush(STREAM s, BRUSH *brush, uint32 present)
+ /* Send a synchronisation PDU */
+ static void
+ rdp_send_synchronise(void)
  {
-         BOOL res = True;
+         STREAM s;
-         if (present & 1)
-                 res = res ? prs_io_uint8(s, &brush->xorigin) : False;
-         if (present & 2)
-                 res = res ? prs_io_uint8(s, &brush->yorigin) : False;
-         if (present & 4)
-                 res = res ? prs_io_uint8(s, &brush->style) : False;
-         if (present & 8)
+         s = rdp_init_data(4);
-                 res = res ? prs_io_uint8(s, &brush->pattern[0]) : False;
-         if (present & 16)
+         out_uint16_le(s, 1);    /* type */
-                 res = res ? prs_io_uint8s(s, &brush->pattern[1], 7) : False;
+         out_uint16_le(s, 1002);
-         return res;
+         s_mark_end(s);
+         rdp_send_data(s, RDP_DATA_PDU_SYNCHRONISE);
  }
- /* Construct a confirm/demand active PDU */
+ /* Send a single input event */
- void rdp_make_active_pdu(RDP_ACTIVE_PDU *pdu, uint32 shareid, uint16 userid,
+ void
-                                 int width, int height)
+ rdp_send_input(uint32 time, uint16 message_type, uint16 device_flags, uint16 param1, uint16 param2)
  {
-         memset(pdu, 0, sizeof(*pdu));
+         STREAM s;
-         pdu->shareid = shareid;
-         pdu->userid  = 1002;
-         pdu->source_len = sizeof(RDP_SOURCE);
-         memcpy(pdu->source, RDP_SOURCE, sizeof(RDP_SOURCE));
-         pdu->caps_len = RDP_CAPLEN_GENERAL + RDP_CAPLEN_BITMAP + RDP_CAPLEN_ORDER
+         s = rdp_init_data(16);
-                 + RDP_CAPLEN_BMPCACHE + RDP_CAPLEN_COLCACHE + RDP_CAPLEN_ACTIVATE
-                 + RDP_CAPLEN_CONTROL + RDP_CAPLEN_POINTER + RDP_CAPLEN_SHARE
-                 + RDP_CAPLEN_UNKNOWN;
-         pdu->num_caps = 0xD;
-         rdp_make_general_caps (&pdu->general_caps );
+         out_uint16_le(s, 1);    /* number of events */
-         rdp_make_bitmap_caps  (&pdu->bitmap_caps, width, height);
+         out_uint16(s, 0);       /* pad */
-         rdp_make_order_caps   (&pdu->order_caps   );
-         rdp_make_bmpcache_caps(&pdu->bmpcache_caps);
+         out_uint32_le(s, time);
-         rdp_make_control_caps (&pdu->control_caps );
+         out_uint16_le(s, message_type);
-         rdp_make_activate_caps(&pdu->activate_caps);
+         out_uint16_le(s, device_flags);
-         rdp_make_pointer_caps (&pdu->pointer_caps );
+         out_uint16_le(s, param1);
-         rdp_make_share_caps   (&pdu->share_caps, userid);
+         out_uint16_le(s, param2);
-         rdp_make_colcache_caps(&pdu->colcache_caps);
+         s_mark_end(s);
+         rdp_send_data(s, RDP_DATA_PDU_INPUT);
  }
- /* Parse a confirm/demand active PDU */
+ /* Send a client window information PDU */
- BOOL rdp_io_active_pdu(STREAM s, RDP_ACTIVE_PDU *pdu, int pdutype)
+ void
+ rdp_send_client_window_status(int status)
  {
-         uint16 capset;
+         STREAM s;
-         uint16 length;
+         static int current_status = 1;
-         BOOL res;
-         int i;
-         res = lsb_io_uint32(s, &pdu->shareid);
+         if (current_status == status)
+                 return;
-         if (pdutype == RDP_PDU_CONFIRM_ACTIVE)
+         s = rdp_init_data(12);
-                 res = res ? lsb_io_uint16(s, &pdu->userid    ) : False;
-         res = res ? lsb_io_uint16(s, &pdu->source_len) : False;
+         out_uint32_le(s, status);
-         res = res ? lsb_io_uint16(s, &pdu->caps_len  ) : False;
-         if (pdu->source_len > 48)
+         switch (status)
          {
-                 ERROR("RDP source descriptor too long\n");
+                 case 0: /* shut the server up */
-                 return False;
+                         break;
+                 case 1: /* receive data again */
+                         out_uint32_le(s, 0);    /* unknown */
+                         out_uint16_le(s, g_width);
+                         out_uint16_le(s, g_height);
+                         break;
          }
-         res = res ? prs_io_uint8s(s,  pdu->source, pdu->source_len) : False;
+         s_mark_end(s);
-         res = res ? lsb_io_uint16(s, &pdu->num_caps  ) : False;
+         rdp_send_data(s, RDP_DATA_PDU_CLIENT_WINDOW_STATUS);
-         res = res ? lsb_io_uint16(s, &pdu->pad       ) : False;
+         current_status = status;
+ }
-         if (s->marshall)
-         {
+ /* Send persistent bitmap cache enumeration PDU's */
-                 capset = RDP_CAPSET_GENERAL;
+ static void
-                 res = res ? lsb_io_uint16(s, &capset) : False;
+ rdp_enum_bmpcache2(void)
-                 res = res ? rdp_io_general_caps(s, &pdu->general_caps) : False;
+ {
+         STREAM s;
-                 capset = RDP_CAPSET_BITMAP;
+         HASH_KEY keylist[BMPCACHE2_NUM_PSTCELLS];
-                 res = res ? lsb_io_uint16(s, &capset) : False;
+         uint32 num_keys, offset, count, flags;
-                 res = res ? rdp_io_bitmap_caps (s, &pdu->bitmap_caps ) : False;
+         offset = 0;
-                 capset = RDP_CAPSET_ORDER;
+         num_keys = pstcache_enumerate(2, keylist);
-                 res = res ? lsb_io_uint16(s, &capset) : False;
-                 res = res ? rdp_io_order_caps  (s, &pdu->order_caps  ) : False;
+         while (offset < num_keys)
+         {
-                 capset = RDP_CAPSET_BMPCACHE;
+                 count = MIN(num_keys - offset, 169);
-                 res = res ? lsb_io_uint16(s, &capset) : False;
-                 res = res ? rdp_io_bmpcache_caps(s, &pdu->bmpcache_caps) : False;
+                 s = rdp_init_data(24 + count * sizeof(HASH_KEY));
-                 capset = RDP_CAPSET_COLCACHE;
+                 flags = 0;
-                 res = res ? lsb_io_uint16(s, &capset) : False;
+                 if (offset == 0)
-                 res = res ? rdp_io_colcache_caps(s, &pdu->colcache_caps) : False;
+                         flags |= PDU_FLAG_FIRST;
+                 if (num_keys - offset <= 169)
-                 capset = RDP_CAPSET_ACTIVATE;
+                         flags |= PDU_FLAG_LAST;
-                 res = res ? lsb_io_uint16(s, &capset) : False;
-                 res = res ? rdp_io_activate_caps(s, &pdu->activate_caps) : False;
+                 /* header */
+                 out_uint32_le(s, 0);
-                 capset = RDP_CAPSET_CONTROL;
+                 out_uint16_le(s, count);
-                 res = res ? lsb_io_uint16(s, &capset) : False;
+                 out_uint16_le(s, 0);
-                 res = res ? rdp_io_control_caps(s, &pdu->control_caps) : False;
+                 out_uint16_le(s, 0);
+                 out_uint16_le(s, 0);
-                 capset = RDP_CAPSET_POINTER;
+                 out_uint16_le(s, 0);
-                 res = res ? lsb_io_uint16(s, &capset) : False;
+                 out_uint16_le(s, num_keys);
-                 res = res ? rdp_io_pointer_caps(s, &pdu->pointer_caps) : False;
+                 out_uint32_le(s, 0);
+                 out_uint32_le(s, flags);
-                 capset = RDP_CAPSET_SHARE;
-                 res = res ? lsb_io_uint16(s, &capset) : False;
+                 /* list */
-                 res = res ? rdp_io_share_caps  (s, &pdu->share_caps  ) : False;
+                 out_uint8a(s, keylist[offset], count * sizeof(HASH_KEY));
-                 capset = RDP_CAPSET_UNKNOWN;
+                 s_mark_end(s);
-                 res = res ? lsb_io_uint16(s, &capset) : False;
+                 rdp_send_data(s, 0x2b);
-                 res = res ? rdp_io_unknown_caps(s, NULL) : False;
+                 offset += 169;
+         }
+ }
+ /* Send an (empty) font information PDU */
+ static void
+ rdp_send_fonts(uint16 seq)
+ {
+         STREAM s;
+         s = rdp_init_data(8);
+         out_uint16(s, 0);       /* number of fonts */
+         out_uint16_le(s, 0);    /* pad? */
+         out_uint16_le(s, seq);  /* unknown */
+         out_uint16_le(s, 0x32); /* entry size */
+         s_mark_end(s);
+         rdp_send_data(s, RDP_DATA_PDU_FONT2);
+ }
+ /* Output general capability set */
+ static void
+ rdp_out_general_caps(STREAM s)
+ {
+         out_uint16_le(s, RDP_CAPSET_GENERAL);
+         out_uint16_le(s, RDP_CAPLEN_GENERAL);
+         out_uint16_le(s, 1);    /* OS major type */
+         out_uint16_le(s, 3);    /* OS minor type */
+         out_uint16_le(s, 0x200);        /* Protocol version */
+         out_uint16(s, 0);       /* Pad */
+         out_uint16(s, 0);       /* Compression types */
+         out_uint16_le(s, g_use_rdp5 ? 0x40d : 0);
+         /* Pad, according to T.128. 0x40d seems to
+            trigger
+            the server to start sending RDP5 packets.
+            However, the value is 0x1d04 with W2KTSK and
+            NT4MS. Hmm.. Anyway, thankyou, Microsoft,
+            for sending such information in a padding
+            field.. */
+         out_uint16(s, 0);       /* Update capability */
+         out_uint16(s, 0);       /* Remote unshare capability */
+         out_uint16(s, 0);       /* Compression level */
+         out_uint16(s, 0);       /* Pad */
+ }
+ /* Output bitmap capability set */
+ static void
+ rdp_out_bitmap_caps(STREAM s)
+ {
+         out_uint16_le(s, RDP_CAPSET_BITMAP);
+         out_uint16_le(s, RDP_CAPLEN_BITMAP);
+         out_uint16_le(s, g_server_depth);       /* Preferred colour depth */
+         out_uint16_le(s, 1);    /* Receive 1 BPP */
+         out_uint16_le(s, 1);    /* Receive 4 BPP */
+         out_uint16_le(s, 1);    /* Receive 8 BPP */
+         out_uint16_le(s, 800);  /* Desktop width */
+         out_uint16_le(s, 600);  /* Desktop height */
+         out_uint16(s, 0);       /* Pad */
+         out_uint16(s, 1);       /* Allow resize */
+         out_uint16_le(s, g_bitmap_compression ? 1 : 0); /* Support compression */
+         out_uint16(s, 0);       /* Unknown */
+         out_uint16_le(s, 1);    /* Unknown */
+         out_uint16(s, 0);       /* Pad */
+ }
+ /* Output order capability set */
+ static void
+ rdp_out_order_caps(STREAM s)
+ {
+         uint8 order_caps[32];
+         memset(order_caps, 0, 32);
+         order_caps[0] = 1;      /* dest blt */
+         order_caps[1] = 1;      /* pat blt */
+         order_caps[2] = 1;      /* screen blt */
+         order_caps[3] = (g_bitmap_cache ? 1 : 0);       /* memblt */
+         order_caps[4] = 0;      /* triblt */
+         order_caps[8] = 1;      /* line */
+         order_caps[9] = 1;      /* line */
+         order_caps[10] = 1;     /* rect */
+         order_caps[11] = (g_desktop_save ? 1 : 0);      /* desksave */
+         order_caps[13] = 1;     /* memblt */
+         order_caps[14] = 1;     /* triblt */
+         order_caps[20] = (g_polygon_ellipse_orders ? 1 : 0);    /* polygon */
+         order_caps[21] = (g_polygon_ellipse_orders ? 1 : 0);    /* polygon2 */
+         order_caps[22] = 1;     /* polyline */
+         order_caps[25] = (g_polygon_ellipse_orders ? 1 : 0);    /* ellipse */
+         order_caps[26] = (g_polygon_ellipse_orders ? 1 : 0);    /* ellipse2 */
+         order_caps[27] = 1;     /* text2 */
+         out_uint16_le(s, RDP_CAPSET_ORDER);
+         out_uint16_le(s, RDP_CAPLEN_ORDER);
+         out_uint8s(s, 20);      /* Terminal desc, pad */
+         out_uint16_le(s, 1);    /* Cache X granularity */
+         out_uint16_le(s, 20);   /* Cache Y granularity */
+         out_uint16(s, 0);       /* Pad */
+         out_uint16_le(s, 1);    /* Max order level */
+         out_uint16_le(s, 0x147);        /* Number of fonts */
+         out_uint16_le(s, 0x2a); /* Capability flags */
+         out_uint8p(s, order_caps, 32);  /* Orders supported */
+         out_uint16_le(s, 0x6a1);        /* Text capability flags */
+         out_uint8s(s, 6);       /* Pad */
+         out_uint32_le(s, g_desktop_save == False ? 0 : 0x38400);        /* Desktop cache size */
+         out_uint32(s, 0);       /* Unknown */
+         out_uint32_le(s, 0x4e4);        /* Unknown */
+ }
+ /* Output bitmap cache capability set */
+ static void
+ rdp_out_bmpcache_caps(STREAM s)
+ {
+         int Bpp;
+         out_uint16_le(s, RDP_CAPSET_BMPCACHE);
+         out_uint16_le(s, RDP_CAPLEN_BMPCACHE);
+         Bpp = (g_server_depth + 7) / 8; /* bytes per pixel */
+         out_uint8s(s, 24);      /* unused */
+         out_uint16_le(s, 0x258);        /* entries */
+         out_uint16_le(s, 0x100 * Bpp);  /* max cell size */
+         out_uint16_le(s, 0x12c);        /* entries */
+         out_uint16_le(s, 0x400 * Bpp);  /* max cell size */
+         out_uint16_le(s, 0x106);        /* entries */
+         out_uint16_le(s, 0x1000 * Bpp); /* max cell size */
+ }
+ /* Output bitmap cache v2 capability set */
+ static void
+ rdp_out_bmpcache2_caps(STREAM s)
+ {
+         out_uint16_le(s, RDP_CAPSET_BMPCACHE2);
+         out_uint16_le(s, RDP_CAPLEN_BMPCACHE2);
+         out_uint16_le(s, g_bitmap_cache_persist_enable ? 2 : 0);        /* version */
+         out_uint16_be(s, 3);    /* number of caches in this set */
+         /* max cell size for cache 0 is 16x16, 1 = 32x32, 2 = 64x64, etc */
+         out_uint32_le(s, BMPCACHE2_C0_CELLS);
+         out_uint32_le(s, BMPCACHE2_C1_CELLS);
+         if (pstcache_init(2))
+         {
+                 out_uint32_le(s, BMPCACHE2_NUM_PSTCELLS | BMPCACHE2_FLAG_PERSIST);
          }
          else
          {
-                 for (i = 0; i < pdu->num_caps; i++)
+                 out_uint32_le(s, BMPCACHE2_C2_CELLS);
-                 {
-                         if (!res)
-                                 return False;
-                         if (!lsb_io_uint16(s, &capset))
-                                 return False;
-                         switch (capset)
-                         {
-                         case RDP_CAPSET_GENERAL:
-                                 res = rdp_io_general_caps (s, &pdu->general_caps );
-                                 break;
-                         case RDP_CAPSET_BITMAP:
-                                 res = rdp_io_bitmap_caps  (s, &pdu->bitmap_caps  );
-                                 break;
-                         case RDP_CAPSET_ORDER:
-                                 res = rdp_io_order_caps   (s, &pdu->order_caps   );
-                                 break;
-                         case RDP_CAPSET_BMPCACHE:
-                                 res = rdp_io_bmpcache_caps(s, &pdu->bmpcache_caps);
-                                 break;
-                         case RDP_CAPSET_CONTROL:
-                                 res = rdp_io_control_caps (s, &pdu->control_caps );
-                                 break;
-                         case RDP_CAPSET_ACTIVATE:
-                                 res = rdp_io_activate_caps(s, &pdu->activate_caps);
-                                 break;
-                         case RDP_CAPSET_POINTER:
-                                 res = rdp_io_pointer_caps (s, &pdu->pointer_caps );
-                                 break;
-                         case RDP_CAPSET_SHARE:
-                                 res = rdp_io_share_caps   (s, &pdu->share_caps   );
-                                 break;
-                         case RDP_CAPSET_COLCACHE:
-                                 res = rdp_io_colcache_caps(s, &pdu->colcache_caps);
-                                 break;
-                         default:
-                                 NOTIMP("capset 0x%x\n", capset);
-                                 if (!lsb_io_uint16(s, &length))
-                                         return False;
-                                 s->offset += (length - 4);
-                         }
-                 }
          }
+         out_uint8s(s, 20);      /* other bitmap caches not used */
-         return res;
  }
- /* Construct a control PDU */
+ /* Output control capability set */
- void rdp_make_control_pdu(RDP_CONTROL_PDU *pdu, uint16 action)
+ static void
+ rdp_out_control_caps(STREAM s)
  {
-         pdu->action = action;
+         out_uint16_le(s, RDP_CAPSET_CONTROL);
-         pdu->userid = 0;
+         out_uint16_le(s, RDP_CAPLEN_CONTROL);
-         pdu->controlid = 0;
- }
- /* Parse a control PDU */
+         out_uint16(s, 0);       /* Control capabilities */
- BOOL rdp_io_control_pdu(STREAM s, RDP_CONTROL_PDU *pdu)
+         out_uint16(s, 0);       /* Remote detach */
- {
+         out_uint16_le(s, 2);    /* Control interest */
-         BOOL res = True;
+         out_uint16_le(s, 2);    /* Detach interest */
-         res = res ? lsb_io_uint16(s, &pdu->action   ) : False;
-         res = res ? lsb_io_uint16(s, &pdu->userid   ) : False;
-         res = res ? lsb_io_uint32(s, &pdu->controlid) : False;
-         return res;
- }
- /* Construct a synchronisation PDU */
- void rdp_make_synchronise_pdu(RDP_SYNCHRONISE_PDU *pdu, uint16 userid)
- {
-         pdu->type = 1;
-         pdu->userid = userid;
  }
- /* Parse a synchronisation PDU */
+ /* Output activation capability set */
- BOOL rdp_io_synchronise_pdu(STREAM s, RDP_SYNCHRONISE_PDU *pdu)
+ static void
+ rdp_out_activate_caps(STREAM s)
  {
-         BOOL res = True;
+         out_uint16_le(s, RDP_CAPSET_ACTIVATE);
+         out_uint16_le(s, RDP_CAPLEN_ACTIVATE);
-         res = res ? lsb_io_uint16(s, &pdu->type  ) : False;
+         out_uint16(s, 0);       /* Help key */
-         res = res ? lsb_io_uint16(s, &pdu->userid) : False;
+         out_uint16(s, 0);       /* Help index key */
+         out_uint16(s, 0);       /* Extended help key */
-         return res;
+         out_uint16(s, 0);       /* Window activate */
  }
- /* Parse a single input event */
+ /* Output pointer capability set */
- BOOL rdp_io_input_event(STREAM s, RDP_INPUT_EVENT *evt)
+ static void
+ rdp_out_pointer_caps(STREAM s)
  {
-         BOOL res = True;
+         out_uint16_le(s, RDP_CAPSET_POINTER);
+         out_uint16_le(s, RDP_CAPLEN_POINTER);
-         res = res ? lsb_io_uint32(s, &evt->event_time)   : False;
-         res = res ? lsb_io_uint16(s, &evt->message_type) : False;
-         res = res ? lsb_io_uint16(s, &evt->device_flags) : False;
-         if (!res)
-                 return False;
-         switch (evt->message_type)
-         {
-         case RDP_INPUT_CODEPOINT:
-         case RDP_INPUT_VIRTKEY:
-                 res = res ? lsb_io_uint16(s, &evt->param1) : False;
-                 break;
-         case RDP_INPUT_SYNCHRONIZE:
-         case RDP_INPUT_SCANCODE:
-         case RDP_INPUT_MOUSE:
-                 res = res ? lsb_io_uint16(s, &evt->param1) : False;
-                 res = res ? lsb_io_uint16(s, &evt->param2) : False;
-                 break;
-         default:
-                 NOTIMP("input type %d\n", evt->message_type);
-                 return False;
-         }
-         return res;
+         out_uint16(s, 0);       /* Colour pointer */
+         out_uint16_le(s, 20);   /* Cache size */
  }
- /* Construct an input PDU */
+ /* Output share capability set */
- void rdp_make_input_pdu(RDP_INPUT_PDU *pdu, uint16 message_type,
+ static void
-                         uint16 device_flags, uint16 param1, uint16 param2)
+ rdp_out_share_caps(STREAM s)
  {
-         uint32 now = time(NULL);
+         out_uint16_le(s, RDP_CAPSET_SHARE);
+         out_uint16_le(s, RDP_CAPLEN_SHARE);
-         pdu->num_events = 1;
+         out_uint16(s, 0);       /* userid */
-         pdu->pad = 0;
+         out_uint16(s, 0);       /* pad */
-         pdu->event[0].event_time = now;
-         pdu->event[0].message_type = message_type;
-         pdu->event[0].device_flags = device_flags;
-         pdu->event[0].param1 = param1;
-         pdu->event[0].param2 = param2;
  }
- /* Parse an input PDU */
+ /* Output colour cache capability set */
- BOOL rdp_io_input_pdu(STREAM s, RDP_INPUT_PDU *pdu)
+ static void
+ rdp_out_colcache_caps(STREAM s)
  {
-         BOOL res = True;
+         out_uint16_le(s, RDP_CAPSET_COLCACHE);
-         int i;
+         out_uint16_le(s, RDP_CAPLEN_COLCACHE);
-         res = res ? lsb_io_uint16(s, &pdu->num_events) : False;
+         out_uint16_le(s, 6);    /* cache size */
-         res = res ? lsb_io_uint16(s, &pdu->pad       ) : False;
+         out_uint16(s, 0);       /* pad */
+ }
-         if (pdu->num_events > RDP_MAX_EVENTS)
+ static uint8 caps_0x0d[] = {
-         {
+x01, 0x00, 0x00, 0x00, 0x09, 0x04, 0x00, 0x00,
-                 ERROR("Too many events in one PDU\n");
+x04, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
-                 return False;
+x0C, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
-         }
+x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+x00, 0x00, 0x00, 0x00
+ };
-         for (i = 0; i < pdu->num_events; i++)
+ static uint8 caps_0x0c[] = { 0x01, 0x00, 0x00, 0x00 };
-         {
-                 res = res ? rdp_io_input_event(s, &pdu->event[i]) : False;
-         }
-         return res;
+ static uint8 caps_0x0e[] = { 0x01, 0x00, 0x00, 0x00 };
- }
- /* Construct a font information PDU */
+ static uint8 caps_0x10[] = {
- void rdp_make_font_pdu(RDP_FONT_PDU *pdu, uint16 seqno)
+xFE, 0x00, 0x04, 0x00, 0xFE, 0x00, 0x04, 0x00,
- {
+xFE, 0x00, 0x08, 0x00, 0xFE, 0x00, 0x08, 0x00,
-         pdu->num_fonts = 0;
+xFE, 0x00, 0x10, 0x00, 0xFE, 0x00, 0x20, 0x00,
-         pdu->unknown1 = 0x3e;
+xFE, 0x00, 0x40, 0x00, 0xFE, 0x00, 0x80, 0x00,
-         pdu->unknown2 = seqno;
+xFE, 0x00, 0x00, 0x01, 0x40, 0x00, 0x00, 0x08,
-         pdu->entry_size = RDP_FONT_INFO_SIZE;
+x00, 0x01, 0x00, 0x01, 0x02, 0x00, 0x00, 0x00
- }
+ };
- /* Parse a font information structure */
+ /* Output unknown capability sets */
- BOOL rdp_io_font_info(STREAM s, RDP_FONT_INFO *font)
+ static void
+ rdp_out_unknown_caps(STREAM s, uint16 id, uint16 length, uint8 * caps)
  {
-         BOOL res = True;
+         out_uint16_le(s, id);
+         out_uint16_le(s, length);
-         res = res ? prs_io_uint8s(s,  font->name, 32 ) : False;
-         res = res ? lsb_io_uint16(s, &font->flags    ) : False;
-         res = res ? lsb_io_uint16(s, &font->width    ) : False;
-         res = res ? lsb_io_uint16(s, &font->height   ) : False;
-         res = res ? lsb_io_uint16(s, &font->xaspect  ) : False;
-         res = res ? lsb_io_uint16(s, &font->yaspect  ) : False;
-         res = res ? lsb_io_uint32(s, &font->signature) : False;
-         res = res ? lsb_io_uint16(s, &font->codepage ) : False;
-         res = res ? lsb_io_uint16(s, &font->ascent   ) : False;
-         return res;
+         out_uint8p(s, caps, length - 4);
  }
- /* Parse a font information PDU */
+ #define RDP5_FLAG 0x0030
- BOOL rdp_io_font_pdu(STREAM s, RDP_FONT_PDU *pdu)
+ /* Send a confirm active PDU */
+ static void
+ rdp_send_confirm_active(void)
  {
-         BOOL res = True;
+         STREAM s;
-         int i;
+         uint32 sec_flags = g_encryption ? (RDP5_FLAG | SEC_ENCRYPT) : RDP5_FLAG;
+         uint16 caplen =
+                 RDP_CAPLEN_GENERAL + RDP_CAPLEN_BITMAP + RDP_CAPLEN_ORDER +
+                 RDP_CAPLEN_BMPCACHE + RDP_CAPLEN_COLCACHE +
+                 RDP_CAPLEN_ACTIVATE + RDP_CAPLEN_CONTROL +
+                 RDP_CAPLEN_POINTER + RDP_CAPLEN_SHARE +
+x58 + 0x08 + 0x08 + 0x34 /* unknown caps */  +
+/* w2k fix, why? */ ;
-         res = res ? lsb_io_uint16(s, &pdu->num_fonts ) : False;
+         s = sec_init(sec_flags, 6 + 14 + caplen + sizeof(RDP_SOURCE));
-         res = res ? lsb_io_uint16(s, &pdu->unknown1  ) : False;
-         res = res ? lsb_io_uint16(s, &pdu->unknown2  ) : False;
-         res = res ? lsb_io_uint16(s, &pdu->entry_size) : False;
-         if (pdu->num_fonts > RDP_MAX_FONTS)
+         out_uint16_le(s, 2 + 14 + caplen + sizeof(RDP_SOURCE));
-         {
+         out_uint16_le(s, (RDP_PDU_CONFIRM_ACTIVE | 0x10));      /* Version 1 */
-                 ERROR("Too many fonts in one PDU\n");
+         out_uint16_le(s, (g_mcs_userid + 1001));
-                 return False;
-         }
-         for (i = 0; i < pdu->num_fonts; i++)
+         out_uint32_le(s, g_rdp_shareid);
-         {
+         out_uint16_le(s, 0x3ea);        /* userid */
-                 res = res ? rdp_io_font_info(s, &pdu->font[i]) : False;
+         out_uint16_le(s, sizeof(RDP_SOURCE));
-         }
+         out_uint16_le(s, caplen);
-         return res;
+         out_uint8p(s, RDP_SOURCE, sizeof(RDP_SOURCE));
- }
+         out_uint16_le(s, 0xd);  /* num_caps */
+         out_uint8s(s, 2);       /* pad */
- /* Parse a pointer PDU */
+         rdp_out_general_caps(s);
- BOOL rdp_io_pointer_pdu(STREAM s, RDP_POINTER_PDU *ptr)
+         rdp_out_bitmap_caps(s);
- {
+         rdp_out_order_caps(s);
-         BOOL res = True;
+         g_use_rdp5 ? rdp_out_bmpcache2_caps(s) : rdp_out_bmpcache_caps(s);
+         rdp_out_colcache_caps(s);
+         rdp_out_activate_caps(s);
+         rdp_out_control_caps(s);
+         rdp_out_pointer_caps(s);
+         rdp_out_share_caps(s);
-         res = res ? lsb_io_uint16(s, &ptr->message) : False;
+         rdp_out_unknown_caps(s, 0x0d, 0x58, caps_0x0d); /* international? */
-         res = res ? lsb_io_uint16(s, &ptr->pad    ) : False;
+         rdp_out_unknown_caps(s, 0x0c, 0x08, caps_0x0c);
+         rdp_out_unknown_caps(s, 0x0e, 0x08, caps_0x0e);
+         rdp_out_unknown_caps(s, 0x10, 0x34, caps_0x10); /* glyph cache? */
-         switch (ptr->message)
+         s_mark_end(s);
-         {
+         sec_send(s, sec_flags);
-                 case RDP_POINTER_MOVE:
-                         res = res ? lsb_io_uint16(s, &ptr->x      ) : False;
-                         res = res ? lsb_io_uint16(s, &ptr->y      ) : False;
-                         break;
-         }
-         return res;
  }
- /* Parse an update PDU */
+ /* Process a general capability set */
- BOOL rdp_io_update_pdu(STREAM s, RDP_UPDATE_PDU *pdu)
+ static void
+ rdp_process_general_caps(STREAM s)
  {
-         BOOL res = True;
+         uint16 pad2octetsB;     /* rdp5 flags? */
-         res = res ? lsb_io_uint16(s, &pdu->update_type) : False;
+         in_uint8s(s, 10);
-         res = res ? lsb_io_uint16(s, &pdu->pad        ) : False;
+         in_uint16_le(s, pad2octetsB);
-         return res;
+         if (!pad2octetsB)
+                 g_use_rdp5 = False;
  }
+ /* Process a bitmap capability set */
- /* PRIMARY ORDERS */
+ static void
+ rdp_process_bitmap_caps(STREAM s)
- /* Parse an destination blt order */
- BOOL rdp_io_destblt_order(STREAM s, DESTBLT_ORDER *os, uint32 present, BOOL delta)
  {
-         if (present & 0x01)
+         uint16 width, height, depth;
-                 rdp_io_coord(s, &os->x, delta);
-         if (present & 0x02)
-                 rdp_io_coord(s, &os->y, delta);
-         if (present & 0x04)
+         in_uint16_le(s, depth);
-                 rdp_io_coord(s, &os->cx, delta);
+         in_uint8s(s, 6);
-         if (present & 0x08)
+         in_uint16_le(s, width);
-                 rdp_io_coord(s, &os->cy, delta);
+         in_uint16_le(s, height);
-         if (present & 0x10)
+         DEBUG(("setting desktop size and depth to: %dx%dx%d\n", width, height, depth));
-                 prs_io_uint8(s, &os->opcode);
-         return PRS_ERROR(s);
+         /*
+          * The server may limit depth and change the size of the desktop (for
+          * example when shadowing another session).
+          */
+         if (g_server_depth != depth)
+         {
+                 warning("Remote desktop does not support colour depth %d; falling back to %d\n",
+                         g_server_depth, depth);
+                 g_server_depth = depth;
+         }
+         if (g_width != width || g_height != height)
+         {
+                 warning("Remote desktop changed from %dx%d to %dx%d.\n", g_width, g_height,
+                         width, height);
+                 g_width = width;
+                 g_height = height;
+                 ui_resize_window();
+         }
  }
- /* Parse an pattern blt order */
+ /* Process server capabilities */
- BOOL rdp_io_patblt_order(STREAM s, PATBLT_ORDER *os, uint32 present, BOOL delta)
+ static void
+ rdp_process_server_caps(STREAM s, uint16 length)
  {
-         if (present & 0x0001)
+         int n;
-                 rdp_io_coord(s, &os->x, delta);
+         uint8 *next, *start;
+         uint16 ncapsets, capset_type, capset_length;
-         if (present & 0x0002)
+         start = s->p;
-                 rdp_io_coord(s, &os->y, delta);
-         if (present & 0x0004)
+         in_uint16_le(s, ncapsets);
-                 rdp_io_coord(s, &os->cx, delta);
+         in_uint8s(s, 2);        /* pad */
-         if (present & 0x0008)
+         for (n = 0; n < ncapsets; n++)
-                 rdp_io_coord(s, &os->cy, delta);
+         {
+                 if (s->p > start + length)
+                         return;
-         if (present & 0x0010)
+                 in_uint16_le(s, capset_type);
-                 prs_io_uint8(s, &os->opcode);
+                 in_uint16_le(s, capset_length);
-         if (present & 0x0020)
+                 next = s->p + capset_length - 4;
-                 rdp_io_colour(s, &os->bgcolour);
-         if (present & 0x0040)
+                 switch (capset_type)
-                 rdp_io_colour(s, &os->fgcolour);
+                 {
+                         case RDP_CAPSET_GENERAL:
+                                 rdp_process_general_caps(s);
+                                 break;
-         rdp_io_brush(s, &os->brush, present >> 7);
+                         case RDP_CAPSET_BITMAP:
+                                 rdp_process_bitmap_caps(s);
+                                 break;
+                 }
-         return PRS_ERROR(s);
+                 s->p = next;
+         }
  }
- /* Parse an screen blt order */
+ /* Respond to a demand active PDU */
- BOOL rdp_io_screenblt_order(STREAM s, SCREENBLT_ORDER *os, uint32 present, BOOL delta)
+ static void
+ process_demand_active(STREAM s)
  {
-         if (present & 0x0001)
+         uint8 type;
-                 rdp_io_coord(s, &os->x, delta);
+         uint16 len_src_descriptor, len_combined_caps;
-         if (present & 0x0002)
-                 rdp_io_coord(s, &os->y, delta);
-         if (present & 0x0004)
+         in_uint32_le(s, g_rdp_shareid);
-                 rdp_io_coord(s, &os->cx, delta);
+         in_uint16_le(s, len_src_descriptor);
+         in_uint16_le(s, len_combined_caps);
+         in_uint8s(s, len_src_descriptor);
-         if (present & 0x0008)
+         DEBUG(("DEMAND_ACTIVE(id=0x%x)\n", g_rdp_shareid));
-                 rdp_io_coord(s, &os->cy, delta);
+         rdp_process_server_caps(s, len_combined_caps);
-         if (present & 0x0010)
+         rdp_send_confirm_active();
-                 prs_io_uint8(s, &os->opcode);
+         rdp_send_synchronise();
+         rdp_send_control(RDP_CTL_COOPERATE);
+         rdp_send_control(RDP_CTL_REQUEST_CONTROL);
+         rdp_recv(&type);        /* RDP_PDU_SYNCHRONIZE */
+         rdp_recv(&type);        /* RDP_CTL_COOPERATE */
+         rdp_recv(&type);        /* RDP_CTL_GRANT_CONTROL */
+         rdp_send_input(0, RDP_INPUT_SYNCHRONIZE, 0,
+                        g_numlock_sync ? ui_get_numlock_state(read_keyboard_state()) : 0, 0);
-         if (present & 0x0020)
+         if (g_use_rdp5)
-                 rdp_io_coord(s, &os->srcx, delta);
+         {
+                 rdp_enum_bmpcache2();
-         if (present & 0x0040)
+                 rdp_send_fonts(3);
-                 rdp_io_coord(s, &os->srcy, delta);
+         }
+         else
+         {
+                 rdp_send_fonts(1);
+                 rdp_send_fonts(2);
+         }
-         return PRS_ERROR(s);
+         rdp_recv(&type);        /* RDP_PDU_UNKNOWN 0x28 (Fonts?) */
+         reset_order_state();
  }
- /* Parse a line order */
+ /* Process a colour pointer PDU */
- BOOL rdp_io_line_order(STREAM s, LINE_ORDER *os, uint32 present, BOOL delta)
+ void
+ process_colour_pointer_pdu(STREAM s)
  {
-         if (present & 0x0001)
+         uint16 x, y, width, height, cache_idx, masklen, datalen;
-                 lsb_io_uint16(s, &os->mixmode);
+         uint8 *mask, *data;
+         HCURSOR cursor;
-         if (present & 0x0002)
+         in_uint16_le(s, cache_idx);
-                 rdp_io_coord(s, &os->startx, delta);
+         in_uint16_le(s, x);
+         in_uint16_le(s, y);
-         if (present & 0x0004)
+         in_uint16_le(s, width);
-                 rdp_io_coord(s, &os->starty, delta);
+         in_uint16_le(s, height);
+         in_uint16_le(s, masklen);
-         if (present & 0x0008)
+         in_uint16_le(s, datalen);
-                 rdp_io_coord(s, &os->endx, delta);
+         in_uint8p(s, data, datalen);
+         in_uint8p(s, mask, masklen);
-         if (present & 0x0010)
+         cursor = ui_create_cursor(x, y, width, height, mask, data);
-                 rdp_io_coord(s, &os->endy, delta);
+         ui_set_cursor(cursor);
+         cache_put_cursor(cache_idx, cursor);
-         if (present & 0x0020)
-                 rdp_io_colour(s, &os->bgcolour);
-         if (present & 0x0040)
-                 prs_io_uint8(s, &os->opcode);
-         rdp_io_pen(s, &os->pen, present >> 7);
-         return PRS_ERROR(s);
  }
- /* Parse an opaque rectangle order */
+ /* Process a cached pointer PDU */
- BOOL rdp_io_rect_order(STREAM s, RECT_ORDER *os, uint32 present, BOOL delta)
+ void
+ process_cached_pointer_pdu(STREAM s)
  {
-         if (present & 0x01)
+         uint16 cache_idx;
-                 rdp_io_coord(s, &os->x, delta);
-         if (present & 0x02)
+         in_uint16_le(s, cache_idx);
-                 rdp_io_coord(s, &os->y, delta);
+         ui_set_cursor(cache_get_cursor(cache_idx));
-         if (present & 0x04)
-                 rdp_io_coord(s, &os->cx, delta);
-         if (present & 0x08)
-                 rdp_io_coord(s, &os->cy, delta);
-         if (present & 0x10)
-                 prs_io_uint8(s, &os->colour);
-         return PRS_ERROR(s);
  }
- /* Parse a desktop save order */
+ /* Process a system pointer PDU */
- BOOL rdp_io_desksave_order(STREAM s, DESKSAVE_ORDER *os, uint32 present, BOOL delta)
+ void
+ process_system_pointer_pdu(STREAM s)
  {
-         if (present & 0x01)
+         uint16 system_pointer_type;
-                 lsb_io_uint32(s, &os->offset);
-         if (present & 0x02)
-                 rdp_io_coord(s, &os->left, delta);
-         if (present & 0x04)
-                 rdp_io_coord(s, &os->top, delta);
-         if (present & 0x08)
+         in_uint16(s, system_pointer_type);
-                 rdp_io_coord(s, &os->right, delta);
+         switch (system_pointer_type)
+         {
-         if (present & 0x10)
+                 case RDP_NULL_POINTER:
-                 rdp_io_coord(s, &os->bottom, delta);
+                         ui_set_null_cursor();
+                         break;
-         if (present & 0x20)
-                 prs_io_uint8(s, &os->action);
-         return PRS_ERROR(s);
+                 default:
+                         unimpl("System pointer message 0x%x\n", system_pointer_type);
+         }
  }
- /* Parse a memory blt order */
+ /* Process a pointer PDU */
- BOOL rdp_io_memblt_order(STREAM s, MEMBLT_ORDER *os, uint32 present, BOOL delta)
+ static void
+ process_pointer_pdu(STREAM s)
  {
-         if (present & 0x0001)
+         uint16 message_type;
-         {
+         uint16 x, y;
-                 prs_io_uint8(s, &os->cache_id);
-                 prs_io_uint8(s, &os->colour_table);
-         }
-         if (present & 0x0002)
-                 rdp_io_coord(s, &os->x, delta);
-         if (present & 0x0004)
-                 rdp_io_coord(s, &os->y, delta);
-         if (present & 0x0008)
+         in_uint16_le(s, message_type);
-                 rdp_io_coord(s, &os->cx, delta);
+         in_uint8s(s, 2);        /* pad */
-         if (present & 0x0010)
+         switch (message_type)
-                 rdp_io_coord(s, &os->cy, delta);
+         {
+                 case RDP_POINTER_MOVE:
-         if (present & 0x0020)
+                         in_uint16_le(s, x);
-                 prs_io_uint8(s, &os->opcode);
+                         in_uint16_le(s, y);
+                         if (s_check(s))
+                                 ui_move_pointer(x, y);
+                         break;
-         if (present & 0x0040)
+                 case RDP_POINTER_COLOR:
-                 rdp_io_coord(s, &os->srcx, delta);
+                         process_colour_pointer_pdu(s);
+                         break;
-         if (present & 0x0080)
+                 case RDP_POINTER_CACHED:
-                 rdp_io_coord(s, &os->srcy, delta);
+                         process_cached_pointer_pdu(s);
+                         break;
-         if (present & 0x0100)
+                 case RDP_POINTER_SYSTEM:
-                 lsb_io_uint16(s, &os->cache_idx);
+                         process_system_pointer_pdu(s);
+                         break;
-         return PRS_ERROR(s);
+                 default:
+                         unimpl("Pointer message 0x%x\n", message_type);
+         }
  }
- /* Parse a 3-way blt order */
+ /* Process bitmap updates */
- BOOL rdp_io_triblt_order(STREAM s, TRIBLT_ORDER *os, uint32 present, BOOL delta)
+ void
+ process_bitmap_updates(STREAM s)
  {
-         if (present & 0x000001)
+         uint16 num_updates;
-         {
+         uint16 left, top, right, bottom, width, height;
-                 prs_io_uint8(s, &os->cache_id);
+         uint16 cx, cy, bpp, Bpp, compress, bufsize, size;
-                 prs_io_uint8(s, &os->colour_table);
+         uint8 *data, *bmpdata;
-         }
+         int i;
-         if (present & 0x000002)
-                 rdp_io_coord(s, &os->x, delta);
-         if (present & 0x000004)
-                 rdp_io_coord(s, &os->y, delta);
-         if (present & 0x000008)
-                 rdp_io_coord(s, &os->cx, delta);
-         if (present & 0x000010)
-                 rdp_io_coord(s, &os->cy, delta);
-         if (present & 0x000020)
-                 prs_io_uint8(s, &os->opcode);
-         if (present & 0x000040)
+         in_uint16_le(s, num_updates);
-                 rdp_io_coord(s, &os->srcx, delta);
-         if (present & 0x000080)
+         for (i = 0; i < num_updates; i++)
-                 rdp_io_coord(s, &os->srcy, delta);
+         {
+                 in_uint16_le(s, left);
+                 in_uint16_le(s, top);
+                 in_uint16_le(s, right);
+                 in_uint16_le(s, bottom);
+                 in_uint16_le(s, width);
+                 in_uint16_le(s, height);
+                 in_uint16_le(s, bpp);
+                 Bpp = (bpp + 7) / 8;
+                 in_uint16_le(s, compress);
+                 in_uint16_le(s, bufsize);
-         if (present & 0x000100)
+                 cx = right - left + 1;
-                 rdp_io_colour(s, &os->bgcolour);
+                 cy = bottom - top + 1;
-         if (present & 0x000200)
+                 DEBUG(("BITMAP_UPDATE(l=%d,t=%d,r=%d,b=%d,w=%d,h=%d,Bpp=%d,cmp=%d)\n",
-                 rdp_io_colour(s, &os->fgcolour);
+                        left, top, right, bottom, width, height, Bpp, compress));
-         rdp_io_brush(s, &os->brush, present >> 10);
+                 if (!compress)
+                 {
+                         int y;
+                         bmpdata = (uint8 *) xmalloc(width * height * Bpp);
+                         for (y = 0; y < height; y++)
+                         {
+                                 in_uint8a(s, &bmpdata[(height - y - 1) * (width * Bpp)],
+                                           width * Bpp);
+                         }
+                         ui_paint_bitmap(left, top, cx, cy, width, height, bmpdata);
+                         xfree(bmpdata);
+                         continue;
+                 }
-         if (present & 0x008000)
-                 lsb_io_uint16(s, &os->cache_idx);
-         if (present & 0x010000)
+                 if (compress & 0x400)
-                 lsb_io_uint16(s, &os->unknown);
+                 {
+                         size = bufsize;
+                 }
+                 else
+                 {
+                         in_uint8s(s, 2);        /* pad */
+                         in_uint16_le(s, size);
+                         in_uint8s(s, 4);        /* line_size, final_size */
+                 }
+                 in_uint8p(s, data, size);
+                 bmpdata = (uint8 *) xmalloc(width * height * Bpp);
+                 if (bitmap_decompress(bmpdata, width, height, data, size, Bpp))
+                 {
+                         ui_paint_bitmap(left, top, cx, cy, width, height, bmpdata);
+                 }
+                 else
+                 {
+                         DEBUG_RDP5(("Failed to decompress data\n"));
+                 }
-         return PRS_ERROR(s);
+                 xfree(bmpdata);
+         }
  }
- /* Parse a text order */
+ /* Process a palette update */
- BOOL rdp_io_text2_order(STREAM s, TEXT2_ORDER *os, uint32 present, BOOL delta)
+ void
+ process_palette(STREAM s)
  {
-         if (present & 0x000001)
+         COLOURENTRY *entry;
-                 prs_io_uint8(s, &os->font);
+         COLOURMAP map;
+         HCOLOURMAP hmap;
-         if (present & 0x000002)
+         int i;
-                 prs_io_uint8(s, &os->flags);
-         if (present & 0x000004)
-                 prs_io_uint8(s, &os->unknown);
-         if (present & 0x000008)
-                 prs_io_uint8(s, &os->mixmode);
-         if (present & 0x000010)
-                 rdp_io_colour(s, &os->fgcolour);
-         if (present & 0x000020)
-                 rdp_io_colour(s, &os->bgcolour);
-         if (present & 0x000040)
-                 lsb_io_uint16(s, &os->clipleft);
-         if (present & 0x000080)
-                 lsb_io_uint16(s, &os->cliptop);
-         if (present & 0x000100)
-                 lsb_io_uint16(s, &os->clipright);
-         if (present & 0x000200)
-                 lsb_io_uint16(s, &os->clipbottom);
-         if (present & 0x000400)
-                 lsb_io_uint16(s, &os->boxleft);
-         if (present & 0x000800)
-                 lsb_io_uint16(s, &os->boxtop);
-         if (present & 0x001000)
-                 lsb_io_uint16(s, &os->boxright);
-         if (present & 0x002000)
+         in_uint8s(s, 2);        /* pad */
-                 lsb_io_uint16(s, &os->boxbottom);
+         in_uint16_le(s, map.ncolours);
+         in_uint8s(s, 2);        /* pad */
-         if (present & 0x080000)
+         map.colours = (COLOURENTRY *) xmalloc(sizeof(COLOURENTRY) * map.ncolours);
-                 lsb_io_uint16(s, &os->x);
-         if (present & 0x100000)
+         DEBUG(("PALETTE(c=%d)\n", map.ncolours));
-                 lsb_io_uint16(s, &os->y);
-         if (present & 0x200000)
+         for (i = 0; i < map.ncolours; i++)
          {
-                 prs_io_uint8(s, &os->length);
+                 entry = &map.colours[i];
-                 prs_io_uint8s(s, os->text, os->length);
+                 in_uint8(s, entry->red);
+                 in_uint8(s, entry->green);
+                 in_uint8(s, entry->blue);
          }
-         return PRS_ERROR(s);
+         hmap = ui_create_colourmap(&map);
- }
+         ui_set_colourmap(hmap);
- /* SECONDARY ORDERS */
- BOOL rdp_io_secondary_order(STREAM s, RDP_SECONDARY_ORDER *rso)
+         xfree(map.colours);
- {
-         BOOL res = True;
-         res = res ? lsb_io_uint16(s, &rso->length) : False;
-         res = res ? lsb_io_uint16(s, &rso->flags ) : False;
-         res = res ? prs_io_uint8 (s, &rso->type  ) : False;
-         return res;
  }
- BOOL rdp_io_raw_bmpcache_order(STREAM s, RDP_RAW_BMPCACHE_ORDER *rbo)
+ /* Process an update PDU */
+ static void
+ process_update_pdu(STREAM s)
  {
-         BOOL res = True;
+         uint16 update_type, count;
-         res = res ? prs_io_uint8 (s, &rbo->cache_id  ) : False;
+         in_uint16_le(s, update_type);
-         res = res ? prs_io_uint8 (s, &rbo->pad1      ) : False;
-         res = res ? prs_io_uint8 (s, &rbo->width     ) : False;
-         res = res ? prs_io_uint8 (s, &rbo->height    ) : False;
-         res = res ? prs_io_uint8 (s, &rbo->bpp       ) : False;
-         res = res ? lsb_io_uint16(s, &rbo->bufsize   ) : False;
-         res = res ? lsb_io_uint16(s, &rbo->cache_idx ) : False;
-         rbo->data = s->data + s->offset;
+         ui_begin_update();
-         s->offset += rbo->bufsize;
+         switch (update_type)
+         {
-         return res;
+                 case RDP_UPDATE_ORDERS:
- }
+                         in_uint8s(s, 2);        /* pad */
+                         in_uint16_le(s, count);
+                         in_uint8s(s, 2);        /* pad */
+                         process_orders(s, count);
+                         break;
- BOOL rdp_io_bmpcache_order(STREAM s, RDP_BMPCACHE_ORDER *rbo)
+                 case RDP_UPDATE_BITMAP:
- {
+                         process_bitmap_updates(s);
-         BOOL res = True;
+                         break;
-         res = res ? prs_io_uint8 (s, &rbo->cache_id  ) : False;
+                 case RDP_UPDATE_PALETTE:
-         res = res ? prs_io_uint8 (s, &rbo->pad1      ) : False;
+                         process_palette(s);
-         res = res ? prs_io_uint8 (s, &rbo->width     ) : False;
+                         break;
-         res = res ? prs_io_uint8 (s, &rbo->height    ) : False;
-         res = res ? prs_io_uint8 (s, &rbo->bpp       ) : False;
-         res = res ? lsb_io_uint16(s, &rbo->bufsize   ) : False;
-         res = res ? lsb_io_uint16(s, &rbo->cache_idx ) : False;
-         res = res ? lsb_io_uint16(s, &rbo->pad2      ) : False;
-         res = res ? lsb_io_uint16(s, &rbo->size      ) : False;
-         res = res ? lsb_io_uint16(s, &rbo->row_size  ) : False;
-         res = res ? lsb_io_uint16(s, &rbo->final_size) : False;
-         rbo->data = s->data + s->offset;
+                 case RDP_UPDATE_SYNCHRONIZE:
-         s->offset += rbo->size;
+                         break;
-         return res;
+                 default:
+                         unimpl("update %d\n", update_type);
+         }
+         ui_end_update();
  }
- BOOL rdp_io_colcache_order(STREAM s, RDP_COLCACHE_ORDER *colours)
+ /* Process a disconnect PDU */
+ void
+ process_disconnect_pdu(STREAM s, uint32 * ext_disc_reason)
  {
-         COLOURENTRY *entry;
+         in_uint32_le(s, *ext_disc_reason);
-         int i;
-         prs_io_uint8(s, &colours->cache_id);
-         lsb_io_uint16(s, &colours->map.ncolours);
-         for (i = 0; i < colours->map.ncolours; i++)
-         {
-                 entry = &colours->map.colours[i];
-                 prs_io_uint8(s, &entry->blue);
-                 prs_io_uint8(s, &entry->green);
-                 prs_io_uint8(s, &entry->red);
-                 s->offset++;
-         }
-         return True;
+         DEBUG(("Received disconnect PDU\n"));
  }
- BOOL rdp_io_fontcache_order(STREAM s, RDP_FONTCACHE_ORDER *font)
+ /* Process data PDU */
+ static BOOL
+ process_data_pdu(STREAM s, uint32 * ext_disc_reason)
  {
-         RDP_FONT_GLYPH *glyph;
+         uint8 data_pdu_type;
-         BOOL res = True;
+         uint8 ctype;
-         int i, j, datasize;
+         uint16 clen;
-         uint8 in, out;
+         uint32 len;
-         res = res ? prs_io_uint8(s, &font->font   ) : False;
+         uint32 roff, rlen;
-         res = res ? prs_io_uint8(s, &font->nglyphs) : False;
-         for (i = 0; i < font->nglyphs; i++)
+         struct stream *ns = &(g_mppc_dict.ns);
-         {
-                 glyph = &font->glyphs[i];
-                 res = res ? lsb_io_uint16(s, &glyph->character) : False;
-                 res = res ? lsb_io_uint16(s, &glyph->unknown  ) : False;
-                 res = res ? lsb_io_uint16(s, &glyph->baseline ) : False;
-                 res = res ? lsb_io_uint16(s, &glyph->width    ) : False;
-                 res = res ? lsb_io_uint16(s, &glyph->height   ) : False;
-                 datasize = (glyph->height * ((glyph->width + 7) / 8) + 3) & ~3;
+         in_uint8s(s, 6);        /* shareid, pad, streamid */
-                 res = res ? prs_io_uint8s(s, glyph->data, datasize) : False;
+         in_uint16(s, len);
-                 for (j = 0; j < datasize; j++)
+         in_uint8(s, data_pdu_type);
-                 {
+         in_uint8(s, ctype);
-                         in = glyph->data[j];
+         in_uint16(s, clen);
-                         out = 0;
+         clen -= 18;
-                         if (in & 1) out |= 128;
-                         if (in & 2) out |= 64;
-                         if (in & 4) out |= 32;
-                         if (in & 8) out |= 16;
-                         if (in & 16) out |= 8;
-                         if (in & 32) out |= 4;
-                         if (in & 64) out |= 2;
-                         if (in & 128) out |= 1;
-                         glyph->data[j] = out;
-                 }
-         }
-         return res;
+         if (ctype & RDP_MPPC_COMPRESSED)
- }
+         {
+                 if (len > RDP_MPPC_DICT_SIZE)
+                         error("error decompressed packet size exceeds max\n");
+                 if (mppc_expand(s->p, clen, ctype, &roff, &rlen) == -1)
+                         error("error while decompressing packet\n");
+                 /* len -= 18; */
- /* CAPABILITIES */
+                 /* allocate memory and copy the uncompressed data into the temporary stream */
+                 ns->data = (uint8 *) xrealloc(ns->data, rlen);
- /* Construct a general capability set */
+                 memcpy((ns->data), (unsigned char *) (g_mppc_dict.hist + roff), rlen);
- void rdp_make_general_caps(RDP_GENERAL_CAPS *caps)
- {
-         caps->os_major_type = 1;
-         caps->os_minor_type = 3;
-         caps->ver_protocol = 0x200;
- }
- /* Parse general capability set */
+                 ns->size = rlen;
- BOOL rdp_io_general_caps(STREAM s, RDP_GENERAL_CAPS *caps)
+                 ns->end = (ns->data + ns->size);
- {
+                 ns->p = ns->data;
-         uint16 length = RDP_CAPLEN_GENERAL;
+                 ns->rdp_hdr = ns->p;
-         uint16 pkt_length = length;
-         BOOL res;
-         res = lsb_io_uint16(s, &pkt_length);
+                 s = ns;
-         if (pkt_length != length)
-         {
-                 ERROR("Unrecognised capabilities size\n");
-                 return False;
          }
-         res = res ? lsb_io_uint16(s, &caps->os_major_type ) : False;
+         switch (data_pdu_type)
-         res = res ? lsb_io_uint16(s, &caps->os_minor_type ) : False;
-         res = res ? lsb_io_uint16(s, &caps->ver_protocol  ) : False;
-         res = res ? lsb_io_uint16(s, &caps->pad1          ) : False;
-         res = res ? lsb_io_uint16(s, &caps->compress_types) : False;
-         res = res ? lsb_io_uint16(s, &caps->pad2          ) : False;
-         res = res ? lsb_io_uint16(s, &caps->cap_update    ) : False;
-         res = res ? lsb_io_uint16(s, &caps->remote_unshare) : False;
-         res = res ? lsb_io_uint16(s, &caps->compress_level) : False;
-         res = res ? lsb_io_uint16(s, &caps->pad3          ) : False;
-         return res;
- }
- /* Construct a bitmap capability set */
- void rdp_make_bitmap_caps(RDP_BITMAP_CAPS *caps, int width, int height)
- {
-         caps->preferred_bpp = 8;
-         caps->receive1bpp = 1;
-         caps->receive4bpp = 1;
-         caps->receive8bpp = 1;
-         caps->width = width;
-         caps->height = height;
-         caps->compression = 1;
-         caps->unknown2 = 1;
- }
- /* Parse bitmap capability set */
- BOOL rdp_io_bitmap_caps(STREAM s, RDP_BITMAP_CAPS *caps)
- {
-         uint16 length = RDP_CAPLEN_BITMAP;
-         uint16 pkt_length = length;
-         BOOL res;
-         res = lsb_io_uint16(s, &pkt_length);
-         if (pkt_length != length)
          {
-                 ERROR("Unrecognised capabilities size\n");
+                 case RDP_DATA_PDU_UPDATE:
-                 return False;
+                         process_update_pdu(s);
-         }
+                         break;
-         res = res ? lsb_io_uint16(s, &caps->preferred_bpp) : False;
+                 case RDP_DATA_PDU_CONTROL:
-         res = res ? lsb_io_uint16(s, &caps->receive1bpp  ) : False;
+                         DEBUG(("Received Control PDU\n"));
-         res = res ? lsb_io_uint16(s, &caps->receive4bpp  ) : False;
+                         break;
-         res = res ? lsb_io_uint16(s, &caps->receive8bpp  ) : False;
-         res = res ? lsb_io_uint16(s, &caps->width        ) : False;
-         res = res ? lsb_io_uint16(s, &caps->height       ) : False;
-         res = res ? lsb_io_uint16(s, &caps->pad1         ) : False;
-         res = res ? lsb_io_uint16(s, &caps->allow_resize ) : False;
-         res = res ? lsb_io_uint16(s, &caps->compression  ) : False;
-         res = res ? lsb_io_uint16(s, &caps->unknown1     ) : False;
-         res = res ? lsb_io_uint16(s, &caps->unknown2     ) : False;
-         res = res ? lsb_io_uint16(s, &caps->pad2         ) : False;
-         return res;
- }
- /* Construct an order capability set */
- void rdp_make_order_caps(RDP_ORDER_CAPS *caps)
- {
-         caps->xgranularity = 1;
-         caps->ygranularity = 20;
-         caps->max_order_level = 1;
-         caps->num_fonts = 0x147;
-         caps->cap_flags = 0x2A;
- //      caps->cap_flags = ORDER_CAP_NEGOTIATE | ORDER_CAP_NOSUPPORT;
-         caps->support[0] = caps->support[1] = caps->support[2]
-                 = caps->support[3] = caps->support[4] = caps->support[5]
-                 = caps->support[6] = caps->support[8] = caps->support[11]
-                 = caps->support[12] = caps->support[22] = caps->support[28]
-                 = caps->support[29] = caps->support[30] = 1;
-         caps->text_cap_flags = 0x6A1;
-         caps->desk_save_size = 0x38400;
-         caps->unknown2 = 0x4E4;
- }
- /* Parse order capability set */
- BOOL rdp_io_order_caps(STREAM s, RDP_ORDER_CAPS *caps)
- {
-         uint16 length = RDP_CAPLEN_ORDER;
-         uint16 pkt_length = length;
-         BOOL res;
-         res = lsb_io_uint16(s, &pkt_length);
+                 case RDP_DATA_PDU_SYNCHRONISE:
-         if (pkt_length != length)
+                         DEBUG(("Received Sync PDU\n"));
-         {
+                         break;
-                 ERROR("Unrecognised capabilities size\n");
-                 return False;
-         }
-         res = res ? prs_io_uint8s(s,  caps->terminal_desc, 16) : False;
+                 case RDP_DATA_PDU_POINTER:
-         res = res ? lsb_io_uint32(s, &caps->pad1             ) : False;
+                         process_pointer_pdu(s);
-         res = res ? lsb_io_uint16(s, &caps->xgranularity     ) : False;
+                         break;
-         res = res ? lsb_io_uint16(s, &caps->ygranularity     ) : False;
-         res = res ? lsb_io_uint16(s, &caps->pad2             ) : False;
-         res = res ? lsb_io_uint16(s, &caps->max_order_level  ) : False;
-         res = res ? lsb_io_uint16(s, &caps->num_fonts        ) : False;
-         res = res ? lsb_io_uint16(s, &caps->cap_flags        ) : False;
-         res = res ? prs_io_uint8s(s,  caps->support      , 32) : False;
-         res = res ? lsb_io_uint16(s, &caps->text_cap_flags   ) : False;
-         res = res ? lsb_io_uint16(s, &caps->pad3             ) : False;
-         res = res ? lsb_io_uint32(s, &caps->pad4             ) : False;
-         res = res ? lsb_io_uint32(s, &caps->desk_save_size   ) : False;
-         res = res ? lsb_io_uint32(s, &caps->unknown1         ) : False;
-         res = res ? lsb_io_uint32(s, &caps->unknown2         ) : False;
-         return res;
- }
- /* Construct a bitmap cache capability set */
- void rdp_make_bmpcache_caps(RDP_BMPCACHE_CAPS *caps)
- {
-         caps->caches[0].entries = 0x258;
-         caps->caches[0].max_cell_size = 0x100;
-         caps->caches[1].entries = 0x12c;
-         caps->caches[1].max_cell_size = 0x400;
-         caps->caches[2].entries = 0x106;
-         caps->caches[2].max_cell_size = 0x1000;
- }
- /* Parse single bitmap cache information structure */
+                 case RDP_DATA_PDU_BELL:
- BOOL rdp_io_bmpcache_info(STREAM s, RDP_BMPCACHE_INFO *info)
+                         ui_bell();
- {
+                         break;
-         if (!lsb_io_uint16(s, &info->entries      ))
-                 return False;
-         if (!lsb_io_uint16(s, &info->max_cell_size))
+                 case RDP_DATA_PDU_LOGON:
-                 return False;
+                         DEBUG(("Received Logon PDU\n"));
+                         /* User logged on */
+                         break;
-         return True;
+                 case RDP_DATA_PDU_DISCONNECT:
- }
+                         process_disconnect_pdu(s, ext_disc_reason);
- /* Parse bitmap cache capability set */
+                         /* We used to return true and disconnect immediately here, but
- BOOL rdp_io_bmpcache_caps(STREAM s, RDP_BMPCACHE_CAPS *caps)
+                          * Windows Vista sends a disconnect PDU with reason 0 when
- {
+                          * reconnecting to a disconnected session, and MSTSC doesn't
-         uint16 length = RDP_CAPLEN_BMPCACHE;
+                          * drop the connection.  I think we should just save the status.
-         uint16 pkt_length = length;
+                          */
-         BOOL res;
+                         break;
-         int i;
-         res = lsb_io_uint16(s, &pkt_length);
+                 default:
-         if (pkt_length != length)
+                         unimpl("data PDU %d\n", data_pdu_type);
-         {
-                 ERROR("Unrecognised capabilities size\n");
-                 return False;
          }
+         return False;
-         for (i = 0; i < 6; i++)
-                 res = res ? lsb_io_uint32(s, &caps->unused[i]) : False;
-         for (i = 0; i < 3; i++)
-                 res = res ? rdp_io_bmpcache_info(s, &caps->caches[i]) : False;
-         return res;
  }
- /* Construct a control capability set */
+ /* Process redirect PDU from Session Directory */
- void rdp_make_control_caps(RDP_CONTROL_CAPS *caps)
+ static BOOL
+ process_redirect_pdu(STREAM s /*, uint32 * ext_disc_reason */ )
  {
-         caps->control_interest = 2;
+         uint32 len;
-         caps->detach_interest = 2;
- }
- /* Parse control capability set */
+         /* these 2 bytes are unknown, seem to be zeros */
- BOOL rdp_io_control_caps(STREAM s, RDP_CONTROL_CAPS *caps)
+         in_uint8s(s, 2);
- {
-         uint16 length = RDP_CAPLEN_CONTROL;
-         uint16 pkt_length = length;
-         BOOL res;
-         res = lsb_io_uint16(s, &pkt_length);
+         /* read connection flags */
-         if (pkt_length != length)
+         in_uint32_le(s, g_redirect_flags);
-         {
-                 ERROR("Unrecognised capabilities size\n");
-                 return False;
-         }
-         res = res ? lsb_io_uint16(s, &caps->control_caps    ) : False;
+         /* read length of ip string */
-         res = res ? lsb_io_uint16(s, &caps->remote_detach   ) : False;
+         in_uint32_le(s, len);
-         res = res ? lsb_io_uint16(s, &caps->control_interest) : False;
-         res = res ? lsb_io_uint16(s, &caps->detach_interest ) : False;
-         return res;
+         /* read ip string */
- }
+         rdp_in_unistr(s, g_redirect_server, len);
- /* Construct an activation capability set */
+         /* read length of cookie string */
- void rdp_make_activate_caps(RDP_ACTIVATE_CAPS *caps)
+         in_uint32_le(s, len);
- {
- }
- /* Parse activation capability set */
+         /* read cookie string (plain ASCII) */
- BOOL rdp_io_activate_caps(STREAM s, RDP_ACTIVATE_CAPS *caps)
+         in_uint8a(s, g_redirect_cookie, len);
- {
+         g_redirect_cookie[len] = 0;
-         uint16 length = RDP_CAPLEN_ACTIVATE;
-         uint16 pkt_length = length;
-         BOOL res;
-         res = lsb_io_uint16(s, &pkt_length);
+         /* read length of username string */
-         if (pkt_length != length)
+         in_uint32_le(s, len);
-         {
-                 ERROR("Unrecognised capabilities size\n");
-                 return False;
-         }
-         res = res ? lsb_io_uint16(s, &caps->help_key         ) : False;
+         /* read username string */
-         res = res ? lsb_io_uint16(s, &caps->help_index_key   ) : False;
+         rdp_in_unistr(s, g_redirect_username, len);
-         res = res ? lsb_io_uint16(s, &caps->help_extended_key) : False;
-         res = res ? lsb_io_uint16(s, &caps->window_activate  ) : False;
-         return res;
+         /* read length of domain string */
- }
+         in_uint32_le(s, len);
- /* Construct a pointer capability set */
+         /* read domain string */
- void rdp_make_pointer_caps(RDP_POINTER_CAPS *caps)
+         rdp_in_unistr(s, g_redirect_domain, len);
- {
-         caps->colour_pointer = 0;
-         caps->cache_size = 20;
- }
- /* Parse pointer capability set */
+         /* read length of password string */
- BOOL rdp_io_pointer_caps(STREAM s, RDP_POINTER_CAPS *caps)
+         in_uint32_le(s, len);
- {
-         uint16 length = RDP_CAPLEN_POINTER;
-         uint16 pkt_length = length;
-         BOOL res;
-         res = lsb_io_uint16(s, &pkt_length);
+         /* read password string */
-         if (pkt_length != length)
+         rdp_in_unistr(s, g_redirect_password, len);
-         {
-                 ERROR("Unrecognised capabilities size\n");
-                 return False;
-         }
-         res = res ? lsb_io_uint16(s, &caps->colour_pointer) : False;
+         g_redirect = True;
-         res = res ? lsb_io_uint16(s, &caps->cache_size    ) : False;
-         return res;
+         return True;
  }
- /* Construct a share capability set */
+ /* Process incoming packets */
- void rdp_make_share_caps(RDP_SHARE_CAPS *caps, uint16 userid)
+ /* nevers gets out of here till app is done */
+ void
+ rdp_main_loop(BOOL * deactivated, uint32 * ext_disc_reason)
  {
+         while (rdp_loop(deactivated, ext_disc_reason))
+                 ;
  }
- /* Parse share capability set */
+ /* used in uiports and rdp_main_loop, processes the rdp packets waiting */
- BOOL rdp_io_share_caps(STREAM s, RDP_SHARE_CAPS *caps)
+ BOOL
+ rdp_loop(BOOL * deactivated, uint32 * ext_disc_reason)
  {
-         uint16 length = RDP_CAPLEN_SHARE;
+         uint8 type;
-         uint16 pkt_length = length;
+         BOOL disc = False;      /* True when a disconnect PDU was received */
-         BOOL res;
+         BOOL cont = True;
+         STREAM s;
-         res = lsb_io_uint16(s, &pkt_length);
+         while (cont)
-         if (pkt_length != length)
          {
-                 ERROR("Unrecognised capabilities size\n");
+                 s = rdp_recv(&type);
-                 return False;
+                 if (s == NULL)
+                         return False;
+                 switch (type)
+                 {
+                         case RDP_PDU_DEMAND_ACTIVE:
+                                 process_demand_active(s);
+                                 *deactivated = False;
+                                 break;
+                         case RDP_PDU_DEACTIVATE:
+                                 DEBUG(("RDP_PDU_DEACTIVATE\n"));
+                                 *deactivated = True;
+                                 break;
+                         case RDP_PDU_REDIRECT:
+                                 return process_redirect_pdu(s);
+                                 break;
+                         case RDP_PDU_DATA:
+                                 disc = process_data_pdu(s, ext_disc_reason);
+                                 break;
+                         case 0:
+                                 break;
+                         default:
+                                 unimpl("PDU %d\n", type);
+                 }
+                 if (disc)
+                         return False;
+                 cont = g_next_packet < s->end;
          }
+         return True;
-         res = res ? lsb_io_uint16(s, &caps->userid) : False;
-         res = res ? lsb_io_uint16(s, &caps->pad   ) : False;
-         return res;
  }
- /* Construct a colour cache capability set */
+ /* Establish a connection up to the RDP layer */
- void rdp_make_colcache_caps(RDP_COLCACHE_CAPS *caps)
+ BOOL
+ rdp_connect(char *server, uint32 flags, char *domain, char *password,
+             char *command, char *directory)
  {
-         caps->cache_size = 6;
+         if (!sec_connect(server, g_username))
+                 return False;
+         rdp_send_logon_info(flags, domain, g_username, password, command, directory);
+         return True;
  }
- /* Parse colour cache capability set */
+ /* Establish a reconnection up to the RDP layer */
- BOOL rdp_io_colcache_caps(STREAM s, RDP_COLCACHE_CAPS *caps)
+ BOOL
+ rdp_reconnect(char *server, uint32 flags, char *domain, char *password,
+               char *command, char *directory, char *cookie)
  {
-         uint16 length = RDP_CAPLEN_COLCACHE;
+         if (!sec_reconnect(server))
-         uint16 pkt_length = length;
-         BOOL res;
-         res = lsb_io_uint16(s, &pkt_length);
-         if (pkt_length != length)
-         {
-                 ERROR("Unrecognised capabilities size\n");
                  return False;
-         }
-         res = res ? lsb_io_uint16(s, &caps->cache_size) : False;
-         res = res ? lsb_io_uint16(s, &caps->pad       ) : False;
-         return res;
+         rdp_send_logon_info(flags, domain, g_username, password, command, directory);
+         return True;
  }
- uint8 canned_caps[] = {
+ /* Called during redirection to reset the state to support redirection */
-x01,0x00,0x00,0x00,0x09,0x04,0x00,0x00,0x04,
+ void
-x00,0x00,0x00,0x00,0x00,0x00,0x00,0x0C,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,
+ rdp_reset_state(void)
-x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,
+ {
-x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,
+         g_next_packet = NULL;   /* reset the packet information */
-x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,
+         g_rdp_shareid = 0;
-x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x0C,0x00,0x08,0x00,0x01,
+         sec_reset_state();
-x00,0x00,0x00,0x0E,0x00,0x08,0x00,0x01,0x00,0x00,0x00,0x10,0x00,0x34,0x00,0xFE,
+ }
-x00,0x04,0x00,0xFE,0x00,0x04,0x00,0xFE,0x00,0x08,0x00,0xFE,0x00,0x08,0x00,0xFE,
-x00,0x10,0x00,0xFE,0x00,0x20,0x00,0xFE,0x00,0x40,0x00,0xFE,0x00,0x80,0x00,0xFE,
-x00,0x00,0x01,0x40,0x00,0x00,0x08,0x00,0x01,0x00,0x01,0x02,0x00,0x00,0x00
- };
- /* Insert canned capabilities */
+ /* Disconnect from the RDP layer */
- BOOL rdp_io_unknown_caps(STREAM s, void *caps)
+ void
+ rdp_disconnect(void)
  {
-         uint16 length = 0x58;
+         sec_disconnect();
-         uint16 pkt_length = length;
-         BOOL res;
-         res = lsb_io_uint16(s, &pkt_length);
-         if (pkt_length != length)
-         {
-                 ERROR("Unrecognised capabilities size\n");
-                 return False;
-         }
-         res = res ? prs_io_uint8s(s, canned_caps, RDP_CAPLEN_UNKNOWN-4) : False;
-         return res;
  }

 Legend:



Removed from v.9
 


changed lines


 
Added in v.1238
 Legend:



Removed from v.9
 


changed lines


 
Added in v.1238
-Removed from v.9
+Added in v.1238

	ViewVC Help
Powered by ViewVC 1.1.26