seamlessrdp/ServerExe/vchannel.c

/* -*- c-basic-offset: 8 -*-
   rdesktop: A Remote Desktop Protocol client.
   Seamless windows - Virtual channel handling

   Copyright (C) Pierre Ossman <ossman@cendio.se> 2006

   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 <assert.h>
#include <stdio.h>
#include <stdarg.h>

#include <windows.h>
#include <wtsapi32.h>
#include <cchannel.h>

#include "vchannel.h"

#define CHANNELNAME "seamrdp"

static HANDLE g_mutex = NULL;
static HANDLE g_vchannel = NULL;

void
debug(char *format, ...)
{
        va_list argp;
        char buf[256];

        sprintf(buf, "DEBUG1,");

        va_start(argp, format);
        _vsnprintf(buf + sizeof("DEBUG1,") - 1, sizeof(buf) - sizeof("DEBUG1,") + 1, format, argp);
        va_end(argp);

        vchannel_write(buf);
}


int
vchannel_open()
{
        g_vchannel = WTSVirtualChannelOpen(WTS_CURRENT_SERVER_HANDLE,
                                           WTS_CURRENT_SESSION, CHANNELNAME);

        if (g_vchannel == NULL)
                return -1;

        g_mutex = CreateMutex(NULL, FALSE, "Local\\SeamlessChannel");
        if (!g_mutex) {
                WTSVirtualChannelClose(g_vchannel);
                g_vchannel = NULL;
                return -1;
        }

        return 0;
}

void
vchannel_close()
{
        if (g_mutex)
                CloseHandle(g_mutex);

        if (g_vchannel)
                WTSVirtualChannelClose(g_vchannel);

        g_mutex = NULL;
        g_vchannel = NULL;
}

int
vchannel_is_open()
{
        if (g_vchannel == NULL)
                return 0;
        else
                return 1;
}

int
vchannel_read(char *buffer, size_t length)
{
        return -1;
}

int
vchannel_write(char *format, ...)
{
        BOOL result;
        va_list argp;
        char buf[1024];
        int size;
        ULONG bytes_written;

        assert(vchannel_is_open());

        va_start(argp, format);
        size = _vsnprintf(buf, sizeof(buf), format, argp);
        va_end(argp);

        assert(size < sizeof(buf));

        WaitForSingleObject(g_mutex, INFINITE);
        result = WTSVirtualChannelWrite(g_vchannel, buf, (ULONG) strlen(buf), &bytes_written);
        result = WTSVirtualChannelWrite(g_vchannel, "\n", (ULONG) 1, &bytes_written);
        ReleaseMutex(g_mutex);

        if (!result)
                return -1;

        return bytes_written;
}
1	ossman_	1073	/* -- c-basic-offset: 8 --
2			rdesktop: A Remote Desktop Protocol client.
3			Seamless windows - Virtual channel handling
4
5			Copyright (C) Pierre Ossman <ossman@cendio.se> 2006
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 2 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, write to the Free Software
19			Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20			*/
21
22			#include <assert.h>
23			#include <stdio.h>
24			#include <stdarg.h>
25
26			#include <windows.h>
27			#include <wtsapi32.h>
28			#include <cchannel.h>
29
30			#include "vchannel.h"
31
32			#define CHANNELNAME "seamrdp"
33
34			static HANDLE g_mutex = NULL;
35			static HANDLE g_vchannel = NULL;
36
37			void
38			debug(char *format, ...)
39			{
40			va_list argp;
41			char buf[256];
42
43			sprintf(buf, "DEBUG1,");
44
45			va_start(argp, format);
46			_vsnprintf(buf + sizeof("DEBUG1,") - 1, sizeof(buf) - sizeof("DEBUG1,") + 1, format, argp);
47			va_end(argp);
48
49			vchannel_write(buf);
50			}
51
52
53			int
54			vchannel_open()
55			{
56			g_vchannel = WTSVirtualChannelOpen(WTS_CURRENT_SERVER_HANDLE,
57			WTS_CURRENT_SESSION, CHANNELNAME);
58
59			if (g_vchannel == NULL)
60			return -1;
61
62			g_mutex = CreateMutex(NULL, FALSE, "Local\\SeamlessChannel");
63			if (!g_mutex) {
64			WTSVirtualChannelClose(g_vchannel);
65			g_vchannel = NULL;
66			return -1;
67			}
68
69			return 0;
70			}
71
72			void
73			vchannel_close()
74			{
75			if (g_mutex)
76			CloseHandle(g_mutex);
77
78			if (g_vchannel)
79			WTSVirtualChannelClose(g_vchannel);
80
81			g_mutex = NULL;
82			g_vchannel = NULL;
83			}
84
85			int
86			vchannel_is_open()
87			{
88			if (g_vchannel == NULL)
89			return 0;
90			else
91			return 1;
92			}
93
94			int
95			vchannel_read(char *buffer, size_t length)
96			{
97			return -1;
98			}
99
100			int
101			vchannel_write(char *format, ...)
102			{
103			BOOL result;
104			va_list argp;
105			char buf[1024];
106			int size;
107			ULONG bytes_written;
108
109			assert(vchannel_is_open());
110
111			va_start(argp, format);
112			size = _vsnprintf(buf, sizeof(buf), format, argp);
113			va_end(argp);
114
115			assert(size < sizeof(buf));
116
117			WaitForSingleObject(g_mutex, INFINITE);
118			result = WTSVirtualChannelWrite(g_vchannel, buf, (ULONG) strlen(buf), &bytes_written);
119			result = WTSVirtualChannelWrite(g_vchannel, "\n", (ULONG) 1, &bytes_written);
120			ReleaseMutex(g_mutex);
121
122			if (!result)
123			return -1;
124
125			return bytes_written;
126			}