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

Comparing rxvt-unicode/src/ptytty.C (file contents):
Revision 1.24 by root, Sat Dec 31 16:06:48 2005 UTC vs.
Revision 1.51 by ayin, Thu Jan 19 18:53:05 2006 UTC

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

Diff Legend

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

Comparing rxvt-unicode/src/ptytty.C (file contents): Revision 1.24 by root, Sat Dec 31 16:06:48 2005 UTC vs. Revision 1.51 by ayin, Thu Jan 19 18:53:05 2006 UTC

Diff Legend

Comparing rxvt-unicode/src/ptytty.C (file contents):
Revision 1.24 by root, Sat Dec 31 16:06:48 2005 UTC vs.
Revision 1.51 by ayin, Thu Jan 19 18:53:05 2006 UTC