/[rdesktop]/sourceforge.net/trunk/rdesktop/rdesktop.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/rdesktop.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 66 by astrand, Thu Jul 18 18:28:12 2002 UTC revision 571 by n-ki, Wed Jan 21 18:17:20 2004 UTC
# Line 1  Line 1 
1  /*  /* -*- c-basic-offset: 8 -*-
2     rdesktop: A Remote Desktop Protocol client.     rdesktop: A Remote Desktop Protocol client.
3     Entrypoint and utility functions     Entrypoint and utility functions
4     Copyright (C) Matthew Chapman 1999-2001     Copyright (C) Matthew Chapman 1999-2003
5      
6     This program is free software; you can redistribute it and/or modify     This program is free software; you can redistribute it and/or modify
7     it under the terms of the GNU General Public License as published by     it under the terms of the GNU General Public License as published by
8     the Free Software Foundation; either version 2 of the License, or     the Free Software Foundation; either version 2 of the License, or
9     (at your option) any later version.     (at your option) any later version.
10      
11     This program is distributed in the hope that it will be useful,     This program is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14     GNU General Public License for more details.     GNU General Public License for more details.
15      
16     You should have received a copy of the GNU General Public License     You should have received a copy of the GNU General Public License
17     along with this program; if not, write to the Free Software     along with this program; if not, write to the Free Software
18     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19  */  */
20    
 #include <stdlib.h>             /* malloc realloc free */  
21  #include <stdarg.h>             /* va_list va_start va_end */  #include <stdarg.h>             /* va_list va_start va_end */
22  #include <unistd.h>             /* read close getuid getgid getpid getppid gethostname */  #include <unistd.h>             /* read close getuid getgid getpid getppid gethostname */
23  #include <fcntl.h>              /* open */  #include <fcntl.h>              /* open */
24  #include <pwd.h>                /* getpwuid */  #include <pwd.h>                /* getpwuid */
25  #include <limits.h>             /* PATH_MAX */  #include <termios.h>            /* tcgetattr tcsetattr */
26  #include <sys/stat.h>           /* stat */  #include <sys/stat.h>           /* stat */
27  #include <sys/time.h>           /* gettimeofday */  #include <sys/time.h>           /* gettimeofday */
28  #include <sys/times.h>          /* times */  #include <sys/times.h>          /* times */
29    #include <ctype.h>              /* toupper */
30    #include <errno.h>
31  #include "rdesktop.h"  #include "rdesktop.h"
32    
33  char username[16];  #ifdef EGD_SOCKET
34    #include <sys/socket.h>         /* socket connect */
35    #include <sys/un.h>             /* sockaddr_un */
36    #endif
37    
38    #ifdef WITH_OPENSSL
39    #include <openssl/md5.h>
40    #else
41    #include "crypto/md5.h"
42    #endif
43    
44    char g_title[64] = "";
45    char g_username[64];
46  char hostname[16];  char hostname[16];
47  char keymapname[16];  char keymapname[16];
48  int keylayout = 0x409;          /* Defaults to US keyboard layout */  int keylayout = 0x409;          /* Defaults to US keyboard layout */
 int width;  
 int height;  
 int tcp_port_rdp = TCP_PORT_RDP;  
 BOOL bitmap_compression = True;  
 BOOL sendmotion = True;  
 BOOL orders = True;  
 BOOL licence = True;  
 BOOL encryption = True;  
 BOOL desktop_save = True;  
 BOOL fullscreen = False;  
49    
50    int g_width = 800;              /* width is special: If 0, the
51                                       geometry will be fetched from
52                                       _NET_WORKAREA. If negative,
53                                       absolute value specifies the
54                                       percent of the whole screen. */
55    int g_height = 600;
56    int tcp_port_rdp = TCP_PORT_RDP;
57    int g_server_bpp = 8;
58    int g_win_button_size = 0;      /* If zero, disable single app mode */
59    BOOL g_bitmap_compression = True;
60    BOOL g_sendmotion = True;
61    BOOL g_orders = True;
62    BOOL g_encryption = True;
63    BOOL packet_encryption = True;
64    BOOL g_desktop_save = True;
65    BOOL g_fullscreen = False;
66    BOOL g_grab_keyboard = True;
67    BOOL g_hide_decorations = False;
68    BOOL g_use_rdp5 = True;
69    BOOL g_console_session = False;
70    BOOL g_numlock_sync = False;
71    extern BOOL g_owncolmap;
72    
73    #ifdef WITH_RDPSND
74    BOOL g_rdpsnd = False;
75    #endif
76    
77    extern RDPDR_DEVICE g_rdpdr_device[];
78    extern uint32 g_num_devices;
79    
80    #ifdef RDP2VNC
81    extern int rfb_port;
82    extern int defer_time;
83    void
84    rdp2vnc_connect(char *server, uint32 flags, char *domain, char *password,
85                    char *shell, char *directory);
86    #endif
87  /* Display usage information */  /* Display usage information */
88  static void  static void
89  usage(char *program)  usage(char *program)
90  {  {
91          printf("Usage: %s [options] server\n", program);          fprintf(stderr, "rdesktop: A Remote Desktop Protocol client.\n");
92          printf("   -u: user name\n");          fprintf(stderr, "Version " VERSION ". Copyright (C) 1999-2003 Matt Chapman.\n");
93          printf("   -d: domain\n");          fprintf(stderr, "See http://www.rdesktop.org/ for more information.\n\n");
94          printf("   -s: shell\n");  
95          printf("   -c: working directory\n");          fprintf(stderr, "Usage: %s [options] server[:port]\n", program);
96          printf("   -p: password (autologon)\n");  #ifdef RDP2VNC
97          printf("   -n: client hostname\n");          fprintf(stderr, "   -V: vnc port\n");
98          printf("   -k: keyboard layout on terminal server (us,sv,gr etc.)\n");          fprintf(stderr, "   -Q: defer time (ms)\n");
99          printf("   -g: desktop geometry (WxH)\n");  #endif
100          printf("   -f: full-screen mode\n");          fprintf(stderr, "   -u: user name\n");
101          printf("   -b: force bitmap updates\n");          fprintf(stderr, "   -d: domain\n");
102          printf("   -e: disable encryption (French TS)\n");          fprintf(stderr, "   -s: shell\n");
103          printf("   -m: do not send motion events\n");          fprintf(stderr, "   -c: working directory\n");
104          printf("   -l: do not request licence\n");          fprintf(stderr, "   -p: password (- to prompt)\n");
105          printf("   -t: rdp tcp port\n\n");          fprintf(stderr, "   -n: client hostname\n");
106            fprintf(stderr, "   -k: keyboard layout on server (en-us, de, sv, etc.)\n");
107            fprintf(stderr, "   -g: desktop geometry (WxH)\n");
108            fprintf(stderr, "   -f: full-screen mode\n");
109            fprintf(stderr, "   -b: force bitmap updates\n");
110            fprintf(stderr, "   -e: disable encryption (French TS)\n");
111            fprintf(stderr, "   -E: disable encryption from client to server\n");
112            fprintf(stderr, "   -m: do not send motion events\n");
113            fprintf(stderr, "   -C: use private colour map\n");
114            fprintf(stderr, "   -D: hide window manager decorations\n");
115            fprintf(stderr, "   -K: keep window manager key bindings\n");
116            fprintf(stderr, "   -S: caption button size (single application mode)\n");
117            fprintf(stderr, "   -T: window title\n");
118            fprintf(stderr, "   -N: enable numlock syncronization\n");
119            fprintf(stderr, "   -a: connection colour depth\n");
120            fprintf(stderr, "   -r: enable specified device redirection (this flag can be repeated)\n");
121            fprintf(stderr, "         '-r comport:COM1=/dev/ttyS0': enable serial redirection of /dev/ttyS0 to COM1\n");
122            fprintf(stderr, "             or    :COM1=/dev/ttyS0:9600,0|1|2,0|2,5|6|7|8:dtr \n");
123            fprintf(stderr, "         '-r disk:A=/mnt/floppy': enable redirection of /mnt/floppy to A:\n");
124            fprintf(stderr, "             or   A=/mnt/floppy,D=/mnt/cdrom'\n");
125            fprintf(stderr, "         '-r lptport:LPT1=/dev/lp0': enable parallel redirection of /dev/lp0 to LPT1\n");
126            fprintf(stderr, "             or       LPT1=/dev/lp0,LPT2=/dev/lp1\n");
127            fprintf(stderr, "         '-r printer:mydeskjet': enable printer redirection\n");
128            fprintf(stderr, "             or       mydeskjet:\"HP Laserjet IIIP\" to enter server driver as well\n");
129            fprintf(stderr, "         '-r sound': enable sound redirection\n");
130            fprintf(stderr, "   -0: attach to console\n");
131            fprintf(stderr, "   -4: use RDP version 4\n");
132            fprintf(stderr, "   -5: use RDP version 5 (default)\n");
133    }
134    
135    static BOOL
136    read_password(char *password, int size)
137    {
138            struct termios tios;
139            BOOL ret = False;
140            int istty = 0;
141            char *p;
142    
143            if (tcgetattr(STDIN_FILENO, &tios) == 0)
144            {
145                    fprintf(stderr, "Password: ");
146                    tios.c_lflag &= ~ECHO;
147                    tcsetattr(STDIN_FILENO, TCSANOW, &tios);
148                    istty = 1;
149            }
150    
151            if (fgets(password, size, stdin) != NULL)
152            {
153                    ret = True;
154    
155                    /* strip final newline */
156                    p = strchr(password, '\n');
157                    if (p != NULL)
158                            *p = 0;
159            }
160    
161            if (istty)
162            {
163                    tios.c_lflag |= ECHO;
164                    tcsetattr(STDIN_FILENO, TCSANOW, &tios);
165                    fprintf(stderr, "\n");
166            }
167    
168            return ret;
169    }
170    
171    static void
172    parse_server_and_port(char *server)
173    {
174            char *p;
175    #ifdef IPv6
176            int addr_colons;
177    #endif
178    
179    #ifdef IPv6
180            p = server;
181            addr_colons = 0;
182            while (*p)
183                    if (*p++ == ':')
184                            addr_colons++;
185            if (addr_colons >= 2)
186            {
187                    /* numeric IPv6 style address format - [1:2:3::4]:port */
188                    p = strchr(server, ']');
189                    if (*server == '[' && p != NULL)
190                    {
191                            if (*(p + 1) == ':' && *(p + 2) != '\0')
192                                    tcp_port_rdp = strtol(p + 2, NULL, 10);
193                            /* remove the port number and brackets from the address */
194                            *p = '\0';
195                            strncpy(server, server + 1, strlen(server));
196                    }
197            }
198            else
199            {
200                    /* dns name or IPv4 style address format - server.example.com:port or 1.2.3.4:port */
201                    p = strchr(server, ':');
202                    if (p != NULL)
203                    {
204                            tcp_port_rdp = strtol(p + 1, NULL, 10);
205                            *p = 0;
206                    }
207            }
208    #else /* no IPv6 support */
209            p = strchr(server, ':');
210            if (p != NULL)
211            {
212                    tcp_port_rdp = strtol(p + 1, NULL, 10);
213                    *p = 0;
214            }
215    #endif /* IPv6 */
216    
217  }  }
218    
219  /* Client program */  /* Client program */
220  int  int
221  main(int argc, char *argv[])  main(int argc, char *argv[])
222  {  {
223            char server[64];
224          char fullhostname[64];          char fullhostname[64];
225          char domain[16];          char domain[16];
226          char password[16];          char password[64];
227          char shell[32];          char shell[128];
228          char directory[32];          char directory[32];
229          char title[32];          BOOL prompt_password, rdp_retval = False;
230          struct passwd *pw;          struct passwd *pw;
         char *server, *p;  
231          uint32 flags;          uint32 flags;
232            char *p;
233          int c;          int c;
234    
235          printf("rdesktop: A Remote Desktop Protocol client.\n");          int username_option = 0;
         printf("Version " VERSION  
                ". Copyright (C) 1999-2001 Matt Chapman.\n");  
         printf("See http://www.rdesktop.org/ for more information.\n\n");  
236    
237          flags = RDP_LOGON_NORMAL;          flags = RDP_LOGON_NORMAL;
238            prompt_password = False;
239          domain[0] = password[0] = shell[0] = directory[0] = 0;          domain[0] = password[0] = shell[0] = directory[0] = 0;
240          strcpy(keymapname, "us");          strcpy(keymapname, "en-us");
241    
242            g_num_devices = 0;
243    
244          while ((c = getopt(argc, argv, "u:d:s:c:p:n:k:g:t:fbemlh?")) != -1)  #ifdef RDP2VNC
245    #define VNCOPT "V:Q:"
246    #else
247    #define VNCOPT
248    #endif
249    
250            while ((c = getopt(argc, argv, VNCOPT "u:d:s:c:p:n:k:g:fbeEmCDKS:T:Na:r:045h?")) != -1)
251          {          {
252                  switch (c)                  switch (c)
253                  {                  {
254    #ifdef RDP2VNC
255                            case 'V':
256                                    rfb_port = strtol(optarg, NULL, 10);
257                                    if (rfb_port < 100)
258                                            rfb_port += 5900;
259                                    break;
260    
261                            case 'Q':
262                                    defer_time = strtol(optarg, NULL, 10);
263                                    if (defer_time < 0)
264                                            defer_time = 0;
265                                    break;
266    #endif
267    
268                          case 'u':                          case 'u':
269                                  STRNCPY(username, optarg, sizeof(username));                                  STRNCPY(g_username, optarg, sizeof(g_username));
270                                    username_option = 1;
271                                  break;                                  break;
272    
273                          case 'd':                          case 'd':
# Line 110  main(int argc, char *argv[]) Line 283  main(int argc, char *argv[])
283                                  break;                                  break;
284    
285                          case 'p':                          case 'p':
286                                    if ((optarg[0] == '-') && (optarg[1] == 0))
287                                    {
288                                            prompt_password = True;
289                                            break;
290                                    }
291    
292                                  STRNCPY(password, optarg, sizeof(password));                                  STRNCPY(password, optarg, sizeof(password));
293                                  flags |= RDP_LOGON_AUTO;                                  flags |= RDP_LOGON_AUTO;
294    
295                                    /* try to overwrite argument so it won't appear in ps */
296                                    p = optarg;
297                                    while (*p)
298                                            *(p++) = 'X';
299                                  break;                                  break;
300    
301                          case 'n':                          case 'n':
# Line 119  main(int argc, char *argv[]) Line 303  main(int argc, char *argv[])
303                                  break;                                  break;
304    
305                          case 'k':                          case 'k':
306                                  STRNCPY(keymapname, optarg,                                  STRNCPY(keymapname, optarg, sizeof(keymapname));
                                         sizeof(keymapname));  
307                                  break;                                  break;
308    
309                          case 'g':                          case 'g':
310                                  width = strtol(optarg, &p, 10);                                  g_fullscreen = False;
311                                    if (!strcmp(optarg, "workarea"))
312                                    {
313                                            g_width = g_height = 0;
314                                            break;
315                                    }
316    
317                                    g_width = strtol(optarg, &p, 10);
318                                    if (g_width <= 0)
319                                    {
320                                            error("invalid geometry\n");
321                                            return 1;
322                                    }
323    
324                                  if (*p == 'x')                                  if (*p == 'x')
325                                          height = strtol(p + 1, NULL, 10);                                          g_height = strtol(p + 1, NULL, 10);
326    
327                                  if ((width == 0) || (height == 0))                                  if (g_height <= 0)
328                                  {                                  {
329                                          error("invalid geometry\n");                                          error("invalid geometry\n");
330                                          return 1;                                          return 1;
331                                  }                                  }
332    
333                                    if (*p == '%')
334                                            g_width = -g_width;
335    
336                                  break;                                  break;
337    
338                          case 'f':                          case 'f':
339                                  fullscreen = True;                                  g_fullscreen = True;
340                                  break;                                  break;
341    
342                          case 'b':                          case 'b':
343                                  orders = False;                                  g_orders = False;
344                                  break;                                  break;
345    
346                          case 'e':                          case 'e':
347                                  encryption = False;                                  g_encryption = False;
348                                    break;
349                            case 'E':
350                                    packet_encryption = False;
351                                  break;                                  break;
   
352                          case 'm':                          case 'm':
353                                  sendmotion = False;                                  g_sendmotion = False;
354                                    break;
355    
356                            case 'C':
357                                    g_owncolmap = True;
358                                    break;
359    
360                            case 'D':
361                                    g_hide_decorations = True;
362                                    break;
363    
364                            case 'K':
365                                    g_grab_keyboard = False;
366                                    break;
367    
368                            case 'S':
369                                    if (!strcmp(optarg, "standard"))
370                                    {
371                                            g_win_button_size = 18;
372                                            break;
373                                    }
374    
375                                    g_win_button_size = strtol(optarg, &p, 10);
376    
377                                    if (*p)
378                                    {
379                                            error("invalid button size\n");
380                                            return 1;
381                                    }
382    
383                                    break;
384    
385                            case 'T':
386                                    STRNCPY(g_title, optarg, sizeof(g_title));
387                                    break;
388    
389                            case 'N':
390                                    g_numlock_sync = True;
391                                    break;
392    
393                            case 'a':
394                                    g_server_bpp = strtol(optarg, NULL, 10);
395                                    if (g_server_bpp != 8 && g_server_bpp != 16 && g_server_bpp != 15
396                                        && g_server_bpp != 24)
397                                    {
398                                            error("invalid server bpp\n");
399                                            return 1;
400                                    }
401                                    break;
402    
403                            case 'r':
404    
405                                    if (strncmp("sound", optarg, 5) == 0)
406                                    {
407    #ifdef WITH_RDPSND
408                                            g_rdpsnd = True;
409    #else
410                                            warning("Not compiled with sound support");
411    #endif
412                                    }
413                                    else if (strncmp("disk", optarg, 4) == 0)
414                                    {
415                                            /* -r disk:h:=/mnt/floppy */
416                                            disk_enum_devices(&g_num_devices, optarg + 4);
417                                    }
418                                    else if (strncmp("comport", optarg, 7) == 0)
419                                    {
420                                            serial_enum_devices(&g_num_devices, optarg + 7);
421                                    }
422                                    else if (strncmp("lptport", optarg, 7) == 0)
423                                    {
424                                            parallel_enum_devices(&g_num_devices, optarg + 7);
425                                    }
426                                    else if (strncmp("printer", optarg, 7) == 0)
427                                    {
428                                            printer_enum_devices(&g_num_devices, optarg + 7);
429                                    }
430                                    else
431                                    {
432                                            warning("Unknown -r argument\n\n\tPossible arguments are: comport, disk, lptport, printer, sound\n");
433                                    }
434                                    break;
435    
436                            case '0':
437                                    g_console_session = True;
438                                  break;                                  break;
439    
440                          case 'l':                          case '4':
441                                  licence = False;                                  g_use_rdp5 = False;
442                                  break;                                  break;
443    
444                          case 't':                          case '5':
445                                  tcp_port_rdp = strtol(optarg, NULL, 10);                                  g_use_rdp5 = True;
446                                  break;                                  break;
447    
448                          case 'h':                          case 'h':
# Line 173  main(int argc, char *argv[]) Line 459  main(int argc, char *argv[])
459                  return 1;                  return 1;
460          }          }
461    
462          server = argv[optind];          STRNCPY(server, argv[optind], sizeof(server));
463            parse_server_and_port(server);
464    
465          if (username[0] == 0)          if (!username_option)
466          {          {
467                  pw = getpwuid(getuid());                  pw = getpwuid(getuid());
468                  if ((pw == NULL) || (pw->pw_name == NULL))                  if ((pw == NULL) || (pw->pw_name == NULL))
# Line 184  main(int argc, char *argv[]) Line 471  main(int argc, char *argv[])
471                          return 1;                          return 1;
472                  }                  }
473    
474                  STRNCPY(username, pw->pw_name, sizeof(username));                  STRNCPY(g_username, pw->pw_name, sizeof(g_username));
475          }          }
476    
477          if (hostname[0] == 0)          if (hostname[0] == 0)
# Line 202  main(int argc, char *argv[]) Line 489  main(int argc, char *argv[])
489                  STRNCPY(hostname, fullhostname, sizeof(hostname));                  STRNCPY(hostname, fullhostname, sizeof(hostname));
490          }          }
491    
492          if (!strcmp(password, "-"))          if (prompt_password && read_password(password, sizeof(password)))
493          {                  flags |= RDP_LOGON_AUTO;
                 p = getpass("Password: ");  
                 if (p == NULL)  
                 {  
                         error("failed to read password\n");  
                         return 0;  
                 }  
                 STRNCPY(password, p, sizeof(password));  
         }  
494    
495          if ((width == 0) || (height == 0))          if (g_title[0] == 0)
496          {          {
497                  width = 800;                  strcpy(g_title, "rdesktop - ");
498                  height = 600;                  strncat(g_title, server, sizeof(g_title) - sizeof("rdesktop - "));
499          }          }
500    
501          strcpy(title, "rdesktop - ");  #ifdef RDP2VNC
502          strncat(title, server, sizeof(title) - sizeof("rdesktop - "));          rdp2vnc_connect(server, flags, domain, password, shell, directory);
503            return 0;
504    #else
505    
506            if (!ui_init())
507                    return 1;
508    
509          xkeymap_init1();  #ifdef WITH_RDPSND
510            if (g_rdpsnd)
511                    rdpsnd_init();
512    #endif
513            rdpdr_init();
514    
515          if (!rdp_connect(server, flags, domain, password, shell, directory))          if (!rdp_connect(server, flags, domain, password, shell, directory))
516                  return 1;                  return 1;
517    
518          printf("Connection successful.\n");          /* By setting encryption to False here, we have an encrypted login
519               packet but unencrypted transfer of other packets */
520            if (!packet_encryption)
521                    g_encryption = False;
522    
523    
524            DEBUG(("Connection successful.\n"));
525            memset(password, 0, sizeof(password));
526    
527          if (ui_create_window(title))          if (ui_create_window())
528          {          {
529                  rdp_main_loop();                  rdp_retval = rdp_main_loop();
530                  ui_destroy_window();                  ui_destroy_window();
531          }          }
532    
533          printf("Disconnecting...\n");          DEBUG(("Disconnecting...\n"));
534          rdp_disconnect();          rdp_disconnect();
535          return 0;          ui_deinit();
536    
537            if (True == rdp_retval)
538                    return 0;
539            else
540                    return 2;
541    
542    #endif
543    
544    }
545    
546    #ifdef EGD_SOCKET
547    /* Read 32 random bytes from PRNGD or EGD socket (based on OpenSSL RAND_egd) */
548    static BOOL
549    generate_random_egd(uint8 * buf)
550    {
551            struct sockaddr_un addr;
552            BOOL ret = False;
553            int fd;
554    
555            fd = socket(AF_UNIX, SOCK_STREAM, 0);
556            if (fd == -1)
557                    return False;
558    
559            addr.sun_family = AF_UNIX;
560            memcpy(addr.sun_path, EGD_SOCKET, sizeof(EGD_SOCKET));
561            if (connect(fd, (struct sockaddr *) &addr, sizeof(addr)) == -1)
562                    goto err;
563    
564            /* PRNGD and EGD use a simple communications protocol */
565            buf[0] = 1;             /* Non-blocking (similar to /dev/urandom) */
566            buf[1] = 32;            /* Number of requested random bytes */
567            if (write(fd, buf, 2) != 2)
568                    goto err;
569    
570            if ((read(fd, buf, 1) != 1) || (buf[0] == 0))   /* Available? */
571                    goto err;
572    
573            if (read(fd, buf, 32) != 32)
574                    goto err;
575    
576            ret = True;
577    
578          err:
579            close(fd);
580            return ret;
581  }  }
582    #endif
583    
584  /* Generate a 32-byte random for the secure transport code. */  /* Generate a 32-byte random for the secure transport code. */
585  void  void
# Line 246  generate_random(uint8 * random) Line 587  generate_random(uint8 * random)
587  {  {
588          struct stat st;          struct stat st;
589          struct tms tmsbuf;          struct tms tmsbuf;
590          uint32 *r = (uint32 *) random;          MD5_CTX md5;
591          int fd;          uint32 *r;
592            int fd, n;
593    
594          /* If we have a kernel random device, use it. */          /* If we have a kernel random device, try that first */
595          if (((fd = open("/dev/urandom", O_RDONLY)) != -1)          if (((fd = open("/dev/urandom", O_RDONLY)) != -1)
596              || ((fd = open("/dev/random", O_RDONLY)) != -1))              || ((fd = open("/dev/random", O_RDONLY)) != -1))
597          {          {
598                  read(fd, random, 32);                  n = read(fd, random, 32);
599                  close(fd);                  close(fd);
600                  return;                  if (n == 32)
601                            return;
602          }          }
603    
604    #ifdef EGD_SOCKET
605            /* As a second preference use an EGD */
606            if (generate_random_egd(random))
607                    return;
608    #endif
609    
610          /* Otherwise use whatever entropy we can gather - ideas welcome. */          /* Otherwise use whatever entropy we can gather - ideas welcome. */
611            r = (uint32 *) random;
612          r[0] = (getpid()) | (getppid() << 16);          r[0] = (getpid()) | (getppid() << 16);
613          r[1] = (getuid()) | (getgid() << 16);          r[1] = (getuid()) | (getgid() << 16);
614          r[2] = times(&tmsbuf);  /* system uptime (clocks) */          r[2] = times(&tmsbuf);  /* system uptime (clocks) */
# Line 267  generate_random(uint8 * random) Line 617  generate_random(uint8 * random)
617          r[5] = st.st_atime;          r[5] = st.st_atime;
618          r[6] = st.st_mtime;          r[6] = st.st_mtime;
619          r[7] = st.st_ctime;          r[7] = st.st_ctime;
620    
621            /* Hash both halves with MD5 to obscure possible patterns */
622            MD5_Init(&md5);
623            MD5_Update(&md5, random, 16);
624            MD5_Final(random, &md5);
625            MD5_Update(&md5, random + 16, 16);
626            MD5_Final(random + 16, &md5);
627  }  }
628    
629  /* malloc; exit if out of memory */  /* malloc; exit if out of memory */
# Line 315  error(char *format, ...) Line 672  error(char *format, ...)
672          va_end(ap);          va_end(ap);
673  }  }
674    
675    /* report a warning */
676    void
677    warning(char *format, ...)
678    {
679            va_list ap;
680    
681            fprintf(stderr, "WARNING: ");
682    
683            va_start(ap, format);
684            vfprintf(stderr, format, ap);
685            va_end(ap);
686    }
687    
688  /* report an unimplemented protocol feature */  /* report an unimplemented protocol feature */
689  void  void
690  unimpl(char *format, ...)  unimpl(char *format, ...)
# Line 333  void Line 703  void
703  hexdump(unsigned char *p, unsigned int len)  hexdump(unsigned char *p, unsigned int len)
704  {  {
705          unsigned char *line = p;          unsigned char *line = p;
706          unsigned int thisline, offset = 0;          int i, thisline, offset = 0;
         int i;  
707    
708          while (offset < len)          while (offset < len)
709          {          {
# Line 350  hexdump(unsigned char *p, unsigned int l Line 719  hexdump(unsigned char *p, unsigned int l
719                          printf("   ");                          printf("   ");
720    
721                  for (i = 0; i < thisline; i++)                  for (i = 0; i < thisline; i++)
722                          printf("%c",                          printf("%c", (line[i] >= 0x20 && line[i] < 0x7f) ? line[i] : '.');
                                (line[i] >= 0x20  
                                 && line[i] < 0x7f) ? line[i] : '.');  
723    
724                  printf("\n");                  printf("\n");
725                  offset += thisline;                  offset += thisline;
# Line 360  hexdump(unsigned char *p, unsigned int l Line 727  hexdump(unsigned char *p, unsigned int l
727          }          }
728  }  }
729    
730    /*
731      input: src is the string we look in for needle
732      return value: returns next src pointer, for
733            succesive executions, like in a while loop
734            if retval is 0, then there are no more args.
735      pitfalls:
736            src is modified. 0x00 chars are inserted to
737            terminate strings.
738            return val, points on the next val chr after ins
739            0x00
740    
741            example usage:
742            while( (pos = next_arg( optarg, ',')) ){
743                    printf("%s\n",optarg);
744                    optarg=pos;
745            }
746    
747    */
748    char *
749    next_arg(char *src, char needle)
750    {
751            char *nextval;
752            char *p;
753    
754            // EOS
755            if (*src == (char) 0x00)
756                    return 0;
757    
758            p = src;
759            // skip escaped needles.
760            while( (nextval = strchr(p, needle) ) )
761            {
762                    if( *(nextval-1) != '\\' )
763                            break;
764                    p = nextval +1;
765            }
766    
767            // more args available
768            if (nextval)
769            {
770                    *nextval = (char) 0x00;
771                    return ++nextval;
772            }
773    
774            // no more args after this, jump to EOS
775            nextval = src + strlen(src);
776            return nextval;
777    }
778    
779    
780    void
781    toupper_str(char* p)
782    {
783            while( *p ){
784                    if( (*p >= 'a') && (*p <= 'z') )
785                            *p = toupper((int) *p);
786                    p++;
787            }
788    }
789    
790    
791    /* not all clibs got ltoa */
792    #define LTOA_BUFSIZE (sizeof(long) * 8 + 1)
793    
794    char *
795    ltoa(long N, int base)
796    {
797            static char ret[LTOA_BUFSIZE];
798    
799            register int i = 2;
800            long uarg;
801            char *tail, *head = ret, buf[LTOA_BUFSIZE];
802    
803            if (36 < base || 2 > base)
804                    base = 10;
805    
806            tail = &buf[LTOA_BUFSIZE - 1];
807            *tail-- = '\0';
808    
809            if (10 == base && N < 0L)
810            {
811                    *head++ = '-';
812                    uarg = -N;
813            }
814            else
815                    uarg = N;
816    
817            if (uarg)
818            {
819                    for (i = 1; uarg; ++i)
820                    {
821                            register ldiv_t r;
822    
823                            r = ldiv(uarg, base);
824                            *tail-- = (char) (r.rem + ((9L < r.rem) ? ('A' - 10L) : '0'));
825                            uarg = r.quot;
826                    }
827            }
828            else
829                    *tail-- = '0';
830    
831            memcpy(head, ++tail, i);
832            return ret;
833    }
834    
835    
836  int  int
837  load_licence(unsigned char **data)  load_licence(unsigned char **data)
838  {  {
839          char path[PATH_MAX];          char *home, *path;
         char *home;  
840          struct stat st;          struct stat st;
841          int fd;          int fd, length;
842    
843          home = getenv("HOME");          home = getenv("HOME");
844          if (home == NULL)          if (home == NULL)
845                  return -1;                  return -1;
846    
847          STRNCPY(path, home, sizeof(path));          path = (char *) xmalloc(strlen(home) + strlen(hostname) + sizeof("/.rdesktop/licence."));
848          strncat(path, "/.rdesktop/licence", sizeof(path) - strlen(path) - 1);          sprintf(path, "%s/.rdesktop/licence.%s", home, hostname);
849    
850          fd = open(path, O_RDONLY);          fd = open(path, O_RDONLY);
851          if (fd == -1)          if (fd == -1)
# Line 382  load_licence(unsigned char **data) Line 854  load_licence(unsigned char **data)
854          if (fstat(fd, &st))          if (fstat(fd, &st))
855                  return -1;                  return -1;
856    
857          *data = xmalloc(st.st_size);          *data = (uint8 *) xmalloc(st.st_size);
858          return read(fd, *data, st.st_size);          length = read(fd, *data, st.st_size);
859            close(fd);
860            xfree(path);
861            return length;
862  }  }
863    
864  void  void
865  save_licence(unsigned char *data, int length)  save_licence(unsigned char *data, int length)
866  {  {
867          char path[PATH_MAX];          char *home, *path, *tmppath;
         char *home;  
868          int fd;          int fd;
869    
870          home = getenv("HOME");          home = getenv("HOME");
871          if (home == NULL)          if (home == NULL)
872                  return;                  return;
873    
874          STRNCPY(path, home, sizeof(path));          path = (char *) xmalloc(strlen(home) + strlen(hostname) + sizeof("/.rdesktop/licence."));
875          strncat(path, "/.rdesktop", sizeof(path) - strlen(path) - 1);  
876          mkdir(path, 0700);          sprintf(path, "%s/.rdesktop", home);
877            if ((mkdir(path, 0700) == -1) && errno != EEXIST)
878            {
879                    perror(path);
880                    return;
881            }
882    
883            /* write licence to licence.hostname.new, then atomically rename to licence.hostname */
884    
885          strncat(path, "/licence", sizeof(path) - strlen(path) - 1);          sprintf(path, "%s/.rdesktop/licence.%s", home, hostname);
886            tmppath = (char *) xmalloc(strlen(path) + sizeof(".new"));
887            strcpy(tmppath, path);
888            strcat(tmppath, ".new");
889    
890          fd = open(path, O_WRONLY | O_CREAT | O_TRUNC, 0600);          fd = open(tmppath, O_WRONLY | O_CREAT | O_TRUNC, 0600);
891          if (fd == -1)          if (fd == -1)
892          {          {
893                  perror("open");                  perror(tmppath);
894                  return;                  return;
895          }          }
896    
897          write(fd, data, length);          if (write(fd, data, length) != length)
898            {
899                    perror(tmppath);
900                    unlink(tmppath);
901            }
902            else if (rename(tmppath, path) == -1)
903            {
904                    perror(path);
905                    unlink(tmppath);
906            }
907    
908          close(fd);          close(fd);
909            xfree(tmppath);
910            xfree(path);
911  }  }
Legend:
Removed from v.66  
changed lines
  Added in v.571
  ViewVC Help
Powered by ViewVC 1.1.26