/[rdesktop]/jpeg/rdesktop/trunk/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 /jpeg/rdesktop/trunk/rdpdr.c

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

revision 590 by n-ki, Thu Jan 29 12:27:21 2004 UTC revision 673 by astrand, Mon Apr 19 08:24:54 2004 UTC
# Line 1  Line 1 
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>
39    #include <dirent.h>             /* opendir, closedir, readdir */
40  #include <time.h>  #include <time.h>
41  #include "rdesktop.h"  #include "rdesktop.h"
42    
 #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  
   
43  #define IRP_MJ_CREATE                   0x00  #define IRP_MJ_CREATE                   0x00
44  #define IRP_MJ_CLOSE                    0x02  #define IRP_MJ_CLOSE                    0x02
45  #define IRP_MJ_READ                     0x03  #define IRP_MJ_READ                     0x03
# Line 18  Line 49 
49  #define IRP_MJ_QUERY_VOLUME_INFORMATION 0x0a  #define IRP_MJ_QUERY_VOLUME_INFORMATION 0x0a
50  #define IRP_MJ_DIRECTORY_CONTROL        0x0c  #define IRP_MJ_DIRECTORY_CONTROL        0x0c
51  #define IRP_MJ_DEVICE_CONTROL           0x0e  #define IRP_MJ_DEVICE_CONTROL           0x0e
52    #define IRP_MJ_LOCK_CONTROL             0x11
53    
54  #define IRP_MN_QUERY_DIRECTORY          0x01  #define IRP_MN_QUERY_DIRECTORY          0x01
55  #define IRP_MN_NOTIFY_CHANGE_DIRECTORY  0x02  #define IRP_MN_NOTIFY_CHANGE_DIRECTORY  0x02
56    
 //#define MAX_ASYNC_IO_REQUESTS   10  
   
57  extern char hostname[16];  extern char hostname[16];
58  extern DEVICE_FNS serial_fns;  extern DEVICE_FNS serial_fns;
59  extern DEVICE_FNS printer_fns;  extern DEVICE_FNS printer_fns;
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[];
63    
64  static VCHANNEL *rdpdr_channel;  static VCHANNEL *rdpdr_channel;
65    
# Line 39  uint32 g_num_devices; Line 69  uint32 g_num_devices;
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;
73    
 #if 0  
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, */
76    /* if multiple ios are being done on the same fd */
77  struct async_iorequest  struct async_iorequest
78  {  {
79          uint32 fd, major, minor, offset, device, id, length;          uint32 fd, major, minor, offset, device, id, length, partial_len;
80          long timeout,           /* Total timeout */          long timeout,           /* Total timeout */
81            itv_timeout;          /* Interval timeout (between serial characters) */            itv_timeout;          /* Interval timeout (between serial characters) */
82          uint8 *buffer;          uint8 *buffer;
83          DEVICE_FNS *fns;          DEVICE_FNS *fns;
84  } g_iorequest[MAX_ASYNC_IO_REQUESTS];  
85  #endif          struct async_iorequest *next;   /* next element in list */
86    };
87    
88    struct async_iorequest *g_iorequest;
89    
90  /* Return device_id for a given handle */  /* Return device_id for a given handle */
91  int  int
# Line 77  convert_to_unix_filename(char *filename) Line 112  convert_to_unix_filename(char *filename)
112          }          }
113  }  }
114    
115  #if 0  static BOOL
116    rdpdr_handle_ok(int device, int handle)
117    {
118            switch (g_rdpdr_device[device].device_type)
119            {
120                    case DEVICE_TYPE_PARALLEL:
121                    case DEVICE_TYPE_SERIAL:
122                    case DEVICE_TYPE_PRINTER:
123                    case DEVICE_TYPE_SCARD:
124                            if (g_rdpdr_device[device].handle != handle)
125                                    return False;
126                            break;
127                    case DEVICE_TYPE_DISK:
128                            if (g_fileinfo[handle].device_id != device)
129                                    return False;
130                            break;
131            }
132            return True;
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, long total_timeout, long interval_timeout, uint8 * buffer)                      DEVICE_FNS * fns, uint32 total_timeout, uint32 interval_timeout, uint8 * buffer,
139                        uint32 offset)
140  {  {
         int i;  
141          struct async_iorequest *iorq;          struct async_iorequest *iorq;
142    
143          for (i = 0; i < MAX_ASYNC_IO_REQUESTS; i++)          if (g_iorequest == NULL)
144          {          {
145                  iorq = &g_iorequest[i];                  g_iorequest = (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));
146                    if (!g_iorequest)
147                            return False;
148                    g_iorequest->fd = 0;
149                    g_iorequest->next = NULL;
150            }
151    
152                  if (iorq->fd == 0)          iorq = g_iorequest;
153    
154            while (iorq->fd != 0)
155            {
156                    // create new element if needed
157                    if (iorq->next == NULL)
158                  {                  {
159                          iorq->device = device;                          iorq->next =
160                          iorq->fd = file;                                  (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));
161                          iorq->id = id;                          if (!iorq->next)
162                          iorq->major = major;                                  return False;
163                          iorq->length = length;                          iorq->next->fd = 0;
164                          iorq->fns = fns;                          iorq->next->next = NULL;
                         iorq->timeout = total_timeout;  
                         iorq->itv_timeout = interval_timeout;  
                         iorq->buffer = buffer;  
                         return True;  
165                  }                  }
166                    iorq = iorq->next;
167          }          }
168          error("IO request table full. Increase MAX_ASYNC_IO_REQUESTS in rdpdr.c!\n");          iorq->device = device;
169          return False;          iorq->fd = file;
170            iorq->id = id;
171            iorq->major = major;
172            iorq->length = length;
173            iorq->partial_len = 0;
174            iorq->fns = fns;
175            iorq->timeout = total_timeout;
176            iorq->itv_timeout = interval_timeout;
177            iorq->buffer = buffer;
178            iorq->offset = offset;
179            return True;
180  }  }
 #endif  
181    
182  void  static void
183  rdpdr_send_connect(void)  rdpdr_send_connect(void)
184  {  {
185          uint8 magic[4] = "rDCC";          uint8 magic[4] = "rDCC";
# Line 125  rdpdr_send_connect(void) Line 195  rdpdr_send_connect(void)
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";
         uint32 hostlen = (strlen(hostname) + 1) * 2;  
202          STREAM s;          STREAM s;
203            uint32 hostlen;
204    
205            if (NULL == g_rdpdr_clientname)
206            {
207                    g_rdpdr_clientname = hostname;
208            }
209            hostlen = (strlen(g_rdpdr_clientname) + 1) * 2;
210    
211          s = channel_init(rdpdr_channel, 16 + hostlen);          s = channel_init(rdpdr_channel, 16 + hostlen);
212          out_uint8a(s, magic, 4);          out_uint8a(s, magic, 4);
# Line 138  rdpdr_send_name(void) Line 214  rdpdr_send_name(void)
214          out_uint16_le(s, 0x72);          out_uint16_le(s, 0x72);
215          out_uint32(s, 0);          out_uint32(s, 0);
216          out_uint32_le(s, hostlen);          out_uint32_le(s, hostlen);
217          rdp_out_unistr(s, hostname, hostlen - 2);          rdp_out_unistr(s, g_rdpdr_clientname, hostlen - 2);
218          s_mark_end(s);          s_mark_end(s);
219          channel_send(s, rdpdr_channel);          channel_send(s, rdpdr_channel);
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;
# Line 171  announcedata_size() Line 247  announcedata_size()
247          return size;          return size;
248  }  }
249    
250  void  static void
251  rdpdr_send_available(void)  rdpdr_send_available(void)
252  {  {
253    
# Line 210  rdpdr_send_available(void) Line 286  rdpdr_send_available(void)
286                                  rdp_out_unistr(s, printerinfo->printer, printerlen - 2);                                  rdp_out_unistr(s, printerinfo->printer, printerlen - 2);
287                                  out_uint8a(s, printerinfo->blob, bloblen);                                  out_uint8a(s, printerinfo->blob, bloblen);
288    
289                                  xfree(printerinfo->blob);       /* Blob is sent twice if reconnecting */                                  if (printerinfo->blob)
290                                            xfree(printerinfo->blob);       /* Blob is sent twice if reconnecting */
291                                  break;                                  break;
292                          default:                          default:
293                                  out_uint32(s, 0);                                  out_uint32(s, 0);
# Line 228  rdpdr_send_available(void) Line 305  rdpdr_send_available(void)
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  {  {
# Line 290  rdpdr_process_irp(STREAM s) Line 367  rdpdr_process_irp(STREAM s)
367                  case DEVICE_TYPE_SERIAL:                  case DEVICE_TYPE_SERIAL:
368    
369                          fns = &serial_fns;                          fns = &serial_fns;
370                          /* should be async when aio is finished */                          rw_blocking = False;
                         /*rw_blocking = False; */  
371                          break;                          break;
372    
373                  case DEVICE_TYPE_PARALLEL:                  case DEVICE_TYPE_PARALLEL:
374    
375                          fns = &parallel_fns;                          fns = &parallel_fns;
376                          /* should be async when aio is finished */                          rw_blocking = False;
                         /*rw_blocking = False;*/  
377                          break;                          break;
378    
379                  case DEVICE_TYPE_PRINTER:                  case DEVICE_TYPE_PRINTER:
# Line 309  rdpdr_process_irp(STREAM s) Line 384  rdpdr_process_irp(STREAM s)
384                  case DEVICE_TYPE_DISK:                  case DEVICE_TYPE_DISK:
385    
386                          fns = &disk_fns;                          fns = &disk_fns;
387                            rw_blocking = False;
388                          break;                          break;
389    
390                  case DEVICE_TYPE_SCARD:                  case DEVICE_TYPE_SCARD:
# Line 374  rdpdr_process_irp(STREAM s) Line 450  rdpdr_process_irp(STREAM s)
450  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
451                          DEBUG(("RDPDR IRP Read (length: %d, offset: %d)\n", length, offset));                          DEBUG(("RDPDR IRP Read (length: %d, offset: %d)\n", length, offset));
452  #endif  #endif
453  //                      if (rw_blocking)        // Complete read immediately                          if (!rdpdr_handle_ok(device, file))
454  //                      {                          {
455                                    status = STATUS_INVALID_HANDLE;
456                                    break;
457                            }
458    
459                            if (rw_blocking)        // Complete read immediately
460                            {
461                                  buffer = (uint8 *) xrealloc((void *) buffer, length);                                  buffer = (uint8 *) xrealloc((void *) buffer, length);
462                                    if (!buffer)
463                                    {
464                                            status = STATUS_CANCELLED;
465                                            break;
466                                    }
467                                  status = fns->read(file, buffer, length, offset, &result);                                  status = fns->read(file, buffer, length, offset, &result);
468                                  buffer_len = result;                                  buffer_len = result;
469                                  break;                                  break;
470  //                      }                          }
471    
 #if 0  
472                          // Add request to table                          // Add request to table
473                          pst_buf = (uint8 *) xmalloc(length);                          pst_buf = (uint8 *) xmalloc(length);
474                            if (!pst_buf)
475                            {
476                                    status = STATUS_CANCELLED;
477                                    break;
478                            }
479                          serial_get_timeout(file, length, &total_timeout, &interval_timeout);                          serial_get_timeout(file, length, &total_timeout, &interval_timeout);
480                          if (add_async_iorequest                          if (add_async_iorequest
481                              (device, file, id, major, length, fns, total_timeout, interval_timeout,                              (device, file, id, major, length, fns, total_timeout, interval_timeout,
482                               pst_buf))                               pst_buf, offset))
483                          {                          {
484                                  status = STATUS_PENDING;                                  status = STATUS_PENDING;
485                                  break;                                  break;
# Line 396  rdpdr_process_irp(STREAM s) Line 487  rdpdr_process_irp(STREAM s)
487    
488                          status = STATUS_CANCELLED;                          status = STATUS_CANCELLED;
489                          break;                          break;
 #endif  
490                  case IRP_MJ_WRITE:                  case IRP_MJ_WRITE:
491    
492                          buffer_len = 1;                          buffer_len = 1;
# Line 413  rdpdr_process_irp(STREAM s) Line 503  rdpdr_process_irp(STREAM s)
503  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
504                          DEBUG(("RDPDR IRP Write (length: %d)\n", result));                          DEBUG(("RDPDR IRP Write (length: %d)\n", result));
505  #endif  #endif
506  //                      if (rw_blocking)        // Complete immediately                          if (!rdpdr_handle_ok(device, file))
507  //                      {                          {
508                                    status = STATUS_INVALID_HANDLE;
509                                    break;
510                            }
511    
512                            if (rw_blocking)        // Complete immediately
513                            {
514                                  status = fns->write(file, s->p, length, offset, &result);                                  status = fns->write(file, s->p, length, offset, &result);
515                                  break;                                  break;
516  //                      }                          }
517  #if 0  
518                          // Add to table                          // Add to table
519                          pst_buf = (uint8 *) xmalloc(length);                          pst_buf = (uint8 *) xmalloc(length);
520                            if (!pst_buf)
521                            {
522                                    status = STATUS_CANCELLED;
523                                    break;
524                            }
525    
526                          in_uint8a(s, pst_buf, length);                          in_uint8a(s, pst_buf, length);
527    
528                          if (add_async_iorequest                          if (add_async_iorequest
529                              (device, file, id, major, length, fns, 0, 0, pst_buf))                              (device, file, id, major, length, fns, 0, 0, pst_buf, offset))
530                          {                          {
531                                  status = STATUS_PENDING;                                  status = STATUS_PENDING;
532                                  break;                                  break;
# Line 432  rdpdr_process_irp(STREAM s) Line 534  rdpdr_process_irp(STREAM s)
534    
535                          status = STATUS_CANCELLED;                          status = STATUS_CANCELLED;
536                          break;                          break;
 #endif  
537    
538                  case IRP_MJ_QUERY_INFORMATION:                  case IRP_MJ_QUERY_INFORMATION:
539    
# Line 545  rdpdr_process_irp(STREAM s) Line 646  rdpdr_process_irp(STREAM s)
646                          in_uint8s(s, 0x14);                          in_uint8s(s, 0x14);
647    
648                          buffer = (uint8 *) xrealloc((void *) buffer, bytes_out + 0x14);                          buffer = (uint8 *) xrealloc((void *) buffer, bytes_out + 0x14);
649                            if (!buffer)
650                            {
651                                    status = STATUS_CANCELLED;
652                                    break;
653                            }
654    
655                          out.data = out.p = buffer;                          out.data = out.p = buffer;
656                          out.size = sizeof(buffer);                          out.size = sizeof(buffer);
657                          status = fns->device_control(file, request, s, &out);                          status = fns->device_control(file, request, s, &out);
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;
# Line 560  rdpdr_process_irp(STREAM s) Line 686  rdpdr_process_irp(STREAM s)
686          {          {
687                  rdpdr_send_completion(device, id, status, result, buffer, buffer_len);                  rdpdr_send_completion(device, id, status, result, buffer, buffer_len);
688          }          }
689          xfree(buffer);          if (buffer)
690                    xfree(buffer);
691            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";
# Line 676  rdpdr_init() Line 803  rdpdr_init()
803          return (rdpdr_channel != NULL);          return (rdpdr_channel != NULL);
804  }  }
805    
 #if 0  
806  /* Add file descriptors of pending io request to select() */  /* Add file descriptors of pending io request to select() */
807  void  void
808  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)
809  {  {
810          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).  
811          struct async_iorequest *iorq;          struct async_iorequest *iorq;
812            char c;
813    
814          for (i = 0; i < MAX_ASYNC_IO_REQUESTS; i++)          iorq = g_iorequest;
815            while (iorq != NULL)
816          {          {
817                  iorq = &g_iorequest[i];                  if (iorq->fd != 0)
   
                 if (iorq->fd != 0)      // Found a pending io request  
818                  {                  {
819                          switch (iorq->major)                          switch (iorq->major)
820                          {                          {
821                                  case IRP_MJ_READ:                                  case IRP_MJ_READ:
822                                            /* Is this FD valid? FDs will
823                                               be invalid when
824                                               reconnecting. FIXME: Real
825                                               support for reconnects. */
826    
827                                            if (read(iorq->fd, &c, 0) != 0)
828                                                    break;
829    
830                                          FD_SET(iorq->fd, rfds);                                          FD_SET(iorq->fd, rfds);
831                                            *n = MAX(*n, iorq->fd);
832    
833                                          // Check if io request timeout is smaller than current (but not 0).                                          // Check if io request timeout is smaller than current (but not 0).
834                                          if (iorq->timeout                                          if (iorq->timeout
# Line 709  rdpdr_add_fds(int *n, fd_set * rfds, fd_ Line 842  rdpdr_add_fds(int *n, fd_set * rfds, fd_
842                                                  tv->tv_usec = (select_timeout % 1000) * 1000;                                                  tv->tv_usec = (select_timeout % 1000) * 1000;
843                                                  *timeout = True;                                                  *timeout = True;
844                                          }                                          }
845    
846                                          break;                                          break;
847    
848                                  case IRP_MJ_WRITE:                                  case IRP_MJ_WRITE:
849                                            /* FD still valid? See above. */
850                                            if (write(iorq->fd, &c, 0) != 0)
851                                                    break;
852    
853                                          FD_SET(iorq->fd, wfds);                                          FD_SET(iorq->fd, wfds);
854                                            *n = MAX(*n, iorq->fd);
855                                          break;                                          break;
856    
857                          }                          }
858                          *n = MAX(*n, iorq->fd);  
859                  }                  }
860    
861                    iorq = iorq->next;
862          }          }
863  }  }
864    
865    struct async_iorequest *
866    rdpdr_remove_iorequest(struct async_iorequest *prev, struct async_iorequest *iorq)
867    {
868            if (!iorq)
869                    return NULL;
870    
871            if (iorq->buffer)
872                    xfree(iorq->buffer);
873            if (prev)
874            {
875                    prev->next = iorq->next;
876                    xfree(iorq);
877                    iorq = prev->next;
878            }
879            else
880            {
881                    // Even if NULL
882                    g_iorequest = iorq->next;
883                    xfree(iorq);
884                    iorq = NULL;
885            }
886            return iorq;
887    }
888    
889  /* 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 */
890  void  void
891  rdpdr_check_fds(fd_set * rfds, fd_set * wfds, BOOL timed_out)  rdpdr_check_fds(fd_set * rfds, fd_set * wfds, BOOL timed_out)
892  {  {
         int i;  
893          NTSTATUS status;          NTSTATUS status;
894          uint32 result = 0, buffer_len = 0;          uint32 result = 0;
895          DEVICE_FNS *fns;          DEVICE_FNS *fns;
896          struct async_iorequest *iorq;          struct async_iorequest *iorq;
897            struct async_iorequest *prev;
898            uint32 req_size = 0;
899    
900          if (timed_out)          if (timed_out)
901          {          {
# Line 739  rdpdr_check_fds(fd_set * rfds, fd_set * Line 903  rdpdr_check_fds(fd_set * rfds, fd_set *
903                  return;                  return;
904          }          }
905    
906          // Walk through array of pending io_rq's          iorq = g_iorequest;
907          for (i = 0; i < MAX_ASYNC_IO_REQUESTS; i++)          prev = NULL;
908            while (iorq != NULL)
909          {          {
                 iorq = &g_iorequest[i];  
   
910                  if (iorq->fd != 0)                  if (iorq->fd != 0)
911                  {                  {
912                          switch (iorq->major)                          switch (iorq->major)
913                          {                          {
   
914                                  case IRP_MJ_READ:                                  case IRP_MJ_READ:
   
915                                          if (FD_ISSET(iorq->fd, rfds))                                          if (FD_ISSET(iorq->fd, rfds))
916                                          {                                          {
917                                                  // Read, and send data.                                                  /* Read the data */
918                                                  fns = iorq->fns;                                                  fns = iorq->fns;
919                                                  status = fns->read(iorq->fd, iorq->buffer,  
920                                                                     iorq->length, 0, &result);                                                  req_size =
921                                                  buffer_len = result;                                                          (iorq->length - iorq->partial_len) >
922                                                            8192 ? 8192 : (iorq->length -
923                                                  rdpdr_send_completion(iorq->device, iorq->id,                                                                         iorq->partial_len);
924                                                                        status, result, iorq->buffer,                                                  /* never read larger chunks than 8k - chances are that it will block */
925                                                                        buffer_len);                                                  status = fns->read(iorq->fd,
926                                                                       iorq->buffer + iorq->partial_len,
927                                                                       req_size, iorq->offset, &result);
928    
929                                                    if (result > 0)
930                                                    {
931                                                            iorq->partial_len += result;
932                                                            iorq->offset += result;
933                                                    }
934  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
935                                                  DEBUG(("RDPDR: %d bytes of data read\n", result));                                                  DEBUG(("RDPDR: %d bytes of data read\n", result));
936  #endif  #endif
937                                                  xfree(iorq->buffer);                                                  /* only delete link if all data has been transfered */
938                                                  iorq->fd = 0;                                                  /* or if result was 0 and status success - EOF      */
939                                                    if ((iorq->partial_len == iorq->length) ||
940                                                        (result == 0))
941                                                    {
942    #if WITH_DEBUG_RDP5
943                                                            DEBUG(("RDPDR: AIO total %u bytes read of %u\n", iorq->partial_len, iorq->length));
944    #endif
945                                                            rdpdr_send_completion(iorq->device,
946                                                                                  iorq->id, status,
947                                                                                  iorq->partial_len,
948                                                                                  iorq->buffer,
949                                                                                  iorq->partial_len);
950                                                            iorq = rdpdr_remove_iorequest(prev, iorq);
951                                                    }
952                                          }                                          }
953                                          break;                                          break;
   
954                                  case IRP_MJ_WRITE:                                  case IRP_MJ_WRITE:
   
955                                          if (FD_ISSET(iorq->fd, wfds))                                          if (FD_ISSET(iorq->fd, wfds))
956                                          {                                          {
957                                                  // Write data and send completion.                                                  /* Write data. */
958                                                  fns = iorq->fns;                                                  fns = iorq->fns;
                                                 status = fns->write(iorq->fd, iorq->buffer,  
                                                                     iorq->length, 0, &result);  
                                                 rdpdr_send_completion(iorq->device, iorq->id,  
                                                                       status, result, "", 1);  
959    
960                                                  xfree(iorq->buffer);                                                  req_size =
961                                                  iorq->fd = 0;                                                          (iorq->length - iorq->partial_len) >
962                                                            8192 ? 8192 : (iorq->length -
963                                                                           iorq->partial_len);
964    
965                                                    /* never write larger chunks than 8k - chances are that it will block */
966                                                    status = fns->write(iorq->fd,
967                                                                        iorq->buffer +
968                                                                        iorq->partial_len, req_size,
969                                                                        iorq->offset, &result);
970    
971                                                    if (result > 0)
972                                                    {
973                                                            iorq->partial_len += result;
974                                                            iorq->offset += result;
975                                                    }
976    
977    #if WITH_DEBUG_RDP5
978                                                    DEBUG(("RDPDR: %d bytes of data written\n",
979                                                           result));
980    #endif
981                                                    /* only delete link if all data has been transfered */
982                                                    /* or we couldn't write */
983                                                    if ((iorq->partial_len == iorq->length)
984                                                        || (result == 0))
985                                                    {
986    #if WITH_DEBUG_RDP5
987                                                            DEBUG(("RDPDR: AIO total %u bytes written of %u\n", iorq->partial_len, iorq->length));
988    #endif
989                                                            rdpdr_send_completion(iorq->device,
990                                                                                  iorq->id, status,
991                                                                                  iorq->partial_len,
992                                                                                  (uint8 *) "", 1);
993    
994                                                            iorq = rdpdr_remove_iorequest(prev, iorq);
995                                                    }
996                                          }                                          }
997                                          break;                                          break;
998                          }                          }
999    
1000                  }                  }
1001                    prev = iorq;
1002                    if (iorq)
1003                            iorq = iorq->next;
1004          }          }
1005    
1006  }  }
1007    
1008  /* Abort a pending io request for a given handle and major */  /* Abort a pending io request for a given handle and major */
# Line 795  BOOL Line 1010  BOOL
1010  rdpdr_abort_io(uint32 fd, uint32 major, NTSTATUS status)  rdpdr_abort_io(uint32 fd, uint32 major, NTSTATUS status)
1011  {  {
1012          uint32 result;          uint32 result;
         int i;  
1013          struct async_iorequest *iorq;          struct async_iorequest *iorq;
1014            struct async_iorequest *prev;
1015    
1016          for (i = 0; i < MAX_ASYNC_IO_REQUESTS; i++)          iorq = g_iorequest;
1017            prev = NULL;
1018            while (iorq != NULL)
1019          {          {
                 iorq = &g_iorequest[i];  
   
1020                  // Only remove from table when major is not set, or when correct major is supplied.                  // Only remove from table when major is not set, or when correct major is supplied.
1021                  // Abort read should not abort a write io request.                  // Abort read should not abort a write io request.
1022                  if ((iorq->fd == fd) && (major == 0 || iorq->major == major))                  if ((iorq->fd == fd) && (major == 0 || iorq->major == major))
1023                  {                  {
1024                          result = 0;                          result = 0;
1025                          rdpdr_send_completion(iorq->device, iorq->id, status, result, "", 1);                          rdpdr_send_completion(iorq->device, iorq->id, status, result, (uint8 *) "",
1026                          xfree(iorq->buffer);                                                1);
1027                          iorq->fd = 0;  
1028                            iorq = rdpdr_remove_iorequest(prev, iorq);
1029                          return True;                          return True;
1030                  }                  }
1031    
1032                    prev = iorq;
1033                    iorq = iorq->next;
1034          }          }
1035    
1036          return False;          return False;
1037  }  }
 #endif  
Legend:
Removed from v.590  
changed lines
  Added in v.673
  ViewVC Help
Powered by ViewVC 1.1.26