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

# 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
61 /* ------------------------------------------------------------------------- *
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 static int
70 get_pty (int *fd_tty, char **ttydev)
71 {
72 int pfd;
73
74 #ifdef PTYS_ARE_OPENPTY
75 char tty_name[sizeof "/dev/pts/????\0"];
76
77 rxvt_privileges(RESTORE);
78 int res = openpty (&pfd, fd_tty, tty_name, NULL, NULL);
79 rxvt_privileges(IGNORE);
80
81 if (res != -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 #if defined(HAVE_GRANTPT) && defined(HAVE_UNLOCKPT)
95 # if defined(PTYS_ARE_GETPT) || defined(PTYS_ARE_POSIX) || defined(PTYS_ARE_PTMX)
96
97 {
98 # ifdef PTYS_ARE_GETPT
99 pfd = getpt();
100 # else
101 # ifdef PTYS_ARE_POSIX
102 pfd = posix_openpt (O_RDWR);
103 # else
104 pfd = open ("/dev/ptmx", O_RDWR | O_NOCTTY, 0);
105 # endif
106 # endif
107
108 if (pfd >= 0)
109 {
110 if (grantpt (pfd) == 0 /* change slave permissions */
111 && unlockpt (pfd) == 0)
112 { /* slave now unlocked */
113 *ttydev = strdup (ptsname (pfd)); /* get slave's name */
114 return pfd;
115 }
116 close (pfd);
117 }
118 }
119 # endif
120 #endif
121
122 #ifdef PTYS_ARE_PTC
123 if ((pfd = open ("/dev/ptc", O_RDWR | O_NOCTTY, 0)) >= 0)
124 {
125 *ttydev = strdup (ttyname (pfd));
126 return pfd;
127 }
128 #endif
129
130 #ifdef PTYS_ARE_CLONE
131 if ((pfd = open ("/dev/ptym/clone", O_RDWR | O_NOCTTY, 0)) >= 0)
132 {
133 *ttydev = strdup (ptsname (pfd));
134 return pfd;
135 }
136 #endif
137
138 #ifdef PTYS_ARE_NUMERIC
139 {
140 int idx;
141 char *c1, *c2;
142 char pty_name[] = "/dev/ptyp???";
143 char tty_name[] = "/dev/ttyp???";
144
145 c1 = &(pty_name[sizeof (pty_name) - 4]);
146 c2 = &(tty_name[sizeof (tty_name) - 4]);
147 for (idx = 0; idx < 256; idx++)
148 {
149 sprintf (c1, "%d", idx);
150 sprintf (c2, "%d", idx);
151 if (access (tty_name, F_OK) < 0)
152 {
153 idx = 256;
154 break;
155 }
156
157 if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) >= 0)
158 {
159 if (access (tty_name, R_OK | W_OK) == 0)
160 {
161 *ttydev = strdup (tty_name);
162 return pfd;
163 }
164
165 close (pfd);
166 }
167 }
168 }
169 #endif
170
171 #ifdef PTYS_ARE_SEARCHED
172 {
173 const char *c1, *c2;
174 char pty_name[] = "/dev/pty??";
175 char tty_name[] = "/dev/tty??";
176
177 # ifndef PTYCHAR1
178 # define PTYCHAR1 "pqrstuvwxyz"
179 # endif
180 # ifndef PTYCHAR2
181 # define PTYCHAR2 "0123456789abcdef"
182 # endif
183
184 for (c1 = PTYCHAR1; *c1; c1++)
185 {
186 pty_name[ (sizeof (pty_name) - 3)] =
187 tty_name[ (sizeof (pty_name) - 3)] = *c1;
188 for (c2 = PTYCHAR2; *c2; c2++)
189 {
190 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 {
194 if (access (tty_name, R_OK | W_OK) == 0)
195 {
196 *ttydev = strdup (tty_name);
197 return pfd;
198 }
199
200 close (pfd);
201 }
202 }
203 }
204 }
205 #endif
206
207 return -1;
208 }
209
210 /*----------------------------------------------------------------------*/
211 /*
212 * Returns tty file descriptor, or -1 on failure
213 */
214 static int
215 get_tty (char *ttydev)
216 {
217 return open (ttydev, O_RDWR | O_NOCTTY, 0);
218 }
219
220 /*----------------------------------------------------------------------*/
221 /*
222 * Make our tty a controlling tty so that /dev/tty points to us
223 */
224 static int
225 control_tty (int fd_tty)
226 {
227 #ifndef __QNX__
228 int fd;
229
230 /* ---------------------------------------- */
231 # ifdef HAVE_SETSID
232 setsid ();
233 # endif
234 # if defined(HAVE_SETPGID)
235 setpgid (0, 0);
236 # elif defined(HAVE_SETPGRP)
237 setpgrp (0, 0);
238 # endif
239
240 /* ---------------------------------------- */
241 # ifdef TIOCNOTTY
242 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 }
248 # endif
249
250 /* ---------------------------------------- */
251 fd = open ("/dev/tty", O_RDWR | O_NOCTTY);
252 if (fd >= 0)
253 close (fd); /* ouch: still have controlling tty */
254
255 /* ---------------------------------------- */
256 # if defined(PTYS_ARE_PTMX) && defined(I_PUSH)
257 /*
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 * After we push the STREAMS modules, the first open () on the slave side
264 * (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 * controlling terminal. We must already have close ()d the master side
268 * fd in this child process before we push STREAMS modules on because the
269 * 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 * through - Geoff Wing
273 */
274 # ifdef HAVE_ISASTREAM
275 if (isastream (fd_tty) == 1)
276 # endif
277 {
278 ioctl (fd_tty, I_PUSH, "ptem");
279 ioctl (fd_tty, I_PUSH, "ldterm");
280 ioctl (fd_tty, I_PUSH, "ttcompat");
281 }
282 # endif
283 /* ---------------------------------------- */
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 #endif /* ! __QNX__ */
292
293 return 0;
294 }
295
296 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 #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 /////////////////////////////////////////////////////////////////////////////
366
367 void
368 rxvt_ptytty_unix::privileges (rxvt_privaction action)
369 {
370 if (!name || !*name)
371 return;
372
373 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 ;
381 else
382 # endif
383 {
384 saved = true;
385 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 if (saved)
396 {
397 chmod (name, savestat.st_mode);
398 chown (name, savestat.st_uid, savestat.st_gid);
399 }
400 # 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 rxvt_ptytty_unix::rxvt_ptytty_unix ()
412 {
413 pty = tty = -1;
414 name = 0;
415 #ifndef NO_SETOWNER_TTYDEV
416 saved = false;
417 #endif
418 #if UTMP_SUPPORT
419 cmd_pid = 0;
420 #endif
421 }
422
423 rxvt_ptytty_unix::~rxvt_ptytty_unix ()
424 {
425 logout ();
426 put ();
427 }
428
429 void
430 rxvt_ptytty_unix::put ()
431 {
432 #ifndef NO_SETOWNER_TTYDEV
433 privileges (RESTORE);
434 #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 rxvt_ptytty_unix::get ()
446 {
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 /*----------------------- end-of-file (C source) -----------------------*/
471