/[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 612 by n-ki, Mon Feb 23 09:58:16 2004 UTC revision 646 by forsberg, Fri Apr 2 15:34:38 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>  #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 28  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 37  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 79  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, uint32 total_timeout, uint32 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    
# Line 121  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 144  void Line 169  void
169  rdpdr_send_name(void)  rdpdr_send_name(void)
170  {  {
171          uint8 magic[4] = "rDNC";          uint8 magic[4] = "rDNC";
172          uint32 hostlen = (strlen(hostname) + 1) * 2;          if (NULL == g_rdpdr_clientname) {
173              g_rdpdr_clientname = hostname;
174            }
175            uint32 hostlen = (strlen(g_rdpdr_clientname) + 1) * 2;
176          STREAM s;          STREAM s;
177    
178          s = channel_init(rdpdr_channel, 16 + hostlen);          s = channel_init(rdpdr_channel, 16 + hostlen);
# Line 153  rdpdr_send_name(void) Line 181  rdpdr_send_name(void)
181          out_uint16_le(s, 0x72);          out_uint16_le(s, 0x72);
182          out_uint32(s, 0);          out_uint32(s, 0);
183          out_uint32_le(s, hostlen);          out_uint32_le(s, hostlen);
184          rdp_out_unistr(s, hostname, hostlen - 2);          rdp_out_unistr(s, g_rdpdr_clientname, hostlen - 2);
185          s_mark_end(s);          s_mark_end(s);
186          channel_send(s, rdpdr_channel);          channel_send(s, rdpdr_channel);
187  }  }
# Line 389  rdpdr_process_irp(STREAM s) Line 417  rdpdr_process_irp(STREAM s)
417  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
418                          DEBUG(("RDPDR IRP Read (length: %d, offset: %d)\n", length, offset));                          DEBUG(("RDPDR IRP Read (length: %d, offset: %d)\n", length, offset));
419  #endif  #endif
420                            if (!rdpdr_handle_ok(device, file))
421                            {
422                                    status = STATUS_INVALID_HANDLE;
423                                    break;
424                            }
425    
426                          if (rw_blocking)        // Complete read immediately                          if (rw_blocking)        // Complete read immediately
427                          {                          {
428                                  buffer = (uint8 *) xrealloc((void *) buffer, length);                                  buffer = (uint8 *) xrealloc((void *) buffer, length);
# Line 412  rdpdr_process_irp(STREAM s) Line 446  rdpdr_process_irp(STREAM s)
446                          serial_get_timeout(file, length, &total_timeout, &interval_timeout);                          serial_get_timeout(file, length, &total_timeout, &interval_timeout);
447                          if (add_async_iorequest                          if (add_async_iorequest
448                              (device, file, id, major, length, fns, total_timeout, interval_timeout,                              (device, file, id, major, length, fns, total_timeout, interval_timeout,
449                               pst_buf))                               pst_buf, offset))
450                          {                          {
451                                  status = STATUS_PENDING;                                  status = STATUS_PENDING;
452                                  break;                                  break;
# Line 436  rdpdr_process_irp(STREAM s) Line 470  rdpdr_process_irp(STREAM s)
470  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
471                          DEBUG(("RDPDR IRP Write (length: %d)\n", result));                          DEBUG(("RDPDR IRP Write (length: %d)\n", result));
472  #endif  #endif
473                            if (!rdpdr_handle_ok(device, file))
474                            {
475                                    status = STATUS_INVALID_HANDLE;
476                                    break;
477                            }
478    
479                          if (rw_blocking)        // Complete immediately                          if (rw_blocking)        // Complete immediately
480                          {                          {
481                                  status = fns->write(file, s->p, length, offset, &result);                                  status = fns->write(file, s->p, length, offset, &result);
# Line 453  rdpdr_process_irp(STREAM s) Line 493  rdpdr_process_irp(STREAM s)
493                          in_uint8a(s, pst_buf, length);                          in_uint8a(s, pst_buf, length);
494    
495                          if (add_async_iorequest                          if (add_async_iorequest
496                              (device, file, id, major, length, fns, 0, 0, pst_buf))                              (device, file, id, major, length, fns, 0, 0, pst_buf, offset))
497                          {                          {
498                                  status = STATUS_PENDING;                                  status = STATUS_PENDING;
499                                  break;                                  break;
# Line 755  rdpdr_add_fds(int *n, fd_set * rfds, fd_ Line 795  rdpdr_add_fds(int *n, fd_set * rfds, fd_
795          }          }
796  }  }
797    
798    struct async_iorequest *
799    rdpdr_remove_iorequest(struct async_iorequest *prev, struct async_iorequest *iorq)
800    {
801            if (!iorq)
802                    return NULL;
803    
804            if (iorq->buffer)
805                    xfree(iorq->buffer);
806            if (prev)
807            {
808                    prev->next = iorq->next;
809                    xfree(iorq);
810                    iorq = prev->next;
811            }
812            else
813            {
814                    // Even if NULL
815                    g_iorequest = iorq->next;
816                    xfree(iorq);
817                    iorq = NULL;
818            }
819            return iorq;
820    }
821    
822  /* 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 */
823  void  void
# Line 794  rdpdr_check_fds(fd_set * rfds, fd_set * Line 857  rdpdr_check_fds(fd_set * rfds, fd_set *
857                                                  /* never read larger chunks than 8k - chances are that it will block */                                                  /* never read larger chunks than 8k - chances are that it will block */
858                                                  status = fns->read(iorq->fd,                                                  status = fns->read(iorq->fd,
859                                                                     iorq->buffer + iorq->partial_len,                                                                     iorq->buffer + iorq->partial_len,
860                                                                     req_size, 0, &result);                                                                     req_size, iorq->offset, &result);
                                                 iorq->partial_len += result;  
861    
862                                                    if (result > 0)
863                                                    {
864                                                            iorq->partial_len += result;
865                                                            iorq->offset += result;
866                                                    }
867  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
868                                                  DEBUG(("RDPDR: %d bytes of data read\n", result));                                                  DEBUG(("RDPDR: %d bytes of data read\n", result));
869  #endif  #endif
# Line 808  rdpdr_check_fds(fd_set * rfds, fd_set * Line 875  rdpdr_check_fds(fd_set * rfds, fd_set *
875  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
876                                                          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));
877  #endif  #endif
                                                         /* send the data */  
                                                         status = STATUS_SUCCESS;  
878                                                          rdpdr_send_completion(iorq->device,                                                          rdpdr_send_completion(iorq->device,
879                                                                                iorq->id, status,                                                                                iorq->id, status,
880                                                                                iorq->partial_len,                                                                                iorq->partial_len,
881                                                                                iorq->buffer,                                                                                iorq->buffer,
882                                                                                iorq->partial_len);                                                                                iorq->partial_len);
883                                                          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);  
                                                         }  
884                                                  }                                                  }
885                                          }                                          }
886                                          break;                                          break;
# Line 845  rdpdr_check_fds(fd_set * rfds, fd_set * Line 898  rdpdr_check_fds(fd_set * rfds, fd_set *
898                                                  /* never write larger chunks than 8k - chances are that it will block */                                                  /* never write larger chunks than 8k - chances are that it will block */
899                                                  status = fns->write(iorq->fd,                                                  status = fns->write(iorq->fd,
900                                                                      iorq->buffer +                                                                      iorq->buffer +
901                                                                      iorq->partial_len, req_size, 0,                                                                      iorq->partial_len, req_size,
902                                                                      &result);                                                                      iorq->offset, &result);
903                                                  iorq->partial_len += result;  
904                                                    if (result > 0)
905                                                    {
906                                                            iorq->partial_len += result;
907                                                            iorq->offset += result;
908                                                    }
909    
910  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
911                                                  DEBUG(("RDPDR: %d bytes of data written\n",                                                  DEBUG(("RDPDR: %d bytes of data written\n",
912                                                         result));                                                         result));
# Line 860  rdpdr_check_fds(fd_set * rfds, fd_set * Line 919  rdpdr_check_fds(fd_set * rfds, fd_set *
919  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
920                                                          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));
921  #endif  #endif
                                                         /* send a status success */  
                                                         status = STATUS_SUCCESS;  
922                                                          rdpdr_send_completion(iorq->device,                                                          rdpdr_send_completion(iorq->device,
923                                                                                iorq->id, status,                                                                                iorq->id, status,
924                                                                                iorq->partial_len,                                                                                iorq->partial_len,
925                                                                                (uint8 *) "", 1);                                                                                (uint8 *) "", 1);
926    
927                                                          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);  
                                                         }  
928                                                  }                                                  }
929                                          }                                          }
930                                          break;                                          break;
# Line 887  rdpdr_check_fds(fd_set * rfds, fd_set * Line 932  rdpdr_check_fds(fd_set * rfds, fd_set *
932    
933                  }                  }
934                  prev = iorq;                  prev = iorq;
935                  iorq = iorq->next;                  if (iorq)
936                            iorq = iorq->next;
937          }          }
938    
939  }  }
# Line 911  rdpdr_abort_io(uint32 fd, uint32 major, Line 957  rdpdr_abort_io(uint32 fd, uint32 major,
957                          result = 0;                          result = 0;
958                          rdpdr_send_completion(iorq->device, iorq->id, status, result, (uint8 *) "",                          rdpdr_send_completion(iorq->device, iorq->id, status, result, (uint8 *) "",
959                                                1);                                                1);
960                          xfree(iorq->buffer);  
961                          iorq->fd = 0;                          iorq = rdpdr_remove_iorequest(prev, iorq);
                         if (prev != NULL)  
                         {  
                                 prev->next = iorq->next;  
                                 xfree(iorq);  
                         }  
                         else  
                         {  
                                 // Even if NULL  
                                 g_iorequest = iorq->next;  
                                 xfree(iorq);  
                         }  
962                          return True;                          return True;
963                  }                  }
964    
Legend:
Removed from v.612  
changed lines
  Added in v.646
  ViewVC Help
Powered by ViewVC 1.1.26