extern/tmate
1
0
Fork 0
mirror of https://github.com/tmate-io/tmate.git synced 2024-12-12 09:51:06 +01:00
Code Issues Packages Projects Releases Wiki Activity

Actually I thought cmd_get_state_client was unnecessary but it will be

Browse Source 
needed.
This commit is contained in:
nicm 2015-12-13 21:17:37 +00:00
parent 72948d9f1d
commit 899bee0056
1 changed files with 33 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
cmd.c
View File

@ -207,8 +207,9 @@ const struct cmd_entry *cmd_table[] = {
NULL NULL
}; };
static void cmd_clear_state(struct cmd_state *); static void cmd_clear_state(struct cmd_state *);
static int cmd_set_state_flag(struct cmd *, struct cmd_q *, char); static struct client *cmd_get_state_client(struct cmd_q *, int);
static int cmd_set_state_flag(struct cmd *, struct cmd_q *, char);
int int
cmd_pack_argv(int argc, char **argv, char *buf, size_t len) cmd_pack_argv(int argc, char **argv, char *buf, size_t len)
@ -579,11 +580,30 @@ complete_everything:
return (0); return (0);
} }
static struct client *
cmd_get_state_client(struct cmd_q *cmdq, int quiet)
{
struct cmd *cmd = cmdq->cmd;
struct args *args = cmd->args;
switch (cmd->entry->flags & (CMD_CLIENT_C|CMD_CLIENT_T)) {
case 0:
return (cmd_find_client(cmdq, NULL, 1));
case CMD_CLIENT_C:
return (cmd_find_client(cmdq, args_get(args, 'c'), quiet));
case CMD_CLIENT_T:
return (cmd_find_client(cmdq, args_get(args, 't'), quiet));
default:
fatalx("both -t and -c for %s", cmd->entry->name);
}
}
int int
cmd_prepare_state(struct cmd *cmd, struct cmd_q *cmdq) cmd_prepare_state(struct cmd *cmd, struct cmd_q *cmdq)
{ {
struct cmd_state *state = &cmdq->state; struct cmd_state *state = &cmdq->state;
struct args *args = cmd->args; struct args *args = cmd->args;
const char *cflag, *tflag;
char *tmp; char *tmp;
int error, quiet; int error, quiet;
@ -606,14 +626,23 @@ cmd_prepare_state(struct cmd *cmd, struct cmd_q *cmdq)
*/ */
switch (cmd->entry->flags & (CMD_CLIENT_C|CMD_CLIENT_T)) { switch (cmd->entry->flags & (CMD_CLIENT_C|CMD_CLIENT_T)) {
case 0: case 0:
state->c = cmd_find_client(cmdq, NULL, 1); state->c = cmd_get_state_client(cmdq, 1);
break; break;
case CMD_CLIENT_C: case CMD_CLIENT_C:
state->c = cmd_find_client(cmdq, args_get(args, 'c'), quiet); cflag = args_get(args, 'c');
if (cflag == NULL)
state->c = cmd_get_state_client(cmdq, quiet);
else
state->c = cmd_find_client(cmdq, cflag, quiet);
if (!quiet && state->c == NULL) if (!quiet && state->c == NULL)
return (-1); return (-1);
break; break;
case CMD_CLIENT_T: case CMD_CLIENT_T:
tflag = args_get(args, 't');
if (tflag == NULL)
state->c = cmd_get_state_client(cmdq, 0);
else
state->c = cmd_find_client(cmdq, tflag, 0);
state->c = cmd_find_client(cmdq, args_get(args, 't'), quiet); state->c = cmd_find_client(cmdq, args_get(args, 't'), quiet);
if (!quiet && state->c == NULL) if (!quiet && state->c == NULL)
return (-1); return (-1);