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

Comparing rxvt-unicode/src/ptytty.C (file contents):
Revision 1.22 by root, Tue Jan 11 04:44:57 2005 UTC vs.
Revision 1.59 by root, Mon Jan 23 12:05:12 2006 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 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	#ifdef HAVE_STDLIB_H
28	# include <cstdlib>	32	#include <cstdlib>
29	#endif	33	#include <cstring>
30	#ifdef HAVE_SYS_TYPES_H	34	#include <csignal>
		35
31	# include <sys/types.h>	36	#include <sys/types.h>
32	#endif	37	#include <sys/socket.h>
33	#ifdef HAVE_UNISTD_H
34	# include <unistd.h>	38	#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>	39	#include <fcntl.h>
41	#endif	40
42	#ifdef HAVE_SYS_IOCTL_H	41	#ifdef HAVE_SYS_IOCTL_H
43	# include <sys/ioctl.h>	42	# include <sys/ioctl.h>
44	#endif	43	#endif
45	#if defined(PTYS_ARE_PTMX) && defined(HAVE_SYS_STROPTS_H)	44	#if defined(HAVE_DEV_PTMX) && defined(HAVE_SYS_STROPTS_H)
46	# include <sys/stropts.h> /* for I_PUSH */	45	# include <sys/stropts.h> /* for I_PUSH */
47	#endif	46	#endif
48	#ifdef HAVE_ISASTREAM	47	#ifdef HAVE_ISASTREAM
49	# include <stropts.h>	48	# include <stropts.h>
50	#endif	49	#endif
51	#ifdef HAVE_PTY_H	50	#if defined(HAVE_PTY_H)
52	# include <pty.h>	51	# include <pty.h>
53	#endif	52	#elif defined(HAVE_LIBUTIL_H)
54
55	// better do this via configure, but....
56	#if defined(__FreeBSD__)
57	# include <libutil.h>	53	# include <libutil.h>
58	#elif defined(__DARWIN__) \|\| (defined (__MACH__) && defined (__APPLE__))	54	#elif defined(HAVE_UTIL_H)
59	# include <util.h>	55	# include <util.h>
60	#endif	56	#endif
		57	#ifdef TTY_GID_SUPPORT
		58	#include <grp.h>
		59	#endif
61		60
62	#include <cstdio>	61	#include <cstdio>
63	#include <grp.h>	62
		63	/////////////////////////////////////////////////////////////////////////////
64		64
65	/* ------------------------------------------------------------------------- *	65	/* ------------------------------------------------------------------------- *
66	* GET PSEUDO TELETYPE - MASTER AND SLAVE *	66	* GET PSEUDO TELETYPE - MASTER AND SLAVE *
67	* ------------------------------------------------------------------------- */	67	* ------------------------------------------------------------------------- */
68	/*	68	/*
69	* Returns pty file descriptor, or -1 on failure	69	* Returns pty file descriptor, or -1 on failure
70	* If successful, ttydev is set to the name of the slave device.	70	* 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	71	* fd_tty _may_ also be set to an open fd to the slave device
72	*/	72	*/
		73	#if defined(UNIX98_PTY)
73	static int	74	static int
74	get_pty (int fd_tty, char *ttydev)	75	get_pty (int fd_tty, char *ttydev)
75	{	76	{
76	int pfd;	77	int pfd;
77		78
78	#ifdef PTYS_ARE_OPENPTY	79	# if defined(HAVE_GETPT)
79	char tty_name[sizeof "/dev/pts/????\0"];	80	pfd = getpt();
		81	# elif defined(HAVE_POSIX_OPENPT)
		82	pfd = posix_openpt (O_RDWR);
		83	# else
		84	pfd = open (CLONE_DEVICE, O_RDWR \| O_NOCTTY, 0);
		85	# endif
		86	if (pfd >= 0)
		87	{
		88	if (grantpt (pfd) == 0 /* change slave permissions */
		89	&& unlockpt (pfd) == 0)
		90	{ /* slave now unlocked */
		91	ttydev = strdup (ptsname (pfd)); / get slave's name */
		92	return pfd;
		93	}
80		94
		95	close (pfd);
		96	}
		97
		98	return -1;
		99	}
		100	#elif defined(HAVE_OPENPTY)
		101	static int
		102	get_pty (int fd_tty, char *ttydev)
		103	{
		104	int pfd;
		105	int res;
		106	char tty_name[32];
		107
81	if (openpty (&pfd, fd_tty, tty_name, NULL, NULL) != -1)	108	res = openpty (&pfd, fd_tty, tty_name, NULL, NULL);
		109	if (res != -1)
82	{	110	{
83	*ttydev = strdup (tty_name);	111	*ttydev = strdup (tty_name);
84	return pfd;	112	return pfd;
85	}	113	}
86	#endif
87		114
88	#ifdef PTYS_ARE__GETPTY	115	return -1;
		116	}
		117	#elif defined(HAVE__GETPTY)
		118	static int
		119	get_pty (int fd_tty, char *ttydev)
		120	{
		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
93		126
94	#ifdef PTYS_ARE_GETPTY	127	return -1;
		128	}
		129	#elif defined(HAVE_DEV_PTC)
		130	static int
		131	get_pty (int fd_tty, char *ttydev)
		132	{
		133	int pfd;
95		134
96	char *ptydev;
97
98	while ((ptydev = getpty ()) != NULL)
99	if ((pfd = open (ptydev, O_RDWR \| O_NOCTTY, 0)) >= 0)
100	{
101	*ttydev = ptydev;
102	return pfd;
103	}
104	#endif
105
106	#if defined(HAVE_GRANTPT) && defined(HAVE_UNLOCKPT)
107	# if defined(PTYS_ARE_GETPT) \|\| defined(PTYS_ARE_PTMX)
108
109	{
110	# ifdef PTYS_ARE_GETPT
111	pfd = getpt ();
112	# else
113	pfd = open ("/dev/ptmx", O_RDWR \| O_NOCTTY, 0);
114	# endif
115
116	if (pfd >= 0)
117	{
118	if (grantpt (pfd) == 0 /* change slave permissions */
119	&& unlockpt (pfd) == 0)
120	{ /* slave now unlocked */
121	ttydev = strdup (ptsname (pfd)); / get slave's name */
122	return pfd;
123	}
124	close (pfd);
125	}
126	}
127	# endif
128	#endif
129
130	#ifdef PTYS_ARE_PTC
131	if ((pfd = open ("/dev/ptc", O_RDWR \| O_NOCTTY, 0)) >= 0)	135	if ((pfd = open ("/dev/ptc", O_RDWR \| O_NOCTTY, 0)) >= 0)
132	{	136	{
133	*ttydev = strdup (ttyname (pfd));	137	*ttydev = strdup (ttyname (pfd));
134	return pfd;	138	return pfd;
135	}	139	}
136	#endif
137		140
138	#ifdef PTYS_ARE_CLONE	141	return -1;
		142	}
		143	#elif defined(HAVE_DEV_CLONE)
		144	static int
		145	get_pty (int fd_tty, char *ttydev)
		146	{
		147	int pfd;
		148
139	if ((pfd = open ("/dev/ptym/clone", O_RDWR \| O_NOCTTY, 0)) >= 0)	149	if ((pfd = open ("/dev/ptym/clone", O_RDWR \| O_NOCTTY, 0)) >= 0)
140	{	150	{
141	*ttydev = strdup (ptsname (pfd));	151	*ttydev = strdup (ptsname (pfd));
142	return pfd;	152	return pfd;
143	}	153	}
144	#endif
145
146	#ifdef PTYS_ARE_NUMERIC
147	{
148	int idx;
149	char c1, c2;
150	char pty_name[] = "/dev/ptyp???";
151	char tty_name[] = "/dev/ttyp???";
152
153	c1 = &(pty_name[sizeof (pty_name) - 4]);
154	c2 = &(tty_name[sizeof (tty_name) - 4]);
155	for (idx = 0; idx < 256; idx++)
156	{
157	sprintf (c1, "%d", idx);
158	sprintf (c2, "%d", idx);
159	if (access (tty_name, F_OK) < 0)
160	{
161	idx = 256;
162	break;
163	}
164
165	if ((pfd = open (pty_name, O_RDWR \| O_NOCTTY, 0)) >= 0)
166	{
167	if (access (tty_name, R_OK \| W_OK) == 0)
168	{
169	*ttydev = strdup (tty_name);
170	return pfd;
171	}
172
173	close (pfd);
174	}
175	}
176	}
177	#endif
178
179	#ifdef PTYS_ARE_SEARCHED
180	{
181	const char c1, c2;
182	char pty_name[] = "/dev/pty??";
183	char tty_name[] = "/dev/tty??";
184
185	# ifndef PTYCHAR1
186	# define PTYCHAR1 "pqrstuvwxyz"
187	# endif
188	# ifndef PTYCHAR2
189	# define PTYCHAR2 "0123456789abcdef"
190	# endif
191
192	for (c1 = PTYCHAR1; *c1; c1++)
193	{
194	pty_name[ (sizeof (pty_name) - 3)] =
195	tty_name[ (sizeof (pty_name) - 3)] = *c1;
196	for (c2 = PTYCHAR2; *c2; c2++)
197	{
198	pty_name[ (sizeof (pty_name) - 2)] =
199	tty_name[ (sizeof (pty_name) - 2)] = *c2;
200	if ((pfd = open (pty_name, O_RDWR \| O_NOCTTY, 0)) >= 0)
201	{
202	if (access (tty_name, R_OK \| W_OK) == 0)
203	{
204	*ttydev = strdup (tty_name);
205	return pfd;
206	}
207
208	close (pfd);
209	}
210	}
211	}
212	}
213	#endif
214		154
215	return -1;	155	return -1;
216	}	156	}
		157	#else
		158	/* Based on the code in openssh/openbsd-compat/bsd-openpty.c */
		159	static int
		160	get_pty (int fd_tty, char *ttydev)
		161	{
		162	int pfd;
		163	int i;
		164	char pty_name[32];
		165	char tty_name[32];
		166	const char *majors = "pqrstuvwxyzabcde";
		167	const char *minors = "0123456789abcdef";
		168	for (i = 0; i < 256; i++)
		169	{
		170	snprintf(pty_name, 32, "/dev/pty%c%c", majors[i / 16], minors[i % 16]);
		171	snprintf(tty_name, 32, "/dev/tty%c%c", majors[i / 16], minors[i % 16]);
		172	if ((pfd = open (pty_name, O_RDWR \| O_NOCTTY, 0)) == -1)
		173	{
		174	snprintf(pty_name, 32, "/dev/ptyp%d", i);
		175	snprintf(tty_name, 32, "/dev/ttyp%d", i);
		176	if ((pfd = open (pty_name, O_RDWR \| O_NOCTTY, 0)) == -1)
		177	continue;
		178	}
		179	if (access (tty_name, R_OK \| W_OK) == 0)
		180	{
		181	*ttydev = strdup (tty_name);
		182	return pfd;
		183	}
		184
		185	close (pfd);
		186	}
		187	}
		188	#endif
217		189
218	/----------------------------------------------------------------------/	190	/----------------------------------------------------------------------/
219	/*	191	/*
220	* Returns tty file descriptor, or -1 on failure	192	* Returns tty file descriptor, or -1 on failure
221	*/	193	*/
…		…
228	/----------------------------------------------------------------------/	200	/----------------------------------------------------------------------/
229	/*	201	/*
230	* Make our tty a controlling tty so that /dev/tty points to us	202	* Make our tty a controlling tty so that /dev/tty points to us
231	*/	203	*/
232	static int	204	static int
233	control_tty (int fd_tty, const char *ttydev)	205	control_tty (int fd_tty)
234	{	206	{
235	#ifndef __QNX__
236	int fd;
237
238	/* ---------------------------------------- */
239	# ifdef HAVE_SETSID
240	setsid ();	207	setsid ();
241	# endif
242	# if defined(HAVE_SETPGID)
243	setpgid (0, 0);
244	# elif defined(HAVE_SETPGRP)
245	setpgrp (0, 0);
246	# endif
247		208
248	/* ---------------------------------------- */
249	# ifdef TIOCNOTTY
250	fd = open ("/dev/tty", O_RDWR \| O_NOCTTY);
251	if (fd >= 0)
252	{
253	ioctl (fd, TIOCNOTTY, NULL); /* void tty associations */
254	close (fd);
255	}
256	# endif
257
258	/* ---------------------------------------- */
259	fd = open ("/dev/tty", O_RDWR \| O_NOCTTY);
260	if (fd >= 0)
261	close (fd); /* ouch: still have controlling tty */
262
263	/* ---------------------------------------- */
264	#if defined(PTYS_ARE_PTMX) && defined(I_PUSH)	209	#if defined(HAVE_DEV_PTMX) && defined(I_PUSH)
265	/*	210	/*
266	* Push STREAMS modules:	211	* Push STREAMS modules:
267	* ptem: pseudo-terminal hardware emulation module.	212	* ptem: pseudo-terminal hardware emulation module.
268	* ldterm: standard terminal line discipline.	213	* ldterm: standard terminal line discipline.
269	* ttcompat: V7, 4BSD and XENIX STREAMS compatibility module.	214	* ttcompat: V7, 4BSD and XENIX STREAMS compatibility module.
…		…
286	ioctl (fd_tty, I_PUSH, "ptem");	231	ioctl (fd_tty, I_PUSH, "ptem");
287	ioctl (fd_tty, I_PUSH, "ldterm");	232	ioctl (fd_tty, I_PUSH, "ldterm");
288	ioctl (fd_tty, I_PUSH, "ttcompat");	233	ioctl (fd_tty, I_PUSH, "ttcompat");
289	}	234	}
290	#endif	235	#endif
291	/* ---------------------------------------- */	236
292	# if defined(TIOCSCTTY)
293	fd = ioctl (fd_tty, TIOCSCTTY, NULL);	237	ioctl (fd_tty, TIOCSCTTY, NULL);
294	# elif defined(TIOCSETCTTY)	238
295	fd = ioctl (fd_tty, TIOCSETCTTY, NULL);
296	# else
297	fd = open (ttydev, O_RDWR);
298	if (fd >= 0)
299	close (fd);
300	# endif
301	/* ---------------------------------------- */
302	fd = open ("/dev/tty", O_WRONLY);	239	int fd = open ("/dev/tty", O_WRONLY);
303	if (fd < 0)	240	if (fd < 0)
304	return -1; /* fatal */	241	return -1; /* fatal */
		242
305	close (fd);	243	close (fd);
306	/* ---------------------------------------- */
307	#endif /* ! __QNX__ */
308		244
309	return 0;	245	return 0;
310	}	246	}
311		247
312	#ifndef NO_SETOWNER_TTYDEV	248	void
		249	ptytty::close_tty ()
		250	{
		251	if (tty < 0)
		252	return;
		253
		254	close (tty);
		255	tty = -1;
		256	}
		257
		258	bool
		259	ptytty::make_controlling_tty ()
		260	{
		261	return control_tty (tty) >= 0;
		262	}
		263
		264	void
		265	ptytty::set_utf8_mode (bool on)
		266	{
		267	#ifdef IUTF8
		268	if (pty < 0)
		269	return;
		270
		271	struct termios tio;
		272
		273	if (tcgetattr (pty, &tio) != -1)
		274	{
		275	tcflag_t new_cflag = tio.c_iflag;
		276
		277	if (on)
		278	new_cflag \|= IUTF8;
		279	else
		280	new_cflag &= ~IUTF8;
		281
		282	if (new_cflag != tio.c_iflag)
		283	{
		284	tio.c_iflag = new_cflag;
		285	tcsetattr (pty, TCSANOW, &tio);
		286	}
		287	}
		288	#endif
		289	}
		290
313	static struct ttyconf {	291	static struct ttyconf {
314	gid_t gid;	292	gid_t gid;
315	mode_t mode;	293	mode_t mode;
316		294
317	ttyconf ()	295	ttyconf ()
…		…
326	}	304	}
327	else	305	else
328	#endif /* TTY_GID_SUPPORT */	306	#endif /* TTY_GID_SUPPORT */
329	{	307	{
330	mode = S_IRUSR \| S_IWUSR \| S_IWGRP \| S_IWOTH;	308	mode = S_IRUSR \| S_IWUSR \| S_IWGRP \| S_IWOTH;
331	gid = getgid ();	309	gid = 0;
332	}	310	}
333	}	311	}
334	} ttyconf;	312	} ttyconf;
335		313
		314	ptytty_unix::ptytty_unix ()
		315	{
		316	name = 0;
		317	#if UTMP_SUPPORT
		318	cmd_pid = 0;
		319	#endif
		320	}
		321
		322	ptytty_unix::~ptytty_unix ()
		323	{
		324	#if UTMP_SUPPORT
		325	logout ();
		326	#endif
		327	put ();
		328	}
		329
336	void	330	void
337	rxvt_ptytty::privileges (rxvt_privaction action)	331	ptytty_unix::put ()
338	{	332	{
339	rxvt_privileges (RESTORE);	333	chmod (name, RESTORE_TTY_MODE);
		334	chown (name, 0, ttyconf.gid);
340		335
341	if (action == SAVE)	336	close_tty ();
342	{
343	//next_tty_action = RESTORE;
344	# ifndef RESET_TTY_TO_COMMON_DEFAULTS
345	/* store original tty status for restoration rxvt_clean_exit () -- rgg 04/12/95 */
346	if (lstat (name, &savestat) < 0) /* you lose out */
347	;//next_tty_action = IGNORE;
348	else
349	# endif
350		337
351	{	338	if (pty >= 0)
352	chown (name, getuid (), ttyconf.gid); /* fail silently */	339	close (pty);
353	chmod (name, ttyconf.mode);
354	# ifdef HAVE_REVOKE
355	revoke (name);
356	# endif
357		340
358	}	341	free (name);
359	}
360	else
361	{ /* action == RESTORE */
362	//next_tty_action = IGNORE;
363	# ifndef RESET_TTY_TO_COMMON_DEFAULTS
364	chmod (name, savestat.st_mode);
365	chown (name, savestat.st_uid, savestat.st_gid);
366	# else
367	chmod (name, (S_IRUSR \| S_IWUSR \| S_IRGRP \| S_IWGRP \| S_IROTH \| S_IWOTH));
368	chown (name, 0, 0);
369	# endif
370		342
371	}
372
373	rxvt_privileges (IGNORE);
374	}
375	#endif
376
377	rxvt_ptytty::rxvt_ptytty ()
378	{
379	pty = tty = -1;	343	pty = tty = -1;
380	name = 0;	344	name = 0;
381	}	345	}
382		346
383	rxvt_ptytty::~rxvt_ptytty ()
384	{
385	put ();
386	}
387
388	void
389
390	rxvt_ptytty::close_tty ()
391	{
392	if (tty >= 0) close (tty);
393	tty = -1;
394	}
395
396	void
397	rxvt_ptytty::put ()
398	{
399	#ifndef NO_SETOWNER_TTYDEV
400	if (tty >= 0)
401	privileges (RESTORE);
402	#endif
403
404	if (pty >= 0) close (pty);
405	close_tty ();
406	free (name);
407
408	pty = tty = -1;
409	name = 0;
410	}
411
412	bool	347	bool
413	rxvt_ptytty::make_controlling_tty ()	348	ptytty_unix::get ()
414	{
415	return control_tty (tty, name) >= 0;
416	}
417
418	bool
419	rxvt_ptytty::get ()
420	{	349	{
421	/* get master (pty) */	350	/* get master (pty) */
422	if ((pty = get_pty (&tty, &name)) < 0)	351	if ((pty = get_pty (&tty, &name)) < 0)
423	return false;	352	return false;
424		353
…		…
426		355
427	/* get slave (tty) */	356	/* get slave (tty) */
428	if (tty < 0)	357	if (tty < 0)
429	{	358	{
430	#ifndef NO_SETOWNER_TTYDEV	359	#ifndef NO_SETOWNER_TTYDEV
431	privileges (SAVE);	360	chown (name, getuid (), ttyconf.gid); /* fail silently */
		361	chmod (name, ttyconf.mode);
		362	# ifdef HAVE_REVOKE
		363	revoke (name);
		364	# endif
432	#endif	365	#endif
433		366
434	if ((tty = get_tty (name)) < 0)	367	if ((tty = get_tty (name)) < 0)
435	{	368	{
436	put ();	369	put ();
…		…
439	}	372	}
440		373
441	return true;	374	return true;
442	}	375	}
443		376
		377	/////////////////////////////////////////////////////////////////////////////
		378	// helper/proxy support
		379
		380	#if PTYTTY_HELPER
		381
		382	static int sock_fd = -1, lock_fd = -1;
		383	static int helper_pid, owner_pid;
		384
		385	struct command
		386	{
		387	enum { get, login, destroy } type;
		388
		389	ptytty *id;
		390
		391	bool login_shell;
		392	int cmd_pid;
		393	char hostname[512]; // arbitrary, but should be plenty
		394	};
		395
		396	struct ptytty_proxy : ptytty
		397	{
		398	ptytty *id;
		399
		400	ptytty_proxy ()
		401	: id(0)
		402	{
		403	}
		404
		405	~ptytty_proxy ();
		406
		407	bool get ();
		408	void login (int cmd_pid, bool login_shell, const char *hostname);
		409	};
		410
		411	#if PTYTTY_REENTRANT
		412	# define NEED_TOKEN do { char ch; read (lock_fd, &ch, 1); } while (0)
		413	# define GIVE_TOKEN do { char ch; write (lock_fd, &ch, 1); } while (0)
		414	#else
		415	# define NEED_TOKEN (void)0
		416	# define GIVE_TOKEN (void)0
		417	#endif
		418
		419	bool
		420	ptytty_proxy::get ()
		421	{
		422	NEED_TOKEN;
		423
		424	command cmd;
		425
		426	cmd.type = command::get;
		427
		428	write (sock_fd, &cmd, sizeof (cmd));
		429
		430	if (read (sock_fd, &id, sizeof (id)) != sizeof (id))
		431	ptytty_fatal ("protocol error while creating pty using helper process, aborting.\n");
		432
		433	if (!id)
		434	{
		435	GIVE_TOKEN;
		436	return false;
		437	}
		438
		439	if ((pty = recv_fd (sock_fd)) < 0
		440	\|\| (tty = recv_fd (sock_fd)) < 0)
		441	ptytty_fatal ("protocol error while reading pty/tty fds from helper process, aborting.\n");
		442
		443	GIVE_TOKEN;
		444	return true;
		445	}
		446
444	void	447	void
445	rxvt_ptytty::set_utf8_mode (bool on)	448	ptytty_proxy::login (int cmd_pid, bool login_shell, const char *hostname)
446	{	449	{
447	#ifdef IUTF8	450	NEED_TOKEN;
448	if (pty != -1)	451
		452	command cmd;
		453
		454	cmd.type = command::login;
		455	cmd.id = id;
		456	cmd.cmd_pid = cmd_pid;
		457	cmd.login_shell = login_shell;
		458	strncpy (cmd.hostname, hostname, sizeof (cmd.hostname));
		459
		460	write (sock_fd, &cmd, sizeof (cmd));
		461
		462	GIVE_TOKEN;
		463	}
		464
		465	ptytty_proxy::~ptytty_proxy ()
		466	{
		467	if (id)
		468	{
		469	NEED_TOKEN;
		470
		471	command cmd;
		472
		473	cmd.type = command::destroy;
		474	cmd.id = id;
		475
		476	write (sock_fd, &cmd, sizeof (cmd));
		477
		478	GIVE_TOKEN;
449	{	479	}
450	struct termios tio;	480	}
451		481
452	if (tcgetattr (pty, &tio) != -1)	482	static
		483	void serve ()
		484	{
		485	command cmd;
		486	vector<ptytty *> ptys;
		487
		488	for (;;)
		489	{
		490	GIVE_TOKEN;
		491
		492	if (read (sock_fd, &cmd, sizeof (command)) != sizeof (command))
		493	break;
		494
		495	if (cmd.type == command::get)
453	{	496	{
454	tcflag_t new_cflag = tio.c_iflag;	497	// -> id ptyfd ttyfd
		498	cmd.id = new ptytty_unix;
455		499
456	if (on)	500	if (cmd.id->get ())
457	new_cflag \|= IUTF8;	501	{
		502	write (sock_fd, &cmd.id, sizeof (cmd.id));
		503	ptys.push_back (cmd.id);
		504
		505	ptytty::send_fd (sock_fd, cmd.id->pty);
		506	ptytty::send_fd (sock_fd, cmd.id->tty);
		507	}
458	else	508	else
459	new_cflag &= ~IUTF8;
460
461	if (new_cflag != tio.c_iflag)
462	{	509	{
463	tio.c_iflag = new_cflag;	510	delete cmd.id;
464	tcsetattr (pty, TCSANOW, &tio);	511	cmd.id = 0;
		512	write (sock_fd, &cmd.id, sizeof (cmd.id));
465	}	513	}
466	}	514	}
		515	else if (cmd.type == command::login)
		516	{
		517	#if UTMP_SUPPORT
		518	if (find (ptys.begin (), ptys.end (), cmd.id) != ptys.end ())
		519	{
		520	cmd.hostname[sizeof (cmd.hostname) - 1] = 0;
		521	cmd.id->login (cmd.cmd_pid, cmd.login_shell, cmd.hostname);
		522	}
		523	#endif
		524	}
		525	else if (cmd.type == command::destroy)
		526	{
		527	vector<ptytty *>::iterator pty = find (ptys.begin (), ptys.end (), cmd.id);
		528
		529	if (pty != ptys.end ())
		530	{
		531	delete *pty;
		532	ptys.erase (pty);
		533	}
		534	}
		535	else
		536	break;
		537
		538	NEED_TOKEN;
		539	}
		540
		541	// destroy all ptys
		542	for (vector<ptytty *>::iterator i = ptys.end (); i-- > ptys.begin (); )
		543	delete *i;
		544	}
		545
		546	void
		547	ptytty::use_helper ()
		548	{
		549	#ifndef PTYTTY_NO_PID_CHECK
		550	int pid = getpid ();
		551	#endif
		552
		553	if (sock_fd >= 0
		554	#ifndef PTYTTY_NO_PID_CHECK
		555	&& pid == owner_pid
		556	#endif
		557	)
		558	return;
		559
		560	#ifndef PTYTTY_NO_PID_CHECK
		561	owner_pid = pid;
		562	#endif
		563
		564	int sv[2];
		565
		566	if (socketpair (AF_UNIX, SOCK_STREAM, 0, sv))
		567	ptytty_fatal ("could not create socket to communicate with pty/sessiondb helper, aborting.\n");
		568
		569	#ifdef PTYTTY_REENTRANT
		570	int lv[2];
		571
		572	if (socketpair (AF_UNIX, SOCK_STREAM, 0, lv))
		573	ptytty_fatal ("could not create socket to communicate with pty/sessiondb helper, aborting.\n");
		574	#endif
		575
		576	helper_pid = fork ();
		577
		578	if (helper_pid < 0)
		579	ptytty_fatal ("could not create pty/sessiondb helper process, aborting.\n");
		580
		581	if (helper_pid)
467	}	582	{
		583	// client, process
		584	sock_fd = sv[0];
		585	close (sv[1]);
		586	fcntl (sock_fd, F_SETFD, FD_CLOEXEC);
		587	#ifdef PTYTTY_REENTRANT
		588	lock_fd = lv[0];
		589	close (lv[1]);
		590	fcntl (lock_fd, F_SETFD, FD_CLOEXEC);
		591	#endif
		592	}
		593	else
		594	{
		595	// server, pty-helper
		596	sock_fd = sv[1];
		597	#ifdef PTYTTY_REENTRANT
		598	lock_fd = lv[1];
		599	#endif
		600
		601	chdir ("/");
		602
		603	signal (SIGHUP, SIG_IGN);
		604	signal (SIGTERM, SIG_IGN);
		605	signal (SIGINT, SIG_IGN);
		606	signal (SIGPIPE, SIG_IGN);
		607
		608	for (int fd = 0; fd < 1023; fd++)
		609	if (fd != sock_fd && fd != lock_fd)
		610	close (fd);
		611
		612	serve ();
		613	_exit (EXIT_SUCCESS);
		614	}
		615	}
		616
		617	#endif
		618
		619	ptytty *
		620	ptytty::create ()
		621	{
		622	#if PTYTTY_HELPER
		623	if (helper_pid
		624	# ifndef PTYTTY_NO_PID_CHECK
		625	&& getpid () == owner_pid
468	#endif	626	# endif
		627	)
		628	// use helper process
		629	return new ptytty_proxy;
		630	else
		631	#endif
		632	return new ptytty_unix;
469	}	633	}
470		634
471	/----------------------- end-of-file (C source) -----------------------/	635	void
		636	ptytty::init ()
		637	{
		638	uid_t uid = getuid ();
		639	gid_t gid = getgid ();
		640
		641	// before doing anything else, check for setuid/setgid operation,
		642	// start the helper process and drop privileges
		643	if (uid != geteuid ()
		644	\|\| gid != getegid ())
		645	{
		646	#if PTYTTY_HELPER
		647	use_helper ();
		648	#else
		649	ptytty_warn ("running setuid/setgid without pty helper compiled in, continuing unprivileged.\n");
		650	#endif
472		651
		652	drop_privileges ();
		653	}
		654	}
		655
		656	void
		657	ptytty::drop_privileges ()
		658	{
		659	uid_t uid = getuid ();
		660	gid_t gid = getgid ();
		661
		662	// drop privileges
		663	#if HAVE_SETRESUID
		664	setresgid (gid, gid, gid);
		665	setresuid (uid, uid, uid);
		666	#elif HAVE_SETREUID
		667	setregid (gid, gid);
		668	setreuid (uid, uid);
		669	#elif HAVE_SETUID
		670	setgid (gid);
		671	setuid (uid);
		672	#endif
		673
		674	if (uid != geteuid ()
		675	\|\| gid != getegid ())
		676	ptytty_fatal ("unable to drop privileges, aborting.\n");
		677	}
		678
		679	/////////////////////////////////////////////////////////////////////////////
		680	// C API
		681
		682	#ifndef PTYTTY_NO_C_API
		683
		684	#define DEFINE_METHOD(retval, name, args1, args2) \
		685	extern "C" retval ptytty_ ## name args1 \
		686	{ return ((struct ptytty *)ptytty)->name args2; }
		687
		688	DEFINE_METHOD(int,pty,(void *ptytty),)
		689	DEFINE_METHOD(int,tty,(void *ptytty),)
		690	DEFINE_METHOD(int,get,(void *ptytty),())
		691	DEFINE_METHOD(void,login,(void ptytty, int cmd_pid, bool login_shell, const char hostname),(cmd_pid,login_shell,hostname))
		692
		693	DEFINE_METHOD(void,close_tty,(void *ptytty),())
		694	DEFINE_METHOD(int,make_controlling_tty,(void *ptytty),())
		695	DEFINE_METHOD(void,set_utf8_mode,(void *ptytty, int on),(on))
		696
		697	#define DEFINE_STATIC(retval, name, args) \
		698	extern "C" retval ptytty_ ## name args \
		699	{ return ptytty::name args; }
		700
		701	DEFINE_STATIC(void,drop_privileges,())
		702	DEFINE_STATIC(void,use_helper,())
		703	DEFINE_STATIC(void,init,())
		704
		705	DEFINE_STATIC(void *,create,())
		706
		707	void ptytty_delete (void *ptytty)
		708	{
		709	delete (struct ptytty *)ptytty;
		710	}
		711
		712	// send_fd, recv_fd not exposed
		713
		714	#endif

Diff Legend

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

Comparing rxvt-unicode/src/ptytty.C (file contents): Revision 1.22 by root, Tue Jan 11 04:44:57 2005 UTC vs. Revision 1.59 by root, Mon Jan 23 12:05:12 2006 UTC

Diff Legend

Comparing rxvt-unicode/src/ptytty.C (file contents):
Revision 1.22 by root, Tue Jan 11 04:44:57 2005 UTC vs.
Revision 1.59 by root, Mon Jan 23 12:05:12 2006 UTC