/[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 583 by n-ki, Tue Jan 27 10:36:49 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,COM2=/dev/ttyS1\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    #ifdef RDP2VNC
245    #define VNCOPT "V:Q:"
246    #else
247    #define VNCOPT
248    #endif
249    
250          while ((c = getopt(argc, argv, "u:d:s:c:p:n:k:g:t:fbemlh?")) != -1)          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;                                  break;
435    
436                          case 'l':                          case '0':
437                                  licence = False;                                  g_console_session = True;
438                                  break;                                  break;
439    
440                          case 't':                          case '4':
441                                  tcp_port_rdp = strtol(optarg, NULL, 10);                                  g_use_rdp5 = False;
442                                    break;
443    
444                            case '5':
445                                    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          if (ui_create_window(title))  
524            DEBUG(("Connection successful.\n"));
525            memset(password, 0, sizeof(password));
526    
527            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             Needle may be escaped by a backslash, in
733             that case we ignore that particular needle.
734      return value: returns next src pointer, for
735            succesive executions, like in a while loop
736            if retval is 0, then there are no more args.
737      pitfalls:
738            src is modified. 0x00 chars are inserted to
739            terminate strings.
740            return val, points on the next val chr after ins
741            0x00
742    
743            example usage:
744            while( (pos = next_arg( optarg, ',')) ){
745                    printf("%s\n",optarg);
746                    optarg=pos;
747            }
748    
749    */
750    char *
751    next_arg(char *src, char needle)
752    {
753            char *nextval;
754            char *p;
755            char *mvp = 0;
756    
757            /* EOS */
758            if (*src == (char) 0x00)
759                    return 0;
760    
761            p = src;
762            /*  skip escaped needles */
763            while ((nextval = strchr(p, needle)))
764            {
765                    mvp = nextval - 1;
766                    /* found backslashed needle */
767                    if (*mvp == '\\' && (mvp > src))
768                    {
769                            /* move string one to the left */
770                            while (*(mvp + 1) != (char) 0x00)
771                            {
772                                    *mvp = *(mvp + 1);
773                                    *mvp++;
774                            }
775                            *mvp = (char) 0x00;
776                            p = nextval;
777                    }
778                    else
779                    {
780                            p = nextval + 1;
781                            break;
782                    }
783    
784            }
785    
786            /* more args available */
787            if (nextval)
788            {
789                    *nextval = (char) 0x00;
790                    return ++nextval;
791            }
792    
793            /* no more args after this, jump to EOS */
794            nextval = src + strlen(src);
795            return nextval;
796    }
797    
798    
799    void
800    toupper_str(char *p)
801    {
802            while (*p)
803            {
804                    if ((*p >= 'a') && (*p <= 'z'))
805                            *p = toupper((int) *p);
806                    p++;
807            }
808    }
809    
810    
811    /* not all clibs got ltoa */
812    #define LTOA_BUFSIZE (sizeof(long) * 8 + 1)
813    
814    char *
815    ltoa(long N, int base)
816    {
817            static char ret[LTOA_BUFSIZE];
818    
819            char *head = ret, buf[LTOA_BUFSIZE], *tail = buf + sizeof(buf);
820    
821            register int divrem;
822    
823            if (base < 36 || 2 > base)
824                    base = 10;
825    
826            if (N < 0)
827            {
828                    *head++ = '-';
829                    N = -N;
830            }
831    
832            tail = buf + sizeof(buf);
833            *--tail = 0;
834    
835            do
836            {
837                    divrem = N % base;
838                    *--tail = (divrem <= 9) ? divrem + '0' : divrem + 'a' - 10;
839                    N /= base;
840            }
841            while (N);
842    
843            strcpy(head, tail);
844            return ret;
845    }
846    
847    
848  int  int
849  load_licence(unsigned char **data)  load_licence(unsigned char **data)
850  {  {
851          char path[PATH_MAX];          char *home, *path;
         char *home;  
852          struct stat st;          struct stat st;
853          int fd;          int fd, length;
854    
855          home = getenv("HOME");          home = getenv("HOME");
856          if (home == NULL)          if (home == NULL)
857                  return -1;                  return -1;
858    
859          STRNCPY(path, home, sizeof(path));          path = (char *) xmalloc(strlen(home) + strlen(hostname) + sizeof("/.rdesktop/licence."));
860          strncat(path, "/.rdesktop/licence", sizeof(path) - strlen(path) - 1);          sprintf(path, "%s/.rdesktop/licence.%s", home, hostname);
861    
862          fd = open(path, O_RDONLY);          fd = open(path, O_RDONLY);
863          if (fd == -1)          if (fd == -1)
# Line 382  load_licence(unsigned char **data) Line 866  load_licence(unsigned char **data)
866          if (fstat(fd, &st))          if (fstat(fd, &st))
867                  return -1;                  return -1;
868    
869          *data = xmalloc(st.st_size);          *data = (uint8 *) xmalloc(st.st_size);
870          return read(fd, *data, st.st_size);          length = read(fd, *data, st.st_size);
871            close(fd);
872            xfree(path);
873            return length;
874  }  }
875    
876  void  void
877  save_licence(unsigned char *data, int length)  save_licence(unsigned char *data, int length)
878  {  {
879          char path[PATH_MAX];          char *home, *path, *tmppath;
         char *home;  
880          int fd;          int fd;
881    
882          home = getenv("HOME");          home = getenv("HOME");
883          if (home == NULL)          if (home == NULL)
884                  return;                  return;
885    
886          STRNCPY(path, home, sizeof(path));          path = (char *) xmalloc(strlen(home) + strlen(hostname) + sizeof("/.rdesktop/licence."));
887          strncat(path, "/.rdesktop", sizeof(path) - strlen(path) - 1);  
888          mkdir(path, 0700);          sprintf(path, "%s/.rdesktop", home);
889            if ((mkdir(path, 0700) == -1) && errno != EEXIST)
890            {
891                    perror(path);
892                    return;
893            }
894    
895            /* write licence to licence.hostname.new, then atomically rename to licence.hostname */
896    
897          strncat(path, "/licence", sizeof(path) - strlen(path) - 1);          sprintf(path, "%s/.rdesktop/licence.%s", home, hostname);
898            tmppath = (char *) xmalloc(strlen(path) + sizeof(".new"));
899            strcpy(tmppath, path);
900            strcat(tmppath, ".new");
901    
902          fd = open(path, O_WRONLY | O_CREAT | O_TRUNC, 0600);          fd = open(tmppath, O_WRONLY | O_CREAT | O_TRUNC, 0600);
903          if (fd == -1)          if (fd == -1)
904          {          {
905                  perror("open");                  perror(tmppath);
906                  return;                  return;
907          }          }
908    
909          write(fd, data, length);          if (write(fd, data, length) != length)
910            {
911                    perror(tmppath);
912                    unlink(tmppath);
913            }
914            else if (rename(tmppath, path) == -1)
915            {
916                    perror(path);
917                    unlink(tmppath);
918            }
919    
920          close(fd);          close(fd);
921            xfree(tmppath);
922            xfree(path);
923  }  }
Legend:
Removed from v.66  
changed lines
  Added in v.583
  ViewVC Help
Powered by ViewVC 1.1.26