tmate-io_tmate-ssh-server/tmate-daemon-encoder.c

#include "tmate.h"
#include "tmate-protocol.h"

#define pack(what, ...) _pack(&tmate_session->daemon_encoder, what, ##__VA_ARGS__)

static void __tmate_notify(const char *msg)
{
	pack(array, 2);
	pack(int, TMATE_IN_NOTIFY);
	pack(string, msg);
}

void tmate_notify(const char *fmt, ...)
{
	va_list ap;
	char msg[1024];

	va_start(ap, fmt);
	vsnprintf(msg, sizeof(msg), fmt, ap);
	va_end(ap);

	__tmate_notify(msg);
}

static void __tmate_notify_later(__unused evutil_socket_t fd,
				 __unused short what, void *arg)
{
	char *msg = arg;
	__tmate_notify(msg);
}

void tmate_notify_later(int timeout, const char *fmt, ...)
{
	struct timeval tv;
	va_list ap;
	char *msg;

	tv.tv_sec = timeout;
	tv.tv_usec = 0;

	va_start(ap, fmt);
	xvasprintf(&msg, fmt, ap);
	va_end(ap);

	/*
	 * FIXME leaks when calling tmate_notify_later()
	 * multiple times.
	 */

	evtimer_set(&tmate_session->ev_notify_timer,
		    __tmate_notify_later, msg);
	evtimer_add(&tmate_session->ev_notify_timer, &tv);
}

void tmate_send_client_ready(void)
{
	if (tmate_session->client_protocol_version < 4)
		return;

	pack(array, 1);
	pack(int, TMATE_IN_READY);
}

void tmate_set_env(const char *name, const char *value)
{
	if (tmate_session->client_protocol_version < 4)
		return;

	pack(array, 3);
	pack(int, TMATE_IN_SET_ENV);
	pack(string, name);
	pack(string, value);
}

void tmate_client_resize(u_int sx, u_int sy)
{
	pack(array, 3);
	pack(int, TMATE_IN_RESIZE);
	/* cast to signed, -1 == no clients */
	pack(int, sx);
	pack(int, sy);
}

void tmate_client_legacy_pane_key(__unused int pane_id, int key)
{
	/*
	 * We don't specify the pane id because the current active pane is
	 * behind, so we'll let master send the key to its active pane.
	 */

	pack(array, 2);
	pack(int, TMATE_IN_LEGACY_PANE_KEY);
	pack(int, key);
}

void tmate_client_pane_key(int pane_id, key_code key)
{
	if (key == KEYC_NONE || key == KEYC_UNKNOWN)
		return;

	/* Mouse keys not supported yet */
	if (KEYC_IS_MOUSE(key))
		return;

	if (tmate_session->client_protocol_version < 5) {
		tmate_translate_legacy_key(pane_id, key);
		return;
	}

	if (tmate_session->client_protocol_version == 5 && key & KEYC_BASE) {
		if ((key & KEYC_MASK_KEY) >= (KEYC_BSPACE & KEYC_MASK_KEY))
			key -= 9;
	}

	pack(array, 3);
	pack(int, TMATE_IN_PANE_KEY);
	pack(int, pane_id);
	pack(uint64, key);
}

extern const struct cmd_entry cmd_bind_key_entry;
extern const struct cmd_entry cmd_unbind_key_entry;
extern const struct cmd_entry cmd_set_option_entry;
extern const struct cmd_entry cmd_set_window_option_entry;
extern const struct cmd_entry cmd_detach_client_entry;
extern const struct cmd_entry cmd_attach_session_entry;

static const struct cmd_entry *local_cmds[] = {
	&cmd_bind_key_entry,
	&cmd_unbind_key_entry,
	&cmd_set_option_entry,
	&cmd_set_window_option_entry,
	&cmd_detach_client_entry,
	&cmd_attach_session_entry,
	NULL
};

int tmate_should_exec_cmd_locally(const struct cmd_entry *cmd)
{
	const struct cmd_entry **ptr;

	for (ptr = local_cmds; *ptr; ptr++)
		if (*ptr == cmd)
			return 1;
	return 0;
}

void tmate_client_cmd_str(int client_id, const char *cmd)
{
	tmate_debug("Remote cmd (cid=%d): %s", client_id, cmd);

	pack(array, 3);
	pack(int, TMATE_IN_EXEC_CMD_STR);
	pack(int, client_id);
	pack(string, cmd);
}

struct args_entry {
	u_char			 flag;
	char			*value;
	RB_ENTRY(args_entry)	 entry;
};

static void extract_cmd(struct cmd *cmd, int *_argc, char ***_argv)
{
	struct args_entry *entry;
	struct args* args = cmd->args;
	int argc = 0;
	char **argv;
	int next = 0, i;

	argc++; /* cmd name */
	RB_FOREACH(entry, args_tree, &args->tree) {
		argc++;
		if (entry->value != NULL)
			argc++;
	}
	argc += args->argc;
	argv = xmalloc(sizeof(char *) * argc);

	argv[next++] = xstrdup(cmd->entry->name);

	RB_FOREACH(entry, args_tree, &args->tree) {
		xasprintf(&argv[next++], "-%c", entry->flag);
		if (entry->value != NULL)
			argv[next++] = xstrdup(entry->value);
	}

	for (i = 0; i < args->argc; i++)
		argv[next++] = xstrdup(args->argv[i]);

	*_argc = argc;
	*_argv = argv;
}

void tmate_client_cmd_args(int client_id, int argc, const char **argv)
{
	int i;

	pack(array, argc + 2);
	pack(int, TMATE_IN_EXEC_CMD);
	pack(int, client_id);

	for (i = 0; i < argc; i++)
		pack(string, argv[i]);
}

void tmate_client_cmd(int client_id, struct cmd *cmd)
{
	char *cmd_str;
	int argc;
	char **argv;

	cmd_str = cmd_print(cmd);
	if (tmate_session->client_protocol_version < 6) {
		tmate_client_cmd_str(client_id, cmd_str);
		free(cmd_str);
		return;
	}
	tmate_debug("Remote cmd (cid=%d): %s", client_id, cmd_str);
	free(cmd_str);

	extract_cmd(cmd, &argc, &argv);
	tmate_client_cmd_args(client_id, argc, (const char **)argv);
	cmd_free_argv(argc, argv);
}

void tmate_client_set_active_pane(int client_id, int win_idx, int pane_id)
{
	char target[64];
	sprintf(target, "%d.%d", win_idx, pane_id);
	tmate_client_cmd_args(client_id, 3, (const char *[]){"select-pane", "-t", target});
}

void tmate_send_mc_obj(msgpack_object *obj)
{
	pack(object, *obj);
}