/[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 435 by astrand, Wed Jul 9 09:18:20 2003 UTC revision 651 by astrand, Thu Apr 15 20:12:42 2004 UTC
# Line 1  Line 1 
1    #include <unistd.h>
2    #include <sys/types.h>
3    #include <sys/time.h>
4    #include <dirent.h>             /* opendir, closedir, readdir */
5    #include <time.h>
6  #include "rdesktop.h"  #include "rdesktop.h"
7    
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
11  #define IRP_MJ_WRITE            0x04  #define IRP_MJ_WRITE                    0x04
12  #define IRP_MJ_DEVICE_CONTROL   0x0e  #define IRP_MJ_QUERY_INFORMATION        0x05
13    #define IRP_MJ_SET_INFORMATION          0x06
14    #define IRP_MJ_QUERY_VOLUME_INFORMATION 0x0a
15    #define IRP_MJ_DIRECTORY_CONTROL        0x0c
16    #define IRP_MJ_DEVICE_CONTROL           0x0e
17    #define IRP_MJ_LOCK_CONTROL             0x11
18    
19    #define IRP_MN_QUERY_DIRECTORY          0x01
20    #define IRP_MN_NOTIFY_CHANGE_DIRECTORY  0x02
21    
22  extern char hostname[16];  extern char hostname[16];
23  extern DEVICE_FNS serial_fns;  extern DEVICE_FNS serial_fns;
24  extern DEVICE_FNS printer_fns;  extern DEVICE_FNS printer_fns;
25    extern DEVICE_FNS parallel_fns;
26    extern DEVICE_FNS disk_fns;
27    extern FILEINFO g_fileinfo[];
28    
29  static VCHANNEL *rdpdr_channel;  static VCHANNEL *rdpdr_channel;
30    
31    /* If select() times out, the request for the device with handle g_min_timeout_fd is aborted */
32    HANDLE g_min_timeout_fd;
33    uint32 g_num_devices;
34    
35    /* Table with information about rdpdr devices */
36    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 */
40    /* using a linked list ensures that they are processed in the right order, */
41    /* if multiple ios are being done on the same fd */
42    struct async_iorequest
43    {
44            uint32 fd, major, minor, offset, device, id, length, partial_len;
45            long timeout,           /* Total timeout */
46              itv_timeout;          /* Interval timeout (between serial characters) */
47            uint8 *buffer;
48            DEVICE_FNS *fns;
49    
50            struct async_iorequest *next;   /* next element in list */
51    };
52    
53    struct async_iorequest *g_iorequest;
54    
55    /* Return device_id for a given handle */
56    int
57    get_device_index(HANDLE handle)
58    {
59            int i;
60            for (i = 0; i < RDPDR_MAX_DEVICES; i++)
61            {
62                    if (g_rdpdr_device[i].handle == handle)
63                            return i;
64            }
65            return -1;
66    }
67    
68    /* Converts a windows path to a unix path */
69    void
70    convert_to_unix_filename(char *filename)
71    {
72            char *p;
73    
74            while ((p = strchr(filename, '\\')))
75            {
76                    *p = '/';
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 */
101    BOOL
102    add_async_iorequest(uint32 device, uint32 file, uint32 id, uint32 major, uint32 length,
103                        DEVICE_FNS * fns, uint32 total_timeout, uint32 interval_timeout, uint8 * buffer,
104                        uint32 offset)
105    {
106            struct async_iorequest *iorq;
107    
108            if (g_iorequest == NULL)
109            {
110                    g_iorequest = (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));
111                    if (!g_iorequest)
112                            return False;
113                    g_iorequest->fd = 0;
114                    g_iorequest->next = NULL;
115            }
116    
117            iorq = g_iorequest;
118    
119            while (iorq->fd != 0)
120            {
121                    // create new element if needed
122                    if (iorq->next == NULL)
123                    {
124                            iorq->next =
125                                    (struct async_iorequest *) xmalloc(sizeof(struct async_iorequest));
126                            if (!iorq->next)
127                                    return False;
128                            iorq->next->fd = 0;
129                            iorq->next->next = NULL;
130                    }
131                    iorq = iorq->next;
132            }
133            iorq->device = device;
134            iorq->fd = file;
135            iorq->id = id;
136            iorq->major = major;
137            iorq->length = length;
138            iorq->partial_len = 0;
139            iorq->fns = fns;
140            iorq->timeout = total_timeout;
141            iorq->itv_timeout = interval_timeout;
142            iorq->buffer = buffer;
143            iorq->offset = offset;
144            return True;
145    }
146    
147  void  void
148  rdpdr_send_connect(void)  rdpdr_send_connect(void)
149  {  {
# Line 27  rdpdr_send_connect(void) Line 159  rdpdr_send_connect(void)
159          channel_send(s, rdpdr_channel);          channel_send(s, rdpdr_channel);
160  }  }
161    
162    
163  void  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 40  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  }  }
186    
187    /* Returns the size of the payload of the announce packet */
188    int
189    announcedata_size()
190    {
191            int size, i;
192            PRINTER *printerinfo;
193    
194            size = 8;               //static announce size
195            size += g_num_devices * 0x14;
196    
197            for (i = 0; i < g_num_devices; i++)
198            {
199                    if (g_rdpdr_device[i].device_type == DEVICE_TYPE_PRINTER)
200                    {
201                            printerinfo = (PRINTER *) g_rdpdr_device[i].pdevice_data;
202                            printerinfo->bloblen =
203                                    printercache_load_blob(printerinfo->printer, &(printerinfo->blob));
204    
205                            size += 0x18;
206                            size += 2 * strlen(printerinfo->driver) + 2;
207                            size += 2 * strlen(printerinfo->printer) + 2;
208                            size += printerinfo->bloblen;
209                    }
210            }
211    
212            return size;
213    }
214    
215  void  void
216  rdpdr_send_available(void)  rdpdr_send_available(void)
217  {  {
218    
219          uint8 magic[4] = "rDAD";          uint8 magic[4] = "rDAD";
220          char *driver = "Digital turbo PrintServer 20";  /* Fairly generic PostScript driver */          uint32 driverlen, printerlen, bloblen;
221          char *printer = "PostScript";          int i;
         uint32 driverlen = (strlen(driver) + 1) * 2;  
         uint32 printerlen = (strlen(printer) + 1) * 2;  
222          STREAM s;          STREAM s;
223            PRINTER *printerinfo;
224    
225          s = channel_init(rdpdr_channel, 8 + 20);          s = channel_init(rdpdr_channel, announcedata_size());
226          out_uint8a(s, magic, 4);          out_uint8a(s, magic, 4);
227          out_uint32_le(s, 1);    /* Number of devices */          out_uint32_le(s, g_num_devices);
228    
229  #if 1          for (i = 0; i < g_num_devices; i++)
230          out_uint32_le(s, 0x1);  /* Device type 0x1 - serial */          {
231          out_uint32_le(s, 0);    /* Handle */                  out_uint32_le(s, g_rdpdr_device[i].device_type);
232          out_uint8p(s, "COM2", 4);                  out_uint32_le(s, i);    /* RDP Device ID */
233          out_uint8s(s, 4);       /* Pad to 8 */                  out_uint8p(s, g_rdpdr_device[i].name, 8);
234          out_uint32(s, 0);  
235  #endif                  switch (g_rdpdr_device[i].device_type)
236  #if 0                  {
237          out_uint32_le(s, 0x2);  /* Device type 0x2 - parallel */                          case DEVICE_TYPE_PRINTER:
238          out_uint32_le(s, 0);                                  printerinfo = (PRINTER *) g_rdpdr_device[i].pdevice_data;
239          out_uint8p(s, "LPT2", 4);  
240          out_uint8s(s, 4);                                  driverlen = 2 * strlen(printerinfo->driver) + 2;
241          out_uint32(s, 0);                                  printerlen = 2 * strlen(printerinfo->printer) + 2;
242  #endif                                  bloblen = printerinfo->bloblen;
243  #if 1  
244          out_uint32_le(s, 0x4);  /* Device type 0x4 - printer */                                  out_uint32_le(s, 24 + driverlen + printerlen + bloblen);        /* length of extra info */
245          out_uint32_le(s, 1);                                  out_uint32_le(s, printerinfo->default_printer ? 2 : 0);
246          out_uint8p(s, "PRN1", 4);                                  out_uint8s(s, 8);       /* unknown */
247          out_uint8s(s, 4);                                  out_uint32_le(s, driverlen);
248          out_uint32_le(s, 24 + driverlen + printerlen);  /* length of extra info */                                  out_uint32_le(s, printerlen);
249          out_uint32_le(s, 2);    /* unknown */                                  out_uint32_le(s, bloblen);
250          out_uint8s(s, 8);       /* unknown */                                  rdp_out_unistr(s, printerinfo->driver, driverlen - 2);
251          out_uint32_le(s, driverlen);    /* length of driver name */                                  rdp_out_unistr(s, printerinfo->printer, printerlen - 2);
252          out_uint32_le(s, printerlen);   /* length of printer name */                                  out_uint8a(s, printerinfo->blob, bloblen);
253          out_uint32(s, 0);       /* unknown */  
254          rdp_out_unistr(s, driver, driverlen - 2);                                  if (printerinfo->blob)
255          rdp_out_unistr(s, printer, printerlen - 2);                                          xfree(printerinfo->blob);       /* Blob is sent twice if reconnecting */
256  #endif                                  break;
257  #if 0                          default:
258          out_uint32_le(s, 0x8);  /* Device type 0x8 - disk */                                  out_uint32(s, 0);
259          out_uint32_le(s, 0);                  }
260          out_uint8p(s, "Z:", 2);          }
         out_uint8s(s, 6);  
         out_uint32(s, 0);  
 #endif  
261  #if 0  #if 0
262          out_uint32_le(s, 0x20); /* Device type 0x20 - smart card */          out_uint32_le(s, 0x20); /* Device type 0x20 - smart card */
263          out_uint32_le(s, 0);          out_uint32_le(s, 0);
# Line 121  rdpdr_send_completion(uint32 device, uin Line 285  rdpdr_send_completion(uint32 device, uin
285          out_uint32_le(s, result);          out_uint32_le(s, result);
286          out_uint8p(s, buffer, length);          out_uint8p(s, buffer, length);
287          s_mark_end(s);          s_mark_end(s);
288          hexdump(s->channel_hdr + 8, s->end - s->channel_hdr - 8);          /* JIF
289               hexdump(s->channel_hdr + 8, s->end - s->channel_hdr - 8); */
290          channel_send(s, rdpdr_channel);          channel_send(s, rdpdr_channel);
291  }  }
292    
293  static void  static void
294  rdpdr_process_irp(STREAM s)  rdpdr_process_irp(STREAM s)
295  {  {
296          uint32 device, file, id, major, minor;          uint32 result = 0,
297          NTSTATUS status = STATUS_INVALID_DEVICE_REQUEST;                  length = 0,
298          uint32 result = 0, length, request, bytes_in, bytes_out;                  desired_access = 0,
299          uint8 buffer[256];                  request,
300          uint32 buffer_len = 1;                  file,
301                    info_level,
302                    buffer_len,
303                    id,
304                    major,
305                    minor,
306                    device,
307                    offset,
308                    bytes_in,
309                    bytes_out,
310                    error_mode,
311                    share_mode, disposition, total_timeout, interval_timeout, flags_and_attributes = 0;
312    
313            char filename[256];
314            uint8 *buffer, *pst_buf;
315          struct stream out;          struct stream out;
316          DEVICE_FNS *fns;          DEVICE_FNS *fns;
317            BOOL rw_blocking = True;
318            NTSTATUS status = STATUS_INVALID_DEVICE_REQUEST;
319    
320          in_uint32_le(s, device);          in_uint32_le(s, device);
321          in_uint32_le(s, file);          in_uint32_le(s, file);
# Line 142  rdpdr_process_irp(STREAM s) Line 323  rdpdr_process_irp(STREAM s)
323          in_uint32_le(s, major);          in_uint32_le(s, major);
324          in_uint32_le(s, minor);          in_uint32_le(s, minor);
325    
326          memset(buffer, 0, sizeof(buffer));          buffer_len = 0;
327            buffer = (uint8 *) xmalloc(1024);
328            buffer[0] = 0;
329    
330          /* FIXME: this should probably be a more dynamic mapping */          switch (g_rdpdr_device[device].device_type)
         switch (device)  
331          {          {
332                  case 0:                  case DEVICE_TYPE_SERIAL:
333    
334                          fns = &serial_fns;                          fns = &serial_fns;
335                  case 1:                          rw_blocking = False;
336                            break;
337    
338                    case DEVICE_TYPE_PARALLEL:
339    
340                            fns = &parallel_fns;
341                            rw_blocking = False;
342                            break;
343    
344                    case DEVICE_TYPE_PRINTER:
345    
346                          fns = &printer_fns;                          fns = &printer_fns;
347                            break;
348    
349                    case DEVICE_TYPE_DISK:
350    
351                            fns = &disk_fns;
352                            rw_blocking = False;
353                            break;
354    
355                    case DEVICE_TYPE_SCARD:
356                  default:                  default:
357    
358                          error("IRP for bad device %ld\n", device);                          error("IRP for bad device %ld\n", device);
359                          return;                          return;
360          }          }
# Line 159  rdpdr_process_irp(STREAM s) Line 362  rdpdr_process_irp(STREAM s)
362          switch (major)          switch (major)
363          {          {
364                  case IRP_MJ_CREATE:                  case IRP_MJ_CREATE:
365                          if (fns->create)  
366                                  status = fns->create(&result);                          in_uint32_be(s, desired_access);
367                            in_uint8s(s, 0x08);     // unknown
368                            in_uint32_le(s, error_mode);
369                            in_uint32_le(s, share_mode);
370                            in_uint32_le(s, disposition);
371                            in_uint32_le(s, flags_and_attributes);
372                            in_uint32_le(s, length);
373    
374                            if (length && (length / 2) < 256)
375                            {
376                                    rdp_in_unistr(s, filename, length);
377                                    convert_to_unix_filename(filename);
378                            }
379                            else
380                            {
381                                    filename[0] = 0;
382                            }
383    
384                            if (!fns->create)
385                            {
386                                    status = STATUS_NOT_SUPPORTED;
387                                    break;
388                            }
389    
390                            status = fns->create(device, desired_access, share_mode, disposition,
391                                                 flags_and_attributes, filename, &result);
392                            buffer_len = 1;
393                          break;                          break;
394    
395                  case IRP_MJ_CLOSE:                  case IRP_MJ_CLOSE:
396                          if (fns->close)                          if (!fns->close)
397                                  status = fns->close(file);                          {
398                                    status = STATUS_NOT_SUPPORTED;
399                                    break;
400                            }
401    
402                            status = fns->close(file);
403                          break;                          break;
404    
405                  case IRP_MJ_READ:                  case IRP_MJ_READ:
406                          if (fns->read)  
407                            if (!fns->read)
408                          {                          {
409                                  if (length > sizeof(buffer))                                  status = STATUS_NOT_SUPPORTED;
410                                          length = sizeof(buffer);                                  break;
411                                  status = fns->read(file, buffer, length, &result);                          }
412    
413                            in_uint32_le(s, length);
414                            in_uint32_le(s, offset);
415    #if WITH_DEBUG_RDP5
416                            DEBUG(("RDPDR IRP Read (length: %d, offset: %d)\n", length, offset));
417    #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
425                            {
426                                    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);
433                                  buffer_len = result;                                  buffer_len = result;
434                                    break;
435                            }
436    
437                            // Add request to table
438                            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);
445                            if (add_async_iorequest
446                                (device, file, id, major, length, fns, total_timeout, interval_timeout,
447                                 pst_buf, offset))
448                            {
449                                    status = STATUS_PENDING;
450                                    break;
451                          }                          }
                         break;  
452    
453                            status = STATUS_CANCELLED;
454                            break;
455                  case IRP_MJ_WRITE:                  case IRP_MJ_WRITE:
456                          if (fns->write)  
457                                  status = fns->write(file, s->p, length, &result);                          buffer_len = 1;
458    
459                            if (!fns->write)
460                            {
461                                    status = STATUS_NOT_SUPPORTED;
462                                    break;
463                            }
464    
465                            in_uint32_le(s, length);
466                            in_uint32_le(s, offset);
467                            in_uint8s(s, 0x18);
468    #if WITH_DEBUG_RDP5
469                            DEBUG(("RDPDR IRP Write (length: %d)\n", result));
470    #endif
471                            if (!rdpdr_handle_ok(device, file))
472                            {
473                                    status = STATUS_INVALID_HANDLE;
474                                    break;
475                            }
476    
477                            if (rw_blocking)        // Complete immediately
478                            {
479                                    status = fns->write(file, s->p, length, offset, &result);
480                                    break;
481                            }
482    
483                            // Add to table
484                            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);
492    
493                            if (add_async_iorequest
494                                (device, file, id, major, length, fns, 0, 0, pst_buf, offset))
495                            {
496                                    status = STATUS_PENDING;
497                                    break;
498                            }
499    
500                            status = STATUS_CANCELLED;
501                            break;
502    
503                    case IRP_MJ_QUERY_INFORMATION:
504    
505                            if (g_rdpdr_device[device].device_type != DEVICE_TYPE_DISK)
506                            {
507                                    status = STATUS_INVALID_HANDLE;
508                                    break;
509                            }
510                            in_uint32_le(s, info_level);
511    
512                            out.data = out.p = buffer;
513                            out.size = sizeof(buffer);
514                            status = disk_query_information(file, info_level, &out);
515                            result = buffer_len = out.p - out.data;
516    
517                            break;
518    
519                    case IRP_MJ_SET_INFORMATION:
520    
521                            if (g_rdpdr_device[device].device_type != DEVICE_TYPE_DISK)
522                            {
523                                    status = STATUS_INVALID_HANDLE;
524                                    break;
525                            }
526    
527                            in_uint32_le(s, info_level);
528    
529                            out.data = out.p = buffer;
530                            out.size = sizeof(buffer);
531                            status = disk_set_information(file, info_level, s, &out);
532                            result = buffer_len = out.p - out.data;
533                            break;
534    
535                    case IRP_MJ_QUERY_VOLUME_INFORMATION:
536    
537                            if (g_rdpdr_device[device].device_type != DEVICE_TYPE_DISK)
538                            {
539                                    status = STATUS_INVALID_HANDLE;
540                                    break;
541                            }
542    
543                            in_uint32_le(s, info_level);
544    
545                            out.data = out.p = buffer;
546                            out.size = sizeof(buffer);
547                            status = disk_query_volume_information(file, info_level, &out);
548                            result = buffer_len = out.p - out.data;
549                            break;
550    
551                    case IRP_MJ_DIRECTORY_CONTROL:
552    
553                            if (g_rdpdr_device[device].device_type != DEVICE_TYPE_DISK)
554                            {
555                                    status = STATUS_INVALID_HANDLE;
556                                    break;
557                            }
558    
559                            switch (minor)
560                            {
561                                    case IRP_MN_QUERY_DIRECTORY:
562    
563                                            in_uint32_le(s, info_level);
564                                            in_uint8s(s, 1);
565                                            in_uint32_le(s, length);
566                                            in_uint8s(s, 0x17);
567                                            if (length && length < 2 * 255)
568                                            {
569                                                    rdp_in_unistr(s, filename, length);
570                                                    convert_to_unix_filename(filename);
571                                            }
572                                            else
573                                            {
574                                                    filename[0] = 0;
575                                            }
576                                            out.data = out.p = buffer;
577                                            out.size = sizeof(buffer);
578                                            status = disk_query_directory(file, info_level, filename,
579                                                                          &out);
580                                            result = buffer_len = out.p - out.data;
581                                            if (!buffer_len)
582                                                    buffer_len++;
583                                            break;
584    
585                                    case IRP_MN_NOTIFY_CHANGE_DIRECTORY:
586    
587                                            /* JIF
588                                               unimpl("IRP major=0x%x minor=0x%x: IRP_MN_NOTIFY_CHANGE_DIRECTORY\n", major, minor); */
589                                            status = STATUS_PENDING;        // Don't send completion packet
590                                            break;
591    
592                                    default:
593    
594                                            status = STATUS_INVALID_PARAMETER;
595                                            /* JIF
596                                               unimpl("IRP major=0x%x minor=0x%x\n", major, minor); */
597                            }
598                          break;                          break;
599    
600                  case IRP_MJ_DEVICE_CONTROL:                  case IRP_MJ_DEVICE_CONTROL:
601                          if (fns->device_control)  
602                            if (!fns->device_control)
603                            {
604                                    status = STATUS_NOT_SUPPORTED;
605                                    break;
606                            }
607    
608                            in_uint32_le(s, bytes_out);
609                            in_uint32_le(s, bytes_in);
610                            in_uint32_le(s, request);
611                            in_uint8s(s, 0x14);
612    
613                            buffer = (uint8 *) xrealloc((void *) buffer, bytes_out + 0x14);
614                            if (!buffer)
615                          {                          {
616                                  in_uint32_le(s, bytes_out);                                  status = STATUS_CANCELLED;
617                                  in_uint32_le(s, bytes_in);                                  break;
                                 in_uint32_le(s, request);  
                                 in_uint8s(s, 0x14);  
                                 out.data = out.p = buffer;  
                                 out.size = sizeof(buffer);  
                                 status = fns->device_control(file, request, s, &out);  
                                 result = buffer_len = out.p - out.data;  
618                          }                          }
619    
620                            out.data = out.p = buffer;
621                            out.size = sizeof(buffer);
622                            status = fns->device_control(file, request, s, &out);
623                            result = buffer_len = out.p - out.data;
624                          break;                          break;
625    
626                  default:                  default:
# Line 202  rdpdr_process_irp(STREAM s) Line 628  rdpdr_process_irp(STREAM s)
628                          break;                          break;
629          }          }
630    
631          rdpdr_send_completion(device, id, status, result, buffer, buffer_len);          if (status != STATUS_PENDING)
632            {
633                    rdpdr_send_completion(device, id, status, result, buffer, buffer_len);
634            }
635            if (buffer)
636                    xfree(buffer);
637            buffer = NULL;
638    }
639    
640    void
641    rdpdr_send_clientcapabilty(void)
642    {
643            uint8 magic[4] = "rDPC";
644            STREAM s;
645    
646            s = channel_init(rdpdr_channel, 0x50);
647            out_uint8a(s, magic, 4);
648            out_uint32_le(s, 5);    /* count */
649            out_uint16_le(s, 1);    /* first */
650            out_uint16_le(s, 0x28); /* length */
651            out_uint32_le(s, 1);
652            out_uint32_le(s, 2);
653            out_uint16_le(s, 2);
654            out_uint16_le(s, 5);
655            out_uint16_le(s, 1);
656            out_uint16_le(s, 5);
657            out_uint16_le(s, 0xFFFF);
658            out_uint16_le(s, 0);
659            out_uint32_le(s, 0);
660            out_uint32_le(s, 3);
661            out_uint32_le(s, 0);
662            out_uint32_le(s, 0);
663            out_uint16_le(s, 2);    /* second */
664            out_uint16_le(s, 8);    /* length */
665            out_uint32_le(s, 1);
666            out_uint16_le(s, 3);    /* third */
667            out_uint16_le(s, 8);    /* length */
668            out_uint32_le(s, 1);
669            out_uint16_le(s, 4);    /* fourth */
670            out_uint16_le(s, 8);    /* length */
671            out_uint32_le(s, 1);
672            out_uint16_le(s, 5);    /* fifth */
673            out_uint16_le(s, 8);    /* length */
674            out_uint32_le(s, 1);
675    
676            s_mark_end(s);
677            channel_send(s, rdpdr_channel);
678  }  }
679    
680  static void  static void
681  rdpdr_process(STREAM s)  rdpdr_process(STREAM s)
682  {  {
683          uint32 handle;          uint32 handle;
684          char *magic;          uint8 *magic;
685    
686          printf("rdpdr_process\n");  #if WITH_DEBUG_RDP5
687            printf("--- rdpdr_process ---\n");
688          hexdump(s->p, s->end - s->p);          hexdump(s->p, s->end - s->p);
689    #endif
690          in_uint8p(s, magic, 4);          in_uint8p(s, magic, 4);
691    
692          if ((magic[0] == 'r') && (magic[1] == 'D'))          if ((magic[0] == 'r') && (magic[1] == 'D'))
# Line 226  rdpdr_process(STREAM s) Line 700  rdpdr_process(STREAM s)
700                  {                  {
701                          rdpdr_send_connect();                          rdpdr_send_connect();
702                          rdpdr_send_name();                          rdpdr_send_name();
                         rdpdr_send_available();  
703                          return;                          return;
704                  }                  }
705                  else if ((magic[2] == 'C') && (magic[3] == 'C'))                  if ((magic[2] == 'C') && (magic[3] == 'C'))
706                  {                  {
707                          /* connect from server */                          /* connect from server */
708                            rdpdr_send_clientcapabilty();
709                            rdpdr_send_available();
710                          return;                          return;
711                  }                  }
712                  else if ((magic[2] == 'r') && (magic[3] == 'd'))                  if ((magic[2] == 'r') && (magic[3] == 'd'))
713                  {                  {
714                          /* connect to a specific resource */                          /* connect to a specific resource */
715                          in_uint32(s, handle);                          in_uint32(s, handle);
716                          printf("Server connected to resource %d\n", handle);  #if WITH_DEBUG_RDP5
717                            DEBUG(("RDPDR: Server connected to resource %d\n", handle));
718    #endif
719                            return;
720                    }
721                    if ((magic[2] == 'P') && (magic[3] == 'S'))
722                    {
723                            /* server capability */
724                            return;
725                    }
726            }
727            if ((magic[0] == 'R') && (magic[1] == 'P'))
728            {
729                    if ((magic[2] == 'C') && (magic[3] == 'P'))
730                    {
731                            printercache_process(s);
732                          return;                          return;
733                  }                  }
734          }          }
# Line 246  rdpdr_process(STREAM s) Line 736  rdpdr_process(STREAM s)
736  }  }
737    
738  BOOL  BOOL
739  rdpdr_init(void)  rdpdr_init()
740  {  {
741          rdpdr_channel =          if (g_num_devices > 0)
742                  channel_register("rdpdr", CHANNEL_OPTION_INITIALIZED | CHANNEL_OPTION_COMPRESS_RDP,          {
743                                   rdpdr_process);                  rdpdr_channel =
744                            channel_register("rdpdr",
745                                             CHANNEL_OPTION_INITIALIZED | CHANNEL_OPTION_COMPRESS_RDP,
746                                             rdpdr_process);
747            }
748    
749          return (rdpdr_channel != NULL);          return (rdpdr_channel != NULL);
750  }  }
751    
752    /* Add file descriptors of pending io request to select() */
753    void
754    rdpdr_add_fds(int *n, fd_set * rfds, fd_set * wfds, struct timeval *tv, BOOL * timeout)
755    {
756            uint32 select_timeout = 0;      // Timeout value to be used for select() (in millisecons).
757            struct async_iorequest *iorq;
758    
759            iorq = g_iorequest;
760            while (iorq != NULL)
761            {
762                    if (iorq->fd != 0)
763                    {
764                            switch (iorq->major)
765                            {
766                                    case IRP_MJ_READ:
767    
768                                            FD_SET(iorq->fd, rfds);
769    
770                                            // Check if io request timeout is smaller than current (but not 0).
771                                            if (iorq->timeout
772                                                && (select_timeout == 0
773                                                    || iorq->timeout < select_timeout))
774                                            {
775                                                    // Set new timeout
776                                                    select_timeout = iorq->timeout;
777                                                    g_min_timeout_fd = iorq->fd;    /* Remember fd */
778                                                    tv->tv_sec = select_timeout / 1000;
779                                                    tv->tv_usec = (select_timeout % 1000) * 1000;
780                                                    *timeout = True;
781                                            }
782                                            break;
783    
784                                    case IRP_MJ_WRITE:
785                                            FD_SET(iorq->fd, wfds);
786                                            break;
787    
788                            }
789                            *n = MAX(*n, iorq->fd);
790                    }
791    
792                    iorq = iorq->next;
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 */
821    void
822    rdpdr_check_fds(fd_set * rfds, fd_set * wfds, BOOL timed_out)
823    {
824            NTSTATUS status;
825            uint32 result = 0;
826            DEVICE_FNS *fns;
827            struct async_iorequest *iorq;
828            struct async_iorequest *prev;
829            uint32 req_size = 0;
830    
831            if (timed_out)
832            {
833                    rdpdr_abort_io(g_min_timeout_fd, 0, STATUS_TIMEOUT);
834                    return;
835            }
836    
837            iorq = g_iorequest;
838            prev = NULL;
839            while (iorq != NULL)
840            {
841                    if (iorq->fd != 0)
842                    {
843                            switch (iorq->major)
844                            {
845                                    case IRP_MJ_READ:
846                                            if (FD_ISSET(iorq->fd, rfds))
847                                            {
848                                                    /* Read the data */
849                                                    fns = iorq->fns;
850    
851                                                    req_size =
852                                                            (iorq->length - iorq->partial_len) >
853                                                            8192 ? 8192 : (iorq->length -
854                                                                           iorq->partial_len);
855                                                    /* never read larger chunks than 8k - chances are that it will block */
856                                                    status = fns->read(iorq->fd,
857                                                                       iorq->buffer + iorq->partial_len,
858                                                                       req_size, iorq->offset, &result);
859    
860                                                    if (result > 0)
861                                                    {
862                                                            iorq->partial_len += result;
863                                                            iorq->offset += result;
864                                                    }
865    #if WITH_DEBUG_RDP5
866                                                    DEBUG(("RDPDR: %d bytes of data read\n", result));
867    #endif
868                                                    /* only delete link if all data has been transfered */
869                                                    /* or if result was 0 and status success - EOF      */
870                                                    if ((iorq->partial_len == iorq->length) ||
871                                                        (result == 0))
872                                                    {
873    #if WITH_DEBUG_RDP5
874                                                            DEBUG(("RDPDR: AIO total %u bytes read of %u\n", iorq->partial_len, iorq->length));
875    #endif
876                                                            rdpdr_send_completion(iorq->device,
877                                                                                  iorq->id, status,
878                                                                                  iorq->partial_len,
879                                                                                  iorq->buffer,
880                                                                                  iorq->partial_len);
881                                                            iorq = rdpdr_remove_iorequest(prev, iorq);
882                                                    }
883                                            }
884                                            break;
885                                    case IRP_MJ_WRITE:
886                                            if (FD_ISSET(iorq->fd, wfds))
887                                            {
888                                                    /* Write data. */
889                                                    fns = iorq->fns;
890    
891                                                    req_size =
892                                                            (iorq->length - iorq->partial_len) >
893                                                            8192 ? 8192 : (iorq->length -
894                                                                           iorq->partial_len);
895    
896                                                    /* never write larger chunks than 8k - chances are that it will block */
897                                                    status = fns->write(iorq->fd,
898                                                                        iorq->buffer +
899                                                                        iorq->partial_len, req_size,
900                                                                        iorq->offset, &result);
901    
902                                                    if (result > 0)
903                                                    {
904                                                            iorq->partial_len += result;
905                                                            iorq->offset += result;
906                                                    }
907    
908    #if WITH_DEBUG_RDP5
909                                                    DEBUG(("RDPDR: %d bytes of data written\n",
910                                                           result));
911    #endif
912                                                    /* only delete link if all data has been transfered */
913                                                    /* or we couldn't write */
914                                                    if ((iorq->partial_len == iorq->length)
915                                                        || (result == 0))
916                                                    {
917    #if WITH_DEBUG_RDP5
918                                                            DEBUG(("RDPDR: AIO total %u bytes written of %u\n", iorq->partial_len, iorq->length));
919    #endif
920                                                            rdpdr_send_completion(iorq->device,
921                                                                                  iorq->id, status,
922                                                                                  iorq->partial_len,
923                                                                                  (uint8 *) "", 1);
924    
925                                                            iorq = rdpdr_remove_iorequest(prev, iorq);
926                                                    }
927                                            }
928                                            break;
929                            }
930    
931                    }
932                    prev = iorq;
933                    if (iorq)
934                            iorq = iorq->next;
935            }
936    
937    }
938    
939    /* Abort a pending io request for a given handle and major */
940    BOOL
941    rdpdr_abort_io(uint32 fd, uint32 major, NTSTATUS status)
942    {
943            uint32 result;
944            struct async_iorequest *iorq;
945            struct async_iorequest *prev;
946    
947            iorq = g_iorequest;
948            prev = NULL;
949            while (iorq != NULL)
950            {
951                    // Only remove from table when major is not set, or when correct major is supplied.
952                    // Abort read should not abort a write io request.
953                    if ((iorq->fd == fd) && (major == 0 || iorq->major == major))
954                    {
955                            result = 0;
956                            rdpdr_send_completion(iorq->device, iorq->id, status, result, (uint8 *) "",
957                                                  1);
958    
959                            iorq = rdpdr_remove_iorequest(prev, iorq);
960                            return True;
961                    }
962    
963                    prev = iorq;
964                    iorq = iorq->next;
965            }
966    
967            return False;
968    }
Legend:
Removed from v.435  
changed lines
  Added in v.651
  ViewVC Help
Powered by ViewVC 1.1.26