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