/[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!
ViewVC logotype

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

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 4 by matty, Wed May 10 07:36:34 2000 UTC revision 340 by forsberg, Thu Mar 6 14:11:17 2003 UTC
# Line 1  Line 1 
1  /*  /*
2     rdesktop: A Remote Desktop Protocol client.     rdesktop: A Remote Desktop Protocol client.
3     Protocol services - RDP layer     Protocol services - RDP layer
4     Copyright (C) Matthew Chapman 1999-2000     Copyright (C) Matthew Chapman 1999-2002
5        
6     This program is free software; you can redistribute it and/or modify     This program is free software; you can redistribute it and/or modify
7     it under the terms of the GNU General Public License as published by     it under the terms of the GNU General Public License as published by
# Line 18  Line 18 
18     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19  */  */
20    
21  #include "includes.h"  #include "rdesktop.h"
22    
23  /* Establish a connection up to the RDP layer */  extern uint16 mcs_userid;
24  HCONN rdp_connect(char *server)  extern char username[16];
25  {  extern BOOL bitmap_compression;
26          HCONN conn;  extern BOOL orders;
27          RDP_ACTIVE_PDU active;  extern BOOL encryption;
28          RDP_DATA_HEADER hdr;  extern BOOL desktop_save;
         RDP_UPDATE_PDU update;  
         RDP_ORDER_STATE os;  
         uint8 type;  
29    
30          memset(&os, 0, sizeof(os));  uint8 *next_packet;
31    uint32 rdp_shareid;
32    
33          if ((conn = mcs_connect(server)) == NULL)  #if WITH_DEBUG
34                  return NULL;  static uint32 packetno;
35    #endif
36    
37          rdp_establish_key(conn);  /* Initialise an RDP packet */
38          mcs_recv(conn, False); /* Server's licensing certificate */  static STREAM
39          rdp_send_cert(conn);  rdp_init(int maxlen)
40          mcs_recv(conn, False);  {
41            STREAM s;
42    
43          if (rdp_recv_pdu(conn, &type) && (type != RDP_PDU_DEMAND_ACTIVE))          s = sec_init(encryption ? SEC_ENCRYPT : 0, maxlen + 6);
44          {          s_push_layer(s, rdp_hdr, 6);
                 fprintf(stderr, "RDP error, expected Demand Active\n");  
                 mcs_disconnect(conn);  
                 return NULL;  
         }  
45    
46          rdp_io_active_pdu(&conn->in, &active, RDP_PDU_DEMAND_ACTIVE);          return s;
47          rdp_send_confirm_active(conn);  }
         rdp_send_synchronize(conn);  
         rdp_send_control(conn, RDP_CTL_COOPERATE);  
         rdp_send_control(conn, RDP_CTL_REQUEST_CONTROL);  
         rdp_recv_pdu(conn, &type); // RDP_PDU_SYNCHRONIZE  
         rdp_recv_pdu(conn, &type); // RDP_CTL_COOPERATE  
         rdp_recv_pdu(conn, &type); // RDP_CTL_GRANT_CONTROL  
         rdp_send_input(conn);  
         rdp_send_fonts(conn, 1);  
         rdp_send_fonts(conn, 2);  
         rdp_recv_pdu(conn, &type); // RDP_PDU_UNKNOWN 0x28  
         while (rdp_recv_pdu(conn, &type))  
         {  
                 if (type != RDP_PDU_DATA)  
                         continue;  
48    
49                  rdp_io_data_header(&conn->in, &hdr);  /* Send an RDP packet */
50    static void
51    rdp_send(STREAM s, uint8 pdu_type)
52    {
53            uint16 length;
54    
55                  switch (hdr.data_pdu_type)          s_pop_layer(s, rdp_hdr);
56                  {          length = s->end - s->p;
57                  case RDP_DATA_PDU_UPDATE:  
58                          rdp_io_update_pdu(&conn->in, &update);          out_uint16_le(s, length);
59                          if (update.update_type == RDP_UPDATE_ORDERS)          out_uint16_le(s, (pdu_type | 0x10));    /* Version 1 */
60                          {          out_uint16_le(s, (mcs_userid + 1001));
                                 fprintf(stderr, "Received orders\n");  
                                 process_orders(conn, &os);  
                         }  
                         break;  
                 }  
         }  
61    
62          return conn;          sec_send(s, encryption ? SEC_ENCRYPT : 0);
63  }  }
64    
65  void prs_io_coord(STREAM s, uint16 *coord, BOOL delta)  /* Receive an RDP packet */
66    static STREAM
67    rdp_recv(uint8 * type)
68  {  {
69          uint8 change;          static STREAM rdp_s;
70            uint16 length, pdu_type;
71    
72          if (delta)          if ((rdp_s == NULL) || (next_packet >= rdp_s->end))
73          {          {
74                  prs_io_uint8(s, &change);                  rdp_s = sec_recv();
75                  *coord += change;                  if (rdp_s == NULL)
76                            return NULL;
77    
78                    next_packet = rdp_s->p;
79          }          }
80          else          else
81          {          {
82                  lsb_io_uint16(s, coord);                  rdp_s->p = next_packet;
83          }          }
 }  
   
 void process_opaque_rect(HCONN conn, RDP_ORDER_STATE *os, BOOL delta)  
 {  
         uint8 present;  
         prs_io_uint8(&conn->in, &present);  
   
         if (present & 1)  
                 prs_io_coord(&conn->in, &os->opaque_rect.x, delta);  
   
         if (present & 2)  
                 prs_io_coord(&conn->in, &os->opaque_rect.y, delta);  
   
         if (present & 4)  
                 prs_io_coord(&conn->in, &os->opaque_rect.cx, delta);  
   
         if (present & 8)  
                 prs_io_coord(&conn->in, &os->opaque_rect.cy, delta);  
   
         if (present & 16)  
                 prs_io_uint8(&conn->in, &os->opaque_rect.colour);  
   
         fprintf(stderr, "Opaque rectangle at %d, %d\n", os->opaque_rect.x, os->opaque_rect.y);  
 }  
   
 void process_orders(HCONN conn, RDP_ORDER_STATE *os)  
 {  
         uint16 num_orders;  
         int processed = 0;  
         BOOL res = True;  
         //      unsigned char *p;  
   
         lsb_io_uint16(&conn->in, &num_orders);  
84    
85          conn->in.offset += 2;          in_uint16_le(rdp_s, length);
86          //      p = &conn->in.data[conn->in.offset];          /* 32k packets are really 8, keepalive fix */
87            if (length == 0x8000)
         //      fprintf(stderr, "%02X %02X %02X %02X\n", p[0], p[1], p[2], p[3]);  
   
         while ((processed < num_orders) && res)  
88          {          {
89                  uint8 order_flags;                  next_packet += 8;
90                    *type = 0;
91                  prs_io_uint8(&conn->in, &order_flags);                  return rdp_s;
   
                 if (!(order_flags & RDP_ORDER_STANDARD))  
                         return;  
   
                 if (order_flags & RDP_ORDER_SECONDARY)  
                 {  
                         RDP_SECONDARY_ORDER rso;  
   
                         rdp_io_secondary_order(&conn->in, &rso);  
                         switch (rso.type)  
                         {  
                         case RDP_ORDER_BMPCACHE:  
                         {  
                                 RDP_BITMAP_HEADER rbh;  
                                 char output[8192];  
   
                                 rdp_io_bitmap_header(&conn->in, &rbh);  
                                 fprintf(stderr, "Decompressing bitmap %d x %d, final size %d\n", rbh.width, rbh.height, rbh.final_size);  
                                 bitmap_decompress(conn->in.data  
                                                   + conn->in.offset, rbh.size,  
                                                   output, rbh.width);  
                                 conn->in.offset += rbh.size;  
                                 break;  
                         }  
                         default:  
                                 fprintf(stderr, "Unknown secondary order %d\n",  
                                         rso.type);  
                                 return;  
                         }  
   
   
                 }  
   
                 if (order_flags & RDP_ORDER_CHANGE)  
                         prs_io_uint8(&conn->in, &os->order_type);  
   
                 switch (os->order_type)  
                 {  
                 case RDP_ORDER_OPAQUE_RECT:  
                         process_opaque_rect(conn, os, order_flags & RDP_ORDER_DELTA);  
                         break;  
                 default:  
                         fprintf(stderr, "Unknown order %d\n", os->order_type);  
                         return;  
                 }  
   
                 processed++;  
92          }          }
93  }          in_uint16_le(rdp_s, pdu_type);
94            in_uint8s(rdp_s, 2);    /* userid */
95            *type = pdu_type & 0xf;
96    
97  /* Work this out later. This is useless anyway when encryption is off. */  #if WITH_DEBUG
98  uint8 precanned_key_packet[] = {          DEBUG(("RDP packet #%d, (type %x):\n", ++packetno, *type));
99     0x48,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x33,0x00,0x00,0x00,0x00,0x00,          hexdump(next_packet, length);
100     0x00,0x00,0x20,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6c,0x86,  #endif /*  */
    0xf7,0x99,0xef,0x60,0xc4,0x49,0x52,0xd0,0xd8,0xea,0xb5,0x4f,0x58,0x19,  
    0x52,0x2a,0x93,0x83,0x57,0x4f,0x4e,0x04,0xde,0x96,0x51,0xab,0x13,0x20,  
    0xd8,0xe5,0x00,0x00,0x00,0x00,0x00,0x00  
 };  
   
 /* Create an RC4 key and transfer it to the server */  
 void rdp_establish_key(HCONN conn)  
 {  
         mcs_init_data(conn);  
         memcpy(conn->out.data + conn->out.offset, precanned_key_packet,  
                sizeof(precanned_key_packet));  
         conn->out.offset += sizeof(precanned_key_packet);  
         MARK_END(conn->out);  
         mcs_send_data(conn, MCS_GLOBAL_CHANNEL, True);  
 }  
   
 /* Horrible horrible certificate stuff. Work out later. */  
 uint8 precanned_cert_packet[] = {  
 0x80,0x00,0x00,0x00,0x12,0x02,0xb4,0x04,0x01,0x00,0x00,  
 0x00,0x00,0x00,0x01,0x02,0x9d,0xa3,0x7a,0x93,0x34,0x7b,0x28,0x37,0x24,0xa0,0x1f,  
 0x61,0x26,0xfd,0x96,0x3a,0x92,0x83,0xf3,0xe9,0x6a,0x2e,0x81,0x7c,0x2c,0xe4,0x72,//  
 0x01,0x18,0xe9,0xa1,0x0f,0x00,0x00,0x48,0x00,0x84,0x23,0x90,0xe6,0xd3,0xf8,0x20,  
 0xdb,0xa8,0x1b,0xb2,0xd0,0x78,0x2c,0x35,0xde,0xe3,0x0e,0x63,0x40,0xca,0xac,0x71,  
 0xc9,0x17,0x49,0x05,0x25,0xeb,0x9b,0xd0,0xa6,0x5c,0x90,0x3e,0x9d,0x4b,0x27,0x01,  
 0x79,0x1c,0x22,0xfb,0x3c,0x2c,0xb9,0x9f,0xf5,0x21,0xf3,0xee,0xd5,0x4d,0x47,0x1c,  
 0x85,0xbe,0x83,0x93,0xe8,0xed,0x8c,0x5c,0x82,0x00,0x00,0x00,0x00,0x00,0x00,0x00,  
 0x00,0x01,0x00,0x10,0x04,0x30,0x82,0x04,0x0c,0x06,0x09,0x2a,0x86,0x48,0x86,0xf7,  
 0x0d,0x01,0x07,0x02,0xa0,0x82,0x03,0xfd,0x30,0x82,0x03,0xf9,0x02,0x01,0x01,0x31,  
 0x00,0x30,0x0b,0x06,0x09,0x2a,0x86,0x48,0x86,0xf7,0x0d,0x01,0x07,0x01,0xa0,0x82,  
 0x03,0xe1,0x30,0x82,0x01,0x77,0x30,0x82,0x01,0x25,0xa0,0x03,0x02,0x01,0x02,0x02,  
 0x08,0x01,0xbf,0x06,0x84,0x9d,0xdb,0x2d,0xe0,0x30,0x0d,0x06,0x09,0x2a,0x86,0x48,  
 0x86,0xf7,0x0d,0x01,0x01,0x01,0x05,0x00,0x30,0x38,0x31,0x36,0x30,0x11,0x06,0x03,  
 0x55,0x04,0x03,0x1e,0x0a,0x00,0x4e,0x00,0x54,0x00,0x54,0x00,0x53,0x00,0x45,0x30,  
 0x21,0x06,0x03,0x55,0x04,0x07,0x1e,0x1a,0x00,0x4d,0x00,0x69,0x00,0x63,0x00,0x72,  
 0x00,0x6f,0x00,0x73,0x00,0x6f,0x00,0x66,0x00,0x74,0x00,0x2e,0x00,0x63,0x00,0x6f,  
 0x00,0x6d,0x30,0x1e,0x17,0x0d,0x39,0x39,0x30,0x39,0x32,0x34,0x31,0x32,0x30,0x32,  
 0x30,0x34,0x5a,0x17,0x0d,0x34,0x39,0x30,0x39,0x32,0x34,0x31,0x32,0x30,0x32,0x30,  
 0x34,0x5a,0x30,0x38,0x31,0x36,0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x1e,0x0a,0x00,  
 0x4e,0x00,0x54,0x00,0x54,0x00,0x53,0x00,0x45,0x30,0x21,0x06,0x03,0x55,0x04,0x07,  
 0x1e,0x1a,0x00,0x4d,0x00,0x69,0x00,0x63,0x00,0x72,0x00,0x6f,0x00,0x73,0x00,0x6f,  
 0x00,0x66,0x00,0x74,0x00,0x2e,0x00,0x63,0x00,0x6f,0x00,0x6d,0x30,0x5c,0x30,0x0d,  
 0x06,0x09,0x2a,0x86,0x48,0x86,0xf7,0x0d,0x01,0x01,0x01,0x05,0x00,0x03,0x4b,0x00,  
 0x30,0x48,0x02,0x41,0x00,0x91,0xb2,0x16,0x1c,0xae,0x4f,0x7f,0x7c,0xaf,0x57,0x2b,  
 0x23,0x4c,0x0c,0x25,0x3c,0x4f,0x66,0x9d,0x25,0xc3,0x4f,0x29,0xee,0x8b,0xda,0x4e,  
 0x95,0xe7,0x3b,0xaa,0xc0,0xa7,0xba,0xaf,0x99,0x8c,0x47,0x24,0x8b,0x09,0x77,0xbc,  
 0x2c,0xf4,0xe7,0x1a,0x07,0x58,0x7b,0x11,0x37,0x2a,0xa8,0x90,0xc3,0x50,0x92,0x80,  
 0x15,0xc5,0xda,0x51,0x8b,0x02,0x03,0x01,0x00,0x01,0xa3,0x13,0x30,0x11,0x30,0x0f,  
 0x06,0x03,0x55,0x1d,0x13,0x04,0x08,0x30,0x06,0x01,0x01,0xff,0x02,0x01,0x00,0x30,  
 0x09,0x06,0x05,0x2b,0x0e,0x03,0x02,0x1d,0x05,0x00,0x03,0x41,0x00,0x14,0x04,0x67,  
 0x28,0xc8,0xd3,0x1f,0x13,0x14,0x2e,0x2c,0x93,0x09,0x25,0xbb,0xbe,0x86,0x6a,0xd3,  
 0x47,0x6f,0x44,0x16,0x7b,0x94,0x8c,0xb2,0xa2,0xd5,0xf7,0x4f,0xb1,0x8f,0x7f,0xde,  
 0x0b,0x88,0x34,0x4a,0x1d,0xdc,0xa1,0xfd,0x26,0xbd,0x43,0xbb,0x38,0xf1,0x87,0x34,  
 0xbb,0xe9,0x3b,0xfa,0x7f,0x1e,0xff,0xe1,0x10,0x7e,0xee,0x6e,0xd8,0x30,0x82,0x02,  
 0x62,0x30,0x82,0x02,0x10,0xa0,0x03,0x02,0x01,0x02,0x02,0x05,0x01,0x00,0x00,0x00,  
 0x01,0x30,0x09,0x06,0x05,0x2b,0x0e,0x03,0x02,0x1d,0x05,0x00,0x30,0x38,0x31,0x36,  
 0x30,0x11,0x06,0x03,0x55,0x04,0x03,0x1e,0x0a,0x00,0x4e,0x00,0x54,0x00,0x54,0x00,  
 0x53,0x00,0x45,0x30,0x21,0x06,0x03,0x55,0x04,0x07,0x1e,0x1a,0x00,0x4d,0x00,0x69,  
 0x00,0x63,0x00,0x72,0x00,0x6f,0x00,0x73,0x00,0x6f,0x00,0x66,0x00,0x74,0x00,0x2e,  
 0x00,0x63,0x00,0x6f,0x00,0x6d,0x30,0x1e,0x17,0x0d,0x39,0x39,0x30,0x39,0x32,0x34,  
 0x30,0x30,0x30,0x30,0x30,0x30,0x5a,0x17,0x0d,0x34,0x39,0x30,0x39,0x32,0x34,0x30,  
 0x30,0x30,0x30,0x30,0x30,0x5a,0x30,0x79,0x31,0x77,0x30,0x17,0x06,0x03,0x55,0x04,  
 0x03,0x1e,0x10,0x00,0x52,0x00,0x45,0x00,0x53,0x00,0x37,0x00,0x2d,0x00,0x4e,0x00,  
 0x45,0x00,0x57,0x30,0x17,0x06,0x03,0x55,0x04,0x07,0x1e,0x10,0x00,0x7a,0x00,0x32,  
 0x00,0x32,0x00,0x33,0x00,0x32,0x00,0x32,0x00,0x30,0x00,0x33,0x30,0x43,0x06,0x03,  
 0x55,0x04,0x05,0x1e,0x3c,0x00,0x31,0x00,0x42,0x00,0x63,0x00,0x4b,0x00,0x65,0x00,  
 0x57,0x00,0x50,0x00,0x6c,0x00,0x37,0x00,0x58,0x00,0x47,0x00,0x61,0x00,0x73,0x00,  
 0x38,0x00,0x4a,0x00,0x79,0x00,0x50,0x00,0x34,0x00,0x30,0x00,0x7a,0x00,0x49,0x00,  
 0x6d,0x00,0x6e,0x00,0x6f,0x00,0x51,0x00,0x5a,0x00,0x59,0x00,0x3d,0x00,0x0d,0x00,  
 0x0a,0x30,0x5c,0x30,0x0d,0x06,0x09,0x2a,0x86,0x48,0x86,0xf7,0x0d,0x01,0x01,0x01,  
 0x05,0x00,0x03,0x4b,0x00,0x30,0x48,0x02,0x41,0x00,0x91,0xb2,0x16,0x1c,0xae,0x4f,  
 0x7f,0x7c,0xaf,0x57,0x2b,0x23,0x4c,0x0c,0x25,0x3c,0x4f,0x66,0x9d,0x25,0xc3,0x4f,  
 0x29,0xee,0x8b,0xda,0x4e,0x95,0xe7,0x3b,0xaa,0xc0,0xa7,0xba,0xaf,0x99,0x8c,0x47,  
 0x24,0x8b,0x09,0x77,0xbc,0x2c,0xf4,0xe7,0x1a,0x07,0x58,0x7b,0x11,0x37,0x2a,0xa8,  
 0x90,0xc3,0x50,0x92,0x80,0x15,0xc5,0xda,0x51,0x8b,0x02,0x03,0x01,0x00,0x01,0xa3,  
 0x81,0xc3,0x30,0x81,0xc0,0x30,0x14,0x06,0x09,0x2b,0x06,0x01,0x04,0x01,0x82,0x37,  
 0x12,0x04,0x01,0x01,0xff,0x04,0x04,0x01,0x00,0x01,0x00,0x30,0x3c,0x06,0x09,0x2b,  
 0x06,0x01,0x04,0x01,0x82,0x37,0x12,0x02,0x01,0x01,0xff,0x04,0x2c,0x4d,0x00,0x69,  
 0x00,0x63,0x00,0x72,0x00,0x6f,0x00,0x73,0x00,0x6f,0x00,0x66,0x00,0x74,0x00,0x20,  
 0x00,0x43,0x00,0x6f,0x00,0x72,0x00,0x70,0x00,0x6f,0x00,0x72,0x00,0x61,0x00,0x74,  
 0x00,0x69,0x00,0x6f,0x00,0x6e,0x00,0x00,0x00,0x30,0x4c,0x06,0x09,0x2b,0x06,0x01,  
 0x04,0x01,0x82,0x37,0x12,0x05,0x01,0x01,0xff,0x04,0x3c,0x00,0x10,0x00,0x00,0x01,  
 0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x09,0x04,0x00,0x00,0x18,0x00,0x18,0x00,0x30,  
 0x00,0x01,0x00,0x32,0x00,0x33,0x00,0x36,0x00,0x2d,0x00,0x34,0x00,0x2e,0x00,0x30,  
 0x00,0x30,0x00,0x2d,0x00,0x45,0x00,0x58,0x00,0x00,0x00,0x04,0x00,0x00,0x00,0x00,  
 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x30,0x1c,0x06,0x03,0x55,0x1d,0x23,0x01,0x01,  
 0xff,0x04,0x12,0x30,0x10,0xa1,0x07,0x81,0x05,0x4e,0x54,0x54,0x53,0x45,0x82,0x05,  
 0x01,0x00,0x00,0x00,0x01,0x30,0x09,0x06,0x05,0x2b,0x0e,0x03,0x02,0x1d,0x05,0x00,  
 0x03,0x41,0x00,0x7b,0x1d,0xfd,0x24,0xea,0xf2,0xe8,0x17,0xdd,0x88,0x7e,0xfd,0xee,  
 0x28,0x61,0x7a,0x02,0xc3,0x73,0xcf,0x32,0x0f,0x7c,0x66,0x87,0x31,0xa7,0xbe,0x1b,  
 0x31,0xe2,0x20,0xa5,0x76,0x91,0x68,0x97,0x53,0x9e,0x80,0xcd,0x2b,0xd0,0x8e,0x8b,  
 0x7f,0x89,0x1b,0x62,0xa8,0xf8,0xee,0x5e,0x56,0xbd,0x9c,0x6b,0x80,0x06,0x54,0xd3,  
 0xf0,0xbf,0xb2,0x31,0x00,0x01,0x00,0x14,0x00,0xc7,0x32,0xf2,0x5b,0x98,0x0e,0x04,  
 0x49,0xa0,0x27,0x7e,0xf5,0xf6,0x0f,0xda,0x08,0x1d,0xe9,0x79,0xd1,0x31,0xc6,0x50,  
 0x90,0x4a,0xd3,0x1f,0x1d,0xf0,0x65,0x0d,0xb6,0x1f,0xaf,0xc9,0x1d  
 };  
101    
102  /* Send license certificate and related data to the server */          next_packet += length;
103  void rdp_send_cert(HCONN conn)          return rdp_s;
 {  
         mcs_init_data(conn);  
         prs_io_uint8s(&conn->out, precanned_cert_packet, sizeof(precanned_cert_packet));  
         MARK_END(conn->out);  
         mcs_send_data(conn, MCS_GLOBAL_CHANNEL, True);  
104  }  }
105    
106  /* Initialise RDP transport packet */  /* Initialise an RDP data packet */
107  void rdp_init(HCONN conn)  static STREAM
108    rdp_init_data(int maxlen)
109  {  {
110          mcs_init_data(conn);          STREAM s;
         PUSH_LAYER(conn->out, rdp_offset, 6);  
 }  
111    
112  /* Transmit RDP transport packet */          s = sec_init(encryption ? SEC_ENCRYPT : 0, maxlen + 18);
113  void rdp_send(HCONN conn, uint16 pdu_type)          s_push_layer(s, rdp_hdr, 18);
 {  
         RDP_HEADER hdr;  
         int length;  
114    
115          POP_LAYER(conn->out, rdp_offset);          return s;
         length = conn->out.end - conn->out.offset;  
         rdp_make_header(&hdr, length, pdu_type, conn->mcs_userid);  
         rdp_io_header(&conn->out, &hdr);  
         mcs_send_data(conn, MCS_GLOBAL_CHANNEL, True);  
116  }  }
117    
118  /* Initialise RDP transport data packet */  /* Send an RDP data packet */
119  void rdp_init_data(HCONN conn)  static void
120    rdp_send_data(STREAM s, uint8 data_pdu_type)
121  {  {
122          mcs_init_data(conn);          uint16 length;
         PUSH_LAYER(conn->out, rdp_offset, 18);  
 }  
123    
124  /* Transmit RDP transport data packet */          s_pop_layer(s, rdp_hdr);
125  void rdp_send_data(HCONN conn, uint16 data_pdu_type)          length = s->end - s->p;
 {  
         RDP_HEADER hdr;  
         RDP_DATA_HEADER datahdr;  
         int length = conn->out.end - conn->out.offset;  
126    
127          POP_LAYER(conn->out, rdp_offset);          out_uint16_le(s, length);
128          length = conn->out.end - conn->out.offset;          out_uint16_le(s, (RDP_PDU_DATA | 0x10));
129          rdp_make_header(&hdr, length, RDP_PDU_DATA, conn->mcs_userid);          out_uint16_le(s, (mcs_userid + 1001));
         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);  
 }  
130    
131  void rdp_send_confirm_active(HCONN conn)          out_uint32_le(s, rdp_shareid);
132  {          out_uint8(s, 0);        /* pad */
133          RDP_ACTIVE_PDU active;          out_uint8(s, 1);        /* streamid */
134            out_uint16_le(s, (length - 14));
135            out_uint8(s, data_pdu_type);
136            out_uint8(s, 0);        /* compress_type */
137            out_uint16(s, 0);       /* compress_len */
138    
139          rdp_init(conn);          sec_send(s, encryption ? SEC_ENCRYPT : 0);
         rdp_make_active_pdu(&active, 0x103ea, conn->mcs_userid);  
         rdp_io_active_pdu(&conn->out, &active, RDP_PDU_CONFIRM_ACTIVE);  
         MARK_END(conn->out);  
         rdp_send(conn, RDP_PDU_CONFIRM_ACTIVE);  
140  }  }
141    
142  void rdp_send_synchronize(HCONN conn)  /* Output a string in Unicode */
143    void
144    rdp_out_unistr(STREAM s, char *string, int len)
145  {  {
146          RDP_SYNCHRONIZE_PDU sync;          int i = 0, j = 0;
147    
148          rdp_init_data(conn);          len += 2;
         rdp_make_synchronize_pdu(&sync, 1002);  
         rdp_io_synchronize_pdu(&conn->out, &sync);  
         MARK_END(conn->out);  
         rdp_send_data(conn, RDP_DATA_PDU_SYNCHRONIZE);  
 }  
149    
150  void rdp_send_control(HCONN conn, uint16 action)          while (i < len)
151  {          {
152          RDP_CONTROL_PDU control;                  s->p[i++] = string[j++];
153                    s->p[i++] = 0;
154            }
155    
156          rdp_init_data(conn);          s->p += len;
         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);  
157  }  }
158    
159  void rdp_send_fonts(HCONN conn, uint16 seqno)  /* Parse a logon info packet */
160    static void
161    rdp_send_logon_info(uint32 flags, char *domain, char *user,
162                        char *password, char *program, char *directory)
163  {  {
164          RDP_FONT_PDU fonts;          int len_domain = 2 * strlen(domain);
165            int len_user = 2 * strlen(user);
166            int len_password = 2 * strlen(password);
167            int len_program = 2 * strlen(program);
168            int len_directory = 2 * strlen(directory);
169            uint32 sec_flags = encryption ? (SEC_LOGON_INFO | SEC_ENCRYPT) : SEC_LOGON_INFO;
170            STREAM s;
171    
172          rdp_init_data(conn);          s = sec_init(sec_flags, 18 + len_domain + len_user + len_password
173          rdp_make_font_pdu(&fonts, seqno);                       + len_program + len_directory + 10);
         rdp_io_font_pdu(&conn->out, &fonts);  
         MARK_END(conn->out);  
         rdp_send_data(conn, RDP_DATA_PDU_FONT2);  
 }  
174    
175  void rdp_send_input(HCONN conn)          out_uint32(s, 0);
176  {          out_uint32_le(s, flags);
177          RDP_INPUT_PDU input;          out_uint16_le(s, len_domain);
178            out_uint16_le(s, len_user);
179            out_uint16_le(s, len_password);
180            out_uint16_le(s, len_program);
181            out_uint16_le(s, len_directory);
182            rdp_out_unistr(s, domain, len_domain);
183            rdp_out_unistr(s, user, len_user);
184            rdp_out_unistr(s, password, len_password);
185            rdp_out_unistr(s, program, len_program);
186            rdp_out_unistr(s, directory, len_directory);
187    
188          rdp_init_data(conn);          s_mark_end(s);
189          rdp_make_input_pdu(&input);          sec_send(s, sec_flags);
         rdp_io_input_pdu(&conn->out, &input);  
         MARK_END(conn->out);  
         rdp_send_data(conn, RDP_DATA_PDU_INPUT);  
190  }  }
191    
192  BOOL rdp_recv_pdu(HCONN conn, uint8 *type)  /* Send a control PDU */
193    static void
194    rdp_send_control(uint16 action)
195  {  {
196          RDP_HEADER hdr;          STREAM s;
197    
198          if (!mcs_recv(conn, False) || !rdp_io_header(&conn->in, &hdr))          s = rdp_init_data(8);
                 return False;  
199    
200          *type = hdr.pdu_type & 0xf;          out_uint16_le(s, action);
201          return True;          out_uint16(s, 0);       /* userid */
202  }          out_uint32(s, 0);       /* control id */
203    
204  /* Disconnect from the RDP layer */          s_mark_end(s);
205  void rdp_disconnect(HCONN conn)          rdp_send_data(s, RDP_DATA_PDU_CONTROL);
 {  
         mcs_disconnect(conn);  
206  }  }
207    
208  void rdp_make_header(RDP_HEADER *hdr, uint16 length, uint16 pdu_type,  /* Send a synchronisation PDU */
209                       uint16 userid)  static void
210    rdp_send_synchronise(void)
211  {  {
212          hdr->length = length;          STREAM s;
         hdr->pdu_type = pdu_type | 0x10; /* Version 1 */  
         hdr->userid = userid + 1001;  
 }  
213    
214  void rdp_make_data_header(RDP_DATA_HEADER *hdr, uint32 shareid,          s = rdp_init_data(4);
                           uint16 length, uint16 data_pdu_type)  
 {  
         hdr->shareid = shareid;  
         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;  
 }  
215    
216  void rdp_make_general_caps(RDP_GENERAL_CAPS *caps)          out_uint16_le(s, 1);    /* type */
217  {          out_uint16_le(s, 1002);
         caps->os_major_type = 1;  
         caps->os_minor_type = 3;  
         caps->ver_protocol = 0x200;  
 }  
218    
219  void rdp_make_bitmap_caps(RDP_BITMAP_CAPS *caps)          s_mark_end(s);
220  {          rdp_send_data(s, RDP_DATA_PDU_SYNCHRONISE);
         caps->preferred_bpp = 8;  
         caps->receive1bpp = 1;  
         caps->receive4bpp = 1;  
         caps->receive8bpp = 1;  
         caps->width = 640;  
         caps->height = 480;  
         caps->compression = 1;  
         caps->unknown2 = 1;  
221  }  }
222    
223  void rdp_make_order_caps(RDP_ORDER_CAPS *caps)  /* Send a single input event */
224    void
225    rdp_send_input(uint32 time, uint16 message_type, uint16 device_flags, uint16 param1, uint16 param2)
226  {  {
227          caps->xgranularity = 1;          STREAM s;
         caps->ygranularity = 20;  
         caps->max_order_level = 1;  
         caps->num_fonts = 0x147;  
         caps->cap_flags = 0x2A;  
228    
229  //      caps->cap_flags = ORDER_CAP_NEGOTIATE | ORDER_CAP_NOSUPPORT;          s = rdp_init_data(16);
230    
231          caps->support[0] = caps->support[1] = caps->support[2]          out_uint16_le(s, 1);    /* number of events */
232                  = caps->support[3] = caps->support[4] = caps->support[5]          out_uint16(s, 0);       /* pad */
                 = 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;  
 }  
233    
234  void rdp_make_bmpcache_caps(RDP_BMPCACHE_CAPS *caps)          out_uint32_le(s, time);
235  {          out_uint16_le(s, message_type);
236          caps->caches[0].entries = 0x258;          out_uint16_le(s, device_flags);
237          caps->caches[0].max_cell_size = 0x100;          out_uint16_le(s, param1);
238          caps->caches[1].entries = 0x12c;          out_uint16_le(s, param2);
         caps->caches[1].max_cell_size = 0x400;  
         caps->caches[2].entries = 0x106;  
         caps->caches[2].max_cell_size = 0x1000;  
 }  
239    
240  void rdp_make_control_caps(RDP_CONTROL_CAPS *caps)          s_mark_end(s);
241  {          rdp_send_data(s, RDP_DATA_PDU_INPUT);
         caps->control_interest = 2;  
         caps->detach_interest = 2;  
242  }  }
243    
244  void rdp_make_activate_caps(RDP_ACTIVATE_CAPS *caps)  /* Send an (empty) font information PDU */
245    static void
246    rdp_send_fonts(uint16 seq)
247  {  {
248  }          STREAM s;
249    
250  void rdp_make_pointer_caps(RDP_POINTER_CAPS *caps)          s = rdp_init_data(8);
 {  
         caps->colour_pointer = 0;  
         caps->cache_size = 20;  
 }  
251    
252  void rdp_make_share_caps(RDP_SHARE_CAPS *caps, uint16 userid)          out_uint16(s, 0);       /* number of fonts */
253  {          out_uint16_le(s, 0x3e); /* unknown */
254  }          out_uint16_le(s, seq);  /* unknown */
255            out_uint16_le(s, 0x32); /* entry size */
256    
257  void rdp_make_colcache_caps(RDP_COLCACHE_CAPS *caps)          s_mark_end(s);
258  {          rdp_send_data(s, RDP_DATA_PDU_FONT2);
         caps->cache_size = 6;  
259  }  }
260    
261  void rdp_make_active_pdu(RDP_ACTIVE_PDU *pdu, uint32 shareid, uint16 userid)  /* Output general capability set */
262    static void
263    rdp_out_general_caps(STREAM s)
264  {  {
265          memset(pdu, 0, sizeof(*pdu));          out_uint16_le(s, RDP_CAPSET_GENERAL);
266          pdu->shareid = shareid;          out_uint16_le(s, RDP_CAPLEN_GENERAL);
         pdu->userid  = 1002;  
         pdu->source_len = sizeof(RDP_SOURCE);  
         memcpy(pdu->source, RDP_SOURCE, sizeof(RDP_SOURCE));  
267    
268          pdu->caps_len = RDP_CAPLEN_GENERAL + RDP_CAPLEN_BITMAP + RDP_CAPLEN_ORDER          out_uint16_le(s, 1);    /* OS major type */
269                  + RDP_CAPLEN_BMPCACHE + RDP_CAPLEN_COLCACHE + RDP_CAPLEN_ACTIVATE          out_uint16_le(s, 3);    /* OS minor type */
270                  + RDP_CAPLEN_CONTROL + RDP_CAPLEN_POINTER + RDP_CAPLEN_SHARE          out_uint16_le(s, 0x200);        /* Protocol version */
271                  + RDP_CAPLEN_UNKNOWN;          out_uint16(s, 0);       /* Pad */
272          pdu->num_caps = 0xD;          out_uint16(s, 0);       /* Compression types */
273            out_uint16(s, 0);       /* Pad */
274          rdp_make_general_caps (&pdu->general_caps );          out_uint16(s, 0);       /* Update capability */
275          rdp_make_bitmap_caps  (&pdu->bitmap_caps  );          out_uint16(s, 0);       /* Remote unshare capability */
276          rdp_make_order_caps   (&pdu->order_caps   );          out_uint16(s, 0);       /* Compression level */
277          rdp_make_bmpcache_caps(&pdu->bmpcache_caps);          out_uint16(s, 0);       /* Pad */
         rdp_make_control_caps (&pdu->control_caps );  
         rdp_make_activate_caps(&pdu->activate_caps);  
         rdp_make_pointer_caps (&pdu->pointer_caps );  
         rdp_make_share_caps   (&pdu->share_caps, userid);  
         rdp_make_colcache_caps(&pdu->colcache_caps);  
278  }  }
279    
280  void rdp_make_control_pdu(RDP_CONTROL_PDU *pdu, uint16 action)  /* Output bitmap capability set */
281    static void
282    rdp_out_bitmap_caps(STREAM s)
283  {  {
284          pdu->action = action;          out_uint16_le(s, RDP_CAPSET_BITMAP);
285          pdu->userid = 0;          out_uint16_le(s, RDP_CAPLEN_BITMAP);
         pdu->controlid = 0;  
 }  
286    
287  void rdp_make_synchronize_pdu(RDP_SYNCHRONIZE_PDU *pdu, uint16 userid)          out_uint16_le(s, 8);    /* Preferred BPP */
288  {          out_uint16_le(s, 1);    /* Receive 1 BPP */
289          pdu->type = 1;          out_uint16_le(s, 1);    /* Receive 4 BPP */
290          pdu->userid = userid;          out_uint16_le(s, 1);    /* Receive 8 BPP */
291  }          out_uint16_le(s, 800);  /* Desktop width */
292            out_uint16_le(s, 600);  /* Desktop height */
293  void rdp_make_font_pdu(RDP_FONT_PDU *pdu, uint16 seqno)          out_uint16(s, 0);       /* Pad */
294  {          out_uint16(s, 0);       /* Allow resize */
295          pdu->num_fonts = 0;          out_uint16_le(s, bitmap_compression ? 1 : 0);   /* Support compression */
296          pdu->unknown1 = 0x3e;          out_uint16(s, 0);       /* Unknown */
297          pdu->unknown2 = seqno;          out_uint16_le(s, 1);    /* Unknown */
298          pdu->entry_size = RDP_FONT_INFO_SIZE;          out_uint16(s, 0);       /* Pad */
299  }  }
300    
301  void rdp_make_input_pdu(RDP_INPUT_PDU *pdu)  /* Output order capability set */
302    static void
303    rdp_out_order_caps(STREAM s)
304  {  {
305          uint32 now = time(NULL);          uint8 order_caps[32];
306    
         pdu->num_events = 3;  
         pdu->pad = 0;  
307    
308          pdu->event[0].event_time = now;          memset(order_caps, 0, 32);
309          pdu->event[0].message_type = RDP_INPUT_SYNCHRONIZE;          order_caps[0] = 1;      /* dest blt */
310          pdu->event[0].device_flags = 0;          order_caps[1] = 1;      /* pat blt */
311          pdu->event[0].mouse_x = 0;          order_caps[2] = 1;      /* screen blt */
312          pdu->event[0].mouse_y = 0;          order_caps[3] = 1;      /* required for memblt? */
313            order_caps[8] = 1;      /* line */
314            order_caps[9] = 1;      /* line */
315            order_caps[10] = 1;     /* rect */
316            order_caps[11] = (desktop_save == False ? 0 : 1);       /* desksave */
317            order_caps[13] = 1;     /* memblt */
318            order_caps[14] = 1;     /* triblt */
319            order_caps[22] = 1;     /* polyline */
320            order_caps[27] = 1;     /* text2 */
321            out_uint16_le(s, RDP_CAPSET_ORDER);
322            out_uint16_le(s, RDP_CAPLEN_ORDER);
323    
324          pdu->event[1].event_time = now;          out_uint8s(s, 20);      /* Terminal desc, pad */
325          pdu->event[1].message_type = RDP_INPUT_UNKNOWN;          out_uint16_le(s, 1);    /* Cache X granularity */
326          pdu->event[1].device_flags = 0x8000;          out_uint16_le(s, 20);   /* Cache Y granularity */
327          pdu->event[1].mouse_x = 15;          out_uint16(s, 0);       /* Pad */
328          pdu->event[1].mouse_y = 0;          out_uint16_le(s, 1);    /* Max order level */
329            out_uint16_le(s, 0x147);        /* Number of fonts */
330          pdu->event[2].event_time = now;          out_uint16_le(s, 0x2a); /* Capability flags */
331          pdu->event[2].message_type = RDP_INPUT_MOUSE;          out_uint8p(s, order_caps, 32);  /* Orders supported */
332          pdu->event[2].device_flags = MOUSE_FLAG_MOVE;          out_uint16_le(s, 0x6a1);        /* Text capability flags */
333          pdu->event[2].mouse_x = 425;          out_uint8s(s, 6);       /* Pad */
334          pdu->event[2].mouse_y = 493;          out_uint32_le(s, desktop_save == False ? 0 : 0x38400);  /* Desktop cache size */
335            out_uint32(s, 0);       /* Unknown */
336            out_uint32_le(s, 0x4e4);        /* Unknown */
337  }  }
338    
339  BOOL rdp_io_header(STREAM s, RDP_HEADER *hdr)  /* Output bitmap cache capability set */
340    static void
341    rdp_out_bmpcache_caps(STREAM s)
342  {  {
343          BOOL res = True;          out_uint16_le(s, RDP_CAPSET_BMPCACHE);
344            out_uint16_le(s, RDP_CAPLEN_BMPCACHE);
         res = res ? lsb_io_uint16(s, &hdr->length  ) : False;  
         res = res ? lsb_io_uint16(s, &hdr->pdu_type) : False;  
         res = res ? lsb_io_uint16(s, &hdr->userid  ) : False;  
345    
346          return res;          out_uint8s(s, 24);      /* unused */
347            out_uint16_le(s, 0x258);        /* entries */
348            out_uint16_le(s, 0x100);        /* max cell size */
349            out_uint16_le(s, 0x12c);        /* entries */
350            out_uint16_le(s, 0x400);        /* max cell size */
351            out_uint16_le(s, 0x106);        /* entries */
352            out_uint16_le(s, 0x1000);       /* max cell size */
353  }  }
354    
355  BOOL rdp_io_data_header(STREAM s, RDP_DATA_HEADER *hdr)  /* Output control capability set */
356    static void
357    rdp_out_control_caps(STREAM s)
358  {  {
359          BOOL res = True;          out_uint16_le(s, RDP_CAPSET_CONTROL);
360            out_uint16_le(s, RDP_CAPLEN_CONTROL);
361    
362          res = res ? lsb_io_uint32(s, &hdr->shareid      ) : False;          out_uint16(s, 0);       /* Control capabilities */
363          res = res ? prs_io_uint8 (s, &hdr->pad          ) : False;          out_uint16(s, 0);       /* Remote detach */
364          res = res ? prs_io_uint8 (s, &hdr->streamid     ) : False;          out_uint16_le(s, 2);    /* Control interest */
365          res = res ? lsb_io_uint16(s, &hdr->length       ) : False;          out_uint16_le(s, 2);    /* Detach interest */
         res = res ? prs_io_uint8 (s, &hdr->data_pdu_type) : False;  
         res = res ? prs_io_uint8 (s, &hdr->compress_type) : False;  
         res = res ? lsb_io_uint16(s, &hdr->compress_len ) : False;  
   
         return res;  
366  }  }
367    
368  BOOL rdp_io_general_caps(STREAM s, RDP_GENERAL_CAPS *caps)  /* Output activation capability set */
369    static void
370    rdp_out_activate_caps(STREAM s)
371  {  {
372          uint16 length = RDP_CAPLEN_GENERAL;          out_uint16_le(s, RDP_CAPSET_ACTIVATE);
373          uint16 pkt_length = length;          out_uint16_le(s, RDP_CAPLEN_ACTIVATE);
         BOOL res;  
   
         res = lsb_io_uint16(s, &pkt_length);  
         if (pkt_length != length)  
         {  
                 fprintf(stderr, "Unrecognised capabilities size\n");  
                 return False;  
         }  
   
         res = res ? lsb_io_uint16(s, &caps->os_major_type ) : False;  
         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;  
 }  
   
 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)  
         {  
                 fprintf(stderr, "Unrecognised capabilities size\n");  
                 return False;  
         }  
374    
375          res = res ? lsb_io_uint16(s, &caps->preferred_bpp) : False;          out_uint16(s, 0);       /* Help key */
376          res = res ? lsb_io_uint16(s, &caps->receive1bpp  ) : False;          out_uint16(s, 0);       /* Help index key */
377          res = res ? lsb_io_uint16(s, &caps->receive4bpp  ) : False;          out_uint16(s, 0);       /* Extended help key */
378          res = res ? lsb_io_uint16(s, &caps->receive8bpp  ) : False;          out_uint16(s, 0);       /* Window activate */
         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;  
 }  
   
 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);  
         if (pkt_length != length)  
         {  
                 fprintf(stderr, "Unrecognised capabilities size\n");  
                 return False;  
         }  
   
         res = res ? prs_io_uint8s(s,  caps->terminal_desc, 16) : False;  
         res = res ? lsb_io_uint32(s, &caps->pad1             ) : False;  
         res = res ? lsb_io_uint16(s, &caps->xgranularity     ) : False;  
         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;  
379  }  }
380    
381  BOOL rdp_io_bmpcache_info(STREAM s, RDP_BMPCACHE_INFO *info)  /* Output pointer capability set */
382    static void
383    rdp_out_pointer_caps(STREAM s)
384  {  {
385          if (!lsb_io_uint16(s, &info->entries      ))          out_uint16_le(s, RDP_CAPSET_POINTER);
386                  return False;          out_uint16_le(s, RDP_CAPLEN_POINTER);
   
         if (!lsb_io_uint16(s, &info->max_cell_size))  
                 return False;  
387    
388          return True;          out_uint16(s, 0);       /* Colour pointer */
389            out_uint16_le(s, 20);   /* Cache size */
390  }  }
391    
392  BOOL rdp_io_bmpcache_caps(STREAM s, RDP_BMPCACHE_CAPS *caps)  /* Output share capability set */
393    static void
394    rdp_out_share_caps(STREAM s)
395  {  {
396          uint16 length = RDP_CAPLEN_BMPCACHE;          out_uint16_le(s, RDP_CAPSET_SHARE);
397          uint16 pkt_length = length;          out_uint16_le(s, RDP_CAPLEN_SHARE);
         BOOL res;  
         int i;  
   
         res = lsb_io_uint16(s, &pkt_length);  
         if (pkt_length != length)  
         {  
                 fprintf(stderr, "Unrecognised capabilities size\n");  
                 return False;  
         }  
   
         for (i = 0; i < 6; i++)  
                 res = res ? lsb_io_uint32(s, &caps->unused[i]) : False;  
398    
399          for (i = 0; i < 3; i++)          out_uint16(s, 0);       /* userid */
400                  res = res ? rdp_io_bmpcache_info(s, &caps->caches[i]) : False;          out_uint16(s, 0);       /* pad */
   
         return res;  
401  }  }
402    
403  BOOL rdp_io_control_caps(STREAM s, RDP_CONTROL_CAPS *caps)  /* Output colour cache capability set */
404    static void
405    rdp_out_colcache_caps(STREAM s)
406  {  {
407          uint16 length = RDP_CAPLEN_CONTROL;          out_uint16_le(s, RDP_CAPSET_COLCACHE);
408          uint16 pkt_length = length;          out_uint16_le(s, RDP_CAPLEN_COLCACHE);
         BOOL res;  
   
         res = lsb_io_uint16(s, &pkt_length);  
         if (pkt_length != length)  
         {  
                 fprintf(stderr, "Unrecognised capabilities size\n");  
                 return False;  
         }  
409    
410          res = res ? lsb_io_uint16(s, &caps->control_caps    ) : False;          out_uint16_le(s, 6);    /* cache size */
411          res = res ? lsb_io_uint16(s, &caps->remote_detach   ) : False;          out_uint16(s, 0);       /* pad */
         res = res ? lsb_io_uint16(s, &caps->control_interest) : False;  
         res = res ? lsb_io_uint16(s, &caps->detach_interest ) : False;  
   
         return res;  
412  }  }
413    
414  BOOL rdp_io_activate_caps(STREAM s, RDP_ACTIVATE_CAPS *caps)  static uint8 canned_caps[] = {
415  {          0x01, 0x00, 0x00, 0x00, 0x09, 0x04, 0x00, 0x00, 0x04,
416          uint16 length = RDP_CAPLEN_ACTIVATE;          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0C, 0x00, 0x00, 0x00,
417          uint16 pkt_length = length;          0x00, 0x00, 0x00, 0x00, 0x00,
418          BOOL res;          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
419            0x00, 0x00, 0x00, 0x00, 0x00,
420          res = lsb_io_uint16(s, &pkt_length);          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
421          if (pkt_length != length)          0x00, 0x00, 0x00, 0x00, 0x00,
422          {          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
423                  fprintf(stderr, "Unrecognised capabilities size\n");          0x00, 0x00, 0x00, 0x00, 0x00,
424                  return False;          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
425          }          0x0C, 0x00, 0x08, 0x00, 0x01,
426            0x00, 0x00, 0x00, 0x0E, 0x00, 0x08, 0x00, 0x01, 0x00, 0x00, 0x00,
427          res = res ? lsb_io_uint16(s, &caps->help_key         ) : False;          0x10, 0x00, 0x34, 0x00, 0xFE,
428          res = res ? lsb_io_uint16(s, &caps->help_index_key   ) : False;          0x00, 0x04, 0x00, 0xFE, 0x00, 0x04, 0x00, 0xFE, 0x00, 0x08, 0x00,
429          res = res ? lsb_io_uint16(s, &caps->help_extended_key) : False;          0xFE, 0x00, 0x08, 0x00, 0xFE,
430          res = res ? lsb_io_uint16(s, &caps->window_activate  ) : False;          0x00, 0x10, 0x00, 0xFE, 0x00, 0x20, 0x00, 0xFE, 0x00, 0x40, 0x00,
431            0xFE, 0x00, 0x80, 0x00, 0xFE,
432          return res;          0x00, 0x00, 0x01, 0x40, 0x00, 0x00, 0x08, 0x00, 0x01, 0x00, 0x01,
433  }          0x02, 0x00, 0x00, 0x00
434    };
 BOOL rdp_io_pointer_caps(STREAM s, RDP_POINTER_CAPS *caps)  
 {  
         uint16 length = RDP_CAPLEN_POINTER;  
         uint16 pkt_length = length;  
         BOOL res;  
   
         res = lsb_io_uint16(s, &pkt_length);  
         if (pkt_length != length)  
         {  
                 fprintf(stderr, "Unrecognised capabilities size\n");  
                 return False;  
         }  
   
         res = res ? lsb_io_uint16(s, &caps->colour_pointer) : False;  
         res = res ? lsb_io_uint16(s, &caps->cache_size    ) : False;  
   
         return res;  
 }  
   
 BOOL rdp_io_share_caps(STREAM s, RDP_SHARE_CAPS *caps)  
 {  
         uint16 length = RDP_CAPLEN_SHARE;  
         uint16 pkt_length = length;  
         BOOL res;  
   
         res = lsb_io_uint16(s, &pkt_length);  
         if (pkt_length != length)  
         {  
                 fprintf(stderr, "Unrecognised capabilities size\n");  
                 return False;  
         }  
   
         res = res ? lsb_io_uint16(s, &caps->userid) : False;  
         res = res ? lsb_io_uint16(s, &caps->pad   ) : False;  
   
         return res;  
 }  
435    
436  BOOL rdp_io_colcache_caps(STREAM s, RDP_COLCACHE_CAPS *caps)  /* Output unknown capability set */
437    static void
438    rdp_out_unknown_caps(STREAM s)
439    {
440            out_uint16_le(s, RDP_CAPSET_UNKNOWN);
441            out_uint16_le(s, 0x58);
442    
443            out_uint8p(s, canned_caps, RDP_CAPLEN_UNKNOWN - 4);
444    }
445    
446    /* Send a confirm active PDU */
447    static void
448    rdp_send_confirm_active(void)
449    {
450            STREAM s;
451            uint16 caplen =
452                    RDP_CAPLEN_GENERAL + RDP_CAPLEN_BITMAP + RDP_CAPLEN_ORDER +
453                    RDP_CAPLEN_BMPCACHE + RDP_CAPLEN_COLCACHE +
454                    RDP_CAPLEN_ACTIVATE + RDP_CAPLEN_CONTROL +
455                    RDP_CAPLEN_POINTER + RDP_CAPLEN_SHARE + RDP_CAPLEN_UNKNOWN + 4 /* w2k fix, why? */ ;
456    
457            s = rdp_init(14 + caplen + sizeof(RDP_SOURCE));
458    
459            out_uint32_le(s, rdp_shareid);
460            out_uint16_le(s, 0x3ea);        /* userid */
461            out_uint16_le(s, sizeof(RDP_SOURCE));
462            out_uint16_le(s, caplen);
463    
464            out_uint8p(s, RDP_SOURCE, sizeof(RDP_SOURCE));
465            out_uint16_le(s, 0xd);  /* num_caps */
466            out_uint8s(s, 2);       /* pad */
467    
468            rdp_out_general_caps(s);
469            rdp_out_bitmap_caps(s);
470            rdp_out_order_caps(s);
471            rdp_out_bmpcache_caps(s);
472            rdp_out_colcache_caps(s);
473            rdp_out_activate_caps(s);
474            rdp_out_control_caps(s);
475            rdp_out_pointer_caps(s);
476            rdp_out_share_caps(s);
477            rdp_out_unknown_caps(s);
478    
479            s_mark_end(s);
480            rdp_send(s, RDP_PDU_CONFIRM_ACTIVE);
481    }
482    
483    /* Respond to a demand active PDU */
484    static void
485    process_demand_active(STREAM s)
486  {  {
487          uint16 length = RDP_CAPLEN_COLCACHE;          uint8 type;
         uint16 pkt_length = length;  
         BOOL res;  
488    
489          res = lsb_io_uint16(s, &pkt_length);          in_uint32_le(s, rdp_shareid);
         if (pkt_length != length)  
         {  
                 fprintf(stderr, "Unrecognised capabilities size\n");  
                 return False;  
         }  
490    
491          res = res ? lsb_io_uint16(s, &caps->cache_size) : False;          DEBUG(("DEMAND_ACTIVE(id=0x%x)\n", rdp_shareid));
         res = res ? lsb_io_uint16(s, &caps->pad       ) : False;  
492    
493          return res;          rdp_send_confirm_active();
494  }          rdp_send_synchronise();
495            rdp_send_control(RDP_CTL_COOPERATE);
496            rdp_send_control(RDP_CTL_REQUEST_CONTROL);
497            rdp_recv(&type);        /* RDP_PDU_SYNCHRONIZE */
498            rdp_recv(&type);        /* RDP_CTL_COOPERATE */
499            rdp_recv(&type);        /* RDP_CTL_GRANT_CONTROL */
500            rdp_send_input(0, RDP_INPUT_SYNCHRONIZE, 0, 0, 0);
501            rdp_send_fonts(1);
502            rdp_send_fonts(2);
503            rdp_recv(&type);        /* RDP_PDU_UNKNOWN 0x28 */
504            reset_order_state();
505    }
506    
507    /* Process a pointer PDU */
508    static void
509    process_pointer_pdu(STREAM s)
510    {
511            uint16 message_type;
512            uint16 x, y, width, height, cache_idx, masklen, datalen;
513            uint8 *mask, *data;
514            HCURSOR cursor;
515    
516            in_uint16_le(s, message_type);
517            in_uint8s(s, 2);        /* pad */
518    
519            switch (message_type)
520            {
521                    case RDP_POINTER_MOVE:
522                            in_uint16_le(s, x);
523                            in_uint16_le(s, y);
524                            if (s_check(s))
525                                    ui_move_pointer(x, y);
526                            break;
527    
528  uint8 canned_caps[] = {                  case RDP_POINTER_COLOR:
529  0x01,0x00,0x00,0x00,0x09,0x04,0x00,0x00,0x04,                          in_uint16_le(s, cache_idx);
530  0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x0C,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,                          in_uint16_le(s, x);
531  0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,                          in_uint16_le(s, y);
532  0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,                          in_uint16_le(s, width);
533  0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,                          in_uint16_le(s, height);
534  0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x0C,0x00,0x08,0x00,0x01,                          in_uint16_le(s, masklen);
535  0x00,0x00,0x00,0x0E,0x00,0x08,0x00,0x01,0x00,0x00,0x00,0x10,0x00,0x34,0x00,0xFE,                          in_uint16_le(s, datalen);
536  0x00,0x04,0x00,0xFE,0x00,0x04,0x00,0xFE,0x00,0x08,0x00,0xFE,0x00,0x08,0x00,0xFE,                          in_uint8p(s, data, datalen);
537  0x00,0x10,0x00,0xFE,0x00,0x20,0x00,0xFE,0x00,0x40,0x00,0xFE,0x00,0x80,0x00,0xFE,                          in_uint8p(s, mask, masklen);
538  0x00,0x00,0x01,0x40,0x00,0x00,0x08,0x00,0x01,0x00,0x01,0x02,0x00,0x00,0x00                          cursor = ui_create_cursor(x, y, width, height, mask, data);
539  };                          ui_set_cursor(cursor);
540                            cache_put_cursor(cache_idx, cursor);
541                            break;
542    
543  BOOL rdp_io_unknown_caps(STREAM s, void *caps)                  case RDP_POINTER_CACHED:
544  {                          in_uint16_le(s, cache_idx);
545          uint16 length = 0x58;                          ui_set_cursor(cache_get_cursor(cache_idx));
546          uint16 pkt_length = length;                          break;
         BOOL res;  
547    
548          res = lsb_io_uint16(s, &pkt_length);                  default:
549          if (pkt_length != length)                          DEBUG(("Pointer message 0x%x\n", message_type));
         {  
                 fprintf(stderr, "Unrecognised capabilities size\n");  
                 return False;  
550          }          }
   
         res = res ? prs_io_uint8s(s, canned_caps, RDP_CAPLEN_UNKNOWN-4) : False;  
   
         return res;  
551  }  }
552    
553  BOOL rdp_io_active_pdu(STREAM s, RDP_ACTIVE_PDU *pdu, int pdutype)  /* Process bitmap updates */
554    static void
555    process_bitmap_updates(STREAM s)
556  {  {
557          uint16 capset;          uint16 num_updates;
558          uint16 length;          uint16 left, top, right, bottom, width, height;
559          BOOL res;          uint16 cx, cy, bpp, Bpp, compress, bufsize, size;
560            uint8 *data, *bmpdata;
561          int i;          int i;
562    
563          res = lsb_io_uint32(s, &pdu->shareid);          in_uint16_le(s, num_updates);
564    
565          if (pdutype == RDP_PDU_CONFIRM_ACTIVE)          for (i = 0; i < num_updates; i++)
                 res = res ? lsb_io_uint16(s, &pdu->userid    ) : False;  
   
         res = res ? lsb_io_uint16(s, &pdu->source_len) : False;  
         res = res ? lsb_io_uint16(s, &pdu->caps_len  ) : False;  
   
         if (pdu->source_len > 48)  
566          {          {
567                  fprintf(stderr, "RDP source descriptor too long\n");                  in_uint16_le(s, left);
568                  return False;                  in_uint16_le(s, top);
569          }                  in_uint16_le(s, right);
570                    in_uint16_le(s, bottom);
571                    in_uint16_le(s, width);
572                    in_uint16_le(s, height);
573                    in_uint16_le(s, bpp);
574                    Bpp = (bpp + 7) / 8;
575                    in_uint16_le(s, compress);
576                    in_uint16_le(s, bufsize);
577    
578          res = res ? prs_io_uint8s(s,  pdu->source, pdu->source_len) : False;                  cx = right - left + 1;
579          res = res ? lsb_io_uint16(s, &pdu->num_caps  ) : False;                  cy = bottom - top + 1;
         res = res ? lsb_io_uint16(s, &pdu->pad       ) : False;  
   
         if (s->marshall)  
         {  
                 capset = RDP_CAPSET_GENERAL;  
                 res = res ? lsb_io_uint16(s, &capset) : False;  
                 res = res ? rdp_io_general_caps(s, &pdu->general_caps) : False;  
   
                 capset = RDP_CAPSET_BITMAP;  
                 res = res ? lsb_io_uint16(s, &capset) : False;  
                 res = res ? rdp_io_bitmap_caps (s, &pdu->bitmap_caps ) : False;  
   
                 capset = RDP_CAPSET_ORDER;  
                 res = res ? lsb_io_uint16(s, &capset) : False;  
                 res = res ? rdp_io_order_caps  (s, &pdu->order_caps  ) : False;  
   
                 capset = RDP_CAPSET_BMPCACHE;  
                 res = res ? lsb_io_uint16(s, &capset) : False;  
                 res = res ? rdp_io_bmpcache_caps(s, &pdu->bmpcache_caps) : False;  
   
                 capset = RDP_CAPSET_COLCACHE;  
                 res = res ? lsb_io_uint16(s, &capset) : False;  
                 res = res ? rdp_io_colcache_caps(s, &pdu->colcache_caps) : False;  
   
                 capset = RDP_CAPSET_ACTIVATE;  
                 res = res ? lsb_io_uint16(s, &capset) : False;  
                 res = res ? rdp_io_activate_caps(s, &pdu->activate_caps) : False;  
   
                 capset = RDP_CAPSET_CONTROL;  
                 res = res ? lsb_io_uint16(s, &capset) : False;  
                 res = res ? rdp_io_control_caps(s, &pdu->control_caps) : False;  
   
                 capset = RDP_CAPSET_POINTER;  
                 res = res ? lsb_io_uint16(s, &capset) : False;  
                 res = res ? rdp_io_pointer_caps(s, &pdu->pointer_caps) : False;  
   
                 capset = RDP_CAPSET_SHARE;  
                 res = res ? lsb_io_uint16(s, &capset) : False;  
                 res = res ? rdp_io_share_caps  (s, &pdu->share_caps  ) : False;  
   
                 capset = RDP_CAPSET_UNKNOWN;  
                 res = res ? lsb_io_uint16(s, &capset) : False;  
                 res = res ? rdp_io_unknown_caps(s, NULL) : False;  
         }  
         else  
         {  
                 for (i = 0; i < pdu->num_caps; i++)  
                 {  
                         if (!res)  
                                 return False;  
580    
581                          if (!lsb_io_uint16(s, &capset))                  DEBUG(("UPDATE(l=%d,t=%d,r=%d,b=%d,w=%d,h=%d,cmp=%d)\n",
582                                  return False;                         left, top, right, bottom, width, height, compress));
583    
584                          switch (capset)                  if (!compress)
585                    {
586                            int y;
587                            bmpdata = xmalloc(width * height * Bpp);
588                            for (y = 0; y < height; y++)
589                          {                          {
590                          case RDP_CAPSET_GENERAL:                                  in_uint8a(s, &bmpdata[(height - y - 1) * (width * Bpp)],
591                                  res = rdp_io_general_caps (s, &pdu->general_caps );                                            width * Bpp);
                                 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:  
                                 fprintf(stderr, "Warning: Unrecognised capset %x\n",  
                                         capset);  
   
                                 if (!lsb_io_uint16(s, &length))  
                                         return False;  
   
                                 s->offset += (length - 4);  
592                          }                          }
593                            ui_paint_bitmap(left, top, cx, cy, width, height, bmpdata);
594                            xfree(bmpdata);
595                            continue;
596                  }                  }
         }  
597    
598          return res;                  in_uint8s(s, 2);        /* pad */
599                    in_uint16_le(s, size);
600                    in_uint8s(s, 4);        /* line_size, final_size */
601                    in_uint8p(s, data, size);
602                    bmpdata = xmalloc(width * height * Bpp);
603                    if (bitmap_decompress(bmpdata, width, height, data, size, Bpp))
604                    {
605                            ui_paint_bitmap(left, top, cx, cy, width, height, bmpdata);
606                    }
607                    xfree(bmpdata);
608            }
609  }  }
610    
611  BOOL rdp_io_control_pdu(STREAM s, RDP_CONTROL_PDU *pdu)  /* Process a palette update */
612    static void
613    process_palette(STREAM s)
614  {  {
615          BOOL res = True;          COLOURENTRY *entry;
616            COLOURMAP map;
617            HCOLOURMAP hmap;
618            int i;
619    
620          res = res ? lsb_io_uint16(s, &pdu->action   ) : False;          in_uint8s(s, 2);        /* pad */
621          res = res ? lsb_io_uint16(s, &pdu->userid   ) : False;          in_uint16_le(s, map.ncolours);
622          res = res ? lsb_io_uint32(s, &pdu->controlid) : False;          in_uint8s(s, 2);        /* pad */
623    
624          return res;          map.colours = xmalloc(3 * map.ncolours);
 }  
625    
626  BOOL rdp_io_synchronize_pdu(STREAM s, RDP_SYNCHRONIZE_PDU *pdu)          for (i = 0; i < map.ncolours; i++)
627  {          {
628          BOOL res = True;                  entry = &map.colours[i];
629                    in_uint8(s, entry->red);
630                    in_uint8(s, entry->green);
631                    in_uint8(s, entry->blue);
632            }
633    
634          res = res ? lsb_io_uint16(s, &pdu->type  ) : False;          hmap = ui_create_colourmap(&map);
635          res = res ? lsb_io_uint16(s, &pdu->userid) : False;          ui_set_colourmap(hmap);
636    
637          return res;          xfree(map.colours);
638  }  }
639    
640  BOOL rdp_io_input_event(STREAM s, RDP_INPUT_EVENT *evt)  /* Process an update PDU */
641    static void
642    process_update_pdu(STREAM s)
643  {  {
644          BOOL res = True;          uint16 update_type;
645    
646          res = res ? lsb_io_uint32(s, &evt->event_time)   : False;          in_uint16_le(s, update_type);
         res = res ? lsb_io_uint16(s, &evt->message_type) : False;  
   
         if (!res)  
                 return False;  
647    
648          switch (evt->message_type)          switch (update_type)
649          {          {
650          case RDP_INPUT_CODEPOINT:                  case RDP_UPDATE_ORDERS:
651          case RDP_INPUT_VIRTKEY:                          process_orders(s);
652                  res = res ? lsb_io_uint16(s, &evt->device_flags) : False;                          break;
                 res = res ? lsb_io_uint16(s, &evt->kbd_keycode ) : False;  
                 break;  
         case RDP_INPUT_SYNCHRONIZE:  
         case RDP_INPUT_UNKNOWN:  
         case RDP_INPUT_MOUSE:  
                 res = res ? lsb_io_uint16(s, &evt->device_flags) : False;  
                 res = res ? lsb_io_uint16(s, &evt->mouse_x     ) : False;  
                 res = res ? lsb_io_uint16(s, &evt->mouse_y     ) : False;  
                 break;  
         default:  
                 fprintf(stderr, "Unknown input type %d\n", evt->message_type);  
                 return False;  
         }  
   
         return res;  
 }  
653    
654  BOOL rdp_io_input_pdu(STREAM s, RDP_INPUT_PDU *pdu)                  case RDP_UPDATE_BITMAP:
655  {                          process_bitmap_updates(s);
656          BOOL res = True;                          break;
         int i;  
657    
658          res = res ? lsb_io_uint16(s, &pdu->num_events) : False;                  case RDP_UPDATE_PALETTE:
659          res = res ? lsb_io_uint16(s, &pdu->pad       ) : False;                          process_palette(s);
660                            break;
661    
662          if (pdu->num_events > RDP_MAX_EVENTS)                  case RDP_UPDATE_SYNCHRONIZE:
663          {                          break;
                 fprintf(stderr, "Too many events in one PDU\n");  
                 return False;  
         }  
664    
665          for (i = 0; i < pdu->num_events; i++)                  default:
666          {                          unimpl("update %d\n", update_type);
                 res = res ? rdp_io_input_event(s, &pdu->event[i]) : False;  
667          }          }
668    
         return res;  
669  }  }
670    
671  BOOL rdp_io_font_info(STREAM s, RDP_FONT_INFO *font)  /* Process data PDU */
672    static void
673    process_data_pdu(STREAM s)
674  {  {
675          BOOL res = True;          uint8 data_pdu_type;
676    
677          res = res ? prs_io_uint8s(s,  font->name, 32 ) : False;          in_uint8s(s, 8);        /* shareid, pad, streamid, length */
678          res = res ? lsb_io_uint16(s, &font->flags    ) : False;          in_uint8(s, data_pdu_type);
679          res = res ? lsb_io_uint16(s, &font->width    ) : False;          in_uint8s(s, 3);        /* compress_type, compress_len */
         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;  
680    
681          return res;          switch (data_pdu_type)
682  }          {
683                    case RDP_DATA_PDU_UPDATE:
684                            process_update_pdu(s);
685                            break;
686    
687  BOOL rdp_io_font_pdu(STREAM s, RDP_FONT_PDU *pdu)                  case RDP_DATA_PDU_POINTER:
688  {                          process_pointer_pdu(s);
689          BOOL res = True;                          break;
         int i;  
690    
691          res = res ? lsb_io_uint16(s, &pdu->num_fonts ) : False;                  case RDP_DATA_PDU_BELL:
692          res = res ? lsb_io_uint16(s, &pdu->unknown1  ) : False;                          ui_bell();
693          res = res ? lsb_io_uint16(s, &pdu->unknown2  ) : False;                          break;
         res = res ? lsb_io_uint16(s, &pdu->entry_size) : False;  
694    
695          if (pdu->num_fonts > RDP_MAX_FONTS)                  case RDP_DATA_PDU_LOGON:
696          {                          /* User logged on */
697                  fprintf(stderr, "Too many fonts in one PDU\n");                          break;
                 return False;  
         }  
698    
699          for (i = 0; i < pdu->num_fonts; i++)                  default:
700          {                          unimpl("data PDU %d\n", data_pdu_type);
                 res = res ? rdp_io_font_info(s, &pdu->font[i]) : False;  
701          }          }
   
         return res;  
702  }  }
703    
704  BOOL rdp_io_update_pdu(STREAM s, RDP_UPDATE_PDU *pdu)  /* Process incoming packets */
705    void
706    rdp_main_loop(void)
707  {  {
708          BOOL res = True;          uint8 type;
709            STREAM s;
710    
711          res = res ? lsb_io_uint16(s, &pdu->update_type) : False;          while ((s = rdp_recv(&type)) != NULL)
712          res = res ? lsb_io_uint16(s, &pdu->pad        ) : False;          {
713                    switch (type)
714                    {
715                            case RDP_PDU_DEMAND_ACTIVE:
716                                    process_demand_active(s);
717                                    break;
718    
719          return res;                          case RDP_PDU_DEACTIVATE:
720  }                                  break;
721    
722  BOOL rdp_io_secondary_order(STREAM s, RDP_SECONDARY_ORDER *rso)                          case RDP_PDU_DATA:
723  {                                  process_data_pdu(s);
724          BOOL res = True;                                  break;
725    
726          res = res ? lsb_io_uint16(s, &rso->length) : False;                          case 0:
727          res = res ? lsb_io_uint16(s, &rso->flags ) : False;                                  break;
         res = res ? prs_io_uint8 (s, &rso->type  ) : False;  
728    
729          return res;                          default:
730                                    unimpl("PDU %d\n", type);
731                    }
732            }
733  }  }
734    
735  BOOL rdp_io_bitmap_header(STREAM s, RDP_BITMAP_HEADER *rdh)  /* Establish a connection up to the RDP layer */
736    BOOL
737    rdp_connect(char *server, uint32 flags, char *domain, char *password,
738                char *command, char *directory)
739  {  {
740          BOOL res = True;          if (!sec_connect(server))
741                    return False;
742    
743          res = res ? prs_io_uint8 (s, &rdh->cache_id  ) : False;          rdp_send_logon_info(flags, domain, username, password, command, directory);
744          res = res ? prs_io_uint8 (s, &rdh->pad1      ) : False;          return True;
745          res = res ? prs_io_uint8 (s, &rdh->width     ) : False;  }
         res = res ? prs_io_uint8 (s, &rdh->height    ) : False;  
         res = res ? prs_io_uint8 (s, &rdh->bpp       ) : False;  
         res = res ? lsb_io_uint16(s, &rdh->bufsize   ) : False;  
         res = res ? lsb_io_uint16(s, &rdh->cache_idx ) : False;  
         res = res ? lsb_io_uint16(s, &rdh->pad2      ) : False;  
         res = res ? lsb_io_uint16(s, &rdh->size      ) : False;  
         res = res ? lsb_io_uint16(s, &rdh->row_size  ) : False;  
         res = res ? lsb_io_uint16(s, &rdh->final_size) : False;  
746    
747          return res;  /* Disconnect from the RDP layer */
748    void
749    rdp_disconnect(void)
750    {
751            sec_disconnect();
752  }  }
Legend:
Removed from v.4  
changed lines
  Added in v.340
  ViewVC Help
Powered by ViewVC 1.1.26