/[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 654 by jsorg71, Thu Apr 15 22:03:24 2004 UTC revision 865 by stargo, Tue Mar 15 11:25:50 2005 UTC
# Line 1  Line 1 
1  /* -*- c-basic-offset: 8 -*-  /* -*- c-basic-offset: 8 -*-
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-2002     Copyright (C) Matthew Chapman 1999-2005
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 19  Line 19 
19  */  */
20    
21  #include <time.h>  #include <time.h>
22    #include <errno.h>
23    #include <unistd.h>
24  #include "rdesktop.h"  #include "rdesktop.h"
25    
26    #ifdef HAVE_ICONV
27    #ifdef HAVE_ICONV_H
28    #include <iconv.h>
29    #endif
30    
31    #ifndef ICONV_CONST
32    #define ICONV_CONST ""
33    #endif
34    #endif
35    
36  extern uint16 g_mcs_userid;  extern uint16 g_mcs_userid;
37  extern char g_username[16];  extern char g_username[64];
38    extern char g_codepage[16];
39  extern BOOL g_bitmap_compression;  extern BOOL g_bitmap_compression;
40  extern BOOL g_orders;  extern BOOL g_orders;
41  extern BOOL g_encryption;  extern BOOL g_encryption;
42  extern BOOL g_desktop_save;  extern BOOL g_desktop_save;
43    extern BOOL g_polygon_ellipse_orders;
44  extern BOOL g_use_rdp5;  extern BOOL g_use_rdp5;
45  extern uint16 g_server_rdp_version;  extern uint16 g_server_rdp_version;
46  extern uint32 g_rdp5_performanceflags;  extern uint32 g_rdp5_performanceflags;
47  extern int g_server_bpp;  extern int g_server_bpp;
48    extern int g_width;
49    extern int g_height;
50    extern BOOL g_bitmap_cache;
51    extern BOOL g_bitmap_cache_persist_enable;
52    
53  uint8 *g_next_packet;  uint8 *g_next_packet;
54  uint32 g_rdp_shareid;  uint32 g_rdp_shareid;
55    
56    extern RDPCOMP g_mppc_dict;
57    
58  #if WITH_DEBUG  #if WITH_DEBUG
59  static uint32 g_packetno;  static uint32 g_packetno;
60  #endif  #endif
61    
62    #ifdef HAVE_ICONV
63    static BOOL g_iconv_works = True;
64    #endif
65    
66  /* Receive an RDP packet */  /* Receive an RDP packet */
67  static STREAM  static STREAM
68  rdp_recv(uint8 * type)  rdp_recv(uint8 * type)
69  {  {
70          static STREAM rdp_s;          static STREAM rdp_s;
71          uint16 length, pdu_type;          uint16 length, pdu_type;
72            uint8 rdpver;
73    
74          if ((rdp_s == NULL) || (g_next_packet >= rdp_s->end))          if ((rdp_s == NULL) || (g_next_packet >= rdp_s->end))
75          {          {
76                  rdp_s = sec_recv();                  rdp_s = sec_recv(&rdpver);
77                  if (rdp_s == NULL)                  if (rdp_s == NULL)
78                          return NULL;                          return NULL;
79                    if (rdpver == 0xff)
80                    {
81                            g_next_packet = rdp_s->end;
82                            *type = 0;
83                            return rdp_s;
84                    }
85                    else if (rdpver != 3)
86                    {
87                            /* rdp5_process should move g_next_packet ok */
88                            rdp5_process(rdp_s);
89                            *type = 0;
90                            return rdp_s;
91                    }
92    
93                  g_next_packet = rdp_s->p;                  g_next_packet = rdp_s->p;
94          }          }
# Line 120  rdp_send_data(STREAM s, uint8 data_pdu_t Line 158  rdp_send_data(STREAM s, uint8 data_pdu_t
158  void  void
159  rdp_out_unistr(STREAM s, char *string, int len)  rdp_out_unistr(STREAM s, char *string, int len)
160  {  {
161          int i = 0, j = 0;  #ifdef HAVE_ICONV
162            size_t ibl = strlen(string), obl = len + 2;
163            static iconv_t iconv_h = (iconv_t) - 1;
164            char *pin = string, *pout = s->p;
165    
166          len += 2;          memset(pout, 0, len + 4);
167    
168          while (i < len)          if (g_iconv_works)
169          {          {
170                  s->p[i++] = string[j++];                  if (iconv_h == (iconv_t) - 1)
171                  s->p[i++] = 0;                  {
172                            size_t i = 1, o = 4;
173                            if ((iconv_h = iconv_open(WINDOWS_CODEPAGE, g_codepage)) == (iconv_t) - 1)
174                            {
175                                    warning("rdp_out_unistr: iconv_open[%s -> %s] fail %d\n",
176                                            g_codepage, WINDOWS_CODEPAGE, (int) iconv_h);
177    
178                                    g_iconv_works = False;
179                                    return (rdp_out_unistr(s, string, len));
180                            }
181                            if (iconv(iconv_h, (ICONV_CONST char **) &pin, &i, &pout, &o) ==
182                                (size_t) - 1)
183                            {
184                                    iconv_close(iconv_h);
185                                    iconv_h = (iconv_t) - 1;
186                                    warning("rdp_out_unistr: iconv(1) fail, errno %d\n", errno);
187    
188                                    g_iconv_works = False;
189                                    return (rdp_out_unistr(s, string, len));
190                            }
191                            pin = string;
192                            pout = (char *) s->p;
193                    }
194    
195                    if (iconv(iconv_h, (ICONV_CONST char **) &pin, &ibl, &pout, &obl) == (size_t) - 1)
196                    {
197                            iconv_close(iconv_h);
198                            iconv_h = (iconv_t) - 1;
199                            warning("rdp_out_unistr: iconv(2) fail, errno %d\n", errno);
200    
201                            g_iconv_works = False;
202                            return (rdp_out_unistr(s, string, len));
203                    }
204    
205                    s->p += len + 2;
206    
207          }          }
208            else
209    #endif
210            {
211                    int i = 0, j = 0;
212    
213                    len += 2;
214    
215                    while (i < len)
216                    {
217                            s->p[i++] = string[j++];
218                            s->p[i++] = 0;
219                    }
220    
221          s->p += len;                  s->p += len;
222            }
223  }  }
224    
225  /* Input a string in Unicode  /* Input a string in Unicode
# Line 140  rdp_out_unistr(STREAM s, char *string, i Line 229  rdp_out_unistr(STREAM s, char *string, i
229  int  int
230  rdp_in_unistr(STREAM s, char *string, int uni_len)  rdp_in_unistr(STREAM s, char *string, int uni_len)
231  {  {
232          int i = 0;  #ifdef HAVE_ICONV
233            size_t ibl = uni_len, obl = uni_len;
234            char *pin = s->p, *pout = string;
235            static iconv_t iconv_h = (iconv_t) - 1;
236    
237          while (i < uni_len / 2)          if (g_iconv_works)
238          {          {
239                  in_uint8a(s, &string[i++], 1);                  if (iconv_h == (iconv_t) - 1)
240                  in_uint8s(s, 1);                  {
241                            if ((iconv_h = iconv_open(g_codepage, WINDOWS_CODEPAGE)) == (iconv_t) - 1)
242                            {
243                                    warning("rdp_in_unistr: iconv_open[%s -> %s] fail %d\n",
244                                            WINDOWS_CODEPAGE, g_codepage, (int) iconv_h);
245    
246                                    g_iconv_works = False;
247                                    return rdp_in_unistr(s, string, uni_len);
248                            }
249                    }
250    
251                    if (iconv(iconv_h, (ICONV_CONST char **) &pin, &ibl, &pout, &obl) == (size_t) - 1)
252                    {
253                            iconv_close(iconv_h);
254                            iconv_h = (iconv_t) - 1;
255                            warning("rdp_in_unistr: iconv fail, errno %d\n", errno);
256    
257                            g_iconv_works = False;
258                            return rdp_in_unistr(s, string, uni_len);
259                    }
260                    return pout - string;
261          }          }
262            else
263    #endif
264            {
265                    int i = 0;
266    
267          return i - 1;                  while (i < uni_len / 2)
268                    {
269                            in_uint8a(s, &string[i++], 1);
270                            in_uint8s(s, 1);
271                    }
272    
273                    return i - 1;
274            }
275  }  }
276    
277    
# Line 172  rdp_send_logon_info(uint32 flags, char * Line 295  rdp_send_logon_info(uint32 flags, char *
295          time_t tzone;          time_t tzone;
296    
297  #if 0  #if 0
298          // enable rdp compression          /* enable rdp compression */
299            /* some problems still exist with rdp5 */
300          flags |= RDP_COMPRESSION;          flags |= RDP_COMPRESSION;
301  #endif  #endif
302    
# Line 198  rdp_send_logon_info(uint32 flags, char * Line 322  rdp_send_logon_info(uint32 flags, char *
322          }          }
323          else          else
324          {          {
325    
326                  flags |= RDP_LOGON_BLOB;                  flags |= RDP_LOGON_BLOB;
327                  DEBUG_RDP5(("Sending RDP5-style Logon packet\n"));                  DEBUG_RDP5(("Sending RDP5-style Logon packet\n"));
328                  packetlen = 4 + /* Unknown uint32 */                  packetlen = 4 + /* Unknown uint32 */
# Line 359  rdp_send_input(uint32 time, uint16 messa Line 484  rdp_send_input(uint32 time, uint16 messa
484          rdp_send_data(s, RDP_DATA_PDU_INPUT);          rdp_send_data(s, RDP_DATA_PDU_INPUT);
485  }  }
486    
487    /* Inform the server on the contents of the persistent bitmap cache */
488    static void
489    rdp_enum_bmpcache2(void)
490    {
491            STREAM s;
492            HASH_KEY keylist[BMPCACHE2_NUM_PSTCELLS];
493            uint32 num_keys, offset, count, flags;
494    
495            offset = 0;
496            num_keys = pstcache_enumerate(2, keylist);
497    
498            while (offset < num_keys)
499            {
500                    count = MIN(num_keys - offset, 169);
501    
502                    s = rdp_init_data(24 + count * sizeof(HASH_KEY));
503    
504                    flags = 0;
505                    if (offset == 0)
506                            flags |= PDU_FLAG_FIRST;
507                    if (num_keys - offset <= 169)
508                            flags |= PDU_FLAG_LAST;
509    
510                    /* header */
511                    out_uint32_le(s, 0);
512                    out_uint16_le(s, count);
513                    out_uint16_le(s, 0);
514                    out_uint16_le(s, 0);
515                    out_uint16_le(s, 0);
516                    out_uint16_le(s, 0);
517                    out_uint16_le(s, num_keys);
518                    out_uint32_le(s, 0);
519                    out_uint32_le(s, flags);
520    
521                    /* list */
522                    out_uint8a(s, keylist[offset], count * sizeof(HASH_KEY));
523    
524                    s_mark_end(s);
525                    rdp_send_data(s, 0x2b);
526    
527                    offset += 169;
528            }
529    }
530    
531  /* Send an (empty) font information PDU */  /* Send an (empty) font information PDU */
532  static void  static void
533  rdp_send_fonts(uint16 seq)  rdp_send_fonts(uint16 seq)
# Line 368  rdp_send_fonts(uint16 seq) Line 537  rdp_send_fonts(uint16 seq)
537          s = rdp_init_data(8);          s = rdp_init_data(8);
538    
539          out_uint16(s, 0);       /* number of fonts */          out_uint16(s, 0);       /* number of fonts */
540          out_uint16_le(s, 0x3e); /* unknown */          out_uint16_le(s, 0);    /* pad? */
541          out_uint16_le(s, seq);  /* unknown */          out_uint16_le(s, seq);  /* unknown */
542          out_uint16_le(s, 0x32); /* entry size */          out_uint16_le(s, 0x32); /* entry size */
543    
# Line 416  rdp_out_bitmap_caps(STREAM s) Line 585  rdp_out_bitmap_caps(STREAM s)
585          out_uint16_le(s, 800);  /* Desktop width */          out_uint16_le(s, 800);  /* Desktop width */
586          out_uint16_le(s, 600);  /* Desktop height */          out_uint16_le(s, 600);  /* Desktop height */
587          out_uint16(s, 0);       /* Pad */          out_uint16(s, 0);       /* Pad */
588          out_uint16(s, 0);       /* Allow resize */          out_uint16(s, 1);       /* Allow resize */
589          out_uint16_le(s, g_bitmap_compression ? 1 : 0); /* Support compression */          out_uint16_le(s, g_bitmap_compression ? 1 : 0); /* Support compression */
590          out_uint16(s, 0);       /* Unknown */          out_uint16(s, 0);       /* Unknown */
591          out_uint16_le(s, 1);    /* Unknown */          out_uint16_le(s, 1);    /* Unknown */
# Line 429  rdp_out_order_caps(STREAM s) Line 598  rdp_out_order_caps(STREAM s)
598  {  {
599          uint8 order_caps[32];          uint8 order_caps[32];
600    
   
601          memset(order_caps, 0, 32);          memset(order_caps, 0, 32);
602          order_caps[0] = 1;      /* dest blt */          order_caps[0] = 1;      /* dest blt */
603          order_caps[1] = 1;      /* pat blt */          order_caps[1] = 1;      /* pat blt */
604          order_caps[2] = 1;      /* screen blt */          order_caps[2] = 1;      /* screen blt */
605          order_caps[3] = 1;      /* required for memblt? */          order_caps[3] = (g_bitmap_cache ? 1 : 0);       /* memblt */
606            order_caps[4] = 0;      /* triblt */
607          order_caps[8] = 1;      /* line */          order_caps[8] = 1;      /* line */
608          order_caps[9] = 1;      /* line */          order_caps[9] = 1;      /* line */
609          order_caps[10] = 1;     /* rect */          order_caps[10] = 1;     /* rect */
610          order_caps[11] = (g_desktop_save == False ? 0 : 1);     /* desksave */          order_caps[11] = (g_desktop_save ? 1 : 0);      /* desksave */
611          order_caps[13] = 1;     /* memblt */          order_caps[13] = 1;     /* memblt */
612          order_caps[14] = 1;     /* triblt */          order_caps[14] = 1;     /* triblt */
613            order_caps[20] = (g_polygon_ellipse_orders ? 1 : 0);    /* polygon */
614            order_caps[21] = (g_polygon_ellipse_orders ? 1 : 0);    /* polygon2 */
615          order_caps[22] = 1;     /* polyline */          order_caps[22] = 1;     /* polyline */
616            order_caps[25] = (g_polygon_ellipse_orders ? 1 : 0);    /* ellipse */
617            order_caps[26] = (g_polygon_ellipse_orders ? 1 : 0);    /* ellipse2 */
618          order_caps[27] = 1;     /* text2 */          order_caps[27] = 1;     /* text2 */
619          out_uint16_le(s, RDP_CAPSET_ORDER);          out_uint16_le(s, RDP_CAPSET_ORDER);
620          out_uint16_le(s, RDP_CAPLEN_ORDER);          out_uint16_le(s, RDP_CAPLEN_ORDER);
# Line 479  rdp_out_bmpcache_caps(STREAM s) Line 652  rdp_out_bmpcache_caps(STREAM s)
652          out_uint16_le(s, 0x1000 * Bpp); /* max cell size */          out_uint16_le(s, 0x1000 * Bpp); /* max cell size */
653  }  }
654    
655    /* Output bitmap cache v2 capability set */
656    static void
657    rdp_out_bmpcache2_caps(STREAM s)
658    {
659            out_uint16_le(s, RDP_CAPSET_BMPCACHE2);
660            out_uint16_le(s, RDP_CAPLEN_BMPCACHE2);
661    
662            out_uint16_le(s, g_bitmap_cache_persist_enable ? 2 : 0);        /* version */
663    
664            out_uint16_le(s, 0x0300);       /* flags? number of caches? */
665    
666            out_uint32_le(s, BMPCACHE2_C0_CELLS);
667            out_uint32_le(s, BMPCACHE2_C1_CELLS);
668            if (pstcache_init(2))
669            {
670                    out_uint32_le(s, BMPCACHE2_NUM_PSTCELLS | BMPCACHE2_FLAG_PERSIST);
671            }
672            else
673            {
674                    out_uint32_le(s, BMPCACHE2_C2_CELLS);
675            }
676            out_uint8s(s, 20);      /* other bitmap caches not used */
677    }
678    
679  /* Output control capability set */  /* Output control capability set */
680  static void  static void
681  rdp_out_control_caps(STREAM s)  rdp_out_control_caps(STREAM s)
# Line 538  rdp_out_colcache_caps(STREAM s) Line 735  rdp_out_colcache_caps(STREAM s)
735          out_uint16(s, 0);       /* pad */          out_uint16(s, 0);       /* pad */
736  }  }
737    
738  static uint8 canned_caps[] = {  static uint8 caps_0x0d[] = {
739          0x01, 0x00, 0x00, 0x00, 0x09, 0x04, 0x00, 0x00, 0x04,          0x01, 0x00, 0x00, 0x00, 0x09, 0x04, 0x00, 0x00,
740          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0C, 0x00, 0x00, 0x00,          0x04, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
741          0x00, 0x00, 0x00, 0x00, 0x00,          0x0C, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
742          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
743          0x00, 0x00, 0x00, 0x00, 0x00,          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
744          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
745          0x00, 0x00, 0x00, 0x00, 0x00,          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
746          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
747          0x00, 0x00, 0x00, 0x00, 0x00,          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
748          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
749          0x0C, 0x00, 0x08, 0x00, 0x01,          0x00, 0x00, 0x00, 0x00
         0x00, 0x00, 0x00, 0x0E, 0x00, 0x08, 0x00, 0x01, 0x00, 0x00, 0x00,  
         0x10, 0x00, 0x34, 0x00, 0xFE,  
         0x00, 0x04, 0x00, 0xFE, 0x00, 0x04, 0x00, 0xFE, 0x00, 0x08, 0x00,  
         0xFE, 0x00, 0x08, 0x00, 0xFE,  
         0x00, 0x10, 0x00, 0xFE, 0x00, 0x20, 0x00, 0xFE, 0x00, 0x40, 0x00,  
         0xFE, 0x00, 0x80, 0x00, 0xFE,  
         0x00, 0x00, 0x01, 0x40, 0x00, 0x00, 0x08, 0x00, 0x01, 0x00, 0x01,  
         0x02, 0x00, 0x00, 0x00  
750  };  };
751    
752  /* Output unknown capability sets (number 13, 12, 14 and 16) */  static uint8 caps_0x0c[] = { 0x01, 0x00, 0x00, 0x00 };
753    
754    static uint8 caps_0x0e[] = { 0x01, 0x00, 0x00, 0x00 };
755    
756    static uint8 caps_0x10[] = {
757            0xFE, 0x00, 0x04, 0x00, 0xFE, 0x00, 0x04, 0x00,
758            0xFE, 0x00, 0x08, 0x00, 0xFE, 0x00, 0x08, 0x00,
759            0xFE, 0x00, 0x10, 0x00, 0xFE, 0x00, 0x20, 0x00,
760            0xFE, 0x00, 0x40, 0x00, 0xFE, 0x00, 0x80, 0x00,
761            0xFE, 0x00, 0x00, 0x01, 0x40, 0x00, 0x00, 0x08,
762            0x00, 0x01, 0x00, 0x01, 0x02, 0x00, 0x00, 0x00
763    };
764    
765    /* Output unknown capability sets */
766  static void  static void
767  rdp_out_unknown_caps(STREAM s)  rdp_out_unknown_caps(STREAM s, uint16 id, uint16 length, uint8 * caps)
768  {  {
769          out_uint16_le(s, RDP_CAPSET_UNKNOWN);          out_uint16_le(s, id);
770          out_uint16_le(s, 0x58);          out_uint16_le(s, length);
771    
772          out_uint8p(s, canned_caps, RDP_CAPLEN_UNKNOWN - 4);          out_uint8p(s, caps, length - 4);
773  }  }
774    
775  #define RDP5_FLAG 0x0030  #define RDP5_FLAG 0x0030
# Line 581  rdp_send_confirm_active(void) Line 783  rdp_send_confirm_active(void)
783                  RDP_CAPLEN_GENERAL + RDP_CAPLEN_BITMAP + RDP_CAPLEN_ORDER +                  RDP_CAPLEN_GENERAL + RDP_CAPLEN_BITMAP + RDP_CAPLEN_ORDER +
784                  RDP_CAPLEN_BMPCACHE + RDP_CAPLEN_COLCACHE +                  RDP_CAPLEN_BMPCACHE + RDP_CAPLEN_COLCACHE +
785                  RDP_CAPLEN_ACTIVATE + RDP_CAPLEN_CONTROL +                  RDP_CAPLEN_ACTIVATE + RDP_CAPLEN_CONTROL +
786                  RDP_CAPLEN_POINTER + RDP_CAPLEN_SHARE + RDP_CAPLEN_UNKNOWN + 4 /* w2k fix, why? */ ;                  RDP_CAPLEN_POINTER + RDP_CAPLEN_SHARE +
787                    0x58 + 0x08 + 0x08 + 0x34 /* unknown caps */  +
788                    4 /* w2k fix, why? */ ;
789    
790          s = sec_init(sec_flags, 6 + 14 + caplen + sizeof(RDP_SOURCE));          s = sec_init(sec_flags, 6 + 14 + caplen + sizeof(RDP_SOURCE));
791    
# Line 601  rdp_send_confirm_active(void) Line 805  rdp_send_confirm_active(void)
805          rdp_out_general_caps(s);          rdp_out_general_caps(s);
806          rdp_out_bitmap_caps(s);          rdp_out_bitmap_caps(s);
807          rdp_out_order_caps(s);          rdp_out_order_caps(s);
808          rdp_out_bmpcache_caps(s);          g_use_rdp5 ? rdp_out_bmpcache2_caps(s) : rdp_out_bmpcache_caps(s);
809          rdp_out_colcache_caps(s);          rdp_out_colcache_caps(s);
810          rdp_out_activate_caps(s);          rdp_out_activate_caps(s);
811          rdp_out_control_caps(s);          rdp_out_control_caps(s);
812          rdp_out_pointer_caps(s);          rdp_out_pointer_caps(s);
813          rdp_out_share_caps(s);          rdp_out_share_caps(s);
814          rdp_out_unknown_caps(s);  
815            rdp_out_unknown_caps(s, 0x0d, 0x58, caps_0x0d); /* international? */
816            rdp_out_unknown_caps(s, 0x0c, 0x08, caps_0x0c);
817            rdp_out_unknown_caps(s, 0x0e, 0x08, caps_0x0e);
818            rdp_out_unknown_caps(s, 0x10, 0x34, caps_0x10); /* glyph cache? */
819    
820          s_mark_end(s);          s_mark_end(s);
821          sec_send(s, sec_flags);          sec_send(s, sec_flags);
822  }  }
823    
824  /* Respond to a demand active PDU */  /* Process a general capability set */
825  static void  static void
826  process_demand_active(STREAM s)  rdp_process_general_caps(STREAM s)
827  {  {
828          uint8 type;          uint16 pad2octetsB;     /* rdp5 flags? */
         uint16 i;  
         uint16 p_bpp;  
829    
830          in_uint32_le(s, g_rdp_shareid);          in_uint8s(s, 10);
831            in_uint16_le(s, pad2octetsB);
832    
833          /* scan for prefered bpp */          if (!pad2octetsB)
834          while (s_check_rem(s, 6))                  g_use_rdp5 = False;
835    }
836    
837    /* Process a bitmap capability set */
838    static void
839    rdp_process_bitmap_caps(STREAM s)
840    {
841            uint16 width, height, bpp;
842    
843            in_uint16_le(s, bpp);
844            in_uint8s(s, 6);
845    
846            in_uint16_le(s, width);
847            in_uint16_le(s, height);
848    
849            DEBUG(("setting desktop size and bpp to: %dx%dx%d\n", width, height, bpp));
850    
851            /*
852             * The server may limit bpp and change the size of the desktop (for
853             * example when shadowing another session).
854             */
855            if (g_server_bpp != bpp)
856          {          {
857                  in_uint16_le(s, i);                  warning("colour depth changed from %d to %d\n", g_server_bpp, bpp);
858                  if (i == RDP_CAPSET_BITMAP)                  g_server_bpp = bpp;
859            }
860            if (g_width != width || g_height != height)
861            {
862                    warning("screen size changed from %dx%d to %dx%d\n", g_width, g_height,
863                            width, height);
864                    g_width = width;
865                    g_height = height;
866                    ui_resize_window();
867            }
868    }
869    
870    /* Process server capabilities */
871    void
872    rdp_process_server_caps(STREAM s, uint16 length)
873    {
874            int n;
875            uint8 *next, *start;
876            uint16 ncapsets, capset_type, capset_length;
877    
878            start = s->p;
879    
880            in_uint16_le(s, ncapsets);
881            in_uint8s(s, 2);        /* pad */
882    
883            for (n = 0; n < ncapsets; n++)
884            {
885                    if (s->p > start + length)
886                            return;
887    
888                    in_uint16_le(s, capset_type);
889                    in_uint16_le(s, capset_length);
890    
891                    next = s->p + capset_length - 4;
892    
893                    switch (capset_type)
894                  {                  {
895                          in_uint16_le(s, i);                          case RDP_CAPSET_GENERAL:
896                          if (i == RDP_CAPLEN_BITMAP)                                  rdp_process_general_caps(s);
897                          {                                  break;
898                                  in_uint16_le(s, p_bpp);  
899                                  if (p_bpp == 8 || p_bpp == 15 || p_bpp == 16 || p_bpp == 24)                          case RDP_CAPSET_BITMAP:
900                                  {                                  rdp_process_bitmap_caps(s);
901                                          if (p_bpp < g_server_bpp)                                  break;
                                         {  
                                                 warning("Server limited colour depth to %d bits\n", p_bpp);  
                                                 g_server_bpp = p_bpp;  
                                         }  
                                         break;  
                                 }  
                         }  
902                  }                  }
903    
904                    s->p = next;
905          }          }
906    }
907    
908    /* Respond to a demand active PDU */
909    static void
910    process_demand_active(STREAM s)
911    {
912            uint8 type;
913            uint16 len_src_descriptor, len_combined_caps;
914    
915            in_uint32_le(s, g_rdp_shareid);
916            in_uint16_le(s, len_src_descriptor);
917            in_uint16_le(s, len_combined_caps);
918            in_uint8s(s, len_src_descriptor);
919    
920          DEBUG(("DEMAND_ACTIVE(id=0x%x)\n", g_rdp_shareid));          DEBUG(("DEMAND_ACTIVE(id=0x%x)\n", g_rdp_shareid));
921            rdp_process_server_caps(s, len_combined_caps);
922    
923          rdp_send_confirm_active();          rdp_send_confirm_active();
924          rdp_send_synchronise();          rdp_send_synchronise();
# Line 657  process_demand_active(STREAM s) Line 928  process_demand_active(STREAM s)
928          rdp_recv(&type);        /* RDP_CTL_COOPERATE */          rdp_recv(&type);        /* RDP_CTL_COOPERATE */
929          rdp_recv(&type);        /* RDP_CTL_GRANT_CONTROL */          rdp_recv(&type);        /* RDP_CTL_GRANT_CONTROL */
930          rdp_send_input(0, RDP_INPUT_SYNCHRONIZE, 0, ui_get_numlock_state(read_keyboard_state()), 0);          rdp_send_input(0, RDP_INPUT_SYNCHRONIZE, 0, ui_get_numlock_state(read_keyboard_state()), 0);
931          rdp_send_fonts(1);  
932          rdp_send_fonts(2);          if (g_use_rdp5)
933          rdp_recv(&type);        /* RDP_PDU_UNKNOWN 0x28 */          {
934                    rdp_enum_bmpcache2();
935                    rdp_send_fonts(3);
936            }
937            else
938            {
939                    rdp_send_fonts(1);
940                    rdp_send_fonts(2);
941            }
942    
943            rdp_recv(&type);        /* RDP_PDU_UNKNOWN 0x28 (Fonts?) */
944          reset_order_state();          reset_order_state();
945  }  }
946    
# Line 859  process_update_pdu(STREAM s) Line 1140  process_update_pdu(STREAM s)
1140    
1141          in_uint16_le(s, update_type);          in_uint16_le(s, update_type);
1142    
1143            ui_begin_update();
1144          switch (update_type)          switch (update_type)
1145          {          {
1146                  case RDP_UPDATE_ORDERS:                  case RDP_UPDATE_ORDERS:
# Line 882  process_update_pdu(STREAM s) Line 1164  process_update_pdu(STREAM s)
1164                  default:                  default:
1165                          unimpl("update %d\n", update_type);                          unimpl("update %d\n", update_type);
1166          }          }
1167            ui_end_update();
1168    }
1169    
1170    /* Process a disconnect PDU */
1171    void
1172    process_disconnect_pdu(STREAM s, uint32 * ext_disc_reason)
1173    {
1174            in_uint32_le(s, *ext_disc_reason);
1175    
1176            DEBUG(("Received disconnect PDU\n"));
1177  }  }
1178    
1179  /* Process data PDU */  /* Process data PDU */
1180  static void  static BOOL
1181  process_data_pdu(STREAM s)  process_data_pdu(STREAM s, uint32 * ext_disc_reason)
1182  {  {
1183          uint8 data_pdu_type;          uint8 data_pdu_type;
1184          uint8 ctype;          uint8 ctype;
1185          uint16 clen;          uint16 clen;
1186          int roff, rlen, len, ret;          uint32 len;
1187          static struct stream ns;  
1188          static signed char *dict = 0;          uint32 roff, rlen;
1189    
1190            struct stream *ns = &(g_mppc_dict.ns);
1191    
1192          in_uint8s(s, 6);        /* shareid, pad, streamid */          in_uint8s(s, 6);        /* shareid, pad, streamid */
1193          in_uint16(s, len);          in_uint16(s, len);
# Line 903  process_data_pdu(STREAM s) Line 1196  process_data_pdu(STREAM s)
1196          in_uint16(s, clen);          in_uint16(s, clen);
1197          clen -= 18;          clen -= 18;
1198    
1199  #if 0          if (ctype & RDP_MPPC_COMPRESSED)
         if (ctype & 0x20)  
1200          {          {
1201                  if (!dict)                  if (len > RDP_MPPC_DICT_SIZE)
1202                  {                          error("error decompressed packet size exceeds max\n");
1203                          dict = (signed char *) malloc(8200 * sizeof(signed char));                  if (mppc_expand(s->p, clen, ctype, &roff, &rlen) == -1)
1204                          dict = (signed char *) memset(dict, 0, 8200 * sizeof(signed char));                          error("error while decompressing packet\n");
                 }  
1205    
1206                  ret = decompress(s->p, clen, ctype, (signed char *) dict, &roff, &rlen);                  /* len -= 18; */
1207    
1208                  len -= 18;                  /* allocate memory and copy the uncompressed data into the temporary stream */
1209                    ns->data = (uint8 *) xrealloc(ns->data, rlen);
1210    
1211                  ns.data = xrealloc(ns.data, len);                  memcpy((ns->data), (unsigned char *) (g_mppc_dict.hist + roff), rlen);
1212    
1213                  ns.data = (unsigned char *) memcpy(ns.data, (unsigned char *) (dict + roff), len);                  ns->size = rlen;
1214                    ns->end = (ns->data + ns->size);
1215                    ns->p = ns->data;
1216                    ns->rdp_hdr = ns->p;
1217    
1218                  ns.size = len;                  s = ns;
                 ns.end = ns.data + ns.size;  
                 ns.p = ns.data;  
                 ns.rdp_hdr = ns.p;  
   
                 s = &ns;  
1219          }          }
 #endif  
1220    
1221          switch (data_pdu_type)          switch (data_pdu_type)
1222          {          {
# Line 935  process_data_pdu(STREAM s) Line 1224  process_data_pdu(STREAM s)
1224                          process_update_pdu(s);                          process_update_pdu(s);
1225                          break;                          break;
1226    
1227                    case RDP_DATA_PDU_CONTROL:
1228                            DEBUG(("Received Control PDU\n"));
1229                            break;
1230    
1231                    case RDP_DATA_PDU_SYNCHRONISE:
1232                            DEBUG(("Received Sync PDU\n"));
1233                            break;
1234    
1235                  case RDP_DATA_PDU_POINTER:                  case RDP_DATA_PDU_POINTER:
1236                          process_pointer_pdu(s);                          process_pointer_pdu(s);
1237                          break;                          break;
# Line 949  process_data_pdu(STREAM s) Line 1246  process_data_pdu(STREAM s)
1246                          break;                          break;
1247    
1248                  case RDP_DATA_PDU_DISCONNECT:                  case RDP_DATA_PDU_DISCONNECT:
1249                          /* Normally received when user logs out or disconnects from a                          process_disconnect_pdu(s, ext_disc_reason);
1250                             console session on Windows XP and 2003 Server */                          return True;
                         DEBUG(("Received disconnect PDU\n"));  
                         break;  
1251    
1252                  default:                  default:
1253                          unimpl("data PDU %d\n", data_pdu_type);                          unimpl("data PDU %d\n", data_pdu_type);
1254          }          }
1255            return False;
1256  }  }
1257    
1258  /* Process incoming packets */  /* Process incoming packets */
1259    /* nevers gets out of here till app is done */
1260    void
1261    rdp_main_loop(BOOL * deactivated, uint32 * ext_disc_reason)
1262    {
1263            while (rdp_loop(deactivated, ext_disc_reason))
1264                    ;
1265    }
1266    
1267    /* used in uiports and rdp_main_loop, processes the rdp packets waiting */
1268  BOOL  BOOL
1269  rdp_main_loop(void)  rdp_loop(BOOL * deactivated, uint32 * ext_disc_reason)
1270  {  {
1271          uint8 type;          uint8 type;
1272            BOOL disc = False;      /* True when a disconnect PDU was received */
1273            BOOL cont = True;
1274          STREAM s;          STREAM s;
1275    
1276          while ((s = rdp_recv(&type)) != NULL)          while (cont)
1277          {          {
1278                    s = rdp_recv(&type);
1279                    if (s == NULL)
1280                            return False;
1281                  switch (type)                  switch (type)
1282                  {                  {
1283                          case RDP_PDU_DEMAND_ACTIVE:                          case RDP_PDU_DEMAND_ACTIVE:
1284                                  process_demand_active(s);                                  process_demand_active(s);
1285                                    *deactivated = False;
1286                                  break;                                  break;
   
1287                          case RDP_PDU_DEACTIVATE:                          case RDP_PDU_DEACTIVATE:
1288                                  DEBUG(("RDP_PDU_DEACTIVATE\n"));                                  DEBUG(("RDP_PDU_DEACTIVATE\n"));
1289                                  /* We thought we could detect a clean                                  *deactivated = True;
                                    shutdown of the session by this  
                                    packet, but it seems Windows 2003  
                                    is sending us one of these when we  
                                    reconnect to a disconnected session  
                                    return True; */  
1290                                  break;                                  break;
   
1291                          case RDP_PDU_DATA:                          case RDP_PDU_DATA:
1292                                  process_data_pdu(s);                                  disc = process_data_pdu(s, ext_disc_reason);
1293                                  break;                                  break;
   
1294                          case 0:                          case 0:
1295                                  break;                                  break;
   
1296                          default:                          default:
1297                                  unimpl("PDU %d\n", type);                                  unimpl("PDU %d\n", type);
1298                  }                  }
1299                    if (disc)
1300                            return False;
1301                    cont = g_next_packet < s->end;
1302          }          }
1303          return True;          return True;
         /* We want to detect if we got a clean shutdown, but we  
            can't. Se above.    
            return False;  */  
1304  }  }
1305    
1306  /* Establish a connection up to the RDP layer */  /* Establish a connection up to the RDP layer */
Legend:
Removed from v.654  
changed lines
  Added in v.865
  ViewVC Help
Powered by ViewVC 1.1.26