|
|
1 | // This file is part of libptytty. Do not make local modifications. |
|
|
2 | // http://software.schmorp.de/pkg/libptytty |
|
|
3 | |
1 | /*--------------------------------*-C-*---------------------------------* |
4 | /*----------------------------------------------------------------------* |
2 | * File: ptytty.C |
5 | * File: ptytty.C |
3 | *----------------------------------------------------------------------* |
6 | *----------------------------------------------------------------------* |
4 | * |
7 | * |
5 | * All portions of code are copyright by their respective author/s. |
8 | * All portions of code are copyright by their respective author/s. |
6 | * Copyright (c) 1999-2001 Geoff Wing <gcw@pobox.com> |
9 | * Copyright (c) 1999-2001 Geoff Wing <gcw@pobox.com> |
7 | * Copyright (c) 2004-2006 Marc Lehmann <pcg@goof.com> |
10 | * Copyright (c) 2004-2006 Marc Lehmann <pcg@goof.com> |
|
|
11 | * Copyright (c) 2006 Emanuele Giaquinta <e.giaquinta@glauco.it> |
8 | * |
12 | * |
9 | * This program is free software; you can redistribute it and/or modify |
13 | * 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 |
14 | * 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 |
15 | * the Free Software Foundation; either version 2 of the License, or |
12 | * (at your option) any later version. |
16 | * (at your option) any later version. |
… | |
… | |
19 | * You should have received a copy of the GNU General Public License |
23 | * You should have received a copy of the GNU General Public License |
20 | * along with this program; if not, write to the Free Software |
24 | * along with this program; if not, write to the Free Software |
21 | * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. |
25 | * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. |
22 | *---------------------------------------------------------------------*/ |
26 | *---------------------------------------------------------------------*/ |
23 | |
27 | |
24 | #include "../config.h" /* NECESSARY */ |
28 | #include "../config.h" |
|
|
29 | |
25 | #include "rxvt.h" |
30 | #include "ptytty.h" |
26 | |
31 | |
27 | #include <cstdlib> |
32 | #include <cstdlib> |
28 | #include <cstring> |
33 | #include <cstring> |
|
|
34 | #include <csignal> |
29 | |
35 | |
30 | #include <sys/types.h> |
36 | #include <sys/types.h> |
31 | #include <sys/socket.h> |
|
|
32 | #include <unistd.h> |
37 | #include <unistd.h> |
33 | #include <fcntl.h> |
38 | #include <fcntl.h> |
34 | |
39 | |
35 | #ifdef HAVE_SYS_IOCTL_H |
40 | #ifdef HAVE_SYS_IOCTL_H |
36 | # include <sys/ioctl.h> |
41 | # include <sys/ioctl.h> |
37 | #endif |
42 | #endif |
38 | #if defined(PTYS_ARE_PTMX) && defined(HAVE_SYS_STROPTS_H) |
43 | #if defined(HAVE_DEV_PTMX) && defined(HAVE_SYS_STROPTS_H) |
39 | # include <sys/stropts.h> /* for I_PUSH */ |
44 | # include <sys/stropts.h> /* for I_PUSH */ |
40 | #endif |
45 | #endif |
41 | #ifdef HAVE_ISASTREAM |
46 | #ifdef HAVE_ISASTREAM |
42 | # include <stropts.h> |
47 | # include <stropts.h> |
43 | #endif |
48 | #endif |
… | |
… | |
46 | #elif defined(HAVE_LIBUTIL_H) |
51 | #elif defined(HAVE_LIBUTIL_H) |
47 | # include <libutil.h> |
52 | # include <libutil.h> |
48 | #elif defined(HAVE_UTIL_H) |
53 | #elif defined(HAVE_UTIL_H) |
49 | # include <util.h> |
54 | # include <util.h> |
50 | #endif |
55 | #endif |
|
|
56 | #ifdef TTY_GID_SUPPORT |
|
|
57 | #include <grp.h> |
|
|
58 | #endif |
51 | |
59 | |
52 | #include <cstdio> |
60 | #include <cstdio> |
53 | #include <grp.h> |
|
|
54 | |
|
|
55 | #include "rxvtutil.h" |
|
|
56 | #include "fdpass.h" |
|
|
57 | #include "ptytty.h" |
|
|
58 | |
61 | |
59 | ///////////////////////////////////////////////////////////////////////////// |
62 | ///////////////////////////////////////////////////////////////////////////// |
60 | |
63 | |
61 | /* ------------------------------------------------------------------------- * |
64 | /* ------------------------------------------------------------------------- * |
62 | * GET PSEUDO TELETYPE - MASTER AND SLAVE * |
65 | * GET PSEUDO TELETYPE - MASTER AND SLAVE * |
… | |
… | |
64 | /* |
67 | /* |
65 | * Returns pty file descriptor, or -1 on failure |
68 | * Returns pty file descriptor, or -1 on failure |
66 | * If successful, ttydev is set to the name of the slave device. |
69 | * 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 |
70 | * fd_tty _may_ also be set to an open fd to the slave device |
68 | */ |
71 | */ |
69 | static inline int |
72 | #if defined(UNIX98_PTY) |
|
|
73 | static int |
70 | get_pty_streams (int *fd_tty, char **ttydev) |
74 | get_pty (int *fd_tty, char **ttydev) |
71 | { |
75 | { |
72 | #ifdef NO_SETOWNER_TTYDEV |
|
|
73 | int pfd; |
76 | int pfd; |
74 | |
77 | |
75 | # if defined(PTYS_ARE_GETPT) |
78 | # if defined(HAVE_GETPT) |
76 | pfd = getpt(); |
79 | pfd = getpt(); |
77 | # elif defined(PTYS_ARE_POSIX) |
80 | # elif defined(HAVE_POSIX_OPENPT) |
78 | pfd = posix_openpt (O_RDWR); |
81 | pfd = posix_openpt (O_RDWR); |
79 | # else |
82 | # else |
80 | pfd = open ("/dev/ptmx", O_RDWR | O_NOCTTY, 0); |
83 | pfd = open (CLONE_DEVICE, O_RDWR | O_NOCTTY, 0); |
81 | # endif |
84 | # endif |
82 | if (pfd >= 0) |
85 | if (pfd >= 0) |
83 | { |
86 | { |
84 | if (grantpt (pfd) == 0 /* change slave permissions */ |
87 | if (grantpt (pfd) == 0 /* change slave permissions */ |
85 | && unlockpt (pfd) == 0) |
88 | && unlockpt (pfd) == 0) |
… | |
… | |
88 | return pfd; |
91 | return pfd; |
89 | } |
92 | } |
90 | |
93 | |
91 | close (pfd); |
94 | close (pfd); |
92 | } |
95 | } |
93 | #endif |
|
|
94 | |
96 | |
95 | return -1; |
97 | return -1; |
96 | } |
98 | } |
97 | |
99 | #elif defined(HAVE_OPENPTY) |
98 | static inline int |
100 | static int |
99 | get_pty_openpty (int *fd_tty, char **ttydev) |
101 | get_pty (int *fd_tty, char **ttydev) |
100 | { |
102 | { |
101 | #ifdef PTYS_ARE_OPENPTY |
|
|
102 | int pfd; |
103 | int pfd; |
103 | int res; |
104 | int res; |
104 | char tty_name[sizeof "/dev/pts/????\0"]; |
105 | char tty_name[32]; |
105 | |
106 | |
106 | res = openpty (&pfd, fd_tty, tty_name, NULL, NULL); |
107 | res = openpty (&pfd, fd_tty, tty_name, NULL, NULL); |
107 | if (res != -1) |
108 | if (res != -1) |
108 | { |
109 | { |
109 | *ttydev = strdup (tty_name); |
110 | *ttydev = strdup (tty_name); |
110 | return pfd; |
111 | return pfd; |
111 | } |
112 | } |
112 | #endif |
|
|
113 | |
113 | |
114 | return -1; |
114 | return -1; |
115 | } |
115 | } |
116 | |
116 | #elif defined(HAVE__GETPTY) |
117 | static inline int |
117 | static int |
118 | get_pty__getpty (int *fd_tty, char **ttydev) |
118 | get_pty (int *fd_tty, char **ttydev) |
119 | { |
119 | { |
120 | #ifdef PTYS_ARE__GETPTY |
|
|
121 | int pfd; |
120 | int pfd; |
122 | |
121 | |
123 | *ttydev = _getpty (&pfd, O_RDWR | O_NONBLOCK | O_NOCTTY, 0622, 0); |
122 | *ttydev = _getpty (&pfd, O_RDWR | O_NONBLOCK | O_NOCTTY, 0622, 0); |
124 | if (*ttydev != NULL) |
123 | if (*ttydev != NULL) |
125 | return pfd; |
124 | return pfd; |
126 | #endif |
|
|
127 | |
125 | |
128 | return -1; |
126 | return -1; |
129 | } |
127 | } |
130 | |
128 | #elif defined(HAVE_DEV_PTC) |
131 | static inline int |
129 | static int |
132 | get_pty_ptc (int *fd_tty, char **ttydev) |
130 | get_pty (int *fd_tty, char **ttydev) |
133 | { |
131 | { |
134 | #ifdef PTYS_ARE_PTC |
|
|
135 | int pfd; |
132 | int pfd; |
136 | |
133 | |
137 | if ((pfd = open ("/dev/ptc", O_RDWR | O_NOCTTY, 0)) >= 0) |
134 | if ((pfd = open ("/dev/ptc", O_RDWR | O_NOCTTY, 0)) >= 0) |
138 | { |
135 | { |
139 | *ttydev = strdup (ttyname (pfd)); |
136 | *ttydev = strdup (ttyname (pfd)); |
140 | return pfd; |
137 | return pfd; |
141 | } |
138 | } |
142 | #endif |
|
|
143 | |
139 | |
144 | return -1; |
140 | return -1; |
145 | } |
141 | } |
146 | |
142 | #elif defined(HAVE_DEV_CLONE) |
147 | static inline int |
143 | static int |
148 | get_pty_clone (int *fd_tty, char **ttydev) |
144 | get_pty (int *fd_tty, char **ttydev) |
149 | { |
145 | { |
150 | #ifdef PTYS_ARE_CLONE |
|
|
151 | int pfd; |
146 | int pfd; |
152 | |
147 | |
153 | if ((pfd = open ("/dev/ptym/clone", O_RDWR | O_NOCTTY, 0)) >= 0) |
148 | if ((pfd = open ("/dev/ptym/clone", O_RDWR | O_NOCTTY, 0)) >= 0) |
154 | { |
149 | { |
155 | *ttydev = strdup (ptsname (pfd)); |
150 | *ttydev = strdup (ptsname (pfd)); |
156 | return pfd; |
151 | return pfd; |
157 | } |
152 | } |
158 | #endif |
|
|
159 | |
153 | |
160 | return -1; |
154 | return -1; |
161 | } |
155 | } |
162 | |
156 | #else |
163 | static inline int |
157 | /* Based on the code in openssh/openbsd-compat/bsd-openpty.c */ |
|
|
158 | static int |
164 | get_pty_numeric (int *fd_tty, char **ttydev) |
159 | get_pty (int *fd_tty, char **ttydev) |
165 | { |
160 | { |
166 | #ifdef PTYS_ARE_NUMERIC |
|
|
167 | int pfd; |
161 | int pfd; |
168 | int idx; |
162 | int i; |
169 | char *c1, *c2; |
163 | char pty_name[32]; |
170 | char pty_name[] = "/dev/ptyp???"; |
164 | char tty_name[32]; |
171 | char tty_name[] = "/dev/ttyp???"; |
165 | const char *majors = "pqrstuvwxyzabcde"; |
172 | |
166 | const char *minors = "0123456789abcdef"; |
173 | c1 = &(pty_name[sizeof (pty_name) - 4]); |
|
|
174 | c2 = &(tty_name[sizeof (tty_name) - 4]); |
|
|
175 | |
|
|
176 | for (idx = 0; idx < 256; idx++) |
167 | for (i = 0; i < 256; i++) |
177 | { |
168 | { |
178 | sprintf (c1, "%d", idx); |
169 | snprintf(pty_name, 32, "/dev/pty%c%c", majors[i / 16], minors[i % 16]); |
179 | sprintf (c2, "%d", idx); |
170 | snprintf(tty_name, 32, "/dev/tty%c%c", majors[i / 16], minors[i % 16]); |
180 | |
|
|
181 | if (access (tty_name, F_OK) < 0) |
|
|
182 | { |
|
|
183 | idx = 256; |
|
|
184 | break; |
|
|
185 | } |
|
|
186 | |
|
|
187 | if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) >= 0) |
171 | if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) == -1) |
|
|
172 | { |
|
|
173 | snprintf(pty_name, 32, "/dev/ptyp%d", i); |
|
|
174 | snprintf(tty_name, 32, "/dev/ttyp%d", i); |
|
|
175 | if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) == -1) |
|
|
176 | continue; |
188 | { |
177 | } |
189 | if (access (tty_name, R_OK | W_OK) == 0) |
178 | if (access (tty_name, R_OK | W_OK) == 0) |
190 | { |
179 | { |
191 | *ttydev = strdup (tty_name); |
180 | *ttydev = strdup (tty_name); |
192 | return pfd; |
181 | return pfd; |
193 | } |
182 | } |
194 | |
183 | |
195 | close (pfd); |
184 | close (pfd); |
196 | } |
|
|
197 | } |
185 | } |
198 | #endif |
|
|
199 | |
186 | |
200 | return -1; |
187 | return -1; |
201 | } |
188 | } |
202 | |
|
|
203 | static inline int |
|
|
204 | get_pty_searched (int *fd_tty, char **ttydev) |
|
|
205 | { |
|
|
206 | #ifdef PTYS_ARE_SEARCHED |
|
|
207 | # ifndef PTYCHAR1 |
|
|
208 | # define PTYCHAR1 "pqrstuvwxyz" |
|
|
209 | # endif |
189 | #endif |
210 | # ifndef PTYCHAR2 |
|
|
211 | # define PTYCHAR2 "0123456789abcdef" |
|
|
212 | # endif |
|
|
213 | int pfd; |
|
|
214 | const char *c1, *c2; |
|
|
215 | char pty_name[] = "/dev/pty??"; |
|
|
216 | char tty_name[] = "/dev/tty??"; |
|
|
217 | |
|
|
218 | for (c1 = PTYCHAR1; *c1; c1++) |
|
|
219 | { |
|
|
220 | pty_name[ (sizeof (pty_name) - 3)] = |
|
|
221 | tty_name[ (sizeof (pty_name) - 3)] = *c1; |
|
|
222 | |
|
|
223 | for (c2 = PTYCHAR2; *c2; c2++) |
|
|
224 | { |
|
|
225 | pty_name[ (sizeof (pty_name) - 2)] = |
|
|
226 | tty_name[ (sizeof (pty_name) - 2)] = *c2; |
|
|
227 | |
|
|
228 | if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) >= 0) |
|
|
229 | { |
|
|
230 | if (access (tty_name, R_OK | W_OK) == 0) |
|
|
231 | { |
|
|
232 | *ttydev = strdup (tty_name); |
|
|
233 | return pfd; |
|
|
234 | } |
|
|
235 | |
|
|
236 | close (pfd); |
|
|
237 | } |
|
|
238 | } |
|
|
239 | } |
|
|
240 | #endif |
|
|
241 | |
|
|
242 | return -1; |
|
|
243 | } |
|
|
244 | |
|
|
245 | static int |
|
|
246 | get_pty (int *fd_tty, char **ttydev) |
|
|
247 | { |
|
|
248 | int pfd; |
|
|
249 | |
|
|
250 | if ((pfd = get_pty_streams (fd_tty, ttydev)) != -1 |
|
|
251 | || (pfd = get_pty_openpty (fd_tty, ttydev)) != -1 |
|
|
252 | || (pfd = get_pty__getpty (fd_tty, ttydev)) != -1 |
|
|
253 | || (pfd = get_pty_ptc (fd_tty, ttydev)) != -1 |
|
|
254 | || (pfd = get_pty_clone (fd_tty, ttydev)) != -1 |
|
|
255 | || (pfd = get_pty_numeric (fd_tty, ttydev)) != -1 |
|
|
256 | || (pfd = get_pty_searched (fd_tty, ttydev)) != -1) |
|
|
257 | return pfd; |
|
|
258 | |
|
|
259 | return -1; |
|
|
260 | } |
|
|
261 | |
190 | |
262 | /*----------------------------------------------------------------------*/ |
191 | /*----------------------------------------------------------------------*/ |
263 | /* |
192 | /* |
264 | * Returns tty file descriptor, or -1 on failure |
193 | * Returns tty file descriptor, or -1 on failure |
265 | */ |
194 | */ |
… | |
… | |
276 | static int |
205 | static int |
277 | control_tty (int fd_tty) |
206 | control_tty (int fd_tty) |
278 | { |
207 | { |
279 | setsid (); |
208 | setsid (); |
280 | |
209 | |
281 | #if defined(PTYS_ARE_PTMX) && defined(I_PUSH) |
210 | #if defined(HAVE_DEV_PTMX) && defined(I_PUSH) |
282 | /* |
211 | /* |
283 | * Push STREAMS modules: |
212 | * Push STREAMS modules: |
284 | * ptem: pseudo-terminal hardware emulation module. |
213 | * ptem: pseudo-terminal hardware emulation module. |
285 | * ldterm: standard terminal line discipline. |
214 | * ldterm: standard terminal line discipline. |
286 | * ttcompat: V7, 4BSD and XENIX STREAMS compatibility module. |
215 | * ttcompat: V7, 4BSD and XENIX STREAMS compatibility module. |
… | |
… | |
308 | |
237 | |
309 | ioctl (fd_tty, TIOCSCTTY, NULL); |
238 | ioctl (fd_tty, TIOCSCTTY, NULL); |
310 | |
239 | |
311 | int fd = open ("/dev/tty", O_WRONLY); |
240 | int fd = open ("/dev/tty", O_WRONLY); |
312 | if (fd < 0) |
241 | if (fd < 0) |
313 | return -1; /* fatal */ |
242 | return -1; /* fatal */ |
314 | |
243 | |
315 | close (fd); |
244 | close (fd); |
316 | |
245 | |
317 | return 0; |
246 | return 0; |
318 | } |
247 | } |
319 | |
248 | |
320 | void |
249 | void |
321 | rxvt_ptytty::close_tty () |
250 | ptytty::close_tty () |
322 | { |
251 | { |
323 | if (tty < 0) |
252 | if (tty < 0) |
324 | return; |
253 | return; |
325 | |
254 | |
326 | close (tty); |
255 | close (tty); |
327 | tty = -1; |
256 | tty = -1; |
328 | } |
257 | } |
329 | |
258 | |
330 | bool |
259 | bool |
331 | rxvt_ptytty::make_controlling_tty () |
260 | ptytty::make_controlling_tty () |
332 | { |
261 | { |
333 | return control_tty (tty) >= 0; |
262 | return control_tty (tty) >= 0; |
334 | } |
263 | } |
335 | |
264 | |
336 | void |
265 | void |
337 | rxvt_ptytty::set_utf8_mode (bool on) |
266 | ptytty::set_utf8_mode (bool on) |
338 | { |
267 | { |
339 | #ifdef IUTF8 |
268 | #ifdef IUTF8 |
340 | if (pty < 0) |
269 | if (pty < 0) |
341 | return; |
270 | return; |
342 | |
271 | |
… | |
… | |
358 | } |
287 | } |
359 | } |
288 | } |
360 | #endif |
289 | #endif |
361 | } |
290 | } |
362 | |
291 | |
363 | ///////////////////////////////////////////////////////////////////////////// |
|
|
364 | |
|
|
365 | #ifndef NO_SETOWNER_TTYDEV |
|
|
366 | static struct ttyconf { |
292 | static struct ttyconf { |
367 | gid_t gid; |
293 | gid_t gid; |
368 | mode_t mode; |
294 | mode_t mode; |
369 | |
295 | |
370 | ttyconf () |
296 | ttyconf () |
… | |
… | |
384 | gid = 0; |
310 | gid = 0; |
385 | } |
311 | } |
386 | } |
312 | } |
387 | } ttyconf; |
313 | } ttyconf; |
388 | |
314 | |
389 | ///////////////////////////////////////////////////////////////////////////// |
315 | ptytty_unix::ptytty_unix () |
|
|
316 | { |
|
|
317 | name = 0; |
|
|
318 | #if UTMP_SUPPORT |
|
|
319 | cmd_pid = 0; |
|
|
320 | #endif |
|
|
321 | } |
|
|
322 | |
|
|
323 | ptytty_unix::~ptytty_unix () |
|
|
324 | { |
|
|
325 | #if UTMP_SUPPORT |
|
|
326 | logout (); |
|
|
327 | #endif |
|
|
328 | put (); |
|
|
329 | } |
390 | |
330 | |
391 | void |
331 | void |
392 | rxvt_ptytty_unix::privileges (rxvt_privaction action) |
332 | ptytty_unix::put () |
393 | { |
333 | { |
394 | if (!name || !*name) |
334 | chmod (name, RESTORE_TTY_MODE); |
|
|
335 | chown (name, 0, ttyconf.gid); |
|
|
336 | |
|
|
337 | close_tty (); |
|
|
338 | |
|
|
339 | if (pty >= 0) |
|
|
340 | close (pty); |
|
|
341 | |
|
|
342 | free (name); |
|
|
343 | |
|
|
344 | pty = tty = -1; |
|
|
345 | name = 0; |
|
|
346 | } |
|
|
347 | |
|
|
348 | bool |
|
|
349 | ptytty_unix::get () |
|
|
350 | { |
|
|
351 | /* get master (pty) */ |
|
|
352 | if ((pty = get_pty (&tty, &name)) < 0) |
395 | return; |
353 | return false; |
396 | |
354 | |
397 | if (action == SAVE) |
355 | fcntl (pty, F_SETFL, O_NONBLOCK); |
|
|
356 | |
|
|
357 | /* get slave (tty) */ |
|
|
358 | if (tty < 0) |
398 | { |
359 | { |
|
|
360 | #ifndef NO_SETOWNER_TTYDEV |
399 | chown (name, getuid (), ttyconf.gid); /* fail silently */ |
361 | chown (name, getuid (), ttyconf.gid); /* fail silently */ |
400 | chmod (name, ttyconf.mode); |
362 | chmod (name, ttyconf.mode); |
401 | # ifdef HAVE_REVOKE |
363 | # ifdef HAVE_REVOKE |
402 | revoke (name); |
364 | revoke (name); |
403 | # endif |
365 | # endif |
404 | } |
|
|
405 | else |
|
|
406 | { /* action == RESTORE */ |
|
|
407 | chmod (name, RESTORE_TTY_MODE); |
|
|
408 | chown (name, 0, ttyconf.gid); |
|
|
409 | } |
|
|
410 | } |
|
|
411 | #endif |
|
|
412 | |
|
|
413 | rxvt_ptytty_unix::rxvt_ptytty_unix () |
|
|
414 | { |
|
|
415 | pty = tty = -1; |
|
|
416 | name = 0; |
|
|
417 | #if UTMP_SUPPORT |
|
|
418 | cmd_pid = 0; |
|
|
419 | #endif |
|
|
420 | } |
|
|
421 | |
|
|
422 | rxvt_ptytty_unix::~rxvt_ptytty_unix () |
|
|
423 | { |
|
|
424 | #if UTMP_SUPPORT |
|
|
425 | logout (); |
|
|
426 | #endif |
|
|
427 | put (); |
|
|
428 | } |
|
|
429 | |
|
|
430 | void |
|
|
431 | rxvt_ptytty_unix::put () |
|
|
432 | { |
|
|
433 | #ifndef NO_SETOWNER_TTYDEV |
|
|
434 | privileges (RESTORE); |
|
|
435 | #endif |
|
|
436 | |
|
|
437 | if (pty >= 0) close (pty); |
|
|
438 | close_tty (); |
|
|
439 | free (name); |
|
|
440 | |
|
|
441 | pty = tty = -1; |
|
|
442 | name = 0; |
|
|
443 | } |
|
|
444 | |
|
|
445 | bool |
|
|
446 | rxvt_ptytty_unix::get () |
|
|
447 | { |
|
|
448 | /* get master (pty) */ |
|
|
449 | if ((pty = get_pty (&tty, &name)) < 0) |
|
|
450 | return false; |
|
|
451 | |
|
|
452 | fcntl (pty, F_SETFL, O_NONBLOCK); |
|
|
453 | |
|
|
454 | /* get slave (tty) */ |
|
|
455 | if (tty < 0) |
|
|
456 | { |
|
|
457 | #ifndef NO_SETOWNER_TTYDEV |
|
|
458 | privileges (SAVE); |
|
|
459 | #endif |
366 | #endif |
460 | |
367 | |
461 | if ((tty = get_tty (name)) < 0) |
368 | if ((tty = get_tty (name)) < 0) |
462 | { |
369 | { |
463 | put (); |
370 | put (); |
… | |
… | |
466 | } |
373 | } |
467 | |
374 | |
468 | return true; |
375 | return true; |
469 | } |
376 | } |
470 | |
377 | |
471 | #if PTYTTY_HELPER |
|
|
472 | |
|
|
473 | static int sock_fd; |
|
|
474 | static int pid; |
|
|
475 | |
|
|
476 | struct command |
|
|
477 | { |
|
|
478 | enum { get, login, destroy } type; |
|
|
479 | |
|
|
480 | rxvt_ptytty *id; |
|
|
481 | |
|
|
482 | bool login_shell; |
|
|
483 | int cmd_pid; |
|
|
484 | char hostname[512]; // arbitrary, but should be plenty |
|
|
485 | }; |
|
|
486 | |
|
|
487 | struct rxvt_ptytty_proxy : zero_initialized, rxvt_ptytty |
|
|
488 | { |
|
|
489 | rxvt_ptytty *id; |
|
|
490 | |
|
|
491 | ~rxvt_ptytty_proxy (); |
|
|
492 | |
|
|
493 | bool get (); |
|
|
494 | void login (int cmd_pid, bool login_shell, const char *hostname); |
|
|
495 | }; |
|
|
496 | |
|
|
497 | bool |
|
|
498 | rxvt_ptytty_proxy::get () |
|
|
499 | { |
|
|
500 | command cmd; |
|
|
501 | |
|
|
502 | cmd.type = command::get; |
|
|
503 | |
|
|
504 | write (sock_fd, &cmd, sizeof (cmd)); |
|
|
505 | |
|
|
506 | if (read (sock_fd, &id, sizeof (id)) != sizeof (id)) |
|
|
507 | rxvt_fatal ("protocol error while creating pty using helper process, aborting.\n"); |
|
|
508 | |
|
|
509 | if (!id) |
|
|
510 | return false; |
|
|
511 | |
|
|
512 | if ((pty = rxvt_recv_fd (sock_fd)) < 0 |
|
|
513 | || (tty = rxvt_recv_fd (sock_fd)) < 0) |
|
|
514 | rxvt_fatal ("protocol error while reading pty/tty fds from helper process, aborting.\n"); |
|
|
515 | |
|
|
516 | return true; |
|
|
517 | } |
|
|
518 | |
|
|
519 | void |
|
|
520 | rxvt_ptytty_proxy::login (int cmd_pid, bool login_shell, const char *hostname) |
|
|
521 | { |
|
|
522 | command cmd; |
|
|
523 | |
|
|
524 | cmd.type = command::login; |
|
|
525 | cmd.id = id; |
|
|
526 | cmd.cmd_pid = cmd_pid; |
|
|
527 | cmd.login_shell = login_shell; |
|
|
528 | strncpy (cmd.hostname, hostname, sizeof (cmd.hostname)); |
|
|
529 | |
|
|
530 | write (sock_fd, &cmd, sizeof (cmd)); |
|
|
531 | } |
|
|
532 | |
|
|
533 | rxvt_ptytty_proxy::~rxvt_ptytty_proxy () |
|
|
534 | { |
|
|
535 | command cmd; |
|
|
536 | |
|
|
537 | cmd.type = command::destroy; |
|
|
538 | cmd.id = id; |
|
|
539 | |
|
|
540 | write (sock_fd, &cmd, sizeof (cmd)); |
|
|
541 | } |
|
|
542 | |
|
|
543 | static |
|
|
544 | void serve () |
|
|
545 | { |
|
|
546 | command cmd; |
|
|
547 | vector<rxvt_ptytty *> ptys; |
|
|
548 | |
|
|
549 | while (read (sock_fd, &cmd, sizeof (command)) == sizeof (command)) |
|
|
550 | { |
|
|
551 | if (cmd.type == command::get) |
|
|
552 | { |
|
|
553 | // -> id ptyfd ttyfd |
|
|
554 | cmd.id = new rxvt_ptytty_unix; |
|
|
555 | |
|
|
556 | if (cmd.id->get ()) |
|
|
557 | { |
|
|
558 | write (sock_fd, &cmd.id, sizeof (cmd.id)); |
|
|
559 | ptys.push_back (cmd.id); |
|
|
560 | |
|
|
561 | rxvt_send_fd (sock_fd, cmd.id->pty); |
|
|
562 | rxvt_send_fd (sock_fd, cmd.id->tty); |
|
|
563 | } |
|
|
564 | else |
|
|
565 | { |
|
|
566 | delete cmd.id; |
|
|
567 | cmd.id = 0; |
|
|
568 | write (sock_fd, &cmd.id, sizeof (cmd.id)); |
|
|
569 | } |
|
|
570 | } |
|
|
571 | else if (cmd.type == command::login) |
|
|
572 | { |
|
|
573 | #if UTMP_SUPPORT |
|
|
574 | if (find (ptys.begin (), ptys.end (), cmd.id)) |
|
|
575 | { |
|
|
576 | cmd.hostname[sizeof (cmd.hostname) - 1] = 0; |
|
|
577 | cmd.id->login (cmd.cmd_pid, cmd.login_shell, cmd.hostname); |
|
|
578 | } |
|
|
579 | #endif |
|
|
580 | } |
|
|
581 | else if (cmd.type == command::destroy) |
|
|
582 | { |
|
|
583 | rxvt_ptytty **pty = find (ptys.begin (), ptys.end (), cmd.id); |
|
|
584 | |
|
|
585 | if (pty) |
|
|
586 | { |
|
|
587 | delete *pty; |
|
|
588 | ptys.erase (pty); |
|
|
589 | } |
|
|
590 | } |
|
|
591 | else |
|
|
592 | break; |
|
|
593 | } |
|
|
594 | |
|
|
595 | // destroy all ptys |
|
|
596 | for (rxvt_ptytty **i = ptys.end (); i-- > ptys.begin (); ) |
|
|
597 | delete *i; |
|
|
598 | } |
|
|
599 | |
|
|
600 | void rxvt_ptytty_server () |
|
|
601 | { |
|
|
602 | int sv[2]; |
|
|
603 | |
|
|
604 | if (socketpair (AF_UNIX, SOCK_STREAM, 0, sv)) |
|
|
605 | rxvt_fatal ("could not create socket to communicate with pty/sessiondb helper, aborting.\n"); |
|
|
606 | |
|
|
607 | pid = fork (); |
|
|
608 | |
|
|
609 | if (pid < 0) |
|
|
610 | rxvt_fatal ("could not create pty/sessiondb helper process, aborting.\n"); |
|
|
611 | |
|
|
612 | if (pid) |
|
|
613 | { |
|
|
614 | // client, urxvt |
|
|
615 | sock_fd = sv[0]; |
|
|
616 | close (sv[1]); |
|
|
617 | fcntl (sock_fd, F_SETFD, FD_CLOEXEC); |
|
|
618 | } |
|
|
619 | else |
|
|
620 | { |
|
|
621 | // server, pty-helper |
|
|
622 | sock_fd = sv[1]; |
|
|
623 | |
|
|
624 | for (int fd = 0; fd < 1023; fd++) |
|
|
625 | if (fd != sock_fd) |
|
|
626 | close (fd); |
|
|
627 | |
|
|
628 | serve (); |
|
|
629 | _exit (EXIT_SUCCESS); |
|
|
630 | } |
|
|
631 | } |
|
|
632 | #endif |
|
|
633 | |
|
|
634 | // a "factory" *g* |
|
|
635 | rxvt_ptytty * |
|
|
636 | rxvt_new_ptytty () |
|
|
637 | { |
|
|
638 | #if PTYTTY_HELPER |
|
|
639 | if (pid > 0) |
|
|
640 | // use helper process |
|
|
641 | return new rxvt_ptytty_proxy; |
|
|
642 | else |
|
|
643 | #endif |
|
|
644 | return new rxvt_ptytty_unix; |
|
|
645 | } |
|
|
646 | |
|
|
647 | /*----------------------- end-of-file (C source) -----------------------*/ |
|
|
648 | |
|
|