/[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

Annotation of /sourceforge.net/trunk/rdesktop/rdesktop.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 111 - (hide annotations)
Sun Sep 1 12:03:33 2002 UTC (21 years, 8 months ago) by astrand
File MIME type: text/plain
File size: 9233 byte(s)
Replaced all printf with fprintf(stderr)

1 matty 10 /*
2     rdesktop: A Remote Desktop Protocol client.
3     Entrypoint and utility functions
4 matty 30 Copyright (C) Matthew Chapman 1999-2001
5 jsorg71 100
6 matty 10 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
8     the Free Software Foundation; either version 2 of the License, or
9     (at your option) any later version.
10 jsorg71 100
11 matty 10 This program is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14     GNU General Public License for more details.
15 jsorg71 100
16 matty 10 You should have received a copy of the GNU General Public License
17     along with this program; if not, write to the Free Software
18     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19     */
20    
21 matty 24 #include <stdlib.h> /* malloc realloc free */
22 matty 30 #include <stdarg.h> /* va_list va_start va_end */
23 matty 24 #include <unistd.h> /* read close getuid getgid getpid getppid gethostname */
24     #include <fcntl.h> /* open */
25     #include <pwd.h> /* getpwuid */
26 matthewc 39 #include <limits.h> /* PATH_MAX */
27 matty 24 #include <sys/stat.h> /* stat */
28     #include <sys/time.h> /* gettimeofday */
29     #include <sys/times.h> /* times */
30 matty 10 #include "rdesktop.h"
31    
32 astrand 107 char title[32] = "";
33 matty 10 char username[16];
34     char hostname[16];
35 matthewc 38 char keymapname[16];
36 astrand 66 int keylayout = 0x409; /* Defaults to US keyboard layout */
37 matty 30 int width;
38     int height;
39 jsorg71 58 int tcp_port_rdp = TCP_PORT_RDP;
40 matty 28 BOOL bitmap_compression = True;
41 matty 29 BOOL sendmotion = True;
42 matty 10 BOOL orders = True;
43     BOOL licence = True;
44 matty 30 BOOL encryption = True;
45 matty 28 BOOL desktop_save = True;
46     BOOL fullscreen = False;
47 astrand 76 BOOL grab_keyboard = True;
48 matty 10
49     /* Display usage information */
50 matty 25 static void
51     usage(char *program)
52 matty 10 {
53 astrand 111 fprintf(stderr, "Usage: %s [options] server\n", program);
54     fprintf(stderr, " -u: user name\n");
55     fprintf(stderr, " -d: domain\n");
56     fprintf(stderr, " -s: shell\n");
57     fprintf(stderr, " -c: working directory\n");
58     fprintf(stderr, " -p: password (autologon)\n");
59     fprintf(stderr, " -P: askpass-program (autologon)\n");
60     fprintf(stderr, " -n: client hostname\n");
61     fprintf(stderr, " -k: keyboard layout on terminal server (us,sv,gr etc.)\n");
62     fprintf(stderr, " -g: desktop geometry (WxH)\n");
63     fprintf(stderr, " -f: full-screen mode\n");
64     fprintf(stderr, " -b: force bitmap updates\n");
65     fprintf(stderr, " -e: disable encryption (French TS)\n");
66     fprintf(stderr, " -m: do not send motion events\n");
67     fprintf(stderr, " -l: do not request licence\n");
68     fprintf(stderr, " -t: rdp tcp port\n");
69     fprintf(stderr, " -K: keep window manager key bindings\n");
70     fprintf(stderr, " -w: window title\n");
71 matty 10 }
72    
73     /* Client program */
74 matty 25 int
75     main(int argc, char *argv[])
76 matty 10 {
77 matty 30 char fullhostname[64];
78 matty 21 char domain[16];
79     char password[16];
80 astrand 96 char *askpass_result;
81 matty 21 char shell[32];
82     char directory[32];
83 matty 30 struct passwd *pw;
84     char *server, *p;
85     uint32 flags;
86 matty 10 int c;
87    
88 astrand 111 fprintf(stderr, "rdesktop: A Remote Desktop Protocol client.\n");
89     fprintf(stderr, "Version " VERSION ". Copyright (C) 1999-2001 Matt Chapman.\n");
90     fprintf(stderr, "See http://www.rdesktop.org/ for more information.\n\n");
91 matty 10
92 matty 21 flags = RDP_LOGON_NORMAL;
93     domain[0] = password[0] = shell[0] = directory[0] = 0;
94 matthewc 38 strcpy(keymapname, "us");
95 matty 21
96 astrand 107 while ((c = getopt(argc, argv, "u:d:s:c:p:P:n:k:g:t:fbemlKw:h?")) != -1)
97 matty 10 {
98     switch (c)
99     {
100     case 'u':
101 matty 30 STRNCPY(username, optarg, sizeof(username));
102 matty 10 break;
103    
104 matty 21 case 'd':
105 matty 30 STRNCPY(domain, optarg, sizeof(domain));
106 matty 21 break;
107    
108     case 's':
109 matty 30 STRNCPY(shell, optarg, sizeof(shell));
110 matty 21 break;
111    
112     case 'c':
113 matty 30 STRNCPY(directory, optarg, sizeof(directory));
114 matty 21 break;
115    
116 matty 30 case 'p':
117     STRNCPY(password, optarg, sizeof(password));
118     flags |= RDP_LOGON_AUTO;
119     break;
120    
121 astrand 96 case 'P':
122     askpass_result = askpass(optarg, "Enter password");
123     if (askpass_result == NULL)
124     exit(1);
125    
126     STRNCPY(password, askpass_result, sizeof(password));
127     free(askpass_result);
128     flags |= RDP_LOGON_AUTO;
129     break;
130    
131 matty 10 case 'n':
132 matty 30 STRNCPY(hostname, optarg, sizeof(hostname));
133 matty 10 break;
134    
135     case 'k':
136 astrand 82 STRNCPY(keymapname, optarg, sizeof(keymapname));
137 matty 10 break;
138    
139 matty 30 case 'g':
140     width = strtol(optarg, &p, 10);
141     if (*p == 'x')
142 astrand 64 height = strtol(p + 1, NULL, 10);
143 matty 30
144     if ((width == 0) || (height == 0))
145     {
146     error("invalid geometry\n");
147     return 1;
148     }
149 matty 10 break;
150    
151 matty 30 case 'f':
152     fullscreen = True;
153     break;
154    
155 matty 10 case 'b':
156     orders = False;
157     break;
158    
159 matty 30 case 'e':
160     encryption = False;
161 matty 10 break;
162    
163 matty 30 case 'm':
164     sendmotion = False;
165 matty 28 break;
166 matty 29
167 matty 30 case 'l':
168     licence = False;
169 matty 28 break;
170 matty 29
171 jsorg71 58 case 't':
172     tcp_port_rdp = strtol(optarg, NULL, 10);
173     break;
174    
175 astrand 76 case 'K':
176     grab_keyboard = False;
177     break;
178    
179 astrand 107 case 'w':
180     strncpy(title, optarg, sizeof(title));
181     break;
182    
183 matty 28 case 'h':
184 matty 10 case '?':
185     default:
186     usage(argv[0]);
187     return 1;
188     }
189     }
190    
191     if (argc - optind < 1)
192     {
193     usage(argv[0]);
194     return 1;
195     }
196    
197     server = argv[optind];
198    
199     if (username[0] == 0)
200     {
201     pw = getpwuid(getuid());
202     if ((pw == NULL) || (pw->pw_name == NULL))
203     {
204 matty 30 error("could not determine username, use -u\n");
205 matty 10 return 1;
206     }
207    
208 matty 30 STRNCPY(username, pw->pw_name, sizeof(username));
209 matty 10 }
210    
211     if (hostname[0] == 0)
212     {
213 matty 30 if (gethostname(fullhostname, sizeof(fullhostname)) == -1)
214 matty 10 {
215 matty 30 error("could not determine local hostname, use -n\n");
216 matty 10 return 1;
217     }
218 matty 30
219     p = strchr(fullhostname, '.');
220     if (p != NULL)
221     *p = 0;
222    
223     STRNCPY(hostname, fullhostname, sizeof(hostname));
224 matty 10 }
225    
226 matty 30 if (!strcmp(password, "-"))
227     {
228     p = getpass("Password: ");
229     if (p == NULL)
230     {
231     error("failed to read password\n");
232     return 0;
233     }
234     STRNCPY(password, p, sizeof(password));
235     }
236    
237     if ((width == 0) || (height == 0))
238     {
239     width = 800;
240     height = 600;
241     }
242 matthewc 103 else
243     {
244     /* make sure width is a multiple of 4 */
245     width = (width + 3) & ~3;
246     }
247 matty 30
248 astrand 107 if (!strlen(title))
249     {
250     strcpy(title, "rdesktop - ");
251     strncat(title, server, sizeof(title) - sizeof("rdesktop - "));
252     }
253 matty 12
254 astrand 66 xkeymap_init1();
255 astrand 82 if (!ui_init())
256     return 1;
257 astrand 66
258 matthewc 53 if (!rdp_connect(server, flags, domain, password, shell, directory))
259     return 1;
260    
261 astrand 111 fprintf(stderr, "Connection successful.\n");
262 matthewc 53
263 jsorg71 100 if (ui_create_window())
264 matty 10 {
265     rdp_main_loop();
266     ui_destroy_window();
267     }
268    
269 astrand 111 fprintf(stderr, "Disconnecting...\n");
270 matthewc 53 rdp_disconnect();
271 matty 10 return 0;
272     }
273    
274     /* Generate a 32-byte random for the secure transport code. */
275 matty 25 void
276 astrand 64 generate_random(uint8 * random)
277 matty 10 {
278     struct stat st;
279 matty 22 struct tms tmsbuf;
280 matty 24 uint32 *r = (uint32 *) random;
281 matty 10 int fd;
282    
283     /* If we have a kernel random device, use it. */
284 matty 30 if (((fd = open("/dev/urandom", O_RDONLY)) != -1)
285     || ((fd = open("/dev/random", O_RDONLY)) != -1))
286 matty 10 {
287     read(fd, random, 32);
288     close(fd);
289     return;
290     }
291    
292     /* Otherwise use whatever entropy we can gather - ideas welcome. */
293     r[0] = (getpid()) | (getppid() << 16);
294     r[1] = (getuid()) | (getgid() << 16);
295 matty 24 r[2] = times(&tmsbuf); /* system uptime (clocks) */
296     gettimeofday((struct timeval *) &r[3], NULL); /* sec and usec */
297 matty 10 stat("/tmp", &st);
298     r[5] = st.st_atime;
299     r[6] = st.st_mtime;
300     r[7] = st.st_ctime;
301     }
302    
303     /* malloc; exit if out of memory */
304 matty 25 void *
305     xmalloc(int size)
306 matty 10 {
307     void *mem = malloc(size);
308     if (mem == NULL)
309     {
310 matty 30 error("xmalloc %d\n", size);
311 matty 10 exit(1);
312     }
313     return mem;
314     }
315    
316     /* realloc; exit if out of memory */
317 matty 25 void *
318     xrealloc(void *oldmem, int size)
319 matty 10 {
320     void *mem = realloc(oldmem, size);
321     if (mem == NULL)
322     {
323 matty 30 error("xrealloc %d\n", size);
324 matty 10 exit(1);
325     }
326     return mem;
327     }
328    
329     /* free */
330 matty 25 void
331     xfree(void *mem)
332 matty 10 {
333     free(mem);
334     }
335    
336 matty 30 /* report an error */
337 matty 25 void
338 matty 30 error(char *format, ...)
339     {
340     va_list ap;
341    
342     fprintf(stderr, "ERROR: ");
343    
344     va_start(ap, format);
345     vfprintf(stderr, format, ap);
346     va_end(ap);
347     }
348    
349     /* report an unimplemented protocol feature */
350     void
351     unimpl(char *format, ...)
352     {
353     va_list ap;
354    
355     fprintf(stderr, "NOT IMPLEMENTED: ");
356    
357     va_start(ap, format);
358     vfprintf(stderr, format, ap);
359     va_end(ap);
360     }
361    
362     /* produce a hex dump */
363     void
364 matty 25 hexdump(unsigned char *p, unsigned int len)
365 matty 10 {
366     unsigned char *line = p;
367     unsigned int thisline, offset = 0;
368     int i;
369    
370     while (offset < len)
371     {
372 astrand 111 fprintf(stderr, "%04x ", offset);
373 matty 10 thisline = len - offset;
374     if (thisline > 16)
375     thisline = 16;
376    
377     for (i = 0; i < thisline; i++)
378 astrand 111 fprintf(stderr, "%02x ", line[i]);
379 matty 10
380 matty 30 for (; i < 16; i++)
381 astrand 111 fprintf(stderr, " ");
382 matty 30
383 matty 10 for (i = 0; i < thisline; i++)
384 astrand 111 fprintf(stderr, "%c", (line[i] >= 0x20 && line[i] < 0x7f) ? line[i] : '.');
385 matty 10
386 astrand 111 fprintf(stderr, "\n");
387 matty 10 offset += thisline;
388     line += thisline;
389     }
390     }
391 matthewc 39
392     int
393     load_licence(unsigned char **data)
394     {
395     char path[PATH_MAX];
396     char *home;
397     struct stat st;
398     int fd;
399    
400     home = getenv("HOME");
401     if (home == NULL)
402     return -1;
403    
404     STRNCPY(path, home, sizeof(path));
405 astrand 64 strncat(path, "/.rdesktop/licence", sizeof(path) - strlen(path) - 1);
406 matthewc 39
407     fd = open(path, O_RDONLY);
408     if (fd == -1)
409     return -1;
410    
411     if (fstat(fd, &st))
412     return -1;
413    
414     *data = xmalloc(st.st_size);
415     return read(fd, *data, st.st_size);
416     }
417    
418     void
419     save_licence(unsigned char *data, int length)
420     {
421     char path[PATH_MAX];
422     char *home;
423     int fd;
424    
425     home = getenv("HOME");
426     if (home == NULL)
427     return;
428    
429     STRNCPY(path, home, sizeof(path));
430 astrand 64 strncat(path, "/.rdesktop", sizeof(path) - strlen(path) - 1);
431 matthewc 39 mkdir(path, 0700);
432    
433 astrand 64 strncat(path, "/licence", sizeof(path) - strlen(path) - 1);
434 matthewc 39
435 astrand 64 fd = open(path, O_WRONLY | O_CREAT | O_TRUNC, 0600);
436 matthewc 39 if (fd == -1)
437     {
438     perror("open");
439     return;
440     }
441    
442     write(fd, data, length);
443     close(fd);
444     }

  ViewVC Help
Powered by ViewVC 1.1.26