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

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

Diff Legend

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

Comparing rxvt-unicode/src/ptytty.C (file contents): Revision 1.34 by ayin, Tue Jan 17 08:54:33 2006 UTC vs. Revision 1.56 by root, Sun Jan 22 04:01:52 2006 UTC

Diff Legend

Comparing rxvt-unicode/src/ptytty.C (file contents):
Revision 1.34 by ayin, Tue Jan 17 08:54:33 2006 UTC vs.
Revision 1.56 by root, Sun Jan 22 04:01:52 2006 UTC