ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/libptytty/src/ptytty.C
Revision: 1.1
Committed: Sat Jan 21 21:40:00 2006 UTC (18 years, 5 months ago) by root
Content type: text/plain
Branch: MAIN
Log Message:
*** empty log message ***

File Contents

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