Fix a crash when completing sessions, from Anindya Mukherjee.

1 files changed, 17 insertions, 6 deletions

diff --git a/status.c b/status.c
index d143b658..7313d2a0 100644
--- a/status.c
+++ b/status.c
@@ -1441,6 +1441,8 @@ status_prompt_complete_prefix(char **list, u_int size)
 	u_int	  i;
 	size_t	  j;
 
+	if (list == NULL || size == 0)
+		return (NULL);
 	out = xstrdup(list[0]);
 	for (i = 1; i < size; i++) {
 		j = strlen(list[i]);
@@ -1649,13 +1651,22 @@ status_prompt_complete_session(char ***list, u_int *size, const char *s,
     char flag)
 {
 	struct session	*loop;
-	char		*out, *tmp;
+	char		*out, *tmp, n[11];
 
 	RB_FOREACH(loop, sessions, &sessions) {
-		if (*s != '\0' && strncmp(loop->name, s, strlen(s)) != 0)
-			continue;
-		*list = xreallocarray(*list, (*size) + 2, sizeof **list);
-		xasprintf(&(*list)[(*size)++], "%s:", loop->name);
+		if (*s == '\0' || strncmp(loop->name, s, strlen(s)) == 0) {
+			*list = xreallocarray(*list, (*size) + 2,
+			    sizeof **list);
+			xasprintf(&(*list)[(*size)++], "%s:", loop->name);
+		} else if (*s == '$') {
+			xsnprintf(n, sizeof n, "%u", loop->id);
+			if (s[1] == '\0' ||
+			    strncmp(n, s + 1, strlen(s) - 1) == 0) {
+				*list = xreallocarray(*list, (*size) + 2,
+				    sizeof **list);
+				xasprintf(&(*list)[(*size)++], "$%s:", n);
+			}
+		}
 	}
 	out = status_prompt_complete_prefix(*list, *size);
 	if (out != NULL && flag != '\0') {
@@ -1670,7 +1681,7 @@ status_prompt_complete_session(char ***list, u_int *size, const char *s,
 static char *
 status_prompt_complete(struct client *c, const char *word, u_int offset)
 {
-	struct session   *session;
+	struct session	 *session;
 	const char	 *s, *colon;
 	char		**list = NULL, *copy = NULL, *out = NULL;
 	char		  flag = '\0';