|
|
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 | |
|
|
30 | #include "ptytty.h" |
25 | |
31 | |
26 | #include <cstdlib> |
32 | #include <cstdlib> |
27 | #include <cstring> |
33 | #include <cstring> |
|
|
34 | #include <csignal> |
28 | |
35 | |
29 | #include <sys/types.h> |
36 | #include <sys/types.h> |
30 | #include <sys/socket.h> |
37 | #include <sys/stat.h> |
31 | #include <unistd.h> |
38 | #include <unistd.h> |
32 | #include <fcntl.h> |
39 | #include <fcntl.h> |
33 | |
40 | |
34 | #ifdef HAVE_SYS_IOCTL_H |
41 | #ifdef HAVE_SYS_IOCTL_H |
35 | # include <sys/ioctl.h> |
42 | # include <sys/ioctl.h> |
36 | #endif |
43 | #endif |
37 | #if defined(HAVE_DEV_PTMX) && defined(HAVE_SYS_STROPTS_H) |
44 | #if defined(HAVE_DEV_PTMX) && defined(HAVE_SYS_STROPTS_H) |
38 | # include <sys/stropts.h> /* for I_PUSH */ |
45 | # include <sys/stropts.h> /* for I_PUSH */ |
39 | #endif |
46 | #endif |
40 | #ifdef HAVE_ISASTREAM |
47 | #if defined(HAVE_ISASTREAM) && defined(HAVE_STROPTS_H) |
41 | # include <stropts.h> |
48 | # include <stropts.h> |
42 | #endif |
49 | #endif |
43 | #if defined(HAVE_PTY_H) |
50 | #if defined(HAVE_PTY_H) |
44 | # include <pty.h> |
51 | # include <pty.h> |
45 | #elif defined(HAVE_LIBUTIL_H) |
52 | #elif defined(HAVE_LIBUTIL_H) |
… | |
… | |
51 | #include <grp.h> |
58 | #include <grp.h> |
52 | #endif |
59 | #endif |
53 | |
60 | |
54 | #include <cstdio> |
61 | #include <cstdio> |
55 | |
62 | |
56 | #include "rxvtutil.h" |
|
|
57 | #include "fdpass.h" |
|
|
58 | #include "ptytty.h" |
|
|
59 | |
|
|
60 | ///////////////////////////////////////////////////////////////////////////// |
63 | ///////////////////////////////////////////////////////////////////////////// |
61 | |
64 | |
62 | /* ------------------------------------------------------------------------- * |
65 | /* ------------------------------------------------------------------------- * |
63 | * GET PSEUDO TELETYPE - MASTER AND SLAVE * |
66 | * GET PSEUDO TELETYPE - MASTER AND SLAVE * |
64 | * ------------------------------------------------------------------------- */ |
67 | * ------------------------------------------------------------------------- */ |
65 | /* |
68 | /* |
66 | * Returns pty file descriptor, or -1 on failure |
69 | * Returns pty file descriptor, or -1 on failure |
67 | * If successful, ttydev is set to the name of the slave device. |
70 | * If successful, ttydev is set to the name of the slave device. |
68 | * fd_tty _may_ also be set to an open fd to the slave device |
71 | * fd_tty _may_ also be set to an open fd to the slave device |
69 | */ |
72 | */ |
70 | #if defined(UNIX98_PTY) |
73 | #if defined(UNIX98_PTY) |
|
|
74 | |
71 | static int |
75 | static int |
72 | get_pty (int *fd_tty, char **ttydev) |
76 | get_pty (int *fd_tty, char **ttydev) |
73 | { |
77 | { |
74 | int pfd; |
78 | int pfd; |
75 | |
79 | |
76 | # if defined(HAVE_GETPT) |
80 | # if defined(HAVE_GETPT) |
77 | pfd = getpt(); |
81 | pfd = getpt(); |
78 | # elif defined(HAVE_POSIX_OPENPT) |
82 | # elif defined(HAVE_POSIX_OPENPT) |
79 | pfd = posix_openpt (O_RDWR); |
83 | pfd = posix_openpt (O_RDWR); |
80 | # else |
84 | # else |
81 | pfd = open (CLONE_DEVICE, O_RDWR | O_NOCTTY, 0); |
85 | pfd = open (CLONE_DEVICE, O_RDWR | O_NOCTTY, 0); |
82 | # endif |
86 | # endif |
|
|
87 | |
83 | if (pfd >= 0) |
88 | if (pfd >= 0) |
84 | { |
89 | { |
85 | if (grantpt (pfd) == 0 /* change slave permissions */ |
90 | if (grantpt (pfd) == 0 /* change slave permissions */ |
86 | && unlockpt (pfd) == 0) |
91 | && unlockpt (pfd) == 0) |
|
|
92 | { |
87 | { /* slave now unlocked */ |
93 | /* slave now unlocked */ |
88 | *ttydev = strdup (ptsname (pfd)); /* get slave's name */ |
94 | *ttydev = strdup (ptsname (pfd)); /* get slave's name */ |
89 | return pfd; |
95 | return pfd; |
90 | } |
96 | } |
91 | |
97 | |
92 | close (pfd); |
98 | close (pfd); |
93 | } |
99 | } |
94 | |
100 | |
95 | return -1; |
101 | return -1; |
96 | } |
102 | } |
|
|
103 | |
97 | #elif defined(HAVE_OPENPTY) |
104 | #elif defined(HAVE_OPENPTY) |
|
|
105 | |
98 | static int |
106 | static int |
99 | get_pty (int *fd_tty, char **ttydev) |
107 | get_pty (int *fd_tty, char **ttydev) |
100 | { |
108 | { |
101 | int pfd; |
109 | int pfd; |
102 | int res; |
110 | int res; |
103 | char tty_name[32]; |
111 | |
104 | |
|
|
105 | res = openpty (&pfd, fd_tty, tty_name, NULL, NULL); |
112 | res = openpty (&pfd, fd_tty, NULL, NULL, NULL); |
|
|
113 | |
106 | if (res != -1) |
114 | if (res != -1) |
107 | { |
115 | { |
108 | *ttydev = strdup (tty_name); |
116 | *ttydev = strdup (ttyname (*fd_tty)); |
109 | return pfd; |
117 | return pfd; |
110 | } |
118 | } |
111 | |
119 | |
112 | return -1; |
120 | return -1; |
113 | } |
121 | } |
|
|
122 | |
114 | #elif defined(HAVE__GETPTY) |
123 | #elif defined(HAVE__GETPTY) |
|
|
124 | |
115 | static int |
125 | static int |
116 | get_pty (int *fd_tty, char **ttydev) |
126 | get_pty (int *fd_tty, char **ttydev) |
117 | { |
127 | { |
118 | int pfd; |
128 | int pfd; |
|
|
129 | char *slave; |
119 | |
130 | |
120 | *ttydev = _getpty (&pfd, O_RDWR | O_NONBLOCK | O_NOCTTY, 0622, 0); |
131 | slave = _getpty (&pfd, O_RDWR | O_NONBLOCK | O_NOCTTY, 0622, 0); |
121 | if (*ttydev != NULL) |
|
|
122 | return pfd; |
|
|
123 | |
132 | |
124 | return -1; |
133 | if (slave != NULL) |
125 | } |
|
|
126 | #elif defined(HAVE_DEV_PTC) |
|
|
127 | static int |
|
|
128 | get_pty (int *fd_tty, char **ttydev) |
|
|
129 | { |
|
|
130 | int pfd; |
|
|
131 | |
|
|
132 | if ((pfd = open ("/dev/ptc", O_RDWR | O_NOCTTY, 0)) >= 0) |
|
|
133 | { |
134 | { |
134 | *ttydev = strdup (ttyname (pfd)); |
135 | *ttydev = strdup (slave); |
135 | return pfd; |
136 | return pfd; |
136 | } |
137 | } |
137 | |
138 | |
138 | return -1; |
139 | return -1; |
139 | } |
|
|
140 | #elif defined(HAVE_DEV_CLONE) |
|
|
141 | static int |
|
|
142 | get_pty (int *fd_tty, char **ttydev) |
|
|
143 | { |
|
|
144 | int pfd; |
|
|
145 | |
|
|
146 | if ((pfd = open ("/dev/ptym/clone", O_RDWR | O_NOCTTY, 0)) >= 0) |
|
|
147 | { |
|
|
148 | *ttydev = strdup (ptsname (pfd)); |
|
|
149 | return pfd; |
|
|
150 | } |
140 | } |
151 | |
141 | |
152 | return -1; |
|
|
153 | } |
|
|
154 | #else |
142 | #else |
|
|
143 | |
155 | /* Based on the code in openssh/openbsd-compat/bsd-openpty.c */ |
144 | /* Based on the code in openssh/openbsd-compat/bsd-openpty.c */ |
156 | static int |
145 | static int |
157 | get_pty (int *fd_tty, char **ttydev) |
146 | get_pty (int *fd_tty, char **ttydev) |
158 | { |
147 | { |
159 | int pfd; |
148 | int pfd; |
160 | int i; |
149 | int i; |
161 | char pty_name[32]; |
150 | char pty_name[32]; |
162 | char tty_name[32]; |
151 | char tty_name[32]; |
163 | const char *majors = "pqrstuvwxyzabcde"; |
152 | const char *majors = "pqrstuvwxyzabcde"; |
164 | const char *minors = "0123456789abcdef"; |
153 | const char *minors = "0123456789abcdef"; |
|
|
154 | |
165 | for (i = 0; i < 256; i++) |
155 | for (i = 0; i < 256; i++) |
166 | { |
156 | { |
167 | snprintf(pty_name, 32, "/dev/pty%c%c", majors[i / 16], minors[i % 16]); |
157 | snprintf(pty_name, 32, "/dev/pty%c%c", majors[i / 16], minors[i % 16]); |
168 | snprintf(tty_name, 32, "/dev/tty%c%c", majors[i / 16], minors[i % 16]); |
158 | snprintf(tty_name, 32, "/dev/tty%c%c", majors[i / 16], minors[i % 16]); |
|
|
159 | |
169 | if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) == -1) |
160 | if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) == -1) |
170 | { |
161 | { |
171 | snprintf(pty_name, 32, "/dev/ptyp%d", i); |
162 | snprintf(pty_name, 32, "/dev/ptyp%d", i); |
172 | snprintf(tty_name, 32, "/dev/ttyp%d", i); |
163 | snprintf(tty_name, 32, "/dev/ttyp%d", i); |
173 | if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) == -1) |
164 | if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) == -1) |
174 | continue; |
165 | continue; |
175 | } |
166 | } |
|
|
167 | |
176 | if (access (tty_name, R_OK | W_OK) == 0) |
168 | if (access (tty_name, R_OK | W_OK) == 0) |
177 | { |
169 | { |
178 | *ttydev = strdup (tty_name); |
170 | *ttydev = strdup (tty_name); |
179 | return pfd; |
171 | return pfd; |
180 | } |
172 | } |
181 | |
173 | |
182 | close (pfd); |
174 | close (pfd); |
183 | } |
175 | } |
184 | } |
176 | |
|
|
177 | return -1; |
|
|
178 | } |
|
|
179 | |
185 | #endif |
180 | #endif |
186 | |
181 | |
187 | /*----------------------------------------------------------------------*/ |
182 | /*----------------------------------------------------------------------*/ |
188 | /* |
183 | /* |
189 | * Returns tty file descriptor, or -1 on failure |
184 | * Returns tty file descriptor, or -1 on failure |
190 | */ |
185 | */ |
191 | static int |
186 | static int |
192 | get_tty (char *ttydev) |
187 | get_tty (char *ttydev) |
193 | { |
188 | { |
194 | return open (ttydev, O_RDWR | O_NOCTTY, 0); |
189 | return open (ttydev, O_RDWR | O_NOCTTY, 0); |
… | |
… | |
199 | * Make our tty a controlling tty so that /dev/tty points to us |
194 | * Make our tty a controlling tty so that /dev/tty points to us |
200 | */ |
195 | */ |
201 | static int |
196 | static int |
202 | control_tty (int fd_tty) |
197 | control_tty (int fd_tty) |
203 | { |
198 | { |
|
|
199 | int fd; |
|
|
200 | |
204 | setsid (); |
201 | setsid (); |
205 | |
202 | |
206 | #if defined(HAVE_DEV_PTMX) && defined(I_PUSH) |
203 | #if defined(HAVE_DEV_PTMX) && defined(I_PUSH) |
207 | /* |
204 | /* |
208 | * Push STREAMS modules: |
205 | * Push STREAMS modules: |
… | |
… | |
219 | * documentation is really unclear about whether it is any close () on |
216 | * documentation is really unclear about whether it is any close () on |
220 | * the master side or the last close () - i.e. a proper STREAMS dismantling |
217 | * the master side or the last close () - i.e. a proper STREAMS dismantling |
221 | * close () - on the master side which causes a hang up to be sent |
218 | * close () - on the master side which causes a hang up to be sent |
222 | * through - Geoff Wing |
219 | * through - Geoff Wing |
223 | */ |
220 | */ |
224 | # ifdef HAVE_ISASTREAM |
221 | #if defined(HAVE_ISASTREAM) && defined(HAVE_STROPTS_H) |
225 | if (isastream (fd_tty) == 1) |
222 | if (isastream (fd_tty) == 1) |
226 | # endif |
223 | # endif |
227 | { |
224 | { |
228 | ioctl (fd_tty, I_PUSH, "ptem"); |
225 | ioctl (fd_tty, I_PUSH, "ptem"); |
229 | ioctl (fd_tty, I_PUSH, "ldterm"); |
226 | ioctl (fd_tty, I_PUSH, "ldterm"); |
230 | ioctl (fd_tty, I_PUSH, "ttcompat"); |
227 | ioctl (fd_tty, I_PUSH, "ttcompat"); |
231 | } |
228 | } |
232 | #endif |
229 | #endif |
233 | |
230 | |
|
|
231 | #ifdef TIOCSCTTY |
234 | ioctl (fd_tty, TIOCSCTTY, NULL); |
232 | ioctl (fd_tty, TIOCSCTTY, NULL); |
|
|
233 | #else |
|
|
234 | fd = open (ttyname (fd_tty), O_RDWR); |
|
|
235 | if (fd >= 0) |
|
|
236 | close (fd); |
|
|
237 | #endif |
235 | |
238 | |
236 | int fd = open ("/dev/tty", O_WRONLY); |
239 | fd = open ("/dev/tty", O_WRONLY); |
237 | if (fd < 0) |
240 | if (fd < 0) |
238 | return -1; /* fatal */ |
241 | return -1; /* fatal */ |
239 | |
242 | |
240 | close (fd); |
243 | close (fd); |
241 | |
244 | |
242 | return 0; |
245 | return 0; |
243 | } |
246 | } |
244 | |
247 | |
245 | void |
248 | void |
246 | rxvt_ptytty::close_tty () |
249 | ptytty::close_tty () |
247 | { |
250 | { |
248 | if (tty < 0) |
251 | if (tty < 0) |
249 | return; |
252 | return; |
250 | |
253 | |
251 | close (tty); |
254 | close (tty); |
252 | tty = -1; |
255 | tty = -1; |
253 | } |
256 | } |
254 | |
257 | |
255 | bool |
258 | bool |
256 | rxvt_ptytty::make_controlling_tty () |
259 | ptytty::make_controlling_tty () |
257 | { |
260 | { |
258 | return control_tty (tty) >= 0; |
261 | return control_tty (tty) >= 0; |
259 | } |
262 | } |
260 | |
263 | |
261 | void |
264 | void |
262 | rxvt_ptytty::set_utf8_mode (bool on) |
265 | ptytty::set_utf8_mode (bool on) |
263 | { |
266 | { |
264 | #ifdef IUTF8 |
267 | #ifdef IUTF8 |
265 | if (pty < 0) |
268 | if (pty < 0) |
266 | return; |
269 | return; |
267 | |
270 | |
… | |
… | |
293 | { |
296 | { |
294 | #ifdef TTY_GID_SUPPORT |
297 | #ifdef TTY_GID_SUPPORT |
295 | struct group *gr = getgrnam ("tty"); |
298 | struct group *gr = getgrnam ("tty"); |
296 | |
299 | |
297 | if (gr) |
300 | if (gr) |
|
|
301 | { |
298 | { /* change group ownership of tty to "tty" */ |
302 | /* change group ownership of tty to "tty" */ |
299 | mode = S_IRUSR | S_IWUSR | S_IWGRP; |
303 | mode = S_IRUSR | S_IWUSR | S_IWGRP; |
300 | gid = gr->gr_gid; |
304 | gid = gr->gr_gid; |
301 | } |
305 | } |
302 | else |
306 | else |
303 | #endif /* TTY_GID_SUPPORT */ |
307 | #endif /* TTY_GID_SUPPORT */ |
304 | { |
308 | { |
305 | mode = S_IRUSR | S_IWUSR | S_IWGRP | S_IWOTH; |
309 | mode = S_IRUSR | S_IWUSR | S_IWGRP | S_IWOTH; |
306 | gid = 0; |
310 | gid = 0; |
307 | } |
311 | } |
308 | } |
312 | } |
309 | } ttyconf; |
313 | } ttyconf; |
310 | |
314 | |
311 | rxvt_ptytty_unix::rxvt_ptytty_unix () |
315 | ptytty_unix::ptytty_unix () |
312 | { |
316 | { |
313 | name = 0; |
317 | name = 0; |
314 | #if UTMP_SUPPORT |
318 | #if UTMP_SUPPORT |
315 | cmd_pid = 0; |
319 | cmd_pid = 0; |
316 | #endif |
320 | #endif |
317 | } |
321 | } |
318 | |
322 | |
319 | rxvt_ptytty_unix::~rxvt_ptytty_unix () |
323 | ptytty_unix::~ptytty_unix () |
320 | { |
324 | { |
321 | #if UTMP_SUPPORT |
325 | #if UTMP_SUPPORT |
322 | logout (); |
326 | logout (); |
323 | #endif |
327 | #endif |
324 | put (); |
328 | put (); |
325 | } |
329 | } |
326 | |
330 | |
327 | void |
331 | void |
328 | rxvt_ptytty_unix::put () |
332 | ptytty_unix::put () |
329 | { |
333 | { |
|
|
334 | if (name) |
|
|
335 | { |
330 | chmod (name, RESTORE_TTY_MODE); |
336 | chmod (name, RESTORE_TTY_MODE); |
331 | chown (name, 0, ttyconf.gid); |
337 | chown (name, 0, ttyconf.gid); |
|
|
338 | } |
332 | |
339 | |
333 | close_tty (); |
340 | close_tty (); |
334 | |
341 | |
335 | if (pty >= 0) |
342 | if (pty >= 0) |
336 | close (pty); |
343 | close (pty); |
… | |
… | |
340 | pty = tty = -1; |
347 | pty = tty = -1; |
341 | name = 0; |
348 | name = 0; |
342 | } |
349 | } |
343 | |
350 | |
344 | bool |
351 | bool |
345 | rxvt_ptytty_unix::get () |
352 | ptytty_unix::get () |
346 | { |
353 | { |
347 | /* get master (pty) */ |
354 | /* get master (pty) */ |
348 | if ((pty = get_pty (&tty, &name)) < 0) |
355 | if ((pty = get_pty (&tty, &name)) < 0) |
349 | return false; |
356 | return false; |
350 | |
357 | |
… | |
… | |
369 | } |
376 | } |
370 | |
377 | |
371 | return true; |
378 | return true; |
372 | } |
379 | } |
373 | |
380 | |
374 | #if PTYTTY_HELPER |
|
|
375 | |
|
|
376 | static int sock_fd; |
|
|
377 | static int pid; |
|
|
378 | |
|
|
379 | struct command |
|
|
380 | { |
|
|
381 | enum { get, login, destroy } type; |
|
|
382 | |
|
|
383 | rxvt_ptytty *id; |
|
|
384 | |
|
|
385 | bool login_shell; |
|
|
386 | int cmd_pid; |
|
|
387 | char hostname[512]; // arbitrary, but should be plenty |
|
|
388 | }; |
|
|
389 | |
|
|
390 | struct rxvt_ptytty_proxy : zero_initialized, rxvt_ptytty |
|
|
391 | { |
|
|
392 | rxvt_ptytty *id; |
|
|
393 | |
|
|
394 | ~rxvt_ptytty_proxy (); |
|
|
395 | |
|
|
396 | bool get (); |
|
|
397 | void login (int cmd_pid, bool login_shell, const char *hostname); |
|
|
398 | }; |
|
|
399 | |
|
|
400 | bool |
|
|
401 | rxvt_ptytty_proxy::get () |
|
|
402 | { |
|
|
403 | command cmd; |
|
|
404 | |
|
|
405 | cmd.type = command::get; |
|
|
406 | |
|
|
407 | write (sock_fd, &cmd, sizeof (cmd)); |
|
|
408 | |
|
|
409 | if (read (sock_fd, &id, sizeof (id)) != sizeof (id)) |
|
|
410 | rxvt_fatal ("protocol error while creating pty using helper process, aborting.\n"); |
|
|
411 | |
|
|
412 | if (!id) |
|
|
413 | return false; |
|
|
414 | |
|
|
415 | if ((pty = rxvt_recv_fd (sock_fd)) < 0 |
|
|
416 | || (tty = rxvt_recv_fd (sock_fd)) < 0) |
|
|
417 | rxvt_fatal ("protocol error while reading pty/tty fds from helper process, aborting.\n"); |
|
|
418 | |
|
|
419 | return true; |
|
|
420 | } |
|
|
421 | |
|
|
422 | void |
|
|
423 | rxvt_ptytty_proxy::login (int cmd_pid, bool login_shell, const char *hostname) |
|
|
424 | { |
|
|
425 | command cmd; |
|
|
426 | |
|
|
427 | cmd.type = command::login; |
|
|
428 | cmd.id = id; |
|
|
429 | cmd.cmd_pid = cmd_pid; |
|
|
430 | cmd.login_shell = login_shell; |
|
|
431 | strncpy (cmd.hostname, hostname, sizeof (cmd.hostname)); |
|
|
432 | |
|
|
433 | write (sock_fd, &cmd, sizeof (cmd)); |
|
|
434 | } |
|
|
435 | |
|
|
436 | rxvt_ptytty_proxy::~rxvt_ptytty_proxy () |
|
|
437 | { |
|
|
438 | command cmd; |
|
|
439 | |
|
|
440 | cmd.type = command::destroy; |
|
|
441 | cmd.id = id; |
|
|
442 | |
|
|
443 | write (sock_fd, &cmd, sizeof (cmd)); |
|
|
444 | } |
|
|
445 | |
|
|
446 | static |
|
|
447 | void serve () |
|
|
448 | { |
|
|
449 | command cmd; |
|
|
450 | vector<rxvt_ptytty *> ptys; |
|
|
451 | |
|
|
452 | while (read (sock_fd, &cmd, sizeof (command)) == sizeof (command)) |
|
|
453 | { |
|
|
454 | if (cmd.type == command::get) |
|
|
455 | { |
|
|
456 | // -> id ptyfd ttyfd |
|
|
457 | cmd.id = new rxvt_ptytty_unix; |
|
|
458 | |
|
|
459 | if (cmd.id->get ()) |
|
|
460 | { |
|
|
461 | write (sock_fd, &cmd.id, sizeof (cmd.id)); |
|
|
462 | ptys.push_back (cmd.id); |
|
|
463 | |
|
|
464 | rxvt_send_fd (sock_fd, cmd.id->pty); |
|
|
465 | rxvt_send_fd (sock_fd, cmd.id->tty); |
|
|
466 | } |
|
|
467 | else |
|
|
468 | { |
|
|
469 | delete cmd.id; |
|
|
470 | cmd.id = 0; |
|
|
471 | write (sock_fd, &cmd.id, sizeof (cmd.id)); |
|
|
472 | } |
|
|
473 | } |
|
|
474 | else if (cmd.type == command::login) |
|
|
475 | { |
|
|
476 | #if UTMP_SUPPORT |
|
|
477 | if (find (ptys.begin (), ptys.end (), cmd.id)) |
|
|
478 | { |
|
|
479 | cmd.hostname[sizeof (cmd.hostname) - 1] = 0; |
|
|
480 | cmd.id->login (cmd.cmd_pid, cmd.login_shell, cmd.hostname); |
|
|
481 | } |
|
|
482 | #endif |
|
|
483 | } |
|
|
484 | else if (cmd.type == command::destroy) |
|
|
485 | { |
|
|
486 | rxvt_ptytty **pty = find (ptys.begin (), ptys.end (), cmd.id); |
|
|
487 | |
|
|
488 | if (pty) |
|
|
489 | { |
|
|
490 | delete *pty; |
|
|
491 | ptys.erase (pty); |
|
|
492 | } |
|
|
493 | } |
|
|
494 | else |
|
|
495 | break; |
|
|
496 | } |
|
|
497 | |
|
|
498 | // destroy all ptys |
|
|
499 | for (rxvt_ptytty **i = ptys.end (); i-- > ptys.begin (); ) |
|
|
500 | delete *i; |
|
|
501 | } |
|
|
502 | |
|
|
503 | void rxvt_ptytty_server () |
|
|
504 | { |
|
|
505 | int sv[2]; |
|
|
506 | |
|
|
507 | if (socketpair (AF_UNIX, SOCK_STREAM, 0, sv)) |
|
|
508 | rxvt_fatal ("could not create socket to communicate with pty/sessiondb helper, aborting.\n"); |
|
|
509 | |
|
|
510 | pid = fork (); |
|
|
511 | |
|
|
512 | if (pid < 0) |
|
|
513 | rxvt_fatal ("could not create pty/sessiondb helper process, aborting.\n"); |
|
|
514 | |
|
|
515 | if (pid) |
|
|
516 | { |
|
|
517 | // client, urxvt |
|
|
518 | sock_fd = sv[0]; |
|
|
519 | close (sv[1]); |
|
|
520 | fcntl (sock_fd, F_SETFD, FD_CLOEXEC); |
|
|
521 | } |
|
|
522 | else |
|
|
523 | { |
|
|
524 | // server, pty-helper |
|
|
525 | sock_fd = sv[1]; |
|
|
526 | |
|
|
527 | for (int fd = 0; fd < 1023; fd++) |
|
|
528 | if (fd != sock_fd) |
|
|
529 | close (fd); |
|
|
530 | |
|
|
531 | serve (); |
|
|
532 | _exit (EXIT_SUCCESS); |
|
|
533 | } |
|
|
534 | } |
|
|
535 | |
|
|
536 | #endif |
|
|
537 | |
|
|
538 | // a "factory" *g* |
|
|
539 | rxvt_ptytty * |
|
|
540 | rxvt_new_ptytty () |
|
|
541 | { |
|
|
542 | #if PTYTTY_HELPER |
|
|
543 | if (pid > 0) |
|
|
544 | // use helper process |
|
|
545 | return new rxvt_ptytty_proxy; |
|
|
546 | else |
|
|
547 | #endif |
|
|
548 | return new rxvt_ptytty_unix; |
|
|
549 | } |
|
|
550 | |
|
|
551 | /*----------------------- end-of-file (C source) -----------------------*/ |
|
|
552 | |
|
|