|
|
1 | // This file is part of libptytty. Do not make local modifications. |
|
|
2 | // http://software.schmorp.de/pkg/libptytty |
|
|
3 | |
1 | /*--------------------------------*-C-*---------------------------------* |
4 | /*----------------------------------------------------------------------* |
2 | * File: ptytty.C |
5 | * File: ptytty.C |
3 | *----------------------------------------------------------------------* |
6 | *----------------------------------------------------------------------* |
4 | * |
7 | * |
5 | * All portions of code are copyright by their respective author/s. |
8 | * All portions of code are copyright by their respective author/s. |
6 | * Copyright (c) 1999-2001 Geoff Wing <gcw@pobox.com> |
9 | * Copyright (c) 1999-2001 Geoff Wing <gcw@pobox.com> |
7 | * Copyright (c) 2004-2006 Marc Lehmann <pcg@goof.com> |
10 | * Copyright (c) 2004-2006 Marc Lehmann <pcg@goof.com> |
|
|
11 | * Copyright (c) 2006 Emanuele Giaquinta <e.giaquinta@glauco.it> |
8 | * |
12 | * |
9 | * This program is free software; you can redistribute it and/or modify |
13 | * This program is free software; you can redistribute it and/or modify |
10 | * it under the terms of the GNU General Public License as published by |
14 | * it under the terms of the GNU General Public License as published by |
11 | * the Free Software Foundation; either version 2 of the License, or |
15 | * the Free Software Foundation; either version 2 of the License, or |
12 | * (at your option) any later version. |
16 | * (at your option) any later version. |
… | |
… | |
19 | * You should have received a copy of the GNU General Public License |
23 | * You should have received a copy of the GNU General Public License |
20 | * along with this program; if not, write to the Free Software |
24 | * along with this program; if not, write to the Free Software |
21 | * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. |
25 | * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. |
22 | *---------------------------------------------------------------------*/ |
26 | *---------------------------------------------------------------------*/ |
23 | |
27 | |
24 | #include "../config.h" /* NECESSARY */ |
28 | #include "../config.h" |
|
|
29 | |
25 | #include "rxvt.h" |
30 | #include "ptytty.h" |
26 | |
31 | |
27 | #include <cstdlib> |
32 | #include <cstdlib> |
28 | #include <cstring> |
33 | #include <cstring> |
|
|
34 | #include <csignal> |
29 | |
35 | |
30 | #include <sys/types.h> |
36 | #include <sys/types.h> |
31 | #include <sys/socket.h> |
|
|
32 | #include <unistd.h> |
37 | #include <unistd.h> |
33 | #include <fcntl.h> |
38 | #include <fcntl.h> |
34 | |
39 | |
35 | #ifdef HAVE_SYS_IOCTL_H |
40 | #ifdef HAVE_SYS_IOCTL_H |
36 | # include <sys/ioctl.h> |
41 | # include <sys/ioctl.h> |
… | |
… | |
51 | #ifdef TTY_GID_SUPPORT |
56 | #ifdef TTY_GID_SUPPORT |
52 | #include <grp.h> |
57 | #include <grp.h> |
53 | #endif |
58 | #endif |
54 | |
59 | |
55 | #include <cstdio> |
60 | #include <cstdio> |
56 | |
|
|
57 | #include "rxvtutil.h" |
|
|
58 | #include "fdpass.h" |
|
|
59 | #include "ptytty.h" |
|
|
60 | |
61 | |
61 | ///////////////////////////////////////////////////////////////////////////// |
62 | ///////////////////////////////////////////////////////////////////////////// |
62 | |
63 | |
63 | /* ------------------------------------------------------------------------- * |
64 | /* ------------------------------------------------------------------------- * |
64 | * GET PSEUDO TELETYPE - MASTER AND SLAVE * |
65 | * GET PSEUDO TELETYPE - MASTER AND SLAVE * |
… | |
… | |
234 | |
235 | |
235 | ioctl (fd_tty, TIOCSCTTY, NULL); |
236 | ioctl (fd_tty, TIOCSCTTY, NULL); |
236 | |
237 | |
237 | int fd = open ("/dev/tty", O_WRONLY); |
238 | int fd = open ("/dev/tty", O_WRONLY); |
238 | if (fd < 0) |
239 | if (fd < 0) |
239 | return -1; /* fatal */ |
240 | return -1; /* fatal */ |
240 | |
241 | |
241 | close (fd); |
242 | close (fd); |
242 | |
243 | |
243 | return 0; |
244 | return 0; |
244 | } |
245 | } |
245 | |
246 | |
246 | void |
247 | void |
247 | rxvt_ptytty::close_tty () |
248 | ptytty::close_tty () |
248 | { |
249 | { |
249 | if (tty < 0) |
250 | if (tty < 0) |
250 | return; |
251 | return; |
251 | |
252 | |
252 | close (tty); |
253 | close (tty); |
253 | tty = -1; |
254 | tty = -1; |
254 | } |
255 | } |
255 | |
256 | |
256 | bool |
257 | bool |
257 | rxvt_ptytty::make_controlling_tty () |
258 | ptytty::make_controlling_tty () |
258 | { |
259 | { |
259 | return control_tty (tty) >= 0; |
260 | return control_tty (tty) >= 0; |
260 | } |
261 | } |
261 | |
262 | |
262 | void |
263 | void |
263 | rxvt_ptytty::set_utf8_mode (bool on) |
264 | ptytty::set_utf8_mode (bool on) |
264 | { |
265 | { |
265 | #ifdef IUTF8 |
266 | #ifdef IUTF8 |
266 | if (pty < 0) |
267 | if (pty < 0) |
267 | return; |
268 | return; |
268 | |
269 | |
… | |
… | |
307 | gid = 0; |
308 | gid = 0; |
308 | } |
309 | } |
309 | } |
310 | } |
310 | } ttyconf; |
311 | } ttyconf; |
311 | |
312 | |
312 | rxvt_ptytty_unix::rxvt_ptytty_unix () |
313 | ptytty_unix::ptytty_unix () |
313 | { |
314 | { |
314 | name = 0; |
315 | name = 0; |
315 | #if UTMP_SUPPORT |
316 | #if UTMP_SUPPORT |
316 | cmd_pid = 0; |
317 | cmd_pid = 0; |
317 | #endif |
318 | #endif |
318 | } |
319 | } |
319 | |
320 | |
320 | rxvt_ptytty_unix::~rxvt_ptytty_unix () |
321 | ptytty_unix::~ptytty_unix () |
321 | { |
322 | { |
322 | #if UTMP_SUPPORT |
323 | #if UTMP_SUPPORT |
323 | logout (); |
324 | logout (); |
324 | #endif |
325 | #endif |
325 | put (); |
326 | put (); |
326 | } |
327 | } |
327 | |
328 | |
328 | void |
329 | void |
329 | rxvt_ptytty_unix::put () |
330 | ptytty_unix::put () |
330 | { |
331 | { |
331 | chmod (name, RESTORE_TTY_MODE); |
332 | chmod (name, RESTORE_TTY_MODE); |
332 | chown (name, 0, ttyconf.gid); |
333 | chown (name, 0, ttyconf.gid); |
333 | |
334 | |
334 | close_tty (); |
335 | close_tty (); |
… | |
… | |
341 | pty = tty = -1; |
342 | pty = tty = -1; |
342 | name = 0; |
343 | name = 0; |
343 | } |
344 | } |
344 | |
345 | |
345 | bool |
346 | bool |
346 | rxvt_ptytty_unix::get () |
347 | ptytty_unix::get () |
347 | { |
348 | { |
348 | /* get master (pty) */ |
349 | /* get master (pty) */ |
349 | if ((pty = get_pty (&tty, &name)) < 0) |
350 | if ((pty = get_pty (&tty, &name)) < 0) |
350 | return false; |
351 | return false; |
351 | |
352 | |
… | |
… | |
370 | } |
371 | } |
371 | |
372 | |
372 | return true; |
373 | return true; |
373 | } |
374 | } |
374 | |
375 | |
375 | #if PTYTTY_HELPER |
|
|
376 | |
|
|
377 | static int sock_fd; |
|
|
378 | static int pid; |
|
|
379 | |
|
|
380 | struct command |
|
|
381 | { |
|
|
382 | enum { get, login, destroy } type; |
|
|
383 | |
|
|
384 | rxvt_ptytty *id; |
|
|
385 | |
|
|
386 | bool login_shell; |
|
|
387 | int cmd_pid; |
|
|
388 | char hostname[512]; // arbitrary, but should be plenty |
|
|
389 | }; |
|
|
390 | |
|
|
391 | struct rxvt_ptytty_proxy : zero_initialized, rxvt_ptytty |
|
|
392 | { |
|
|
393 | rxvt_ptytty *id; |
|
|
394 | |
|
|
395 | ~rxvt_ptytty_proxy (); |
|
|
396 | |
|
|
397 | bool get (); |
|
|
398 | void login (int cmd_pid, bool login_shell, const char *hostname); |
|
|
399 | }; |
|
|
400 | |
|
|
401 | bool |
|
|
402 | rxvt_ptytty_proxy::get () |
|
|
403 | { |
|
|
404 | command cmd; |
|
|
405 | |
|
|
406 | cmd.type = command::get; |
|
|
407 | |
|
|
408 | write (sock_fd, &cmd, sizeof (cmd)); |
|
|
409 | |
|
|
410 | if (read (sock_fd, &id, sizeof (id)) != sizeof (id)) |
|
|
411 | rxvt_fatal ("protocol error while creating pty using helper process, aborting.\n"); |
|
|
412 | |
|
|
413 | if (!id) |
|
|
414 | return false; |
|
|
415 | |
|
|
416 | if ((pty = rxvt_recv_fd (sock_fd)) < 0 |
|
|
417 | || (tty = rxvt_recv_fd (sock_fd)) < 0) |
|
|
418 | rxvt_fatal ("protocol error while reading pty/tty fds from helper process, aborting.\n"); |
|
|
419 | |
|
|
420 | return true; |
|
|
421 | } |
|
|
422 | |
|
|
423 | void |
|
|
424 | rxvt_ptytty_proxy::login (int cmd_pid, bool login_shell, const char *hostname) |
|
|
425 | { |
|
|
426 | command cmd; |
|
|
427 | |
|
|
428 | cmd.type = command::login; |
|
|
429 | cmd.id = id; |
|
|
430 | cmd.cmd_pid = cmd_pid; |
|
|
431 | cmd.login_shell = login_shell; |
|
|
432 | strncpy (cmd.hostname, hostname, sizeof (cmd.hostname)); |
|
|
433 | |
|
|
434 | write (sock_fd, &cmd, sizeof (cmd)); |
|
|
435 | } |
|
|
436 | |
|
|
437 | rxvt_ptytty_proxy::~rxvt_ptytty_proxy () |
|
|
438 | { |
|
|
439 | command cmd; |
|
|
440 | |
|
|
441 | cmd.type = command::destroy; |
|
|
442 | cmd.id = id; |
|
|
443 | |
|
|
444 | write (sock_fd, &cmd, sizeof (cmd)); |
|
|
445 | } |
|
|
446 | |
|
|
447 | static |
|
|
448 | void serve () |
|
|
449 | { |
|
|
450 | command cmd; |
|
|
451 | vector<rxvt_ptytty *> ptys; |
|
|
452 | |
|
|
453 | while (read (sock_fd, &cmd, sizeof (command)) == sizeof (command)) |
|
|
454 | { |
|
|
455 | if (cmd.type == command::get) |
|
|
456 | { |
|
|
457 | // -> id ptyfd ttyfd |
|
|
458 | cmd.id = new rxvt_ptytty_unix; |
|
|
459 | |
|
|
460 | if (cmd.id->get ()) |
|
|
461 | { |
|
|
462 | write (sock_fd, &cmd.id, sizeof (cmd.id)); |
|
|
463 | ptys.push_back (cmd.id); |
|
|
464 | |
|
|
465 | rxvt_send_fd (sock_fd, cmd.id->pty); |
|
|
466 | rxvt_send_fd (sock_fd, cmd.id->tty); |
|
|
467 | } |
|
|
468 | else |
|
|
469 | { |
|
|
470 | delete cmd.id; |
|
|
471 | cmd.id = 0; |
|
|
472 | write (sock_fd, &cmd.id, sizeof (cmd.id)); |
|
|
473 | } |
|
|
474 | } |
|
|
475 | else if (cmd.type == command::login) |
|
|
476 | { |
|
|
477 | #if UTMP_SUPPORT |
|
|
478 | if (find (ptys.begin (), ptys.end (), cmd.id)) |
|
|
479 | { |
|
|
480 | cmd.hostname[sizeof (cmd.hostname) - 1] = 0; |
|
|
481 | cmd.id->login (cmd.cmd_pid, cmd.login_shell, cmd.hostname); |
|
|
482 | } |
|
|
483 | #endif |
|
|
484 | } |
|
|
485 | else if (cmd.type == command::destroy) |
|
|
486 | { |
|
|
487 | rxvt_ptytty **pty = find (ptys.begin (), ptys.end (), cmd.id); |
|
|
488 | |
|
|
489 | if (pty) |
|
|
490 | { |
|
|
491 | delete *pty; |
|
|
492 | ptys.erase (pty); |
|
|
493 | } |
|
|
494 | } |
|
|
495 | else |
|
|
496 | break; |
|
|
497 | } |
|
|
498 | |
|
|
499 | // destroy all ptys |
|
|
500 | for (rxvt_ptytty **i = ptys.end (); i-- > ptys.begin (); ) |
|
|
501 | delete *i; |
|
|
502 | } |
|
|
503 | |
|
|
504 | void rxvt_ptytty_server () |
|
|
505 | { |
|
|
506 | int sv[2]; |
|
|
507 | |
|
|
508 | if (socketpair (AF_UNIX, SOCK_STREAM, 0, sv)) |
|
|
509 | rxvt_fatal ("could not create socket to communicate with pty/sessiondb helper, aborting.\n"); |
|
|
510 | |
|
|
511 | pid = fork (); |
|
|
512 | |
|
|
513 | if (pid < 0) |
|
|
514 | rxvt_fatal ("could not create pty/sessiondb helper process, aborting.\n"); |
|
|
515 | |
|
|
516 | if (pid) |
|
|
517 | { |
|
|
518 | // client, urxvt |
|
|
519 | sock_fd = sv[0]; |
|
|
520 | close (sv[1]); |
|
|
521 | fcntl (sock_fd, F_SETFD, FD_CLOEXEC); |
|
|
522 | } |
|
|
523 | else |
|
|
524 | { |
|
|
525 | // server, pty-helper |
|
|
526 | sock_fd = sv[1]; |
|
|
527 | |
|
|
528 | for (int fd = 0; fd < 1023; fd++) |
|
|
529 | if (fd != sock_fd) |
|
|
530 | close (fd); |
|
|
531 | |
|
|
532 | serve (); |
|
|
533 | _exit (EXIT_SUCCESS); |
|
|
534 | } |
|
|
535 | } |
|
|
536 | |
|
|
537 | #endif |
|
|
538 | |
|
|
539 | // a "factory" *g* |
|
|
540 | rxvt_ptytty * |
|
|
541 | rxvt_new_ptytty () |
|
|
542 | { |
|
|
543 | #if PTYTTY_HELPER |
|
|
544 | if (pid > 0) |
|
|
545 | // use helper process |
|
|
546 | return new rxvt_ptytty_proxy; |
|
|
547 | else |
|
|
548 | #endif |
|
|
549 | return new rxvt_ptytty_unix; |
|
|
550 | } |
|
|
551 | |
|
|
552 | /*----------------------- end-of-file (C source) -----------------------*/ |
|
|
553 | |
|
|