/[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 651 by astrand, Thu Apr 15 20:12:42 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    
 #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  
   
8  #define IRP_MJ_CREATE                   0x00  #define IRP_MJ_CREATE                   0x00
9  #define IRP_MJ_CLOSE                    0x02  #define IRP_MJ_CLOSE                    0x02
10  #define IRP_MJ_READ                     0x03  #define IRP_MJ_READ                     0x03
# Line 18  Line 14 
14  #define IRP_MJ_QUERY_VOLUME_INFORMATION 0x0a  #define IRP_MJ_QUERY_VOLUME_INFORMATION 0x0a
15  #define IRP_MJ_DIRECTORY_CONTROL        0x0c  #define IRP_MJ_DIRECTORY_CONTROL        0x0c
16  #define IRP_MJ_DEVICE_CONTROL           0x0e  #define IRP_MJ_DEVICE_CONTROL           0x0e
17    #define IRP_MJ_LOCK_CONTROL             0x11
18    
19  #define IRP_MN_QUERY_DIRECTORY          0x01  #define IRP_MN_QUERY_DIRECTORY          0x01
20  #define IRP_MN_NOTIFY_CHANGE_DIRECTORY  0x02  #define IRP_MN_NOTIFY_CHANGE_DIRECTORY  0x02
# Line 27  extern DEVICE_FNS serial_fns; Line 24  extern DEVICE_FNS serial_fns;
24  extern DEVICE_FNS printer_fns;  extern DEVICE_FNS printer_fns;
25  extern DEVICE_FNS parallel_fns;  extern DEVICE_FNS parallel_fns;
26  extern DEVICE_FNS disk_fns;  extern DEVICE_FNS disk_fns;
27    extern FILEINFO g_fileinfo[];
28    
29  static VCHANNEL *rdpdr_channel;  static VCHANNEL *rdpdr_channel;
30    
# Line 36  uint32 g_num_devices; Line 34  uint32 g_num_devices;
34    
35  /* Table with information about rdpdr devices */  /* Table with information about rdpdr devices */
36  RDPDR_DEVICE g_rdpdr_device[RDPDR_MAX_DEVICES];  RDPDR_DEVICE g_rdpdr_device[RDPDR_MAX_DEVICES];
37    char *g_rdpdr_clientname = NULL;
38    
39  /* 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 */
40  /* 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 78  convert_to_unix_filename(char *filename) Line 77  convert_to_unix_filename(char *filename)
77          }          }
78  }  }
79    
80    BOOL
81    rdpdr_handle_ok(int device, int handle)
82    {
83            switch (g_rdpdr_device[device].device_type)
84            {
85                    case DEVICE_TYPE_PARALLEL:
86                    case DEVICE_TYPE_SERIAL:
87                    case DEVICE_TYPE_PRINTER:
88                    case DEVICE_TYPE_SCARD:
89                            if (g_rdpdr_device[device].handle != handle)
90                                    return False;
91                            break;
92                    case DEVICE_TYPE_DISK:
93                            if (g_fileinfo[handle].device_id != device)
94                                    return False;
95                            break;
96            }
97            return True;
98    }
99    
100  /* 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 */
101  BOOL  BOOL
102  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,
103                      DEVICE_FNS * fns, long total_timeout, long interval_timeout, uint8 * buffer)                      DEVICE_FNS * fns, uint32 total_timeout, uint32 interval_timeout, uint8 * buffer,
104                        uint32 offset)
105  {  {
106          struct async_iorequest *iorq;          struct async_iorequest *iorq;
107    
108          if (g_iorequest == NULL)          if (g_iorequest == NULL)
109          {          {
110                  g_iorequest = (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));                  g_iorequest = (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));
111                    if (!g_iorequest)
112                            return False;
113                  g_iorequest->fd = 0;                  g_iorequest->fd = 0;
114                  g_iorequest->next = NULL;                  g_iorequest->next = NULL;
115          }          }
# Line 101  add_async_iorequest(uint32 device, uint3 Line 123  add_async_iorequest(uint32 device, uint3
123                  {                  {
124                          iorq->next =                          iorq->next =
125                                  (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));                                  (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));
126                            if (!iorq->next)
127                                    return False;
128                          iorq->next->fd = 0;                          iorq->next->fd = 0;
129                          iorq->next->next = NULL;                          iorq->next->next = NULL;
130                  }                  }
# Line 116  add_async_iorequest(uint32 device, uint3 Line 140  add_async_iorequest(uint32 device, uint3
140          iorq->timeout = total_timeout;          iorq->timeout = total_timeout;
141          iorq->itv_timeout = interval_timeout;          iorq->itv_timeout = interval_timeout;
142          iorq->buffer = buffer;          iorq->buffer = buffer;
143            iorq->offset = offset;
144          return True;          return True;
145  }  }
146    
# Line 139  void Line 164  void
164  rdpdr_send_name(void)  rdpdr_send_name(void)
165  {  {
166          uint8 magic[4] = "rDNC";          uint8 magic[4] = "rDNC";
         uint32 hostlen = (strlen(hostname) + 1) * 2;  
167          STREAM s;          STREAM s;
168            uint32 hostlen;
169    
170            if (NULL == g_rdpdr_clientname)
171            {
172                    g_rdpdr_clientname = hostname;
173            }
174            hostlen = (strlen(g_rdpdr_clientname) + 1) * 2;
175    
176          s = channel_init(rdpdr_channel, 16 + hostlen);          s = channel_init(rdpdr_channel, 16 + hostlen);
177          out_uint8a(s, magic, 4);          out_uint8a(s, magic, 4);
# Line 148  rdpdr_send_name(void) Line 179  rdpdr_send_name(void)
179          out_uint16_le(s, 0x72);          out_uint16_le(s, 0x72);
180          out_uint32(s, 0);          out_uint32(s, 0);
181          out_uint32_le(s, hostlen);          out_uint32_le(s, hostlen);
182          rdp_out_unistr(s, hostname, hostlen - 2);          rdp_out_unistr(s, g_rdpdr_clientname, hostlen - 2);
183          s_mark_end(s);          s_mark_end(s);
184          channel_send(s, rdpdr_channel);          channel_send(s, rdpdr_channel);
185  }  }
# Line 384  rdpdr_process_irp(STREAM s) Line 415  rdpdr_process_irp(STREAM s)
415  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
416                          DEBUG(("RDPDR IRP Read (length: %d, offset: %d)\n", length, offset));                          DEBUG(("RDPDR IRP Read (length: %d, offset: %d)\n", length, offset));
417  #endif  #endif
418                            if (!rdpdr_handle_ok(device, file))
419                            {
420                                    status = STATUS_INVALID_HANDLE;
421                                    break;
422                            }
423    
424                          if (rw_blocking)        // Complete read immediately                          if (rw_blocking)        // Complete read immediately
425                          {                          {
426                                  buffer = (uint8 *) xrealloc((void *) buffer, length);                                  buffer = (uint8 *) xrealloc((void *) buffer, length);
427                                    if (!buffer)
428                                    {
429                                            status = STATUS_CANCELLED;
430                                            break;
431                                    }
432                                  status = fns->read(file, buffer, length, offset, &result);                                  status = fns->read(file, buffer, length, offset, &result);
433                                  buffer_len = result;                                  buffer_len = result;
434                                  break;                                  break;
# Line 394  rdpdr_process_irp(STREAM s) Line 436  rdpdr_process_irp(STREAM s)
436    
437                          // Add request to table                          // Add request to table
438                          pst_buf = (uint8 *) xmalloc(length);                          pst_buf = (uint8 *) xmalloc(length);
439                            if (!pst_buf)
440                            {
441                                    status = STATUS_CANCELLED;
442                                    break;
443                            }
444                          serial_get_timeout(file, length, &total_timeout, &interval_timeout);                          serial_get_timeout(file, length, &total_timeout, &interval_timeout);
445                          if (add_async_iorequest                          if (add_async_iorequest
446                              (device, file, id, major, length, fns, total_timeout, interval_timeout,                              (device, file, id, major, length, fns, total_timeout, interval_timeout,
447                               pst_buf))                               pst_buf, offset))
448                          {                          {
449                                  status = STATUS_PENDING;                                  status = STATUS_PENDING;
450                                  break;                                  break;
# Line 421  rdpdr_process_irp(STREAM s) Line 468  rdpdr_process_irp(STREAM s)
468  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
469                          DEBUG(("RDPDR IRP Write (length: %d)\n", result));                          DEBUG(("RDPDR IRP Write (length: %d)\n", result));
470  #endif  #endif
471                            if (!rdpdr_handle_ok(device, file))
472                            {
473                                    status = STATUS_INVALID_HANDLE;
474                                    break;
475                            }
476    
477                          if (rw_blocking)        // Complete immediately                          if (rw_blocking)        // Complete immediately
478                          {                          {
479                                  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 482  rdpdr_process_irp(STREAM s)
482    
483                          // Add to table                          // Add to table
484                          pst_buf = (uint8 *) xmalloc(length);                          pst_buf = (uint8 *) xmalloc(length);
485                            if (!pst_buf)
486                            {
487                                    status = STATUS_CANCELLED;
488                                    break;
489                            }
490    
491                          in_uint8a(s, pst_buf, length);                          in_uint8a(s, pst_buf, length);
492    
493                          if (add_async_iorequest                          if (add_async_iorequest
494                              (device, file, id, major, length, fns, 0, 0, pst_buf))                              (device, file, id, major, length, fns, 0, 0, pst_buf, offset))
495                          {                          {
496                                  status = STATUS_PENDING;                                  status = STATUS_PENDING;
497                                  break;                                  break;
# Line 552  rdpdr_process_irp(STREAM s) Line 611  rdpdr_process_irp(STREAM s)
611                          in_uint8s(s, 0x14);                          in_uint8s(s, 0x14);
612    
613                          buffer = (uint8 *) xrealloc((void *) buffer, bytes_out + 0x14);                          buffer = (uint8 *) xrealloc((void *) buffer, bytes_out + 0x14);
614                            if (!buffer)
615                            {
616                                    status = STATUS_CANCELLED;
617                                    break;
618                            }
619    
620                          out.data = out.p = buffer;                          out.data = out.p = buffer;
621                          out.size = sizeof(buffer);                          out.size = sizeof(buffer);
622                          status = fns->device_control(file, request, s, &out);                          status = fns->device_control(file, request, s, &out);
# Line 567  rdpdr_process_irp(STREAM s) Line 632  rdpdr_process_irp(STREAM s)
632          {          {
633                  rdpdr_send_completion(device, id, status, result, buffer, buffer_len);                  rdpdr_send_completion(device, id, status, result, buffer, buffer_len);
634          }          }
635          xfree(buffer);          if (buffer)
636                    xfree(buffer);
637            buffer = NULL;
638  }  }
639    
640  void  void
# Line 687  rdpdr_init() Line 753  rdpdr_init()
753  void  void
754  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)
755  {  {
756          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).
757          struct async_iorequest *iorq;          struct async_iorequest *iorq;
758    
759          iorq = g_iorequest;          iorq = g_iorequest;
# Line 727  rdpdr_add_fds(int *n, fd_set * rfds, fd_ Line 793  rdpdr_add_fds(int *n, fd_set * rfds, fd_
793          }          }
794  }  }
795    
796    struct async_iorequest *
797    rdpdr_remove_iorequest(struct async_iorequest *prev, struct async_iorequest *iorq)
798    {
799            if (!iorq)
800                    return NULL;
801    
802            if (iorq->buffer)
803                    xfree(iorq->buffer);
804            if (prev)
805            {
806                    prev->next = iorq->next;
807                    xfree(iorq);
808                    iorq = prev->next;
809            }
810            else
811            {
812                    // Even if NULL
813                    g_iorequest = iorq->next;
814                    xfree(iorq);
815                    iorq = NULL;
816            }
817            return iorq;
818    }
819    
820  /* 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 */
821  void  void
# Line 766  rdpdr_check_fds(fd_set * rfds, fd_set * Line 855  rdpdr_check_fds(fd_set * rfds, fd_set *
855                                                  /* never read larger chunks than 8k - chances are that it will block */                                                  /* never read larger chunks than 8k - chances are that it will block */
856                                                  status = fns->read(iorq->fd,                                                  status = fns->read(iorq->fd,
857                                                                     iorq->buffer + iorq->partial_len,                                                                     iorq->buffer + iorq->partial_len,
858                                                                     req_size, 0, &result);                                                                     req_size, iorq->offset, &result);
                                                 iorq->partial_len += result;  
859    
860                                                    if (result > 0)
861                                                    {
862                                                            iorq->partial_len += result;
863                                                            iorq->offset += result;
864                                                    }
865  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
866                                                  DEBUG(("RDPDR: %d bytes of data read\n", result));                                                  DEBUG(("RDPDR: %d bytes of data read\n", result));
867  #endif  #endif
# Line 780  rdpdr_check_fds(fd_set * rfds, fd_set * Line 873  rdpdr_check_fds(fd_set * rfds, fd_set *
873  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
874                                                          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));
875  #endif  #endif
                                                         /* send the data */  
                                                         status = STATUS_SUCCESS;  
876                                                          rdpdr_send_completion(iorq->device,                                                          rdpdr_send_completion(iorq->device,
877                                                                                iorq->id, status,                                                                                iorq->id, status,
878                                                                                iorq->partial_len,                                                                                iorq->partial_len,
879                                                                                iorq->buffer,                                                                                iorq->buffer,
880                                                                                iorq->partial_len);                                                                                iorq->partial_len);
881                                                          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);  
                                                         }  
882                                                  }                                                  }
883                                          }                                          }
884                                          break;                                          break;
# Line 817  rdpdr_check_fds(fd_set * rfds, fd_set * Line 896  rdpdr_check_fds(fd_set * rfds, fd_set *
896                                                  /* never write larger chunks than 8k - chances are that it will block */                                                  /* never write larger chunks than 8k - chances are that it will block */
897                                                  status = fns->write(iorq->fd,                                                  status = fns->write(iorq->fd,
898                                                                      iorq->buffer +                                                                      iorq->buffer +
899                                                                      iorq->partial_len, req_size, 0,                                                                      iorq->partial_len, req_size,
900                                                                      &result);                                                                      iorq->offset, &result);
901                                                  iorq->partial_len += result;  
902                                                    if (result > 0)
903                                                    {
904                                                            iorq->partial_len += result;
905                                                            iorq->offset += result;
906                                                    }
907    
908  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
909                                                  DEBUG(("RDPDR: %d bytes of data written\n",                                                  DEBUG(("RDPDR: %d bytes of data written\n",
910                                                         result));                                                         result));
# Line 832  rdpdr_check_fds(fd_set * rfds, fd_set * Line 917  rdpdr_check_fds(fd_set * rfds, fd_set *
917  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
918                                                          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));
919  #endif  #endif
                                                         /* send a status success */  
                                                         status = STATUS_SUCCESS;  
920                                                          rdpdr_send_completion(iorq->device,                                                          rdpdr_send_completion(iorq->device,
921                                                                                iorq->id, status,                                                                                iorq->id, status,
922                                                                                iorq->partial_len, (uint8*)"",                                                                                iorq->partial_len,
923                                                                                1);                                                                                (uint8 *) "", 1);
924    
925                                                          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);  
                                                         }  
926                                                  }                                                  }
927                                          }                                          }
928                                          break;                                          break;
# Line 859  rdpdr_check_fds(fd_set * rfds, fd_set * Line 930  rdpdr_check_fds(fd_set * rfds, fd_set *
930    
931                  }                  }
932                  prev = iorq;                  prev = iorq;
933                  iorq = iorq->next;                  if (iorq)
934                            iorq = iorq->next;
935          }          }
936    
937  }  }
# Line 881  rdpdr_abort_io(uint32 fd, uint32 major, Line 953  rdpdr_abort_io(uint32 fd, uint32 major,
953                  if ((iorq->fd == fd) && (major == 0 || iorq->major == major))                  if ((iorq->fd == fd) && (major == 0 || iorq->major == major))
954                  {                  {
955                          result = 0;                          result = 0;
956                          rdpdr_send_completion(iorq->device, iorq->id, status, result, (uint8*)"", 1);                          rdpdr_send_completion(iorq->device, iorq->id, status, result, (uint8 *) "",
957                          xfree(iorq->buffer);                                                1);
958                          iorq->fd = 0;  
959                          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);  
                         }  
960                          return True;                          return True;
961                  }                  }
962    
Legend:
Removed from v.602  
changed lines
  Added in v.651
  ViewVC Help
Powered by ViewVC 1.1.26