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

Comparing rxvt-unicode/src/ptytty.C (file contents):
Revision 1.44 by ayin, Tue Jan 17 17:59:31 2006 UTC vs.
Revision 1.68 by ayin, Tue Dec 11 21:49:12 2007 UTC

…		…
		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>
37	#endif	42	#endif
38	#if defined(PTYS_ARE_PTMX) && defined(HAVE_SYS_STROPTS_H)	43	#if defined(HAVE_DEV_PTMX) && defined(HAVE_SYS_STROPTS_H)
39	# include <sys/stropts.h> /* for I_PUSH */	44	# include <sys/stropts.h> /* for I_PUSH */
40	#endif	45	#endif
41	#ifdef HAVE_ISASTREAM	46	#ifdef HAVE_ISASTREAM
42	# include <stropts.h>	47	# include <stropts.h>
43	#endif	48	#endif
…		…
46	#elif defined(HAVE_LIBUTIL_H)	51	#elif defined(HAVE_LIBUTIL_H)
47	# include <libutil.h>	52	# include <libutil.h>
48	#elif defined(HAVE_UTIL_H)	53	#elif defined(HAVE_UTIL_H)
49	# include <util.h>	54	# include <util.h>
50	#endif	55	#endif
		56	#ifdef TTY_GID_SUPPORT
		57	#include <grp.h>
		58	#endif
51		59
52	#include <cstdio>	60	#include <cstdio>
53	#include <grp.h>
54
55	#include "rxvtutil.h"
56	#include "fdpass.h"
57	#include "ptytty.h"
58		61
59	/////////////////////////////////////////////////////////////////////////////	62	/////////////////////////////////////////////////////////////////////////////
60		63
61	/* ------------------------------------------------------------------------- *	64	/* ------------------------------------------------------------------------- *
62	* GET PSEUDO TELETYPE - MASTER AND SLAVE *	65	* GET PSEUDO TELETYPE - MASTER AND SLAVE *
63	* ------------------------------------------------------------------------- */	66	* ------------------------------------------------------------------------- */
64	/*	67	/*
65	* Returns pty file descriptor, or -1 on failure	68	* Returns pty file descriptor, or -1 on failure
66	* If successful, ttydev is set to the name of the slave device.	69	* 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	70	* fd_tty _may_ also be set to an open fd to the slave device
68	*/	71	*/
69	static inline int	72	#if defined(UNIX98_PTY)
		73
		74	static int
70	get_pty_streams (int fd_tty, char *ttydev)	75	get_pty (int fd_tty, char *ttydev)
71	{	76	{
72	#ifdef NO_SETOWNER_TTYDEV
73	int pfd;	77	int pfd;
74		78
75	# ifdef PTYS_ARE_GETPT	79	# if defined(HAVE_GETPT)
76	pfd = getpt();	80	pfd = getpt();
		81	# elif defined(HAVE_POSIX_OPENPT)
		82	pfd = posix_openpt (O_RDWR);
77	# else	83	# else
78	# ifdef PTYS_ARE_POSIX
79	pfd = posix_openpt (O_RDWR);
80	# else
81	pfd = open ("/dev/ptmx", O_RDWR \| O_NOCTTY, 0);	84	pfd = open (CLONE_DEVICE, O_RDWR \| O_NOCTTY, 0);
82	# endif	85	# endif
83	# endif	86
84	if (pfd >= 0)	87	if (pfd >= 0)
85	{	88	{
86	if (grantpt (pfd) == 0 /* change slave permissions */	89	if (grantpt (pfd) == 0 /* change slave permissions */
87	&& unlockpt (pfd) == 0)	90	&& unlockpt (pfd) == 0)
		91	{
88	{ /* slave now unlocked */	92	/* slave now unlocked */
89	ttydev = strdup (ptsname (pfd)); / get slave's name */	93	ttydev = strdup (ptsname (pfd)); / get slave's name */
90	return pfd;	94	return pfd;
91	}	95	}
		96
92	close (pfd);	97	close (pfd);
93	}	98	}
94	#endif	99
95	return -1;	100	return -1;
96	}	101	}
97		102
98	static inline int	103	#elif defined(HAVE_OPENPTY)
		104
		105	static int
99	get_pty_openpty (int fd_tty, char *ttydev)	106	get_pty (int fd_tty, char *ttydev)
100	{	107	{
101	#ifdef PTYS_ARE_OPENPTY
102	int pfd;	108	int pfd;
103	int res;	109	int res;
104	char tty_name[sizeof "/dev/pts/????\0"];	110
105
106	res = openpty (&pfd, fd_tty, tty_name, NULL, NULL);	111	res = openpty (&pfd, fd_tty, NULL, NULL, NULL);
		112
107	if (res != -1)	113	if (res != -1)
108	{	114	{
		115	ttydev = strdup (ttyname (fd_tty));
		116	return pfd;
		117	}
		118
		119	return -1;
		120	}
		121
		122	#elif defined(HAVE__GETPTY)
		123
		124	static int
		125	get_pty (int fd_tty, char *ttydev)
		126	{
		127	int pfd;
		128	char *slave;
		129
		130	slave = _getpty (&pfd, O_RDWR \| O_NONBLOCK \| O_NOCTTY, 0622, 0);
		131
		132	if (slave != NULL) {
109	*ttydev = strdup (tty_name);	133	*ttydev = strdup (slave);
110	return pfd;	134	return pfd;
111	}	135	}
112	#endif	136
113	return -1;	137	return -1;
114	}	138	}
115		139
116	static inline int	140	#elif defined(HAVE_DEV_PTC)
117	get_pty__getpty (int fd_tty, char *ttydev)
118	{
119	#ifdef PTYS_ARE__GETPTY
120	int pfd;
121		141
122	*ttydev = _getpty (&pfd, O_RDWR \| O_NONBLOCK \| O_NOCTTY, 0622, 0);	142	static int
123	if (*ttydev != NULL)
124	return pfd;
125	#endif
126	return -1;
127	}
128
129	static inline int
130	get_pty_ptc (int fd_tty, char *ttydev)	143	get_pty (int fd_tty, char *ttydev)
131	{	144	{
132	#ifdef PTYS_ARE_PTC
133	int pfd;	145	int pfd;
134		146
135	if ((pfd = open ("/dev/ptc", O_RDWR \| O_NOCTTY, 0)) >= 0)	147	if ((pfd = open ("/dev/ptc", O_RDWR \| O_NOCTTY, 0)) >= 0)
136	{
137	*ttydev = strdup (ttyname (pfd));
138	return pfd;
139	}
140	#endif
141	return -1;
142	}
143
144	static inline int
145	get_pty_clone (int fd_tty, char *ttydev)
146	{
147	#ifdef PTYS_ARE_CLONE
148	int pfd;
149
150	if ((pfd = open ("/dev/ptym/clone", O_RDWR \| O_NOCTTY, 0)) >= 0)
151	{
152	*ttydev = strdup (ptsname (pfd));
153	return pfd;
154	}
155	#endif
156	return -1;
157	}
158
159	static inline int
160	get_pty_numeric (int fd_tty, char *ttydev)
161	{
162	#ifdef PTYS_ARE_NUMERIC
163	int pfd;
164	int idx;
165	char c1, c2;
166	char pty_name[] = "/dev/ptyp???";
167	char tty_name[] = "/dev/ttyp???";
168
169	c1 = &(pty_name[sizeof (pty_name) - 4]);
170	c2 = &(tty_name[sizeof (tty_name) - 4]);
171	for (idx = 0; idx < 256; idx++)
172	{	148	{
173	sprintf (c1, "%d", idx);	149	*ttydev = strdup (ttyname (pfd));
174	sprintf (c2, "%d", idx);	150	return pfd;
175	if (access (tty_name, F_OK) < 0)	151	}
		152
		153	return -1;
		154	}
		155
		156	#elif defined(HAVE_DEV_CLONE)
		157
		158	static int
		159	get_pty (int fd_tty, char *ttydev)
		160	{
		161	int pfd;
		162
		163	if ((pfd = open ("/dev/ptym/clone", O_RDWR \| O_NOCTTY, 0)) >= 0)
		164	{
		165	*ttydev = strdup (ptsname (pfd));
		166	return pfd;
		167	}
		168
		169	return -1;
		170	}
		171
		172	#else
		173
		174	/* Based on the code in openssh/openbsd-compat/bsd-openpty.c */
		175	static int
		176	get_pty (int fd_tty, char *ttydev)
		177	{
		178	int pfd;
		179	int i;
		180	char pty_name[32];
		181	char tty_name[32];
		182	const char *majors = "pqrstuvwxyzabcde";
		183	const char *minors = "0123456789abcdef";
		184
		185	for (i = 0; i < 256; i++)
		186	{
		187	snprintf(pty_name, 32, "/dev/pty%c%c", majors[i / 16], minors[i % 16]);
		188	snprintf(tty_name, 32, "/dev/tty%c%c", majors[i / 16], minors[i % 16]);
		189
		190	if ((pfd = open (pty_name, O_RDWR \| O_NOCTTY, 0)) == -1)
176	{	191	{
177	idx = 256;	192	snprintf(pty_name, 32, "/dev/ptyp%d", i);
178	break;	193	snprintf(tty_name, 32, "/dev/ttyp%d", i);
		194	if ((pfd = open (pty_name, O_RDWR \| O_NOCTTY, 0)) == -1)
		195	continue;
179	}	196	}
180		197
181	if ((pfd = open (pty_name, O_RDWR \| O_NOCTTY, 0)) >= 0)	198	if (access (tty_name, R_OK \| W_OK) == 0)
182	{	199	{
183	if (access (tty_name, R_OK \| W_OK) == 0)
184	{
185	*ttydev = strdup (tty_name);	200	*ttydev = strdup (tty_name);
186	return pfd;	201	return pfd;
187	}
188
189	close (pfd);
190	}	202	}
		203
		204	close (pfd);
191	}	205	}
192	#endif
193	return -1;
194	}
195		206
196	static inline int
197	get_pty_searched (int fd_tty, char *ttydev)
198	{
199	#ifdef PTYS_ARE_SEARCHED
200	# ifndef PTYCHAR1
201	# define PTYCHAR1 "pqrstuvwxyz"
202	# endif
203	# ifndef PTYCHAR2
204	# define PTYCHAR2 "0123456789abcdef"
205	# endif
206	int pfd;
207	const char c1, c2;
208	char pty_name[] = "/dev/pty??";
209	char tty_name[] = "/dev/tty??";
210
211	for (c1 = PTYCHAR1; *c1; c1++)
212	{
213	pty_name[ (sizeof (pty_name) - 3)] =
214	tty_name[ (sizeof (pty_name) - 3)] = *c1;
215	for (c2 = PTYCHAR2; *c2; c2++)
216	{
217	pty_name[ (sizeof (pty_name) - 2)] =
218	tty_name[ (sizeof (pty_name) - 2)] = *c2;
219	if ((pfd = open (pty_name, O_RDWR \| O_NOCTTY, 0)) >= 0)
220	{
221	if (access (tty_name, R_OK \| W_OK) == 0)
222	{
223	*ttydev = strdup (tty_name);
224	return pfd;
225	}
226
227	close (pfd);
228	}
229	}
230	}
231	#endif
232	return -1;
233	}
234
235	static int
236	get_pty (int fd_tty, char *ttydev)
237	{
238	int pfd;
239
240	if ((pfd = get_pty_streams (fd_tty, ttydev)) != -1
241	\|\| (pfd = get_pty_openpty (fd_tty, ttydev)) != -1
242	\|\| (pfd = get_pty__getpty (fd_tty, ttydev)) != -1
243	\|\| (pfd = get_pty_ptc (fd_tty, ttydev)) != -1
244	\|\| (pfd = get_pty_clone (fd_tty, ttydev)) != -1
245	\|\| (pfd = get_pty_numeric (fd_tty, ttydev)) != -1
246	\|\| (pfd = get_pty_searched (fd_tty, ttydev)) != -1)
247	return pfd;
248	return -1;	207	return -1;
249	}	208	}
		209
		210	#endif
250		211
251	/----------------------------------------------------------------------/	212	/----------------------------------------------------------------------/
252	/*	213	/*
253	* Returns tty file descriptor, or -1 on failure	214	* Returns tty file descriptor, or -1 on failure
254	*/	215	*/
255	static int	216	static int
256	get_tty (char *ttydev)	217	get_tty (char *ttydev)
257	{	218	{
258	return open (ttydev, O_RDWR \| O_NOCTTY, 0);	219	return open (ttydev, O_RDWR \| O_NOCTTY, 0);
…		…
265	static int	226	static int
266	control_tty (int fd_tty)	227	control_tty (int fd_tty)
267	{	228	{
268	int fd;	229	int fd;
269		230
270	/* ---------------------------------------- */
271	setsid ();	231	setsid ();
272		232
273	/* ---------------------------------------- */
274	# if defined(PTYS_ARE_PTMX) && defined(I_PUSH)	233	#if defined(HAVE_DEV_PTMX) && defined(I_PUSH)
275	/*	234	/*
276	* Push STREAMS modules:	235	* Push STREAMS modules:
277	* ptem: pseudo-terminal hardware emulation module.	236	* ptem: pseudo-terminal hardware emulation module.
278	* ldterm: standard terminal line discipline.	237	* ldterm: standard terminal line discipline.
279	* ttcompat: V7, 4BSD and XENIX STREAMS compatibility module.	238	* ttcompat: V7, 4BSD and XENIX STREAMS compatibility module.
…		…
287	* documentation is really unclear about whether it is any close () on	246	* documentation is really unclear about whether it is any close () on
288	* the master side or the last close () - i.e. a proper STREAMS dismantling	247	* the master side or the last close () - i.e. a proper STREAMS dismantling
289	* close () - on the master side which causes a hang up to be sent	248	* close () - on the master side which causes a hang up to be sent
290	* through - Geoff Wing	249	* through - Geoff Wing
291	*/	250	*/
292	# ifdef HAVE_ISASTREAM	251	# ifdef HAVE_ISASTREAM
293	if (isastream (fd_tty) == 1)	252	if (isastream (fd_tty) == 1)
294	# endif	253	# endif
295	{	254	{
296	ioctl (fd_tty, I_PUSH, "ptem");	255	ioctl (fd_tty, I_PUSH, "ptem");
297	ioctl (fd_tty, I_PUSH, "ldterm");	256	ioctl (fd_tty, I_PUSH, "ldterm");
298	ioctl (fd_tty, I_PUSH, "ttcompat");	257	ioctl (fd_tty, I_PUSH, "ttcompat");
299	}	258	}
300	# endif	259	#endif
301	/* ---------------------------------------- */	260
		261	#ifdef TIOCSCTTY
302	fd = ioctl (fd_tty, TIOCSCTTY, NULL);	262	ioctl (fd_tty, TIOCSCTTY, NULL);
303	/* ---------------------------------------- */	263	#else
		264	fd = open (ttyname (fd_tty), O_RDWR);
		265	if (fd >= 0)
		266	close (fd);
		267	#endif
		268
304	fd = open ("/dev/tty", O_WRONLY);	269	fd = open ("/dev/tty", O_WRONLY);
305	if (fd < 0)	270	if (fd < 0)
306	return -1; /* fatal */	271	return -1; /* fatal */
		272
307	close (fd);	273	close (fd);
308	/* ---------------------------------------- */
309		274
310	return 0;	275	return 0;
311	}	276	}
312		277
313	void	278	void
314	rxvt_ptytty::close_tty ()	279	ptytty::close_tty ()
315	{	280	{
316	if (tty < 0)	281	if (tty < 0)
317	return;	282	return;
318		283
319	close (tty);	284	close (tty);
320	tty = -1;	285	tty = -1;
321	}	286	}
322		287
323	bool	288	bool
324	rxvt_ptytty::make_controlling_tty ()	289	ptytty::make_controlling_tty ()
325	{	290	{
326	return control_tty (tty) >= 0;	291	return control_tty (tty) >= 0;
327	}	292	}
328		293
329	void	294	void
330	rxvt_ptytty::set_utf8_mode (bool on)	295	ptytty::set_utf8_mode (bool on)
331	{	296	{
332	#ifdef IUTF8	297	#ifdef IUTF8
333	if (pty < 0)	298	if (pty < 0)
334	return;	299	return;
335		300
…		…
351	}	316	}
352	}	317	}
353	#endif	318	#endif
354	}	319	}
355		320
356	/////////////////////////////////////////////////////////////////////////////
357
358	#ifndef NO_SETOWNER_TTYDEV
359	static struct ttyconf {	321	static struct ttyconf {
360	gid_t gid;	322	gid_t gid;
361	mode_t mode;	323	mode_t mode;
362		324
363	ttyconf ()	325	ttyconf ()
364	{	326	{
365	#ifdef TTY_GID_SUPPORT	327	#ifdef TTY_GID_SUPPORT
366	struct group *gr = getgrnam ("tty");	328	struct group *gr = getgrnam ("tty");
367		329
368	if (gr)	330	if (gr)
		331	{
369	{ /* change group ownership of tty to "tty" */	332	/* change group ownership of tty to "tty" */
370	mode = S_IRUSR \| S_IWUSR \| S_IWGRP;	333	mode = S_IRUSR \| S_IWUSR \| S_IWGRP;
371	gid = gr->gr_gid;	334	gid = gr->gr_gid;
372	}	335	}
373	else	336	else
374	#endif /* TTY_GID_SUPPORT */	337	#endif /* TTY_GID_SUPPORT */
375	{	338	{
376	mode = S_IRUSR \| S_IWUSR \| S_IWGRP \| S_IWOTH;	339	mode = S_IRUSR \| S_IWUSR \| S_IWGRP \| S_IWOTH;
377	gid = getgid ();	340	gid = 0;
378	}	341	}
379	}	342	}
380	} ttyconf;	343	} ttyconf;
381		344
382	/////////////////////////////////////////////////////////////////////////////	345	ptytty_unix::ptytty_unix ()
		346	{
		347	name = 0;
		348	#if UTMP_SUPPORT
		349	cmd_pid = 0;
		350	#endif
		351	}
		352
		353	ptytty_unix::~ptytty_unix ()
		354	{
		355	#if UTMP_SUPPORT
		356	logout ();
		357	#endif
		358	put ();
		359	}
383		360
384	void	361	void
385	rxvt_ptytty_unix::privileges (rxvt_privaction action)	362	ptytty_unix::put ()
386	{	363	{
387	if (!name \|\| !*name)	364	if (name)
388	return;
389
390	if (action == SAVE)
391	{	365	{
392	# ifndef RESET_TTY_TO_COMMON_DEFAULTS	366	chmod (name, RESTORE_TTY_MODE);
393	/* store original tty status for restoration rxvt_clean_exit () -- rgg 04/12/95 */
394	if (lstat (name, &savestat) < 0) /* you lose out */
395	;
396	else
397	# endif
398	{
399	saved = true;
400	chown (name, getuid (), ttyconf.gid); /* fail silently */
401	chmod (name, ttyconf.mode);
402	# ifdef HAVE_REVOKE
403	revoke (name);
404	# endif
405	}
406	}
407	else
408	{ /* action == RESTORE */
409	# ifndef RESET_TTY_TO_COMMON_DEFAULTS
410	if (saved)
411	{
412	chmod (name, savestat.st_mode);
413	chown (name, savestat.st_uid, savestat.st_gid);
414	}
415	# else
416	chmod (name, (S_IRUSR \| S_IWUSR \| S_IRGRP \| S_IWGRP \| S_IROTH \| S_IWOTH));
417	chown (name, 0, 0);	367	chown (name, 0, ttyconf.gid);
418	# endif
419
420	}	368	}
421	}
422	#endif
423		369
424	rxvt_ptytty_unix::rxvt_ptytty_unix ()	370	close_tty ();
425	{	371
		372	if (pty >= 0)
		373	close (pty);
		374
		375	free (name);
		376
426	pty = tty = -1;	377	pty = tty = -1;
427	name = 0;	378	name = 0;
428	#ifndef NO_SETOWNER_TTYDEV
429	saved = false;
430	#endif
431	#if UTMP_SUPPORT
432	cmd_pid = 0;
433	#endif
434	}
435
436	rxvt_ptytty_unix::~rxvt_ptytty_unix ()
437	{
438	#if UTMP_SUPPORT
439	logout ();
440	#endif
441	put ();
442	}
443
444	void
445	rxvt_ptytty_unix::put ()
446	{
447	#ifndef NO_SETOWNER_TTYDEV
448	privileges (RESTORE);
449	#endif
450
451	if (pty >= 0) close (pty);
452	close_tty ();
453	free (name);
454
455	pty = tty = -1;
456	name = 0;
457	}	379	}
458		380
459	bool	381	bool
460	rxvt_ptytty_unix::get ()	382	ptytty_unix::get ()
461	{	383	{
462	/* get master (pty) */	384	/* get master (pty) */
463	if ((pty = get_pty (&tty, &name)) < 0)	385	if ((pty = get_pty (&tty, &name)) < 0)
464	return false;	386	return false;
465		387
…		…
467		389
468	/* get slave (tty) */	390	/* get slave (tty) */
469	if (tty < 0)	391	if (tty < 0)
470	{	392	{
471	#ifndef NO_SETOWNER_TTYDEV	393	#ifndef NO_SETOWNER_TTYDEV
472	privileges (SAVE);	394	chown (name, getuid (), ttyconf.gid); /* fail silently */
		395	chmod (name, ttyconf.mode);
		396	# ifdef HAVE_REVOKE
		397	revoke (name);
		398	# endif
473	#endif	399	#endif
474		400
475	if ((tty = get_tty (name)) < 0)	401	if ((tty = get_tty (name)) < 0)
476	{	402	{
477	put ();	403	put ();
…		…
480	}	406	}
481		407
482	return true;	408	return true;
483	}	409	}
484		410
485	#if PTYTTY_HELPER
486
487	static int sock_fd;
488	static int pid;
489
490	struct command
491	{
492	enum { get, login, destroy } type;
493
494	rxvt_ptytty *id;
495
496	bool login_shell;
497	int cmd_pid;
498	char hostname[512]; // arbitrary, but should be plenty
499	};
500
501	struct rxvt_ptytty_proxy : zero_initialized, rxvt_ptytty
502	{
503	rxvt_ptytty *id;
504
505	~rxvt_ptytty_proxy ();
506
507	bool get ();
508	void login (int cmd_pid, bool login_shell, const char *hostname);
509	};
510
511	bool
512	rxvt_ptytty_proxy::get ()
513	{
514	command cmd;
515
516	cmd.type = command::get;
517
518	write (sock_fd, &cmd, sizeof (cmd));
519
520	if (read (sock_fd, &id, sizeof (id)) != sizeof (id))
521	rxvt_fatal ("protocol error while creating pty using helper process, aborting.\n");
522
523	if (!id)
524	return false;
525
526	if ((pty = rxvt_recv_fd (sock_fd)) < 0
527	\|\| (tty = rxvt_recv_fd (sock_fd)) < 0)
528	rxvt_fatal ("protocol error while reading pty/tty fds from helper process, aborting.\n");
529
530	return true;
531	}
532
533	void
534	rxvt_ptytty_proxy::login (int cmd_pid, bool login_shell, const char *hostname)
535	{
536	command cmd;
537
538	cmd.type = command::login;
539	cmd.id = id;
540	cmd.cmd_pid = cmd_pid;
541	cmd.login_shell = login_shell;
542	strncpy (cmd.hostname, hostname, sizeof (cmd.hostname));
543
544	write (sock_fd, &cmd, sizeof (cmd));
545	}
546
547	rxvt_ptytty_proxy::~rxvt_ptytty_proxy ()
548	{
549	command cmd;
550
551	cmd.type = command::destroy;
552	cmd.id = id;
553
554	write (sock_fd, &cmd, sizeof (cmd));
555	}
556
557	static
558	void serve ()
559	{
560	command cmd;
561	vector<rxvt_ptytty *> ptys;
562
563	while (read (sock_fd, &cmd, sizeof (command)) == sizeof (command))
564	{
565	if (cmd.type == command::get)
566	{
567	// -> id ptyfd ttyfd
568	cmd.id = new rxvt_ptytty_unix;
569
570	if (cmd.id->get ())
571	{
572	write (sock_fd, &cmd.id, sizeof (cmd.id));
573	ptys.push_back (cmd.id);
574
575	rxvt_send_fd (sock_fd, cmd.id->pty);
576	rxvt_send_fd (sock_fd, cmd.id->tty);
577	}
578	else
579	{
580	delete cmd.id;
581	cmd.id = 0;
582	write (sock_fd, &cmd.id, sizeof (cmd.id));
583	}
584	}
585	else if (cmd.type == command::login)
586	{
587	#if UTMP_SUPPORT
588	if (find (ptys.begin (), ptys.end (), cmd.id))
589	{
590	cmd.hostname[sizeof (cmd.hostname) - 1] = 0;
591	cmd.id->login (cmd.cmd_pid, cmd.login_shell, cmd.hostname);
592	}
593	#endif
594	}
595	else if (cmd.type == command::destroy)
596	{
597	rxvt_ptytty **pty = find (ptys.begin (), ptys.end (), cmd.id);
598
599	if (pty)
600	{
601	delete *pty;
602	ptys.erase (pty);
603	}
604	}
605	else
606	break;
607	}
608
609	// destroy all ptys
610	for (rxvt_ptytty **i = ptys.end (); i-- > ptys.begin (); )
611	delete *i;
612	}
613
614	void rxvt_ptytty_server ()
615	{
616	int sv[2];
617
618	if (socketpair (AF_UNIX, SOCK_STREAM, 0, sv))
619	rxvt_fatal ("could not create socket to communicate with pty/sessiondb helper, aborting.\n");
620
621	pid = fork ();
622
623	if (pid < 0)
624	rxvt_fatal ("could not create pty/sessiondb helper process, aborting.\n");
625
626	if (pid)
627	{
628	// client, urxvt
629	sock_fd = sv[0];
630	close (sv[1]);
631	fcntl (sock_fd, F_SETFD, FD_CLOEXEC);
632	}
633	else
634	{
635	setgid (getegid ());
636	setuid (geteuid ());
637
638	// server, pty-helper
639	sock_fd = sv[1];
640
641	for (int fd = 0; fd < 1023; fd++)
642	if (fd != sock_fd && fd != 1)
643	close (fd);
644
645	serve ();
646	_exit (EXIT_SUCCESS);
647	}
648	}
649	#endif
650
651	// a "factory" g
652	rxvt_ptytty *
653	rxvt_new_ptytty ()
654	{
655	#if PTYTTY_HELPER
656	if (pid > 0)
657	// use helper process
658	return new rxvt_ptytty_proxy;
659	else
660	#endif
661	return new rxvt_ptytty_unix;
662	}
663
664	/----------------------- end-of-file (C source) -----------------------/
665

Diff Legend

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

Comparing rxvt-unicode/src/ptytty.C (file contents): Revision 1.44 by ayin, Tue Jan 17 17:59:31 2006 UTC vs. Revision 1.68 by ayin, Tue Dec 11 21:49:12 2007 UTC

Diff Legend

Comparing rxvt-unicode/src/ptytty.C (file contents):
Revision 1.44 by ayin, Tue Jan 17 17:59:31 2006 UTC vs.
Revision 1.68 by ayin, Tue Dec 11 21:49:12 2007 UTC