/[rdesktop]/sourceforge.net/trunk/rdesktop/rdpdr.c
This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!
ViewVC logotype

Diff of /sourceforge.net/trunk/rdesktop/rdpdr.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 576 by stargo, Thu Jan 22 20:31:59 2004 UTC revision 1365 by jsorg71, Thu Jan 4 05:39:39 2007 UTC
# Line 1  Line 1 
1    /* -*- c-basic-offset: 8 -*-
2       rdesktop: A Remote Desktop Protocol client.
3       Copyright (C) Matthew Chapman 1999-2007
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>
39    #include <dirent.h>             /* opendir, closedir, readdir */
40  #include <time.h>  #include <time.h>
41    #include <errno.h>
42  #include "rdesktop.h"  #include "rdesktop.h"
43    
 #define IRP_MJ_CREATE           0x00  
 #define IRP_MJ_CLOSE            0x02  
 #define IRP_MJ_READ             0x03  
 #define IRP_MJ_WRITE            0x04  
 #define IRP_MJ_DEVICE_CONTROL   0x0e  
   
44  #define IRP_MJ_CREATE                   0x00  #define IRP_MJ_CREATE                   0x00
45  #define IRP_MJ_CLOSE                    0x02  #define IRP_MJ_CLOSE                    0x02
46  #define IRP_MJ_READ                     0x03  #define IRP_MJ_READ                     0x03
# Line 18  Line 50 
50  #define IRP_MJ_QUERY_VOLUME_INFORMATION 0x0a  #define IRP_MJ_QUERY_VOLUME_INFORMATION 0x0a
51  #define IRP_MJ_DIRECTORY_CONTROL        0x0c  #define IRP_MJ_DIRECTORY_CONTROL        0x0c
52  #define IRP_MJ_DEVICE_CONTROL           0x0e  #define IRP_MJ_DEVICE_CONTROL           0x0e
53    #define IRP_MJ_LOCK_CONTROL             0x11
54    
55  #define IRP_MN_QUERY_DIRECTORY          0x01  #define IRP_MN_QUERY_DIRECTORY          0x01
56  #define IRP_MN_NOTIFY_CHANGE_DIRECTORY  0x02  #define IRP_MN_NOTIFY_CHANGE_DIRECTORY  0x02
57    
58  #define MAX_ASYNC_IO_REQUESTS   10  extern char g_hostname[16];
   
 extern char hostname[16];  
59  extern DEVICE_FNS serial_fns;  extern DEVICE_FNS serial_fns;
60  extern DEVICE_FNS printer_fns;  extern DEVICE_FNS printer_fns;
61  extern DEVICE_FNS parallel_fns;  extern DEVICE_FNS parallel_fns;
62  extern DEVICE_FNS disk_fns;  extern DEVICE_FNS disk_fns;
63    #ifdef WITH_SCARD
64    extern DEVICE_FNS scard_fns;
65    #endif
66    extern FILEINFO g_fileinfo[];
67    extern BOOL g_notify_stamp;
68    
69  static VCHANNEL *rdpdr_channel;  static VCHANNEL *rdpdr_channel;
70    
71  /* 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 */
72  HANDLE g_min_timeout_fd;  RD_NTHANDLE g_min_timeout_fd;
73  uint32 g_num_devices;  uint32 g_num_devices;
74    
75  /* Table with information about rdpdr devices */  /* Table with information about rdpdr devices */
76  RDPDR_DEVICE g_rdpdr_device[RDPDR_MAX_DEVICES];  RDPDR_DEVICE g_rdpdr_device[RDPDR_MAX_DEVICES];
77    char *g_rdpdr_clientname = NULL;
78    
79  /* 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 */
80    /* using a linked list ensures that they are processed in the right order, */
81    /* if multiple ios are being done on the same fd */
82  struct async_iorequest  struct async_iorequest
83  {  {
84          uint32 fd, major, minor, offset, device, id, length;          uint32 fd, major, minor, offset, device, id, length, partial_len;
85          long timeout,           /* Total timeout */          long timeout,           /* Total timeout */
86            itv_timeout;          /* Interval timeout (between serial characters) */            itv_timeout;          /* Interval timeout (between serial characters) */
87          uint8 *buffer;          uint8 *buffer;
88          DEVICE_FNS *fns;          DEVICE_FNS *fns;
89  } g_iorequest[MAX_ASYNC_IO_REQUESTS];  
90            struct async_iorequest *next;   /* next element in list */
91    };
92    
93    struct async_iorequest *g_iorequest;
94    
95  /* Return device_id for a given handle */  /* Return device_id for a given handle */
96  int  int
97  get_device_index(HANDLE handle)  get_device_index(RD_NTHANDLE handle)
98  {  {
99          int i;          int i;
100          for (i = 0; i < RDPDR_MAX_DEVICES; i++)          for (i = 0; i < RDPDR_MAX_DEVICES; i++)
# Line 75  convert_to_unix_filename(char *filename) Line 117  convert_to_unix_filename(char *filename)
117          }          }
118  }  }
119    
120    static BOOL
121    rdpdr_handle_ok(int device, int handle)
122    {
123            switch (g_rdpdr_device[device].device_type)
124            {
125                    case DEVICE_TYPE_PARALLEL:
126                    case DEVICE_TYPE_SERIAL:
127                    case DEVICE_TYPE_PRINTER:
128                    case DEVICE_TYPE_SCARD:
129                            if (g_rdpdr_device[device].handle != handle)
130                                    return False;
131                            break;
132                    case DEVICE_TYPE_DISK:
133                            if (g_fileinfo[handle].device_id != device)
134                                    return False;
135                            break;
136            }
137            return True;
138    }
139    
140  /* 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 */
141  BOOL  static BOOL
142  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,
143                      DEVICE_FNS * fns, long total_timeout, long interval_timeout, uint8 * buffer)                      DEVICE_FNS * fns, uint32 total_timeout, uint32 interval_timeout, uint8 * buffer,
144                        uint32 offset)
145  {  {
         int i;  
146          struct async_iorequest *iorq;          struct async_iorequest *iorq;
147    
148          for (i = 0; i < MAX_ASYNC_IO_REQUESTS; i++)          if (g_iorequest == NULL)
149          {          {
150                  iorq = &g_iorequest[i];                  g_iorequest = (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));
151                    if (!g_iorequest)
152                            return False;
153                    g_iorequest->fd = 0;
154                    g_iorequest->next = NULL;
155            }
156    
157            iorq = g_iorequest;
158    
159                  if (iorq->fd == 0)          while (iorq->fd != 0)
160            {
161                    /* create new element if needed */
162                    if (iorq->next == NULL)
163                  {                  {
164                          iorq->device = device;                          iorq->next =
165                          iorq->fd = file;                                  (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));
166                          iorq->id = id;                          if (!iorq->next)
167                          iorq->major = major;                                  return False;
168                          iorq->length = length;                          iorq->next->fd = 0;
169                          iorq->fns = fns;                          iorq->next->next = NULL;
                         iorq->timeout = total_timeout;  
                         iorq->itv_timeout = interval_timeout;  
                         iorq->buffer = buffer;  
                         return True;  
170                  }                  }
171                    iorq = iorq->next;
172          }          }
173          error("IO request table full. Increase MAX_ASYNC_IO_REQUESTS in rdpdr.c!\n");          iorq->device = device;
174          return False;          iorq->fd = file;
175            iorq->id = id;
176            iorq->major = major;
177            iorq->length = length;
178            iorq->partial_len = 0;
179            iorq->fns = fns;
180            iorq->timeout = total_timeout;
181            iorq->itv_timeout = interval_timeout;
182            iorq->buffer = buffer;
183            iorq->offset = offset;
184            return True;
185  }  }
186    
187    static void
 void  
188  rdpdr_send_connect(void)  rdpdr_send_connect(void)
189  {  {
190          uint8 magic[4] = "rDCC";          uint8 magic[4] = "rDCC";
# Line 122  rdpdr_send_connect(void) Line 200  rdpdr_send_connect(void)
200  }  }
201    
202    
203  void  static void
204  rdpdr_send_name(void)  rdpdr_send_name(void)
205  {  {
206          uint8 magic[4] = "rDNC";          uint8 magic[4] = "rDNC";
         uint32 hostlen = (strlen(hostname) + 1) * 2;  
207          STREAM s;          STREAM s;
208            uint32 hostlen;
209    
210            if (NULL == g_rdpdr_clientname)
211            {
212                    g_rdpdr_clientname = g_hostname;
213            }
214            hostlen = (strlen(g_rdpdr_clientname) + 1) * 2;
215    
216          s = channel_init(rdpdr_channel, 16 + hostlen);          s = channel_init(rdpdr_channel, 16 + hostlen);
217          out_uint8a(s, magic, 4);          out_uint8a(s, magic, 4);
# Line 135  rdpdr_send_name(void) Line 219  rdpdr_send_name(void)
219          out_uint16_le(s, 0x72);          out_uint16_le(s, 0x72);
220          out_uint32(s, 0);          out_uint32(s, 0);
221          out_uint32_le(s, hostlen);          out_uint32_le(s, hostlen);
222          rdp_out_unistr(s, hostname, hostlen - 2);          rdp_out_unistr(s, g_rdpdr_clientname, hostlen - 2);
223          s_mark_end(s);          s_mark_end(s);
224          channel_send(s, rdpdr_channel);          channel_send(s, rdpdr_channel);
225  }  }
226    
227  /* Returns the size of the payload of the announce packet */  /* Returns the size of the payload of the announce packet */
228  int  static int
229  announcedata_size()  announcedata_size()
230  {  {
231          int size, i;          int size, i;
232          PRINTER *printerinfo;          PRINTER *printerinfo;
233    
234          size = 8;               //static announce size          size = 8;               /* static announce size */
235          size += g_num_devices * 0x14;          size += g_num_devices * 0x14;
236    
237          for (i = 0; i < g_num_devices; i++)          for (i = 0; i < g_num_devices; i++)
# Line 168  announcedata_size() Line 252  announcedata_size()
252          return size;          return size;
253  }  }
254    
255  void  static void
256  rdpdr_send_available(void)  rdpdr_send_available(void)
257  {  {
258    
# Line 186  rdpdr_send_available(void) Line 270  rdpdr_send_available(void)
270          {          {
271                  out_uint32_le(s, g_rdpdr_device[i].device_type);                  out_uint32_le(s, g_rdpdr_device[i].device_type);
272                  out_uint32_le(s, i);    /* RDP Device ID */                  out_uint32_le(s, i);    /* RDP Device ID */
273                    /* Is it possible to use share names longer than 8 chars?
274                       /astrand */
275                  out_uint8p(s, g_rdpdr_device[i].name, 8);                  out_uint8p(s, g_rdpdr_device[i].name, 8);
276    
277                  if (g_rdpdr_device[i].device_type == DEVICE_TYPE_PRINTER)                  switch (g_rdpdr_device[i].device_type)
278                  {                  {
279                          printerinfo = (PRINTER *) g_rdpdr_device[i].pdevice_data;                          case DEVICE_TYPE_PRINTER:
280                                    printerinfo = (PRINTER *) g_rdpdr_device[i].pdevice_data;
281    
282                          driverlen = 2 * strlen(printerinfo->driver) + 2;                                  driverlen = 2 * strlen(printerinfo->driver) + 2;
283                          printerlen = 2 * strlen(printerinfo->printer) + 2;                                  printerlen = 2 * strlen(printerinfo->printer) + 2;
284                          bloblen = printerinfo->bloblen;                                  bloblen = printerinfo->bloblen;
285    
286                          out_uint32_le(s, 24 + driverlen + printerlen + bloblen);        /* length of extra info */                                  out_uint32_le(s, 24 + driverlen + printerlen + bloblen);        /* length of extra info */
287                          out_uint32_le(s, printerinfo->default_printer ? 2 : 0);                                  out_uint32_le(s, printerinfo->default_printer ? 2 : 0);
288                          out_uint8s(s, 8);       /* unknown */                                  out_uint8s(s, 8);       /* unknown */
289                          out_uint32_le(s, driverlen);                                  out_uint32_le(s, driverlen);
290                          out_uint32_le(s, printerlen);                                  out_uint32_le(s, printerlen);
291                          out_uint32_le(s, bloblen);                                  out_uint32_le(s, bloblen);
292                          rdp_out_unistr(s, printerinfo->driver, driverlen - 2);                                  rdp_out_unistr(s, printerinfo->driver, driverlen - 2);
293                          rdp_out_unistr(s, printerinfo->printer, printerlen - 2);                                  rdp_out_unistr(s, printerinfo->printer, printerlen - 2);
294                          out_uint8a(s, printerinfo->blob, bloblen);                                  out_uint8a(s, printerinfo->blob, bloblen);
295    
296                          xfree(printerinfo->blob);       /* Blob is sent twice if reconnecting */                                  if (printerinfo->blob)
297                  }                                          xfree(printerinfo->blob);       /* Blob is sent twice if reconnecting */
298                  else                                  break;
299                  {                          default:
300                          out_uint32(s, 0);                                  out_uint32(s, 0);
301                  }                  }
302          }          }
303  #if 0  #if 0
# Line 232  rdpdr_send_completion(uint32 device, uin Line 319  rdpdr_send_completion(uint32 device, uin
319          uint8 magic[4] = "rDCI";          uint8 magic[4] = "rDCI";
320          STREAM s;          STREAM s;
321    
322    #ifdef WITH_SCARD
323            scard_lock(SCARD_LOCK_RDPDR);
324    #endif
325          s = channel_init(rdpdr_channel, 20 + length);          s = channel_init(rdpdr_channel, 20 + length);
326          out_uint8a(s, magic, 4);          out_uint8a(s, magic, 4);
327          out_uint32_le(s, device);          out_uint32_le(s, device);
# Line 240  rdpdr_send_completion(uint32 device, uin Line 330  rdpdr_send_completion(uint32 device, uin
330          out_uint32_le(s, result);          out_uint32_le(s, result);
331          out_uint8p(s, buffer, length);          out_uint8p(s, buffer, length);
332          s_mark_end(s);          s_mark_end(s);
333          /* JIF          /* JIF */
334             hexdump(s->channel_hdr + 8, s->end - s->channel_hdr - 8); */  #ifdef WITH_DEBUG_RDP5
335            printf("--> rdpdr_send_completion\n");
336            /* hexdump(s->channel_hdr + 8, s->end - s->channel_hdr - 8); */
337    #endif
338          channel_send(s, rdpdr_channel);          channel_send(s, rdpdr_channel);
339    #ifdef WITH_SCARD
340            scard_unlock(SCARD_LOCK_RDPDR);
341    #endif
342  }  }
343    
344  static void  static void
# Line 265  rdpdr_process_irp(STREAM s) Line 361  rdpdr_process_irp(STREAM s)
361                  error_mode,                  error_mode,
362                  share_mode, disposition, total_timeout, interval_timeout, flags_and_attributes = 0;                  share_mode, disposition, total_timeout, interval_timeout, flags_and_attributes = 0;
363    
364          char filename[256];          char filename[PATH_MAX];
365          uint8 *buffer, *pst_buf;          uint8 *buffer, *pst_buf;
366          struct stream out;          struct stream out;
367          DEVICE_FNS *fns;          DEVICE_FNS *fns;
368          BOOL rw_blocking = True;          BOOL rw_blocking = True;
369          NTSTATUS status = STATUS_INVALID_DEVICE_REQUEST;          RD_NTSTATUS status = RD_STATUS_INVALID_DEVICE_REQUEST;
370    
371          in_uint32_le(s, device);          in_uint32_le(s, device);
372          in_uint32_le(s, file);          in_uint32_le(s, file);
# Line 282  rdpdr_process_irp(STREAM s) Line 378  rdpdr_process_irp(STREAM s)
378          buffer = (uint8 *) xmalloc(1024);          buffer = (uint8 *) xmalloc(1024);
379          buffer[0] = 0;          buffer[0] = 0;
380    
   
381          switch (g_rdpdr_device[device].device_type)          switch (g_rdpdr_device[device].device_type)
382          {          {
383                  case DEVICE_TYPE_SERIAL:                  case DEVICE_TYPE_SERIAL:
# Line 305  rdpdr_process_irp(STREAM s) Line 400  rdpdr_process_irp(STREAM s)
400                  case DEVICE_TYPE_DISK:                  case DEVICE_TYPE_DISK:
401    
402                          fns = &disk_fns;                          fns = &disk_fns;
403                            rw_blocking = False;
404                          break;                          break;
405    
406                  case DEVICE_TYPE_SCARD:                  case DEVICE_TYPE_SCARD:
407    #ifdef WITH_SCARD
408                            fns = &scard_fns;
409                            rw_blocking = False;
410                            break;
411    #endif
412                  default:                  default:
413    
414                          error("IRP for bad device %ld\n", device);                          error("IRP for bad device %ld\n", device);
# Line 319  rdpdr_process_irp(STREAM s) Line 420  rdpdr_process_irp(STREAM s)
420                  case IRP_MJ_CREATE:                  case IRP_MJ_CREATE:
421    
422                          in_uint32_be(s, desired_access);                          in_uint32_be(s, desired_access);
423                          in_uint8s(s, 0x08);     // unknown                          in_uint8s(s, 0x08);     /* unknown */
424                          in_uint32_le(s, error_mode);                          in_uint32_le(s, error_mode);
425                          in_uint32_le(s, share_mode);                          in_uint32_le(s, share_mode);
426                          in_uint32_le(s, disposition);                          in_uint32_le(s, disposition);
# Line 338  rdpdr_process_irp(STREAM s) Line 439  rdpdr_process_irp(STREAM s)
439    
440                          if (!fns->create)                          if (!fns->create)
441                          {                          {
442                                  status = STATUS_NOT_SUPPORTED;                                  status = RD_STATUS_NOT_SUPPORTED;
443                                  break;                                  break;
444                          }                          }
445    
# Line 350  rdpdr_process_irp(STREAM s) Line 451  rdpdr_process_irp(STREAM s)
451                  case IRP_MJ_CLOSE:                  case IRP_MJ_CLOSE:
452                          if (!fns->close)                          if (!fns->close)
453                          {                          {
454                                  status = STATUS_NOT_SUPPORTED;                                  status = RD_STATUS_NOT_SUPPORTED;
455                                  break;                                  break;
456                          }                          }
457    
# Line 361  rdpdr_process_irp(STREAM s) Line 462  rdpdr_process_irp(STREAM s)
462    
463                          if (!fns->read)                          if (!fns->read)
464                          {                          {
465                                  status = STATUS_NOT_SUPPORTED;                                  status = RD_STATUS_NOT_SUPPORTED;
466                                  break;                                  break;
467                          }                          }
468    
# Line 370  rdpdr_process_irp(STREAM s) Line 471  rdpdr_process_irp(STREAM s)
471  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
472                          DEBUG(("RDPDR IRP Read (length: %d, offset: %d)\n", length, offset));                          DEBUG(("RDPDR IRP Read (length: %d, offset: %d)\n", length, offset));
473  #endif  #endif
474                          if (rw_blocking)        // Complete read immediately                          if (!rdpdr_handle_ok(device, file))
475                            {
476                                    status = RD_STATUS_INVALID_HANDLE;
477                                    break;
478                            }
479    
480                            if (rw_blocking)        /* Complete read immediately */
481                          {                          {
482                                  buffer = (uint8 *) xrealloc((void *) buffer, length);                                  buffer = (uint8 *) xrealloc((void *) buffer, length);
483                                    if (!buffer)
484                                    {
485                                            status = RD_STATUS_CANCELLED;
486                                            break;
487                                    }
488                                  status = fns->read(file, buffer, length, offset, &result);                                  status = fns->read(file, buffer, length, offset, &result);
489                                  buffer_len = result;                                  buffer_len = result;
490                                  break;                                  break;
491                          }                          }
492    
493                          // Add request to table                          /* Add request to table */
494                          pst_buf = (uint8 *) xmalloc(length);                          pst_buf = (uint8 *) xmalloc(length);
495                            if (!pst_buf)
496                            {
497                                    status = RD_STATUS_CANCELLED;
498                                    break;
499                            }
500                          serial_get_timeout(file, length, &total_timeout, &interval_timeout);                          serial_get_timeout(file, length, &total_timeout, &interval_timeout);
501                          if (add_async_iorequest                          if (add_async_iorequest
502                              (device, file, id, major, length, fns, total_timeout, interval_timeout,                              (device, file, id, major, length, fns, total_timeout, interval_timeout,
503                               pst_buf))                               pst_buf, offset))
504                          {                          {
505                                  status = STATUS_PENDING;                                  status = RD_STATUS_PENDING;
506                                  break;                                  break;
507                          }                          }
508    
509                          status = STATUS_CANCELLED;                          status = RD_STATUS_CANCELLED;
510                          break;                          break;
   
511                  case IRP_MJ_WRITE:                  case IRP_MJ_WRITE:
512    
513                          buffer_len = 1;                          buffer_len = 1;
514    
515                          if (!fns->write)                          if (!fns->write)
516                          {                          {
517                                  status = STATUS_NOT_SUPPORTED;                                  status = RD_STATUS_NOT_SUPPORTED;
518                                  break;                                  break;
519                          }                          }
520    
# Line 408  rdpdr_process_irp(STREAM s) Line 524  rdpdr_process_irp(STREAM s)
524  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
525                          DEBUG(("RDPDR IRP Write (length: %d)\n", result));                          DEBUG(("RDPDR IRP Write (length: %d)\n", result));
526  #endif  #endif
527                          if (rw_blocking)        // Complete immediately                          if (!rdpdr_handle_ok(device, file))
528                            {
529                                    status = RD_STATUS_INVALID_HANDLE;
530                                    break;
531                            }
532    
533                            if (rw_blocking)        /* Complete immediately */
534                          {                          {
535                                  status = fns->write(file, s->p, length, offset, &result);                                  status = fns->write(file, s->p, length, offset, &result);
536                                  break;                                  break;
537                          }                          }
538    
539                          // Add to table                          /* Add to table */
540                          pst_buf = (uint8 *) xmalloc(length);                          pst_buf = (uint8 *) xmalloc(length);
541                            if (!pst_buf)
542                            {
543                                    status = RD_STATUS_CANCELLED;
544                                    break;
545                            }
546    
547                          in_uint8a(s, pst_buf, length);                          in_uint8a(s, pst_buf, length);
548    
549                          if (add_async_iorequest                          if (add_async_iorequest
550                              (device, file, id, major, length, fns, 0, 0, pst_buf))                              (device, file, id, major, length, fns, 0, 0, pst_buf, offset))
551                          {                          {
552                                  status = STATUS_PENDING;                                  status = RD_STATUS_PENDING;
553                                  break;                                  break;
554                          }                          }
555    
556                          status = STATUS_CANCELLED;                          status = RD_STATUS_CANCELLED;
557                          break;                          break;
558    
559                  case IRP_MJ_QUERY_INFORMATION:                  case IRP_MJ_QUERY_INFORMATION:
560    
561                          if (g_rdpdr_device[device].device_type != DEVICE_TYPE_DISK)                          if (g_rdpdr_device[device].device_type != DEVICE_TYPE_DISK)
562                          {                          {
563                                  status = STATUS_INVALID_HANDLE;                                  status = RD_STATUS_INVALID_HANDLE;
564                                  break;                                  break;
565                          }                          }
566                          in_uint32_le(s, info_level);                          in_uint32_le(s, info_level);
# Line 448  rdpdr_process_irp(STREAM s) Line 576  rdpdr_process_irp(STREAM s)
576    
577                          if (g_rdpdr_device[device].device_type != DEVICE_TYPE_DISK)                          if (g_rdpdr_device[device].device_type != DEVICE_TYPE_DISK)
578                          {                          {
579                                  status = STATUS_INVALID_HANDLE;                                  status = RD_STATUS_INVALID_HANDLE;
580                                  break;                                  break;
581                          }                          }
582    
# Line 464  rdpdr_process_irp(STREAM s) Line 592  rdpdr_process_irp(STREAM s)
592    
593                          if (g_rdpdr_device[device].device_type != DEVICE_TYPE_DISK)                          if (g_rdpdr_device[device].device_type != DEVICE_TYPE_DISK)
594                          {                          {
595                                  status = STATUS_INVALID_HANDLE;                                  status = RD_STATUS_INVALID_HANDLE;
596                                  break;                                  break;
597                          }                          }
598    
# Line 480  rdpdr_process_irp(STREAM s) Line 608  rdpdr_process_irp(STREAM s)
608    
609                          if (g_rdpdr_device[device].device_type != DEVICE_TYPE_DISK)                          if (g_rdpdr_device[device].device_type != DEVICE_TYPE_DISK)
610                          {                          {
611                                  status = STATUS_INVALID_HANDLE;                                  status = RD_STATUS_INVALID_HANDLE;
612                                  break;                                  break;
613                          }                          }
614    
# Line 513  rdpdr_process_irp(STREAM s) Line 641  rdpdr_process_irp(STREAM s)
641                                  case IRP_MN_NOTIFY_CHANGE_DIRECTORY:                                  case IRP_MN_NOTIFY_CHANGE_DIRECTORY:
642    
643                                          /* JIF                                          /* JIF
644                                             unimpl("IRP major=0x%x minor=0x%x: IRP_MN_NOTIFY_CHANGE_DIRECTORY\n", major, minor); */                                             unimpl("IRP major=0x%x minor=0x%x: IRP_MN_NOTIFY_CHANGE_DIRECTORY\n", major, minor);  */
645                                          status = STATUS_PENDING;        // Don't send completion packet  
646                                            in_uint32_le(s, info_level);    /* notify mask */
647    
648                                            g_notify_stamp = True;
649    
650                                            status = disk_create_notify(file, info_level);
651                                            result = 0;
652    
653                                            if (status == RD_STATUS_PENDING)
654                                                    add_async_iorequest(device, file, id, major, length,
655                                                                        fns, 0, 0, NULL, 0);
656                                          break;                                          break;
657    
658                                  default:                                  default:
659    
660                                          status = STATUS_INVALID_PARAMETER;                                          status = RD_STATUS_INVALID_PARAMETER;
661                                          /* JIF                                          /* JIF */
662                                             unimpl("IRP major=0x%x minor=0x%x\n", major, minor); */                                          unimpl("IRP major=0x%x minor=0x%x\n", major, minor);
663                          }                          }
664                          break;                          break;
665    
# Line 529  rdpdr_process_irp(STREAM s) Line 667  rdpdr_process_irp(STREAM s)
667    
668                          if (!fns->device_control)                          if (!fns->device_control)
669                          {                          {
670                                  status = STATUS_NOT_SUPPORTED;                                  status = RD_STATUS_NOT_SUPPORTED;
671                                  break;                                  break;
672                          }                          }
673    
# Line 539  rdpdr_process_irp(STREAM s) Line 677  rdpdr_process_irp(STREAM s)
677                          in_uint8s(s, 0x14);                          in_uint8s(s, 0x14);
678    
679                          buffer = (uint8 *) xrealloc((void *) buffer, bytes_out + 0x14);                          buffer = (uint8 *) xrealloc((void *) buffer, bytes_out + 0x14);
680                            if (!buffer)
681                            {
682                                    status = RD_STATUS_CANCELLED;
683                                    break;
684                            }
685    
686                          out.data = out.p = buffer;                          out.data = out.p = buffer;
687                          out.size = sizeof(buffer);                          out.size = sizeof(buffer);
688    
689                            DEBUG_SCARD(("[SMART-CARD TRACE]\n"));
690                            DEBUG_SCARD(("device 0x%.8x\n", device));
691                            DEBUG_SCARD(("file   0x%.8x\n", file));
692                            DEBUG_SCARD(("id     0x%.8x\n", id));
693    #ifdef WITH_SCARD
694                            scardSetInfo(device, id, bytes_out + 0x14);
695    #endif
696                          status = fns->device_control(file, request, s, &out);                          status = fns->device_control(file, request, s, &out);
697                          result = buffer_len = out.p - out.data;                          result = buffer_len = out.p - out.data;
698    
699                            DEBUG_SCARD(("[SMART-CARD TRACE] OUT 0x%.8x\n", status));
700    
701                            /* Serial SERIAL_WAIT_ON_MASK */
702                            if (status == RD_STATUS_PENDING)
703                            {
704                                    if (add_async_iorequest
705                                        (device, file, id, major, length, fns, 0, 0, NULL, 0))
706                                    {
707                                            status = RD_STATUS_PENDING;
708                                            break;
709                                    }
710                            }
711    #ifdef WITH_SCARD
712                            else if (status == (RD_STATUS_PENDING | 0xC0000000))
713                                    status = RD_STATUS_PENDING;
714    #endif
715                            break;
716    
717    
718                    case IRP_MJ_LOCK_CONTROL:
719    
720                            if (g_rdpdr_device[device].device_type != DEVICE_TYPE_DISK)
721                            {
722                                    status = RD_STATUS_INVALID_HANDLE;
723                                    break;
724                            }
725    
726                            in_uint32_le(s, info_level);
727    
728                            out.data = out.p = buffer;
729                            out.size = sizeof(buffer);
730                            /* FIXME: Perhaps consider actually *do*
731                               something here :-) */
732                            status = RD_STATUS_SUCCESS;
733                            result = buffer_len = out.p - out.data;
734                          break;                          break;
735    
736                  default:                  default:
# Line 550  rdpdr_process_irp(STREAM s) Line 738  rdpdr_process_irp(STREAM s)
738                          break;                          break;
739          }          }
740    
741          if (status != STATUS_PENDING)          if (status != RD_STATUS_PENDING)
742          {          {
743                  rdpdr_send_completion(device, id, status, result, buffer, buffer_len);                  rdpdr_send_completion(device, id, status, result, buffer, buffer_len);
744          }          }
745          xfree(buffer);          if (buffer)
746                    xfree(buffer);
747            buffer = NULL;
748  }  }
749    
750  void  static void
751  rdpdr_send_clientcapabilty(void)  rdpdr_send_clientcapabilty(void)
752  {  {
753          uint8 magic[4] = "rDPC";          uint8 magic[4] = "rDPC";
# Line 661  rdpdr_init() Line 850  rdpdr_init()
850  {  {
851          if (g_num_devices > 0)          if (g_num_devices > 0)
852          {          {
853                  rdpdr_channel = channel_register("rdpdr", CHANNEL_OPTION_INITIALIZED | CHANNEL_OPTION_COMPRESS_RDP, rdpdr_process);                  rdpdr_channel =
854                            channel_register("rdpdr",
855                                             CHANNEL_OPTION_INITIALIZED | CHANNEL_OPTION_COMPRESS_RDP,
856                                             rdpdr_process);
857          }          }
858    
859          return (rdpdr_channel != NULL);          return (rdpdr_channel != NULL);
# Line 671  rdpdr_init() Line 863  rdpdr_init()
863  void  void
864  rdpdr_add_fds(int *n, fd_set * rfds, fd_set * wfds, struct timeval *tv, BOOL * timeout)  rdpdr_add_fds(int *n, fd_set * rfds, fd_set * wfds, struct timeval *tv, BOOL * timeout)
865  {  {
866          int i;          uint32 select_timeout = 0;      /* Timeout value to be used for select() (in millisecons). */
         long select_timeout = 0;        // Timeout value to be used for select() (in millisecons).  
867          struct async_iorequest *iorq;          struct async_iorequest *iorq;
868            char c;
869    
870          for (i = 0; i < MAX_ASYNC_IO_REQUESTS; i++)          iorq = g_iorequest;
871            while (iorq != NULL)
872          {          {
873                  iorq = &g_iorequest[i];                  if (iorq->fd != 0)
   
                 if (iorq->fd != 0)      // Found a pending io request  
874                  {                  {
875                          switch (iorq->major)                          switch (iorq->major)
876                          {                          {
877                                  case IRP_MJ_READ:                                  case IRP_MJ_READ:
878                                            /* Is this FD valid? FDs will
879                                               be invalid when
880                                               reconnecting. FIXME: Real
881                                               support for reconnects. */
882    
883                                          FD_SET(iorq->fd, rfds);                                          FD_SET(iorq->fd, rfds);
884                                            *n = MAX(*n, iorq->fd);
885    
886                                          // Check if io request timeout is smaller than current (but not 0).                                          /* Check if io request timeout is smaller than current (but not 0). */
887                                          if (iorq->timeout                                          if (iorq->timeout
888                                              && (select_timeout == 0                                              && (select_timeout == 0
889                                                  || iorq->timeout < select_timeout))                                                  || iorq->timeout < select_timeout))
890                                          {                                          {
891                                                  // Set new timeout                                                  /* Set new timeout */
892                                                  select_timeout = iorq->timeout;                                                  select_timeout = iorq->timeout;
893                                                  g_min_timeout_fd = iorq->fd;    /* Remember fd */                                                  g_min_timeout_fd = iorq->fd;    /* Remember fd */
894                                                  tv->tv_sec = select_timeout / 1000;                                                  tv->tv_sec = select_timeout / 1000;
895                                                  tv->tv_usec = (select_timeout % 1000) * 1000;                                                  tv->tv_usec = (select_timeout % 1000) * 1000;
896                                                  *timeout = True;                                                  *timeout = True;
897                                                    break;
898                                            }
899                                            if (iorq->itv_timeout && iorq->partial_len > 0
900                                                && (select_timeout == 0
901                                                    || iorq->itv_timeout < select_timeout))
902                                            {
903                                                    /* Set new timeout */
904                                                    select_timeout = iorq->itv_timeout;
905                                                    g_min_timeout_fd = iorq->fd;    /* Remember fd */
906                                                    tv->tv_sec = select_timeout / 1000;
907                                                    tv->tv_usec = (select_timeout % 1000) * 1000;
908                                                    *timeout = True;
909                                                    break;
910                                          }                                          }
911                                          break;                                          break;
912    
913                                  case IRP_MJ_WRITE:                                  case IRP_MJ_WRITE:
914                                            /* FD still valid? See above. */
915                                            if ((write(iorq->fd, &c, 0) != 0) && (errno == EBADF))
916                                                    break;
917    
918                                          FD_SET(iorq->fd, wfds);                                          FD_SET(iorq->fd, wfds);
919                                            *n = MAX(*n, iorq->fd);
920                                            break;
921    
922                                    case IRP_MJ_DEVICE_CONTROL:
923                                            if (select_timeout > 5)
924                                                    select_timeout = 5;     /* serial event queue */
925                                          break;                                          break;
926    
927                          }                          }
928                          *n = MAX(*n, iorq->fd);  
929                  }                  }
930    
931                    iorq = iorq->next;
932          }          }
933  }  }
934    
935    struct async_iorequest *
936    rdpdr_remove_iorequest(struct async_iorequest *prev, struct async_iorequest *iorq)
937    {
938            if (!iorq)
939                    return NULL;
940    
941            if (iorq->buffer)
942                    xfree(iorq->buffer);
943            if (prev)
944            {
945                    prev->next = iorq->next;
946                    xfree(iorq);
947                    iorq = prev->next;
948            }
949            else
950            {
951                    /* Even if NULL */
952                    g_iorequest = iorq->next;
953                    xfree(iorq);
954                    iorq = NULL;
955            }
956            return iorq;
957    }
958    
959  /* Check if select() returned with one of the rdpdr file descriptors, and complete io if it did */  /* Check if select() returned with one of the rdpdr file descriptors, and complete io if it did */
960  void  static void
961  rdpdr_check_fds(fd_set * rfds, fd_set * wfds, BOOL timed_out)  _rdpdr_check_fds(fd_set * rfds, fd_set * wfds, BOOL timed_out)
962  {  {
963          int i;          RD_NTSTATUS status;
964          NTSTATUS status;          uint32 result = 0;
         uint32 result = 0, buffer_len = 0;  
965          DEVICE_FNS *fns;          DEVICE_FNS *fns;
966          struct async_iorequest *iorq;          struct async_iorequest *iorq;
967            struct async_iorequest *prev;
968            uint32 req_size = 0;
969            uint32 buffer_len;
970            struct stream out;
971            uint8 *buffer = NULL;
972    
973    
974          if (timed_out)          if (timed_out)
975          {          {
976                  rdpdr_abort_io(g_min_timeout_fd, 0, STATUS_TIMEOUT);                  /* check serial iv_timeout */
977    
978                    iorq = g_iorequest;
979                    prev = NULL;
980                    while (iorq != NULL)
981                    {
982                            if (iorq->fd == g_min_timeout_fd)
983                            {
984                                    if ((iorq->partial_len > 0) &&
985                                        (g_rdpdr_device[iorq->device].device_type ==
986                                         DEVICE_TYPE_SERIAL))
987                                    {
988    
989                                            /* iv_timeout between 2 chars, send partial_len */
990                                            /*printf("RDPDR: IVT total %u bytes read of %u\n", iorq->partial_len, iorq->length); */
991                                            rdpdr_send_completion(iorq->device,
992                                                                  iorq->id, RD_STATUS_SUCCESS,
993                                                                  iorq->partial_len,
994                                                                  iorq->buffer, iorq->partial_len);
995                                            iorq = rdpdr_remove_iorequest(prev, iorq);
996                                            return;
997                                    }
998                                    else
999                                    {
1000                                            break;
1001                                    }
1002    
1003                            }
1004                            else
1005                            {
1006                                    break;
1007                            }
1008    
1009    
1010                            prev = iorq;
1011                            if (iorq)
1012                                    iorq = iorq->next;
1013    
1014                    }
1015    
1016                    rdpdr_abort_io(g_min_timeout_fd, 0, RD_STATUS_TIMEOUT);
1017                  return;                  return;
1018          }          }
1019    
1020          // Walk through array of pending io_rq's          iorq = g_iorequest;
1021          for (i = 0; i < MAX_ASYNC_IO_REQUESTS; i++)          prev = NULL;
1022            while (iorq != NULL)
1023          {          {
                 iorq = &g_iorequest[i];  
   
1024                  if (iorq->fd != 0)                  if (iorq->fd != 0)
1025                  {                  {
1026                          switch (iorq->major)                          switch (iorq->major)
1027                          {                          {
   
1028                                  case IRP_MJ_READ:                                  case IRP_MJ_READ:
   
1029                                          if (FD_ISSET(iorq->fd, rfds))                                          if (FD_ISSET(iorq->fd, rfds))
1030                                          {                                          {
1031                                                  // Read, and send data.                                                  /* Read the data */
1032                                                  fns = iorq->fns;                                                  fns = iorq->fns;
                                                 status = fns->read(iorq->fd, iorq->buffer,  
                                                                    iorq->length, 0, &result);  
                                                 buffer_len = result;  
1033    
1034                                                  rdpdr_send_completion(iorq->device, iorq->id,                                                  req_size =
1035                                                                        status, result, iorq->buffer,                                                          (iorq->length - iorq->partial_len) >
1036                                                                        buffer_len);                                                          8192 ? 8192 : (iorq->length -
1037                                                                           iorq->partial_len);
1038                                                    /* never read larger chunks than 8k - chances are that it will block */
1039                                                    status = fns->read(iorq->fd,
1040                                                                       iorq->buffer + iorq->partial_len,
1041                                                                       req_size, iorq->offset, &result);
1042    
1043                                                    if ((long) result > 0)
1044                                                    {
1045                                                            iorq->partial_len += result;
1046                                                            iorq->offset += result;
1047                                                    }
1048  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
1049                                                  DEBUG(("RDPDR: %d bytes of data read\n", result));                                                  DEBUG(("RDPDR: %d bytes of data read\n", result));
1050  #endif  #endif
1051                                                  xfree(iorq->buffer);                                                  /* only delete link if all data has been transfered */
1052                                                  iorq->fd = 0;                                                  /* or if result was 0 and status success - EOF      */
1053                                                    if ((iorq->partial_len == iorq->length) ||
1054                                                        (result == 0))
1055                                                    {
1056    #if WITH_DEBUG_RDP5
1057                                                            DEBUG(("RDPDR: AIO total %u bytes read of %u\n", iorq->partial_len, iorq->length));
1058    #endif
1059                                                            rdpdr_send_completion(iorq->device,
1060                                                                                  iorq->id, status,
1061                                                                                  iorq->partial_len,
1062                                                                                  iorq->buffer,
1063                                                                                  iorq->partial_len);
1064                                                            iorq = rdpdr_remove_iorequest(prev, iorq);
1065                                                    }
1066                                          }                                          }
1067                                          break;                                          break;
   
1068                                  case IRP_MJ_WRITE:                                  case IRP_MJ_WRITE:
   
1069                                          if (FD_ISSET(iorq->fd, wfds))                                          if (FD_ISSET(iorq->fd, wfds))
1070                                          {                                          {
1071                                                  // Write data and send completion.                                                  /* Write data. */
1072                                                  fns = iorq->fns;                                                  fns = iorq->fns;
1073                                                  status = fns->write(iorq->fd, iorq->buffer,  
1074                                                                      iorq->length, 0, &result);                                                  req_size =
1075                                                            (iorq->length - iorq->partial_len) >
1076                                                            8192 ? 8192 : (iorq->length -
1077                                                                           iorq->partial_len);
1078    
1079                                                    /* never write larger chunks than 8k - chances are that it will block */
1080                                                    status = fns->write(iorq->fd,
1081                                                                        iorq->buffer +
1082                                                                        iorq->partial_len, req_size,
1083                                                                        iorq->offset, &result);
1084    
1085                                                    if ((long) result > 0)
1086                                                    {
1087                                                            iorq->partial_len += result;
1088                                                            iorq->offset += result;
1089                                                    }
1090    
1091    #if WITH_DEBUG_RDP5
1092                                                    DEBUG(("RDPDR: %d bytes of data written\n",
1093                                                           result));
1094    #endif
1095                                                    /* only delete link if all data has been transfered */
1096                                                    /* or we couldn't write */
1097                                                    if ((iorq->partial_len == iorq->length)
1098                                                        || (result == 0))
1099                                                    {
1100    #if WITH_DEBUG_RDP5
1101                                                            DEBUG(("RDPDR: AIO total %u bytes written of %u\n", iorq->partial_len, iorq->length));
1102    #endif
1103                                                            rdpdr_send_completion(iorq->device,
1104                                                                                  iorq->id, status,
1105                                                                                  iorq->partial_len,
1106                                                                                  (uint8 *) "", 1);
1107    
1108                                                            iorq = rdpdr_remove_iorequest(prev, iorq);
1109                                                    }
1110                                            }
1111                                            break;
1112                                    case IRP_MJ_DEVICE_CONTROL:
1113                                            if (serial_get_event(iorq->fd, &result))
1114                                            {
1115                                                    buffer = (uint8 *) xrealloc((void *) buffer, 0x14);
1116                                                    out.data = out.p = buffer;
1117                                                    out.size = sizeof(buffer);
1118                                                    out_uint32_le(&out, result);
1119                                                    result = buffer_len = out.p - out.data;
1120                                                    status = RD_STATUS_SUCCESS;
1121                                                  rdpdr_send_completion(iorq->device, iorq->id,                                                  rdpdr_send_completion(iorq->device, iorq->id,
1122                                                                        status, result, "", 1);                                                                        status, result, buffer,
1123                                                                          buffer_len);
1124                                                    xfree(buffer);
1125                                                    iorq = rdpdr_remove_iorequest(prev, iorq);
1126                                            }
1127    
1128                                            break;
1129                            }
1130    
1131                    }
1132                    prev = iorq;
1133                    if (iorq)
1134                            iorq = iorq->next;
1135            }
1136    
1137            /* Check notify */
1138            iorq = g_iorequest;
1139            prev = NULL;
1140            while (iorq != NULL)
1141            {
1142                    if (iorq->fd != 0)
1143                    {
1144                            switch (iorq->major)
1145                            {
1146    
1147                                                  xfree(iorq->buffer);                                  case IRP_MJ_DIRECTORY_CONTROL:
1148                                                  iorq->fd = 0;                                          if (g_rdpdr_device[iorq->device].device_type ==
1149                                                DEVICE_TYPE_DISK)
1150                                            {
1151    
1152                                                    if (g_notify_stamp)
1153                                                    {
1154                                                            g_notify_stamp = False;
1155                                                            status = disk_check_notify(iorq->fd);
1156                                                            if (status != RD_STATUS_PENDING)
1157                                                            {
1158                                                                    rdpdr_send_completion(iorq->device,
1159                                                                                          iorq->id,
1160                                                                                          status, 0,
1161                                                                                          NULL, 0);
1162                                                                    iorq = rdpdr_remove_iorequest(prev,
1163                                                                                                  iorq);
1164                                                            }
1165                                                    }
1166                                          }                                          }
1167                                          break;                                          break;
1168    
1169    
1170    
1171                          }                          }
1172                  }                  }
1173    
1174                    prev = iorq;
1175                    if (iorq)
1176                            iorq = iorq->next;
1177          }          }
1178    
1179    }
1180    
1181    void
1182    rdpdr_check_fds(fd_set * rfds, fd_set * wfds, BOOL timed_out)
1183    {
1184            fd_set dummy;
1185    
1186    
1187            FD_ZERO(&dummy);
1188    
1189    
1190            /* fist check event queue only,
1191               any serial wait event must be done before read block will be sent
1192             */
1193    
1194            _rdpdr_check_fds(&dummy, &dummy, False);
1195            _rdpdr_check_fds(rfds, wfds, timed_out);
1196  }  }
1197    
1198    
1199  /* Abort a pending io request for a given handle and major */  /* Abort a pending io request for a given handle and major */
1200  BOOL  BOOL
1201  rdpdr_abort_io(uint32 fd, uint32 major, NTSTATUS status)  rdpdr_abort_io(uint32 fd, uint32 major, RD_NTSTATUS status)
1202  {  {
1203          uint32 result;          uint32 result;
         int i;  
1204          struct async_iorequest *iorq;          struct async_iorequest *iorq;
1205            struct async_iorequest *prev;
1206    
1207          for (i = 0; i < MAX_ASYNC_IO_REQUESTS; i++)          iorq = g_iorequest;
1208            prev = NULL;
1209            while (iorq != NULL)
1210          {          {
1211                  iorq = &g_iorequest[i];                  /* Only remove from table when major is not set, or when correct major is supplied.
1212                       Abort read should not abort a write io request. */
                 // Only remove from table when major is not set, or when correct major is supplied.  
                 // Abort read should not abort a write io request.  
1213                  if ((iorq->fd == fd) && (major == 0 || iorq->major == major))                  if ((iorq->fd == fd) && (major == 0 || iorq->major == major))
1214                  {                  {
1215                          result = 0;                          result = 0;
1216                          rdpdr_send_completion(iorq->device, iorq->id, status, result, "", 1);                          rdpdr_send_completion(iorq->device, iorq->id, status, result, (uint8 *) "",
1217                          xfree(iorq->buffer);                                                1);
1218                          iorq->fd = 0;  
1219                            iorq = rdpdr_remove_iorequest(prev, iorq);
1220                          return True;                          return True;
1221                  }                  }
1222    
1223                    prev = iorq;
1224                    iorq = iorq->next;
1225          }          }
1226    
1227          return False;          return False;
1228  }  }
Legend:
Removed from v.576  
changed lines
  Added in v.1365
  ViewVC Help
Powered by ViewVC 1.1.26