/[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 592 by n-ki, Fri Jan 30 14:10:32 2004 UTC revision 647 by astrand, Mon Apr 5 19:23:08 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 <dirent.h>             /* opendir, closedir, readdir */
5  #include <time.h>  #include <time.h>
6  #include "rdesktop.h"  #include "rdesktop.h"
7    
# Line 27  extern DEVICE_FNS serial_fns; Line 29  extern DEVICE_FNS serial_fns;
29  extern DEVICE_FNS printer_fns;  extern DEVICE_FNS printer_fns;
30  extern DEVICE_FNS parallel_fns;  extern DEVICE_FNS parallel_fns;
31  extern DEVICE_FNS disk_fns;  extern DEVICE_FNS disk_fns;
32    extern FILEINFO g_fileinfo[];
33    
34  static VCHANNEL *rdpdr_channel;  static VCHANNEL *rdpdr_channel;
35    
# Line 36  uint32 g_num_devices; Line 39  uint32 g_num_devices;
39    
40  /* Table with information about rdpdr devices */  /* Table with information about rdpdr devices */
41  RDPDR_DEVICE g_rdpdr_device[RDPDR_MAX_DEVICES];  RDPDR_DEVICE g_rdpdr_device[RDPDR_MAX_DEVICES];
42    char * g_rdpdr_clientname = NULL;
43    
44  /* 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 */
45  /* 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 49  struct async_iorequest Line 53  struct async_iorequest
53          DEVICE_FNS *fns;          DEVICE_FNS *fns;
54    
55          struct async_iorequest *next;   /* next element in list */          struct async_iorequest *next;   /* next element in list */
56  } g_iorequest;  };
57    
58    struct async_iorequest *g_iorequest;
59    
60  /* Return device_id for a given handle */  /* Return device_id for a given handle */
61  int  int
# Line 76  convert_to_unix_filename(char *filename) Line 82  convert_to_unix_filename(char *filename)
82          }          }
83  }  }
84    
85    BOOL
86    rdpdr_handle_ok(int device, int handle)
87    {
88            switch (g_rdpdr_device[device].device_type)
89            {
90                    case DEVICE_TYPE_PARALLEL:
91                    case DEVICE_TYPE_SERIAL:
92                    case DEVICE_TYPE_PRINTER:
93                    case DEVICE_TYPE_SCARD:
94                            if (g_rdpdr_device[device].handle != handle)
95                                    return False;
96                            break;
97                    case DEVICE_TYPE_DISK:
98                            if (g_fileinfo[handle].device_id != device)
99                                    return False;
100                            break;
101            }
102            return True;
103    }
104    
105  /* 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 */
106  BOOL  BOOL
107  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,
108                      DEVICE_FNS * fns, long total_timeout, long interval_timeout, uint8 * buffer)                      DEVICE_FNS * fns, uint32 total_timeout, uint32 interval_timeout, uint8 * buffer,
109                        uint32 offset)
110  {  {
111          struct async_iorequest *iorq;          struct async_iorequest *iorq;
112    
113          iorq = &g_iorequest;          if (g_iorequest == NULL)
114            {
115                    g_iorequest = (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));
116                    if (!g_iorequest)
117                            return False;
118                    g_iorequest->fd = 0;
119                    g_iorequest->next = NULL;
120            }
121    
122            iorq = g_iorequest;
123    
124          while (iorq->fd != 0)          while (iorq->fd != 0)
125          {          {
126                  // create new element if needed                  // create new element if needed
127                  if (iorq->next == NULL)                  if (iorq->next == NULL)
                         iorq->next =  
                                 (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));  
   
                 iorq = iorq->next;  
         }  
   
         /* first element is special since it doesn't get deleted */  
         /* don't want to get io out of order */  
         if (g_iorequest.fd == 0)  
         {  
                 iorq = &g_iorequest;  
                 /* look for first occurrence of fd */  
                 while (iorq->next != NULL)  
128                  {                  {
                         if (iorq->fd == file)  
                                 break;  
                         iorq = iorq->next;  
                 }  
                 /* if same create new link at end of chain instead */  
                 if (iorq->fd == file)  
                 {  
                         while (iorq->next != NULL)  
                                 iorq = iorq->next;  
129                          iorq->next =                          iorq->next =
130                                  (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));                                  (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));
131                          iorq = iorq->next;                          if (!iorq->next)
132                                    return False;
133                            iorq->next->fd = 0;
134                            iorq->next->next = NULL;
135                  }                  }
136                  else                  iorq = iorq->next;
                         iorq = &g_iorequest;    /* didn't find fs use first entry */  
137          }          }
   
138          iorq->device = device;          iorq->device = device;
139          iorq->fd = file;          iorq->fd = file;
140          iorq->id = id;          iorq->id = id;
# Line 129  add_async_iorequest(uint32 device, uint3 Line 145  add_async_iorequest(uint32 device, uint3
145          iorq->timeout = total_timeout;          iorq->timeout = total_timeout;
146          iorq->itv_timeout = interval_timeout;          iorq->itv_timeout = interval_timeout;
147          iorq->buffer = buffer;          iorq->buffer = buffer;
148            iorq->offset = offset;
149          return True;          return True;
150  }  }
151    
# Line 152  void Line 169  void
169  rdpdr_send_name(void)  rdpdr_send_name(void)
170  {  {
171          uint8 magic[4] = "rDNC";          uint8 magic[4] = "rDNC";
         uint32 hostlen = (strlen(hostname) + 1) * 2;  
172          STREAM s;          STREAM s;
173            uint32 hostlen;
174    
175            if (NULL == g_rdpdr_clientname) {
176              g_rdpdr_clientname = hostname;
177            }
178            hostlen = (strlen(g_rdpdr_clientname) + 1) * 2;
179    
180          s = channel_init(rdpdr_channel, 16 + hostlen);          s = channel_init(rdpdr_channel, 16 + hostlen);
181          out_uint8a(s, magic, 4);          out_uint8a(s, magic, 4);
# Line 161  rdpdr_send_name(void) Line 183  rdpdr_send_name(void)
183          out_uint16_le(s, 0x72);          out_uint16_le(s, 0x72);
184          out_uint32(s, 0);          out_uint32(s, 0);
185          out_uint32_le(s, hostlen);          out_uint32_le(s, hostlen);
186          rdp_out_unistr(s, hostname, hostlen - 2);          rdp_out_unistr(s, g_rdpdr_clientname, hostlen - 2);
187          s_mark_end(s);          s_mark_end(s);
188          channel_send(s, rdpdr_channel);          channel_send(s, rdpdr_channel);
189  }  }
# Line 233  rdpdr_send_available(void) Line 255  rdpdr_send_available(void)
255                                  rdp_out_unistr(s, printerinfo->printer, printerlen - 2);                                  rdp_out_unistr(s, printerinfo->printer, printerlen - 2);
256                                  out_uint8a(s, printerinfo->blob, bloblen);                                  out_uint8a(s, printerinfo->blob, bloblen);
257    
258                                  xfree(printerinfo->blob);       /* Blob is sent twice if reconnecting */                                  if (printerinfo->blob)
259                                            xfree(printerinfo->blob);       /* Blob is sent twice if reconnecting */
260                                  break;                                  break;
261                          default:                          default:
262                                  out_uint32(s, 0);                                  out_uint32(s, 0);
# Line 329  rdpdr_process_irp(STREAM s) Line 352  rdpdr_process_irp(STREAM s)
352    
353                  case DEVICE_TYPE_DISK:                  case DEVICE_TYPE_DISK:
354    
                         /*rw_blocking = False; */  
355                          fns = &disk_fns;                          fns = &disk_fns;
356                            rw_blocking = False;
357                          break;                          break;
358    
359                  case DEVICE_TYPE_SCARD:                  case DEVICE_TYPE_SCARD:
# Line 396  rdpdr_process_irp(STREAM s) Line 419  rdpdr_process_irp(STREAM s)
419  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
420                          DEBUG(("RDPDR IRP Read (length: %d, offset: %d)\n", length, offset));                          DEBUG(("RDPDR IRP Read (length: %d, offset: %d)\n", length, offset));
421  #endif  #endif
422                            if (!rdpdr_handle_ok(device, file))
423                            {
424                                    status = STATUS_INVALID_HANDLE;
425                                    break;
426                            }
427    
428                          if (rw_blocking)        // Complete read immediately                          if (rw_blocking)        // Complete read immediately
429                          {                          {
430                                  buffer = (uint8 *) xrealloc((void *) buffer, length);                                  buffer = (uint8 *) xrealloc((void *) buffer, length);
431                                    if (!buffer)
432                                    {
433                                            status = STATUS_CANCELLED;
434                                            break;
435                                    }
436                                  status = fns->read(file, buffer, length, offset, &result);                                  status = fns->read(file, buffer, length, offset, &result);
437                                  buffer_len = result;                                  buffer_len = result;
438                                  break;                                  break;
# Line 406  rdpdr_process_irp(STREAM s) Line 440  rdpdr_process_irp(STREAM s)
440    
441                          // Add request to table                          // Add request to table
442                          pst_buf = (uint8 *) xmalloc(length);                          pst_buf = (uint8 *) xmalloc(length);
443                            if (!pst_buf)
444                            {
445                                    status = STATUS_CANCELLED;
446                                    break;
447                            }
448                          serial_get_timeout(file, length, &total_timeout, &interval_timeout);                          serial_get_timeout(file, length, &total_timeout, &interval_timeout);
449                          if (add_async_iorequest                          if (add_async_iorequest
450                              (device, file, id, major, length, fns, total_timeout, interval_timeout,                              (device, file, id, major, length, fns, total_timeout, interval_timeout,
451                               pst_buf))                               pst_buf, offset))
452                          {                          {
453                                  status = STATUS_PENDING;                                  status = STATUS_PENDING;
454                                  break;                                  break;
# Line 433  rdpdr_process_irp(STREAM s) Line 472  rdpdr_process_irp(STREAM s)
472  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
473                          DEBUG(("RDPDR IRP Write (length: %d)\n", result));                          DEBUG(("RDPDR IRP Write (length: %d)\n", result));
474  #endif  #endif
475                            if (!rdpdr_handle_ok(device, file))
476                            {
477                                    status = STATUS_INVALID_HANDLE;
478                                    break;
479                            }
480    
481                          if (rw_blocking)        // Complete immediately                          if (rw_blocking)        // Complete immediately
482                          {                          {
483                                  status = fns->write(file, s->p, length, offset, &result);                                  status = fns->write(file, s->p, length, offset, &result);
# Line 441  rdpdr_process_irp(STREAM s) Line 486  rdpdr_process_irp(STREAM s)
486    
487                          // Add to table                          // Add to table
488                          pst_buf = (uint8 *) xmalloc(length);                          pst_buf = (uint8 *) xmalloc(length);
489                            if (!pst_buf)
490                            {
491                                    status = STATUS_CANCELLED;
492                                    break;
493                            }
494    
495                          in_uint8a(s, pst_buf, length);                          in_uint8a(s, pst_buf, length);
496    
497                          if (add_async_iorequest                          if (add_async_iorequest
498                              (device, file, id, major, length, fns, 0, 0, pst_buf))                              (device, file, id, major, length, fns, 0, 0, pst_buf, offset))
499                          {                          {
500                                  status = STATUS_PENDING;                                  status = STATUS_PENDING;
501                                  break;                                  break;
# Line 564  rdpdr_process_irp(STREAM s) Line 615  rdpdr_process_irp(STREAM s)
615                          in_uint8s(s, 0x14);                          in_uint8s(s, 0x14);
616    
617                          buffer = (uint8 *) xrealloc((void *) buffer, bytes_out + 0x14);                          buffer = (uint8 *) xrealloc((void *) buffer, bytes_out + 0x14);
618                            if (!buffer)
619                            {
620                                    status = STATUS_CANCELLED;
621                                    break;
622                            }
623    
624                          out.data = out.p = buffer;                          out.data = out.p = buffer;
625                          out.size = sizeof(buffer);                          out.size = sizeof(buffer);
626                          status = fns->device_control(file, request, s, &out);                          status = fns->device_control(file, request, s, &out);
# Line 579  rdpdr_process_irp(STREAM s) Line 636  rdpdr_process_irp(STREAM s)
636          {          {
637                  rdpdr_send_completion(device, id, status, result, buffer, buffer_len);                  rdpdr_send_completion(device, id, status, result, buffer, buffer_len);
638          }          }
639          xfree(buffer);          if (buffer)
640                    xfree(buffer);
641            buffer = NULL;
642  }  }
643    
644  void  void
# Line 699  rdpdr_init() Line 757  rdpdr_init()
757  void  void
758  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)
759  {  {
760          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).
761          struct async_iorequest *iorq;          struct async_iorequest *iorq;
762    
763          iorq = &g_iorequest;          iorq = g_iorequest;
764          while (iorq != NULL)          while (iorq != NULL)
765          {          {
766                  if (iorq->fd != 0)                  if (iorq->fd != 0)
# Line 739  rdpdr_add_fds(int *n, fd_set * rfds, fd_ Line 797  rdpdr_add_fds(int *n, fd_set * rfds, fd_
797          }          }
798  }  }
799    
800    struct async_iorequest *
801    rdpdr_remove_iorequest(struct async_iorequest *prev, struct async_iorequest *iorq)
802    {
803            if (!iorq)
804                    return NULL;
805    
806            if (iorq->buffer)
807                    xfree(iorq->buffer);
808            if (prev)
809            {
810                    prev->next = iorq->next;
811                    xfree(iorq);
812                    iorq = prev->next;
813            }
814            else
815            {
816                    // Even if NULL
817                    g_iorequest = iorq->next;
818                    xfree(iorq);
819                    iorq = NULL;
820            }
821            return iorq;
822    }
823    
824  /* 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 */
825  void  void
# Line 749  rdpdr_check_fds(fd_set * rfds, fd_set * Line 830  rdpdr_check_fds(fd_set * rfds, fd_set *
830          DEVICE_FNS *fns;          DEVICE_FNS *fns;
831          struct async_iorequest *iorq;          struct async_iorequest *iorq;
832          struct async_iorequest *prev;          struct async_iorequest *prev;
833            uint32 req_size = 0;
834    
835          if (timed_out)          if (timed_out)
836          {          {
# Line 756  rdpdr_check_fds(fd_set * rfds, fd_set * Line 838  rdpdr_check_fds(fd_set * rfds, fd_set *
838                  return;                  return;
839          }          }
840    
841          iorq = &g_iorequest;          iorq = g_iorequest;
842          prev = NULL;          prev = NULL;
843          while (iorq != NULL)          while (iorq != NULL)
844          {          {
   
845                  if (iorq->fd != 0)                  if (iorq->fd != 0)
846                  {                  {
847                          switch (iorq->major)                          switch (iorq->major)
# Line 770  rdpdr_check_fds(fd_set * rfds, fd_set * Line 851  rdpdr_check_fds(fd_set * rfds, fd_set *
851                                          {                                          {
852                                                  /* Read the data */                                                  /* Read the data */
853                                                  fns = iorq->fns;                                                  fns = iorq->fns;
854    
855                                                    req_size =
856                                                            (iorq->length - iorq->partial_len) >
857                                                            8192 ? 8192 : (iorq->length -
858                                                                           iorq->partial_len);
859                                                    /* never read larger chunks than 8k - chances are that it will block */
860                                                  status = fns->read(iorq->fd,                                                  status = fns->read(iorq->fd,
861                                                                     iorq->buffer + iorq->partial_len,                                                                     iorq->buffer + iorq->partial_len,
862                                                                     iorq->length - iorq->partial_len,                                                                     req_size, iorq->offset, &result);
863                                                                     0, &result);  
864                                                  iorq->partial_len += result;                                                  if (result > 0)
865                                                    {
866                                                            iorq->partial_len += result;
867                                                            iorq->offset += result;
868                                                    }
869  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
870                                                  DEBUG(("RDPDR: %d bytes of data read\n", result));                                                  DEBUG(("RDPDR: %d bytes of data read\n", result));
871  #endif  #endif
872                                                  /* only delete link if all data has been transfered */                                                  /* only delete link if all data has been transfered */
873                                                  if (iorq->partial_len == iorq->length)                                                  /* or if result was 0 and status success - EOF      */
874                                                    if ((iorq->partial_len == iorq->length) ||
875                                                        (result == 0))
876                                                  {                                                  {
877                                                          /* send the data */  #if WITH_DEBUG_RDP5
878                                                          status = STATUS_SUCCESS;                                                          DEBUG(("RDPDR: AIO total %u bytes read of %u\n", iorq->partial_len, iorq->length));
879    #endif
880                                                          rdpdr_send_completion(iorq->device,                                                          rdpdr_send_completion(iorq->device,
881                                                                                iorq->id, status,                                                                                iorq->id, status,
882                                                                                iorq->length,                                                                                iorq->partial_len,
883                                                                                iorq->buffer, result);                                                                                iorq->buffer,
884                                                                                  iorq->partial_len);
885                                                          xfree(iorq->buffer);                                                          iorq = rdpdr_remove_iorequest(prev, iorq);
                                                         iorq->fd = 0;  
                                                         if (prev != NULL)  
                                                         {  
                                                                 prev->next = iorq->next;  
                                                                 xfree(iorq);  
                                                         }  
886                                                  }                                                  }
887                                          }                                          }
888                                          break;                                          break;
# Line 803  rdpdr_check_fds(fd_set * rfds, fd_set * Line 891  rdpdr_check_fds(fd_set * rfds, fd_set *
891                                          {                                          {
892                                                  /* Write data. */                                                  /* Write data. */
893                                                  fns = iorq->fns;                                                  fns = iorq->fns;
894    
895                                                    req_size =
896                                                            (iorq->length - iorq->partial_len) >
897                                                            8192 ? 8192 : (iorq->length -
898                                                                           iorq->partial_len);
899    
900                                                    /* never write larger chunks than 8k - chances are that it will block */
901                                                  status = fns->write(iorq->fd,                                                  status = fns->write(iorq->fd,
902                                                                      iorq->buffer +                                                                      iorq->buffer +
903                                                                      iorq->partial_len,                                                                      iorq->partial_len, req_size,
904                                                                      iorq->length -                                                                      iorq->offset, &result);
905                                                                      iorq->partial_len, 0, &result);  
906                                                  iorq->partial_len += result;                                                  if (result > 0)
907                                                    {
908                                                            iorq->partial_len += result;
909                                                            iorq->offset += result;
910                                                    }
911    
912  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
913                                                  DEBUG(("RDPDR: %d bytes of data written\n",                                                  DEBUG(("RDPDR: %d bytes of data written\n",
914                                                         result));                                                         result));
915  #endif  #endif
916                                                  /* only delete link if all data has been transfered */                                                  /* only delete link if all data has been transfered */
917                                                  if (iorq->partial_len == iorq->length)                                                  /* or we couldn't write */
918                                                    if ((iorq->partial_len == iorq->length)
919                                                        || (result == 0))
920                                                  {                                                  {
921                                                          /* send a status success */  #if WITH_DEBUG_RDP5
922                                                          status = STATUS_SUCCESS;                                                          DEBUG(("RDPDR: AIO total %u bytes written of %u\n", iorq->partial_len, iorq->length));
923    #endif
924                                                          rdpdr_send_completion(iorq->device,                                                          rdpdr_send_completion(iorq->device,
925                                                                                iorq->id, status,                                                                                iorq->id, status,
926                                                                                iorq->length, "", 1);                                                                                iorq->partial_len,
927                                                                                  (uint8 *) "", 1);
928    
929                                                          xfree(iorq->buffer);                                                          iorq = rdpdr_remove_iorequest(prev, iorq);
                                                         iorq->fd = 0;  
                                                         if (prev != NULL)  
                                                         {  
                                                                 prev->next = iorq->next;  
                                                                 xfree(iorq);  
                                                         }  
930                                                  }                                                  }
931                                          }                                          }
932                                          break;                                          break;
# Line 836  rdpdr_check_fds(fd_set * rfds, fd_set * Line 934  rdpdr_check_fds(fd_set * rfds, fd_set *
934    
935                  }                  }
936                  prev = iorq;                  prev = iorq;
937                  iorq = iorq->next;                  if (iorq)
938                            iorq = iorq->next;
939          }          }
940    
941  }  }
# Line 849  rdpdr_abort_io(uint32 fd, uint32 major, Line 948  rdpdr_abort_io(uint32 fd, uint32 major,
948          struct async_iorequest *iorq;          struct async_iorequest *iorq;
949          struct async_iorequest *prev;          struct async_iorequest *prev;
950    
951          iorq = &g_iorequest;          iorq = g_iorequest;
952          prev = NULL;          prev = NULL;
953          while (iorq != NULL)          while (iorq != NULL)
954          {          {
# Line 858  rdpdr_abort_io(uint32 fd, uint32 major, Line 957  rdpdr_abort_io(uint32 fd, uint32 major,
957                  if ((iorq->fd == fd) && (major == 0 || iorq->major == major))                  if ((iorq->fd == fd) && (major == 0 || iorq->major == major))
958                  {                  {
959                          result = 0;                          result = 0;
960                          rdpdr_send_completion(iorq->device, iorq->id, status, result, "", 1);                          rdpdr_send_completion(iorq->device, iorq->id, status, result, (uint8 *) "",
961                          xfree(iorq->buffer);                                                1);
962                          iorq->fd = 0;  
963                          if (prev != NULL)                          iorq = rdpdr_remove_iorequest(prev, iorq);
                         {  
                                 prev->next = iorq->next;  
                                 xfree(iorq);  
                         }  
964                          return True;                          return True;
965                  }                  }
966    
Legend:
Removed from v.592  
changed lines
  Added in v.647
  ViewVC Help
Powered by ViewVC 1.1.26