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