… | |
… | |
22 | *---------------------------------------------------------------------*/ |
22 | *---------------------------------------------------------------------*/ |
23 | |
23 | |
24 | #include "../config.h" /* NECESSARY */ |
24 | #include "../config.h" /* NECESSARY */ |
25 | #include "rxvt.h" |
25 | #include "rxvt.h" |
26 | |
26 | |
27 | # include <cstdlib> |
27 | #include <cstdlib> |
28 | # include <cstring> |
28 | #include <cstring> |
29 | |
29 | |
30 | #ifdef HAVE_SYS_TYPES_H |
|
|
31 | # include <sys/types.h> |
30 | #include <sys/types.h> |
32 | #endif |
31 | #include <sys/socket.h> |
33 | #ifdef HAVE_UNISTD_H |
|
|
34 | # include <unistd.h> |
32 | #include <unistd.h> |
35 | #endif |
|
|
36 | #ifdef HAVE_FCNTL_H |
|
|
37 | # include <fcntl.h> |
33 | #include <fcntl.h> |
38 | #endif |
34 | |
39 | #ifdef HAVE_SYS_IOCTL_H |
35 | #ifdef HAVE_SYS_IOCTL_H |
40 | # include <sys/ioctl.h> |
36 | # include <sys/ioctl.h> |
41 | #endif |
37 | #endif |
42 | #if defined(PTYS_ARE_PTMX) && defined(HAVE_SYS_STROPTS_H) |
38 | #if defined(PTYS_ARE_PTMX) && defined(HAVE_SYS_STROPTS_H) |
43 | # include <sys/stropts.h> /* for I_PUSH */ |
39 | # include <sys/stropts.h> /* for I_PUSH */ |
… | |
… | |
53 | # include <util.h> |
49 | # include <util.h> |
54 | #endif |
50 | #endif |
55 | |
51 | |
56 | #include <cstdio> |
52 | #include <cstdio> |
57 | #include <grp.h> |
53 | #include <grp.h> |
|
|
54 | |
|
|
55 | #include "rxvtutil.h" |
|
|
56 | #include "fdpass.h" |
|
|
57 | #include "ptytty.h" |
58 | |
58 | |
59 | ///////////////////////////////////////////////////////////////////////////// |
59 | ///////////////////////////////////////////////////////////////////////////// |
60 | |
60 | |
61 | /* ------------------------------------------------------------------------- * |
61 | /* ------------------------------------------------------------------------- * |
62 | * GET PSEUDO TELETYPE - MASTER AND SLAVE * |
62 | * GET PSEUDO TELETYPE - MASTER AND SLAVE * |
… | |
… | |
64 | /* |
64 | /* |
65 | * Returns pty file descriptor, or -1 on failure |
65 | * Returns pty file descriptor, or -1 on failure |
66 | * If successful, ttydev is set to the name of the slave device. |
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 |
67 | * fd_tty _may_ also be set to an open fd to the slave device |
68 | */ |
68 | */ |
69 | static int |
69 | static inline int |
70 | get_pty (int *fd_tty, char **ttydev) |
70 | get_pty_streams (int *fd_tty, char **ttydev) |
71 | { |
71 | { |
|
|
72 | #ifdef NO_SETOWNER_TTYDEV |
72 | int pfd; |
73 | int pfd; |
73 | |
74 | |
|
|
75 | # if defined(PTYS_ARE_GETPT) |
|
|
76 | pfd = getpt(); |
|
|
77 | # elif defined(PTYS_ARE_POSIX) |
|
|
78 | pfd = posix_openpt (O_RDWR); |
|
|
79 | # else |
|
|
80 | pfd = open ("/dev/ptmx", O_RDWR | O_NOCTTY, 0); |
|
|
81 | # endif |
|
|
82 | if (pfd >= 0) |
|
|
83 | { |
|
|
84 | if (grantpt (pfd) == 0 /* change slave permissions */ |
|
|
85 | && unlockpt (pfd) == 0) |
|
|
86 | { /* slave now unlocked */ |
|
|
87 | *ttydev = strdup (ptsname (pfd)); /* get slave's name */ |
|
|
88 | return pfd; |
|
|
89 | } |
|
|
90 | |
|
|
91 | close (pfd); |
|
|
92 | } |
|
|
93 | #endif |
|
|
94 | |
|
|
95 | return -1; |
|
|
96 | } |
|
|
97 | |
|
|
98 | static inline int |
|
|
99 | get_pty_openpty (int *fd_tty, char **ttydev) |
|
|
100 | { |
74 | #ifdef PTYS_ARE_OPENPTY |
101 | #ifdef PTYS_ARE_OPENPTY |
|
|
102 | int pfd; |
|
|
103 | int res; |
75 | char tty_name[sizeof "/dev/pts/????\0"]; |
104 | char tty_name[sizeof "/dev/pts/????\0"]; |
76 | |
105 | |
77 | rxvt_privileges(RESTORE); |
|
|
78 | int res = openpty (&pfd, fd_tty, tty_name, NULL, NULL); |
106 | res = openpty (&pfd, fd_tty, tty_name, NULL, NULL); |
79 | rxvt_privileges(IGNORE); |
|
|
80 | |
|
|
81 | if (res != -1) |
107 | if (res != -1) |
82 | { |
108 | { |
83 | *ttydev = strdup (tty_name); |
109 | *ttydev = strdup (tty_name); |
84 | return pfd; |
110 | return pfd; |
85 | } |
111 | } |
86 | #endif |
112 | #endif |
87 | |
113 | |
|
|
114 | return -1; |
|
|
115 | } |
|
|
116 | |
|
|
117 | static inline int |
|
|
118 | get_pty__getpty (int *fd_tty, char **ttydev) |
|
|
119 | { |
88 | #ifdef PTYS_ARE__GETPTY |
120 | #ifdef PTYS_ARE__GETPTY |
|
|
121 | int pfd; |
|
|
122 | |
89 | *ttydev = _getpty (&pfd, O_RDWR | O_NONBLOCK | O_NOCTTY, 0622, 0); |
123 | *ttydev = _getpty (&pfd, O_RDWR | O_NONBLOCK | O_NOCTTY, 0622, 0); |
90 | if (*ttydev != NULL) |
124 | if (*ttydev != NULL) |
91 | return pfd; |
125 | return pfd; |
92 | #endif |
126 | #endif |
93 | |
127 | |
94 | #if defined(HAVE_GRANTPT) && defined(HAVE_UNLOCKPT) |
128 | return -1; |
95 | # if defined(PTYS_ARE_GETPT) || defined(PTYS_ARE_POSIX) || defined(PTYS_ARE_PTMX) |
129 | } |
96 | |
130 | |
97 | { |
131 | static inline int |
98 | # ifdef PTYS_ARE_GETPT |
132 | get_pty_ptc (int *fd_tty, char **ttydev) |
99 | pfd = getpt(); |
133 | { |
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 |
134 | #ifdef PTYS_ARE_PTC |
|
|
135 | int pfd; |
|
|
136 | |
123 | if ((pfd = open ("/dev/ptc", O_RDWR | O_NOCTTY, 0)) >= 0) |
137 | if ((pfd = open ("/dev/ptc", O_RDWR | O_NOCTTY, 0)) >= 0) |
124 | { |
138 | { |
125 | *ttydev = strdup (ttyname (pfd)); |
139 | *ttydev = strdup (ttyname (pfd)); |
126 | return pfd; |
140 | return pfd; |
127 | } |
141 | } |
128 | #endif |
142 | #endif |
129 | |
143 | |
|
|
144 | return -1; |
|
|
145 | } |
|
|
146 | |
|
|
147 | static inline int |
|
|
148 | get_pty_clone (int *fd_tty, char **ttydev) |
|
|
149 | { |
130 | #ifdef PTYS_ARE_CLONE |
150 | #ifdef PTYS_ARE_CLONE |
|
|
151 | int pfd; |
|
|
152 | |
131 | if ((pfd = open ("/dev/ptym/clone", O_RDWR | O_NOCTTY, 0)) >= 0) |
153 | if ((pfd = open ("/dev/ptym/clone", O_RDWR | O_NOCTTY, 0)) >= 0) |
132 | { |
154 | { |
133 | *ttydev = strdup (ptsname (pfd)); |
155 | *ttydev = strdup (ptsname (pfd)); |
134 | return pfd; |
156 | return pfd; |
135 | } |
157 | } |
136 | #endif |
158 | #endif |
137 | |
159 | |
|
|
160 | return -1; |
|
|
161 | } |
|
|
162 | |
|
|
163 | static inline int |
|
|
164 | get_pty_numeric (int *fd_tty, char **ttydev) |
|
|
165 | { |
138 | #ifdef PTYS_ARE_NUMERIC |
166 | #ifdef PTYS_ARE_NUMERIC |
139 | { |
167 | int pfd; |
140 | int idx; |
168 | int idx; |
141 | char *c1, *c2; |
169 | char *c1, *c2; |
142 | char pty_name[] = "/dev/ptyp???"; |
170 | char pty_name[] = "/dev/ptyp???"; |
143 | char tty_name[] = "/dev/ttyp???"; |
171 | char tty_name[] = "/dev/ttyp???"; |
144 | |
172 | |
145 | c1 = &(pty_name[sizeof (pty_name) - 4]); |
173 | c1 = &(pty_name[sizeof (pty_name) - 4]); |
146 | c2 = &(tty_name[sizeof (tty_name) - 4]); |
174 | c2 = &(tty_name[sizeof (tty_name) - 4]); |
|
|
175 | |
147 | for (idx = 0; idx < 256; idx++) |
176 | for (idx = 0; idx < 256; idx++) |
148 | { |
177 | { |
149 | sprintf (c1, "%d", idx); |
178 | sprintf (c1, "%d", idx); |
150 | sprintf (c2, "%d", idx); |
179 | sprintf (c2, "%d", idx); |
|
|
180 | |
151 | if (access (tty_name, F_OK) < 0) |
181 | if (access (tty_name, F_OK) < 0) |
152 | { |
182 | { |
153 | idx = 256; |
183 | idx = 256; |
154 | break; |
184 | break; |
155 | } |
185 | } |
156 | |
186 | |
157 | if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) >= 0) |
187 | if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) >= 0) |
158 | { |
188 | { |
159 | if (access (tty_name, R_OK | W_OK) == 0) |
189 | if (access (tty_name, R_OK | W_OK) == 0) |
160 | { |
190 | { |
161 | *ttydev = strdup (tty_name); |
191 | *ttydev = strdup (tty_name); |
162 | return pfd; |
192 | return pfd; |
163 | } |
193 | } |
164 | |
194 | |
165 | close (pfd); |
195 | close (pfd); |
166 | } |
196 | } |
167 | } |
197 | } |
168 | } |
|
|
169 | #endif |
198 | #endif |
170 | |
199 | |
|
|
200 | return -1; |
|
|
201 | } |
|
|
202 | |
|
|
203 | static inline int |
|
|
204 | get_pty_searched (int *fd_tty, char **ttydev) |
|
|
205 | { |
171 | #ifdef PTYS_ARE_SEARCHED |
206 | #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 |
207 | # ifndef PTYCHAR1 |
178 | # define PTYCHAR1 "pqrstuvwxyz" |
208 | # define PTYCHAR1 "pqrstuvwxyz" |
179 | # endif |
209 | # endif |
180 | # ifndef PTYCHAR2 |
210 | # ifndef PTYCHAR2 |
181 | # define PTYCHAR2 "0123456789abcdef" |
211 | # define PTYCHAR2 "0123456789abcdef" |
182 | # endif |
212 | # endif |
|
|
213 | int pfd; |
|
|
214 | const char *c1, *c2; |
|
|
215 | char pty_name[] = "/dev/pty??"; |
|
|
216 | char tty_name[] = "/dev/tty??"; |
183 | |
217 | |
184 | for (c1 = PTYCHAR1; *c1; c1++) |
218 | for (c1 = PTYCHAR1; *c1; c1++) |
185 | { |
219 | { |
186 | pty_name[ (sizeof (pty_name) - 3)] = |
220 | pty_name[ (sizeof (pty_name) - 3)] = |
187 | tty_name[ (sizeof (pty_name) - 3)] = *c1; |
221 | tty_name[ (sizeof (pty_name) - 3)] = *c1; |
|
|
222 | |
188 | for (c2 = PTYCHAR2; *c2; c2++) |
223 | for (c2 = PTYCHAR2; *c2; c2++) |
189 | { |
224 | { |
190 | pty_name[ (sizeof (pty_name) - 2)] = |
225 | pty_name[ (sizeof (pty_name) - 2)] = |
191 | tty_name[ (sizeof (pty_name) - 2)] = *c2; |
226 | tty_name[ (sizeof (pty_name) - 2)] = *c2; |
|
|
227 | |
192 | if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) >= 0) |
228 | if ((pfd = open (pty_name, O_RDWR | O_NOCTTY, 0)) >= 0) |
193 | { |
229 | { |
194 | if (access (tty_name, R_OK | W_OK) == 0) |
230 | if (access (tty_name, R_OK | W_OK) == 0) |
195 | { |
231 | { |
196 | *ttydev = strdup (tty_name); |
232 | *ttydev = strdup (tty_name); |
197 | return pfd; |
233 | return pfd; |
198 | } |
234 | } |
199 | |
235 | |
200 | close (pfd); |
236 | close (pfd); |
201 | } |
237 | } |
202 | } |
238 | } |
203 | } |
239 | } |
204 | } |
|
|
205 | #endif |
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; |
206 | |
258 | |
207 | return -1; |
259 | return -1; |
208 | } |
260 | } |
209 | |
261 | |
210 | /*----------------------------------------------------------------------*/ |
262 | /*----------------------------------------------------------------------*/ |
… | |
… | |
222 | * Make our tty a controlling tty so that /dev/tty points to us |
274 | * Make our tty a controlling tty so that /dev/tty points to us |
223 | */ |
275 | */ |
224 | static int |
276 | static int |
225 | control_tty (int fd_tty) |
277 | control_tty (int fd_tty) |
226 | { |
278 | { |
227 | int fd; |
|
|
228 | |
|
|
229 | /* ---------------------------------------- */ |
|
|
230 | # ifdef HAVE_SETSID |
|
|
231 | setsid (); |
279 | setsid (); |
232 | # endif |
|
|
233 | # if defined(HAVE_SETPGID) |
|
|
234 | setpgid (0, 0); |
|
|
235 | # elif defined(HAVE_SETPGRP) |
|
|
236 | setpgrp (0, 0); |
|
|
237 | # endif |
|
|
238 | |
280 | |
239 | /* ---------------------------------------- */ |
|
|
240 | # ifdef TIOCNOTTY |
|
|
241 | fd = open ("/dev/tty", O_RDWR | O_NOCTTY); |
|
|
242 | if (fd >= 0) |
|
|
243 | { |
|
|
244 | ioctl (fd, TIOCNOTTY, NULL); /* void tty associations */ |
|
|
245 | close (fd); |
|
|
246 | } |
|
|
247 | # endif |
|
|
248 | |
|
|
249 | /* ---------------------------------------- */ |
|
|
250 | fd = open ("/dev/tty", O_RDWR | O_NOCTTY); |
|
|
251 | if (fd >= 0) |
|
|
252 | close (fd); /* ouch: still have controlling tty */ |
|
|
253 | |
|
|
254 | /* ---------------------------------------- */ |
|
|
255 | # if defined(PTYS_ARE_PTMX) && defined(I_PUSH) |
281 | # if defined(PTYS_ARE_PTMX) && defined(I_PUSH) |
256 | /* |
282 | /* |
257 | * Push STREAMS modules: |
283 | * Push STREAMS modules: |
258 | * ptem: pseudo-terminal hardware emulation module. |
284 | * ptem: pseudo-terminal hardware emulation module. |
259 | * ldterm: standard terminal line discipline. |
285 | * ldterm: standard terminal line discipline. |
… | |
… | |
277 | ioctl (fd_tty, I_PUSH, "ptem"); |
303 | ioctl (fd_tty, I_PUSH, "ptem"); |
278 | ioctl (fd_tty, I_PUSH, "ldterm"); |
304 | ioctl (fd_tty, I_PUSH, "ldterm"); |
279 | ioctl (fd_tty, I_PUSH, "ttcompat"); |
305 | ioctl (fd_tty, I_PUSH, "ttcompat"); |
280 | } |
306 | } |
281 | # endif |
307 | # endif |
282 | /* ---------------------------------------- */ |
308 | |
283 | fd = ioctl (fd_tty, TIOCSCTTY, NULL); |
309 | ioctl (fd_tty, TIOCSCTTY, NULL); |
284 | /* ---------------------------------------- */ |
310 | |
285 | fd = open ("/dev/tty", O_WRONLY); |
311 | int fd = open ("/dev/tty", O_WRONLY); |
286 | if (fd < 0) |
312 | if (fd < 0) |
287 | return -1; /* fatal */ |
313 | return -1; /* fatal */ |
|
|
314 | |
288 | close (fd); |
315 | close (fd); |
289 | /* ---------------------------------------- */ |
|
|
290 | |
316 | |
291 | return 0; |
317 | return 0; |
292 | } |
318 | } |
293 | |
319 | |
294 | void |
320 | void |
… | |
… | |
365 | void |
391 | void |
366 | rxvt_ptytty_unix::privileges (rxvt_privaction action) |
392 | rxvt_ptytty_unix::privileges (rxvt_privaction action) |
367 | { |
393 | { |
368 | if (!name || !*name) |
394 | if (!name || !*name) |
369 | return; |
395 | return; |
370 | |
|
|
371 | rxvt_privileges (RESTORE); |
|
|
372 | |
396 | |
373 | if (action == SAVE) |
397 | if (action == SAVE) |
374 | { |
398 | { |
375 | # ifndef RESET_TTY_TO_COMMON_DEFAULTS |
399 | # ifndef RESET_TTY_TO_COMMON_DEFAULTS |
376 | /* store original tty status for restoration rxvt_clean_exit () -- rgg 04/12/95 */ |
400 | /* store original tty status for restoration rxvt_clean_exit () -- rgg 04/12/95 */ |
… | |
… | |
397 | } |
421 | } |
398 | # else |
422 | # else |
399 | chmod (name, (S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH)); |
423 | chmod (name, (S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH)); |
400 | chown (name, 0, 0); |
424 | chown (name, 0, 0); |
401 | # endif |
425 | # endif |
402 | |
|
|
403 | } |
426 | } |
404 | |
|
|
405 | rxvt_privileges (IGNORE); |
|
|
406 | } |
427 | } |
407 | #endif |
428 | #endif |
408 | |
429 | |
409 | rxvt_ptytty_unix::rxvt_ptytty_unix () |
430 | rxvt_ptytty_unix::rxvt_ptytty_unix () |
410 | { |
431 | { |
… | |
… | |
418 | #endif |
439 | #endif |
419 | } |
440 | } |
420 | |
441 | |
421 | rxvt_ptytty_unix::~rxvt_ptytty_unix () |
442 | rxvt_ptytty_unix::~rxvt_ptytty_unix () |
422 | { |
443 | { |
|
|
444 | #if UTMP_SUPPORT |
423 | logout (); |
445 | logout (); |
|
|
446 | #endif |
424 | put (); |
447 | put (); |
425 | } |
448 | } |
426 | |
449 | |
427 | void |
450 | void |
428 | rxvt_ptytty_unix::put () |
451 | rxvt_ptytty_unix::put () |
… | |
… | |
463 | } |
486 | } |
464 | |
487 | |
465 | return true; |
488 | return true; |
466 | } |
489 | } |
467 | |
490 | |
|
|
491 | #if PTYTTY_HELPER |
|
|
492 | |
|
|
493 | static int sock_fd; |
|
|
494 | static int pid; |
|
|
495 | |
|
|
496 | struct command |
|
|
497 | { |
|
|
498 | enum { get, login, destroy } type; |
|
|
499 | |
|
|
500 | rxvt_ptytty *id; |
|
|
501 | |
|
|
502 | bool login_shell; |
|
|
503 | int cmd_pid; |
|
|
504 | char hostname[512]; // arbitrary, but should be plenty |
|
|
505 | }; |
|
|
506 | |
|
|
507 | struct rxvt_ptytty_proxy : zero_initialized, rxvt_ptytty |
|
|
508 | { |
|
|
509 | rxvt_ptytty *id; |
|
|
510 | |
|
|
511 | ~rxvt_ptytty_proxy (); |
|
|
512 | |
|
|
513 | bool get (); |
|
|
514 | void login (int cmd_pid, bool login_shell, const char *hostname); |
|
|
515 | }; |
|
|
516 | |
|
|
517 | bool |
|
|
518 | rxvt_ptytty_proxy::get () |
|
|
519 | { |
|
|
520 | command cmd; |
|
|
521 | |
|
|
522 | cmd.type = command::get; |
|
|
523 | |
|
|
524 | write (sock_fd, &cmd, sizeof (cmd)); |
|
|
525 | |
|
|
526 | if (read (sock_fd, &id, sizeof (id)) != sizeof (id)) |
|
|
527 | rxvt_fatal ("protocol error while creating pty using helper process, aborting.\n"); |
|
|
528 | |
|
|
529 | if (!id) |
|
|
530 | return false; |
|
|
531 | |
|
|
532 | if ((pty = rxvt_recv_fd (sock_fd)) < 0 |
|
|
533 | || (tty = rxvt_recv_fd (sock_fd)) < 0) |
|
|
534 | rxvt_fatal ("protocol error while reading pty/tty fds from helper process, aborting.\n"); |
|
|
535 | |
|
|
536 | return true; |
|
|
537 | } |
|
|
538 | |
|
|
539 | void |
|
|
540 | rxvt_ptytty_proxy::login (int cmd_pid, bool login_shell, const char *hostname) |
|
|
541 | { |
|
|
542 | command cmd; |
|
|
543 | |
|
|
544 | cmd.type = command::login; |
|
|
545 | cmd.id = id; |
|
|
546 | cmd.cmd_pid = cmd_pid; |
|
|
547 | cmd.login_shell = login_shell; |
|
|
548 | strncpy (cmd.hostname, hostname, sizeof (cmd.hostname)); |
|
|
549 | |
|
|
550 | write (sock_fd, &cmd, sizeof (cmd)); |
|
|
551 | } |
|
|
552 | |
|
|
553 | rxvt_ptytty_proxy::~rxvt_ptytty_proxy () |
|
|
554 | { |
|
|
555 | command cmd; |
|
|
556 | |
|
|
557 | cmd.type = command::destroy; |
|
|
558 | cmd.id = id; |
|
|
559 | |
|
|
560 | write (sock_fd, &cmd, sizeof (cmd)); |
|
|
561 | } |
|
|
562 | |
|
|
563 | static |
|
|
564 | void serve () |
|
|
565 | { |
|
|
566 | command cmd; |
|
|
567 | vector<rxvt_ptytty *> ptys; |
|
|
568 | |
|
|
569 | while (read (sock_fd, &cmd, sizeof (command)) == sizeof (command)) |
|
|
570 | { |
|
|
571 | if (cmd.type == command::get) |
|
|
572 | { |
|
|
573 | // -> id ptyfd ttyfd |
|
|
574 | cmd.id = new rxvt_ptytty_unix; |
|
|
575 | |
|
|
576 | if (cmd.id->get ()) |
|
|
577 | { |
|
|
578 | write (sock_fd, &cmd.id, sizeof (cmd.id)); |
|
|
579 | ptys.push_back (cmd.id); |
|
|
580 | |
|
|
581 | rxvt_send_fd (sock_fd, cmd.id->pty); |
|
|
582 | rxvt_send_fd (sock_fd, cmd.id->tty); |
|
|
583 | } |
|
|
584 | else |
|
|
585 | { |
|
|
586 | delete cmd.id; |
|
|
587 | cmd.id = 0; |
|
|
588 | write (sock_fd, &cmd.id, sizeof (cmd.id)); |
|
|
589 | } |
|
|
590 | } |
|
|
591 | else if (cmd.type == command::login) |
|
|
592 | { |
|
|
593 | #if UTMP_SUPPORT |
|
|
594 | if (find (ptys.begin (), ptys.end (), cmd.id)) |
|
|
595 | { |
|
|
596 | cmd.hostname[sizeof (cmd.hostname) - 1] = 0; |
|
|
597 | cmd.id->login (cmd.cmd_pid, cmd.login_shell, cmd.hostname); |
|
|
598 | } |
|
|
599 | #endif |
|
|
600 | } |
|
|
601 | else if (cmd.type == command::destroy) |
|
|
602 | { |
|
|
603 | rxvt_ptytty **pty = find (ptys.begin (), ptys.end (), cmd.id); |
|
|
604 | |
|
|
605 | if (pty) |
|
|
606 | { |
|
|
607 | delete *pty; |
|
|
608 | ptys.erase (pty); |
|
|
609 | } |
|
|
610 | } |
|
|
611 | else |
|
|
612 | break; |
|
|
613 | } |
|
|
614 | |
|
|
615 | // destroy all ptys |
|
|
616 | for (rxvt_ptytty **i = ptys.end (); i-- > ptys.begin (); ) |
|
|
617 | delete *i; |
|
|
618 | } |
|
|
619 | |
|
|
620 | void rxvt_ptytty_server () |
|
|
621 | { |
|
|
622 | int sv[2]; |
|
|
623 | |
|
|
624 | if (socketpair (AF_UNIX, SOCK_STREAM, 0, sv)) |
|
|
625 | rxvt_fatal ("could not create socket to communicate with pty/sessiondb helper, aborting.\n"); |
|
|
626 | |
|
|
627 | pid = fork (); |
|
|
628 | |
|
|
629 | if (pid < 0) |
|
|
630 | rxvt_fatal ("could not create pty/sessiondb helper process, aborting.\n"); |
|
|
631 | |
|
|
632 | if (pid) |
|
|
633 | { |
|
|
634 | // client, urxvt |
|
|
635 | sock_fd = sv[0]; |
|
|
636 | close (sv[1]); |
|
|
637 | fcntl (sock_fd, F_SETFD, FD_CLOEXEC); |
|
|
638 | } |
|
|
639 | else |
|
|
640 | { |
|
|
641 | setgid (getegid ()); |
|
|
642 | setuid (geteuid ()); |
|
|
643 | |
|
|
644 | // server, pty-helper |
|
|
645 | sock_fd = sv[1]; |
|
|
646 | |
|
|
647 | for (int fd = 0; fd < 1023; fd++) |
|
|
648 | if (fd != sock_fd && fd != 1) |
|
|
649 | close (fd); |
|
|
650 | |
|
|
651 | serve (); |
|
|
652 | _exit (EXIT_SUCCESS); |
|
|
653 | } |
|
|
654 | } |
|
|
655 | #endif |
|
|
656 | |
|
|
657 | // a "factory" *g* |
|
|
658 | rxvt_ptytty * |
|
|
659 | rxvt_new_ptytty () |
|
|
660 | { |
|
|
661 | #if PTYTTY_HELPER |
|
|
662 | if (pid > 0) |
|
|
663 | // use helper process |
|
|
664 | return new rxvt_ptytty_proxy; |
|
|
665 | else |
|
|
666 | #endif |
|
|
667 | return new rxvt_ptytty_unix; |
|
|
668 | } |
|
|
669 | |
468 | /*----------------------- end-of-file (C source) -----------------------*/ |
670 | /*----------------------- end-of-file (C source) -----------------------*/ |
469 | |
671 | |