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

Comparing rxvt-unicode/src/ptytty.C (file contents):
Revision 1.30 by root, Fri Jan 6 05:37:59 2006 UTC vs.
Revision 1.48 by ayin, Thu Jan 19 10:44:44 2006 UTC

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

Diff Legend

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

Comparing rxvt-unicode/src/ptytty.C (file contents): Revision 1.30 by root, Fri Jan 6 05:37:59 2006 UTC vs. Revision 1.48 by ayin, Thu Jan 19 10:44:44 2006 UTC

Diff Legend

Comparing rxvt-unicode/src/ptytty.C (file contents):
Revision 1.30 by root, Fri Jan 6 05:37:59 2006 UTC vs.
Revision 1.48 by ayin, Thu Jan 19 10:44:44 2006 UTC