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

File Contents

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