ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/rxvt-unicode/src/ptytty.C
Revision: 1.31
Committed: Tue Jan 10 18:09:22 2006 UTC (18 years, 4 months ago) by root
Content type: text/plain
Branch: MAIN
Changes since 1.30: +5 -11 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 pcg 1.1 /* ------------------------------------------------------------------------- *
60     * GET PSEUDO TELETYPE - MASTER AND SLAVE *
61     * ------------------------------------------------------------------------- */
62     /*
63     * Returns pty file descriptor, or -1 on failure
64     * 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
66     */
67 root 1.16 static int
68     get_pty (int *fd_tty, char **ttydev)
69 pcg 1.1 {
70 pcg 1.4 int pfd;
71 pcg 1.1
72     #ifdef PTYS_ARE_OPENPTY
73 root 1.26 char tty_name[sizeof "/dev/pts/????\0"];
74 pcg 1.4
75 root 1.27 rxvt_privileges(RESTORE);
76     int res = openpty (&pfd, fd_tty, tty_name, NULL, NULL);
77     rxvt_privileges(IGNORE);
78    
79     if (res != -1)
80 pcg 1.4 {
81 pcg 1.6 *ttydev = strdup (tty_name);
82 pcg 1.4 return pfd;
83 pcg 1.1 }
84     #endif
85    
86     #ifdef PTYS_ARE__GETPTY
87 pcg 1.6 *ttydev = _getpty (&pfd, O_RDWR | O_NONBLOCK | O_NOCTTY, 0622, 0);
88 pcg 1.4 if (*ttydev != NULL)
89     return pfd;
90 pcg 1.1 #endif
91    
92     #ifdef PTYS_ARE_GETPTY
93 pcg 1.4 char *ptydev;
94    
95 pcg 1.6 while ((ptydev = getpty ()) != NULL)
96     if ((pfd = open (ptydev, O_RDWR | O_NOCTTY, 0)) >= 0)
97 pcg 1.4 {
98     *ttydev = ptydev;
99     return pfd;
100     }
101 pcg 1.1 #endif
102    
103     #if defined(HAVE_GRANTPT) && defined(HAVE_UNLOCKPT)
104     # if defined(PTYS_ARE_GETPT) || defined(PTYS_ARE_PTMX)
105 pcg 1.4
106     {
107 pcg 1.1 # ifdef PTYS_ARE_GETPT
108 pcg 1.6 pfd = getpt ();
109 pcg 1.1 # else
110 pcg 1.6 pfd = open ("/dev/ptmx", O_RDWR | O_NOCTTY, 0);
111 pcg 1.1 # endif
112 pcg 1.4
113     if (pfd >= 0)
114     {
115 pcg 1.6 if (grantpt (pfd) == 0 /* change slave permissions */
116     && unlockpt (pfd) == 0)
117 pcg 1.4 { /* slave now unlocked */
118 root 1.16 *ttydev = strdup (ptsname (pfd)); /* get slave's name */
119 pcg 1.4 return pfd;
120     }
121 pcg 1.6 close (pfd);
122 pcg 1.4 }
123     }
124 pcg 1.1 # endif
125     #endif
126    
127     #ifdef PTYS_ARE_PTC
128 pcg 1.6 if ((pfd = open ("/dev/ptc", O_RDWR | O_NOCTTY, 0)) >= 0)
129 pcg 1.4 {
130 root 1.16 *ttydev = strdup (ttyname (pfd));
131 pcg 1.4 return pfd;
132 pcg 1.1 }
133     #endif
134    
135     #ifdef PTYS_ARE_CLONE
136 pcg 1.6 if ((pfd = open ("/dev/ptym/clone", O_RDWR | O_NOCTTY, 0)) >= 0)
137 pcg 1.4 {
138 root 1.16 *ttydev = strdup (ptsname (pfd));
139 pcg 1.4 return pfd;
140 pcg 1.1 }
141     #endif
142    
143     #ifdef PTYS_ARE_NUMERIC
144 pcg 1.4 {
145 root 1.16 int idx;
146     char *c1, *c2;
147     char pty_name[] = "/dev/ptyp???";
148     char tty_name[] = "/dev/ttyp???";
149 pcg 1.4
150 root 1.16 c1 = &(pty_name[sizeof (pty_name) - 4]);
151     c2 = &(tty_name[sizeof (tty_name) - 4]);
152 pcg 1.4 for (idx = 0; idx < 256; idx++)
153     {
154 pcg 1.6 sprintf (c1, "%d", idx);
155     sprintf (c2, "%d", idx);
156     if (access (tty_name, F_OK) < 0)
157 pcg 1.4 {
158     idx = 256;
159     break;
160     }
161 root 1.16
162 pcg 1.6 if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) >= 0)
163 pcg 1.4 {
164 pcg 1.6 if (access (tty_name, R_OK | W_OK) == 0)
165 pcg 1.4 {
166 pcg 1.6 *ttydev = strdup (tty_name);
167 pcg 1.4 return pfd;
168     }
169 root 1.16
170 pcg 1.6 close (pfd);
171 pcg 1.4 }
172     }
173     }
174 pcg 1.1 #endif
175 root 1.16
176 pcg 1.1 #ifdef PTYS_ARE_SEARCHED
177 pcg 1.4 {
178 root 1.16 const char *c1, *c2;
179     char pty_name[] = "/dev/pty??";
180     char tty_name[] = "/dev/tty??";
181 pcg 1.1
182     # ifndef PTYCHAR1
183     # define PTYCHAR1 "pqrstuvwxyz"
184     # endif
185     # ifndef PTYCHAR2
186     # define PTYCHAR2 "0123456789abcdef"
187     # endif
188 pcg 1.4
189     for (c1 = PTYCHAR1; *c1; c1++)
190     {
191 pcg 1.6 pty_name[ (sizeof (pty_name) - 3)] =
192     tty_name[ (sizeof (pty_name) - 3)] = *c1;
193 pcg 1.4 for (c2 = PTYCHAR2; *c2; c2++)
194     {
195 pcg 1.6 pty_name[ (sizeof (pty_name) - 2)] =
196     tty_name[ (sizeof (pty_name) - 2)] = *c2;
197     if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) >= 0)
198 pcg 1.4 {
199 pcg 1.6 if (access (tty_name, R_OK | W_OK) == 0)
200 pcg 1.4 {
201 pcg 1.6 *ttydev = strdup (tty_name);
202 pcg 1.4 return pfd;
203     }
204 root 1.16
205 pcg 1.6 close (pfd);
206 pcg 1.4 }
207     }
208     }
209     }
210 pcg 1.1 #endif
211 root 1.16
212 pcg 1.4 return -1;
213 pcg 1.1 }
214    
215     /*----------------------------------------------------------------------*/
216     /*
217     * Returns tty file descriptor, or -1 on failure
218     */
219 root 1.16 static int
220     get_tty (char *ttydev)
221 pcg 1.1 {
222 pcg 1.4 return open (ttydev, O_RDWR | O_NOCTTY, 0);
223 pcg 1.1 }
224    
225     /*----------------------------------------------------------------------*/
226     /*
227     * Make our tty a controlling tty so that /dev/tty points to us
228     */
229 root 1.16 static int
230     control_tty (int fd_tty, const char *ttydev)
231 pcg 1.1 {
232     #ifndef __QNX__
233 pcg 1.4 int fd;
234 pcg 1.1
235 pcg 1.4 /* ---------------------------------------- */
236 pcg 1.1 # ifdef HAVE_SETSID
237 pcg 1.4 setsid ();
238 pcg 1.1 # endif
239     # if defined(HAVE_SETPGID)
240 pcg 1.4 setpgid (0, 0);
241 pcg 1.1 # elif defined(HAVE_SETPGRP)
242 pcg 1.4 setpgrp (0, 0);
243 pcg 1.1 # endif
244 root 1.16
245 pcg 1.4 /* ---------------------------------------- */
246 pcg 1.1 # ifdef TIOCNOTTY
247 pcg 1.4 fd = open ("/dev/tty", O_RDWR | O_NOCTTY);
248     if (fd >= 0)
249     {
250     ioctl (fd, TIOCNOTTY, NULL); /* void tty associations */
251     close (fd);
252 pcg 1.1 }
253     # endif
254 root 1.16
255 pcg 1.4 /* ---------------------------------------- */
256     fd = open ("/dev/tty", O_RDWR | O_NOCTTY);
257     if (fd >= 0)
258     close (fd); /* ouch: still have controlling tty */
259 root 1.16
260 pcg 1.4 /* ---------------------------------------- */
261 pcg 1.1 #if defined(PTYS_ARE_PTMX) && defined(I_PUSH)
262 pcg 1.4 /*
263     * Push STREAMS modules:
264     * ptem: pseudo-terminal hardware emulation module.
265     * ldterm: standard terminal line discipline.
266     * ttcompat: V7, 4BSD and XENIX STREAMS compatibility module.
267     *
268 pcg 1.6 * After we push the STREAMS modules, the first open () on the slave side
269 pcg 1.4 * (i.e. the next section between the dashes giving us "tty opened OK")
270     * should make the "ptem" (or "ldterm" depending upon either which OS
271     * version or which set of manual pages you have) module give us a
272 pcg 1.6 * controlling terminal. We must already have close ()d the master side
273 pcg 1.4 * fd in this child process before we push STREAMS modules on because the
274 pcg 1.6 * documentation is really unclear about whether it is any close () on
275     * the master side or the last close () - i.e. a proper STREAMS dismantling
276     * close () - on the master side which causes a hang up to be sent
277 pcg 1.4 * through - Geoff Wing
278     */
279 pcg 1.1 # ifdef HAVE_ISASTREAM
280 pcg 1.4 if (isastream (fd_tty) == 1)
281 pcg 1.1 # endif
282     {
283 pcg 1.4 ioctl (fd_tty, I_PUSH, "ptem");
284     ioctl (fd_tty, I_PUSH, "ldterm");
285     ioctl (fd_tty, I_PUSH, "ttcompat");
286 pcg 1.1 }
287     #endif
288 pcg 1.4 /* ---------------------------------------- */
289 pcg 1.1 # if defined(TIOCSCTTY)
290 pcg 1.4 fd = ioctl (fd_tty, TIOCSCTTY, NULL);
291 pcg 1.1 # elif defined(TIOCSETCTTY)
292 pcg 1.4 fd = ioctl (fd_tty, TIOCSETCTTY, NULL);
293 pcg 1.1 # else
294 pcg 1.4 fd = open (ttydev, O_RDWR);
295     if (fd >= 0)
296 pcg 1.6 close (fd);
297 pcg 1.1 # endif
298 pcg 1.4 /* ---------------------------------------- */
299     fd = open ("/dev/tty", O_WRONLY);
300     if (fd < 0)
301     return -1; /* fatal */
302     close (fd);
303     /* ---------------------------------------- */
304 pcg 1.1 #endif /* ! __QNX__ */
305 pcg 1.4
306     return 0;
307 pcg 1.1 }
308 root 1.16
309     #ifndef NO_SETOWNER_TTYDEV
310     static struct ttyconf {
311     gid_t gid;
312     mode_t mode;
313    
314     ttyconf ()
315     {
316     #ifdef TTY_GID_SUPPORT
317     struct group *gr = getgrnam ("tty");
318    
319     if (gr)
320     { /* change group ownership of tty to "tty" */
321     mode = S_IRUSR | S_IWUSR | S_IWGRP;
322     gid = gr->gr_gid;
323     }
324     else
325     #endif /* TTY_GID_SUPPORT */
326     {
327     mode = S_IRUSR | S_IWUSR | S_IWGRP | S_IWOTH;
328     gid = getgid ();
329     }
330     }
331     } ttyconf;
332    
333     void
334     rxvt_ptytty::privileges (rxvt_privaction action)
335     {
336 root 1.26 if (!name || !*name)
337 root 1.23 return;
338    
339 root 1.16 rxvt_privileges (RESTORE);
340    
341     if (action == SAVE)
342     {
343     # ifndef RESET_TTY_TO_COMMON_DEFAULTS
344     /* store original tty status for restoration rxvt_clean_exit () -- rgg 04/12/95 */
345     if (lstat (name, &savestat) < 0) /* you lose out */
346 root 1.26 ;
347 root 1.16 else
348     # endif
349     {
350 root 1.26 saved = true;
351 root 1.16 chown (name, getuid (), ttyconf.gid); /* fail silently */
352     chmod (name, ttyconf.mode);
353     # ifdef HAVE_REVOKE
354     revoke (name);
355     # endif
356     }
357     }
358     else
359     { /* action == RESTORE */
360     # ifndef RESET_TTY_TO_COMMON_DEFAULTS
361 root 1.26 if (saved)
362     {
363     chmod (name, savestat.st_mode);
364     chown (name, savestat.st_uid, savestat.st_gid);
365     }
366 root 1.16 # else
367     chmod (name, (S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH));
368     chown (name, 0, 0);
369     # endif
370    
371     }
372    
373     rxvt_privileges (IGNORE);
374     }
375     #endif
376    
377     rxvt_ptytty::rxvt_ptytty ()
378     {
379     pty = tty = -1;
380     name = 0;
381 root 1.26 #ifndef NO_SETOWNER_TTYDEV
382     saved = false;
383     #endif
384 root 1.16 }
385    
386     rxvt_ptytty::~rxvt_ptytty ()
387     {
388     put ();
389     }
390    
391     void
392    
393     rxvt_ptytty::close_tty ()
394     {
395 root 1.29 if (tty < 0)
396     return;
397    
398     close (tty);
399 root 1.16 tty = -1;
400     }
401    
402     void
403     rxvt_ptytty::put ()
404     {
405     #ifndef NO_SETOWNER_TTYDEV
406 root 1.28 privileges (RESTORE);
407 root 1.16 #endif
408    
409     if (pty >= 0) close (pty);
410     close_tty ();
411     free (name);
412    
413     pty = tty = -1;
414     name = 0;
415     }
416    
417     bool
418     rxvt_ptytty::make_controlling_tty ()
419     {
420     return control_tty (tty, name) >= 0;
421     }
422    
423     bool
424     rxvt_ptytty::get ()
425     {
426     /* get master (pty) */
427     if ((pty = get_pty (&tty, &name)) < 0)
428     return false;
429    
430     fcntl (pty, F_SETFL, O_NONBLOCK);
431    
432     /* get slave (tty) */
433     if (tty < 0)
434     {
435     #ifndef NO_SETOWNER_TTYDEV
436     privileges (SAVE);
437     #endif
438    
439     if ((tty = get_tty (name)) < 0)
440     {
441     put ();
442     return false;
443     }
444     }
445    
446     return true;
447     }
448    
449 root 1.22 void
450     rxvt_ptytty::set_utf8_mode (bool on)
451     {
452     #ifdef IUTF8
453 root 1.29 if (pty < 0)
454     return;
455    
456     struct termios tio;
457    
458     if (tcgetattr (pty, &tio) != -1)
459 root 1.22 {
460 root 1.29 tcflag_t new_cflag = tio.c_iflag;
461    
462     if (on)
463     new_cflag |= IUTF8;
464     else
465     new_cflag &= ~IUTF8;
466 root 1.22
467 root 1.29 if (new_cflag != tio.c_iflag)
468 root 1.22 {
469 root 1.29 tio.c_iflag = new_cflag;
470     tcsetattr (pty, TCSANOW, &tio);
471 root 1.22 }
472     }
473     #endif
474     }
475    
476 pcg 1.1 /*----------------------- end-of-file (C source) -----------------------*/
477 root 1.16