31 |
extern uint16 g_server_rdp_version; |
extern uint16 g_server_rdp_version; |
32 |
extern uint32 g_rdp5_performanceflags; |
extern uint32 g_rdp5_performanceflags; |
33 |
extern int g_server_bpp; |
extern int g_server_bpp; |
34 |
|
extern int g_width; |
35 |
|
extern int g_height; |
36 |
|
extern BOOL g_bitmap_cache; |
37 |
|
|
38 |
uint8 *g_next_packet; |
uint8 *g_next_packet; |
39 |
uint32 g_rdp_shareid; |
uint32 g_rdp_shareid; |
40 |
|
|
41 |
|
extern RDPCOMP mppc_dict; |
42 |
|
|
43 |
#if WITH_DEBUG |
#if WITH_DEBUG |
44 |
static uint32 g_packetno; |
static uint32 g_packetno; |
45 |
#endif |
#endif |
176 |
time_t t = time(NULL); |
time_t t = time(NULL); |
177 |
time_t tzone; |
time_t tzone; |
178 |
|
|
|
#if 0 |
|
|
// enable rdp compression |
|
|
flags |= RDP_COMPRESSION; |
|
|
#endif |
|
|
|
|
179 |
if (!g_use_rdp5 || 1 == g_server_rdp_version) |
if (!g_use_rdp5 || 1 == g_server_rdp_version) |
180 |
{ |
{ |
181 |
DEBUG_RDP5(("Sending RDP4-style Logon packet\n")); |
DEBUG_RDP5(("Sending RDP4-style Logon packet\n")); |
182 |
|
|
183 |
|
#if 0 |
184 |
|
/* enable rdp compression */ |
185 |
|
/* decompression also works with rdp5 */ |
186 |
|
/* but there are some unknown opcodes */ |
187 |
|
flags |= RDP_COMPRESSION; |
188 |
|
#endif |
189 |
s = sec_init(sec_flags, 18 + len_domain + len_user + len_password |
s = sec_init(sec_flags, 18 + len_domain + len_user + len_password |
190 |
+ len_program + len_directory + 10); |
+ len_program + len_directory + 10); |
191 |
|
|
204 |
} |
} |
205 |
else |
else |
206 |
{ |
{ |
207 |
|
|
208 |
flags |= RDP_LOGON_BLOB; |
flags |= RDP_LOGON_BLOB; |
209 |
DEBUG_RDP5(("Sending RDP5-style Logon packet\n")); |
DEBUG_RDP5(("Sending RDP5-style Logon packet\n")); |
210 |
packetlen = 4 + /* Unknown uint32 */ |
packetlen = 4 + /* Unknown uint32 */ |
375 |
s = rdp_init_data(8); |
s = rdp_init_data(8); |
376 |
|
|
377 |
out_uint16(s, 0); /* number of fonts */ |
out_uint16(s, 0); /* number of fonts */ |
378 |
out_uint16_le(s, 0x3e); /* unknown */ |
out_uint16_le(s, 0); /* pad? */ |
379 |
out_uint16_le(s, seq); /* unknown */ |
out_uint16_le(s, seq); /* unknown */ |
380 |
out_uint16_le(s, 0x32); /* entry size */ |
out_uint16_le(s, 0x32); /* entry size */ |
381 |
|
|
423 |
out_uint16_le(s, 800); /* Desktop width */ |
out_uint16_le(s, 800); /* Desktop width */ |
424 |
out_uint16_le(s, 600); /* Desktop height */ |
out_uint16_le(s, 600); /* Desktop height */ |
425 |
out_uint16(s, 0); /* Pad */ |
out_uint16(s, 0); /* Pad */ |
426 |
out_uint16(s, 0); /* Allow resize */ |
out_uint16(s, 1); /* Allow resize */ |
427 |
out_uint16_le(s, g_bitmap_compression ? 1 : 0); /* Support compression */ |
out_uint16_le(s, g_bitmap_compression ? 1 : 0); /* Support compression */ |
428 |
out_uint16(s, 0); /* Unknown */ |
out_uint16(s, 0); /* Unknown */ |
429 |
out_uint16_le(s, 1); /* Unknown */ |
out_uint16_le(s, 1); /* Unknown */ |
441 |
order_caps[0] = 1; /* dest blt */ |
order_caps[0] = 1; /* dest blt */ |
442 |
order_caps[1] = 1; /* pat blt */ |
order_caps[1] = 1; /* pat blt */ |
443 |
order_caps[2] = 1; /* screen blt */ |
order_caps[2] = 1; /* screen blt */ |
444 |
order_caps[3] = 1; /* required for memblt? */ |
order_caps[3] = (g_bitmap_cache ? 1 : 0); /* memblt */ |
445 |
order_caps[8] = 1; /* line */ |
order_caps[8] = 1; /* line */ |
446 |
order_caps[9] = 1; /* line */ |
order_caps[9] = 1; /* line */ |
447 |
order_caps[10] = 1; /* rect */ |
order_caps[10] = 1; /* rect */ |
620 |
sec_send(s, sec_flags); |
sec_send(s, sec_flags); |
621 |
} |
} |
622 |
|
|
623 |
|
/* Process a general capability set */ |
624 |
|
static void |
625 |
|
rdp_process_general_caps(STREAM s) |
626 |
|
{ |
627 |
|
uint16 pad2octetsB; /* rdp5 flags? */ |
628 |
|
|
629 |
|
in_uint8s(s, 10); |
630 |
|
in_uint16_le(s, pad2octetsB); |
631 |
|
|
632 |
|
if (!pad2octetsB) |
633 |
|
g_use_rdp5 = False; |
634 |
|
} |
635 |
|
|
636 |
|
/* Process a bitmap capability set */ |
637 |
|
static void |
638 |
|
rdp_process_bitmap_caps(STREAM s) |
639 |
|
{ |
640 |
|
uint16 width, height, bpp; |
641 |
|
|
642 |
|
in_uint16_le(s, bpp); |
643 |
|
in_uint8s(s, 6); |
644 |
|
|
645 |
|
in_uint16_le(s, width); |
646 |
|
in_uint16_le(s, height); |
647 |
|
|
648 |
|
DEBUG(("setting desktop size and bpp to: %dx%dx%d\n", width, height, bpp)); |
649 |
|
|
650 |
|
/* |
651 |
|
* The server may limit bpp and change the size of the desktop (for |
652 |
|
* example when shadowing another session). |
653 |
|
*/ |
654 |
|
g_server_bpp = bpp; |
655 |
|
g_width = width; |
656 |
|
g_height = height; |
657 |
|
|
658 |
|
ui_resize_window(); |
659 |
|
} |
660 |
|
|
661 |
/* Respond to a demand active PDU */ |
/* Respond to a demand active PDU */ |
662 |
static void |
static void |
663 |
process_demand_active(STREAM s) |
process_demand_active(STREAM s) |
664 |
{ |
{ |
665 |
uint8 type; |
int n; |
666 |
uint16 i; |
uint8 type, *next; |
667 |
uint16 p_bpp; |
uint16 len_src_descriptor, len_combined_caps, num_capsets, capset_type, capset_length; |
668 |
|
|
669 |
in_uint32_le(s, g_rdp_shareid); |
in_uint32_le(s, g_rdp_shareid); |
670 |
|
in_uint16_le(s, len_src_descriptor); |
671 |
|
in_uint16_le(s, len_combined_caps); |
672 |
|
in_uint8s(s, len_src_descriptor); |
673 |
|
|
674 |
/* scan for prefered bpp */ |
in_uint16_le(s, num_capsets); |
675 |
while (s_check_rem(s, 6)) |
in_uint8s(s, 2); /* pad */ |
676 |
|
|
677 |
|
DEBUG(("DEMAND_ACTIVE(id=0x%x,num_caps=%d)\n", g_rdp_shareid, num_capsets)); |
678 |
|
|
679 |
|
for (n = 0; n < num_capsets; n++) |
680 |
{ |
{ |
681 |
in_uint16_le(s, i); |
in_uint16_le(s, capset_type); |
682 |
if (i == RDP_CAPSET_BITMAP) |
in_uint16_le(s, capset_length); |
683 |
|
|
684 |
|
next = s->p + capset_length - 4; |
685 |
|
|
686 |
|
switch (capset_type) |
687 |
{ |
{ |
688 |
in_uint16_le(s, i); |
case RDP_CAPSET_GENERAL: |
689 |
if (i == RDP_CAPLEN_BITMAP) |
rdp_process_general_caps(s); |
690 |
{ |
break; |
|
in_uint16_le(s, p_bpp); |
|
|
if (p_bpp == 8 || p_bpp == 15 || p_bpp == 16 || p_bpp == 24) |
|
|
{ |
|
|
if (p_bpp < g_server_bpp) |
|
|
{ |
|
|
warning("Server limited colour depth to %d bits\n", p_bpp); |
|
|
g_server_bpp = p_bpp; |
|
|
} |
|
|
break; |
|
|
} |
|
|
} |
|
|
} |
|
|
} |
|
691 |
|
|
692 |
|
case RDP_CAPSET_BITMAP: |
693 |
|
rdp_process_bitmap_caps(s); |
694 |
|
break; |
695 |
|
} |
696 |
|
|
697 |
DEBUG(("DEMAND_ACTIVE(id=0x%x)\n", g_rdp_shareid)); |
s->p = next; |
698 |
|
} |
699 |
|
|
700 |
rdp_send_confirm_active(); |
rdp_send_confirm_active(); |
701 |
rdp_send_synchronise(); |
rdp_send_synchronise(); |
705 |
rdp_recv(&type); /* RDP_CTL_COOPERATE */ |
rdp_recv(&type); /* RDP_CTL_COOPERATE */ |
706 |
rdp_recv(&type); /* RDP_CTL_GRANT_CONTROL */ |
rdp_recv(&type); /* RDP_CTL_GRANT_CONTROL */ |
707 |
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); |
708 |
rdp_send_fonts(1); |
|
709 |
rdp_send_fonts(2); |
if (g_use_rdp5) |
710 |
rdp_recv(&type); /* RDP_PDU_UNKNOWN 0x28 */ |
{ |
711 |
|
rdp_send_fonts(3); |
712 |
|
} |
713 |
|
else |
714 |
|
{ |
715 |
|
rdp_send_fonts(1); |
716 |
|
rdp_send_fonts(2); |
717 |
|
} |
718 |
|
|
719 |
|
rdp_recv(&type); /* RDP_PDU_UNKNOWN 0x28 (Fonts?) */ |
720 |
reset_order_state(); |
reset_order_state(); |
721 |
} |
} |
722 |
|
|
942 |
|
|
943 |
} |
} |
944 |
|
|
945 |
|
/* Process a disconnect PDU */ |
946 |
|
void |
947 |
|
process_disconnect_pdu(STREAM s, uint32 * ext_disc_reason) |
948 |
|
{ |
949 |
|
in_uint32_le(s, *ext_disc_reason); |
950 |
|
|
951 |
|
DEBUG(("Received disconnect PDU\n")); |
952 |
|
} |
953 |
|
|
954 |
/* Process data PDU */ |
/* Process data PDU */ |
955 |
static void |
static BOOL |
956 |
process_data_pdu(STREAM s) |
process_data_pdu(STREAM s, uint32 * ext_disc_reason) |
957 |
{ |
{ |
958 |
uint8 data_pdu_type; |
uint8 data_pdu_type; |
959 |
uint8 ctype; |
uint8 ctype; |
960 |
uint16 clen; |
uint16 clen; |
961 |
int roff, rlen, len, ret; |
uint32 len; |
962 |
static struct stream ns; |
|
963 |
static signed char *dict = 0; |
uint32 roff, rlen; |
964 |
|
|
965 |
|
struct stream *ns = &(mppc_dict.ns); |
966 |
|
uint8 *dict = (mppc_dict.hist); |
967 |
|
|
968 |
in_uint8s(s, 6); /* shareid, pad, streamid */ |
in_uint8s(s, 6); /* shareid, pad, streamid */ |
969 |
in_uint16(s, len); |
in_uint16(s, len); |
972 |
in_uint16(s, clen); |
in_uint16(s, clen); |
973 |
clen -= 18; |
clen -= 18; |
974 |
|
|
975 |
#if 0 |
if (ctype & RDP_MPPC_COMPRESSED) |
|
if (ctype & 0x20) |
|
976 |
{ |
{ |
|
if (!dict) |
|
|
{ |
|
|
dict = (signed char *) malloc(8200 * sizeof(signed char)); |
|
|
dict = (signed char *) memset(dict, 0, 8200 * sizeof(signed char)); |
|
|
} |
|
977 |
|
|
978 |
ret = decompress(s->p, clen, ctype, (signed char *) dict, &roff, &rlen); |
if (mppc_expand(s->p, clen, ctype, &roff, &rlen) == -1) |
979 |
|
error("error while decompressing packet\n"); |
980 |
|
|
981 |
len -= 18; |
len -= 18; |
982 |
|
|
983 |
ns.data = xrealloc(ns.data, len); |
/* this should never happen */ |
984 |
|
if (len != rlen) |
985 |
|
error("decompression error len != rlen\n"); |
986 |
|
|
987 |
|
/* allocate memory and copy the uncompressed data into the temporary stream */ |
988 |
|
ns->data = xrealloc(ns->data, len); |
989 |
|
|
990 |
ns.data = (unsigned char *) memcpy(ns.data, (unsigned char *) (dict + roff), len); |
memcpy((ns->data), (unsigned char *) (mppc_dict.hist + roff), len); |
991 |
|
|
992 |
ns.size = len; |
ns->size = len; |
993 |
ns.end = ns.data + ns.size; |
ns->end = (ns->data + ns->size); |
994 |
ns.p = ns.data; |
ns->p = ns->data; |
995 |
ns.rdp_hdr = ns.p; |
ns->rdp_hdr = ns->p; |
996 |
|
|
997 |
s = &ns; |
s = ns; |
998 |
} |
} |
|
#endif |
|
999 |
|
|
1000 |
switch (data_pdu_type) |
switch (data_pdu_type) |
1001 |
{ |
{ |
1003 |
process_update_pdu(s); |
process_update_pdu(s); |
1004 |
break; |
break; |
1005 |
|
|
1006 |
|
case RDP_DATA_PDU_CONTROL: |
1007 |
|
DEBUG(("Received Control PDU\n")); |
1008 |
|
break; |
1009 |
|
|
1010 |
|
case RDP_DATA_PDU_SYNCHRONISE: |
1011 |
|
DEBUG(("Received Sync PDU\n")); |
1012 |
|
break; |
1013 |
|
|
1014 |
case RDP_DATA_PDU_POINTER: |
case RDP_DATA_PDU_POINTER: |
1015 |
process_pointer_pdu(s); |
process_pointer_pdu(s); |
1016 |
break; |
break; |
1025 |
break; |
break; |
1026 |
|
|
1027 |
case RDP_DATA_PDU_DISCONNECT: |
case RDP_DATA_PDU_DISCONNECT: |
1028 |
/* Normally received when user logs out or disconnects from a |
process_disconnect_pdu(s, ext_disc_reason); |
1029 |
console session on Windows XP and 2003 Server */ |
return True; |
|
DEBUG(("Received disconnect PDU\n")); |
|
|
break; |
|
1030 |
|
|
1031 |
default: |
default: |
1032 |
unimpl("data PDU %d\n", data_pdu_type); |
unimpl("data PDU %d\n", data_pdu_type); |
1033 |
} |
} |
1034 |
|
return False; |
1035 |
} |
} |
1036 |
|
|
1037 |
/* Process incoming packets */ |
/* Process incoming packets */ |
1038 |
BOOL |
void |
1039 |
rdp_main_loop(void) |
rdp_main_loop(BOOL * deactivated, uint32 * ext_disc_reason) |
1040 |
{ |
{ |
1041 |
uint8 type; |
uint8 type; |
1042 |
|
BOOL disc = False; /* True when a disconnect PDU was received */ |
1043 |
STREAM s; |
STREAM s; |
1044 |
|
|
1045 |
while ((s = rdp_recv(&type)) != NULL) |
while ((s = rdp_recv(&type)) != NULL) |
1048 |
{ |
{ |
1049 |
case RDP_PDU_DEMAND_ACTIVE: |
case RDP_PDU_DEMAND_ACTIVE: |
1050 |
process_demand_active(s); |
process_demand_active(s); |
1051 |
|
*deactivated = False; |
1052 |
break; |
break; |
1053 |
|
|
1054 |
case RDP_PDU_DEACTIVATE: |
case RDP_PDU_DEACTIVATE: |
1055 |
DEBUG(("RDP_PDU_DEACTIVATE\n")); |
DEBUG(("RDP_PDU_DEACTIVATE\n")); |
1056 |
/* 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; */ |
|
1057 |
break; |
break; |
1058 |
|
|
1059 |
case RDP_PDU_DATA: |
case RDP_PDU_DATA: |
1060 |
process_data_pdu(s); |
disc = process_data_pdu(s, ext_disc_reason); |
1061 |
break; |
break; |
1062 |
|
|
1063 |
case 0: |
case 0: |
1066 |
default: |
default: |
1067 |
unimpl("PDU %d\n", type); |
unimpl("PDU %d\n", type); |
1068 |
} |
} |
1069 |
|
|
1070 |
|
if (disc) |
1071 |
|
{ |
1072 |
|
return; |
1073 |
|
} |
1074 |
} |
} |
1075 |
return True; |
return; |
|
/* We want to detect if we got a clean shutdown, but we |
|
|
can't. Se above. |
|
|
return False; */ |
|
1076 |
} |
} |
1077 |
|
|
1078 |
/* Establish a connection up to the RDP layer */ |
/* Establish a connection up to the RDP layer */ |