1 |
|
|
/* $OpenBSD: cmd-run-shell.c,v 1.51 2017/08/30 10:33:57 nicm Exp $ */ |
2 |
|
|
|
3 |
|
|
/* |
4 |
|
|
* Copyright (c) 2009 Tiago Cunha <me@tiagocunha.org> |
5 |
|
|
* Copyright (c) 2009 Nicholas Marriott <nicm@openbsd.org> |
6 |
|
|
* |
7 |
|
|
* Permission to use, copy, modify, and distribute this software for any |
8 |
|
|
* purpose with or without fee is hereby granted, provided that the above |
9 |
|
|
* copyright notice and this permission notice appear in all copies. |
10 |
|
|
* |
11 |
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES |
12 |
|
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF |
13 |
|
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR |
14 |
|
|
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES |
15 |
|
|
* WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER |
16 |
|
|
* IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING |
17 |
|
|
* OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. |
18 |
|
|
*/ |
19 |
|
|
|
20 |
|
|
#include <sys/types.h> |
21 |
|
|
#include <sys/wait.h> |
22 |
|
|
|
23 |
|
|
#include <stdlib.h> |
24 |
|
|
#include <string.h> |
25 |
|
|
|
26 |
|
|
#include "tmux.h" |
27 |
|
|
|
28 |
|
|
/* |
29 |
|
|
* Runs a command without a window. |
30 |
|
|
*/ |
31 |
|
|
|
32 |
|
|
static enum cmd_retval cmd_run_shell_exec(struct cmd *, struct cmdq_item *); |
33 |
|
|
|
34 |
|
|
static void cmd_run_shell_callback(struct job *); |
35 |
|
|
static void cmd_run_shell_free(void *); |
36 |
|
|
static void cmd_run_shell_print(struct job *, const char *); |
37 |
|
|
|
38 |
|
|
const struct cmd_entry cmd_run_shell_entry = { |
39 |
|
|
.name = "run-shell", |
40 |
|
|
.alias = "run", |
41 |
|
|
|
42 |
|
|
.args = { "bt:", 1, 1 }, |
43 |
|
|
.usage = "[-b] " CMD_TARGET_PANE_USAGE " shell-command", |
44 |
|
|
|
45 |
|
|
.target = { 't', CMD_FIND_PANE, CMD_FIND_CANFAIL }, |
46 |
|
|
|
47 |
|
|
.flags = 0, |
48 |
|
|
.exec = cmd_run_shell_exec |
49 |
|
|
}; |
50 |
|
|
|
51 |
|
|
struct cmd_run_shell_data { |
52 |
|
|
char *cmd; |
53 |
|
|
struct cmdq_item *item; |
54 |
|
|
int wp_id; |
55 |
|
|
}; |
56 |
|
|
|
57 |
|
|
static void |
58 |
|
|
cmd_run_shell_print(struct job *job, const char *msg) |
59 |
|
|
{ |
60 |
|
|
struct cmd_run_shell_data *cdata = job->data; |
61 |
|
|
struct window_pane *wp = NULL; |
62 |
|
|
struct cmd_find_state fs; |
63 |
|
|
|
64 |
|
|
if (cdata->wp_id != -1) |
65 |
|
|
wp = window_pane_find_by_id(cdata->wp_id); |
66 |
|
|
if (wp == NULL) { |
67 |
|
|
if (cdata->item != NULL) { |
68 |
|
|
cmdq_print(cdata->item, "%s", msg); |
69 |
|
|
return; |
70 |
|
|
} |
71 |
|
|
if (cmd_find_from_nothing(&fs, 0) != 0) |
72 |
|
|
return; |
73 |
|
|
wp = fs.wp; |
74 |
|
|
if (wp == NULL) |
75 |
|
|
return; |
76 |
|
|
} |
77 |
|
|
|
78 |
|
|
if (window_pane_set_mode(wp, &window_copy_mode, NULL, NULL) == 0) |
79 |
|
|
window_copy_init_for_output(wp); |
80 |
|
|
if (wp->mode == &window_copy_mode) |
81 |
|
|
window_copy_add(wp, "%s", msg); |
82 |
|
|
} |
83 |
|
|
|
84 |
|
|
static enum cmd_retval |
85 |
|
|
cmd_run_shell_exec(struct cmd *self, struct cmdq_item *item) |
86 |
|
|
{ |
87 |
|
|
struct args *args = self->args; |
88 |
|
|
struct cmd_run_shell_data *cdata; |
89 |
|
|
struct client *c = cmd_find_client(item, NULL, 1); |
90 |
|
|
struct session *s = item->target.s; |
91 |
|
|
struct winlink *wl = item->target.wl; |
92 |
|
|
struct window_pane *wp = item->target.wp; |
93 |
|
|
const char *cwd; |
94 |
|
|
|
95 |
|
|
if (item->client != NULL && item->client->session == NULL) |
96 |
|
|
cwd = item->client->cwd; |
97 |
|
|
else if (s != NULL) |
98 |
|
|
cwd = s->cwd; |
99 |
|
|
else |
100 |
|
|
cwd = NULL; |
101 |
|
|
|
102 |
|
|
cdata = xcalloc(1, sizeof *cdata); |
103 |
|
|
cdata->cmd = format_single(item, args->argv[0], c, s, wl, wp); |
104 |
|
|
|
105 |
|
|
if (args_has(args, 't') && wp != NULL) |
106 |
|
|
cdata->wp_id = wp->id; |
107 |
|
|
else |
108 |
|
|
cdata->wp_id = -1; |
109 |
|
|
|
110 |
|
|
if (!args_has(args, 'b')) |
111 |
|
|
cdata->item = item; |
112 |
|
|
|
113 |
|
|
job_run(cdata->cmd, s, cwd, NULL, cmd_run_shell_callback, |
114 |
|
|
cmd_run_shell_free, cdata); |
115 |
|
|
|
116 |
|
|
if (args_has(args, 'b')) |
117 |
|
|
return (CMD_RETURN_NORMAL); |
118 |
|
|
return (CMD_RETURN_WAIT); |
119 |
|
|
} |
120 |
|
|
|
121 |
|
|
static void |
122 |
|
|
cmd_run_shell_callback(struct job *job) |
123 |
|
|
{ |
124 |
|
|
struct cmd_run_shell_data *cdata = job->data; |
125 |
|
|
char *cmd = cdata->cmd, *msg, *line; |
126 |
|
|
size_t size; |
127 |
|
|
int retcode; |
128 |
|
|
|
129 |
|
|
do { |
130 |
|
|
if ((line = evbuffer_readline(job->event->input)) != NULL) { |
131 |
|
|
cmd_run_shell_print(job, line); |
132 |
|
|
free(line); |
133 |
|
|
} |
134 |
|
|
} while (line != NULL); |
135 |
|
|
|
136 |
|
|
size = EVBUFFER_LENGTH(job->event->input); |
137 |
|
|
if (size != 0) { |
138 |
|
|
line = xmalloc(size + 1); |
139 |
|
|
memcpy(line, EVBUFFER_DATA(job->event->input), size); |
140 |
|
|
line[size] = '\0'; |
141 |
|
|
|
142 |
|
|
cmd_run_shell_print(job, line); |
143 |
|
|
|
144 |
|
|
free(line); |
145 |
|
|
} |
146 |
|
|
|
147 |
|
|
msg = NULL; |
148 |
|
|
if (WIFEXITED(job->status)) { |
149 |
|
|
if ((retcode = WEXITSTATUS(job->status)) != 0) |
150 |
|
|
xasprintf(&msg, "'%s' returned %d", cmd, retcode); |
151 |
|
|
} else if (WIFSIGNALED(job->status)) { |
152 |
|
|
retcode = WTERMSIG(job->status); |
153 |
|
|
xasprintf(&msg, "'%s' terminated by signal %d", cmd, retcode); |
154 |
|
|
} |
155 |
|
|
if (msg != NULL) |
156 |
|
|
cmd_run_shell_print(job, msg); |
157 |
|
|
free(msg); |
158 |
|
|
|
159 |
|
|
if (cdata->item != NULL) |
160 |
|
|
cdata->item->flags &= ~CMDQ_WAITING; |
161 |
|
|
} |
162 |
|
|
|
163 |
|
|
static void |
164 |
|
|
cmd_run_shell_free(void *data) |
165 |
|
|
{ |
166 |
|
|
struct cmd_run_shell_data *cdata = data; |
167 |
|
|
|
168 |
|
|
free(cdata->cmd); |
169 |
|
|
free(cdata); |
170 |
|
|
} |