]> git.sesse.net Git - rdpsrv/blob - iso.c
a098ad5f50826b3e566cf3d849a62816a5488105
[rdpsrv] / iso.c
1 /* -*- c-basic-offset: 8 -*-
2    rdesktop: A Remote Desktop Protocol client.
3    Protocol services - ISO layer
4    Copyright (C) Matthew Chapman 1999-2002
5    
6    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
8    the Free Software Foundation; either version 2 of the License, or
9    (at your option) any later version.
10    
11    This program is distributed in the hope that it will be useful,
12    but WITHOUT ANY WARRANTY; without even the implied warranty of
13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14    GNU General Public License for more details.
15    
16    You should have received a copy of the GNU General Public License
17    along with this program; if not, write to the Free Software
18    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19 */
20
21 #include "rdesktop.h"
22
23 /* Send a self-contained ISO PDU */
24 static void
25 iso_send_msg(uint8 code, uint8 class)
26 {
27         STREAM s;
28
29         s = tcp_init(11);
30
31         out_uint8(s, 3);        /* version */
32         out_uint8(s, 0);        /* reserved */
33         out_uint16_be(s, 11);   /* length */
34
35         out_uint8(s, 6);        /* hdrlen */
36         out_uint8(s, code);
37         out_uint16(s, 0);               /* dst_ref */
38         out_uint16_be(s, 0x1234);       /* src_ref */
39         out_uint8(s, class);    /* class */
40
41         s_mark_end(s);
42         tcp_send(s);
43 }
44
45 static void
46 iso_send_connection_request(char *username)
47 {
48         STREAM s;
49         int length = 30 + strlen(username);
50
51         s = tcp_init(length);
52
53         out_uint8(s, 3);        /* version */
54         out_uint8(s, 0);        /* reserved */
55         out_uint16_be(s, length);       /* length */
56
57         out_uint8(s, length - 5);       /* hdrlen */
58         out_uint8(s, ISO_PDU_CR);
59         out_uint16(s, 0);       /* dst_ref */
60         out_uint16(s, 0);       /* src_ref */
61         out_uint8(s, 0);        /* class */
62
63         out_uint8p(s, "Cookie: mstshash=", strlen("Cookie: mstshash="));
64         out_uint8p(s, username, strlen(username));
65
66         out_uint8(s, 0x0d);     /* Unknown */
67         out_uint8(s, 0x0a);     /* Unknown */
68
69         s_mark_end(s);
70         tcp_send(s);
71 }
72
73 /* Receive a message on the ISO layer, return code */
74 static STREAM
75 iso_recv_msg(uint8 * code)
76 {
77         STREAM s;
78         uint16 length;
79         uint8 version;
80
81       next_packet:
82         s = tcp_recv(NULL, 4);
83         if (s == NULL)
84                 return NULL;
85
86         in_uint8(s, version);
87         switch (version & 3)
88         {
89                 case 0:
90                         in_uint8(s, length);
91                         if (length & 0x80)
92                         {
93                                 length &= ~0x80;
94                                 next_be(s, length);
95                         }
96                         break;
97
98                 case 3:
99                         in_uint8s(s, 1);        /* pad */
100                         in_uint16_be(s, length);
101                         break;
102
103                 default:
104                         error("TPKT v%d\n", version);
105                         return NULL;
106         }
107
108         s = tcp_recv(s, length - 4);
109         if (s == NULL)
110                 return NULL;
111
112         if ((version & 3) == 0)
113         {
114                 rdp5_process(s, version & 0x80);
115                 return NULL;
116         }
117
118         in_uint8s(s, 1);        /* hdrlen */
119         in_uint8(s, *code);
120
121         if (*code == ISO_PDU_DT)
122         {
123                 in_uint8s(s, 1);        /* eot */
124                 return s;
125         }
126
127         in_uint8s(s, 5);        /* dst_ref, src_ref, class */
128         return s;
129 }
130
131 /* Initialise ISO transport data packet */
132 STREAM
133 iso_init(int length)
134 {
135         STREAM s;
136
137         s = tcp_init(length + 7);
138         s_push_layer(s, iso_hdr, 7);
139
140         return s;
141 }
142
143 /* Send an ISO data PDU */
144 void
145 iso_send(STREAM s)
146 {
147         uint16 length;
148
149         s_pop_layer(s, iso_hdr);
150         length = s->end - s->p;
151
152         out_uint8(s, 3);        /* version */
153         out_uint8(s, 0);        /* reserved */
154         out_uint16_be(s, length);
155
156         out_uint8(s, 2);        /* hdrlen */
157         out_uint8(s, ISO_PDU_DT);       /* code */
158         out_uint8(s, 0x80);     /* eot */
159
160         tcp_send(s);
161 }
162
163 /* Receive ISO transport data packet */
164 STREAM
165 iso_recv(void)
166 {
167         STREAM s;
168         uint8 code;
169
170         s = iso_recv_msg(&code);
171         if (s == NULL)
172                 return NULL;
173
174         if (code != ISO_PDU_DT)
175         {
176                 error("expected DT, got 0x%x\n", code);
177                 return NULL;
178         }
179
180         return s;
181 }
182
183 BOOL
184 iso_recv_connect(int server_sock)
185 {
186         STREAM s, t;
187         uint8 code;
188
189         tcp_recv_connect(server_sock);
190         
191         s = iso_recv_msg(&code);
192         if (s == NULL)
193                 return 0;
194
195         if (code != ISO_PDU_CR)
196         {
197                 error("expected CR, got 0x%x\n", code);
198                 return 0;
199         }
200
201         DEBUG(("Got ISO connection request\n"));
202
203         iso_send_msg(ISO_PDU_CC, 0);
204         return 1;
205 }
206
207 /* Disconnect from the ISO layer */
208 void
209 iso_disconnect(void)
210 {
211         iso_send_msg(ISO_PDU_DR, 0);
212         tcp_disconnect();
213 }