/[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 365 by matthewc, Wed Apr 16 08:19:15 2003 UTC revision 861 by stargo, Sun Mar 13 17:40:51 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
8     the Free Software Foundation; either version 2 of the License, or     the Free Software Foundation; either version 2 of the License, or
9     (at your option) any later version.     (at your option) any later version.
10      
11     This program is distributed in the hope that it will be useful,     This program is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14     GNU General Public License for more details.     GNU General Public License for more details.
15      
16     You should have received a copy of the GNU General Public License     You should have received a copy of the GNU General Public License
17     along with this program; if not, write to the Free Software     along with this program; if not, write to the Free Software
18     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19  */  */
20    
21    #include <time.h>
22    #include <errno.h>
23    #include <unistd.h>
24  #include "rdesktop.h"  #include "rdesktop.h"
25    
26  extern uint16 mcs_userid;  #ifdef HAVE_ICONV
27  extern char username[16];  #ifdef HAVE_ICONV_H
28  extern BOOL bitmap_compression;  #include <iconv.h>
29  extern BOOL orders;  #endif
30  extern BOOL encryption;  
31  extern BOOL desktop_save;  #ifndef ICONV_CONST
32  extern BOOL use_rdp5;  #define ICONV_CONST ""
33  extern uint16 server_rdp_version;  #endif
34    #endif
35    
36  uint8 *next_packet;  extern uint16 g_mcs_userid;
37  uint32 rdp_shareid;  extern char g_username[64];
38    extern char g_codepage[16];
39    extern BOOL g_bitmap_compression;
40    extern BOOL g_orders;
41    extern BOOL g_encryption;
42    extern BOOL g_desktop_save;
43    extern BOOL g_polygon_ellipse_orders;
44    extern BOOL g_use_rdp5;
45    extern uint16 g_server_rdp_version;
46    extern uint32 g_rdp5_performanceflags;
47    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;
54    uint32 g_rdp_shareid;
55    
56    extern RDPCOMP g_mppc_dict;
57    
58  #if WITH_DEBUG  #if WITH_DEBUG
59  static uint32 packetno;  static uint32 g_packetno;
60  #endif  #endif
61    
62  /* Receive an RDP packet */  /* Receive an RDP packet */
# Line 42  rdp_recv(uint8 * type) Line 65  rdp_recv(uint8 * type)
65  {  {
66          static STREAM rdp_s;          static STREAM rdp_s;
67          uint16 length, pdu_type;          uint16 length, pdu_type;
68            uint8 rdpver;
69    
70          if ((rdp_s == NULL) || (next_packet >= rdp_s->end))          if ((rdp_s == NULL) || (g_next_packet >= rdp_s->end))
71          {          {
72                  rdp_s = sec_recv();                  rdp_s = sec_recv(&rdpver);
73                  if (rdp_s == NULL)                  if (rdp_s == NULL)
74                          return NULL;                          return NULL;
75                    if (rdpver == 0xff)
76                    {
77                            g_next_packet = rdp_s->end;
78                            *type = 0;
79                            return rdp_s;
80                    }
81                    else if (rdpver != 3)
82                    {
83                            /* rdp5_process should move g_next_packet ok */
84                            rdp5_process(rdp_s);
85                            *type = 0;
86                            return rdp_s;
87                    }
88    
89                  next_packet = rdp_s->p;                  g_next_packet = rdp_s->p;
90          }          }
91          else          else
92          {          {
93                  rdp_s->p = next_packet;                  rdp_s->p = g_next_packet;
94          }          }
95    
96          in_uint16_le(rdp_s, length);          in_uint16_le(rdp_s, length);
97          /* 32k packets are really 8, keepalive fix */          /* 32k packets are really 8, keepalive fix */
98          if (length == 0x8000)          if (length == 0x8000)
99          {          {
100                  next_packet += 8;                  g_next_packet += 8;
101                  *type = 0;                  *type = 0;
102                  return rdp_s;                  return rdp_s;
103          }          }
# Line 69  rdp_recv(uint8 * type) Line 106  rdp_recv(uint8 * type)
106          *type = pdu_type & 0xf;          *type = pdu_type & 0xf;
107    
108  #if WITH_DEBUG  #if WITH_DEBUG
109          DEBUG(("RDP packet #%d, (type %x)\n", ++packetno, *type));          DEBUG(("RDP packet #%d, (type %x)\n", ++g_packetno, *type));
110          //      hexdump(next_packet, length);          hexdump(g_next_packet, length);
111  #endif /*  */  #endif /*  */
112    
113          next_packet += length;          g_next_packet += length;
114          return rdp_s;          return rdp_s;
115  }  }
116    
# Line 83  rdp_init_data(int maxlen) Line 120  rdp_init_data(int maxlen)
120  {  {
121          STREAM s;          STREAM s;
122    
123          s = sec_init(encryption ? SEC_ENCRYPT : 0, maxlen + 18);          s = sec_init(g_encryption ? SEC_ENCRYPT : 0, maxlen + 18);
124          s_push_layer(s, rdp_hdr, 18);          s_push_layer(s, rdp_hdr, 18);
125    
126          return s;          return s;
# Line 100  rdp_send_data(STREAM s, uint8 data_pdu_t Line 137  rdp_send_data(STREAM s, uint8 data_pdu_t
137    
138          out_uint16_le(s, length);          out_uint16_le(s, length);
139          out_uint16_le(s, (RDP_PDU_DATA | 0x10));          out_uint16_le(s, (RDP_PDU_DATA | 0x10));
140          out_uint16_le(s, (mcs_userid + 1001));          out_uint16_le(s, (g_mcs_userid + 1001));
141    
142          out_uint32_le(s, rdp_shareid);          out_uint32_le(s, g_rdp_shareid);
143          out_uint8(s, 0);        /* pad */          out_uint8(s, 0);        /* pad */
144          out_uint8(s, 1);        /* streamid */          out_uint8(s, 1);        /* streamid */
145          out_uint16_le(s, (length - 14));          out_uint16_le(s, (length - 14));
# Line 110  rdp_send_data(STREAM s, uint8 data_pdu_t Line 147  rdp_send_data(STREAM s, uint8 data_pdu_t
147          out_uint8(s, 0);        /* compress_type */          out_uint8(s, 0);        /* compress_type */
148          out_uint16(s, 0);       /* compress_len */          out_uint16(s, 0);       /* compress_len */
149    
150          sec_send(s, encryption ? SEC_ENCRYPT : 0);          sec_send(s, g_encryption ? SEC_ENCRYPT : 0);
151  }  }
152    
153  /* Output a string in Unicode */  /* Output a string in Unicode */
154  void  void
155  rdp_out_unistr(STREAM s, char *string, int len)  rdp_out_unistr(STREAM s, char *string, int len)
156  {  {
157    #ifdef HAVE_ICONV
158            size_t ibl = strlen(string), obl = len + 2;
159            static iconv_t iconv_h = (iconv_t) - 1;
160            char *pin = string, *pout;
161    #ifdef B_ENDIAN
162            char ss[4096];          // FIXME: global MAX_BUF_SIZE macro need
163    
164            pout = ss;
165    #else
166            pout = s->p;
167    #endif
168    
169            memset(pout, 0, len + 4);
170    
171            if (iconv_h == (iconv_t) - 1)
172            {
173                    size_t i = 1, o = 4;
174                    if ((iconv_h = iconv_open(WINDOWS_CODEPAGE, g_codepage)) == (iconv_t) - 1)
175                    {
176                            printf("rdp_out_unistr: iconv_open[%s -> %s] fail %d\n",
177                                   g_codepage, WINDOWS_CODEPAGE, (int) iconv_h);
178                            return;
179                    }
180                    if (iconv(iconv_h, (ICONV_CONST char **) &pin, &i, &pout, &o) == (size_t) - 1)
181                    {
182                            iconv_close(iconv_h);
183                            iconv_h = (iconv_t) - 1;
184                            printf("rdp_out_unistr: iconv(1) fail, errno %d\n", errno);
185                            return;
186                    }
187                    pin = string;
188                    pout = (char *) s->p;
189            }
190    
191            if (iconv(iconv_h, (ICONV_CONST char **) &pin, &ibl, &pout, &obl) == (size_t) - 1)
192            {
193                    iconv_close(iconv_h);
194                    iconv_h = (iconv_t) - 1;
195                    printf("rdp_out_unistr: iconv(2) fail, errno %d\n", errno);
196                    return;
197            }
198    
199    #ifdef B_ENDIAN
200            swab(ss, (char *) s->p, len + 4);
201    #endif
202    
203            s->p += len + 2;
204    
205    #else /*HAVE_ICONV undef */
206          int i = 0, j = 0;          int i = 0, j = 0;
207    
208          len += 2;          len += 2;
# Line 128  rdp_out_unistr(STREAM s, char *string, i Line 214  rdp_out_unistr(STREAM s, char *string, i
214          }          }
215    
216          s->p += len;          s->p += len;
217    #endif
218  }  }
219    
220    /* Input a string in Unicode
221     *
222     * Returns str_len of string
223     */
224    int
225    rdp_in_unistr(STREAM s, char *string, int uni_len)
226    {
227    #ifdef HAVE_ICONV
228            size_t ibl = uni_len, obl = uni_len;
229            char *pin, *pout = string;
230            static iconv_t iconv_h = (iconv_t) - 1;
231    #ifdef B_ENDIAN
232            char ss[4096];          // FIXME: global MAX_BUF_SIZE macro need
233    
234            swab((char *) s->p, ss, uni_len);
235            pin = ss;
236    #else
237            pin = s->p;
238    #endif
239    
240            if (iconv_h == (iconv_t) - 1)
241            {
242                    if ((iconv_h = iconv_open(g_codepage, WINDOWS_CODEPAGE)) == (iconv_t) - 1)
243                    {
244                            printf("rdp_in_unistr: iconv_open[%s -> %s] fail %d\n",
245                                   WINDOWS_CODEPAGE, g_codepage, (int) iconv_h);
246                            return 0;
247                    }
248            }
249    
250            if (iconv(iconv_h, (ICONV_CONST char **) &pin, &ibl, &pout, &obl) == (size_t) - 1)
251            {
252                    iconv_close(iconv_h);
253                    iconv_h = (iconv_t) - 1;
254                    printf("rdp_in_unistr: iconv fail, errno %d\n", errno);
255                    return 0;
256            }
257            return pout - string;
258    #else /* HAVE_ICONV undef */
259            int i = 0;
260    
261            while (i < uni_len / 2)
262            {
263                    in_uint8a(s, &string[i++], 1);
264                    in_uint8s(s, 1);
265            }
266    
267            return i - 1;
268    #endif
269    }
270    
271    
272  /* Parse a logon info packet */  /* Parse a logon info packet */
273  static void  static void
274  rdp_send_logon_info(uint32 flags, char *domain, char *user,  rdp_send_logon_info(uint32 flags, char *domain, char *user,
275                      char *password, char *program, char *directory)                      char *password, char *program, char *directory)
276  {  {
277            char *ipaddr = tcp_get_address();
278          int len_domain = 2 * strlen(domain);          int len_domain = 2 * strlen(domain);
279          int len_user = 2 * strlen(user);          int len_user = 2 * strlen(user);
280          int len_password = 2 * strlen(password);          int len_password = 2 * strlen(password);
281          int len_program = 2 * strlen(program);          int len_program = 2 * strlen(program);
282          int len_directory = 2 * strlen(directory);          int len_directory = 2 * strlen(directory);
283          uint32 sec_flags = encryption ? (SEC_LOGON_INFO | SEC_ENCRYPT) : SEC_LOGON_INFO;          int len_ip = 2 * strlen(ipaddr);
284            int len_dll = 2 * strlen("C:\\WINNT\\System32\\mstscax.dll");
285            int packetlen = 0;
286            uint32 sec_flags = g_encryption ? (SEC_LOGON_INFO | SEC_ENCRYPT) : SEC_LOGON_INFO;
287          STREAM s;          STREAM s;
288            time_t t = time(NULL);
289            time_t tzone;
290    
291    #if 0
292            /* enable rdp compression */
293            /* some problems still exist with rdp5 */
294            flags |= RDP_COMPRESSION;
295    #endif
296    
297          if (1 == server_rdp_version)          if (!g_use_rdp5 || 1 == g_server_rdp_version)
298          {          {
299                  DEBUG_RDP5(("Sending RDP4-style Logon packet\n"));                  DEBUG_RDP5(("Sending RDP4-style Logon packet\n"));
300    
# Line 165  rdp_send_logon_info(uint32 flags, char * Line 316  rdp_send_logon_info(uint32 flags, char *
316          }          }
317          else          else
318          {          {
319    
320                    flags |= RDP_LOGON_BLOB;
321                  DEBUG_RDP5(("Sending RDP5-style Logon packet\n"));                  DEBUG_RDP5(("Sending RDP5-style Logon packet\n"));
322                  s = sec_init(sec_flags, 12 + (flags & RDP_LOGON_AUTO ? 2 : 0) + 6 + (flags & RDP_LOGON_AUTO ? len_password : 0) + len_domain + len_user + 4 + len_program + len_directory + 30 + 2 + 60 + 32 + 20 + 32 + 20);   /* Phew! */                  packetlen = 4 + /* Unknown uint32 */
323                            4 +     /* flags */
324                            2 +     /* len_domain */
325                            2 +     /* len_user */
326                            (flags & RDP_LOGON_AUTO ? 2 : 0) +      /* len_password */
327                            (flags & RDP_LOGON_BLOB ? 2 : 0) +      /* Length of BLOB */
328                            2 +     /* len_program */
329                            2 +     /* len_directory */
330                            (0 < len_domain ? len_domain : 2) +     /* domain */
331                            len_user + (flags & RDP_LOGON_AUTO ? len_password : 0) + 0 +    /* We have no 512 byte BLOB. Perhaps we must? */
332                            (flags & RDP_LOGON_BLOB && !(flags & RDP_LOGON_AUTO) ? 2 : 0) + /* After the BLOB is a unknown int16. If there is a BLOB, that is. */
333                            (0 < len_program ? len_program : 2) + (0 < len_directory ? len_directory : 2) + 2 +     /* Unknown (2) */
334                            2 +     /* Client ip length */
335                            len_ip +        /* Client ip */
336                            2 +     /* DLL string length */
337                            len_dll +       /* DLL string */
338                            2 +     /* Unknown */
339                            2 +     /* Unknown */
340                            64 +    /* Time zone #0 */
341                            2 +     /* Unknown */
342                            64 +    /* Time zone #1 */
343                            32;     /* Unknown */
344    
345                  out_uint32(s, 0);                  s = sec_init(sec_flags, packetlen);
346                    DEBUG_RDP5(("Called sec_init with packetlen %d\n", packetlen));
347    
348                    out_uint32(s, 0);       /* Unknown */
349                  out_uint32_le(s, flags);                  out_uint32_le(s, flags);
350                  out_uint16_le(s, len_domain);                  out_uint16_le(s, len_domain);
351                  out_uint16_le(s, len_user);                  out_uint16_le(s, len_user);
352                  if (flags & RDP_LOGON_AUTO)                  if (flags & RDP_LOGON_AUTO)
353                  {                  {
354                          out_uint16_le(s, len_password);                          out_uint16_le(s, len_password);
355    
356                    }
357                    if (flags & RDP_LOGON_BLOB && !(flags & RDP_LOGON_AUTO))
358                    {
359                            out_uint16_le(s, 0);
360                  }                  }
                 out_uint16(s, 0);       /* Seems to be length of a 512 byte blob with  
                                            completely unknown data, but hopefully we'll do  
                                            with a 0 length block as well */  
361                  out_uint16_le(s, len_program);                  out_uint16_le(s, len_program);
362                  out_uint16_le(s, len_directory);                  out_uint16_le(s, len_directory);
363                    if (0 < len_domain)
364                            rdp_out_unistr(s, domain, len_domain);
365                    else
366                            out_uint16_le(s, 0);
367                    rdp_out_unistr(s, user, len_user);
368                  if (flags & RDP_LOGON_AUTO)                  if (flags & RDP_LOGON_AUTO)
369                  {                  {
370                          rdp_out_unistr(s, password, len_password);                          rdp_out_unistr(s, password, len_password);
371                  }                  }
372                  rdp_out_unistr(s, domain, len_domain);                  if (flags & RDP_LOGON_BLOB && !(flags & RDP_LOGON_AUTO))
373                  rdp_out_unistr(s, user, len_user);                  {
374                  out_uint16(s, 0);                          out_uint16_le(s, 0);
375                  out_uint16(s, 0);                  }
376                  if (0 < len_program)                  if (0 < len_program)
377                    {
378                          rdp_out_unistr(s, program, len_program);                          rdp_out_unistr(s, program, len_program);
379    
380                    }
381                    else
382                    {
383                            out_uint16_le(s, 0);
384                    }
385                  if (0 < len_directory)                  if (0 < len_directory)
386                    {
387                          rdp_out_unistr(s, directory, len_directory);                          rdp_out_unistr(s, directory, len_directory);
388                  out_uint8s(s, 30);      /* Some kind of client data - let's see if the server                  }
389                                             handles zeros well.. */                  else
390                  out_uint16_le(s, 60);                  {
391                  rdp_out_unistr(s, "C:\\WINNT\\System32\\mstscax.dll", 58);                          out_uint16_le(s, 0);
392                  out_uint32_be(s, 0x88ffffff);                  }
393                  rdp_out_unistr(s, "GTB, normaltid", 2 * strlen("GTB, normaltid") - 2);                  out_uint16_le(s, 2);
394                  out_uint8s(s, 30 - 2 * strlen("GTP, normaltid"));                  out_uint16_le(s, len_ip + 2);   /* Length of client ip */
395                    rdp_out_unistr(s, ipaddr, len_ip);
396                    out_uint16_le(s, len_dll + 2);
397                    rdp_out_unistr(s, "C:\\WINNT\\System32\\mstscax.dll", len_dll);
398    
399                    tzone = (mktime(gmtime(&t)) - mktime(localtime(&t))) / 60;
400                    out_uint32_le(s, tzone);
401    
402                    rdp_out_unistr(s, "GTB, normaltid", 2 * strlen("GTB, normaltid"));
403                    out_uint8s(s, 62 - 2 * strlen("GTB, normaltid"));
404    
405                  out_uint32_le(s, 0x0a0000);                  out_uint32_le(s, 0x0a0000);
406                  out_uint32_le(s, 0x050000);                  out_uint32_le(s, 0x050000);
407                  out_uint32_le(s, 2);                  out_uint32_le(s, 3);
408                  out_uint32(s, 0);                  out_uint32_le(s, 0);
409                  out_uint32_le(s, 0xffffffc4);                  out_uint32_le(s, 0);
                 out_uint32_le(s, 0xfffffffe);  
                 out_uint32_le(s, 0x0f);  
                 out_uint32(s, 0);  
410    
411                  rdp_out_unistr(s, "GTB, sommartid", 2 * strlen("GTB, sommartid") - 1);                  rdp_out_unistr(s, "GTB, sommartid", 2 * strlen("GTB, sommartid"));
412                  out_uint8s(s, 30 - 2 * strlen("GTP, sommartid"));                  out_uint8s(s, 62 - 2 * strlen("GTB, sommartid"));
413    
414                  out_uint32_le(s, 0x030000);                  out_uint32_le(s, 0x30000);
415                  out_uint32_le(s, 0x050000);                  out_uint32_le(s, 0x050000);
416                  out_uint32_le(s, 2);                  out_uint32_le(s, 2);
417                  out_uint32(s, 0);                  out_uint32(s, 0);
418                  out_uint32_le(s, 0xffffffc4);                  out_uint32_le(s, 0xffffffc4);
419                  out_uint32_le(s, 0xfffffffe);                  out_uint32_le(s, 0xfffffffe);
420                  out_uint32_le(s, 0x0f);                  out_uint32_le(s, g_rdp5_performanceflags);
421                  out_uint32(s, 0);                  out_uint32(s, 0);
422    
423    
424          }          }
425          s_mark_end(s);          s_mark_end(s);
426          sec_send(s, sec_flags);          sec_send(s, sec_flags);
# Line 279  rdp_send_input(uint32 time, uint16 messa Line 478  rdp_send_input(uint32 time, uint16 messa
478          rdp_send_data(s, RDP_DATA_PDU_INPUT);          rdp_send_data(s, RDP_DATA_PDU_INPUT);
479  }  }
480    
481    /* Inform the server on the contents of the persistent bitmap cache */
482    static void
483    rdp_enum_bmpcache2(void)
484    {
485            STREAM s;
486            HASH_KEY keylist[BMPCACHE2_NUM_PSTCELLS];
487            uint32 num_keys, offset, count, flags;
488    
489            offset = 0;
490            num_keys = pstcache_enumerate(2, keylist);
491    
492            while (offset < num_keys)
493            {
494                    count = MIN(num_keys - offset, 169);
495    
496                    s = rdp_init_data(24 + count * sizeof(HASH_KEY));
497    
498                    flags = 0;
499                    if (offset == 0)
500                            flags |= PDU_FLAG_FIRST;
501                    if (num_keys - offset <= 169)
502                            flags |= PDU_FLAG_LAST;
503    
504                    /* header */
505                    out_uint32_le(s, 0);
506                    out_uint16_le(s, count);
507                    out_uint16_le(s, 0);
508                    out_uint16_le(s, 0);
509                    out_uint16_le(s, 0);
510                    out_uint16_le(s, 0);
511                    out_uint16_le(s, num_keys);
512                    out_uint32_le(s, 0);
513                    out_uint32_le(s, flags);
514    
515                    /* list */
516                    out_uint8a(s, keylist[offset], count * sizeof(HASH_KEY));
517    
518                    s_mark_end(s);
519                    rdp_send_data(s, 0x2b);
520    
521                    offset += 169;
522            }
523    }
524    
525  /* Send an (empty) font information PDU */  /* Send an (empty) font information PDU */
526  static void  static void
527  rdp_send_fonts(uint16 seq)  rdp_send_fonts(uint16 seq)
# Line 288  rdp_send_fonts(uint16 seq) Line 531  rdp_send_fonts(uint16 seq)
531          s = rdp_init_data(8);          s = rdp_init_data(8);
532    
533          out_uint16(s, 0);       /* number of fonts */          out_uint16(s, 0);       /* number of fonts */
534          out_uint16_le(s, 0x3e); /* unknown */          out_uint16_le(s, 0);    /* pad? */
535          out_uint16_le(s, seq);  /* unknown */          out_uint16_le(s, seq);  /* unknown */
536          out_uint16_le(s, 0x32); /* entry size */          out_uint16_le(s, 0x32); /* entry size */
537    
# Line 308  rdp_out_general_caps(STREAM s) Line 551  rdp_out_general_caps(STREAM s)
551          out_uint16_le(s, 0x200);        /* Protocol version */          out_uint16_le(s, 0x200);        /* Protocol version */
552          out_uint16(s, 0);       /* Pad */          out_uint16(s, 0);       /* Pad */
553          out_uint16(s, 0);       /* Compression types */          out_uint16(s, 0);       /* Compression types */
554          out_uint16_le(s, use_rdp5 ? 0x40d : 0);          out_uint16_le(s, g_use_rdp5 ? 0x40d : 0);
555          /* Pad, according to T.128. 0x40d seems to          /* Pad, according to T.128. 0x40d seems to
556             trigger             trigger
557             the server to start sending RDP5 packets.             the server to start sending RDP5 packets.
# Line 329  rdp_out_bitmap_caps(STREAM s) Line 572  rdp_out_bitmap_caps(STREAM s)
572          out_uint16_le(s, RDP_CAPSET_BITMAP);          out_uint16_le(s, RDP_CAPSET_BITMAP);
573          out_uint16_le(s, RDP_CAPLEN_BITMAP);          out_uint16_le(s, RDP_CAPLEN_BITMAP);
574    
575          out_uint16_le(s, 8);    /* Preferred BPP */          out_uint16_le(s, g_server_bpp); /* Preferred BPP */
576          out_uint16_le(s, 1);    /* Receive 1 BPP */          out_uint16_le(s, 1);    /* Receive 1 BPP */
577          out_uint16_le(s, 1);    /* Receive 4 BPP */          out_uint16_le(s, 1);    /* Receive 4 BPP */
578          out_uint16_le(s, 1);    /* Receive 8 BPP */          out_uint16_le(s, 1);    /* Receive 8 BPP */
579          out_uint16_le(s, 800);  /* Desktop width */          out_uint16_le(s, 800);  /* Desktop width */
580          out_uint16_le(s, 600);  /* Desktop height */          out_uint16_le(s, 600);  /* Desktop height */
581          out_uint16(s, 0);       /* Pad */          out_uint16(s, 0);       /* Pad */
582          out_uint16(s, 0);       /* Allow resize */          out_uint16(s, 1);       /* Allow resize */
583          out_uint16_le(s, bitmap_compression ? 1 : 0);   /* Support compression */          out_uint16_le(s, g_bitmap_compression ? 1 : 0); /* Support compression */
584          out_uint16(s, 0);       /* Unknown */          out_uint16(s, 0);       /* Unknown */
585          out_uint16_le(s, 1);    /* Unknown */          out_uint16_le(s, 1);    /* Unknown */
586          out_uint16(s, 0);       /* Pad */          out_uint16(s, 0);       /* Pad */
# Line 349  rdp_out_order_caps(STREAM s) Line 592  rdp_out_order_caps(STREAM s)
592  {  {
593          uint8 order_caps[32];          uint8 order_caps[32];
594    
   
595          memset(order_caps, 0, 32);          memset(order_caps, 0, 32);
596          order_caps[0] = 1;      /* dest blt */          order_caps[0] = 1;      /* dest blt */
597          order_caps[1] = 1;      /* pat blt */          order_caps[1] = 1;      /* pat blt */
598          order_caps[2] = 1;      /* screen blt */          order_caps[2] = 1;      /* screen blt */
599          order_caps[3] = 1;      /* required for memblt? */          order_caps[3] = (g_bitmap_cache ? 1 : 0);       /* memblt */
600            order_caps[4] = 0;      /* triblt */
601          order_caps[8] = 1;      /* line */          order_caps[8] = 1;      /* line */
602          order_caps[9] = 1;      /* line */          order_caps[9] = 1;      /* line */
603          order_caps[10] = 1;     /* rect */          order_caps[10] = 1;     /* rect */
604          order_caps[11] = (desktop_save == False ? 0 : 1);       /* desksave */          order_caps[11] = (g_desktop_save ? 1 : 0);      /* desksave */
605          order_caps[13] = 1;     /* memblt */          order_caps[13] = 1;     /* memblt */
606          order_caps[14] = 1;     /* triblt */          order_caps[14] = 1;     /* triblt */
607            order_caps[20] = (g_polygon_ellipse_orders ? 1 : 0);    /* polygon */
608            order_caps[21] = (g_polygon_ellipse_orders ? 1 : 0);    /* polygon2 */
609          order_caps[22] = 1;     /* polyline */          order_caps[22] = 1;     /* polyline */
610            order_caps[25] = (g_polygon_ellipse_orders ? 1 : 0);    /* ellipse */
611            order_caps[26] = (g_polygon_ellipse_orders ? 1 : 0);    /* ellipse2 */
612          order_caps[27] = 1;     /* text2 */          order_caps[27] = 1;     /* text2 */
613          out_uint16_le(s, RDP_CAPSET_ORDER);          out_uint16_le(s, RDP_CAPSET_ORDER);
614          out_uint16_le(s, RDP_CAPLEN_ORDER);          out_uint16_le(s, RDP_CAPLEN_ORDER);
# Line 376  rdp_out_order_caps(STREAM s) Line 623  rdp_out_order_caps(STREAM s)
623          out_uint8p(s, order_caps, 32);  /* Orders supported */          out_uint8p(s, order_caps, 32);  /* Orders supported */
624          out_uint16_le(s, 0x6a1);        /* Text capability flags */          out_uint16_le(s, 0x6a1);        /* Text capability flags */
625          out_uint8s(s, 6);       /* Pad */          out_uint8s(s, 6);       /* Pad */
626          out_uint32_le(s, desktop_save == False ? 0 : 0x38400);  /* Desktop cache size */          out_uint32_le(s, g_desktop_save == False ? 0 : 0x38400);        /* Desktop cache size */
627          out_uint32(s, 0);       /* Unknown */          out_uint32(s, 0);       /* Unknown */
628          out_uint32_le(s, 0x4e4);        /* Unknown */          out_uint32_le(s, 0x4e4);        /* Unknown */
629  }  }
# Line 385  rdp_out_order_caps(STREAM s) Line 632  rdp_out_order_caps(STREAM s)
632  static void  static void
633  rdp_out_bmpcache_caps(STREAM s)  rdp_out_bmpcache_caps(STREAM s)
634  {  {
635            int Bpp;
636          out_uint16_le(s, RDP_CAPSET_BMPCACHE);          out_uint16_le(s, RDP_CAPSET_BMPCACHE);
637          out_uint16_le(s, RDP_CAPLEN_BMPCACHE);          out_uint16_le(s, RDP_CAPLEN_BMPCACHE);
638    
639            Bpp = (g_server_bpp + 7) / 8;
640          out_uint8s(s, 24);      /* unused */          out_uint8s(s, 24);      /* unused */
641          out_uint16_le(s, 0x258);        /* entries */          out_uint16_le(s, 0x258);        /* entries */
642          out_uint16_le(s, 0x100);        /* max cell size */          out_uint16_le(s, 0x100 * Bpp);  /* max cell size */
643          out_uint16_le(s, 0x12c);        /* entries */          out_uint16_le(s, 0x12c);        /* entries */
644          out_uint16_le(s, 0x400);        /* max cell size */          out_uint16_le(s, 0x400 * Bpp);  /* max cell size */
645          out_uint16_le(s, 0x106);        /* entries */          out_uint16_le(s, 0x106);        /* entries */
646          out_uint16_le(s, 0x1000);       /* max cell size */          out_uint16_le(s, 0x1000 * Bpp); /* max cell size */
647    }
648    
649    /* Output bitmap cache v2 capability set */
650    static void
651    rdp_out_bmpcache2_caps(STREAM s)
652    {
653            out_uint16_le(s, RDP_CAPSET_BMPCACHE2);
654            out_uint16_le(s, RDP_CAPLEN_BMPCACHE2);
655    
656            out_uint16_le(s, g_bitmap_cache_persist_enable ? 2 : 0);        /* version */
657    
658            out_uint16_le(s, 0x0300);       /* flags? number of caches? */
659    
660            out_uint32_le(s, BMPCACHE2_C0_CELLS);
661            out_uint32_le(s, BMPCACHE2_C1_CELLS);
662            if (pstcache_init(2))
663            {
664                    out_uint32_le(s, BMPCACHE2_NUM_PSTCELLS | BMPCACHE2_FLAG_PERSIST);
665            }
666            else
667            {
668                    out_uint32_le(s, BMPCACHE2_C2_CELLS);
669            }
670            out_uint8s(s, 20);      /* other bitmap caches not used */
671  }  }
672    
673  /* Output control capability set */  /* Output control capability set */
# Line 456  rdp_out_colcache_caps(STREAM s) Line 729  rdp_out_colcache_caps(STREAM s)
729          out_uint16(s, 0);       /* pad */          out_uint16(s, 0);       /* pad */
730  }  }
731    
732  static uint8 canned_caps[] = {  static uint8 caps_0x0d[] = {
733          0x01, 0x00, 0x00, 0x00, 0x09, 0x04, 0x00, 0x00, 0x04,          0x01, 0x00, 0x00, 0x00, 0x09, 0x04, 0x00, 0x00,
734          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0C, 0x00, 0x00, 0x00,          0x04, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
735          0x00, 0x00, 0x00, 0x00, 0x00,          0x0C, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
736          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
737          0x00, 0x00, 0x00, 0x00, 0x00,          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
738          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
739          0x00, 0x00, 0x00, 0x00, 0x00,          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
740          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,          0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
741          0x00, 0x00, 0x00, 0x00, 0x00,          0x00, 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          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  
744  };  };
745    
746  /* Output unknown capability sets (number 13, 12, 14 and 16) */  static uint8 caps_0x0c[] = { 0x01, 0x00, 0x00, 0x00 };
747    
748    static uint8 caps_0x0e[] = { 0x01, 0x00, 0x00, 0x00 };
749    
750    static uint8 caps_0x10[] = {
751            0xFE, 0x00, 0x04, 0x00, 0xFE, 0x00, 0x04, 0x00,
752            0xFE, 0x00, 0x08, 0x00, 0xFE, 0x00, 0x08, 0x00,
753            0xFE, 0x00, 0x10, 0x00, 0xFE, 0x00, 0x20, 0x00,
754            0xFE, 0x00, 0x40, 0x00, 0xFE, 0x00, 0x80, 0x00,
755            0xFE, 0x00, 0x00, 0x01, 0x40, 0x00, 0x00, 0x08,
756            0x00, 0x01, 0x00, 0x01, 0x02, 0x00, 0x00, 0x00
757    };
758    
759    /* Output unknown capability sets */
760  static void  static void
761  rdp_out_unknown_caps(STREAM s)  rdp_out_unknown_caps(STREAM s, uint16 id, uint16 length, uint8 * caps)
762  {  {
763          out_uint16_le(s, RDP_CAPSET_UNKNOWN);          out_uint16_le(s, id);
764          out_uint16_le(s, 0x58);          out_uint16_le(s, length);
765    
766          out_uint8p(s, canned_caps, RDP_CAPLEN_UNKNOWN - 4);          out_uint8p(s, caps, length - 4);
767  }  }
768    
769  #define RDP5_FLAG 0x0030  #define RDP5_FLAG 0x0030
# Line 494  static void Line 772  static void
772  rdp_send_confirm_active(void)  rdp_send_confirm_active(void)
773  {  {
774          STREAM s;          STREAM s;
775          uint32 sec_flags = encryption ? (RDP5_FLAG | SEC_ENCRYPT) : RDP5_FLAG;          uint32 sec_flags = g_encryption ? (RDP5_FLAG | SEC_ENCRYPT) : RDP5_FLAG;
776          uint16 caplen =          uint16 caplen =
777                  RDP_CAPLEN_GENERAL + RDP_CAPLEN_BITMAP + RDP_CAPLEN_ORDER +                  RDP_CAPLEN_GENERAL + RDP_CAPLEN_BITMAP + RDP_CAPLEN_ORDER +
778                  RDP_CAPLEN_BMPCACHE + RDP_CAPLEN_COLCACHE +                  RDP_CAPLEN_BMPCACHE + RDP_CAPLEN_COLCACHE +
779                  RDP_CAPLEN_ACTIVATE + RDP_CAPLEN_CONTROL +                  RDP_CAPLEN_ACTIVATE + RDP_CAPLEN_CONTROL +
780                  RDP_CAPLEN_POINTER + RDP_CAPLEN_SHARE + RDP_CAPLEN_UNKNOWN + 4 /* w2k fix, why? */ ;                  RDP_CAPLEN_POINTER + RDP_CAPLEN_SHARE +
781                    0x58 + 0x08 + 0x08 + 0x34 /* unknown caps */  +
782                    4 /* w2k fix, why? */ ;
783    
784          s = sec_init(sec_flags, 6 + 14 + caplen + sizeof(RDP_SOURCE));          s = sec_init(sec_flags, 6 + 14 + caplen + sizeof(RDP_SOURCE));
785    
786          out_uint16_le(s, 2 + 14 + caplen + sizeof(RDP_SOURCE));          out_uint16_le(s, 2 + 14 + caplen + sizeof(RDP_SOURCE));
787          out_uint16_le(s, (RDP_PDU_CONFIRM_ACTIVE | 0x10));      /* Version 1 */          out_uint16_le(s, (RDP_PDU_CONFIRM_ACTIVE | 0x10));      /* Version 1 */
788          out_uint16_le(s, (mcs_userid + 1001));          out_uint16_le(s, (g_mcs_userid + 1001));
789    
790          out_uint32_le(s, rdp_shareid);          out_uint32_le(s, g_rdp_shareid);
791          out_uint16_le(s, 0x3ea);        /* userid */          out_uint16_le(s, 0x3ea);        /* userid */
792          out_uint16_le(s, sizeof(RDP_SOURCE));          out_uint16_le(s, sizeof(RDP_SOURCE));
793          out_uint16_le(s, caplen);          out_uint16_le(s, caplen);
# Line 519  rdp_send_confirm_active(void) Line 799  rdp_send_confirm_active(void)
799          rdp_out_general_caps(s);          rdp_out_general_caps(s);
800          rdp_out_bitmap_caps(s);          rdp_out_bitmap_caps(s);
801          rdp_out_order_caps(s);          rdp_out_order_caps(s);
802          rdp_out_bmpcache_caps(s);          g_use_rdp5 ? rdp_out_bmpcache2_caps(s) : rdp_out_bmpcache_caps(s);
803          rdp_out_colcache_caps(s);          rdp_out_colcache_caps(s);
804          rdp_out_activate_caps(s);          rdp_out_activate_caps(s);
805          rdp_out_control_caps(s);          rdp_out_control_caps(s);
806          rdp_out_pointer_caps(s);          rdp_out_pointer_caps(s);
807          rdp_out_share_caps(s);          rdp_out_share_caps(s);
808          rdp_out_unknown_caps(s);  
809            rdp_out_unknown_caps(s, 0x0d, 0x58, caps_0x0d); /* international? */
810            rdp_out_unknown_caps(s, 0x0c, 0x08, caps_0x0c);
811            rdp_out_unknown_caps(s, 0x0e, 0x08, caps_0x0e);
812            rdp_out_unknown_caps(s, 0x10, 0x34, caps_0x10); /* glyph cache? */
813    
814          s_mark_end(s);          s_mark_end(s);
815          sec_send(s, sec_flags);          sec_send(s, sec_flags);
816  }  }
817    
818    /* Process a general capability set */
819    static void
820    rdp_process_general_caps(STREAM s)
821    {
822            uint16 pad2octetsB;     /* rdp5 flags? */
823    
824            in_uint8s(s, 10);
825            in_uint16_le(s, pad2octetsB);
826    
827            if (!pad2octetsB)
828                    g_use_rdp5 = False;
829    }
830    
831    /* Process a bitmap capability set */
832    static void
833    rdp_process_bitmap_caps(STREAM s)
834    {
835            uint16 width, height, bpp;
836    
837            in_uint16_le(s, bpp);
838            in_uint8s(s, 6);
839    
840            in_uint16_le(s, width);
841            in_uint16_le(s, height);
842    
843            DEBUG(("setting desktop size and bpp to: %dx%dx%d\n", width, height, bpp));
844    
845            /*
846             * The server may limit bpp and change the size of the desktop (for
847             * example when shadowing another session).
848             */
849            if (g_server_bpp != bpp)
850            {
851                    warning("colour depth changed from %d to %d\n", g_server_bpp, bpp);
852                    g_server_bpp = bpp;
853            }
854            if (g_width != width || g_height != height)
855            {
856                    warning("screen size changed from %dx%d to %dx%d\n", g_width, g_height,
857                            width, height);
858                    g_width = width;
859                    g_height = height;
860                    ui_resize_window();
861            }
862    }
863    
864    /* Process server capabilities */
865    void
866    rdp_process_server_caps(STREAM s, uint16 length)
867    {
868            int n;
869            uint8 *next, *start;
870            uint16 ncapsets, capset_type, capset_length;
871    
872            start = s->p;
873    
874            in_uint16_le(s, ncapsets);
875            in_uint8s(s, 2);        /* pad */
876    
877            for (n = 0; n < ncapsets; n++)
878            {
879                    if (s->p > start + length)
880                            return;
881    
882                    in_uint16_le(s, capset_type);
883                    in_uint16_le(s, capset_length);
884    
885                    next = s->p + capset_length - 4;
886    
887                    switch (capset_type)
888                    {
889                            case RDP_CAPSET_GENERAL:
890                                    rdp_process_general_caps(s);
891                                    break;
892    
893                            case RDP_CAPSET_BITMAP:
894                                    rdp_process_bitmap_caps(s);
895                                    break;
896                    }
897    
898                    s->p = next;
899            }
900    }
901    
902  /* Respond to a demand active PDU */  /* Respond to a demand active PDU */
903  static void  static void
904  process_demand_active(STREAM s)  process_demand_active(STREAM s)
905  {  {
906          uint8 type;          uint8 type;
907            uint16 len_src_descriptor, len_combined_caps;
908    
909          in_uint32_le(s, rdp_shareid);          in_uint32_le(s, g_rdp_shareid);
910            in_uint16_le(s, len_src_descriptor);
911            in_uint16_le(s, len_combined_caps);
912            in_uint8s(s, len_src_descriptor);
913    
914          DEBUG(("DEMAND_ACTIVE(id=0x%x)\n", rdp_shareid));          DEBUG(("DEMAND_ACTIVE(id=0x%x)\n", g_rdp_shareid));
915            rdp_process_server_caps(s, len_combined_caps);
916    
917          rdp_send_confirm_active();          rdp_send_confirm_active();
918          rdp_send_synchronise();          rdp_send_synchronise();
# Line 548  process_demand_active(STREAM s) Line 921  process_demand_active(STREAM s)
921          rdp_recv(&type);        /* RDP_PDU_SYNCHRONIZE */          rdp_recv(&type);        /* RDP_PDU_SYNCHRONIZE */
922          rdp_recv(&type);        /* RDP_CTL_COOPERATE */          rdp_recv(&type);        /* RDP_CTL_COOPERATE */
923          rdp_recv(&type);        /* RDP_CTL_GRANT_CONTROL */          rdp_recv(&type);        /* RDP_CTL_GRANT_CONTROL */
924          rdp_send_input(0, RDP_INPUT_SYNCHRONIZE, 0, 0, 0);          rdp_send_input(0, RDP_INPUT_SYNCHRONIZE, 0, ui_get_numlock_state(read_keyboard_state()), 0);
         rdp_send_fonts(1);  
         rdp_send_fonts(2);  
         rdp_recv(&type);        /* RDP_PDU_UNKNOWN 0x28 */  
         reset_order_state();  
 }  
925    
926  /* Process a null system pointer PDU */          if (g_use_rdp5)
927  void          {
928  process_null_system_pointer_pdu(STREAM s)                  rdp_enum_bmpcache2();
929  {                  rdp_send_fonts(3);
930          // FIXME: We should probably set another cursor here,          }
931          // like the X window system base cursor or something.          else
932          ui_set_cursor(cache_get_cursor(0));          {
933                    rdp_send_fonts(1);
934                    rdp_send_fonts(2);
935            }
936    
937            rdp_recv(&type);        /* RDP_PDU_UNKNOWN 0x28 (Fonts?) */
938            reset_order_state();
939  }  }
940    
941  /* Process a colour pointer PDU */  /* Process a colour pointer PDU */
# Line 596  process_cached_pointer_pdu(STREAM s) Line 970  process_cached_pointer_pdu(STREAM s)
970          ui_set_cursor(cache_get_cursor(cache_idx));          ui_set_cursor(cache_get_cursor(cache_idx));
971  }  }
972    
973    /* Process a system pointer PDU */
974    void
975    process_system_pointer_pdu(STREAM s)
976    {
977            uint16 system_pointer_type;
978    
979            in_uint16(s, system_pointer_type);
980            switch (system_pointer_type)
981            {
982                    case RDP_NULL_POINTER:
983                            ui_set_null_cursor();
984                            break;
985    
986                    default:
987                            unimpl("System pointer message 0x%x\n", system_pointer_type);
988            }
989    }
990    
991  /* Process a pointer PDU */  /* Process a pointer PDU */
992  static void  static void
# Line 624  process_pointer_pdu(STREAM s) Line 1015  process_pointer_pdu(STREAM s)
1015                          process_cached_pointer_pdu(s);                          process_cached_pointer_pdu(s);
1016                          break;                          break;
1017    
1018                    case RDP_POINTER_SYSTEM:
1019                            process_system_pointer_pdu(s);
1020                            break;
1021    
1022                  default:                  default:
1023                          DEBUG(("Pointer message 0x%x\n", message_type));                          unimpl("Pointer message 0x%x\n", message_type);
1024          }          }
1025  }  }
1026    
# Line 663  process_bitmap_updates(STREAM s) Line 1058  process_bitmap_updates(STREAM s)
1058                  if (!compress)                  if (!compress)
1059                  {                  {
1060                          int y;                          int y;
1061                          bmpdata = xmalloc(width * height * Bpp);                          bmpdata = (uint8 *) xmalloc(width * height * Bpp);
1062                          for (y = 0; y < height; y++)                          for (y = 0; y < height; y++)
1063                          {                          {
1064                                  in_uint8a(s, &bmpdata[(height - y - 1) * (width * Bpp)],                                  in_uint8a(s, &bmpdata[(height - y - 1) * (width * Bpp)],
# Line 686  process_bitmap_updates(STREAM s) Line 1081  process_bitmap_updates(STREAM s)
1081                          in_uint8s(s, 4);        /* line_size, final_size */                          in_uint8s(s, 4);        /* line_size, final_size */
1082                  }                  }
1083                  in_uint8p(s, data, size);                  in_uint8p(s, data, size);
1084                  bmpdata = xmalloc(width * height * Bpp);                  bmpdata = (uint8 *) xmalloc(width * height * Bpp);
1085                  if (bitmap_decompress(bmpdata, width, height, data, size, Bpp))                  if (bitmap_decompress(bmpdata, width, height, data, size, Bpp))
1086                  {                  {
1087                          ui_paint_bitmap(left, top, cx, cy, width, height, bmpdata);                          ui_paint_bitmap(left, top, cx, cy, width, height, bmpdata);
# Line 713  process_palette(STREAM s) Line 1108  process_palette(STREAM s)
1108          in_uint16_le(s, map.ncolours);          in_uint16_le(s, map.ncolours);
1109          in_uint8s(s, 2);        /* pad */          in_uint8s(s, 2);        /* pad */
1110    
1111          map.colours = xmalloc(3 * map.ncolours);          map.colours = (COLOURENTRY *) xmalloc(sizeof(COLOURENTRY) * map.ncolours);
1112    
1113          DEBUG(("PALETTE(c=%d)\n", map.ncolours));          DEBUG(("PALETTE(c=%d)\n", map.ncolours));
1114    
# Line 739  process_update_pdu(STREAM s) Line 1134  process_update_pdu(STREAM s)
1134    
1135          in_uint16_le(s, update_type);          in_uint16_le(s, update_type);
1136    
1137            ui_begin_update();
1138          switch (update_type)          switch (update_type)
1139          {          {
1140                  case RDP_UPDATE_ORDERS:                  case RDP_UPDATE_ORDERS:
# Line 762  process_update_pdu(STREAM s) Line 1158  process_update_pdu(STREAM s)
1158                  default:                  default:
1159                          unimpl("update %d\n", update_type);                          unimpl("update %d\n", update_type);
1160          }          }
1161            ui_end_update();
1162    }
1163    
1164    /* Process a disconnect PDU */
1165    void
1166    process_disconnect_pdu(STREAM s, uint32 * ext_disc_reason)
1167    {
1168            in_uint32_le(s, *ext_disc_reason);
1169    
1170            DEBUG(("Received disconnect PDU\n"));
1171  }  }
1172    
1173  /* Process data PDU */  /* Process data PDU */
1174  static void  static BOOL
1175  process_data_pdu(STREAM s)  process_data_pdu(STREAM s, uint32 * ext_disc_reason)
1176  {  {
1177          uint8 data_pdu_type;          uint8 data_pdu_type;
1178            uint8 ctype;
1179            uint16 clen;
1180            uint32 len;
1181    
1182            uint32 roff, rlen;
1183    
1184          in_uint8s(s, 8);        /* shareid, pad, streamid, length */          struct stream *ns = &(g_mppc_dict.ns);
1185    
1186            in_uint8s(s, 6);        /* shareid, pad, streamid */
1187            in_uint16(s, len);
1188          in_uint8(s, data_pdu_type);          in_uint8(s, data_pdu_type);
1189          in_uint8s(s, 3);        /* compress_type, compress_len */          in_uint8(s, ctype);
1190            in_uint16(s, clen);
1191            clen -= 18;
1192    
1193            if (ctype & RDP_MPPC_COMPRESSED)
1194            {
1195                    if (len > RDP_MPPC_DICT_SIZE)
1196                            error("error decompressed packet size exceeds max\n");
1197                    if (mppc_expand(s->p, clen, ctype, &roff, &rlen) == -1)
1198                            error("error while decompressing packet\n");
1199    
1200                    //len -= 18;
1201    
1202                    /* allocate memory and copy the uncompressed data into the temporary stream */
1203                    ns->data = (uint8 *) xrealloc(ns->data, rlen);
1204    
1205                    memcpy((ns->data), (unsigned char *) (g_mppc_dict.hist + roff), rlen);
1206    
1207                    ns->size = rlen;
1208                    ns->end = (ns->data + ns->size);
1209                    ns->p = ns->data;
1210                    ns->rdp_hdr = ns->p;
1211    
1212                    s = ns;
1213            }
1214    
1215          switch (data_pdu_type)          switch (data_pdu_type)
1216          {          {
# Line 781  process_data_pdu(STREAM s) Line 1218  process_data_pdu(STREAM s)
1218                          process_update_pdu(s);                          process_update_pdu(s);
1219                          break;                          break;
1220    
1221                    case RDP_DATA_PDU_CONTROL:
1222                            DEBUG(("Received Control PDU\n"));
1223                            break;
1224    
1225                    case RDP_DATA_PDU_SYNCHRONISE:
1226                            DEBUG(("Received Sync PDU\n"));
1227                            break;
1228    
1229                  case RDP_DATA_PDU_POINTER:                  case RDP_DATA_PDU_POINTER:
1230                          process_pointer_pdu(s);                          process_pointer_pdu(s);
1231                          break;                          break;
# Line 794  process_data_pdu(STREAM s) Line 1239  process_data_pdu(STREAM s)
1239                          /* User logged on */                          /* User logged on */
1240                          break;                          break;
1241    
1242                    case RDP_DATA_PDU_DISCONNECT:
1243                            process_disconnect_pdu(s, ext_disc_reason);
1244                            return True;
1245    
1246                  default:                  default:
1247                          unimpl("data PDU %d\n", data_pdu_type);                          unimpl("data PDU %d\n", data_pdu_type);
1248          }          }
1249            return False;
1250  }  }
1251    
1252  /* Process incoming packets */  /* Process incoming packets */
1253    /* nevers gets out of here till app is done */
1254  void  void
1255  rdp_main_loop(void)  rdp_main_loop(BOOL * deactivated, uint32 * ext_disc_reason)
1256    {
1257            while (rdp_loop(deactivated, ext_disc_reason))
1258                    ;
1259    }
1260    
1261    /* used in uiports and rdp_main_loop, processes the rdp packets waiting */
1262    BOOL
1263    rdp_loop(BOOL * deactivated, uint32 * ext_disc_reason)
1264  {  {
1265          uint8 type;          uint8 type;
1266            BOOL disc = False;      /* True when a disconnect PDU was received */
1267            BOOL cont = True;
1268          STREAM s;          STREAM s;
1269    
1270          while ((s = rdp_recv(&type)) != NULL)          while (cont)
1271          {          {
1272                    s = rdp_recv(&type);
1273                    if (s == NULL)
1274                            return False;
1275                  switch (type)                  switch (type)
1276                  {                  {
1277                          case RDP_PDU_DEMAND_ACTIVE:                          case RDP_PDU_DEMAND_ACTIVE:
1278                                  process_demand_active(s);                                  process_demand_active(s);
1279                                    *deactivated = False;
1280                                  break;                                  break;
   
1281                          case RDP_PDU_DEACTIVATE:                          case RDP_PDU_DEACTIVATE:
1282                                    DEBUG(("RDP_PDU_DEACTIVATE\n"));
1283                                    *deactivated = True;
1284                                  break;                                  break;
   
1285                          case RDP_PDU_DATA:                          case RDP_PDU_DATA:
1286                                  process_data_pdu(s);                                  disc = process_data_pdu(s, ext_disc_reason);
1287                                  break;                                  break;
   
1288                          case 0:                          case 0:
1289                                  break;                                  break;
   
1290                          default:                          default:
1291                                  unimpl("PDU %d\n", type);                                  unimpl("PDU %d\n", type);
1292                  }                  }
1293                    if (disc)
1294                            return False;
1295                    cont = g_next_packet < s->end;
1296          }          }
1297            return True;
1298  }  }
1299    
1300  /* Establish a connection up to the RDP layer */  /* Establish a connection up to the RDP layer */
# Line 835  BOOL Line 1302  BOOL
1302  rdp_connect(char *server, uint32 flags, char *domain, char *password,  rdp_connect(char *server, uint32 flags, char *domain, char *password,
1303              char *command, char *directory)              char *command, char *directory)
1304  {  {
1305          if (!sec_connect(server, username))          if (!sec_connect(server, g_username))
1306                  return False;                  return False;
1307    
1308          rdp_send_logon_info(flags, domain, username, password, command, directory);          rdp_send_logon_info(flags, domain, g_username, password, command, directory);
1309          return True;          return True;
1310  }  }
1311    
Legend:
Removed from v.365  
changed lines
  Added in v.861
  ViewVC Help
Powered by ViewVC 1.1.26