/[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 122 - (hide annotations)
Sat Sep 14 11:54:11 2002 UTC (21 years, 8 months ago) by matthewc
File MIME type: text/plain
File size: 9198 byte(s)
Eliminate noise in output - only print version/copyright in usage(),
make connecting/disconnecting message a DEBUG.

1 matty 10 /*
2     rdesktop: A Remote Desktop Protocol client.
3     Entrypoint and utility functions
4 matthewc 122 Copyright (C) Matthew Chapman 1999-2002
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 matthewc 122 fprintf(stderr, "rdesktop: A Remote Desktop Protocol client.\n");
54     fprintf(stderr, "Version " VERSION ". Copyright (C) 1999-2002 Matt Chapman.\n");
55     fprintf(stderr, "See http://www.rdesktop.org/ for more information.\n\n");
56    
57 astrand 111 fprintf(stderr, "Usage: %s [options] server\n", program);
58     fprintf(stderr, " -u: user name\n");
59     fprintf(stderr, " -d: domain\n");
60     fprintf(stderr, " -s: shell\n");
61     fprintf(stderr, " -c: working directory\n");
62     fprintf(stderr, " -p: password (autologon)\n");
63     fprintf(stderr, " -P: askpass-program (autologon)\n");
64     fprintf(stderr, " -n: client hostname\n");
65     fprintf(stderr, " -k: keyboard layout on terminal server (us,sv,gr etc.)\n");
66     fprintf(stderr, " -g: desktop geometry (WxH)\n");
67     fprintf(stderr, " -f: full-screen mode\n");
68     fprintf(stderr, " -b: force bitmap updates\n");
69     fprintf(stderr, " -e: disable encryption (French TS)\n");
70     fprintf(stderr, " -m: do not send motion events\n");
71     fprintf(stderr, " -l: do not request licence\n");
72     fprintf(stderr, " -t: rdp tcp port\n");
73     fprintf(stderr, " -K: keep window manager key bindings\n");
74     fprintf(stderr, " -w: window title\n");
75 matty 10 }
76    
77     /* Client program */
78 matty 25 int
79     main(int argc, char *argv[])
80 matty 10 {
81 matty 30 char fullhostname[64];
82 matty 21 char domain[16];
83     char password[16];
84 astrand 96 char *askpass_result;
85 matty 21 char shell[32];
86     char directory[32];
87 matty 30 struct passwd *pw;
88     char *server, *p;
89     uint32 flags;
90 matty 10 int c;
91    
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 114 if (!strlen(title))
249 astrand 107 {
250     strcpy(title, "rdesktop - ");
251     strncat(title, server, sizeof(title) - sizeof("rdesktop - "));
252     }
253 matty 12
254 astrand 82 if (!ui_init())
255     return 1;
256 astrand 66
257 matthewc 53 if (!rdp_connect(server, flags, domain, password, shell, directory))
258     return 1;
259    
260 matthewc 122 DEBUG(("Connection successful.\n"));
261 matthewc 53
262 jsorg71 100 if (ui_create_window())
263 matty 10 {
264     rdp_main_loop();
265     ui_destroy_window();
266     }
267    
268 matthewc 122 DEBUG(("Disconnecting...\n"));
269 matthewc 53 rdp_disconnect();
270 matty 10 return 0;
271     }
272    
273     /* Generate a 32-byte random for the secure transport code. */
274 matty 25 void
275 astrand 64 generate_random(uint8 * random)
276 matty 10 {
277     struct stat st;
278 matty 22 struct tms tmsbuf;
279 matty 24 uint32 *r = (uint32 *) random;
280 matty 10 int fd;
281    
282     /* If we have a kernel random device, use it. */
283 matty 30 if (((fd = open("/dev/urandom", O_RDONLY)) != -1)
284     || ((fd = open("/dev/random", O_RDONLY)) != -1))
285 matty 10 {
286     read(fd, random, 32);
287     close(fd);
288     return;
289     }
290    
291     /* Otherwise use whatever entropy we can gather - ideas welcome. */
292     r[0] = (getpid()) | (getppid() << 16);
293     r[1] = (getuid()) | (getgid() << 16);
294 matty 24 r[2] = times(&tmsbuf); /* system uptime (clocks) */
295     gettimeofday((struct timeval *) &r[3], NULL); /* sec and usec */
296 matty 10 stat("/tmp", &st);
297     r[5] = st.st_atime;
298     r[6] = st.st_mtime;
299     r[7] = st.st_ctime;
300     }
301    
302     /* malloc; exit if out of memory */
303 matty 25 void *
304     xmalloc(int size)
305 matty 10 {
306     void *mem = malloc(size);
307     if (mem == NULL)
308     {
309 matty 30 error("xmalloc %d\n", size);
310 matty 10 exit(1);
311     }
312     return mem;
313     }
314    
315     /* realloc; exit if out of memory */
316 matty 25 void *
317     xrealloc(void *oldmem, int size)
318 matty 10 {
319     void *mem = realloc(oldmem, size);
320     if (mem == NULL)
321     {
322 matty 30 error("xrealloc %d\n", size);
323 matty 10 exit(1);
324     }
325     return mem;
326     }
327    
328     /* free */
329 matty 25 void
330     xfree(void *mem)
331 matty 10 {
332     free(mem);
333     }
334    
335 matty 30 /* report an error */
336 matty 25 void
337 matty 30 error(char *format, ...)
338     {
339     va_list ap;
340    
341     fprintf(stderr, "ERROR: ");
342    
343     va_start(ap, format);
344     vfprintf(stderr, format, ap);
345     va_end(ap);
346     }
347    
348     /* report an unimplemented protocol feature */
349     void
350     unimpl(char *format, ...)
351     {
352     va_list ap;
353    
354     fprintf(stderr, "NOT IMPLEMENTED: ");
355    
356     va_start(ap, format);
357     vfprintf(stderr, format, ap);
358     va_end(ap);
359     }
360    
361     /* produce a hex dump */
362     void
363 matty 25 hexdump(unsigned char *p, unsigned int len)
364 matty 10 {
365     unsigned char *line = p;
366     unsigned int thisline, offset = 0;
367     int i;
368    
369     while (offset < len)
370     {
371 astrand 111 fprintf(stderr, "%04x ", offset);
372 matty 10 thisline = len - offset;
373     if (thisline > 16)
374     thisline = 16;
375    
376     for (i = 0; i < thisline; i++)
377 astrand 111 fprintf(stderr, "%02x ", line[i]);
378 matty 10
379 matty 30 for (; i < 16; i++)
380 astrand 111 fprintf(stderr, " ");
381 matty 30
382 matty 10 for (i = 0; i < thisline; i++)
383 astrand 111 fprintf(stderr, "%c", (line[i] >= 0x20 && line[i] < 0x7f) ? line[i] : '.');
384 matty 10
385 astrand 111 fprintf(stderr, "\n");
386 matty 10 offset += thisline;
387     line += thisline;
388     }
389     }
390 matthewc 39
391     int
392     load_licence(unsigned char **data)
393     {
394     char path[PATH_MAX];
395     char *home;
396     struct stat st;
397     int fd;
398    
399     home = getenv("HOME");
400     if (home == NULL)
401     return -1;
402    
403     STRNCPY(path, home, sizeof(path));
404 astrand 64 strncat(path, "/.rdesktop/licence", sizeof(path) - strlen(path) - 1);
405 matthewc 39
406     fd = open(path, O_RDONLY);
407     if (fd == -1)
408     return -1;
409    
410     if (fstat(fd, &st))
411     return -1;
412    
413     *data = xmalloc(st.st_size);
414     return read(fd, *data, st.st_size);
415     }
416    
417     void
418     save_licence(unsigned char *data, int length)
419     {
420     char path[PATH_MAX];
421     char *home;
422     int fd;
423    
424     home = getenv("HOME");
425     if (home == NULL)
426     return;
427    
428     STRNCPY(path, home, sizeof(path));
429 astrand 64 strncat(path, "/.rdesktop", sizeof(path) - strlen(path) - 1);
430 matthewc 39 mkdir(path, 0700);
431    
432 astrand 64 strncat(path, "/licence", sizeof(path) - strlen(path) - 1);
433 matthewc 39
434 astrand 64 fd = open(path, O_WRONLY | O_CREAT | O_TRUNC, 0600);
435 matthewc 39 if (fd == -1)
436     {
437     perror("open");
438     return;
439     }
440    
441     write(fd, data, length);
442     close(fd);
443     }

  ViewVC Help
Powered by ViewVC 1.1.26