trunk/rdesktop/ipc.c

/* -*- c-basic-offset: 8 -*-
   rdesktop: A Remote Desktop Protocol client.
   Communication between different rdesktop processes using X properties
   Copyright (C) Erik Forsberg <forsberg@cendio.se> 2003

   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 <X11/Xlib.h>
#include <X11/Xatom.h>
#include "rdesktop.h"
#include "xproto.h"

Atom ipc_atom;

extern Display *display;
extern Window wnd;

static struct stream in;
static struct stream out;
static const uint16 headerlen = 12;

struct _ipc_channel;

typedef struct _ipc_channel
{
        uint16 messagetype;
        void (*callback) (unsigned char *, uint16 length);
        struct _ipc_channel *next;
}
ipc_channel;

static ipc_channel *ipc_channels = NULL;

void
ipc_register_ipcnotify(uint16 messagetype, void (*notifycallback) (unsigned char *, uint16))
{
        ipc_channel *this;
        if (NULL != ipc_channels)
        {
                this = ipc_channels;
                while (NULL != this->next)
                        this = this->next;
                this->next = xmalloc(sizeof(ipc_channel));
                this->next->next = NULL;
                this = this->next;
        }
        else
        {
                this = xmalloc(sizeof(ipc_channel));
                this->next = NULL;
                ipc_channels = this;
        }
        this->messagetype = messagetype;
        this->callback = notifycallback;
}

void
ipc_deregister_ipcnotify(uint16 messagetype)
{
        ipc_channel *this, *prev;
        prev = this = ipc_channels;
        while (NULL != this)
        {
                if (this->messagetype == messagetype)
                {
                        if (prev == this)
                                ipc_channels = this->next;
                        else
                                prev->next = this->next;
                        xfree(this);
                        return;
                }
        }
}

void
ipc_recv_message(XPropertyEvent * xev)
{
        int actual_format_return;
        unsigned long nitems_return, bytes_after_return;
        unsigned char *prop_return;
        unsigned char *data = NULL;
        uint16 totalsize, rdesktop_ipc_version, messagetype;
        uint32 sender_wnd;
        Atom actual_type_return;
        ipc_channel *channel = ipc_channels;

        DEBUG_RDP5(("Got event in ipc_recv_message\n"));

        in.end = in.p = in.data;

        XGetWindowProperty(display, DefaultRootWindow(display), ipc_atom, 0, 1, False,  // Delete
                           AnyPropertyType,
                           &actual_type_return,
                           &actual_format_return,
                           &nitems_return, &bytes_after_return, &prop_return);

        memcpy(in.end, prop_return, 2);
        XFree(prop_return);
        in_uint32_le(&in, totalsize);
        in.end += 4;

        DEBUG_RDP5(("ipc totalsize is %d\n", totalsize));

        XGetWindowProperty(display, DefaultRootWindow(display), ipc_atom, 1, (totalsize - 1) / 4, False,        // Delete
                           AnyPropertyType,
                           &actual_type_return,
                           &actual_format_return,
                           &nitems_return, &bytes_after_return, &prop_return);

        memcpy(in.end, prop_return, totalsize - 4);
        XFree(prop_return);

        in_uint16_le(&in, rdesktop_ipc_version);

        DEBUG_RDP5(("Got rdesktop_ipc_version %d\n", rdesktop_ipc_version));

        if (rdesktop_ipc_version > RDESKTOP_IPC_VERSION)
        {
                warning("IPC version of sending Rdesktop is higher than ours. Returning without trying to parse message\n");
                return;
        }

        in_uint16_le(&in, messagetype);
        in_uint32_le(&in, sender_wnd);

        DEBUG_RDP5(("header: %d, %d, %d\n", rdesktop_ipc_version, messagetype, sender_wnd));


        if (sender_wnd == wnd)
        {
                DEBUG_RDP5(("We sent this message, returning..\n"));
                return;         /* We are not interested in our own events.. */
        }

        DEBUG_RDP5(("Not our window..\n"));

        /* Check if we are interested by traversing our callback list */
        while (NULL != channel)
        {
                DEBUG_RDP5(("Found channel for messagetype %d\n", channel->messagetype));
                if (messagetype == channel->messagetype)
                {
                        data = xmalloc(in.size - headerlen);
                        in_uint8p(&in, data, totalsize - headerlen);
                        channel->callback(data, totalsize - headerlen);
                        return;
                }
                /* Callback is responsible for freeing data */
                channel = channel->next;
        }
}

void
ipc_init(void)
{
        ipc_atom = XInternAtom(display, "_RDESKTOP_IPC", False);
        xwin_register_propertynotify(DefaultRootWindow(display), ipc_atom, ipc_recv_message);
        in.size = 4096;
        in.data = xmalloc(in.size);
        out.size = 4096;
        out.data = xmalloc(out.size);
}

void
ipc_send_message(uint16 messagetype, unsigned char *data, uint16 length)
{

        if ((length + headerlen) > out.size)
        {
                out.data = xrealloc(out.data, length + headerlen);
                out.size = length + headerlen;
        }
        out.p = out.data;
        out.end = out.data + out.size;

        out_uint32_le(&out, length + headerlen);

        out_uint16_le(&out, RDESKTOP_IPC_VERSION);
        out_uint16_le(&out, messagetype);
        out_uint32_le(&out, wnd);

        out_uint8p(&out, data, length);
        s_mark_end(&out);

        DEBUG_RDP5(("length+headerlen is %d\n", length + headerlen));

        XChangeProperty(display,
                        DefaultRootWindow(display),
                        ipc_atom, XA_STRING, 8, PropModeReplace, out.data, out.end - out.data);
        XFlush(display);
}
1	/* -- c-basic-offset: 8 --
2	rdesktop: A Remote Desktop Protocol client.
3	Communication between different rdesktop processes using X properties
4	Copyright (C) Erik Forsberg <forsberg@cendio.se> 2003
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 <X11/Xlib.h>
22	#include <X11/Xatom.h>
23	#include "rdesktop.h"
24	#include "xproto.h"
25
26	Atom ipc_atom;
27
28	extern Display *display;
29	extern Window wnd;
30
31	static struct stream in;
32	static struct stream out;
33	static const uint16 headerlen = 12;
34
35	struct _ipc_channel;
36
37	typedef struct _ipc_channel
38	{
39	uint16 messagetype;
40	void (callback) (unsigned char , uint16 length);
41	struct _ipc_channel *next;
42	}
43	ipc_channel;
44
45	static ipc_channel *ipc_channels = NULL;
46
47	void
48	ipc_register_ipcnotify(uint16 messagetype, void (notifycallback) (unsigned char , uint16))
49	{
50	ipc_channel *this;
51	if (NULL != ipc_channels)
52	{
53	this = ipc_channels;
54	while (NULL != this->next)
55	this = this->next;
56	this->next = xmalloc(sizeof(ipc_channel));
57	this->next->next = NULL;
58	this = this->next;
59	}
60	else
61	{
62	this = xmalloc(sizeof(ipc_channel));
63	this->next = NULL;
64	ipc_channels = this;
65	}
66	this->messagetype = messagetype;
67	this->callback = notifycallback;
68	}
69
70	void
71	ipc_deregister_ipcnotify(uint16 messagetype)
72	{
73	ipc_channel this, prev;
74	prev = this = ipc_channels;
75	while (NULL != this)
76	{
77	if (this->messagetype == messagetype)
78	{
79	if (prev == this)
80	ipc_channels = this->next;
81	else
82	prev->next = this->next;
83	xfree(this);
84	return;
85	}
86	}
87	}
88
89	void
90	ipc_recv_message(XPropertyEvent * xev)
91	{
92	int actual_format_return;
93	unsigned long nitems_return, bytes_after_return;
94	unsigned char *prop_return;
95	unsigned char *data = NULL;
96	uint16 totalsize, rdesktop_ipc_version, messagetype;
97	uint32 sender_wnd;
98	Atom actual_type_return;
99	ipc_channel *channel = ipc_channels;
100
101	DEBUG_RDP5(("Got event in ipc_recv_message\n"));
102
103	in.end = in.p = in.data;
104
105	XGetWindowProperty(display, DefaultRootWindow(display), ipc_atom, 0, 1, False, // Delete
106	AnyPropertyType,
107	&actual_type_return,
108	&actual_format_return,
109	&nitems_return, &bytes_after_return, &prop_return);
110
111	memcpy(in.end, prop_return, 2);
112	XFree(prop_return);
113	in_uint32_le(&in, totalsize);
114	in.end += 4;
115
116	DEBUG_RDP5(("ipc totalsize is %d\n", totalsize));
117
118	XGetWindowProperty(display, DefaultRootWindow(display), ipc_atom, 1, (totalsize - 1) / 4, False, // Delete
119	AnyPropertyType,
120	&actual_type_return,
121	&actual_format_return,
122	&nitems_return, &bytes_after_return, &prop_return);
123
124	memcpy(in.end, prop_return, totalsize - 4);
125	XFree(prop_return);
126
127	in_uint16_le(&in, rdesktop_ipc_version);
128
129	DEBUG_RDP5(("Got rdesktop_ipc_version %d\n", rdesktop_ipc_version));
130
131	if (rdesktop_ipc_version > RDESKTOP_IPC_VERSION)
132	{
133	warning("IPC version of sending Rdesktop is higher than ours. Returning without trying to parse message\n");
134	return;
135	}
136
137	in_uint16_le(&in, messagetype);
138	in_uint32_le(&in, sender_wnd);
139
140	DEBUG_RDP5(("header: %d, %d, %d\n", rdesktop_ipc_version, messagetype, sender_wnd));
141
142
143	if (sender_wnd == wnd)
144	{
145	DEBUG_RDP5(("We sent this message, returning..\n"));
146	return; /* We are not interested in our own events.. */
147	}
148
149	DEBUG_RDP5(("Not our window..\n"));
150
151	/* Check if we are interested by traversing our callback list */
152	while (NULL != channel)
153	{
154	DEBUG_RDP5(("Found channel for messagetype %d\n", channel->messagetype));
155	if (messagetype == channel->messagetype)
156	{
157	data = xmalloc(in.size - headerlen);
158	in_uint8p(&in, data, totalsize - headerlen);
159	channel->callback(data, totalsize - headerlen);
160	return;
161	}
162	/* Callback is responsible for freeing data */
163	channel = channel->next;
164	}
165	}
166
167	void
168	ipc_init(void)
169	{
170	ipc_atom = XInternAtom(display, "_RDESKTOP_IPC", False);
171	xwin_register_propertynotify(DefaultRootWindow(display), ipc_atom, ipc_recv_message);
172	in.size = 4096;
173	in.data = xmalloc(in.size);
174	out.size = 4096;
175	out.data = xmalloc(out.size);
176	}
177
178	void
179	ipc_send_message(uint16 messagetype, unsigned char *data, uint16 length)
180	{
181
182	if ((length + headerlen) > out.size)
183	{
184	out.data = xrealloc(out.data, length + headerlen);
185	out.size = length + headerlen;
186	}
187	out.p = out.data;
188	out.end = out.data + out.size;
189
190	out_uint32_le(&out, length + headerlen);
191
192	out_uint16_le(&out, RDESKTOP_IPC_VERSION);
193	out_uint16_le(&out, messagetype);
194	out_uint32_le(&out, wnd);
195
196	out_uint8p(&out, data, length);
197	s_mark_end(&out);
198
199	DEBUG_RDP5(("length+headerlen is %d\n", length + headerlen));
200
201	XChangeProperty(display,
202	DefaultRootWindow(display),
203	ipc_atom, XA_STRING, 8, PropModeReplace, out.data, out.end - out.data);
204	XFlush(display);
205	}