/[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 613 by n-ki, Mon Feb 23 10:34:18 2004 UTC
# Line 1  Line 1 
1  #include <unistd.h>  #include <unistd.h>
2  #include <sys/types.h>  #include <sys/types.h>
3    #include <sys/time.h>
4  #include <time.h>  #include <time.h>
5  #include "rdesktop.h"  #include "rdesktop.h"
6    
# Line 22  Line 23 
23  #define IRP_MN_QUERY_DIRECTORY          0x01  #define IRP_MN_QUERY_DIRECTORY          0x01
24  #define IRP_MN_NOTIFY_CHANGE_DIRECTORY  0x02  #define IRP_MN_NOTIFY_CHANGE_DIRECTORY  0x02
25    
 #define MAX_ASYNC_IO_REQUESTS   10  
   
26  extern char hostname[16];  extern char hostname[16];
27  extern DEVICE_FNS serial_fns;  extern DEVICE_FNS serial_fns;
28  extern DEVICE_FNS printer_fns;  extern DEVICE_FNS printer_fns;
29  extern DEVICE_FNS parallel_fns;  extern DEVICE_FNS parallel_fns;
30  extern DEVICE_FNS disk_fns;  extern DEVICE_FNS disk_fns;
31    
   
32  static VCHANNEL *rdpdr_channel;  static VCHANNEL *rdpdr_channel;
33    
34  /* 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 */
# Line 41  uint32 g_num_devices; Line 39  uint32 g_num_devices;
39  RDPDR_DEVICE g_rdpdr_device[RDPDR_MAX_DEVICES];  RDPDR_DEVICE g_rdpdr_device[RDPDR_MAX_DEVICES];
40    
41  /* 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 */
42    /* using a linked list ensures that they are processed in the right order, */
43    /* if multiple ios are being done on the same fd */
44  struct async_iorequest  struct async_iorequest
45  {  {
46          uint32 fd, major, minor, offset, device, id, length;          uint32 fd, major, minor, offset, device, id, length, partial_len;
47          long timeout,           /* Total timeout */          long timeout,           /* Total timeout */
48            itv_timeout;          /* Interval timeout (between serial characters) */            itv_timeout;          /* Interval timeout (between serial characters) */
49          uint8 *buffer;          uint8 *buffer;
50          DEVICE_FNS *fns;          DEVICE_FNS *fns;
51  } g_iorequest[MAX_ASYNC_IO_REQUESTS];  
52            struct async_iorequest *next;   /* next element in list */
53    };
54    
55    struct async_iorequest *g_iorequest;
56    
57  /* Return device_id for a given handle */  /* Return device_id for a given handle */
58  int  int
# Line 78  convert_to_unix_filename(char *filename) Line 82  convert_to_unix_filename(char *filename)
82  /* 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 */
83  BOOL  BOOL
84  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,
85                      DEVICE_FNS * fns, long total_timeout, long interval_timeout, uint8 * buffer)                      DEVICE_FNS * fns, uint32 total_timeout, uint32 interval_timeout, uint8 * buffer,
86                        uint32 offset)
87  {  {
         int i;  
88          struct async_iorequest *iorq;          struct async_iorequest *iorq;
89    
90          for (i = 0; i < MAX_ASYNC_IO_REQUESTS; i++)          if (g_iorequest == NULL)
91          {          {
92                  iorq = &g_iorequest[i];                  g_iorequest = (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));
93                    if (!g_iorequest)
94                  if (iorq->fd == 0)                          return False;
95                  {                  g_iorequest->fd = 0;
96                          iorq->device = device;                  g_iorequest->next = NULL;
97                          iorq->fd = file;          }
98                          iorq->id = id;  
99                          iorq->major = major;          iorq = g_iorequest;
100                          iorq->length = length;  
101                          iorq->fns = fns;          while (iorq->fd != 0)
102                          iorq->timeout = total_timeout;          {
103                          iorq->itv_timeout = interval_timeout;                  // create new element if needed
104                          iorq->buffer = buffer;                  if (iorq->next == NULL)
105                          return True;                  {
106                  }                          iorq->next =
107          }                                  (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));
108          error("IO request table full. Increase MAX_ASYNC_IO_REQUESTS in rdpdr.c!\n");                          if (!iorq->next)
109          return False;                                  return False;
110                            iorq->next->fd = 0;
111                            iorq->next->next = NULL;
112                    }
113                    iorq = iorq->next;
114            }
115            iorq->device = device;
116            iorq->fd = file;
117            iorq->id = id;
118            iorq->major = major;
119            iorq->length = length;
120            iorq->partial_len = 0;
121            iorq->fns = fns;
122            iorq->timeout = total_timeout;
123            iorq->itv_timeout = interval_timeout;
124            iorq->buffer = buffer;
125            iorq->offset = offset;
126            return True;
127  }  }
128    
   
129  void  void
130  rdpdr_send_connect(void)  rdpdr_send_connect(void)
131  {  {
# Line 188  rdpdr_send_available(void) Line 208  rdpdr_send_available(void)
208                  out_uint32_le(s, i);    /* RDP Device ID */                  out_uint32_le(s, i);    /* RDP Device ID */
209                  out_uint8p(s, g_rdpdr_device[i].name, 8);                  out_uint8p(s, g_rdpdr_device[i].name, 8);
210    
211                  if (g_rdpdr_device[i].device_type == DEVICE_TYPE_PRINTER)                  switch (g_rdpdr_device[i].device_type)
212                  {                  {
213                          printerinfo = (PRINTER *) g_rdpdr_device[i].pdevice_data;                          case DEVICE_TYPE_PRINTER:
214                                    printerinfo = (PRINTER *) g_rdpdr_device[i].pdevice_data;
215    
216                          driverlen = 2 * strlen(printerinfo->driver) + 2;                                  driverlen = 2 * strlen(printerinfo->driver) + 2;
217                          printerlen = 2 * strlen(printerinfo->printer) + 2;                                  printerlen = 2 * strlen(printerinfo->printer) + 2;
218                          bloblen = printerinfo->bloblen;                                  bloblen = printerinfo->bloblen;
219    
220                          out_uint32_le(s, 24 + driverlen + printerlen + bloblen);        /* length of extra info */                                  out_uint32_le(s, 24 + driverlen + printerlen + bloblen);        /* length of extra info */
221                          out_uint32_le(s, printerinfo->default_printer ? 2 : 0);                                  out_uint32_le(s, printerinfo->default_printer ? 2 : 0);
222                          out_uint8s(s, 8);       /* unknown */                                  out_uint8s(s, 8);       /* unknown */
223                          out_uint32_le(s, driverlen);                                  out_uint32_le(s, driverlen);
224                          out_uint32_le(s, printerlen);                                  out_uint32_le(s, printerlen);
225                          out_uint32_le(s, bloblen);                                  out_uint32_le(s, bloblen);
226                          rdp_out_unistr(s, printerinfo->driver, driverlen - 2);                                  rdp_out_unistr(s, printerinfo->driver, driverlen - 2);
227                          rdp_out_unistr(s, printerinfo->printer, printerlen - 2);                                  rdp_out_unistr(s, printerinfo->printer, printerlen - 2);
228                          out_uint8a(s, printerinfo->blob, bloblen);                                  out_uint8a(s, printerinfo->blob, bloblen);
229    
230                          xfree(printerinfo->blob);       /* Blob is sent twice if reconnecting */                                  if (printerinfo->blob)
231                  }                                          xfree(printerinfo->blob);       /* Blob is sent twice if reconnecting */
232                  else                                  break;
233                  {                          default:
234                          out_uint32(s, 0);                                  out_uint32(s, 0);
235                  }                  }
236          }          }
237  #if 0  #if 0
# Line 282  rdpdr_process_irp(STREAM s) Line 303  rdpdr_process_irp(STREAM s)
303          buffer = (uint8 *) xmalloc(1024);          buffer = (uint8 *) xmalloc(1024);
304          buffer[0] = 0;          buffer[0] = 0;
305    
   
306          switch (g_rdpdr_device[device].device_type)          switch (g_rdpdr_device[device].device_type)
307          {          {
308                  case DEVICE_TYPE_SERIAL:                  case DEVICE_TYPE_SERIAL:
# Line 305  rdpdr_process_irp(STREAM s) Line 325  rdpdr_process_irp(STREAM s)
325                  case DEVICE_TYPE_DISK:                  case DEVICE_TYPE_DISK:
326    
327                          fns = &disk_fns;                          fns = &disk_fns;
328                            rw_blocking = False;
329                          break;                          break;
330    
331                  case DEVICE_TYPE_SCARD:                  case DEVICE_TYPE_SCARD:
# Line 373  rdpdr_process_irp(STREAM s) Line 394  rdpdr_process_irp(STREAM s)
394                          if (rw_blocking)        // Complete read immediately                          if (rw_blocking)        // Complete read immediately
395                          {                          {
396                                  buffer = (uint8 *) xrealloc((void *) buffer, length);                                  buffer = (uint8 *) xrealloc((void *) buffer, length);
397                                    if (!buffer)
398                                    {
399                                            status = STATUS_CANCELLED;
400                                            break;
401                                    }
402                                  status = fns->read(file, buffer, length, offset, &result);                                  status = fns->read(file, buffer, length, offset, &result);
403                                  buffer_len = result;                                  buffer_len = result;
404                                  break;                                  break;
# Line 380  rdpdr_process_irp(STREAM s) Line 406  rdpdr_process_irp(STREAM s)
406    
407                          // Add request to table                          // Add request to table
408                          pst_buf = (uint8 *) xmalloc(length);                          pst_buf = (uint8 *) xmalloc(length);
409                            if (!pst_buf)
410                            {
411                                    status = STATUS_CANCELLED;
412                                    break;
413                            }
414                          serial_get_timeout(file, length, &total_timeout, &interval_timeout);                          serial_get_timeout(file, length, &total_timeout, &interval_timeout);
415                          if (add_async_iorequest                          if (add_async_iorequest
416                              (device, file, id, major, length, fns, total_timeout, interval_timeout,                              (device, file, id, major, length, fns, total_timeout, interval_timeout,
417                               pst_buf))                               pst_buf, offset))
418                          {                          {
419                                  status = STATUS_PENDING;                                  status = STATUS_PENDING;
420                                  break;                                  break;
# Line 391  rdpdr_process_irp(STREAM s) Line 422  rdpdr_process_irp(STREAM s)
422    
423                          status = STATUS_CANCELLED;                          status = STATUS_CANCELLED;
424                          break;                          break;
   
425                  case IRP_MJ_WRITE:                  case IRP_MJ_WRITE:
426    
427                          buffer_len = 1;                          buffer_len = 1;
# Line 416  rdpdr_process_irp(STREAM s) Line 446  rdpdr_process_irp(STREAM s)
446    
447                          // Add to table                          // Add to table
448                          pst_buf = (uint8 *) xmalloc(length);                          pst_buf = (uint8 *) xmalloc(length);
449                            if (!pst_buf)
450                            {
451                                    status = STATUS_CANCELLED;
452                                    break;
453                            }
454    
455                          in_uint8a(s, pst_buf, length);                          in_uint8a(s, pst_buf, length);
456    
457                          if (add_async_iorequest                          if (add_async_iorequest
458                              (device, file, id, major, length, fns, 0, 0, pst_buf))                              (device, file, id, major, length, fns, 0, 0, pst_buf, offset))
459                          {                          {
460                                  status = STATUS_PENDING;                                  status = STATUS_PENDING;
461                                  break;                                  break;
# Line 539  rdpdr_process_irp(STREAM s) Line 575  rdpdr_process_irp(STREAM s)
575                          in_uint8s(s, 0x14);                          in_uint8s(s, 0x14);
576    
577                          buffer = (uint8 *) xrealloc((void *) buffer, bytes_out + 0x14);                          buffer = (uint8 *) xrealloc((void *) buffer, bytes_out + 0x14);
578                            if (!buffer)
579                            {
580                                    status = STATUS_CANCELLED;
581                                    break;
582                            }
583    
584                          out.data = out.p = buffer;                          out.data = out.p = buffer;
585                          out.size = sizeof(buffer);                          out.size = sizeof(buffer);
586                          status = fns->device_control(file, request, s, &out);                          status = fns->device_control(file, request, s, &out);
# Line 554  rdpdr_process_irp(STREAM s) Line 596  rdpdr_process_irp(STREAM s)
596          {          {
597                  rdpdr_send_completion(device, id, status, result, buffer, buffer_len);                  rdpdr_send_completion(device, id, status, result, buffer, buffer_len);
598          }          }
599          xfree(buffer);          if (buffer)
600                    xfree(buffer);
601            buffer = NULL;
602  }  }
603    
604  void  void
# Line 661  rdpdr_init() Line 704  rdpdr_init()
704  {  {
705          if (g_num_devices > 0)          if (g_num_devices > 0)
706          {          {
707                  rdpdr_channel = channel_register("rdpdr", CHANNEL_OPTION_INITIALIZED | CHANNEL_OPTION_COMPRESS_RDP, rdpdr_process);                  rdpdr_channel =
708                            channel_register("rdpdr",
709                                             CHANNEL_OPTION_INITIALIZED | CHANNEL_OPTION_COMPRESS_RDP,
710                                             rdpdr_process);
711          }          }
712    
713          return (rdpdr_channel != NULL);          return (rdpdr_channel != NULL);
# Line 671  rdpdr_init() Line 717  rdpdr_init()
717  void  void
718  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)
719  {  {
720          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).  
721          struct async_iorequest *iorq;          struct async_iorequest *iorq;
722    
723          for (i = 0; i < MAX_ASYNC_IO_REQUESTS; i++)          iorq = g_iorequest;
724            while (iorq != NULL)
725          {          {
726                  iorq = &g_iorequest[i];                  if (iorq->fd != 0)
   
                 if (iorq->fd != 0)      // Found a pending io request  
727                  {                  {
728                          switch (iorq->major)                          switch (iorq->major)
729                          {                          {
# Line 702  rdpdr_add_fds(int *n, fd_set * rfds, fd_ Line 746  rdpdr_add_fds(int *n, fd_set * rfds, fd_
746                                          break;                                          break;
747    
748                                  case IRP_MJ_WRITE:                                  case IRP_MJ_WRITE:
   
749                                          FD_SET(iorq->fd, wfds);                                          FD_SET(iorq->fd, wfds);
750                                          break;                                          break;
751    
752                          }                          }
753                          *n = MAX(*n, iorq->fd);                          *n = MAX(*n, iorq->fd);
754                  }                  }
755    
756                    iorq = iorq->next;
757          }          }
758  }  }
759    
760    
761  /* 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 */
762  void  void
763  rdpdr_check_fds(fd_set * rfds, fd_set * wfds, BOOL timed_out)  rdpdr_check_fds(fd_set * rfds, fd_set * wfds, BOOL timed_out)
764  {  {
         int i;  
765          NTSTATUS status;          NTSTATUS status;
766          uint32 result = 0, buffer_len = 0;          uint32 result = 0;
767          DEVICE_FNS *fns;          DEVICE_FNS *fns;
768          struct async_iorequest *iorq;          struct async_iorequest *iorq;
769            struct async_iorequest *prev;
770            uint32 req_size = 0;
771    
772          if (timed_out)          if (timed_out)
773          {          {
# Line 728  rdpdr_check_fds(fd_set * rfds, fd_set * Line 775  rdpdr_check_fds(fd_set * rfds, fd_set *
775                  return;                  return;
776          }          }
777    
778          // Walk through array of pending io_rq's          iorq = g_iorequest;
779          for (i = 0; i < MAX_ASYNC_IO_REQUESTS; i++)          prev = NULL;
780            while (iorq != NULL)
781          {          {
                 iorq = &g_iorequest[i];  
   
782                  if (iorq->fd != 0)                  if (iorq->fd != 0)
783                  {                  {
784                          switch (iorq->major)                          switch (iorq->major)
785                          {                          {
   
786                                  case IRP_MJ_READ:                                  case IRP_MJ_READ:
   
787                                          if (FD_ISSET(iorq->fd, rfds))                                          if (FD_ISSET(iorq->fd, rfds))
788                                          {                                          {
789                                                  // Read, and send data.                                                  /* Read the data */
790                                                  fns = iorq->fns;                                                  fns = iorq->fns;
791                                                  status = fns->read(iorq->fd, iorq->buffer,  
792                                                                     iorq->length, 0, &result);                                                  req_size =
793                                                  buffer_len = result;                                                          (iorq->length - iorq->partial_len) >
794                                                            8192 ? 8192 : (iorq->length -
795                                                  rdpdr_send_completion(iorq->device, iorq->id,                                                                         iorq->partial_len);
796                                                                        status, result, iorq->buffer,                                                  /* never read larger chunks than 8k - chances are that it will block */
797                                                                        buffer_len);                                                  status = fns->read(iorq->fd,
798                                                                       iorq->buffer + iorq->partial_len,
799                                                                       req_size, iorq->offset, &result);
800                                                    iorq->partial_len += result;
801                                                    iorq->offset += result;
802    
803  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
804                                                  DEBUG(("RDPDR: %d bytes of data read\n", result));                                                  DEBUG(("RDPDR: %d bytes of data read\n", result));
805  #endif  #endif
806                                                  xfree(iorq->buffer);                                                  /* only delete link if all data has been transfered */
807                                                  iorq->fd = 0;                                                  /* or if result was 0 and status success - EOF      */
808                                                    if ((iorq->partial_len == iorq->length) ||
809                                                        (result == 0))
810                                                    {
811    #if WITH_DEBUG_RDP5
812                                                            DEBUG(("RDPDR: AIO total %u bytes read of %u\n", iorq->partial_len, iorq->length));
813    #endif
814                                                            /* send the data */
815                                                            status = STATUS_SUCCESS;
816                                                            rdpdr_send_completion(iorq->device,
817                                                                                  iorq->id, status,
818                                                                                  iorq->partial_len,
819                                                                                  iorq->buffer,
820                                                                                  iorq->partial_len);
821                                                            xfree(iorq->buffer);
822                                                            iorq->fd = 0;
823                                                            if (prev != NULL)
824                                                            {
825                                                                    prev->next = iorq->next;
826                                                                    xfree(iorq);
827                                                            }
828                                                            else
829                                                            {
830                                                                    // Even if NULL
831                                                                    g_iorequest = iorq->next;
832                                                                    xfree(iorq);
833                                                            }
834                                                    }
835                                          }                                          }
836                                          break;                                          break;
   
837                                  case IRP_MJ_WRITE:                                  case IRP_MJ_WRITE:
   
838                                          if (FD_ISSET(iorq->fd, wfds))                                          if (FD_ISSET(iorq->fd, wfds))
839                                          {                                          {
840                                                  // Write data and send completion.                                                  /* Write data. */
841                                                  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);  
842    
843                                                  xfree(iorq->buffer);                                                  req_size =
844                                                  iorq->fd = 0;                                                          (iorq->length - iorq->partial_len) >
845                                                            8192 ? 8192 : (iorq->length -
846                                                                           iorq->partial_len);
847    
848                                                    /* never write larger chunks than 8k - chances are that it will block */
849                                                    status = fns->write(iorq->fd,
850                                                                        iorq->buffer +
851                                                                        iorq->partial_len, req_size,
852                                                                        iorq->offset, &result);
853                                                    iorq->partial_len += result;
854                                                    iorq->offset += result;
855    #if WITH_DEBUG_RDP5
856                                                    DEBUG(("RDPDR: %d bytes of data written\n",
857                                                           result));
858    #endif
859                                                    /* only delete link if all data has been transfered */
860                                                    /* or we couldn't write */
861                                                    if ((iorq->partial_len == iorq->length)
862                                                        || (result == 0))
863                                                    {
864    #if WITH_DEBUG_RDP5
865                                                            DEBUG(("RDPDR: AIO total %u bytes written of %u\n", iorq->partial_len, iorq->length));
866    #endif
867                                                            /* send a status success */
868                                                            status = STATUS_SUCCESS;
869                                                            rdpdr_send_completion(iorq->device,
870                                                                                  iorq->id, status,
871                                                                                  iorq->partial_len,
872                                                                                  (uint8 *) "", 1);
873    
874                                                            xfree(iorq->buffer);
875                                                            iorq->fd = 0;
876                                                            if (prev != NULL)
877                                                            {
878                                                                    prev->next = iorq->next;
879                                                                    xfree(iorq);
880                                                            }
881                                                            else
882                                                            {
883                                                                    // Even if NULL
884                                                                    g_iorequest = iorq->next;
885                                                                    xfree(iorq);
886                                                            }
887                                                    }
888                                          }                                          }
889                                          break;                                          break;
890                          }                          }
891    
892                  }                  }
893                    prev = iorq;
894                    iorq = iorq->next;
895          }          }
896    
897  }  }
898    
899  /* Abort a pending io request for a given handle and major */  /* Abort a pending io request for a given handle and major */
# Line 784  BOOL Line 901  BOOL
901  rdpdr_abort_io(uint32 fd, uint32 major, NTSTATUS status)  rdpdr_abort_io(uint32 fd, uint32 major, NTSTATUS status)
902  {  {
903          uint32 result;          uint32 result;
         int i;  
904          struct async_iorequest *iorq;          struct async_iorequest *iorq;
905            struct async_iorequest *prev;
906    
907          for (i = 0; i < MAX_ASYNC_IO_REQUESTS; i++)          iorq = g_iorequest;
908            prev = NULL;
909            while (iorq != NULL)
910          {          {
                 iorq = &g_iorequest[i];  
   
911                  // 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.
912                  // Abort read should not abort a write io request.                  // Abort read should not abort a write io request.
913                  if ((iorq->fd == fd) && (major == 0 || iorq->major == major))                  if ((iorq->fd == fd) && (major == 0 || iorq->major == major))
914                  {                  {
915                          result = 0;                          result = 0;
916                          rdpdr_send_completion(iorq->device, iorq->id, status, result, "", 1);                          rdpdr_send_completion(iorq->device, iorq->id, status, result, (uint8 *) "",
917                                                  1);
918                          xfree(iorq->buffer);                          xfree(iorq->buffer);
919                          iorq->fd = 0;                          iorq->fd = 0;
920                            if (prev != NULL)
921                            {
922                                    prev->next = iorq->next;
923                                    xfree(iorq);
924                            }
925                            else
926                            {
927                                    // Even if NULL
928                                    g_iorequest = iorq->next;
929                                    xfree(iorq);
930                            }
931                          return True;                          return True;
932                  }                  }
933    
934                    prev = iorq;
935                    iorq = iorq->next;
936          }          }
937    
938          return False;          return False;
939  }  }
Legend:
Removed from v.576  
changed lines
  Added in v.613
  ViewVC Help
Powered by ViewVC 1.1.26