/[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 602 by stargo, Sat Feb 7 17:32:21 2004 UTC revision 776 by jsorg71, Sat Oct 2 01:30:33 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 <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  extern char hostname[16];  extern char g_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    extern FILEINFO g_fileinfo[];
64    
65  static VCHANNEL *rdpdr_channel;  static VCHANNEL *rdpdr_channel;
66    
67  /* 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 */
68  HANDLE g_min_timeout_fd;  NTHANDLE g_min_timeout_fd;
69  uint32 g_num_devices;  uint32 g_num_devices;
70    
71  /* Table with information about rdpdr devices */  /* Table with information about rdpdr devices */
72  RDPDR_DEVICE g_rdpdr_device[RDPDR_MAX_DEVICES];  RDPDR_DEVICE g_rdpdr_device[RDPDR_MAX_DEVICES];
73    char *g_rdpdr_clientname = NULL;
74    
75  /* 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 */
76  /* using a linked list ensures that they are processed in the right order, */  /* using a linked list ensures that they are processed in the right order, */
# Line 55  struct async_iorequest *g_iorequest; Line 90  struct async_iorequest *g_iorequest;
90    
91  /* Return device_id for a given handle */  /* Return device_id for a given handle */
92  int  int
93  get_device_index(HANDLE handle)  get_device_index(NTHANDLE handle)
94  {  {
95          int i;          int i;
96          for (i = 0; i < RDPDR_MAX_DEVICES; i++)          for (i = 0; i < RDPDR_MAX_DEVICES; i++)
# Line 78  convert_to_unix_filename(char *filename) Line 113  convert_to_unix_filename(char *filename)
113          }          }
114  }  }
115    
116    static BOOL
117    rdpdr_handle_ok(int device, int handle)
118    {
119            switch (g_rdpdr_device[device].device_type)
120            {
121                    case DEVICE_TYPE_PARALLEL:
122                    case DEVICE_TYPE_SERIAL:
123                    case DEVICE_TYPE_PRINTER:
124                    case DEVICE_TYPE_SCARD:
125                            if (g_rdpdr_device[device].handle != handle)
126                                    return False;
127                            break;
128                    case DEVICE_TYPE_DISK:
129                            if (g_fileinfo[handle].device_id != device)
130                                    return False;
131                            break;
132            }
133            return True;
134    }
135    
136  /* 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 */
137  BOOL  static BOOL
138  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,
139                      DEVICE_FNS * fns, long total_timeout, long interval_timeout, uint8 * buffer)                      DEVICE_FNS * fns, uint32 total_timeout, uint32 interval_timeout, uint8 * buffer,
140                        uint32 offset)
141  {  {
142          struct async_iorequest *iorq;          struct async_iorequest *iorq;
143    
144          if (g_iorequest == NULL)          if (g_iorequest == NULL)
145          {          {
146                  g_iorequest = (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));                  g_iorequest = (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));
147                    if (!g_iorequest)
148                            return False;
149                  g_iorequest->fd = 0;                  g_iorequest->fd = 0;
150                  g_iorequest->next = NULL;                  g_iorequest->next = NULL;
151          }          }
# Line 101  add_async_iorequest(uint32 device, uint3 Line 159  add_async_iorequest(uint32 device, uint3
159                  {                  {
160                          iorq->next =                          iorq->next =
161                                  (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));                                  (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));
162                            if (!iorq->next)
163                                    return False;
164                          iorq->next->fd = 0;                          iorq->next->fd = 0;
165                          iorq->next->next = NULL;                          iorq->next->next = NULL;
166                  }                  }
# Line 116  add_async_iorequest(uint32 device, uint3 Line 176  add_async_iorequest(uint32 device, uint3
176          iorq->timeout = total_timeout;          iorq->timeout = total_timeout;
177          iorq->itv_timeout = interval_timeout;          iorq->itv_timeout = interval_timeout;
178          iorq->buffer = buffer;          iorq->buffer = buffer;
179            iorq->offset = offset;
180          return True;          return True;
181  }  }
182    
183  void  static void
184  rdpdr_send_connect(void)  rdpdr_send_connect(void)
185  {  {
186          uint8 magic[4] = "rDCC";          uint8 magic[4] = "rDCC";
# Line 135  rdpdr_send_connect(void) Line 196  rdpdr_send_connect(void)
196  }  }
197    
198    
199  void  static void
200  rdpdr_send_name(void)  rdpdr_send_name(void)
201  {  {
202          uint8 magic[4] = "rDNC";          uint8 magic[4] = "rDNC";
         uint32 hostlen = (strlen(hostname) + 1) * 2;  
203          STREAM s;          STREAM s;
204            uint32 hostlen;
205    
206            if (NULL == g_rdpdr_clientname)
207            {
208                    g_rdpdr_clientname = g_hostname;
209            }
210            hostlen = (strlen(g_rdpdr_clientname) + 1) * 2;
211    
212          s = channel_init(rdpdr_channel, 16 + hostlen);          s = channel_init(rdpdr_channel, 16 + hostlen);
213          out_uint8a(s, magic, 4);          out_uint8a(s, magic, 4);
# Line 148  rdpdr_send_name(void) Line 215  rdpdr_send_name(void)
215          out_uint16_le(s, 0x72);          out_uint16_le(s, 0x72);
216          out_uint32(s, 0);          out_uint32(s, 0);
217          out_uint32_le(s, hostlen);          out_uint32_le(s, hostlen);
218          rdp_out_unistr(s, hostname, hostlen - 2);          rdp_out_unistr(s, g_rdpdr_clientname, hostlen - 2);
219          s_mark_end(s);          s_mark_end(s);
220          channel_send(s, rdpdr_channel);          channel_send(s, rdpdr_channel);
221  }  }
222    
223  /* Returns the size of the payload of the announce packet */  /* Returns the size of the payload of the announce packet */
224  int  static int
225  announcedata_size()  announcedata_size()
226  {  {
227          int size, i;          int size, i;
# Line 181  announcedata_size() Line 248  announcedata_size()
248          return size;          return size;
249  }  }
250    
251  void  static void
252  rdpdr_send_available(void)  rdpdr_send_available(void)
253  {  {
254    
# Line 199  rdpdr_send_available(void) Line 266  rdpdr_send_available(void)
266          {          {
267                  out_uint32_le(s, g_rdpdr_device[i].device_type);                  out_uint32_le(s, g_rdpdr_device[i].device_type);
268                  out_uint32_le(s, i);    /* RDP Device ID */                  out_uint32_le(s, i);    /* RDP Device ID */
269                    /* Is it possible to use share names longer than 8 chars?
270                       /astrand */
271                  out_uint8p(s, g_rdpdr_device[i].name, 8);                  out_uint8p(s, g_rdpdr_device[i].name, 8);
272    
273                  switch (g_rdpdr_device[i].device_type)                  switch (g_rdpdr_device[i].device_type)
# Line 239  rdpdr_send_available(void) Line 308  rdpdr_send_available(void)
308          channel_send(s, rdpdr_channel);          channel_send(s, rdpdr_channel);
309  }  }
310    
311  void  static void
312  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,
313                        uint32 length)                        uint32 length)
314  {  {
# Line 384  rdpdr_process_irp(STREAM s) Line 453  rdpdr_process_irp(STREAM s)
453  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
454                          DEBUG(("RDPDR IRP Read (length: %d, offset: %d)\n", length, offset));                          DEBUG(("RDPDR IRP Read (length: %d, offset: %d)\n", length, offset));
455  #endif  #endif
456                            if (!rdpdr_handle_ok(device, file))
457                            {
458                                    status = STATUS_INVALID_HANDLE;
459                                    break;
460                            }
461    
462                          if (rw_blocking)        // Complete read immediately                          if (rw_blocking)        // Complete read immediately
463                          {                          {
464                                  buffer = (uint8 *) xrealloc((void *) buffer, length);                                  buffer = (uint8 *) xrealloc((void *) buffer, length);
465                                    if (!buffer)
466                                    {
467                                            status = STATUS_CANCELLED;
468                                            break;
469                                    }
470                                  status = fns->read(file, buffer, length, offset, &result);                                  status = fns->read(file, buffer, length, offset, &result);
471                                  buffer_len = result;                                  buffer_len = result;
472                                  break;                                  break;
# Line 394  rdpdr_process_irp(STREAM s) Line 474  rdpdr_process_irp(STREAM s)
474    
475                          // Add request to table                          // Add request to table
476                          pst_buf = (uint8 *) xmalloc(length);                          pst_buf = (uint8 *) xmalloc(length);
477                            if (!pst_buf)
478                            {
479                                    status = STATUS_CANCELLED;
480                                    break;
481                            }
482                          serial_get_timeout(file, length, &total_timeout, &interval_timeout);                          serial_get_timeout(file, length, &total_timeout, &interval_timeout);
483                          if (add_async_iorequest                          if (add_async_iorequest
484                              (device, file, id, major, length, fns, total_timeout, interval_timeout,                              (device, file, id, major, length, fns, total_timeout, interval_timeout,
485                               pst_buf))                               pst_buf, offset))
486                          {                          {
487                                  status = STATUS_PENDING;                                  status = STATUS_PENDING;
488                                  break;                                  break;
# Line 421  rdpdr_process_irp(STREAM s) Line 506  rdpdr_process_irp(STREAM s)
506  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
507                          DEBUG(("RDPDR IRP Write (length: %d)\n", result));                          DEBUG(("RDPDR IRP Write (length: %d)\n", result));
508  #endif  #endif
509                            if (!rdpdr_handle_ok(device, file))
510                            {
511                                    status = STATUS_INVALID_HANDLE;
512                                    break;
513                            }
514    
515                          if (rw_blocking)        // Complete immediately                          if (rw_blocking)        // Complete immediately
516                          {                          {
517                                  status = fns->write(file, s->p, length, offset, &result);                                  status = fns->write(file, s->p, length, offset, &result);
# Line 429  rdpdr_process_irp(STREAM s) Line 520  rdpdr_process_irp(STREAM s)
520    
521                          // Add to table                          // Add to table
522                          pst_buf = (uint8 *) xmalloc(length);                          pst_buf = (uint8 *) xmalloc(length);
523                            if (!pst_buf)
524                            {
525                                    status = STATUS_CANCELLED;
526                                    break;
527                            }
528    
529                          in_uint8a(s, pst_buf, length);                          in_uint8a(s, pst_buf, length);
530    
531                          if (add_async_iorequest                          if (add_async_iorequest
532                              (device, file, id, major, length, fns, 0, 0, pst_buf))                              (device, file, id, major, length, fns, 0, 0, pst_buf, offset))
533                          {                          {
534                                  status = STATUS_PENDING;                                  status = STATUS_PENDING;
535                                  break;                                  break;
# Line 552  rdpdr_process_irp(STREAM s) Line 649  rdpdr_process_irp(STREAM s)
649                          in_uint8s(s, 0x14);                          in_uint8s(s, 0x14);
650    
651                          buffer = (uint8 *) xrealloc((void *) buffer, bytes_out + 0x14);                          buffer = (uint8 *) xrealloc((void *) buffer, bytes_out + 0x14);
652                            if (!buffer)
653                            {
654                                    status = STATUS_CANCELLED;
655                                    break;
656                            }
657    
658                          out.data = out.p = buffer;                          out.data = out.p = buffer;
659                          out.size = sizeof(buffer);                          out.size = sizeof(buffer);
660                          status = fns->device_control(file, request, s, &out);                          status = fns->device_control(file, request, s, &out);
661                          result = buffer_len = out.p - out.data;                          result = buffer_len = out.p - out.data;
662                          break;                          break;
663    
664    
665                    case IRP_MJ_LOCK_CONTROL:
666    
667                            if (g_rdpdr_device[device].device_type != DEVICE_TYPE_DISK)
668                            {
669                                    status = STATUS_INVALID_HANDLE;
670                                    break;
671                            }
672    
673                            in_uint32_le(s, info_level);
674    
675                            out.data = out.p = buffer;
676                            out.size = sizeof(buffer);
677                            /* FIXME: Perhaps consider actually *do*
678                               something here :-) */
679                            status = STATUS_SUCCESS;
680                            result = buffer_len = out.p - out.data;
681                            break;
682    
683                  default:                  default:
684                          unimpl("IRP major=0x%x minor=0x%x\n", major, minor);                          unimpl("IRP major=0x%x minor=0x%x\n", major, minor);
685                          break;                          break;
# Line 567  rdpdr_process_irp(STREAM s) Line 689  rdpdr_process_irp(STREAM s)
689          {          {
690                  rdpdr_send_completion(device, id, status, result, buffer, buffer_len);                  rdpdr_send_completion(device, id, status, result, buffer, buffer_len);
691          }          }
692          xfree(buffer);          if (buffer)
693                    xfree(buffer);
694            buffer = NULL;
695  }  }
696    
697  void  static void
698  rdpdr_send_clientcapabilty(void)  rdpdr_send_clientcapabilty(void)
699  {  {
700          uint8 magic[4] = "rDPC";          uint8 magic[4] = "rDPC";
# Line 687  rdpdr_init() Line 810  rdpdr_init()
810  void  void
811  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)
812  {  {
813          long select_timeout = 0;        // Timeout value to be used for select() (in millisecons).          uint32 select_timeout = 0;      // Timeout value to be used for select() (in millisecons).
814          struct async_iorequest *iorq;          struct async_iorequest *iorq;
815            char c;
816    
817          iorq = g_iorequest;          iorq = g_iorequest;
818          while (iorq != NULL)          while (iorq != NULL)
# Line 698  rdpdr_add_fds(int *n, fd_set * rfds, fd_ Line 822  rdpdr_add_fds(int *n, fd_set * rfds, fd_
822                          switch (iorq->major)                          switch (iorq->major)
823                          {                          {
824                                  case IRP_MJ_READ:                                  case IRP_MJ_READ:
825                                            /* Is this FD valid? FDs will
826                                               be invalid when
827                                               reconnecting. FIXME: Real
828                                               support for reconnects. */
829    
830                                            if ((read(iorq->fd, &c, 0) != 0) && (errno == EBADF))
831                                                    break;
832    
833                                          FD_SET(iorq->fd, rfds);                                          FD_SET(iorq->fd, rfds);
834                                            *n = MAX(*n, iorq->fd);
835    
836                                          // Check if io request timeout is smaller than current (but not 0).                                          // Check if io request timeout is smaller than current (but not 0).
837                                          if (iorq->timeout                                          if (iorq->timeout
# Line 713  rdpdr_add_fds(int *n, fd_set * rfds, fd_ Line 845  rdpdr_add_fds(int *n, fd_set * rfds, fd_
845                                                  tv->tv_usec = (select_timeout % 1000) * 1000;                                                  tv->tv_usec = (select_timeout % 1000) * 1000;
846                                                  *timeout = True;                                                  *timeout = True;
847                                          }                                          }
848    
849                                          break;                                          break;
850    
851                                  case IRP_MJ_WRITE:                                  case IRP_MJ_WRITE:
852                                            /* FD still valid? See above. */
853                                            if ((write(iorq->fd, &c, 0) != 0) && (errno == EBADF))
854                                                    break;
855    
856                                          FD_SET(iorq->fd, wfds);                                          FD_SET(iorq->fd, wfds);
857                                            *n = MAX(*n, iorq->fd);
858                                          break;                                          break;
859    
860                          }                          }
861                          *n = MAX(*n, iorq->fd);  
862                  }                  }
863    
864                  iorq = iorq->next;                  iorq = iorq->next;
865          }          }
866  }  }
867    
868    struct async_iorequest *
869    rdpdr_remove_iorequest(struct async_iorequest *prev, struct async_iorequest *iorq)
870    {
871            if (!iorq)
872                    return NULL;
873    
874            if (iorq->buffer)
875                    xfree(iorq->buffer);
876            if (prev)
877            {
878                    prev->next = iorq->next;
879                    xfree(iorq);
880                    iorq = prev->next;
881            }
882            else
883            {
884                    // Even if NULL
885                    g_iorequest = iorq->next;
886                    xfree(iorq);
887                    iorq = NULL;
888            }
889            return iorq;
890    }
891    
892  /* 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 */
893  void  void
# Line 766  rdpdr_check_fds(fd_set * rfds, fd_set * Line 927  rdpdr_check_fds(fd_set * rfds, fd_set *
927                                                  /* never read larger chunks than 8k - chances are that it will block */                                                  /* never read larger chunks than 8k - chances are that it will block */
928                                                  status = fns->read(iorq->fd,                                                  status = fns->read(iorq->fd,
929                                                                     iorq->buffer + iorq->partial_len,                                                                     iorq->buffer + iorq->partial_len,
930                                                                     req_size, 0, &result);                                                                     req_size, iorq->offset, &result);
                                                 iorq->partial_len += result;  
931    
932                                                    if (result > 0)
933                                                    {
934                                                            iorq->partial_len += result;
935                                                            iorq->offset += result;
936                                                    }
937  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
938                                                  DEBUG(("RDPDR: %d bytes of data read\n", result));                                                  DEBUG(("RDPDR: %d bytes of data read\n", result));
939  #endif  #endif
940                                                  /* only delete link if all data has been transfered */                                                  /* only delete link if all data has been transfered */
941                                                  /* or if result was 0 and status success - EOF      */                                                  /* or if result was 0 and status success - EOF      */
942                                                  if ((iorq->partial_len == iorq->length) ||                                                  if ((iorq->partial_len == iorq->length) ||
943                                                      (result == 0))                                                      (g_rdpdr_device[iorq->device].device_type ==
944                                                         DEVICE_TYPE_SERIAL) || (result == 0))
945                                                  {                                                  {
946  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
947                                                          DEBUG(("RDPDR: AIO total %u bytes read of %u\n", iorq->partial_len, iorq->length));                                                          DEBUG(("RDPDR: AIO total %u bytes read of %u\n", iorq->partial_len, iorq->length));
948  #endif  #endif
                                                         /* send the data */  
                                                         status = STATUS_SUCCESS;  
949                                                          rdpdr_send_completion(iorq->device,                                                          rdpdr_send_completion(iorq->device,
950                                                                                iorq->id, status,                                                                                iorq->id, status,
951                                                                                iorq->partial_len,                                                                                iorq->partial_len,
952                                                                                iorq->buffer,                                                                                iorq->buffer,
953                                                                                iorq->partial_len);                                                                                iorq->partial_len);
954                                                          xfree(iorq->buffer);                                                          iorq = rdpdr_remove_iorequest(prev, iorq);
                                                         iorq->fd = 0;  
                                                         if (prev != NULL)  
                                                         {  
                                                                 prev->next = iorq->next;  
                                                                 xfree(iorq);  
                                                         }  
                                                         else  
                                                         {  
                                                                 // Even if NULL  
                                                                 g_iorequest = iorq->next;  
                                                                 xfree(iorq);  
                                                         }  
955                                                  }                                                  }
956                                          }                                          }
957                                          break;                                          break;
# Line 817  rdpdr_check_fds(fd_set * rfds, fd_set * Line 969  rdpdr_check_fds(fd_set * rfds, fd_set *
969                                                  /* never write larger chunks than 8k - chances are that it will block */                                                  /* never write larger chunks than 8k - chances are that it will block */
970                                                  status = fns->write(iorq->fd,                                                  status = fns->write(iorq->fd,
971                                                                      iorq->buffer +                                                                      iorq->buffer +
972                                                                      iorq->partial_len, req_size, 0,                                                                      iorq->partial_len, req_size,
973                                                                      &result);                                                                      iorq->offset, &result);
974                                                  iorq->partial_len += result;  
975                                                    if (result > 0)
976                                                    {
977                                                            iorq->partial_len += result;
978                                                            iorq->offset += result;
979                                                    }
980    
981  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
982                                                  DEBUG(("RDPDR: %d bytes of data written\n",                                                  DEBUG(("RDPDR: %d bytes of data written\n",
983                                                         result));                                                         result));
# Line 832  rdpdr_check_fds(fd_set * rfds, fd_set * Line 990  rdpdr_check_fds(fd_set * rfds, fd_set *
990  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
991                                                          DEBUG(("RDPDR: AIO total %u bytes written of %u\n", iorq->partial_len, iorq->length));                                                          DEBUG(("RDPDR: AIO total %u bytes written of %u\n", iorq->partial_len, iorq->length));
992  #endif  #endif
                                                         /* send a status success */  
                                                         status = STATUS_SUCCESS;  
993                                                          rdpdr_send_completion(iorq->device,                                                          rdpdr_send_completion(iorq->device,
994                                                                                iorq->id, status,                                                                                iorq->id, status,
995                                                                                iorq->partial_len, (uint8*)"",                                                                                iorq->partial_len,
996                                                                                1);                                                                                (uint8 *) "", 1);
997    
998                                                          xfree(iorq->buffer);                                                          iorq = rdpdr_remove_iorequest(prev, iorq);
                                                         iorq->fd = 0;  
                                                         if (prev != NULL)  
                                                         {  
                                                                 prev->next = iorq->next;  
                                                                 xfree(iorq);  
                                                         }  
                                                         else  
                                                         {  
                                                                 // Even if NULL  
                                                                 g_iorequest = iorq->next;  
                                                                 xfree(iorq);  
                                                         }  
999                                                  }                                                  }
1000                                          }                                          }
1001                                          break;                                          break;
# Line 859  rdpdr_check_fds(fd_set * rfds, fd_set * Line 1003  rdpdr_check_fds(fd_set * rfds, fd_set *
1003    
1004                  }                  }
1005                  prev = iorq;                  prev = iorq;
1006                  iorq = iorq->next;                  if (iorq)
1007                            iorq = iorq->next;
1008          }          }
1009    
1010  }  }
# Line 881  rdpdr_abort_io(uint32 fd, uint32 major, Line 1026  rdpdr_abort_io(uint32 fd, uint32 major,
1026                  if ((iorq->fd == fd) && (major == 0 || iorq->major == major))                  if ((iorq->fd == fd) && (major == 0 || iorq->major == major))
1027                  {                  {
1028                          result = 0;                          result = 0;
1029                          rdpdr_send_completion(iorq->device, iorq->id, status, result, (uint8*)"", 1);                          rdpdr_send_completion(iorq->device, iorq->id, status, result, (uint8 *) "",
1030                          xfree(iorq->buffer);                                                1);
1031                          iorq->fd = 0;  
1032                          if (prev != NULL)                          iorq = rdpdr_remove_iorequest(prev, iorq);
                         {  
                                 prev->next = iorq->next;  
                                 xfree(iorq);  
                         }  
                         else  
                         {  
                                 // Even if NULL  
                                 g_iorequest = iorq->next;  
                                 xfree(iorq);  
                         }  
1033                          return True;                          return True;
1034                  }                  }
1035    
Legend:
Removed from v.602  
changed lines
  Added in v.776
  ViewVC Help
Powered by ViewVC 1.1.26