|
|
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> |
… | |
… | |
19 | * You should have received a copy of the GNU General Public License |
22 | * You should have received a copy of the GNU General Public License |
20 | * along with this program; if not, write to the Free Software |
23 | * along with this program; if not, write to the Free Software |
21 | * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. |
24 | * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. |
22 | *---------------------------------------------------------------------*/ |
25 | *---------------------------------------------------------------------*/ |
23 | |
26 | |
24 | #include "../config.h" /* NECESSARY */ |
27 | #include "../config.h" |
25 | |
28 | |
26 | #include "fdpass.h" |
29 | #include "fdpass.h" |
27 | #include "ptytty.h" |
30 | #include "ptytty.h" |
28 | |
31 | |
29 | #include <cstdlib> |
32 | #include <cstdlib> |
… | |
… | |
240 | |
243 | |
241 | return 0; |
244 | return 0; |
242 | } |
245 | } |
243 | |
246 | |
244 | void |
247 | void |
245 | rxvt_ptytty::close_tty () |
248 | ptytty::close_tty () |
246 | { |
249 | { |
247 | if (tty < 0) |
250 | if (tty < 0) |
248 | return; |
251 | return; |
249 | |
252 | |
250 | close (tty); |
253 | close (tty); |
251 | tty = -1; |
254 | tty = -1; |
252 | } |
255 | } |
253 | |
256 | |
254 | bool |
257 | bool |
255 | rxvt_ptytty::make_controlling_tty () |
258 | ptytty::make_controlling_tty () |
256 | { |
259 | { |
257 | return control_tty (tty) >= 0; |
260 | return control_tty (tty) >= 0; |
258 | } |
261 | } |
259 | |
262 | |
260 | void |
263 | void |
261 | rxvt_ptytty::set_utf8_mode (bool on) |
264 | ptytty::set_utf8_mode (bool on) |
262 | { |
265 | { |
263 | #ifdef IUTF8 |
266 | #ifdef IUTF8 |
264 | if (pty < 0) |
267 | if (pty < 0) |
265 | return; |
268 | return; |
266 | |
269 | |
… | |
… | |
305 | gid = 0; |
308 | gid = 0; |
306 | } |
309 | } |
307 | } |
310 | } |
308 | } ttyconf; |
311 | } ttyconf; |
309 | |
312 | |
310 | rxvt_ptytty_unix::rxvt_ptytty_unix () |
313 | ptytty_unix::ptytty_unix () |
311 | { |
314 | { |
312 | name = 0; |
315 | name = 0; |
313 | #if UTMP_SUPPORT |
316 | #if UTMP_SUPPORT |
314 | cmd_pid = 0; |
317 | cmd_pid = 0; |
315 | #endif |
318 | #endif |
316 | } |
319 | } |
317 | |
320 | |
318 | rxvt_ptytty_unix::~rxvt_ptytty_unix () |
321 | ptytty_unix::~ptytty_unix () |
319 | { |
322 | { |
320 | #if UTMP_SUPPORT |
323 | #if UTMP_SUPPORT |
321 | logout (); |
324 | logout (); |
322 | #endif |
325 | #endif |
323 | put (); |
326 | put (); |
324 | } |
327 | } |
325 | |
328 | |
326 | void |
329 | void |
327 | rxvt_ptytty_unix::put () |
330 | ptytty_unix::put () |
328 | { |
331 | { |
329 | chmod (name, RESTORE_TTY_MODE); |
332 | chmod (name, RESTORE_TTY_MODE); |
330 | chown (name, 0, ttyconf.gid); |
333 | chown (name, 0, ttyconf.gid); |
331 | |
334 | |
332 | close_tty (); |
335 | close_tty (); |
… | |
… | |
339 | pty = tty = -1; |
342 | pty = tty = -1; |
340 | name = 0; |
343 | name = 0; |
341 | } |
344 | } |
342 | |
345 | |
343 | bool |
346 | bool |
344 | rxvt_ptytty_unix::get () |
347 | ptytty_unix::get () |
345 | { |
348 | { |
346 | /* get master (pty) */ |
349 | /* get master (pty) */ |
347 | if ((pty = get_pty (&tty, &name)) < 0) |
350 | if ((pty = get_pty (&tty, &name)) < 0) |
348 | return false; |
351 | return false; |
349 | |
352 | |
… | |
… | |
377 | |
380 | |
378 | struct command |
381 | struct command |
379 | { |
382 | { |
380 | enum { get, login, destroy } type; |
383 | enum { get, login, destroy } type; |
381 | |
384 | |
382 | rxvt_ptytty *id; |
385 | ptytty *id; |
383 | |
386 | |
384 | bool login_shell; |
387 | bool login_shell; |
385 | int cmd_pid; |
388 | int cmd_pid; |
386 | char hostname[512]; // arbitrary, but should be plenty |
389 | char hostname[512]; // arbitrary, but should be plenty |
387 | }; |
390 | }; |
388 | |
391 | |
389 | struct rxvt_ptytty_proxy : zero_initialized, rxvt_ptytty |
392 | struct ptytty_proxy : ptytty |
390 | { |
393 | { |
391 | rxvt_ptytty *id; |
394 | ptytty *id; |
392 | |
395 | |
|
|
396 | ptytty_proxy () |
|
|
397 | : id(0) |
|
|
398 | { |
|
|
399 | } |
|
|
400 | |
393 | ~rxvt_ptytty_proxy (); |
401 | ~ptytty_proxy (); |
394 | |
402 | |
395 | bool get (); |
403 | bool get (); |
396 | void login (int cmd_pid, bool login_shell, const char *hostname); |
404 | void login (int cmd_pid, bool login_shell, const char *hostname); |
397 | }; |
405 | }; |
398 | |
406 | |
399 | bool |
407 | bool |
400 | rxvt_ptytty_proxy::get () |
408 | ptytty_proxy::get () |
401 | { |
409 | { |
402 | command cmd; |
410 | command cmd; |
403 | |
411 | |
404 | cmd.type = command::get; |
412 | cmd.type = command::get; |
405 | |
413 | |
406 | write (sock_fd, &cmd, sizeof (cmd)); |
414 | write (sock_fd, &cmd, sizeof (cmd)); |
407 | |
415 | |
408 | if (read (sock_fd, &id, sizeof (id)) != sizeof (id)) |
416 | if (read (sock_fd, &id, sizeof (id)) != sizeof (id)) |
409 | rxvt_fatal ("protocol error while creating pty using helper process, aborting.\n"); |
417 | fatal ("protocol error while creating pty using helper process, aborting.\n"); |
410 | |
418 | |
411 | if (!id) |
419 | if (!id) |
412 | return false; |
420 | return false; |
413 | |
421 | |
414 | if ((pty = rxvt_recv_fd (sock_fd)) < 0 |
422 | if ((pty = ptytty_recv_fd (sock_fd)) < 0 |
415 | || (tty = rxvt_recv_fd (sock_fd)) < 0) |
423 | || (tty = ptytty_recv_fd (sock_fd)) < 0) |
416 | rxvt_fatal ("protocol error while reading pty/tty fds from helper process, aborting.\n"); |
424 | fatal ("protocol error while reading pty/tty fds from helper process, aborting.\n"); |
417 | |
425 | |
418 | return true; |
426 | return true; |
419 | } |
427 | } |
420 | |
428 | |
421 | void |
429 | void |
422 | rxvt_ptytty_proxy::login (int cmd_pid, bool login_shell, const char *hostname) |
430 | ptytty_proxy::login (int cmd_pid, bool login_shell, const char *hostname) |
423 | { |
431 | { |
424 | command cmd; |
432 | command cmd; |
425 | |
433 | |
426 | cmd.type = command::login; |
434 | cmd.type = command::login; |
427 | cmd.id = id; |
435 | cmd.id = id; |
… | |
… | |
430 | strncpy (cmd.hostname, hostname, sizeof (cmd.hostname)); |
438 | strncpy (cmd.hostname, hostname, sizeof (cmd.hostname)); |
431 | |
439 | |
432 | write (sock_fd, &cmd, sizeof (cmd)); |
440 | write (sock_fd, &cmd, sizeof (cmd)); |
433 | } |
441 | } |
434 | |
442 | |
435 | rxvt_ptytty_proxy::~rxvt_ptytty_proxy () |
443 | ptytty_proxy::~ptytty_proxy () |
436 | { |
444 | { |
|
|
445 | if (id) |
|
|
446 | { |
437 | command cmd; |
447 | command cmd; |
438 | |
448 | |
439 | cmd.type = command::destroy; |
449 | cmd.type = command::destroy; |
440 | cmd.id = id; |
450 | cmd.id = id; |
441 | |
451 | |
442 | write (sock_fd, &cmd, sizeof (cmd)); |
452 | write (sock_fd, &cmd, sizeof (cmd)); |
|
|
453 | } |
443 | } |
454 | } |
444 | |
455 | |
445 | static |
456 | static |
446 | void serve () |
457 | void serve () |
447 | { |
458 | { |
448 | command cmd; |
459 | command cmd; |
449 | vector<rxvt_ptytty *> ptys; |
460 | vector<ptytty *> ptys; |
450 | |
461 | |
451 | while (read (sock_fd, &cmd, sizeof (command)) == sizeof (command)) |
462 | while (read (sock_fd, &cmd, sizeof (command)) == sizeof (command)) |
452 | { |
463 | { |
453 | if (cmd.type == command::get) |
464 | if (cmd.type == command::get) |
454 | { |
465 | { |
455 | // -> id ptyfd ttyfd |
466 | // -> id ptyfd ttyfd |
456 | cmd.id = new rxvt_ptytty_unix; |
467 | cmd.id = new ptytty_unix; |
457 | |
468 | |
458 | if (cmd.id->get ()) |
469 | if (cmd.id->get ()) |
459 | { |
470 | { |
460 | write (sock_fd, &cmd.id, sizeof (cmd.id)); |
471 | write (sock_fd, &cmd.id, sizeof (cmd.id)); |
461 | ptys.push_back (cmd.id); |
472 | ptys.push_back (cmd.id); |
462 | |
473 | |
463 | rxvt_send_fd (sock_fd, cmd.id->pty); |
474 | ptytty_send_fd (sock_fd, cmd.id->pty); |
464 | rxvt_send_fd (sock_fd, cmd.id->tty); |
475 | ptytty_send_fd (sock_fd, cmd.id->tty); |
465 | } |
476 | } |
466 | else |
477 | else |
467 | { |
478 | { |
468 | delete cmd.id; |
479 | delete cmd.id; |
469 | cmd.id = 0; |
480 | cmd.id = 0; |
… | |
… | |
480 | } |
491 | } |
481 | #endif |
492 | #endif |
482 | } |
493 | } |
483 | else if (cmd.type == command::destroy) |
494 | else if (cmd.type == command::destroy) |
484 | { |
495 | { |
485 | rxvt_ptytty **pty = find (ptys.begin (), ptys.end (), cmd.id); |
496 | ptytty **pty = find (ptys.begin (), ptys.end (), cmd.id); |
486 | |
497 | |
487 | if (pty) |
498 | if (pty) |
488 | { |
499 | { |
489 | delete *pty; |
500 | delete *pty; |
490 | ptys.erase (pty); |
501 | ptys.erase (pty); |
… | |
… | |
493 | else |
504 | else |
494 | break; |
505 | break; |
495 | } |
506 | } |
496 | |
507 | |
497 | // destroy all ptys |
508 | // destroy all ptys |
498 | for (rxvt_ptytty **i = ptys.end (); i-- > ptys.begin (); ) |
509 | for (ptytty **i = ptys.end (); i-- > ptys.begin (); ) |
499 | delete *i; |
510 | delete *i; |
500 | } |
511 | } |
501 | |
512 | |
502 | void rxvt_ptytty_server () |
513 | void ptytty_server () |
503 | { |
514 | { |
504 | int sv[2]; |
515 | int sv[2]; |
505 | |
516 | |
506 | if (socketpair (AF_UNIX, SOCK_STREAM, 0, sv)) |
517 | if (socketpair (AF_UNIX, SOCK_STREAM, 0, sv)) |
507 | rxvt_fatal ("could not create socket to communicate with pty/sessiondb helper, aborting.\n"); |
518 | fatal ("could not create socket to communicate with pty/sessiondb helper, aborting.\n"); |
508 | |
519 | |
509 | pid = fork (); |
520 | pid = fork (); |
510 | |
521 | |
511 | if (pid < 0) |
522 | if (pid < 0) |
512 | rxvt_fatal ("could not create pty/sessiondb helper process, aborting.\n"); |
523 | fatal ("could not create pty/sessiondb helper process, aborting.\n"); |
513 | |
524 | |
514 | if (pid) |
525 | if (pid) |
515 | { |
526 | { |
516 | // client, urxvt |
527 | // client, process |
517 | sock_fd = sv[0]; |
528 | sock_fd = sv[0]; |
518 | close (sv[1]); |
529 | close (sv[1]); |
519 | fcntl (sock_fd, F_SETFD, FD_CLOEXEC); |
530 | fcntl (sock_fd, F_SETFD, FD_CLOEXEC); |
520 | } |
531 | } |
521 | else |
532 | else |
… | |
… | |
533 | } |
544 | } |
534 | |
545 | |
535 | #endif |
546 | #endif |
536 | |
547 | |
537 | // a "factory" *g* |
548 | // a "factory" *g* |
538 | rxvt_ptytty * |
549 | ptytty * |
539 | rxvt_new_ptytty () |
550 | new_ptytty () |
540 | { |
551 | { |
541 | #if PTYTTY_HELPER |
552 | #if PTYTTY_HELPER |
542 | if (pid > 0) |
553 | if (pid > 0) |
543 | // use helper process |
554 | // use helper process |
544 | return new rxvt_ptytty_proxy; |
555 | return new ptytty_proxy; |
545 | else |
556 | else |
546 | #endif |
557 | #endif |
547 | return new rxvt_ptytty_unix; |
558 | return new ptytty_unix; |
548 | } |
559 | } |
549 | |
560 | |
550 | /*----------------------- end-of-file (C source) -----------------------*/ |
561 | /*----------------------- end-of-file (C source) -----------------------*/ |
551 | |
562 | |