ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/rxvt-unicode/src/ptytty.C
Revision: 1.27
Committed: Tue Jan 3 16:32:33 2006 UTC (18 years, 4 months ago) by root
Content type: text/plain
Branch: MAIN
Changes since 1.26: +5 -1 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 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 rxvt_privileges(RESTORE);
82 int res = openpty (&pfd, fd_tty, tty_name, NULL, NULL);
83 rxvt_privileges(IGNORE);
84
85 if (res != -1)
86 {
87 *ttydev = strdup (tty_name);
88 return pfd;
89 }
90 #endif
91
92 #ifdef PTYS_ARE__GETPTY
93 *ttydev = _getpty (&pfd, O_RDWR | O_NONBLOCK | O_NOCTTY, 0622, 0);
94 if (*ttydev != NULL)
95 return pfd;
96 #endif
97
98 #ifdef PTYS_ARE_GETPTY
99 char *ptydev;
100
101 while ((ptydev = getpty ()) != NULL)
102 if ((pfd = open (ptydev, O_RDWR | O_NOCTTY, 0)) >= 0)
103 {
104 *ttydev = ptydev;
105 return pfd;
106 }
107 #endif
108
109 #if defined(HAVE_GRANTPT) && defined(HAVE_UNLOCKPT)
110 # if defined(PTYS_ARE_GETPT) || defined(PTYS_ARE_PTMX)
111
112 {
113 # ifdef PTYS_ARE_GETPT
114 pfd = getpt ();
115 # else
116 pfd = open ("/dev/ptmx", O_RDWR | O_NOCTTY, 0);
117 # endif
118
119 if (pfd >= 0)
120 {
121 if (grantpt (pfd) == 0 /* change slave permissions */
122 && unlockpt (pfd) == 0)
123 { /* slave now unlocked */
124 *ttydev = strdup (ptsname (pfd)); /* get slave's name */
125 return pfd;
126 }
127 close (pfd);
128 }
129 }
130 # endif
131 #endif
132
133 #ifdef PTYS_ARE_PTC
134 if ((pfd = open ("/dev/ptc", O_RDWR | O_NOCTTY, 0)) >= 0)
135 {
136 *ttydev = strdup (ttyname (pfd));
137 return pfd;
138 }
139 #endif
140
141 #ifdef PTYS_ARE_CLONE
142 if ((pfd = open ("/dev/ptym/clone", O_RDWR | O_NOCTTY, 0)) >= 0)
143 {
144 *ttydev = strdup (ptsname (pfd));
145 return pfd;
146 }
147 #endif
148
149 #ifdef PTYS_ARE_NUMERIC
150 {
151 int idx;
152 char *c1, *c2;
153 char pty_name[] = "/dev/ptyp???";
154 char tty_name[] = "/dev/ttyp???";
155
156 c1 = &(pty_name[sizeof (pty_name) - 4]);
157 c2 = &(tty_name[sizeof (tty_name) - 4]);
158 for (idx = 0; idx < 256; idx++)
159 {
160 sprintf (c1, "%d", idx);
161 sprintf (c2, "%d", idx);
162 if (access (tty_name, F_OK) < 0)
163 {
164 idx = 256;
165 break;
166 }
167
168 if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) >= 0)
169 {
170 if (access (tty_name, R_OK | W_OK) == 0)
171 {
172 *ttydev = strdup (tty_name);
173 return pfd;
174 }
175
176 close (pfd);
177 }
178 }
179 }
180 #endif
181
182 #ifdef PTYS_ARE_SEARCHED
183 {
184 const char *c1, *c2;
185 char pty_name[] = "/dev/pty??";
186 char tty_name[] = "/dev/tty??";
187
188 # ifndef PTYCHAR1
189 # define PTYCHAR1 "pqrstuvwxyz"
190 # endif
191 # ifndef PTYCHAR2
192 # define PTYCHAR2 "0123456789abcdef"
193 # endif
194
195 for (c1 = PTYCHAR1; *c1; c1++)
196 {
197 pty_name[ (sizeof (pty_name) - 3)] =
198 tty_name[ (sizeof (pty_name) - 3)] = *c1;
199 for (c2 = PTYCHAR2; *c2; c2++)
200 {
201 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 {
205 if (access (tty_name, R_OK | W_OK) == 0)
206 {
207 *ttydev = strdup (tty_name);
208 return pfd;
209 }
210
211 close (pfd);
212 }
213 }
214 }
215 }
216 #endif
217
218 return -1;
219 }
220
221 /*----------------------------------------------------------------------*/
222 /*
223 * Returns tty file descriptor, or -1 on failure
224 */
225 static int
226 get_tty (char *ttydev)
227 {
228 return open (ttydev, O_RDWR | O_NOCTTY, 0);
229 }
230
231 /*----------------------------------------------------------------------*/
232 /*
233 * Make our tty a controlling tty so that /dev/tty points to us
234 */
235 static int
236 control_tty (int fd_tty, const char *ttydev)
237 {
238 #ifndef __QNX__
239 int fd;
240
241 /* ---------------------------------------- */
242 # ifdef HAVE_SETSID
243 setsid ();
244 # endif
245 # if defined(HAVE_SETPGID)
246 setpgid (0, 0);
247 # elif defined(HAVE_SETPGRP)
248 setpgrp (0, 0);
249 # endif
250
251 /* ---------------------------------------- */
252 # ifdef TIOCNOTTY
253 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 }
259 # endif
260
261 /* ---------------------------------------- */
262 fd = open ("/dev/tty", O_RDWR | O_NOCTTY);
263 if (fd >= 0)
264 close (fd); /* ouch: still have controlling tty */
265
266 /* ---------------------------------------- */
267 #if defined(PTYS_ARE_PTMX) && defined(I_PUSH)
268 /*
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 * After we push the STREAMS modules, the first open () on the slave side
275 * (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 * controlling terminal. We must already have close ()d the master side
279 * fd in this child process before we push STREAMS modules on because the
280 * 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 * through - Geoff Wing
284 */
285 # ifdef HAVE_ISASTREAM
286 if (isastream (fd_tty) == 1)
287 # endif
288 {
289 ioctl (fd_tty, I_PUSH, "ptem");
290 ioctl (fd_tty, I_PUSH, "ldterm");
291 ioctl (fd_tty, I_PUSH, "ttcompat");
292 }
293 #endif
294 /* ---------------------------------------- */
295 # if defined(TIOCSCTTY)
296 fd = ioctl (fd_tty, TIOCSCTTY, NULL);
297 # elif defined(TIOCSETCTTY)
298 fd = ioctl (fd_tty, TIOCSETCTTY, NULL);
299 # else
300 fd = open (ttydev, O_RDWR);
301 if (fd >= 0)
302 close (fd);
303 # endif
304 /* ---------------------------------------- */
305 fd = open ("/dev/tty", O_WRONLY);
306 if (fd < 0)
307 return -1; /* fatal */
308 close (fd);
309 /* ---------------------------------------- */
310 #endif /* ! __QNX__ */
311
312 return 0;
313 }
314
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 if (!name || !*name)
343 return;
344
345 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 ;
353 else
354 # endif
355 {
356 saved = true;
357 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 if (saved)
368 {
369 chmod (name, savestat.st_mode);
370 chown (name, savestat.st_uid, savestat.st_gid);
371 }
372 # 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 #ifndef NO_SETOWNER_TTYDEV
388 saved = false;
389 #endif
390 }
391
392 rxvt_ptytty::~rxvt_ptytty ()
393 {
394 put ();
395 }
396
397 void
398
399 rxvt_ptytty::close_tty ()
400 {
401 if (tty >= 0) close (tty);
402 tty = -1;
403 }
404
405 void
406 rxvt_ptytty::put ()
407 {
408 #ifndef NO_SETOWNER_TTYDEV
409 if (tty >= 0)
410 privileges (RESTORE);
411 #endif
412
413 if (pty >= 0) close (pty);
414 close_tty ();
415 free (name);
416
417 pty = tty = -1;
418 name = 0;
419 }
420
421 bool
422 rxvt_ptytty::make_controlling_tty ()
423 {
424 return control_tty (tty, name) >= 0;
425 }
426
427 bool
428 rxvt_ptytty::get ()
429 {
430 /* get master (pty) */
431 if ((pty = get_pty (&tty, &name)) < 0)
432 return false;
433
434 fcntl (pty, F_SETFL, O_NONBLOCK);
435
436 /* get slave (tty) */
437 if (tty < 0)
438 {
439 #ifndef NO_SETOWNER_TTYDEV
440 privileges (SAVE);
441 #endif
442
443 if ((tty = get_tty (name)) < 0)
444 {
445 put ();
446 return false;
447 }
448 }
449
450 return true;
451 }
452
453 void
454 rxvt_ptytty::set_utf8_mode (bool on)
455 {
456 #ifdef IUTF8
457 if (pty != -1)
458 {
459 struct termios tio;
460
461 if (tcgetattr (pty, &tio) != -1)
462 {
463 tcflag_t new_cflag = tio.c_iflag;
464
465 if (on)
466 new_cflag |= IUTF8;
467 else
468 new_cflag &= ~IUTF8;
469
470 if (new_cflag != tio.c_iflag)
471 {
472 tio.c_iflag = new_cflag;
473 tcsetattr (pty, TCSANOW, &tio);
474 }
475 }
476 }
477 #endif
478 }
479
480 /*----------------------- end-of-file (C source) -----------------------*/
481