[ViewVC] Diff of: cvs/rxvt-unicode/src/ptytty.C

Comparing rxvt-unicode/src/ptytty.C (file contents):
Revision 1.37 by root, Tue Jan 17 11:02:10 2006 UTC vs.
Revision 1.46 by root, Thu Jan 19 09:57:20 2006 UTC

…		…
22	---------------------------------------------------------------------/	22	---------------------------------------------------------------------/
23		23
24	#include "../config.h" /* NECESSARY */	24	#include "../config.h" /* NECESSARY */
25	#include "rxvt.h"	25	#include "rxvt.h"
26		26
27	# include <cstdlib>	27	#include <cstdlib>
28	# include <cstring>	28	#include <cstring>
29		29
30	#ifdef HAVE_SYS_TYPES_H
31	# include <sys/types.h>	30	#include <sys/types.h>
32	#endif	31	#include <sys/socket.h>
33	#ifdef HAVE_UNISTD_H
34	# include <unistd.h>	32	#include <unistd.h>
35	#endif
36	#ifdef HAVE_FCNTL_H
37	# include <fcntl.h>	33	#include <fcntl.h>
38	#endif	34
39	#ifdef HAVE_SYS_IOCTL_H	35	#ifdef HAVE_SYS_IOCTL_H
40	# include <sys/ioctl.h>	36	# include <sys/ioctl.h>
41	#endif	37	#endif
42	#if defined(PTYS_ARE_PTMX) && defined(HAVE_SYS_STROPTS_H)	38	#if defined(PTYS_ARE_PTMX) && defined(HAVE_SYS_STROPTS_H)
43	# include <sys/stropts.h> /* for I_PUSH */	39	# include <sys/stropts.h> /* for I_PUSH */
…		…
53	# include <util.h>	49	# include <util.h>
54	#endif	50	#endif
55		51
56	#include <cstdio>	52	#include <cstdio>
57	#include <grp.h>	53	#include <grp.h>
		54
		55	#include "rxvtutil.h"
		56	#include "fdpass.h"
		57	#include "ptytty.h"
58		58
59	/////////////////////////////////////////////////////////////////////////////	59	/////////////////////////////////////////////////////////////////////////////
60		60
61	/* ------------------------------------------------------------------------- *	61	/* ------------------------------------------------------------------------- *
62	* GET PSEUDO TELETYPE - MASTER AND SLAVE *	62	* GET PSEUDO TELETYPE - MASTER AND SLAVE *
…		…
64	/*	64	/*
65	* Returns pty file descriptor, or -1 on failure	65	* Returns pty file descriptor, or -1 on failure
66	* If successful, ttydev is set to the name of the slave device.	66	* If successful, ttydev is set to the name of the slave device.
67	* fd_tty _may_ also be set to an open fd to the slave device	67	* fd_tty _may_ also be set to an open fd to the slave device
68	*/	68	*/
69	static int	69	static inline int
70	get_pty (int fd_tty, char *ttydev)	70	get_pty_streams (int fd_tty, char *ttydev)
71	{	71	{
		72	#ifdef NO_SETOWNER_TTYDEV
72	int pfd;	73	int pfd;
73		74
		75	# if defined(PTYS_ARE_GETPT)
		76	pfd = getpt();
		77	# elif defined(PTYS_ARE_POSIX)
		78	pfd = posix_openpt (O_RDWR);
		79	# else
		80	pfd = open ("/dev/ptmx", O_RDWR \| O_NOCTTY, 0);
		81	# endif
		82	if (pfd >= 0)
		83	{
		84	if (grantpt (pfd) == 0 /* change slave permissions */
		85	&& unlockpt (pfd) == 0)
		86	{ /* slave now unlocked */
		87	ttydev = strdup (ptsname (pfd)); / get slave's name */
		88	return pfd;
		89	}
		90
		91	close (pfd);
		92	}
		93	#endif
		94
		95	return -1;
		96	}
		97
		98	static inline int
		99	get_pty_openpty (int fd_tty, char *ttydev)
		100	{
74	#ifdef PTYS_ARE_OPENPTY	101	#ifdef PTYS_ARE_OPENPTY
		102	int pfd;
		103	int res;
75	char tty_name[sizeof "/dev/pts/????\0"];	104	char tty_name[sizeof "/dev/pts/????\0"];
76		105
77	rxvt_privileges(RESTORE);
78	int res = openpty (&pfd, fd_tty, tty_name, NULL, NULL);	106	res = openpty (&pfd, fd_tty, tty_name, NULL, NULL);
79	rxvt_privileges(IGNORE);
80
81	if (res != -1)	107	if (res != -1)
82	{	108	{
83	*ttydev = strdup (tty_name);	109	*ttydev = strdup (tty_name);
84	return pfd;	110	return pfd;
85	}	111	}
86	#endif	112	#endif
87		113
		114	return -1;
		115	}
		116
		117	static inline int
		118	get_pty__getpty (int fd_tty, char *ttydev)
		119	{
88	#ifdef PTYS_ARE__GETPTY	120	#ifdef PTYS_ARE__GETPTY
		121	int pfd;
		122
89	*ttydev = _getpty (&pfd, O_RDWR \| O_NONBLOCK \| O_NOCTTY, 0622, 0);	123	*ttydev = _getpty (&pfd, O_RDWR \| O_NONBLOCK \| O_NOCTTY, 0622, 0);
90	if (*ttydev != NULL)	124	if (*ttydev != NULL)
91	return pfd;	125	return pfd;
92	#endif	126	#endif
93		127
94	#if defined(HAVE_GRANTPT) && defined(HAVE_UNLOCKPT)	128	return -1;
95	# if defined(PTYS_ARE_GETPT) \|\| defined(PTYS_ARE_POSIX) \|\| defined(PTYS_ARE_PTMX)	129	}
96		130
97	{	131	static inline int
98	# ifdef PTYS_ARE_GETPT	132	get_pty_ptc (int fd_tty, char *ttydev)
99	pfd = getpt();	133	{
100	# else
101	# ifdef PTYS_ARE_POSIX
102	pfd = posix_openpt (O_RDWR);
103	# else
104	pfd = open ("/dev/ptmx", O_RDWR \| O_NOCTTY, 0);
105	# endif
106	# endif
107
108	if (pfd >= 0)
109	{
110	if (grantpt (pfd) == 0 /* change slave permissions */
111	&& unlockpt (pfd) == 0)
112	{ /* slave now unlocked */
113	ttydev = strdup (ptsname (pfd)); / get slave's name */
114	return pfd;
115	}
116	close (pfd);
117	}
118	}
119	# endif
120	#endif
121
122	#ifdef PTYS_ARE_PTC	134	#ifdef PTYS_ARE_PTC
		135	int pfd;
		136
123	if ((pfd = open ("/dev/ptc", O_RDWR \| O_NOCTTY, 0)) >= 0)	137	if ((pfd = open ("/dev/ptc", O_RDWR \| O_NOCTTY, 0)) >= 0)
124	{	138	{
125	*ttydev = strdup (ttyname (pfd));	139	*ttydev = strdup (ttyname (pfd));
126	return pfd;	140	return pfd;
127	}	141	}
128	#endif	142	#endif
129		143
		144	return -1;
		145	}
		146
		147	static inline int
		148	get_pty_clone (int fd_tty, char *ttydev)
		149	{
130	#ifdef PTYS_ARE_CLONE	150	#ifdef PTYS_ARE_CLONE
		151	int pfd;
		152
131	if ((pfd = open ("/dev/ptym/clone", O_RDWR \| O_NOCTTY, 0)) >= 0)	153	if ((pfd = open ("/dev/ptym/clone", O_RDWR \| O_NOCTTY, 0)) >= 0)
132	{	154	{
133	*ttydev = strdup (ptsname (pfd));	155	*ttydev = strdup (ptsname (pfd));
134	return pfd;	156	return pfd;
135	}	157	}
136	#endif	158	#endif
137		159
		160	return -1;
		161	}
		162
		163	static inline int
		164	get_pty_numeric (int fd_tty, char *ttydev)
		165	{
138	#ifdef PTYS_ARE_NUMERIC	166	#ifdef PTYS_ARE_NUMERIC
139	{	167	int pfd;
140	int idx;	168	int idx;
141	char c1, c2;	169	char c1, c2;
142	char pty_name[] = "/dev/ptyp???";	170	char pty_name[] = "/dev/ptyp???";
143	char tty_name[] = "/dev/ttyp???";	171	char tty_name[] = "/dev/ttyp???";
144		172
145	c1 = &(pty_name[sizeof (pty_name) - 4]);	173	c1 = &(pty_name[sizeof (pty_name) - 4]);
146	c2 = &(tty_name[sizeof (tty_name) - 4]);	174	c2 = &(tty_name[sizeof (tty_name) - 4]);
		175
147	for (idx = 0; idx < 256; idx++)	176	for (idx = 0; idx < 256; idx++)
148	{	177	{
149	sprintf (c1, "%d", idx);	178	sprintf (c1, "%d", idx);
150	sprintf (c2, "%d", idx);	179	sprintf (c2, "%d", idx);
		180
151	if (access (tty_name, F_OK) < 0)	181	if (access (tty_name, F_OK) < 0)
152	{	182	{
153	idx = 256;	183	idx = 256;
154	break;	184	break;
155	}	185	}
156		186
157	if ((pfd = open (pty_name, O_RDWR \| O_NOCTTY, 0)) >= 0)	187	if ((pfd = open (pty_name, O_RDWR \| O_NOCTTY, 0)) >= 0)
158	{	188	{
159	if (access (tty_name, R_OK \| W_OK) == 0)	189	if (access (tty_name, R_OK \| W_OK) == 0)
160	{	190	{
161	*ttydev = strdup (tty_name);	191	*ttydev = strdup (tty_name);
162	return pfd;	192	return pfd;
163	}	193	}
164		194
165	close (pfd);	195	close (pfd);
166	}	196	}
167	}	197	}
168	}
169	#endif	198	#endif
170		199
		200	return -1;
		201	}
		202
		203	static inline int
		204	get_pty_searched (int fd_tty, char *ttydev)
		205	{
171	#ifdef PTYS_ARE_SEARCHED	206	#ifdef PTYS_ARE_SEARCHED
172	{
173	const char c1, c2;
174	char pty_name[] = "/dev/pty??";
175	char tty_name[] = "/dev/tty??";
176
177	# ifndef PTYCHAR1	207	# ifndef PTYCHAR1
178	# define PTYCHAR1 "pqrstuvwxyz"	208	# define PTYCHAR1 "pqrstuvwxyz"
179	# endif	209	# endif
180	# ifndef PTYCHAR2	210	# ifndef PTYCHAR2
181	# define PTYCHAR2 "0123456789abcdef"	211	# define PTYCHAR2 "0123456789abcdef"
182	# endif	212	# endif
		213	int pfd;
		214	const char c1, c2;
		215	char pty_name[] = "/dev/pty??";
		216	char tty_name[] = "/dev/tty??";
183		217
184	for (c1 = PTYCHAR1; *c1; c1++)	218	for (c1 = PTYCHAR1; *c1; c1++)
185	{	219	{
186	pty_name[ (sizeof (pty_name) - 3)] =	220	pty_name[ (sizeof (pty_name) - 3)] =
187	tty_name[ (sizeof (pty_name) - 3)] = *c1;	221	tty_name[ (sizeof (pty_name) - 3)] = *c1;
		222
188	for (c2 = PTYCHAR2; *c2; c2++)	223	for (c2 = PTYCHAR2; *c2; c2++)
189	{	224	{
190	pty_name[ (sizeof (pty_name) - 2)] =	225	pty_name[ (sizeof (pty_name) - 2)] =
191	tty_name[ (sizeof (pty_name) - 2)] = *c2;	226	tty_name[ (sizeof (pty_name) - 2)] = *c2;
		227
192	if ((pfd = open (pty_name, O_RDWR \| O_NOCTTY, 0)) >= 0)	228	if ((pfd = open (pty_name, O_RDWR \| O_NOCTTY, 0)) >= 0)
193	{	229	{
194	if (access (tty_name, R_OK \| W_OK) == 0)	230	if (access (tty_name, R_OK \| W_OK) == 0)
195	{	231	{
196	*ttydev = strdup (tty_name);	232	*ttydev = strdup (tty_name);
197	return pfd;	233	return pfd;
198	}	234	}
199		235
200	close (pfd);	236	close (pfd);
201	}	237	}
202	}	238	}
203	}	239	}
204	}
205	#endif	240	#endif
		241
		242	return -1;
		243	}
		244
		245	static int
		246	get_pty (int fd_tty, char *ttydev)
		247	{
		248	int pfd;
		249
		250	if ((pfd = get_pty_streams (fd_tty, ttydev)) != -1
		251	\|\| (pfd = get_pty_openpty (fd_tty, ttydev)) != -1
		252	\|\| (pfd = get_pty__getpty (fd_tty, ttydev)) != -1
		253	\|\| (pfd = get_pty_ptc (fd_tty, ttydev)) != -1
		254	\|\| (pfd = get_pty_clone (fd_tty, ttydev)) != -1
		255	\|\| (pfd = get_pty_numeric (fd_tty, ttydev)) != -1
		256	\|\| (pfd = get_pty_searched (fd_tty, ttydev)) != -1)
		257	return pfd;
206		258
207	return -1;	259	return -1;
208	}	260	}
209		261
210	/----------------------------------------------------------------------/	262	/----------------------------------------------------------------------/
…		…
222	* Make our tty a controlling tty so that /dev/tty points to us	274	* Make our tty a controlling tty so that /dev/tty points to us
223	*/	275	*/
224	static int	276	static int
225	control_tty (int fd_tty)	277	control_tty (int fd_tty)
226	{	278	{
227	int fd;
228
229	/* ---------------------------------------- */
230	# ifdef HAVE_SETSID
231	setsid ();	279	setsid ();
232	# endif
233	# if defined(HAVE_SETPGID)
234	setpgid (0, 0);
235	# elif defined(HAVE_SETPGRP)
236	setpgrp (0, 0);
237	# endif
238		280
239	/* ---------------------------------------- */
240	# ifdef TIOCNOTTY
241	fd = open ("/dev/tty", O_RDWR \| O_NOCTTY);
242	if (fd >= 0)
243	{
244	ioctl (fd, TIOCNOTTY, NULL); /* void tty associations */
245	close (fd);
246	}
247	# endif
248
249	/* ---------------------------------------- */
250	fd = open ("/dev/tty", O_RDWR \| O_NOCTTY);
251	if (fd >= 0)
252	close (fd); /* ouch: still have controlling tty */
253
254	/* ---------------------------------------- */
255	# if defined(PTYS_ARE_PTMX) && defined(I_PUSH)	281	# if defined(PTYS_ARE_PTMX) && defined(I_PUSH)
256	/*	282	/*
257	* Push STREAMS modules:	283	* Push STREAMS modules:
258	* ptem: pseudo-terminal hardware emulation module.	284	* ptem: pseudo-terminal hardware emulation module.
259	* ldterm: standard terminal line discipline.	285	* ldterm: standard terminal line discipline.
…		…
277	ioctl (fd_tty, I_PUSH, "ptem");	303	ioctl (fd_tty, I_PUSH, "ptem");
278	ioctl (fd_tty, I_PUSH, "ldterm");	304	ioctl (fd_tty, I_PUSH, "ldterm");
279	ioctl (fd_tty, I_PUSH, "ttcompat");	305	ioctl (fd_tty, I_PUSH, "ttcompat");
280	}	306	}
281	# endif	307	# endif
282	/* ---------------------------------------- */	308
283	fd = ioctl (fd_tty, TIOCSCTTY, NULL);	309	ioctl (fd_tty, TIOCSCTTY, NULL);
284	/* ---------------------------------------- */	310
285	fd = open ("/dev/tty", O_WRONLY);	311	int fd = open ("/dev/tty", O_WRONLY);
286	if (fd < 0)	312	if (fd < 0)
287	return -1; /* fatal */	313	return -1; /* fatal */
		314
288	close (fd);	315	close (fd);
289	/* ---------------------------------------- */
290		316
291	return 0;	317	return 0;
292	}	318	}
293		319
294	void	320	void
…		…
365	void	391	void
366	rxvt_ptytty_unix::privileges (rxvt_privaction action)	392	rxvt_ptytty_unix::privileges (rxvt_privaction action)
367	{	393	{
368	if (!name \|\| !*name)	394	if (!name \|\| !*name)
369	return;	395	return;
370
371	rxvt_privileges (RESTORE);
372		396
373	if (action == SAVE)	397	if (action == SAVE)
374	{	398	{
375	# ifndef RESET_TTY_TO_COMMON_DEFAULTS	399	# ifndef RESET_TTY_TO_COMMON_DEFAULTS
376	/* store original tty status for restoration rxvt_clean_exit () -- rgg 04/12/95 */	400	/* store original tty status for restoration rxvt_clean_exit () -- rgg 04/12/95 */
…		…
397	}	421	}
398	# else	422	# else
399	chmod (name, (S_IRUSR \| S_IWUSR \| S_IRGRP \| S_IWGRP \| S_IROTH \| S_IWOTH));	423	chmod (name, (S_IRUSR \| S_IWUSR \| S_IRGRP \| S_IWGRP \| S_IROTH \| S_IWOTH));
400	chown (name, 0, 0);	424	chown (name, 0, 0);
401	# endif	425	# endif
402
403	}	426	}
404
405	rxvt_privileges (IGNORE);
406	}	427	}
407	#endif	428	#endif
408		429
409	rxvt_ptytty_unix::rxvt_ptytty_unix ()	430	rxvt_ptytty_unix::rxvt_ptytty_unix ()
410	{	431	{
…		…
418	#endif	439	#endif
419	}	440	}
420		441
421	rxvt_ptytty_unix::~rxvt_ptytty_unix ()	442	rxvt_ptytty_unix::~rxvt_ptytty_unix ()
422	{	443	{
		444	#if UTMP_SUPPORT
423	logout ();	445	logout ();
		446	#endif
424	put ();	447	put ();
425	}	448	}
426		449
427	void	450	void
428	rxvt_ptytty_unix::put ()	451	rxvt_ptytty_unix::put ()
…		…
463	}	486	}
464		487
465	return true;	488	return true;
466	}	489	}
467		490
		491	#if PTYTTY_HELPER
		492
		493	static int sock_fd;
		494	static int pid;
		495
		496	struct command
		497	{
		498	enum { get, login, destroy } type;
		499
		500	rxvt_ptytty *id;
		501
		502	bool login_shell;
		503	int cmd_pid;
		504	char hostname[512]; // arbitrary, but should be plenty
		505	};
		506
		507	struct rxvt_ptytty_proxy : zero_initialized, rxvt_ptytty
		508	{
		509	rxvt_ptytty *id;
		510
		511	~rxvt_ptytty_proxy ();
		512
		513	bool get ();
		514	void login (int cmd_pid, bool login_shell, const char *hostname);
		515	};
		516
		517	bool
		518	rxvt_ptytty_proxy::get ()
		519	{
		520	command cmd;
		521
		522	cmd.type = command::get;
		523
		524	write (sock_fd, &cmd, sizeof (cmd));
		525
		526	if (read (sock_fd, &id, sizeof (id)) != sizeof (id))
		527	rxvt_fatal ("protocol error while creating pty using helper process, aborting.\n");
		528
		529	if (!id)
		530	return false;
		531
		532	if ((pty = rxvt_recv_fd (sock_fd)) < 0
		533	\|\| (tty = rxvt_recv_fd (sock_fd)) < 0)
		534	rxvt_fatal ("protocol error while reading pty/tty fds from helper process, aborting.\n");
		535
		536	return true;
		537	}
		538
		539	void
		540	rxvt_ptytty_proxy::login (int cmd_pid, bool login_shell, const char *hostname)
		541	{
		542	command cmd;
		543
		544	cmd.type = command::login;
		545	cmd.id = id;
		546	cmd.cmd_pid = cmd_pid;
		547	cmd.login_shell = login_shell;
		548	strncpy (cmd.hostname, hostname, sizeof (cmd.hostname));
		549
		550	write (sock_fd, &cmd, sizeof (cmd));
		551	}
		552
		553	rxvt_ptytty_proxy::~rxvt_ptytty_proxy ()
		554	{
		555	command cmd;
		556
		557	cmd.type = command::destroy;
		558	cmd.id = id;
		559
		560	write (sock_fd, &cmd, sizeof (cmd));
		561	}
		562
		563	static
		564	void serve ()
		565	{
		566	command cmd;
		567	vector<rxvt_ptytty *> ptys;
		568
		569	while (read (sock_fd, &cmd, sizeof (command)) == sizeof (command))
		570	{
		571	if (cmd.type == command::get)
		572	{
		573	// -> id ptyfd ttyfd
		574	cmd.id = new rxvt_ptytty_unix;
		575
		576	if (cmd.id->get ())
		577	{
		578	write (sock_fd, &cmd.id, sizeof (cmd.id));
		579	ptys.push_back (cmd.id);
		580
		581	rxvt_send_fd (sock_fd, cmd.id->pty);
		582	rxvt_send_fd (sock_fd, cmd.id->tty);
		583	}
		584	else
		585	{
		586	delete cmd.id;
		587	cmd.id = 0;
		588	write (sock_fd, &cmd.id, sizeof (cmd.id));
		589	}
		590	}
		591	else if (cmd.type == command::login)
		592	{
		593	#if UTMP_SUPPORT
		594	if (find (ptys.begin (), ptys.end (), cmd.id))
		595	{
		596	cmd.hostname[sizeof (cmd.hostname) - 1] = 0;
		597	cmd.id->login (cmd.cmd_pid, cmd.login_shell, cmd.hostname);
		598	}
		599	#endif
		600	}
		601	else if (cmd.type == command::destroy)
		602	{
		603	rxvt_ptytty **pty = find (ptys.begin (), ptys.end (), cmd.id);
		604
		605	if (pty)
		606	{
		607	delete *pty;
		608	ptys.erase (pty);
		609	}
		610	}
		611	else
		612	break;
		613	}
		614
		615	// destroy all ptys
		616	for (rxvt_ptytty **i = ptys.end (); i-- > ptys.begin (); )
		617	delete *i;
		618	}
		619
		620	void rxvt_ptytty_server ()
		621	{
		622	int sv[2];
		623
		624	if (socketpair (AF_UNIX, SOCK_STREAM, 0, sv))
		625	rxvt_fatal ("could not create socket to communicate with pty/sessiondb helper, aborting.\n");
		626
		627	pid = fork ();
		628
		629	if (pid < 0)
		630	rxvt_fatal ("could not create pty/sessiondb helper process, aborting.\n");
		631
		632	if (pid)
		633	{
		634	// client, urxvt
		635	sock_fd = sv[0];
		636	close (sv[1]);
		637	fcntl (sock_fd, F_SETFD, FD_CLOEXEC);
		638	}
		639	else
		640	{
		641	// server, pty-helper
		642	sock_fd = sv[1];
		643
		644	for (int fd = 0; fd < 1023; fd++)
		645	if (fd != sock_fd)
		646	close (fd);
		647
		648	serve ();
		649	_exit (EXIT_SUCCESS);
		650	}
		651	}
		652	#endif
		653
		654	// a "factory" g
		655	rxvt_ptytty *
		656	rxvt_new_ptytty ()
		657	{
		658	#if PTYTTY_HELPER
		659	if (pid > 0)
		660	// use helper process
		661	return new rxvt_ptytty_proxy;
		662	else
		663	#endif
		664	return new rxvt_ptytty_unix;
		665	}
		666
468	/----------------------- end-of-file (C source) -----------------------/	667	/----------------------- end-of-file (C source) -----------------------/
469		668

Diff Legend

-–
+Removed lines
-+
+Added lines
-<
+Changed lines
->
+Changed lines

Comparing rxvt-unicode/src/ptytty.C (file contents): Revision 1.37 by root, Tue Jan 17 11:02:10 2006 UTC vs. Revision 1.46 by root, Thu Jan 19 09:57:20 2006 UTC

Diff Legend

Comparing rxvt-unicode/src/ptytty.C (file contents):
Revision 1.37 by root, Tue Jan 17 11:02:10 2006 UTC vs.
Revision 1.46 by root, Thu Jan 19 09:57:20 2006 UTC