trunk/rdesktop/iso.c

/*
   rdesktop: A Remote Desktop Protocol client.
   Protocol services - ISO layer
   Copyright (C) Matthew Chapman 1999-2000
   
   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., 675 Mass Ave, Cambridge, MA 02139, USA.
*/

#include "rdesktop.h"

/* Send a self-contained ISO PDU */
static void iso_send_msg(uint8 code)
{
        STREAM s;

        s = tcp_init(11);

        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);
        tcp_send(s);
}

/* Receive a message on the ISO layer, return code */
static STREAM iso_recv_msg(uint8 *code)
{
        STREAM s;
        uint16 length;
        uint8 version;

        s = tcp_recv(4);
        if (s == NULL)
                return False;

        in_uint8(s, version);
        if (version != 3)
        {
                ERROR("TPKT v%d\n", version);
                return False;
        }

        in_uint8s(s, 1); /* pad */
        in_uint16_be(s, length);

        s = tcp_recv(length - 4);
        if (s == NULL)
                return False;

        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(int length)
{
        STREAM s;

        s = tcp_init(length + 7);
        s_push_layer(s, iso_hdr, 7);

        return s;
}

/* Send an ISO data PDU */
void iso_send(STREAM s)
{
        uint16 length;

        s_pop_layer(s, iso_hdr);
        length = 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 */

        tcp_send(s);
}

/* Receive ISO transport data packet */
STREAM iso_recv()
{
        STREAM s;
        uint8 code;

        s = iso_recv_msg(&code);
        if ((s == NULL) || (code != ISO_PDU_DT))
        {
                ERROR("expected DT, got %d\n", code);
                return False;
        }

        return s;
}

/* Establish a connection up to the ISO layer */
BOOL iso_connect(char *server)
{
        uint8 code;

        if (!tcp_connect(server))
                return False;

        iso_send_msg(ISO_PDU_CR);

        if ((iso_recv_msg(&code) == NULL) || (code != ISO_PDU_CC))
        {
                ERROR("expected CC, got %d\n", code);
                tcp_disconnect();
                return False;
        }

        return True;
}

/* Disconnect from the ISO layer */
void iso_disconnect()
{
        iso_send_msg(ISO_PDU_DR);
        tcp_disconnect();
}
1	/*
2	rdesktop: A Remote Desktop Protocol client.
3	Protocol services - ISO layer
4	Copyright (C) Matthew Chapman 1999-2000
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 iso_send_msg(uint8 code)
25	{
26	STREAM s;
27
28	s = tcp_init(11);
29
30	out_uint8(s, 3); /* version */
31	out_uint8(s, 0); /* reserved */
32	out_uint16_be(s, 11); /* length */
33
34	out_uint8(s, 6); /* hdrlen */
35	out_uint8(s, code);
36	out_uint16(s, 0); /* dst_ref */
37	out_uint16(s, 0); /* src_ref */
38	out_uint8(s, 0); /* class */
39
40	s_mark_end(s);
41	tcp_send(s);
42	}
43
44	/* Receive a message on the ISO layer, return code */
45	static STREAM iso_recv_msg(uint8 *code)
46	{
47	STREAM s;
48	uint16 length;
49	uint8 version;
50
51	s = tcp_recv(4);
52	if (s == NULL)
53	return False;
54
55	in_uint8(s, version);
56	if (version != 3)
57	{
58	ERROR("TPKT v%d\n", version);
59	return False;
60	}
61
62	in_uint8s(s, 1); /* pad */
63	in_uint16_be(s, length);
64
65	s = tcp_recv(length - 4);
66	if (s == NULL)
67	return False;
68
69	in_uint8s(s, 1); /* hdrlen */
70	in_uint8(s, *code);
71
72	if (*code == ISO_PDU_DT)
73	{
74	in_uint8s(s, 1); /* eot */
75	return s;
76	}
77
78	in_uint8s(s, 5); /* dst_ref, src_ref, class */
79	return s;
80	}
81
82	/* Initialise ISO transport data packet */
83	STREAM iso_init(int length)
84	{
85	STREAM s;
86
87	s = tcp_init(length + 7);
88	s_push_layer(s, iso_hdr, 7);
89
90	return s;
91	}
92
93	/* Send an ISO data PDU */
94	void iso_send(STREAM s)
95	{
96	uint16 length;
97
98	s_pop_layer(s, iso_hdr);
99	length = s->end - s->p;
100
101	out_uint8(s, 3); /* version */
102	out_uint8(s, 0); /* reserved */
103	out_uint16_be(s, length);
104
105	out_uint8(s, 2); /* hdrlen */
106	out_uint8(s, ISO_PDU_DT); /* code */
107	out_uint8(s, 0x80); /* eot */
108
109	tcp_send(s);
110	}
111
112	/* Receive ISO transport data packet */
113	STREAM iso_recv()
114	{
115	STREAM s;
116	uint8 code;
117
118	s = iso_recv_msg(&code);
119	if ((s == NULL) \|\| (code != ISO_PDU_DT))
120	{
121	ERROR("expected DT, got %d\n", code);
122	return False;
123	}
124
125	return s;
126	}
127
128	/* Establish a connection up to the ISO layer */
129	BOOL iso_connect(char *server)
130	{
131	uint8 code;
132
133	if (!tcp_connect(server))
134	return False;
135
136	iso_send_msg(ISO_PDU_CR);
137
138	if ((iso_recv_msg(&code) == NULL) \|\| (code != ISO_PDU_CC))
139	{
140	ERROR("expected CC, got %d\n", code);
141	tcp_disconnect();
142	return False;
143	}
144
145	return True;
146	}
147
148	/* Disconnect from the ISO layer */
149	void iso_disconnect()
150	{
151	iso_send_msg(ISO_PDU_DR);
152	tcp_disconnect();
153	}