/[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 595 by n-ki, Tue Feb 3 14:02:59 2004 UTC revision 657 by astrand, Fri Apr 16 11:28:34 2004 UTC
# Line 1  Line 1 
1    /* -*- c-basic-offset: 8 -*-
2       rdesktop: A Remote Desktop Protocol client.
3       Copyright (C) Matthew Chapman 1999-2004
4    
5       This program is free software; you can redistribute it and/or modify
6       it under the terms of the GNU General Public License as published by
7       the Free Software Foundation; either version 2 of the License, or
8       (at your option) any later version.
9    
10       This program is distributed in the hope that it will be useful,
11       but WITHOUT ANY WARRANTY; without even the implied warranty of
12       MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13       GNU General Public License for more details.
14    
15       You should have received a copy of the GNU General Public License
16       along with this program; if not, write to the Free Software
17       Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
18    */
19    
20    /*
21      Here are some resources, for your IRP hacking pleasure:
22    
23      http://cvs.sourceforge.net/viewcvs.py/mingw/w32api/include/ddk/winddk.h?view=markup
24    
25      http://win32.mvps.org/ntfs/streams.cpp
26    
27      http://www.acc.umu.se/~bosse/ntifs.h
28    
29      http://undocumented.ntinternals.net/UserMode/Undocumented%20Functions/NT%20Objects/File/
30    
31      http://us1.samba.org/samba/ftp/specs/smb-nt01.txt
32    
33      http://www.osronline.com/
34    */
35    
36  #include <unistd.h>  #include <unistd.h>
37  #include <sys/types.h>  #include <sys/types.h>
38    #include <sys/time.h>
39    #include <dirent.h>             /* opendir, closedir, readdir */
40  #include <time.h>  #include <time.h>
41  #include "rdesktop.h"  #include "rdesktop.h"
42    
 #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  
   
43  #define IRP_MJ_CREATE                   0x00  #define IRP_MJ_CREATE                   0x00
44  #define IRP_MJ_CLOSE                    0x02  #define IRP_MJ_CLOSE                    0x02
45  #define IRP_MJ_READ                     0x03  #define IRP_MJ_READ                     0x03
# Line 18  Line 49 
49  #define IRP_MJ_QUERY_VOLUME_INFORMATION 0x0a  #define IRP_MJ_QUERY_VOLUME_INFORMATION 0x0a
50  #define IRP_MJ_DIRECTORY_CONTROL        0x0c  #define IRP_MJ_DIRECTORY_CONTROL        0x0c
51  #define IRP_MJ_DEVICE_CONTROL           0x0e  #define IRP_MJ_DEVICE_CONTROL           0x0e
52    #define IRP_MJ_LOCK_CONTROL             0x11
53    
54  #define IRP_MN_QUERY_DIRECTORY          0x01  #define IRP_MN_QUERY_DIRECTORY          0x01
55  #define IRP_MN_NOTIFY_CHANGE_DIRECTORY  0x02  #define IRP_MN_NOTIFY_CHANGE_DIRECTORY  0x02
# Line 27  extern DEVICE_FNS serial_fns; Line 59  extern DEVICE_FNS serial_fns;
59  extern DEVICE_FNS printer_fns;  extern DEVICE_FNS printer_fns;
60  extern DEVICE_FNS parallel_fns;  extern DEVICE_FNS parallel_fns;
61  extern DEVICE_FNS disk_fns;  extern DEVICE_FNS disk_fns;
62    extern FILEINFO g_fileinfo[];
63    
64  static VCHANNEL *rdpdr_channel;  static VCHANNEL *rdpdr_channel;
65    
# Line 36  uint32 g_num_devices; Line 69  uint32 g_num_devices;
69    
70  /* Table with information about rdpdr devices */  /* Table with information about rdpdr devices */
71  RDPDR_DEVICE g_rdpdr_device[RDPDR_MAX_DEVICES];  RDPDR_DEVICE g_rdpdr_device[RDPDR_MAX_DEVICES];
72    char *g_rdpdr_clientname = NULL;
73    
74  /* 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 */
75  /* 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 112  convert_to_unix_filename(char *filename)
112          }          }
113  }  }
114    
115    BOOL
116    rdpdr_handle_ok(int device, int handle)
117    {
118            switch (g_rdpdr_device[device].device_type)
119            {
120                    case DEVICE_TYPE_PARALLEL:
121                    case DEVICE_TYPE_SERIAL:
122                    case DEVICE_TYPE_PRINTER:
123                    case DEVICE_TYPE_SCARD:
124                            if (g_rdpdr_device[device].handle != handle)
125                                    return False;
126                            break;
127                    case DEVICE_TYPE_DISK:
128                            if (g_fileinfo[handle].device_id != device)
129                                    return False;
130                            break;
131            }
132            return True;
133    }
134    
135  /* 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 */
136  BOOL  BOOL
137  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,
138                      DEVICE_FNS * fns, long total_timeout, long interval_timeout, uint8 * buffer)                      DEVICE_FNS * fns, uint32 total_timeout, uint32 interval_timeout, uint8 * buffer,
139                        uint32 offset)
140  {  {
141          struct async_iorequest *iorq;          struct async_iorequest *iorq;
142    
143          if (g_iorequest == NULL)          if (g_iorequest == NULL)
144          {          {
145                  g_iorequest = (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));                  g_iorequest = (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));
146                    if (!g_iorequest)
147                            return False;
148                  g_iorequest->fd = 0;                  g_iorequest->fd = 0;
149                  g_iorequest->next = NULL;                  g_iorequest->next = NULL;
150          }          }
# Line 101  add_async_iorequest(uint32 device, uint3 Line 158  add_async_iorequest(uint32 device, uint3
158                  {                  {
159                          iorq->next =                          iorq->next =
160                                  (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));                                  (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));
161                            if (!iorq->next)
162                                    return False;
163                          iorq->next->fd = 0;                          iorq->next->fd = 0;
164                          iorq->next->next = NULL;                          iorq->next->next = NULL;
165                  }                  }
# Line 116  add_async_iorequest(uint32 device, uint3 Line 175  add_async_iorequest(uint32 device, uint3
175          iorq->timeout = total_timeout;          iorq->timeout = total_timeout;
176          iorq->itv_timeout = interval_timeout;          iorq->itv_timeout = interval_timeout;
177          iorq->buffer = buffer;          iorq->buffer = buffer;
178            iorq->offset = offset;
179          return True;          return True;
180  }  }
181    
# Line 139  void Line 199  void
199  rdpdr_send_name(void)  rdpdr_send_name(void)
200  {  {
201          uint8 magic[4] = "rDNC";          uint8 magic[4] = "rDNC";
         uint32 hostlen = (strlen(hostname) + 1) * 2;  
202          STREAM s;          STREAM s;
203            uint32 hostlen;
204    
205            if (NULL == g_rdpdr_clientname)
206            {
207                    g_rdpdr_clientname = hostname;
208            }
209            hostlen = (strlen(g_rdpdr_clientname) + 1) * 2;
210    
211          s = channel_init(rdpdr_channel, 16 + hostlen);          s = channel_init(rdpdr_channel, 16 + hostlen);
212          out_uint8a(s, magic, 4);          out_uint8a(s, magic, 4);
# Line 148  rdpdr_send_name(void) Line 214  rdpdr_send_name(void)
214          out_uint16_le(s, 0x72);          out_uint16_le(s, 0x72);
215          out_uint32(s, 0);          out_uint32(s, 0);
216          out_uint32_le(s, hostlen);          out_uint32_le(s, hostlen);
217          rdp_out_unistr(s, hostname, hostlen - 2);          rdp_out_unistr(s, g_rdpdr_clientname, hostlen - 2);
218          s_mark_end(s);          s_mark_end(s);
219          channel_send(s, rdpdr_channel);          channel_send(s, rdpdr_channel);
220  }  }
# Line 220  rdpdr_send_available(void) Line 286  rdpdr_send_available(void)
286                                  rdp_out_unistr(s, printerinfo->printer, printerlen - 2);                                  rdp_out_unistr(s, printerinfo->printer, printerlen - 2);
287                                  out_uint8a(s, printerinfo->blob, bloblen);                                  out_uint8a(s, printerinfo->blob, bloblen);
288    
289                                  xfree(printerinfo->blob);       /* Blob is sent twice if reconnecting */                                  if (printerinfo->blob)
290                                            xfree(printerinfo->blob);       /* Blob is sent twice if reconnecting */
291                                  break;                                  break;
292                          default:                          default:
293                                  out_uint32(s, 0);                                  out_uint32(s, 0);
# Line 383  rdpdr_process_irp(STREAM s) Line 450  rdpdr_process_irp(STREAM s)
450  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
451                          DEBUG(("RDPDR IRP Read (length: %d, offset: %d)\n", length, offset));                          DEBUG(("RDPDR IRP Read (length: %d, offset: %d)\n", length, offset));
452  #endif  #endif
453                            if (!rdpdr_handle_ok(device, file))
454                            {
455                                    status = STATUS_INVALID_HANDLE;
456                                    break;
457                            }
458    
459                          if (rw_blocking)        // Complete read immediately                          if (rw_blocking)        // Complete read immediately
460                          {                          {
461                                  buffer = (uint8 *) xrealloc((void *) buffer, length);                                  buffer = (uint8 *) xrealloc((void *) buffer, length);
462                                    if (!buffer)
463                                    {
464                                            status = STATUS_CANCELLED;
465                                            break;
466                                    }
467                                  status = fns->read(file, buffer, length, offset, &result);                                  status = fns->read(file, buffer, length, offset, &result);
468                                  buffer_len = result;                                  buffer_len = result;
469                                  break;                                  break;
# Line 393  rdpdr_process_irp(STREAM s) Line 471  rdpdr_process_irp(STREAM s)
471    
472                          // Add request to table                          // Add request to table
473                          pst_buf = (uint8 *) xmalloc(length);                          pst_buf = (uint8 *) xmalloc(length);
474                            if (!pst_buf)
475                            {
476                                    status = STATUS_CANCELLED;
477                                    break;
478                            }
479                          serial_get_timeout(file, length, &total_timeout, &interval_timeout);                          serial_get_timeout(file, length, &total_timeout, &interval_timeout);
480                          if (add_async_iorequest                          if (add_async_iorequest
481                              (device, file, id, major, length, fns, total_timeout, interval_timeout,                              (device, file, id, major, length, fns, total_timeout, interval_timeout,
482                               pst_buf))                               pst_buf, offset))
483                          {                          {
484                                  status = STATUS_PENDING;                                  status = STATUS_PENDING;
485                                  break;                                  break;
# Line 420  rdpdr_process_irp(STREAM s) Line 503  rdpdr_process_irp(STREAM s)
503  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
504                          DEBUG(("RDPDR IRP Write (length: %d)\n", result));                          DEBUG(("RDPDR IRP Write (length: %d)\n", result));
505  #endif  #endif
506                            if (!rdpdr_handle_ok(device, file))
507                            {
508                                    status = STATUS_INVALID_HANDLE;
509                                    break;
510                            }
511    
512                          if (rw_blocking)        // Complete immediately                          if (rw_blocking)        // Complete immediately
513                          {                          {
514                                  status = fns->write(file, s->p, length, offset, &result);                                  status = fns->write(file, s->p, length, offset, &result);
# Line 428  rdpdr_process_irp(STREAM s) Line 517  rdpdr_process_irp(STREAM s)
517    
518                          // Add to table                          // Add to table
519                          pst_buf = (uint8 *) xmalloc(length);                          pst_buf = (uint8 *) xmalloc(length);
520                            if (!pst_buf)
521                            {
522                                    status = STATUS_CANCELLED;
523                                    break;
524                            }
525    
526                          in_uint8a(s, pst_buf, length);                          in_uint8a(s, pst_buf, length);
527    
528                          if (add_async_iorequest                          if (add_async_iorequest
529                              (device, file, id, major, length, fns, 0, 0, pst_buf))                              (device, file, id, major, length, fns, 0, 0, pst_buf, offset))
530                          {                          {
531                                  status = STATUS_PENDING;                                  status = STATUS_PENDING;
532                                  break;                                  break;
# Line 551  rdpdr_process_irp(STREAM s) Line 646  rdpdr_process_irp(STREAM s)
646                          in_uint8s(s, 0x14);                          in_uint8s(s, 0x14);
647    
648                          buffer = (uint8 *) xrealloc((void *) buffer, bytes_out + 0x14);                          buffer = (uint8 *) xrealloc((void *) buffer, bytes_out + 0x14);
649                            if (!buffer)
650                            {
651                                    status = STATUS_CANCELLED;
652                                    break;
653                            }
654    
655                          out.data = out.p = buffer;                          out.data = out.p = buffer;
656                          out.size = sizeof(buffer);                          out.size = sizeof(buffer);
657                          status = fns->device_control(file, request, s, &out);                          status = fns->device_control(file, request, s, &out);
# Line 566  rdpdr_process_irp(STREAM s) Line 667  rdpdr_process_irp(STREAM s)
667          {          {
668                  rdpdr_send_completion(device, id, status, result, buffer, buffer_len);                  rdpdr_send_completion(device, id, status, result, buffer, buffer_len);
669          }          }
670          xfree(buffer);          if (buffer)
671                    xfree(buffer);
672            buffer = NULL;
673  }  }
674    
675  void  void
# Line 686  rdpdr_init() Line 788  rdpdr_init()
788  void  void
789  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)
790  {  {
791          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).
792          struct async_iorequest *iorq;          struct async_iorequest *iorq;
793    
794          iorq = g_iorequest;          iorq = g_iorequest;
# Line 726  rdpdr_add_fds(int *n, fd_set * rfds, fd_ Line 828  rdpdr_add_fds(int *n, fd_set * rfds, fd_
828          }          }
829  }  }
830    
831    struct async_iorequest *
832    rdpdr_remove_iorequest(struct async_iorequest *prev, struct async_iorequest *iorq)
833    {
834            if (!iorq)
835                    return NULL;
836    
837            if (iorq->buffer)
838                    xfree(iorq->buffer);
839            if (prev)
840            {
841                    prev->next = iorq->next;
842                    xfree(iorq);
843                    iorq = prev->next;
844            }
845            else
846            {
847                    // Even if NULL
848                    g_iorequest = iorq->next;
849                    xfree(iorq);
850                    iorq = NULL;
851            }
852            return iorq;
853    }
854    
855  /* 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 */
856  void  void
# Line 765  rdpdr_check_fds(fd_set * rfds, fd_set * Line 890  rdpdr_check_fds(fd_set * rfds, fd_set *
890                                                  /* never read larger chunks than 8k - chances are that it will block */                                                  /* never read larger chunks than 8k - chances are that it will block */
891                                                  status = fns->read(iorq->fd,                                                  status = fns->read(iorq->fd,
892                                                                     iorq->buffer + iorq->partial_len,                                                                     iorq->buffer + iorq->partial_len,
893                                                                     req_size, 0, &result);                                                                     req_size, iorq->offset, &result);
                                                 iorq->partial_len += result;  
894    
895                                                    if (result > 0)
896                                                    {
897                                                            iorq->partial_len += result;
898                                                            iorq->offset += result;
899                                                    }
900  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
901                                                  DEBUG(("RDPDR: %d bytes of data read\n", result));                                                  DEBUG(("RDPDR: %d bytes of data read\n", result));
902  #endif  #endif
# Line 779  rdpdr_check_fds(fd_set * rfds, fd_set * Line 908  rdpdr_check_fds(fd_set * rfds, fd_set *
908  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
909                                                          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));
910  #endif  #endif
                                                         /* send the data */  
                                                         status = STATUS_SUCCESS;  
911                                                          rdpdr_send_completion(iorq->device,                                                          rdpdr_send_completion(iorq->device,
912                                                                                iorq->id, status,                                                                                iorq->id, status,
913                                                                                iorq->partial_len,                                                                                iorq->partial_len,
914                                                                                iorq->buffer,                                                                                iorq->buffer,
915                                                                                iorq->partial_len);                                                                                iorq->partial_len);
916                                                          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);  
                                                         }  
917                                                  }                                                  }
918                                          }                                          }
919                                          break;                                          break;
# Line 816  rdpdr_check_fds(fd_set * rfds, fd_set * Line 931  rdpdr_check_fds(fd_set * rfds, fd_set *
931                                                  /* never write larger chunks than 8k - chances are that it will block */                                                  /* never write larger chunks than 8k - chances are that it will block */
932                                                  status = fns->write(iorq->fd,                                                  status = fns->write(iorq->fd,
933                                                                      iorq->buffer +                                                                      iorq->buffer +
934                                                                      iorq->partial_len, req_size, 0,                                                                      iorq->partial_len, req_size,
935                                                                      &result);                                                                      iorq->offset, &result);
936                                                  iorq->partial_len += result;  
937                                                    if (result > 0)
938                                                    {
939                                                            iorq->partial_len += result;
940                                                            iorq->offset += result;
941                                                    }
942    
943  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
944                                                  DEBUG(("RDPDR: %d bytes of data written\n",                                                  DEBUG(("RDPDR: %d bytes of data written\n",
945                                                         result));                                                         result));
# Line 831  rdpdr_check_fds(fd_set * rfds, fd_set * Line 952  rdpdr_check_fds(fd_set * rfds, fd_set *
952  #if WITH_DEBUG_RDP5  #if WITH_DEBUG_RDP5
953                                                          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));
954  #endif  #endif
                                                         /* send a status success */  
                                                         status = STATUS_SUCCESS;  
955                                                          rdpdr_send_completion(iorq->device,                                                          rdpdr_send_completion(iorq->device,
956                                                                                iorq->id, status,                                                                                iorq->id, status,
957                                                                                iorq->partial_len, "",                                                                                iorq->partial_len,
958                                                                                1);                                                                                (uint8 *) "", 1);
959    
960                                                          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);  
                                                         }  
961                                                  }                                                  }
962                                          }                                          }
963                                          break;                                          break;
# Line 858  rdpdr_check_fds(fd_set * rfds, fd_set * Line 965  rdpdr_check_fds(fd_set * rfds, fd_set *
965    
966                  }                  }
967                  prev = iorq;                  prev = iorq;
968                  iorq = iorq->next;                  if (iorq)
969                            iorq = iorq->next;
970          }          }
971    
972  }  }
# Line 871  rdpdr_abort_io(uint32 fd, uint32 major, Line 979  rdpdr_abort_io(uint32 fd, uint32 major,
979          struct async_iorequest *iorq;          struct async_iorequest *iorq;
980          struct async_iorequest *prev;          struct async_iorequest *prev;
981    
982          iorq = &g_iorequest;          iorq = g_iorequest;
983          prev = NULL;          prev = NULL;
984          while (iorq != NULL)          while (iorq != NULL)
985          {          {
# Line 880  rdpdr_abort_io(uint32 fd, uint32 major, Line 988  rdpdr_abort_io(uint32 fd, uint32 major,
988                  if ((iorq->fd == fd) && (major == 0 || iorq->major == major))                  if ((iorq->fd == fd) && (major == 0 || iorq->major == major))
989                  {                  {
990                          result = 0;                          result = 0;
991                          rdpdr_send_completion(iorq->device, iorq->id, status, result, "", 1);                          rdpdr_send_completion(iorq->device, iorq->id, status, result, (uint8 *) "",
992                          xfree(iorq->buffer);                                                1);
993                          iorq->fd = 0;  
994                          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);  
                         }  
995                          return True;                          return True;
996                  }                  }
997    
Legend:
Removed from v.595  
changed lines
  Added in v.657
  ViewVC Help
Powered by ViewVC 1.1.26