1 /* 2 rdesktop: A Remote Desktop Protocol client. 3 Parsing primitives 4 Copyright (C) Matthew Chapman 1999-2008 5 Copyright 2012 Henrik Andersson <hean01@cendio.se> for Cendio AB 6 7 This program is free software: you can redistribute it and/or modify 8 it under the terms of the GNU General Public License as published by 9 the Free Software Foundation, either version 3 of the License, or 10 (at your option) any later version. 11 12 This program is distributed in the hope that it will be useful, 13 but WITHOUT ANY WARRANTY; without even the implied warranty of 14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 15 GNU General Public License for more details. 16 17 You should have received a copy of the GNU General Public License 18 along with this program. If not, see <http://www.gnu.org/licenses/>. 19 */ 20 21 /* Parser state */ 22 typedef struct stream 23 { 24 unsigned char *p; 25 unsigned char *end; 26 unsigned char *data; 27 unsigned int size; 28 29 /* Offsets of various headers */ 30 unsigned char *iso_hdr; 31 unsigned char *mcs_hdr; 32 unsigned char *sec_hdr; 33 unsigned char *rdp_hdr; 34 unsigned char *channel_hdr; 35 36 } 37 *STREAM; 38 39 #define s_push_layer(s,h,n) { (s)->h = (s)->p; (s)->p += n; } 40 #define s_pop_layer(s,h) (s)->p = (s)->h; 41 #define s_mark_end(s) (s)->end = (s)->p; 42 #define s_check(s) ((s)->p <= (s)->end) 43 #define s_check_rem(s,n) ((s)->p + n <= (s)->end) 44 #define s_check_end(s) ((s)->p == (s)->end) 45 #define s_length(s) ((s)->end - (s)->data) 46 #define s_reset(s) ((s)->end = (s)->p = (s)->data) 47 48 #if defined(L_ENDIAN) && !defined(NEED_ALIGN) 49 #define in_uint16_le(s,v) { v = *(uint16 *)((s)->p); (s)->p += 2; } 50 #define in_uint32_le(s,v) { v = *(uint32 *)((s)->p); (s)->p += 4; } 51 #define out_uint16_le(s,v) { *(uint16 *)((s)->p) = v; (s)->p += 2; } 52 #define out_uint32_le(s,v) { *(uint32 *)((s)->p) = v; (s)->p += 4; } 53 54 #else 55 #define in_uint16_le(s,v) { v = *((s)->p++); v += *((s)->p++) << 8; } 56 #define in_uint32_le(s,v) { in_uint16_le(s,v) \ 57 v += *((s)->p++) << 16; v += *((s)->p++) << 24; } 58 #define out_uint16_le(s,v) { *((s)->p++) = (v) & 0xff; *((s)->p++) = ((v) >> 8) & 0xff; } 59 #define out_uint32_le(s,v) { out_uint16_le(s, (v) & 0xffff); out_uint16_le(s, ((v) >> 16) & 0xffff); } 60 #endif 61 62 #if defined(B_ENDIAN) && !defined(NEED_ALIGN) 63 #define in_uint16_be(s,v) { v = *(uint16 *)((s)->p); (s)->p += 2; } 64 #define in_uint32_be(s,v) { v = *(uint32 *)((s)->p); (s)->p += 4; } 65 #define out_uint16_be(s,v) { *(uint16 *)((s)->p) = v; (s)->p += 2; } 66 #define out_uint32_be(s,v) { *(uint32 *)((s)->p) = v; (s)->p += 4; } 67 68 #define B_ENDIAN_PREFERRED 69 #define in_uint16(s,v) in_uint16_be(s,v) 70 #define in_uint32(s,v) in_uint32_be(s,v) 71 #define out_uint16(s,v) out_uint16_be(s,v) 72 #define out_uint32(s,v) out_uint32_be(s,v) 73 74 #else 75 #define in_uint16_be(s,v) { v = *((s)->p++); next_be(s,v); } 76 #define in_uint32_be(s,v) { in_uint16_be(s,v); next_be(s,v); next_be(s,v); } 77 #define out_uint16_be(s,v) { *((s)->p++) = ((v) >> 8) & 0xff; *((s)->p++) = (v) & 0xff; } 78 #define out_uint32_be(s,v) { out_uint16_be(s, ((v) >> 16) & 0xffff); out_uint16_be(s, (v) & 0xffff); } 79 #endif 80 81 #ifndef B_ENDIAN_PREFERRED 82 #define in_uint16(s,v) in_uint16_le(s,v) 83 #define in_uint32(s,v) in_uint32_le(s,v) 84 #define out_uint16(s,v) out_uint16_le(s,v) 85 #define out_uint32(s,v) out_uint32_le(s,v) 86 #endif 87 88 #define in_uint8(s,v) v = *((s)->p++); 89 #define in_uint8p(s,v,n) { v = (s)->p; (s)->p += n; } 90 #define in_uint8a(s,v,n) { memcpy(v,(s)->p,n); (s)->p += n; } 91 #define in_uint8s(s,n) (s)->p += n; 92 #define out_uint8(s,v) *((s)->p++) = v; 93 #define out_uint8p(s,v,n) { memcpy((s)->p,v,n); (s)->p += n; } 94 #define out_uint8a(s,v,n) out_uint8p(s,v,n); 95 #define out_uint8s(s,n) { memset((s)->p,0,n); (s)->p += n; } 96 97 #define next_be(s,v) v = ((v) << 8) + *((s)->p++); 98