ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/rxvt-unicode/src/ptytty.C
Revision: 1.36
Committed: Tue Jan 17 10:59:32 2006 UTC (18 years, 4 months ago) by root
Content type: text/plain
Branch: MAIN
Changes since 1.35: +59 -62 lines
Log Message:
*** empty log message ***

File Contents

# User Rev Content
1 pcg 1.1 /*--------------------------------*-C-*---------------------------------*
2 pcg 1.11 * File: ptytty.C
3 pcg 1.1 *----------------------------------------------------------------------*
4     *
5     * All portions of code are copyright by their respective author/s.
6     * Copyright (c) 1999-2001 Geoff Wing <gcw@pobox.com>
7 root 1.30 * Copyright (c) 2004-2006 Marc Lehmann <pcg@goof.com>
8 pcg 1.1 *
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
11     * the Free Software Foundation; either version 2 of the License, or
12     * (at your option) any later version.
13     *
14     * This program is distributed in the hope that it will be useful,
15     * but WITHOUT ANY WARRANTY; without even the implied warranty of
16     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17     * GNU General Public License for more details.
18     *
19     * You should have received a copy of the GNU General Public License
20     * along with this program; if not, write to the Free Software
21     * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
22     *---------------------------------------------------------------------*/
23    
24     #include "../config.h" /* NECESSARY */
25 root 1.16 #include "rxvt.h"
26 pcg 1.1
27 pcg 1.8 # include <cstdlib>
28 root 1.31 # include <cstring>
29    
30 pcg 1.1 #ifdef HAVE_SYS_TYPES_H
31     # include <sys/types.h>
32     #endif
33     #ifdef HAVE_UNISTD_H
34     # include <unistd.h>
35     #endif
36     #ifdef HAVE_FCNTL_H
37     # include <fcntl.h>
38     #endif
39     #ifdef HAVE_SYS_IOCTL_H
40     # include <sys/ioctl.h>
41     #endif
42 root 1.12 #if defined(PTYS_ARE_PTMX) && defined(HAVE_SYS_STROPTS_H)
43 pcg 1.1 # include <sys/stropts.h> /* for I_PUSH */
44     #endif
45 pcg 1.10 #ifdef HAVE_ISASTREAM
46     # include <stropts.h>
47     #endif
48 root 1.31 #if defined(HAVE_PTY_H)
49 root 1.22 # include <pty.h>
50 root 1.31 #elif defined(HAVE_LIBUTIL_H)
51 root 1.16 # include <libutil.h>
52 root 1.31 #elif defined(HAVE_UTIL_H)
53 root 1.17 # include <util.h>
54 pcg 1.1 #endif
55    
56 root 1.16 #include <cstdio>
57     #include <grp.h>
58    
59 root 1.36 /////////////////////////////////////////////////////////////////////////////
60    
61 pcg 1.1 /* ------------------------------------------------------------------------- *
62     * GET PSEUDO TELETYPE - MASTER AND SLAVE *
63     * ------------------------------------------------------------------------- */
64     /*
65     * Returns pty file descriptor, or -1 on failure
66     * If successful, ttydev is set to the name of the slave device.
67     * fd_tty _may_ also be set to an open fd to the slave device
68     */
69 root 1.16 static int
70     get_pty (int *fd_tty, char **ttydev)
71 pcg 1.1 {
72 pcg 1.4 int pfd;
73 pcg 1.1
74     #ifdef PTYS_ARE_OPENPTY
75 root 1.26 char tty_name[sizeof "/dev/pts/????\0"];
76 pcg 1.4
77 root 1.27 rxvt_privileges(RESTORE);
78     int res = openpty (&pfd, fd_tty, tty_name, NULL, NULL);
79     rxvt_privileges(IGNORE);
80    
81     if (res != -1)
82 pcg 1.4 {
83 pcg 1.6 *ttydev = strdup (tty_name);
84 pcg 1.4 return pfd;
85 pcg 1.1 }
86     #endif
87    
88     #ifdef PTYS_ARE__GETPTY
89 pcg 1.6 *ttydev = _getpty (&pfd, O_RDWR | O_NONBLOCK | O_NOCTTY, 0622, 0);
90 pcg 1.4 if (*ttydev != NULL)
91     return pfd;
92 pcg 1.1 #endif
93    
94     #if defined(HAVE_GRANTPT) && defined(HAVE_UNLOCKPT)
95 ayin 1.34 # if defined(PTYS_ARE_GETPT) || defined(PTYS_ARE_POSIX) || defined(PTYS_ARE_PTMX)
96 pcg 1.4
97     {
98 ayin 1.34 # ifdef PTYS_ARE_GETPT
99     pfd = getpt();
100     # else
101 root 1.33 # ifdef PTYS_ARE_POSIX
102     pfd = posix_openpt (O_RDWR);
103 pcg 1.1 # else
104 pcg 1.6 pfd = open ("/dev/ptmx", O_RDWR | O_NOCTTY, 0);
105 pcg 1.1 # endif
106 ayin 1.34 # endif
107 pcg 1.4
108     if (pfd >= 0)
109     {
110 pcg 1.6 if (grantpt (pfd) == 0 /* change slave permissions */
111     && unlockpt (pfd) == 0)
112 pcg 1.4 { /* slave now unlocked */
113 root 1.16 *ttydev = strdup (ptsname (pfd)); /* get slave's name */
114 pcg 1.4 return pfd;
115     }
116 pcg 1.6 close (pfd);
117 pcg 1.4 }
118     }
119 pcg 1.1 # endif
120     #endif
121    
122     #ifdef PTYS_ARE_PTC
123 pcg 1.6 if ((pfd = open ("/dev/ptc", O_RDWR | O_NOCTTY, 0)) >= 0)
124 pcg 1.4 {
125 root 1.16 *ttydev = strdup (ttyname (pfd));
126 pcg 1.4 return pfd;
127 pcg 1.1 }
128     #endif
129    
130     #ifdef PTYS_ARE_CLONE
131 pcg 1.6 if ((pfd = open ("/dev/ptym/clone", O_RDWR | O_NOCTTY, 0)) >= 0)
132 pcg 1.4 {
133 root 1.16 *ttydev = strdup (ptsname (pfd));
134 pcg 1.4 return pfd;
135 pcg 1.1 }
136     #endif
137    
138     #ifdef PTYS_ARE_NUMERIC
139 pcg 1.4 {
140 root 1.16 int idx;
141     char *c1, *c2;
142     char pty_name[] = "/dev/ptyp???";
143     char tty_name[] = "/dev/ttyp???";
144 pcg 1.4
145 root 1.16 c1 = &(pty_name[sizeof (pty_name) - 4]);
146     c2 = &(tty_name[sizeof (tty_name) - 4]);
147 pcg 1.4 for (idx = 0; idx < 256; idx++)
148     {
149 pcg 1.6 sprintf (c1, "%d", idx);
150     sprintf (c2, "%d", idx);
151     if (access (tty_name, F_OK) < 0)
152 pcg 1.4 {
153     idx = 256;
154     break;
155     }
156 root 1.16
157 pcg 1.6 if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) >= 0)
158 pcg 1.4 {
159 pcg 1.6 if (access (tty_name, R_OK | W_OK) == 0)
160 pcg 1.4 {
161 pcg 1.6 *ttydev = strdup (tty_name);
162 pcg 1.4 return pfd;
163     }
164 root 1.16
165 pcg 1.6 close (pfd);
166 pcg 1.4 }
167     }
168     }
169 pcg 1.1 #endif
170 root 1.16
171 pcg 1.1 #ifdef PTYS_ARE_SEARCHED
172 pcg 1.4 {
173 root 1.16 const char *c1, *c2;
174     char pty_name[] = "/dev/pty??";
175     char tty_name[] = "/dev/tty??";
176 pcg 1.1
177     # ifndef PTYCHAR1
178     # define PTYCHAR1 "pqrstuvwxyz"
179     # endif
180     # ifndef PTYCHAR2
181     # define PTYCHAR2 "0123456789abcdef"
182     # endif
183 pcg 1.4
184     for (c1 = PTYCHAR1; *c1; c1++)
185     {
186 pcg 1.6 pty_name[ (sizeof (pty_name) - 3)] =
187     tty_name[ (sizeof (pty_name) - 3)] = *c1;
188 pcg 1.4 for (c2 = PTYCHAR2; *c2; c2++)
189     {
190 pcg 1.6 pty_name[ (sizeof (pty_name) - 2)] =
191     tty_name[ (sizeof (pty_name) - 2)] = *c2;
192     if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) >= 0)
193 pcg 1.4 {
194 pcg 1.6 if (access (tty_name, R_OK | W_OK) == 0)
195 pcg 1.4 {
196 pcg 1.6 *ttydev = strdup (tty_name);
197 pcg 1.4 return pfd;
198     }
199 root 1.16
200 pcg 1.6 close (pfd);
201 pcg 1.4 }
202     }
203     }
204     }
205 pcg 1.1 #endif
206 root 1.16
207 pcg 1.4 return -1;
208 pcg 1.1 }
209    
210     /*----------------------------------------------------------------------*/
211     /*
212     * Returns tty file descriptor, or -1 on failure
213     */
214 root 1.16 static int
215     get_tty (char *ttydev)
216 pcg 1.1 {
217 pcg 1.4 return open (ttydev, O_RDWR | O_NOCTTY, 0);
218 pcg 1.1 }
219    
220     /*----------------------------------------------------------------------*/
221     /*
222     * Make our tty a controlling tty so that /dev/tty points to us
223     */
224 root 1.16 static int
225 root 1.36 control_tty (int fd_tty)
226 pcg 1.1 {
227     #ifndef __QNX__
228 pcg 1.4 int fd;
229 pcg 1.1
230 pcg 1.4 /* ---------------------------------------- */
231 pcg 1.1 # ifdef HAVE_SETSID
232 pcg 1.4 setsid ();
233 pcg 1.1 # endif
234     # if defined(HAVE_SETPGID)
235 pcg 1.4 setpgid (0, 0);
236 pcg 1.1 # elif defined(HAVE_SETPGRP)
237 pcg 1.4 setpgrp (0, 0);
238 pcg 1.1 # endif
239 root 1.16
240 pcg 1.4 /* ---------------------------------------- */
241 pcg 1.1 # ifdef TIOCNOTTY
242 pcg 1.4 fd = open ("/dev/tty", O_RDWR | O_NOCTTY);
243     if (fd >= 0)
244     {
245     ioctl (fd, TIOCNOTTY, NULL); /* void tty associations */
246     close (fd);
247 pcg 1.1 }
248     # endif
249 root 1.16
250 pcg 1.4 /* ---------------------------------------- */
251     fd = open ("/dev/tty", O_RDWR | O_NOCTTY);
252     if (fd >= 0)
253     close (fd); /* ouch: still have controlling tty */
254 root 1.16
255 pcg 1.4 /* ---------------------------------------- */
256 root 1.36 # if defined(PTYS_ARE_PTMX) && defined(I_PUSH)
257 pcg 1.4 /*
258     * Push STREAMS modules:
259     * ptem: pseudo-terminal hardware emulation module.
260     * ldterm: standard terminal line discipline.
261     * ttcompat: V7, 4BSD and XENIX STREAMS compatibility module.
262     *
263 pcg 1.6 * After we push the STREAMS modules, the first open () on the slave side
264 pcg 1.4 * (i.e. the next section between the dashes giving us "tty opened OK")
265     * should make the "ptem" (or "ldterm" depending upon either which OS
266     * version or which set of manual pages you have) module give us a
267 pcg 1.6 * controlling terminal. We must already have close ()d the master side
268 pcg 1.4 * fd in this child process before we push STREAMS modules on because the
269 pcg 1.6 * documentation is really unclear about whether it is any close () on
270     * the master side or the last close () - i.e. a proper STREAMS dismantling
271     * close () - on the master side which causes a hang up to be sent
272 pcg 1.4 * through - Geoff Wing
273     */
274 root 1.36 # ifdef HAVE_ISASTREAM
275 pcg 1.4 if (isastream (fd_tty) == 1)
276 root 1.36 # endif
277 pcg 1.1 {
278 pcg 1.4 ioctl (fd_tty, I_PUSH, "ptem");
279     ioctl (fd_tty, I_PUSH, "ldterm");
280     ioctl (fd_tty, I_PUSH, "ttcompat");
281 pcg 1.1 }
282 root 1.36 # endif
283 pcg 1.4 /* ---------------------------------------- */
284     fd = ioctl (fd_tty, TIOCSCTTY, NULL);
285     /* ---------------------------------------- */
286     fd = open ("/dev/tty", O_WRONLY);
287     if (fd < 0)
288     return -1; /* fatal */
289     close (fd);
290     /* ---------------------------------------- */
291 pcg 1.1 #endif /* ! __QNX__ */
292 pcg 1.4
293     return 0;
294 pcg 1.1 }
295 root 1.16
296 root 1.36 void
297     rxvt_ptytty::close_tty ()
298     {
299     if (tty < 0)
300     return;
301    
302     close (tty);
303     tty = -1;
304     }
305    
306     bool
307     rxvt_ptytty::make_controlling_tty ()
308     {
309     return control_tty (tty) >= 0;
310     }
311    
312     void
313     rxvt_ptytty::set_utf8_mode (bool on)
314     {
315     #ifdef IUTF8
316     if (pty < 0)
317     return;
318    
319     struct termios tio;
320    
321     if (tcgetattr (pty, &tio) != -1)
322     {
323     tcflag_t new_cflag = tio.c_iflag;
324    
325     if (on)
326     new_cflag |= IUTF8;
327     else
328     new_cflag &= ~IUTF8;
329    
330     if (new_cflag != tio.c_iflag)
331     {
332     tio.c_iflag = new_cflag;
333     tcsetattr (pty, TCSANOW, &tio);
334     }
335     }
336     #endif
337     }
338    
339     /////////////////////////////////////////////////////////////////////////////
340    
341 root 1.16 #ifndef NO_SETOWNER_TTYDEV
342     static struct ttyconf {
343     gid_t gid;
344     mode_t mode;
345    
346     ttyconf ()
347     {
348     #ifdef TTY_GID_SUPPORT
349     struct group *gr = getgrnam ("tty");
350    
351     if (gr)
352     { /* change group ownership of tty to "tty" */
353     mode = S_IRUSR | S_IWUSR | S_IWGRP;
354     gid = gr->gr_gid;
355     }
356     else
357     #endif /* TTY_GID_SUPPORT */
358     {
359     mode = S_IRUSR | S_IWUSR | S_IWGRP | S_IWOTH;
360     gid = getgid ();
361     }
362     }
363     } ttyconf;
364    
365 root 1.36 /////////////////////////////////////////////////////////////////////////////
366    
367 root 1.16 void
368 root 1.36 rxvt_ptytty_unix::privileges (rxvt_privaction action)
369 root 1.16 {
370 root 1.26 if (!name || !*name)
371 root 1.23 return;
372    
373 root 1.16 rxvt_privileges (RESTORE);
374    
375     if (action == SAVE)
376     {
377     # ifndef RESET_TTY_TO_COMMON_DEFAULTS
378     /* store original tty status for restoration rxvt_clean_exit () -- rgg 04/12/95 */
379     if (lstat (name, &savestat) < 0) /* you lose out */
380 root 1.26 ;
381 root 1.16 else
382     # endif
383     {
384 root 1.26 saved = true;
385 root 1.16 chown (name, getuid (), ttyconf.gid); /* fail silently */
386     chmod (name, ttyconf.mode);
387     # ifdef HAVE_REVOKE
388     revoke (name);
389     # endif
390     }
391     }
392     else
393     { /* action == RESTORE */
394     # ifndef RESET_TTY_TO_COMMON_DEFAULTS
395 root 1.26 if (saved)
396     {
397     chmod (name, savestat.st_mode);
398     chown (name, savestat.st_uid, savestat.st_gid);
399     }
400 root 1.16 # else
401     chmod (name, (S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH));
402     chown (name, 0, 0);
403     # endif
404    
405     }
406    
407     rxvt_privileges (IGNORE);
408     }
409     #endif
410    
411 root 1.36 rxvt_ptytty_unix::rxvt_ptytty_unix ()
412 root 1.16 {
413     pty = tty = -1;
414     name = 0;
415 root 1.26 #ifndef NO_SETOWNER_TTYDEV
416     saved = false;
417     #endif
418 root 1.35 #if UTMP_SUPPORT
419     cmd_pid = 0;
420     #endif
421 root 1.16 }
422    
423 root 1.36 rxvt_ptytty_unix::~rxvt_ptytty_unix ()
424 root 1.16 {
425 root 1.35 logout ();
426 root 1.16 put ();
427     }
428    
429     void
430 root 1.36 rxvt_ptytty_unix::put ()
431 root 1.16 {
432     #ifndef NO_SETOWNER_TTYDEV
433 root 1.28 privileges (RESTORE);
434 root 1.16 #endif
435    
436     if (pty >= 0) close (pty);
437     close_tty ();
438     free (name);
439    
440     pty = tty = -1;
441     name = 0;
442     }
443    
444     bool
445 root 1.36 rxvt_ptytty_unix::get ()
446 root 1.16 {
447     /* get master (pty) */
448     if ((pty = get_pty (&tty, &name)) < 0)
449     return false;
450    
451     fcntl (pty, F_SETFL, O_NONBLOCK);
452    
453     /* get slave (tty) */
454     if (tty < 0)
455     {
456     #ifndef NO_SETOWNER_TTYDEV
457     privileges (SAVE);
458     #endif
459    
460     if ((tty = get_tty (name)) < 0)
461     {
462     put ();
463     return false;
464     }
465     }
466    
467     return true;
468     }
469    
470 pcg 1.1 /*----------------------- end-of-file (C source) -----------------------*/
471 root 1.16