1 |
|
/* -*- c-basic-offset: 8 -*- |
2 |
|
rdesktop: A Remote Desktop Protocol client. |
3 |
|
Copyright (C) Matthew Chapman 1999-2004 |
4 |
|
|
5 |
|
This program is free software; you can redistribute it and/or modify |
6 |
|
it under the terms of the GNU General Public License as published by |
7 |
|
the Free Software Foundation; either version 2 of the License, or |
8 |
|
(at your option) any later version. |
9 |
|
|
10 |
|
This program is distributed in the hope that it will be useful, |
11 |
|
but WITHOUT ANY WARRANTY; without even the implied warranty of |
12 |
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
13 |
|
GNU General Public License for more details. |
14 |
|
|
15 |
|
You should have received a copy of the GNU General Public License |
16 |
|
along with this program; if not, write to the Free Software |
17 |
|
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. |
18 |
|
*/ |
19 |
|
|
20 |
|
/* |
21 |
|
Here are some resources, for your IRP hacking pleasure: |
22 |
|
|
23 |
|
http://cvs.sourceforge.net/viewcvs.py/mingw/w32api/include/ddk/winddk.h?view=markup |
24 |
|
|
25 |
|
http://win32.mvps.org/ntfs/streams.cpp |
26 |
|
|
27 |
|
http://www.acc.umu.se/~bosse/ntifs.h |
28 |
|
|
29 |
|
http://undocumented.ntinternals.net/UserMode/Undocumented%20Functions/NT%20Objects/File/ |
30 |
|
|
31 |
|
http://us1.samba.org/samba/ftp/specs/smb-nt01.txt |
32 |
|
|
33 |
|
http://www.osronline.com/ |
34 |
|
*/ |
35 |
|
|
36 |
#include <unistd.h> |
#include <unistd.h> |
37 |
#include <sys/types.h> |
#include <sys/types.h> |
38 |
#include <sys/time.h> |
#include <sys/time.h> |
60 |
extern DEVICE_FNS parallel_fns; |
extern DEVICE_FNS parallel_fns; |
61 |
extern DEVICE_FNS disk_fns; |
extern DEVICE_FNS disk_fns; |
62 |
extern FILEINFO g_fileinfo[]; |
extern FILEINFO g_fileinfo[]; |
63 |
|
|
64 |
static VCHANNEL *rdpdr_channel; |
static VCHANNEL *rdpdr_channel; |
65 |
|
|
66 |
/* If select() times out, the request for the device with handle g_min_timeout_fd is aborted */ |
/* If select() times out, the request for the device with handle g_min_timeout_fd is aborted */ |
69 |
|
|
70 |
/* Table with information about rdpdr devices */ |
/* Table with information about rdpdr devices */ |
71 |
RDPDR_DEVICE g_rdpdr_device[RDPDR_MAX_DEVICES]; |
RDPDR_DEVICE g_rdpdr_device[RDPDR_MAX_DEVICES]; |
72 |
char * g_rdpdr_clientname = NULL; |
char *g_rdpdr_clientname = NULL; |
73 |
|
|
74 |
/* Used to store incoming io request, until they are ready to be completed */ |
/* Used to store incoming io request, until they are ready to be completed */ |
75 |
/* using a linked list ensures that they are processed in the right order, */ |
/* using a linked list ensures that they are processed in the right order, */ |
112 |
} |
} |
113 |
} |
} |
114 |
|
|
115 |
BOOL |
static BOOL |
116 |
rdpdr_handle_ok(int device, int handle) |
rdpdr_handle_ok(int device, int handle) |
117 |
{ |
{ |
118 |
switch (g_rdpdr_device[device].device_type) |
switch (g_rdpdr_device[device].device_type) |
133 |
} |
} |
134 |
|
|
135 |
/* Add a new io request to the table containing pending io requests so it won't block rdesktop */ |
/* Add a new io request to the table containing pending io requests so it won't block rdesktop */ |
136 |
BOOL |
static BOOL |
137 |
add_async_iorequest(uint32 device, uint32 file, uint32 id, uint32 major, uint32 length, |
add_async_iorequest(uint32 device, uint32 file, uint32 id, uint32 major, uint32 length, |
138 |
DEVICE_FNS * fns, uint32 total_timeout, uint32 interval_timeout, uint8 * buffer, |
DEVICE_FNS * fns, uint32 total_timeout, uint32 interval_timeout, uint8 * buffer, |
139 |
uint32 offset) |
uint32 offset) |
179 |
return True; |
return True; |
180 |
} |
} |
181 |
|
|
182 |
void |
static void |
183 |
rdpdr_send_connect(void) |
rdpdr_send_connect(void) |
184 |
{ |
{ |
185 |
uint8 magic[4] = "rDCC"; |
uint8 magic[4] = "rDCC"; |
195 |
} |
} |
196 |
|
|
197 |
|
|
198 |
void |
static void |
199 |
rdpdr_send_name(void) |
rdpdr_send_name(void) |
200 |
{ |
{ |
201 |
uint8 magic[4] = "rDNC"; |
uint8 magic[4] = "rDNC"; |
202 |
STREAM s; |
STREAM s; |
203 |
uint32 hostlen; |
uint32 hostlen; |
204 |
|
|
205 |
if (NULL == g_rdpdr_clientname) { |
if (NULL == g_rdpdr_clientname) |
206 |
g_rdpdr_clientname = hostname; |
{ |
207 |
|
g_rdpdr_clientname = hostname; |
208 |
} |
} |
209 |
hostlen = (strlen(g_rdpdr_clientname) + 1) * 2; |
hostlen = (strlen(g_rdpdr_clientname) + 1) * 2; |
210 |
|
|
220 |
} |
} |
221 |
|
|
222 |
/* Returns the size of the payload of the announce packet */ |
/* Returns the size of the payload of the announce packet */ |
223 |
int |
static int |
224 |
announcedata_size() |
announcedata_size() |
225 |
{ |
{ |
226 |
int size, i; |
int size, i; |
247 |
return size; |
return size; |
248 |
} |
} |
249 |
|
|
250 |
void |
static void |
251 |
rdpdr_send_available(void) |
rdpdr_send_available(void) |
252 |
{ |
{ |
253 |
|
|
305 |
channel_send(s, rdpdr_channel); |
channel_send(s, rdpdr_channel); |
306 |
} |
} |
307 |
|
|
308 |
void |
static void |
309 |
rdpdr_send_completion(uint32 device, uint32 id, uint32 status, uint32 result, uint8 * buffer, |
rdpdr_send_completion(uint32 device, uint32 id, uint32 status, uint32 result, uint8 * buffer, |
310 |
uint32 length) |
uint32 length) |
311 |
{ |
{ |
658 |
result = buffer_len = out.p - out.data; |
result = buffer_len = out.p - out.data; |
659 |
break; |
break; |
660 |
|
|
661 |
|
|
662 |
|
case IRP_MJ_LOCK_CONTROL: |
663 |
|
|
664 |
|
if (g_rdpdr_device[device].device_type != DEVICE_TYPE_DISK) |
665 |
|
{ |
666 |
|
status = STATUS_INVALID_HANDLE; |
667 |
|
break; |
668 |
|
} |
669 |
|
|
670 |
|
in_uint32_le(s, info_level); |
671 |
|
|
672 |
|
out.data = out.p = buffer; |
673 |
|
out.size = sizeof(buffer); |
674 |
|
/* FIXME: Perhaps consider actually *do* |
675 |
|
something here :-) */ |
676 |
|
status = STATUS_SUCCESS; |
677 |
|
result = buffer_len = out.p - out.data; |
678 |
|
break; |
679 |
|
|
680 |
default: |
default: |
681 |
unimpl("IRP major=0x%x minor=0x%x\n", major, minor); |
unimpl("IRP major=0x%x minor=0x%x\n", major, minor); |
682 |
break; |
break; |
691 |
buffer = NULL; |
buffer = NULL; |
692 |
} |
} |
693 |
|
|
694 |
void |
static void |
695 |
rdpdr_send_clientcapabilty(void) |
rdpdr_send_clientcapabilty(void) |
696 |
{ |
{ |
697 |
uint8 magic[4] = "rDPC"; |
uint8 magic[4] = "rDPC"; |
809 |
{ |
{ |
810 |
uint32 select_timeout = 0; // Timeout value to be used for select() (in millisecons). |
uint32 select_timeout = 0; // Timeout value to be used for select() (in millisecons). |
811 |
struct async_iorequest *iorq; |
struct async_iorequest *iorq; |
812 |
|
char c; |
813 |
|
|
814 |
iorq = g_iorequest; |
iorq = g_iorequest; |
815 |
while (iorq != NULL) |
while (iorq != NULL) |
816 |
{ |
{ |
817 |
if (iorq->fd != 0) |
/* We need to test that the fd is still valid */ |
818 |
|
if ((iorq->fd != 0) && (read(iorq->fd, &c, 0) == 0)) |
819 |
{ |
{ |
820 |
switch (iorq->major) |
switch (iorq->major) |
821 |
{ |
{ |