23 |
#include <unistd.h> |
#include <unistd.h> |
24 |
#include "rdesktop.h" |
#include "rdesktop.h" |
25 |
|
|
26 |
|
#ifdef HAVE_ICONV |
27 |
#ifdef HAVE_ICONV_H |
#ifdef HAVE_ICONV_H |
28 |
#include <iconv.h> |
#include <iconv.h> |
29 |
#endif |
#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[64]; |
extern char g_username[64]; |
38 |
extern char g_codepage[16]; |
extern char g_codepage[16]; |
49 |
extern int g_height; |
extern int g_height; |
50 |
extern BOOL g_bitmap_cache; |
extern BOOL g_bitmap_cache; |
51 |
extern BOOL g_bitmap_cache_persist_enable; |
extern BOOL g_bitmap_cache_persist_enable; |
52 |
|
extern BOOL g_rdp_compression; |
53 |
|
|
54 |
uint8 *g_next_packet; |
uint8 *g_next_packet; |
55 |
uint32 g_rdp_shareid; |
uint32 g_rdp_shareid; |
60 |
static uint32 g_packetno; |
static uint32 g_packetno; |
61 |
#endif |
#endif |
62 |
|
|
63 |
|
#ifdef HAVE_ICONV |
64 |
|
static BOOL g_iconv_works = True; |
65 |
|
#endif |
66 |
|
|
67 |
/* Receive an RDP packet */ |
/* Receive an RDP packet */ |
68 |
static STREAM |
static STREAM |
69 |
rdp_recv(uint8 * type) |
rdp_recv(uint8 * type) |
159 |
void |
void |
160 |
rdp_out_unistr(STREAM s, char *string, int len) |
rdp_out_unistr(STREAM s, char *string, int len) |
161 |
{ |
{ |
162 |
#ifdef HAVE_ICONV |
#ifdef HAVE_ICONV |
163 |
size_t ibl = strlen(string), obl = len + 2; |
size_t ibl = strlen(string), obl = len + 2; |
164 |
static iconv_t iconv_h = (iconv_t)-1; |
static iconv_t iconv_h = (iconv_t) - 1; |
165 |
char *pin = string, *pout; |
char *pin = string, *pout = (char *) s->p; |
|
#ifdef B_ENDIAN |
|
|
char ss[4096]; // FIXME: global MAX_BUF_SIZE macro need |
|
|
|
|
|
pout = ss; |
|
|
#else |
|
|
pout = s->p; |
|
|
#endif |
|
166 |
|
|
167 |
memset(pout, 0, len + 4); |
memset(pout, 0, len + 4); |
168 |
|
|
169 |
if (iconv_h == (iconv_t)-1) |
if (g_iconv_works) |
170 |
{ |
{ |
171 |
size_t i = 1, o = 4; |
if (iconv_h == (iconv_t) - 1) |
|
if ((iconv_h = iconv_open(WINDOWS_CODEPAGE, g_codepage)) == (iconv_t)-1) |
|
172 |
{ |
{ |
173 |
printf("rdp_out_unistr: iconv_open[%s -> %s] fail %d\n", |
size_t i = 1, o = 4; |
174 |
g_codepage, WINDOWS_CODEPAGE, (int)iconv_h); |
if ((iconv_h = iconv_open(WINDOWS_CODEPAGE, g_codepage)) == (iconv_t) - 1) |
175 |
return; |
{ |
176 |
|
warning("rdp_out_unistr: iconv_open[%s -> %s] fail %d\n", |
177 |
|
g_codepage, WINDOWS_CODEPAGE, (int) iconv_h); |
178 |
|
|
179 |
|
g_iconv_works = False; |
180 |
|
rdp_out_unistr(s, string, len); |
181 |
|
return; |
182 |
|
} |
183 |
|
if (iconv(iconv_h, (ICONV_CONST char **) &pin, &i, &pout, &o) == |
184 |
|
(size_t) - 1) |
185 |
|
{ |
186 |
|
iconv_close(iconv_h); |
187 |
|
iconv_h = (iconv_t) - 1; |
188 |
|
warning("rdp_out_unistr: iconv(1) fail, errno %d\n", errno); |
189 |
|
|
190 |
|
g_iconv_works = False; |
191 |
|
rdp_out_unistr(s, string, len); |
192 |
|
return; |
193 |
|
} |
194 |
|
pin = string; |
195 |
|
pout = (char *) s->p; |
196 |
} |
} |
197 |
if (iconv(iconv_h, (const char**)&pin, &i, &pout, &o) == (size_t)-1) |
|
198 |
|
if (iconv(iconv_h, (ICONV_CONST char **) &pin, &ibl, &pout, &obl) == (size_t) - 1) |
199 |
{ |
{ |
200 |
iconv_close(iconv_h); |
iconv_close(iconv_h); |
201 |
iconv_h = (iconv_t)-1; |
iconv_h = (iconv_t) - 1; |
202 |
printf("rdp_out_unistr: iconv(1) fail, errno %d\n", errno); |
warning("rdp_out_unistr: iconv(2) fail, errno %d\n", errno); |
203 |
|
|
204 |
|
g_iconv_works = False; |
205 |
|
rdp_out_unistr(s, string, len); |
206 |
return; |
return; |
207 |
} |
} |
|
pin = string; pout = s->p; |
|
|
} |
|
208 |
|
|
209 |
if (iconv(iconv_h, (const char**)&pin, &ibl, &pout, &obl) == (size_t)-1) |
s->p += len + 2; |
|
{ |
|
|
iconv_close(iconv_h); |
|
|
iconv_h = (iconv_t)-1; |
|
|
printf("rdp_out_unistr: iconv(2) fail, errno %d\n", errno); |
|
|
return; |
|
|
} |
|
210 |
|
|
211 |
#ifdef B_ENDIAN |
} |
212 |
swab(ss, s->p, len + 4); |
else |
213 |
#endif |
#endif |
214 |
|
{ |
215 |
|
int i = 0, j = 0; |
216 |
|
|
217 |
s->p += len + 2; |
len += 2; |
|
|
|
|
#else /*HAVE_ICONV undef*/ |
|
|
int i = 0, j = 0; |
|
218 |
|
|
219 |
len += 2; |
while (i < len) |
220 |
|
{ |
221 |
|
s->p[i++] = string[j++]; |
222 |
|
s->p[i++] = 0; |
223 |
|
} |
224 |
|
|
225 |
while (i < len) |
s->p += len; |
|
{ |
|
|
s->p[i++] = string[j++]; |
|
|
s->p[i++] = 0; |
|
226 |
} |
} |
|
|
|
|
s->p += len; |
|
|
#endif |
|
227 |
} |
} |
228 |
|
|
229 |
/* Input a string in Unicode |
/* Input a string in Unicode |
233 |
int |
int |
234 |
rdp_in_unistr(STREAM s, char *string, int uni_len) |
rdp_in_unistr(STREAM s, char *string, int uni_len) |
235 |
{ |
{ |
236 |
#ifdef HAVE_ICONV |
#ifdef HAVE_ICONV |
237 |
size_t ibl = uni_len, obl = uni_len; |
size_t ibl = uni_len, obl = uni_len; |
238 |
char *pin, *pout = string; |
char *pin = (char *) s->p, *pout = string; |
239 |
static iconv_t iconv_h = (iconv_t)-1; |
static iconv_t iconv_h = (iconv_t) - 1; |
|
#ifdef B_ENDIAN |
|
|
char ss[4096]; // FIXME: global MAX_BUF_SIZE macro need |
|
|
|
|
|
swab(s->p, ss, uni_len); |
|
|
pin = ss; |
|
|
#else |
|
|
pin = s->p; |
|
|
#endif |
|
240 |
|
|
241 |
if (iconv_h == (iconv_t)-1) |
if (g_iconv_works) |
242 |
{ |
{ |
243 |
if ((iconv_h = iconv_open(g_codepage, WINDOWS_CODEPAGE)) == (iconv_t)-1) |
if (iconv_h == (iconv_t) - 1) |
244 |
|
{ |
245 |
|
if ((iconv_h = iconv_open(g_codepage, WINDOWS_CODEPAGE)) == (iconv_t) - 1) |
246 |
|
{ |
247 |
|
warning("rdp_in_unistr: iconv_open[%s -> %s] fail %d\n", |
248 |
|
WINDOWS_CODEPAGE, g_codepage, (int) iconv_h); |
249 |
|
|
250 |
|
g_iconv_works = False; |
251 |
|
return rdp_in_unistr(s, string, uni_len); |
252 |
|
} |
253 |
|
} |
254 |
|
|
255 |
|
if (iconv(iconv_h, (ICONV_CONST char **) &pin, &ibl, &pout, &obl) == (size_t) - 1) |
256 |
{ |
{ |
257 |
printf("rdp_in_unistr: iconv_open[%s -> %s] fail %d\n", |
iconv_close(iconv_h); |
258 |
WINDOWS_CODEPAGE, g_codepage, (int)iconv_h); |
iconv_h = (iconv_t) - 1; |
259 |
return 0; |
warning("rdp_in_unistr: iconv fail, errno %d\n", errno); |
260 |
|
|
261 |
|
g_iconv_works = False; |
262 |
|
return rdp_in_unistr(s, string, uni_len); |
263 |
} |
} |
264 |
|
return pout - string; |
265 |
} |
} |
266 |
|
else |
267 |
if (iconv(iconv_h, (const char**)&pin, &ibl, &pout, &obl) == (size_t)-1) |
#endif |
268 |
{ |
{ |
269 |
iconv_close(iconv_h); |
int i = 0; |
|
iconv_h = (iconv_t)-1; |
|
|
printf("rdp_in_unistr: iconv fail, errno %d\n", errno); |
|
|
return 0; |
|
|
} |
|
|
return pout - string; |
|
|
#else /* HAVE_ICONV undef */ |
|
|
int i = 0; |
|
270 |
|
|
271 |
while (i < uni_len / 2) |
while (i < uni_len / 2) |
272 |
{ |
{ |
273 |
in_uint8a(s, &string[i++], 1); |
in_uint8a(s, &string[i++], 1); |
274 |
in_uint8s(s, 1); |
in_uint8s(s, 1); |
275 |
} |
} |
276 |
|
|
277 |
return i - 1; |
return i - 1; |
278 |
#endif |
} |
279 |
} |
} |
280 |
|
|
281 |
|
|
660 |
static void |
static void |
661 |
rdp_out_bmpcache2_caps(STREAM s) |
rdp_out_bmpcache2_caps(STREAM s) |
662 |
{ |
{ |
663 |
|
uint16 cellsize; |
664 |
|
|
665 |
out_uint16_le(s, RDP_CAPSET_BMPCACHE2); |
out_uint16_le(s, RDP_CAPSET_BMPCACHE2); |
666 |
out_uint16_le(s, RDP_CAPLEN_BMPCACHE2); |
out_uint16_le(s, RDP_CAPLEN_BMPCACHE2); |
667 |
|
|
668 |
out_uint16_le(s, g_bitmap_cache_persist_enable ? 2 : 0); /* version */ |
out_uint16_le(s, g_bitmap_cache_persist_enable ? 2 : 0); /* version */ |
669 |
|
|
670 |
out_uint16_le(s, 0x0300); /* flags? number of caches? */ |
/* Cellsize: |
671 |
|
01 = 16x16, 02 = 32x32, 03 = 64x64 |
672 |
|
log2(cell size) - 3 |
673 |
|
*/ |
674 |
|
|
675 |
|
cellsize = 0x03; |
676 |
|
|
677 |
|
if (g_rdp_compression) |
678 |
|
{ |
679 |
|
switch (g_server_bpp) |
680 |
|
{ |
681 |
|
case 24: |
682 |
|
case 16: |
683 |
|
case 15: |
684 |
|
cellsize = 0x02; |
685 |
|
break; |
686 |
|
} |
687 |
|
} |
688 |
|
|
689 |
|
out_uint16_le(s, (0x0000 | (cellsize << 8))); /* flags? number of caches? */ |
690 |
|
|
691 |
out_uint32_le(s, BMPCACHE2_C0_CELLS); |
out_uint32_le(s, BMPCACHE2_C0_CELLS); |
692 |
out_uint32_le(s, BMPCACHE2_C1_CELLS); |
out_uint32_le(s, BMPCACHE2_C1_CELLS); |
1228 |
if (mppc_expand(s->p, clen, ctype, &roff, &rlen) == -1) |
if (mppc_expand(s->p, clen, ctype, &roff, &rlen) == -1) |
1229 |
error("error while decompressing packet\n"); |
error("error while decompressing packet\n"); |
1230 |
|
|
1231 |
//len -= 18; |
/* len -= 18; */ |
1232 |
|
|
1233 |
/* allocate memory and copy the uncompressed data into the temporary stream */ |
/* allocate memory and copy the uncompressed data into the temporary stream */ |
1234 |
ns->data = (uint8 *) xrealloc(ns->data, rlen); |
ns->data = (uint8 *) xrealloc(ns->data, rlen); |