4 * Copyright (c) 2009 Nicholas Marriott <nicm@users.sourceforge.net>
6 * Permission to use, copy, modify, and distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14 * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
15 * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
16 * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
19 #include <sys/types.h>
30 void server_client_handle_key(int, struct mouse_event
*, void *);
31 void server_client_repeat_timer(int, short, void *);
32 void server_client_check_exit(struct client
*);
33 void server_client_check_backoff(struct client
*);
34 void server_client_check_redraw(struct client
*);
35 void server_client_set_title(struct client
*);
36 void server_client_reset_state(struct client
*);
37 void server_client_in_callback(struct bufferevent
*, short, void *);
38 void server_client_out_callback(struct bufferevent
*, short, void *);
39 void server_client_err_callback(struct bufferevent
*, short, void *);
41 int server_client_msg_dispatch(struct client
*);
42 void server_client_msg_command(struct client
*, struct msg_command_data
*);
43 void server_client_msg_identify(
44 struct client
*, struct msg_identify_data
*, int);
45 void server_client_msg_shell(struct client
*);
47 void printflike2
server_client_msg_error(struct cmd_ctx
*, const char *, ...);
48 void printflike2
server_client_msg_print(struct cmd_ctx
*, const char *, ...);
49 void printflike2
server_client_msg_info(struct cmd_ctx
*, const char *, ...);
51 /* Create a new client. */
53 server_client_create(int fd
)
60 c
= xcalloc(1, sizeof *c
);
62 imsg_init(&c
->ibuf
, fd
);
63 server_update_event(c
);
65 if (gettimeofday(&c
->creation_time
, NULL
) != 0)
66 fatal("gettimeofday failed");
67 memcpy(&c
->activity_time
, &c
->creation_time
, sizeof c
->activity_time
);
69 c
->stdin_event
= NULL
;
70 c
->stdout_event
= NULL
;
71 c
->stderr_event
= NULL
;
77 c
->last_session
= NULL
;
81 screen_init(&c
->status
, c
->tty
.sx
, 1, 0);
82 RB_INIT(&c
->status_new
);
83 RB_INIT(&c
->status_old
);
85 c
->message_string
= NULL
;
86 ARRAY_INIT(&c
->message_log
);
88 c
->prompt_string
= NULL
;
89 c
->prompt_buffer
= NULL
;
92 c
->last_mouse
.b
= MOUSE_UP
;
93 c
->last_mouse
.x
= c
->last_mouse
.y
= -1;
95 evtimer_set(&c
->repeat_timer
, server_client_repeat_timer
, c
);
97 for (i
= 0; i
< ARRAY_LENGTH(&clients
); i
++) {
98 if (ARRAY_ITEM(&clients
, i
) == NULL
) {
99 ARRAY_SET(&clients
, i
, c
);
103 ARRAY_ADD(&clients
, c
);
104 log_debug("new client %d", fd
);
109 server_client_lost(struct client
*c
)
111 struct message_entry
*msg
;
114 for (i
= 0; i
< ARRAY_LENGTH(&clients
); i
++) {
115 if (ARRAY_ITEM(&clients
, i
) == c
)
116 ARRAY_SET(&clients
, i
, NULL
);
118 log_debug("lost client %d", c
->ibuf
.fd
);
121 * If CLIENT_TERMINAL hasn't been set, then tty_init hasn't been called
122 * and tty_free might close an unrelated fd.
124 if (c
->flags
& CLIENT_TERMINAL
)
127 if (c
->stdin_fd
!= -1) {
128 setblocking(c
->stdin_fd
, 1);
131 if (c
->stdin_event
!= NULL
)
132 bufferevent_free(c
->stdin_event
);
133 if (c
->stdout_fd
!= -1) {
134 setblocking(c
->stdout_fd
, 1);
137 if (c
->stdout_event
!= NULL
)
138 bufferevent_free(c
->stdout_event
);
139 if (c
->stderr_fd
!= -1) {
140 setblocking(c
->stderr_fd
, 1);
143 if (c
->stderr_event
!= NULL
)
144 bufferevent_free(c
->stderr_event
);
146 status_free_jobs(&c
->status_new
);
147 status_free_jobs(&c
->status_old
);
148 screen_free(&c
->status
);
150 if (c
->title
!= NULL
)
153 evtimer_del(&c
->repeat_timer
);
155 evtimer_del(&c
->identify_timer
);
157 if (c
->message_string
!= NULL
)
158 xfree(c
->message_string
);
159 evtimer_del(&c
->message_timer
);
160 for (i
= 0; i
< ARRAY_LENGTH(&c
->message_log
); i
++) {
161 msg
= &ARRAY_ITEM(&c
->message_log
, i
);
164 ARRAY_FREE(&c
->message_log
);
166 if (c
->prompt_string
!= NULL
)
167 xfree(c
->prompt_string
);
168 if (c
->prompt_buffer
!= NULL
)
169 xfree(c
->prompt_buffer
);
174 environ_free(&c
->environ
);
177 imsg_clear(&c
->ibuf
);
178 event_del(&c
->event
);
180 for (i
= 0; i
< ARRAY_LENGTH(&dead_clients
); i
++) {
181 if (ARRAY_ITEM(&dead_clients
, i
) == NULL
) {
182 ARRAY_SET(&dead_clients
, i
, c
);
186 if (i
== ARRAY_LENGTH(&dead_clients
))
187 ARRAY_ADD(&dead_clients
, c
);
188 c
->flags
|= CLIENT_DEAD
;
191 server_check_unattached();
192 server_update_socket();
195 /* Process a single client event. */
197 server_client_callback(int fd
, short events
, void *data
)
199 struct client
*c
= data
;
201 if (c
->flags
& CLIENT_DEAD
)
204 if (fd
== c
->ibuf
.fd
) {
205 if (events
& EV_WRITE
&& msgbuf_write(&c
->ibuf
.w
) < 0)
208 if (c
->flags
& CLIENT_BAD
) {
209 if (c
->ibuf
.w
.queued
== 0)
214 if (events
& EV_READ
&& server_client_msg_dispatch(c
) != 0)
218 server_update_event(c
);
222 server_client_lost(c
);
225 /* Handle client status timer. */
227 server_client_status_timer(void)
236 if (gettimeofday(&tv
, NULL
) != 0)
237 fatal("gettimeofday failed");
239 for (i
= 0; i
< ARRAY_LENGTH(&clients
); i
++) {
240 c
= ARRAY_ITEM(&clients
, i
);
241 if (c
== NULL
|| c
->session
== NULL
)
243 if (c
->message_string
!= NULL
|| c
->prompt_string
!= NULL
) {
245 * Don't need timed redraw for messages/prompts so bail
246 * now. The status timer isn't reset when they are
253 if (!options_get_number(&s
->options
, "status"))
255 interval
= options_get_number(&s
->options
, "status-interval");
257 difference
= tv
.tv_sec
- c
->status_timer
.tv_sec
;
258 if (difference
>= interval
) {
259 status_update_jobs(c
);
260 c
->flags
|= CLIENT_STATUS
;
265 /* Handle data key input from client. */
267 server_client_handle_key(int key
, struct mouse_event
*mouse
, void *data
)
269 struct client
*c
= data
;
272 struct window_pane
*wp
;
275 struct key_binding
*bd
;
276 struct keylist
*keylist
;
277 int xtimeout
, isprefix
;
280 /* Check the client is good to accept input. */
281 if ((c
->flags
& (CLIENT_DEAD
|CLIENT_SUSPENDED
)) != 0)
283 if (c
->session
== NULL
)
287 /* Update the activity timer. */
288 if (gettimeofday(&c
->activity_time
, NULL
) != 0)
289 fatal("gettimeofday failed");
290 memcpy(&s
->activity_time
, &c
->activity_time
, sizeof s
->activity_time
);
292 w
= c
->session
->curw
->window
;
294 oo
= &c
->session
->options
;
296 /* Special case: number keys jump to pane in identify mode. */
297 if (c
->flags
& CLIENT_IDENTIFY
&& key
>= '0' && key
<= '9') {
298 if (c
->flags
& CLIENT_READONLY
)
300 wp
= window_pane_at_index(w
, key
- '0');
301 if (wp
!= NULL
&& window_pane_visible(wp
))
302 window_set_active_pane(w
, wp
);
303 server_clear_identify(c
);
307 /* Handle status line. */
308 if (!(c
->flags
& CLIENT_READONLY
)) {
309 status_message_clear(c
);
310 server_clear_identify(c
);
312 if (c
->prompt_string
!= NULL
) {
313 if (!(c
->flags
& CLIENT_READONLY
))
314 status_prompt_key(c
, key
);
318 /* Check for mouse keys. */
319 if (key
== KEYC_MOUSE
) {
320 if (c
->flags
& CLIENT_READONLY
)
322 if (options_get_number(oo
, "mouse-select-pane") &&
323 ((!(mouse
->b
& MOUSE_DRAG
) && mouse
->b
!= MOUSE_UP
) ||
324 wp
->mode
!= &window_copy_mode
)) {
326 * Allow pane switching in copy mode only by mouse down
329 window_set_active_at(w
, mouse
->x
, mouse
->y
);
330 server_redraw_window_borders(w
);
333 if (mouse
->y
+ 1 == c
->tty
.sy
&&
334 options_get_number(oo
, "mouse-select-window") &&
335 options_get_number(oo
, "status")) {
336 if (mouse
->b
== MOUSE_UP
&&
337 c
->last_mouse
.b
!= MOUSE_UP
) {
338 status_set_window_at(c
, mouse
->x
);
341 if (mouse
->b
& MOUSE_45
) {
342 if ((mouse
->b
& MOUSE_BUTTON
) == MOUSE_1
) {
343 session_previous(c
->session
, 0);
344 server_redraw_session(s
);
346 if ((mouse
->b
& MOUSE_BUTTON
) == MOUSE_2
) {
347 session_next(c
->session
, 0);
348 server_redraw_session(s
);
353 if (options_get_number(oo
, "mouse-resize-pane"))
354 layout_resize_pane_mouse(c
, mouse
);
355 memcpy(&c
->last_mouse
, mouse
, sizeof c
->last_mouse
);
356 window_pane_mouse(wp
, c
->session
, mouse
);
360 /* Is this a prefix key? */
361 keylist
= options_get_data(&c
->session
->options
, "prefix");
363 for (i
= 0; i
< ARRAY_LENGTH(keylist
); i
++) {
364 if (key
== ARRAY_ITEM(keylist
, i
)) {
370 /* No previous prefix key. */
371 if (!(c
->flags
& CLIENT_PREFIX
)) {
373 c
->flags
|= CLIENT_PREFIX
;
375 /* Try as a non-prefix key binding. */
376 if ((bd
= key_bindings_lookup(key
)) == NULL
) {
377 if (!(c
->flags
& CLIENT_READONLY
))
378 window_pane_key(wp
, c
->session
, key
);
380 key_bindings_dispatch(bd
, c
);
385 /* Prefix key already pressed. Reset prefix and lookup key. */
386 c
->flags
&= ~CLIENT_PREFIX
;
387 if ((bd
= key_bindings_lookup(key
| KEYC_PREFIX
)) == NULL
) {
388 /* If repeating, treat this as a key, else ignore. */
389 if (c
->flags
& CLIENT_REPEAT
) {
390 c
->flags
&= ~CLIENT_REPEAT
;
392 c
->flags
|= CLIENT_PREFIX
;
393 else if (!(c
->flags
& CLIENT_READONLY
))
394 window_pane_key(wp
, c
->session
, key
);
399 /* If already repeating, but this key can't repeat, skip it. */
400 if (c
->flags
& CLIENT_REPEAT
&& !bd
->can_repeat
) {
401 c
->flags
&= ~CLIENT_REPEAT
;
403 c
->flags
|= CLIENT_PREFIX
;
404 else if (!(c
->flags
& CLIENT_READONLY
))
405 window_pane_key(wp
, c
->session
, key
);
409 /* If this key can repeat, reset the repeat flags and timer. */
410 xtimeout
= options_get_number(&c
->session
->options
, "repeat-time");
411 if (xtimeout
!= 0 && bd
->can_repeat
) {
412 c
->flags
|= CLIENT_PREFIX
|CLIENT_REPEAT
;
414 tv
.tv_sec
= xtimeout
/ 1000;
415 tv
.tv_usec
= (xtimeout
% 1000) * 1000L;
416 evtimer_del(&c
->repeat_timer
);
417 evtimer_add(&c
->repeat_timer
, &tv
);
420 /* Dispatch the command. */
421 key_bindings_dispatch(bd
, c
);
424 /* Client functions that need to happen every loop. */
426 server_client_loop(void)
430 struct window_pane
*wp
;
433 for (i
= 0; i
< ARRAY_LENGTH(&clients
); i
++) {
434 c
= ARRAY_ITEM(&clients
, i
);
438 server_client_check_exit(c
);
439 if (c
->session
!= NULL
) {
440 server_client_check_redraw(c
);
441 server_client_reset_state(c
);
446 * Any windows will have been redrawn as part of clients, so clear
449 for (i
= 0; i
< ARRAY_LENGTH(&windows
); i
++) {
450 w
= ARRAY_ITEM(&windows
, i
);
454 w
->flags
&= ~WINDOW_REDRAW
;
455 TAILQ_FOREACH(wp
, &w
->panes
, entry
)
456 wp
->flags
&= ~PANE_REDRAW
;
461 * Update cursor position and mode settings. The scroll region and attributes
462 * are cleared when idle (waiting for an event) as this is the most likely time
463 * a user may interrupt tmux, for example with ~^Z in ssh(1). This is a
464 * compromise between excessive resets and likelihood of an interrupt.
466 * tty_region/tty_reset/tty_update_mode already take care of not resetting
467 * things that are already in their default state.
470 server_client_reset_state(struct client
*c
)
472 struct window
*w
= c
->session
->curw
->window
;
473 struct window_pane
*wp
= w
->active
;
474 struct screen
*s
= wp
->screen
;
475 struct options
*oo
= &c
->session
->options
;
476 struct options
*wo
= &w
->options
;
479 if (c
->flags
& CLIENT_SUSPENDED
)
482 tty_region(&c
->tty
, 0, c
->tty
.sy
- 1);
484 status
= options_get_number(oo
, "status");
485 if (!window_pane_visible(wp
) || wp
->yoff
+ s
->cy
>= c
->tty
.sy
- status
)
486 tty_cursor(&c
->tty
, 0, 0);
488 tty_cursor(&c
->tty
, wp
->xoff
+ s
->cx
, wp
->yoff
+ s
->cy
);
491 * Resizing panes with the mouse requires at least button mode to give
492 * a smooth appearance.
495 if ((c
->last_mouse
.b
& MOUSE_RESIZE_PANE
) &&
496 !(mode
& (MODE_MOUSE_BUTTON
|MODE_MOUSE_ANY
)))
497 mode
|= MODE_MOUSE_BUTTON
;
500 * Any mode will do for mouse-select-pane, but set standard mode if
503 if ((mode
& ALL_MOUSE_MODES
) == 0) {
504 if (TAILQ_NEXT(TAILQ_FIRST(&w
->panes
), entry
) != NULL
&&
505 options_get_number(oo
, "mouse-select-pane"))
506 mode
|= MODE_MOUSE_STANDARD
;
507 else if (options_get_number(oo
, "mouse-resize-pane"))
508 mode
|= MODE_MOUSE_STANDARD
;
509 else if (options_get_number(oo
, "mouse-select-window"))
510 mode
|= MODE_MOUSE_STANDARD
;
511 else if (options_get_number(wo
, "mode-mouse"))
512 mode
|= MODE_MOUSE_STANDARD
;
516 * Set UTF-8 mouse input if required. If the terminal is UTF-8, the
517 * user has set mouse-utf8 and any mouse mode is in effect, turn on
518 * UTF-8 mouse input. If the receiving terminal hasn't requested it
519 * (that is, it isn't in s->mode), then it'll be converted in
522 if ((c
->tty
.flags
& TTY_UTF8
) &&
523 (mode
& ALL_MOUSE_MODES
) && options_get_number(oo
, "mouse-utf8"))
524 mode
|= MODE_MOUSE_UTF8
;
526 mode
&= ~MODE_MOUSE_UTF8
;
528 /* Set the terminal mode and reset attributes. */
529 tty_update_mode(&c
->tty
, mode
, s
);
533 /* Repeat time callback. */
536 server_client_repeat_timer(unused
int fd
, unused
short events
, void *data
)
538 struct client
*c
= data
;
540 if (c
->flags
& CLIENT_REPEAT
)
541 c
->flags
&= ~(CLIENT_PREFIX
|CLIENT_REPEAT
);
544 /* Check if client should be exited. */
546 server_client_check_exit(struct client
*c
)
548 struct msg_exit_data exitdata
;
550 if (!(c
->flags
& CLIENT_EXIT
))
553 if (c
->stdout_fd
!= -1 && c
->stdout_event
!= NULL
&&
554 EVBUFFER_LENGTH(c
->stdout_event
->output
) != 0)
556 if (c
->stderr_fd
!= -1 && c
->stderr_event
!= NULL
&&
557 EVBUFFER_LENGTH(c
->stderr_event
->output
) != 0)
560 exitdata
.retcode
= c
->retcode
;
561 server_write_client(c
, MSG_EXIT
, &exitdata
, sizeof exitdata
);
563 c
->flags
&= ~CLIENT_EXIT
;
567 * Check if the client should backoff. During backoff, data from external
568 * programs is not written to the terminal. When the existing data drains, the
571 * There are two backoff phases - both the tty and client have backoff flags -
572 * the first to allow existing data to drain and the latter to ensure backoff
573 * is disabled until the redraw has finished and prevent the redraw triggering
577 server_client_check_backoff(struct client
*c
)
579 struct tty
*tty
= &c
->tty
;
582 used
= EVBUFFER_LENGTH(tty
->event
->output
);
585 * If in the second backoff phase (redrawing), don't check backoff
586 * until the redraw has completed (or enough of it to drop below the
587 * backoff threshold).
589 if (c
->flags
& CLIENT_BACKOFF
) {
590 if (used
> BACKOFF_THRESHOLD
)
592 c
->flags
&= ~CLIENT_BACKOFF
;
596 /* Once drained, allow data through again and schedule redraw. */
597 if (tty
->flags
& TTY_BACKOFF
) {
600 tty
->flags
&= ~TTY_BACKOFF
;
601 c
->flags
|= (CLIENT_BACKOFF
|CLIENT_REDRAWWINDOW
|CLIENT_STATUS
);
605 /* If too much data, start backoff. */
606 if (used
> BACKOFF_THRESHOLD
)
607 tty
->flags
|= TTY_BACKOFF
;
610 /* Check for client redraws. */
612 server_client_check_redraw(struct client
*c
)
614 struct session
*s
= c
->session
;
615 struct window_pane
*wp
;
618 flags
= c
->tty
.flags
& TTY_FREEZE
;
619 c
->tty
.flags
&= ~TTY_FREEZE
;
621 if (c
->flags
& (CLIENT_REDRAW
|CLIENT_STATUS
)) {
622 if (options_get_number(&s
->options
, "set-titles"))
623 server_client_set_title(c
);
625 if (c
->message_string
!= NULL
)
626 redraw
= status_message_redraw(c
);
627 else if (c
->prompt_string
!= NULL
)
628 redraw
= status_prompt_redraw(c
);
630 redraw
= status_redraw(c
);
632 c
->flags
&= ~CLIENT_STATUS
;
635 if (c
->flags
& CLIENT_REDRAW
) {
636 screen_redraw_screen(c
, 0, 0);
637 c
->flags
&= ~(CLIENT_STATUS
|CLIENT_BORDERS
);
638 } else if (c
->flags
& CLIENT_REDRAWWINDOW
) {
639 TAILQ_FOREACH(wp
, &c
->session
->curw
->window
->panes
, entry
)
640 screen_redraw_pane(c
, wp
);
641 c
->flags
&= ~CLIENT_REDRAWWINDOW
;
643 TAILQ_FOREACH(wp
, &c
->session
->curw
->window
->panes
, entry
) {
644 if (wp
->flags
& PANE_REDRAW
)
645 screen_redraw_pane(c
, wp
);
649 if (c
->flags
& CLIENT_BORDERS
)
650 screen_redraw_screen(c
, 0, 1);
652 if (c
->flags
& CLIENT_STATUS
)
653 screen_redraw_screen(c
, 1, 0);
655 c
->tty
.flags
|= flags
;
657 c
->flags
&= ~(CLIENT_REDRAW
|CLIENT_STATUS
|CLIENT_BORDERS
);
660 /* Set client title. */
662 server_client_set_title(struct client
*c
)
664 struct session
*s
= c
->session
;
665 const char *template;
668 template = options_get_string(&s
->options
, "set-titles-string");
670 title
= status_replace(c
, NULL
, NULL
, NULL
, template, time(NULL
), 1);
671 if (c
->title
== NULL
|| strcmp(title
, c
->title
) != 0) {
672 if (c
->title
!= NULL
)
674 c
->title
= xstrdup(title
);
675 tty_set_title(&c
->tty
, c
->title
);
681 * Error callback for client stdin. Caller must increase reference count when
685 server_client_in_callback(
686 unused
struct bufferevent
*bufev
, unused
short what
, void *data
)
688 struct client
*c
= data
;
691 if (c
->flags
& CLIENT_DEAD
)
694 bufferevent_disable(c
->stdin_event
, EV_READ
|EV_WRITE
);
695 setblocking(c
->stdin_fd
, 1);
699 if (c
->stdin_callback
!= NULL
)
700 c
->stdin_callback(c
, c
->stdin_data
);
703 /* Error callback for client stdout. */
705 server_client_out_callback(
706 unused
struct bufferevent
*bufev
, unused
short what
, unused
void *data
)
708 struct client
*c
= data
;
710 bufferevent_disable(c
->stdout_event
, EV_READ
|EV_WRITE
);
711 setblocking(c
->stdout_fd
, 1);
716 /* Error callback for client stderr. */
718 server_client_err_callback(
719 unused
struct bufferevent
*bufev
, unused
short what
, unused
void *data
)
721 struct client
*c
= data
;
723 bufferevent_disable(c
->stderr_event
, EV_READ
|EV_WRITE
);
724 setblocking(c
->stderr_fd
, 1);
729 /* Dispatch message from client. */
731 server_client_msg_dispatch(struct client
*c
)
734 struct msg_command_data commanddata
;
735 struct msg_identify_data identifydata
;
736 struct msg_environ_data environdata
;
739 if ((n
= imsg_read(&c
->ibuf
)) == -1 || n
== 0)
743 if ((n
= imsg_get(&c
->ibuf
, &imsg
)) == -1)
747 datalen
= imsg
.hdr
.len
- IMSG_HEADER_SIZE
;
749 if (imsg
.hdr
.peerid
!= PROTOCOL_VERSION
) {
750 server_write_client(c
, MSG_VERSION
, NULL
, 0);
751 c
->flags
|= CLIENT_BAD
;
756 log_debug("got %d from client %d", imsg
.hdr
.type
, c
->ibuf
.fd
);
757 switch (imsg
.hdr
.type
) {
759 if (datalen
!= sizeof commanddata
)
760 fatalx("bad MSG_COMMAND size");
761 memcpy(&commanddata
, imsg
.data
, sizeof commanddata
);
763 server_client_msg_command(c
, &commanddata
);
766 if (datalen
!= sizeof identifydata
)
767 fatalx("bad MSG_IDENTIFY size");
769 fatalx("MSG_IDENTIFY missing fd");
770 memcpy(&identifydata
, imsg
.data
, sizeof identifydata
);
772 c
->stdin_fd
= imsg
.fd
;
773 c
->stdin_event
= bufferevent_new(c
->stdin_fd
,
774 NULL
, NULL
, server_client_in_callback
, c
);
775 if (c
->stdin_event
== NULL
)
776 fatalx("failed to create stdin event");
777 setblocking(c
->stdin_fd
, 0);
779 server_client_msg_identify(c
, &identifydata
, imsg
.fd
);
783 fatalx("bad MSG_STDOUT size");
785 fatalx("MSG_STDOUT missing fd");
787 c
->stdout_fd
= imsg
.fd
;
788 c
->stdout_event
= bufferevent_new(c
->stdout_fd
,
789 NULL
, NULL
, server_client_out_callback
, c
);
790 if (c
->stdout_event
== NULL
)
791 fatalx("failed to create stdout event");
792 setblocking(c
->stdout_fd
, 0);
797 fatalx("bad MSG_STDERR size");
799 fatalx("MSG_STDERR missing fd");
801 c
->stderr_fd
= imsg
.fd
;
802 c
->stderr_event
= bufferevent_new(c
->stderr_fd
,
803 NULL
, NULL
, server_client_err_callback
, c
);
804 if (c
->stderr_event
== NULL
)
805 fatalx("failed to create stderr event");
806 setblocking(c
->stderr_fd
, 0);
811 fatalx("bad MSG_RESIZE size");
813 if (tty_resize(&c
->tty
)) {
815 server_redraw_client(c
);
820 fatalx("bad MSG_EXITING size");
824 server_write_client(c
, MSG_EXITED
, NULL
, 0);
829 fatalx("bad MSG_WAKEUP size");
831 if (!(c
->flags
& CLIENT_SUSPENDED
))
833 c
->flags
&= ~CLIENT_SUSPENDED
;
835 if (gettimeofday(&c
->activity_time
, NULL
) != 0)
836 fatal("gettimeofday");
837 if (c
->session
!= NULL
)
838 session_update_activity(c
->session
);
840 tty_start_tty(&c
->tty
);
841 server_redraw_client(c
);
845 if (datalen
!= sizeof environdata
)
846 fatalx("bad MSG_ENVIRON size");
847 memcpy(&environdata
, imsg
.data
, sizeof environdata
);
849 environdata
.var
[(sizeof environdata
.var
) - 1] = '\0';
850 if (strchr(environdata
.var
, '=') != NULL
)
851 environ_put(&c
->environ
, environdata
.var
);
855 fatalx("bad MSG_SHELL size");
857 server_client_msg_shell(c
);
860 fatalx("unexpected message");
867 /* Callback to send error message to client. */
869 server_client_msg_error(struct cmd_ctx
*ctx
, const char *fmt
, ...)
874 evbuffer_add_vprintf(ctx
->cmdclient
->stderr_event
->output
, fmt
, ap
);
877 bufferevent_write(ctx
->cmdclient
->stderr_event
, "\n", 1);
878 ctx
->cmdclient
->retcode
= 1;
881 /* Callback to send print message to client. */
883 server_client_msg_print(struct cmd_ctx
*ctx
, const char *fmt
, ...)
888 evbuffer_add_vprintf(ctx
->cmdclient
->stdout_event
->output
, fmt
, ap
);
891 bufferevent_write(ctx
->cmdclient
->stdout_event
, "\n", 1);
894 /* Callback to send print message to client, if not quiet. */
896 server_client_msg_info(struct cmd_ctx
*ctx
, const char *fmt
, ...)
900 if (options_get_number(&global_options
, "quiet"))
904 evbuffer_add_vprintf(ctx
->cmdclient
->stdout_event
->output
, fmt
, ap
);
907 bufferevent_write(ctx
->cmdclient
->stdout_event
, "\n", 1);
910 /* Handle command message. */
912 server_client_msg_command(struct client
*c
, struct msg_command_data
*data
)
915 struct cmd_list
*cmdlist
= NULL
;
919 ctx
.error
= server_client_msg_error
;
920 ctx
.print
= server_client_msg_print
;
921 ctx
.info
= server_client_msg_info
;
924 ctx
.curclient
= NULL
;
929 data
->argv
[(sizeof data
->argv
) - 1] = '\0';
930 if (cmd_unpack_argv(data
->argv
, sizeof data
->argv
, argc
, &argv
) != 0) {
931 server_client_msg_error(&ctx
, "command too long");
937 argv
= xcalloc(1, sizeof *argv
);
938 *argv
= xstrdup("new-session");
941 if ((cmdlist
= cmd_list_parse(argc
, argv
, &cause
)) == NULL
) {
942 server_client_msg_error(&ctx
, "%s", cause
);
943 cmd_free_argv(argc
, argv
);
946 cmd_free_argv(argc
, argv
);
948 if (cmd_list_exec(cmdlist
, &ctx
) != 1)
949 c
->flags
|= CLIENT_EXIT
;
950 cmd_list_free(cmdlist
);
955 cmd_list_free(cmdlist
);
956 c
->flags
|= CLIENT_EXIT
;
959 /* Handle identify message. */
961 server_client_msg_identify(
962 struct client
*c
, struct msg_identify_data
*data
, int fd
)
967 data
->cwd
[(sizeof data
->cwd
) - 1] = '\0';
968 if (*data
->cwd
!= '\0')
969 c
->cwd
= xstrdup(data
->cwd
);
973 if ((tty_fd
= dup(fd
)) == -1)
975 data
->term
[(sizeof data
->term
) - 1] = '\0';
976 tty_init(&c
->tty
, tty_fd
, data
->term
);
977 if (data
->flags
& IDENTIFY_UTF8
)
978 c
->tty
.flags
|= TTY_UTF8
;
979 if (data
->flags
& IDENTIFY_256COLOURS
)
980 c
->tty
.term_flags
|= TERM_256COLOURS
;
981 else if (data
->flags
& IDENTIFY_88COLOURS
)
982 c
->tty
.term_flags
|= TERM_88COLOURS
;
983 c
->tty
.key_callback
= server_client_handle_key
;
988 c
->flags
|= CLIENT_TERMINAL
;
991 /* Handle shell message. */
993 server_client_msg_shell(struct client
*c
)
995 struct msg_shell_data data
;
998 shell
= options_get_string(&global_s_options
, "default-shell");
1000 if (*shell
== '\0' || areshell(shell
))
1001 shell
= _PATH_BSHELL
;
1002 if (strlcpy(data
.shell
, shell
, sizeof data
.shell
) >= sizeof data
.shell
)
1003 strlcpy(data
.shell
, _PATH_BSHELL
, sizeof data
.shell
);
1005 server_write_client(c
, MSG_SHELL
, &data
, sizeof data
);
1006 c
->flags
|= CLIENT_BAD
; /* it will die after exec */