1 |
/* |
/* |
2 |
rdesktop: A Remote Desktop Protocol client. |
rdesktop: A Remote Desktop Protocol client. |
3 |
Protocol services - parsing layer |
Parsing primitives |
4 |
Copyright (C) Matthew Chapman 1999-2000 |
Copyright (C) Matthew Chapman 1999-2002 |
5 |
|
|
6 |
This program is free software; you can redistribute it and/or modify |
This program is free software; you can redistribute it and/or modify |
7 |
it under the terms of the GNU General Public License as published by |
it under the terms of the GNU General Public License as published by |
21 |
/* Parser state */ |
/* Parser state */ |
22 |
typedef struct stream |
typedef struct stream |
23 |
{ |
{ |
24 |
/* Parsing layer */ |
unsigned char *p; |
25 |
unsigned char *data; |
unsigned char *end; |
26 |
|
unsigned char *data; |
27 |
unsigned int size; |
unsigned int size; |
|
unsigned int offset; |
|
|
unsigned int end; |
|
|
BOOL marshall; |
|
|
BOOL error; |
|
|
|
|
|
/* Other layers */ |
|
|
int iso_offset; |
|
|
int mcs_offset; |
|
|
int rdp_offset; |
|
28 |
|
|
29 |
} *STREAM; |
/* Offsets of various headers */ |
30 |
|
unsigned char *iso_hdr; |
31 |
/* Connection state */ |
unsigned char *mcs_hdr; |
32 |
typedef struct connection |
unsigned char *sec_hdr; |
33 |
{ |
unsigned char *rdp_hdr; |
34 |
/* User interface */ |
unsigned char *channel_hdr; |
35 |
HWINDOW wnd; |
|
36 |
HBITMAP bmpcache[8]; |
} |
37 |
|
*STREAM; |
38 |
/* Parsing layer */ |
|
39 |
struct stream in; |
#define s_push_layer(s,h,n) { (s)->h = (s)->p; (s)->p += n; } |
40 |
struct stream out; |
#define s_pop_layer(s,h) (s)->p = (s)->h; |
41 |
|
#define s_mark_end(s) (s)->end = (s)->p; |
42 |
/* TCP layer */ |
#define s_check(s) ((s)->p <= (s)->end) |
43 |
int tcp_socket; |
#define s_check_rem(s,n) ((s)->p + n <= (s)->end) |
44 |
|
#define s_check_end(s) ((s)->p == (s)->end) |
45 |
/* MCS layer */ |
|
46 |
uint16 mcs_userid; |
#if defined(L_ENDIAN) && !defined(NEED_ALIGN) |
47 |
|
#define in_uint16_le(s,v) { v = *(uint16 *)((s)->p); (s)->p += 2; } |
48 |
} *HCONN; |
#define in_uint32_le(s,v) { v = *(uint32 *)((s)->p); (s)->p += 4; } |
49 |
|
#define out_uint16_le(s,v) { *(uint16 *)((s)->p) = v; (s)->p += 2; } |
50 |
#define STREAM_INIT(s,m) { s.data = xmalloc(2048); s.end = s.size = 2048; s.offset = 0; s.marshall = m; s.error = False; } |
#define out_uint32_le(s,v) { *(uint32 *)((s)->p) = v; (s)->p += 4; } |
51 |
#define STREAM_SIZE(s,l) { if (l > s.size) { s.data = xrealloc(s.data,l); s.end = s.size = l; } } |
|
52 |
#define REMAINING(s) ( s->end - s->offset ) |
#else |
53 |
#define PUSH_LAYER(s,v,l) { s.v = s.offset; s.offset += l; } |
#define in_uint16_le(s,v) { v = *((s)->p++); v += *((s)->p++) << 8; } |
54 |
#define POP_LAYER(s,v) { s.offset = s.v; } |
#define in_uint32_le(s,v) { in_uint16_le(s,v) \ |
55 |
#define MARK_END(s) { s.end = s.offset; } |
v += *((s)->p++) << 16; v += *((s)->p++) << 24; } |
56 |
|
#define out_uint16_le(s,v) { *((s)->p++) = (v) & 0xff; *((s)->p++) = ((v) >> 8) & 0xff; } |
57 |
|
#define out_uint32_le(s,v) { out_uint16_le(s, (v) & 0xffff); out_uint16_le(s, ((v) >> 16) & 0xffff); } |
58 |
|
#endif |
59 |
|
|
60 |
|
#if defined(B_ENDIAN) && !defined(NEED_ALIGN) |
61 |
|
#define in_uint16_be(s,v) { v = *(uint16 *)((s)->p); (s)->p += 2; } |
62 |
|
#define in_uint32_be(s,v) { v = *(uint32 *)((s)->p); (s)->p += 4; } |
63 |
|
#define out_uint16_be(s,v) { *(uint16 *)((s)->p) = v; (s)->p += 2; } |
64 |
|
#define out_uint32_be(s,v) { *(uint32 *)((s)->p) = v; (s)->p += 4; } |
65 |
|
|
66 |
|
#define B_ENDIAN_PREFERRED |
67 |
|
#define in_uint16(s,v) in_uint16_be(s,v) |
68 |
|
#define in_uint32(s,v) in_uint32_be(s,v) |
69 |
|
#define out_uint16(s,v) out_uint16_be(s,v) |
70 |
|
#define out_uint32(s,v) out_uint32_be(s,v) |
71 |
|
|
72 |
|
#else |
73 |
|
#define next_be(s,v) v = ((v) << 8) + *((s)->p++); |
74 |
|
#define in_uint16_be(s,v) { v = *((s)->p++); next_be(s,v); } |
75 |
|
#define in_uint32_be(s,v) { in_uint16_be(s,v); next_be(s,v); next_be(s,v); } |
76 |
|
#define out_uint16_be(s,v) { *((s)->p++) = ((v) >> 8) & 0xff; *((s)->p++) = (v) & 0xff; } |
77 |
|
#define out_uint32_be(s,v) { out_uint16_be(s, ((v) >> 16) & 0xffff); out_uint16_be(s, (v) & 0xffff); } |
78 |
|
#endif |
79 |
|
|
80 |
|
#ifndef B_ENDIAN_PREFERRED |
81 |
|
#define in_uint16(s,v) in_uint16_le(s,v) |
82 |
|
#define in_uint32(s,v) in_uint32_le(s,v) |
83 |
|
#define out_uint16(s,v) out_uint16_le(s,v) |
84 |
|
#define out_uint32(s,v) out_uint32_le(s,v) |
85 |
|
#endif |
86 |
|
|
87 |
|
#define in_uint8(s,v) v = *((s)->p++); |
88 |
|
#define in_uint8p(s,v,n) { v = (s)->p; (s)->p += n; } |
89 |
|
#define in_uint8a(s,v,n) { memcpy(v,(s)->p,n); (s)->p += n; } |
90 |
|
#define in_uint8s(s,n) (s)->p += n; |
91 |
|
#define out_uint8(s,v) *((s)->p++) = v; |
92 |
|
#define out_uint8p(s,v,n) { memcpy((s)->p,v,n); (s)->p += n; } |
93 |
|
#define out_uint8a(s,v,n) out_uint8p(s,v,n); |
94 |
|
#define out_uint8s(s,n) { memset((s)->p,0,n); (s)->p += n; } |