--- /dev/null
+/* -*- c-basic-offset: 8 -*-
+ rdesktop: A Remote Desktop Protocol client.
+ Protocol services - ISO layer
+ Copyright (C) Matthew Chapman 1999-2005
+
+ This program is free software; you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation; either version 2 of the License, or
+ (at your option) any later version.
+
+ This program is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License along
+ with this program; if not, write to the Free Software Foundation, Inc.,
+ 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+*/
+
+#include "rdesktop.h"
+
+/* Send a self-contained ISO PDU */
+static BOOL
+iso_send_msg(RDPCLIENT * This, uint8 code)
+{
+ STREAM s;
+
+ s = tcp_init(This, 11);
+
+ if(s == NULL)
+ return False;
+
+ out_uint8(s, 3); /* version */
+ out_uint8(s, 0); /* reserved */
+ out_uint16_be(s, 11); /* length */
+
+ out_uint8(s, 6); /* hdrlen */
+ out_uint8(s, code);
+ out_uint16(s, 0); /* dst_ref */
+ out_uint16(s, 0); /* src_ref */
+ out_uint8(s, 0); /* class */
+
+ s_mark_end(s);
+ return tcp_send(This, s);
+}
+
+static BOOL
+iso_send_connection_request(RDPCLIENT * This, char *cookie)
+{
+ STREAM s;
+ int cookielen = (int)strlen(cookie);
+ int length = 11 + cookielen;
+
+ s = tcp_init(This, length);
+
+ if(s == NULL)
+ return False;
+
+ out_uint8(s, 3); /* version */
+ out_uint8(s, 0); /* reserved */
+ out_uint16_be(s, length); /* length */
+
+ out_uint8(s, length - 5); /* hdrlen */
+ out_uint8(s, ISO_PDU_CR);
+ out_uint16(s, 0); /* dst_ref */
+ out_uint16(s, 0); /* src_ref */
+ out_uint8(s, 0); /* class */
+
+ out_uint8p(s, cookie, cookielen);
+
+ s_mark_end(s);
+ return tcp_send(This, s);
+}
+
+/* Receive a message on the ISO layer, return code */
+static STREAM
+iso_recv_msg(RDPCLIENT * This, uint8 * code, uint8 * rdpver)
+{
+ STREAM s;
+ uint16 length;
+ uint8 version;
+
+ s = tcp_recv(This, NULL, 4);
+ if (s == NULL)
+ return NULL;
+ in_uint8(s, version);
+ if (rdpver != NULL)
+ *rdpver = version;
+ if (version == 3)
+ {
+ in_uint8s(s, 1); /* pad */
+ in_uint16_be(s, length);
+ }
+ else
+ {
+ in_uint8(s, length);
+ if (length & 0x80)
+ {
+ length &= ~0x80;
+ next_be(s, length);
+ }
+ }
+ s = tcp_recv(This, s, length - 4);
+ if (s == NULL)
+ return NULL;
+ if (version != 3)
+ return s;
+ in_uint8s(s, 1); /* hdrlen */
+ in_uint8(s, *code);
+ if (*code == ISO_PDU_DT)
+ {
+ in_uint8s(s, 1); /* eot */
+ return s;
+ }
+ in_uint8s(s, 5); /* dst_ref, src_ref, class */
+ return s;
+}
+
+/* Initialise ISO transport data packet */
+STREAM
+iso_init(RDPCLIENT * This, int length)
+{
+ STREAM s;
+
+ s = tcp_init(This, length + 7);
+
+ if(s == NULL)
+ return NULL;
+
+ s_push_layer(s, iso_hdr, 7);
+
+ return s;
+}
+
+/* Send an ISO data PDU */
+BOOL
+iso_send(RDPCLIENT * This, STREAM s)
+{
+ uint16 length;
+
+ s_pop_layer(s, iso_hdr);
+ length = (uint16)(s->end - s->p);
+
+ out_uint8(s, 3); /* version */
+ out_uint8(s, 0); /* reserved */
+ out_uint16_be(s, length);
+
+ out_uint8(s, 2); /* hdrlen */
+ out_uint8(s, ISO_PDU_DT); /* code */
+ out_uint8(s, 0x80); /* eot */
+
+ return tcp_send(This, s);
+}
+
+/* Receive ISO transport data packet */
+STREAM
+iso_recv(RDPCLIENT * This, uint8 * rdpver)
+{
+ STREAM s;
+ uint8 code = 0;
+
+ s = iso_recv_msg(This, &code, rdpver);
+ if (s == NULL)
+ return NULL;
+ if (rdpver != NULL)
+ if (*rdpver != 3)
+ return s;
+ if (code != ISO_PDU_DT)
+ {
+ error("expected DT, got 0x%x\n", code);
+ return NULL;
+ }
+ return s;
+}
+
+/* Establish a connection up to the ISO layer */
+BOOL
+iso_connect(RDPCLIENT * This, char *server, char *cookie)
+{
+ uint8 code = 0;
+
+ if (!tcp_connect(This, server))
+ return False;
+
+ if (!iso_send_connection_request(This, cookie))
+ return False;
+
+ if (iso_recv_msg(This, &code, NULL) == NULL)
+ return False;
+
+ if (code != ISO_PDU_CC)
+ {
+ error("expected CC, got 0x%x\n", code);
+ tcp_disconnect(This);
+ return False;
+ }
+
+ return True;
+}
+
+/* Establish a reconnection up to the ISO layer */
+BOOL
+iso_reconnect(RDPCLIENT * This, char *server, char *cookie)
+{
+ uint8 code = 0;
+
+ if (!tcp_connect(This, server))
+ return False;
+
+ if (!iso_send_connection_request(This, cookie)) // BUGBUG should we really pass the cookie here?
+ return False;
+
+ if (iso_recv_msg(This, &code, NULL) == NULL)
+ return False;
+
+ if (code != ISO_PDU_CC)
+ {
+ error("expected CC, got 0x%x\n", code);
+ tcp_disconnect(This);
+ return False;
+ }
+
+ return True;
+}
+
+/* Disconnect from the ISO layer */
+BOOL
+iso_disconnect(RDPCLIENT * This)
+{
+ iso_send_msg(This, ISO_PDU_DR);
+ return tcp_disconnect(This);
+}
+
+/* reset the state to support reconnecting */
+void
+iso_reset_state(RDPCLIENT * This)
+{
+ tcp_reset_state(This);
+}
projects / reactos.git / blobdiff