4 * Copyright (c) 2007 Nicholas Marriott <nicholas.marriott@gmail.com>
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>
29 struct sessions sessions
;
30 u_int next_session_id
;
31 struct session_groups session_groups
= RB_INITIALIZER(&session_groups
);
33 static void session_free(int, short, void *);
35 static void session_lock_timer(int, short, void *);
37 static struct winlink
*session_next_alert(struct winlink
*);
38 static struct winlink
*session_previous_alert(struct winlink
*);
40 static void session_group_remove(struct session
*);
41 static void session_group_synchronize1(struct session
*, struct session
*);
44 session_cmp(struct session
*s1
, struct session
*s2
)
46 return (strcmp(s1
->name
, s2
->name
));
48 RB_GENERATE(sessions
, session
, entry
, session_cmp
);
51 session_group_cmp(struct session_group
*s1
, struct session_group
*s2
)
53 return (strcmp(s1
->name
, s2
->name
));
55 RB_GENERATE_STATIC(session_groups
, session_group
, entry
, session_group_cmp
);
58 * Find if session is still alive. This is true if it is still on the global
62 session_alive(struct session
*s
)
64 struct session
*s_loop
;
66 RB_FOREACH(s_loop
, sessions
, &sessions
) {
73 /* Find session by name. */
75 session_find(const char *name
)
79 s
.name
= (char *) name
;
80 return (RB_FIND(sessions
, &sessions
, &s
));
83 /* Find session by id parsed from a string. */
85 session_find_by_id_str(const char *s
)
93 id
= strtonum(s
+ 1, 0, UINT_MAX
, &errstr
);
96 return (session_find_by_id(id
));
99 /* Find session by id. */
101 session_find_by_id(u_int id
)
105 RB_FOREACH(s
, sessions
, &sessions
) {
112 /* Create a new session. */
114 session_create(const char *prefix
, const char *name
, const char *cwd
,
115 struct environ
*env
, struct options
*oo
, struct termios
*tio
)
119 s
= xcalloc(1, sizeof *s
);
123 s
->cwd
= xstrdup(cwd
);
125 TAILQ_INIT(&s
->lastw
);
126 RB_INIT(&s
->windows
);
131 status_update_cache(s
);
135 s
->tio
= xmalloc(sizeof *s
->tio
);
136 memcpy(s
->tio
, tio
, sizeof *s
->tio
);
140 s
->name
= xstrdup(name
);
141 s
->id
= next_session_id
++;
144 s
->id
= next_session_id
++;
147 xasprintf(&s
->name
, "%s-%u", prefix
, s
->id
);
149 xasprintf(&s
->name
, "%u", s
->id
);
150 } while (RB_FIND(sessions
, &sessions
, s
) != NULL
);
152 RB_INSERT(sessions
, &sessions
, s
);
154 log_debug("new session %s $%u", s
->name
, s
->id
);
156 if (gettimeofday(&s
->creation_time
, NULL
) != 0)
157 fatal("gettimeofday failed");
158 session_update_activity(s
, &s
->creation_time
);
163 /* Add a reference to a session. */
165 session_add_ref(struct session
*s
, const char *from
)
168 log_debug("%s: %s %s, now %d", __func__
, s
->name
, from
, s
->references
);
171 /* Remove a reference from a session. */
173 session_remove_ref(struct session
*s
, const char *from
)
176 log_debug("%s: %s %s, now %d", __func__
, s
->name
, from
, s
->references
);
178 if (s
->references
== 0)
179 event_once(-1, EV_TIMEOUT
, session_free
, s
, NULL
);
184 session_free(__unused
int fd
, __unused
short events
, void *arg
)
186 struct session
*s
= arg
;
188 log_debug("session %s freed (%d references)", s
->name
, s
->references
);
190 if (s
->references
== 0) {
191 environ_free(s
->environ
);
192 options_free(s
->options
);
199 /* Destroy a session. */
201 session_destroy(struct session
*s
, int notify
, const char *from
)
205 log_debug("session %s destroyed (%s)", s
->name
, from
);
211 RB_REMOVE(sessions
, &sessions
, s
);
213 notify_session("session-closed", s
);
217 if (event_initialized(&s
->lock_timer
))
218 event_del(&s
->lock_timer
);
220 session_group_remove(s
);
222 while (!TAILQ_EMPTY(&s
->lastw
))
223 winlink_stack_remove(&s
->lastw
, TAILQ_FIRST(&s
->lastw
));
224 while (!RB_EMPTY(&s
->windows
)) {
225 wl
= RB_ROOT(&s
->windows
);
226 notify_session_window("window-unlinked", s
, wl
->window
);
227 winlink_remove(&s
->windows
, wl
);
230 free((void *)s
->cwd
);
232 session_remove_ref(s
, __func__
);
235 /* Sanitize session name. */
237 session_check_name(const char *name
)
239 char *copy
, *cp
, *new_name
;
243 copy
= xstrdup(name
);
244 for (cp
= copy
; *cp
!= '\0'; cp
++) {
245 if (*cp
== ':' || *cp
== '.')
248 utf8_stravis(&new_name
, copy
, VIS_OCTAL
|VIS_CSTYLE
|VIS_TAB
|VIS_NL
);
253 /* Lock session if it has timed out. */
255 session_lock_timer(__unused
int fd
, __unused
short events
, void *arg
)
257 struct session
*s
= arg
;
259 if (s
->attached
== 0)
262 log_debug("session %s locked, activity time %lld", s
->name
,
263 (long long)s
->activity_time
.tv_sec
);
265 server_lock_session(s
);
269 /* Update activity time. */
271 session_update_activity(struct session
*s
, struct timeval
*from
)
273 struct timeval
*last
= &s
->last_activity_time
;
276 memcpy(last
, &s
->activity_time
, sizeof *last
);
278 gettimeofday(&s
->activity_time
, NULL
);
280 memcpy(&s
->activity_time
, from
, sizeof s
->activity_time
);
282 log_debug("session $%u %s activity %lld.%06d (last %lld.%06d)", s
->id
,
283 s
->name
, (long long)s
->activity_time
.tv_sec
,
284 (int)s
->activity_time
.tv_usec
, (long long)last
->tv_sec
,
287 if (evtimer_initialized(&s
->lock_timer
))
288 evtimer_del(&s
->lock_timer
);
290 evtimer_set(&s
->lock_timer
, session_lock_timer
, s
);
292 if (s
->attached
!= 0) {
294 tv
.tv_sec
= options_get_number(s
->options
, "lock-after-time");
296 evtimer_add(&s
->lock_timer
, &tv
);
300 /* Find the next usable session. */
302 session_next_session(struct session
*s
)
306 if (RB_EMPTY(&sessions
) || !session_alive(s
))
309 s2
= RB_NEXT(sessions
, &sessions
, s
);
311 s2
= RB_MIN(sessions
, &sessions
);
317 /* Find the previous usable session. */
319 session_previous_session(struct session
*s
)
323 if (RB_EMPTY(&sessions
) || !session_alive(s
))
326 s2
= RB_PREV(sessions
, &sessions
, s
);
328 s2
= RB_MAX(sessions
, &sessions
);
334 /* Attach a window to a session. */
336 session_attach(struct session
*s
, struct window
*w
, int idx
, char **cause
)
340 if ((wl
= winlink_add(&s
->windows
, idx
)) == NULL
) {
341 xasprintf(cause
, "index in use: %d", idx
);
345 winlink_set_window(wl
, w
);
346 notify_session_window("window-linked", s
, w
);
348 session_group_synchronize_from(s
);
352 /* Detach a window from a session. */
354 session_detach(struct session
*s
, struct winlink
*wl
)
357 session_last(s
) != 0 &&
358 session_previous(s
, 0) != 0)
361 wl
->flags
&= ~WINLINK_ALERTFLAGS
;
362 notify_session_window("window-unlinked", s
, wl
->window
);
363 winlink_stack_remove(&s
->lastw
, wl
);
364 winlink_remove(&s
->windows
, wl
);
366 session_group_synchronize_from(s
);
368 if (RB_EMPTY(&s
->windows
))
373 /* Return if session has window. */
375 session_has(struct session
*s
, struct window
*w
)
379 TAILQ_FOREACH(wl
, &w
->winlinks
, wentry
) {
380 if (wl
->session
== s
)
387 * Return 1 if a window is linked outside this session (not including session
388 * groups). The window must be in this session!
391 session_is_linked(struct session
*s
, struct window
*w
)
393 struct session_group
*sg
;
395 if ((sg
= session_group_contains(s
)) != NULL
)
396 return (w
->references
!= session_group_count(sg
));
397 return (w
->references
!= 1);
400 static struct winlink
*
401 session_next_alert(struct winlink
*wl
)
404 if (wl
->flags
& WINLINK_ALERTFLAGS
)
406 wl
= winlink_next(wl
);
411 /* Move session to next window. */
413 session_next(struct session
*s
, int alert
)
420 wl
= winlink_next(s
->curw
);
422 wl
= session_next_alert(wl
);
424 wl
= RB_MIN(winlinks
, &s
->windows
);
425 if (alert
&& ((wl
= session_next_alert(wl
)) == NULL
))
428 return (session_set_current(s
, wl
));
431 static struct winlink
*
432 session_previous_alert(struct winlink
*wl
)
435 if (wl
->flags
& WINLINK_ALERTFLAGS
)
437 wl
= winlink_previous(wl
);
442 /* Move session to previous window. */
444 session_previous(struct session
*s
, int alert
)
451 wl
= winlink_previous(s
->curw
);
453 wl
= session_previous_alert(wl
);
455 wl
= RB_MAX(winlinks
, &s
->windows
);
456 if (alert
&& (wl
= session_previous_alert(wl
)) == NULL
)
459 return (session_set_current(s
, wl
));
462 /* Move session to specific window. */
464 session_select(struct session
*s
, int idx
)
468 wl
= winlink_find_by_index(&s
->windows
, idx
);
469 return (session_set_current(s
, wl
));
472 /* Move session to last used window. */
474 session_last(struct session
*s
)
478 wl
= TAILQ_FIRST(&s
->lastw
);
484 return (session_set_current(s
, wl
));
487 /* Set current winlink to wl .*/
489 session_set_current(struct session
*s
, struct winlink
*wl
)
491 struct winlink
*old
= s
->curw
;
498 winlink_stack_remove(&s
->lastw
, wl
);
499 winlink_stack_push(&s
->lastw
, s
->curw
);
501 if (options_get_number(global_options
, "focus-events")) {
503 window_update_focus(old
->window
);
504 window_update_focus(wl
->window
);
506 winlink_clear_flags(wl
);
507 window_update_activity(wl
->window
);
508 tty_update_window_offset(wl
->window
);
509 notify_session("session-window-changed", s
);
513 /* Find the session group containing a session. */
514 struct session_group
*
515 session_group_contains(struct session
*target
)
517 struct session_group
*sg
;
520 RB_FOREACH(sg
, session_groups
, &session_groups
) {
521 TAILQ_FOREACH(s
, &sg
->sessions
, gentry
) {
529 /* Find session group by name. */
530 struct session_group
*
531 session_group_find(const char *name
)
533 struct session_group sg
;
536 return (RB_FIND(session_groups
, &session_groups
, &sg
));
539 /* Create a new session group. */
540 struct session_group
*
541 session_group_new(const char *name
)
543 struct session_group
*sg
;
545 if ((sg
= session_group_find(name
)) != NULL
)
548 sg
= xcalloc(1, sizeof *sg
);
549 sg
->name
= xstrdup(name
);
550 TAILQ_INIT(&sg
->sessions
);
552 RB_INSERT(session_groups
, &session_groups
, sg
);
556 /* Add a session to a session group. */
558 session_group_add(struct session_group
*sg
, struct session
*s
)
560 if (session_group_contains(s
) == NULL
)
561 TAILQ_INSERT_TAIL(&sg
->sessions
, s
, gentry
);
564 /* Remove a session from its group and destroy the group if empty. */
566 session_group_remove(struct session
*s
)
568 struct session_group
*sg
;
570 if ((sg
= session_group_contains(s
)) == NULL
)
572 TAILQ_REMOVE(&sg
->sessions
, s
, gentry
);
573 if (TAILQ_EMPTY(&sg
->sessions
)) {
574 RB_REMOVE(session_groups
, &session_groups
, sg
);
575 free((void *)sg
->name
);
580 /* Count number of sessions in session group. */
582 session_group_count(struct session_group
*sg
)
588 TAILQ_FOREACH(s
, &sg
->sessions
, gentry
)
593 /* Count number of clients attached to sessions in session group. */
595 session_group_attached_count(struct session_group
*sg
)
601 TAILQ_FOREACH(s
, &sg
->sessions
, gentry
)
606 /* Synchronize a session to its session group. */
608 session_group_synchronize_to(struct session
*s
)
610 struct session_group
*sg
;
611 struct session
*target
;
613 if ((sg
= session_group_contains(s
)) == NULL
)
617 TAILQ_FOREACH(target
, &sg
->sessions
, gentry
) {
622 session_group_synchronize1(target
, s
);
625 /* Synchronize a session group to a session. */
627 session_group_synchronize_from(struct session
*target
)
629 struct session_group
*sg
;
632 if ((sg
= session_group_contains(target
)) == NULL
)
635 TAILQ_FOREACH(s
, &sg
->sessions
, gentry
) {
637 session_group_synchronize1(target
, s
);
642 * Synchronize a session with a target session. This means destroying all
643 * winlinks then recreating them, then updating the current window, last window
647 session_group_synchronize1(struct session
*target
, struct session
*s
)
649 struct winlinks old_windows
, *ww
;
650 struct winlink_stack old_lastw
;
651 struct winlink
*wl
, *wl2
;
653 /* Don't do anything if the session is empty (it'll be destroyed). */
654 ww
= &target
->windows
;
658 /* If the current window has vanished, move to the next now. */
659 if (s
->curw
!= NULL
&&
660 winlink_find_by_index(ww
, s
->curw
->idx
) == NULL
&&
661 session_last(s
) != 0 && session_previous(s
, 0) != 0)
664 /* Save the old pointer and reset it. */
665 memcpy(&old_windows
, &s
->windows
, sizeof old_windows
);
666 RB_INIT(&s
->windows
);
668 /* Link all the windows from the target. */
669 RB_FOREACH(wl
, winlinks
, ww
) {
670 wl2
= winlink_add(&s
->windows
, wl
->idx
);
672 winlink_set_window(wl2
, wl
->window
);
673 notify_session_window("window-linked", s
, wl2
->window
);
674 wl2
->flags
|= wl
->flags
& WINLINK_ALERTFLAGS
;
677 /* Fix up the current window. */
679 s
->curw
= winlink_find_by_index(&s
->windows
, s
->curw
->idx
);
681 s
->curw
= winlink_find_by_index(&s
->windows
, target
->curw
->idx
);
683 /* Fix up the last window stack. */
684 memcpy(&old_lastw
, &s
->lastw
, sizeof old_lastw
);
685 TAILQ_INIT(&s
->lastw
);
686 TAILQ_FOREACH(wl
, &old_lastw
, sentry
) {
687 wl2
= winlink_find_by_index(&s
->windows
, wl
->idx
);
689 TAILQ_INSERT_TAIL(&s
->lastw
, wl2
, sentry
);
690 wl2
->flags
|= WINLINK_VISITED
;
694 /* Then free the old winlinks list. */
695 while (!RB_EMPTY(&old_windows
)) {
696 wl
= RB_ROOT(&old_windows
);
697 wl2
= winlink_find_by_window_id(&s
->windows
, wl
->window
->id
);
699 notify_session_window("window-unlinked", s
, wl
->window
);
700 winlink_remove(&old_windows
, wl
);
704 /* Renumber the windows across winlinks attached to a specific session. */
706 session_renumber_windows(struct session
*s
)
708 struct winlink
*wl
, *wl1
, *wl_new
;
709 struct winlinks old_wins
;
710 struct winlink_stack old_lastw
;
711 int new_idx
, new_curw_idx
, marked_idx
= -1;
713 /* Save and replace old window list. */
714 memcpy(&old_wins
, &s
->windows
, sizeof old_wins
);
715 RB_INIT(&s
->windows
);
717 /* Start renumbering from the base-index if it's set. */
718 new_idx
= options_get_number(s
->options
, "base-index");
721 /* Go through the winlinks and assign new indexes. */
722 RB_FOREACH(wl
, winlinks
, &old_wins
) {
723 wl_new
= winlink_add(&s
->windows
, new_idx
);
725 winlink_set_window(wl_new
, wl
->window
);
726 wl_new
->flags
|= wl
->flags
& WINLINK_ALERTFLAGS
;
728 if (wl
== marked_pane
.wl
)
729 marked_idx
= wl_new
->idx
;
731 new_curw_idx
= wl_new
->idx
;
736 /* Fix the stack of last windows now. */
737 memcpy(&old_lastw
, &s
->lastw
, sizeof old_lastw
);
738 TAILQ_INIT(&s
->lastw
);
739 TAILQ_FOREACH(wl
, &old_lastw
, sentry
) {
740 wl
->flags
&= ~WINLINK_VISITED
;
741 wl_new
= winlink_find_by_window(&s
->windows
, wl
->window
);
742 if (wl_new
!= NULL
) {
743 TAILQ_INSERT_TAIL(&s
->lastw
, wl_new
, sentry
);
744 wl_new
->flags
|= WINLINK_VISITED
;
748 /* Set the current window. */
749 if (marked_idx
!= -1) {
750 marked_pane
.wl
= winlink_find_by_index(&s
->windows
, marked_idx
);
751 if (marked_pane
.wl
== NULL
)
752 server_clear_marked();
754 s
->curw
= winlink_find_by_index(&s
->windows
, new_curw_idx
);
756 /* Free the old winlinks (reducing window references too). */
757 RB_FOREACH_SAFE(wl
, winlinks
, &old_wins
, wl1
)
758 winlink_remove(&old_wins
, wl
);