1 /* $OpenBSD$ */ 2 3 /* 4 * Copyright (c) 2009 Jonathan Alvarado <radobobo (at) users.sourceforge.net> 5 * 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. 9 * 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. 17 */ 18 19 #include <sys/types.h> 20 21 #include <stdlib.h> 22 #include <string.h> 23 24 #include "tmux.h" 25 26 /* 27 * Write the entire contents of a pane to a buffer or stdout. 28 */ 29 30 static enum cmd_retval cmd_capture_pane_exec(struct cmd *, struct cmdq_item *); 31 32 static char *cmd_capture_pane_append(char *, size_t *, char *, size_t); 33 static char *cmd_capture_pane_pending(struct args *, struct window_pane *, 34 size_t *); 35 static char *cmd_capture_pane_history(struct args *, struct cmdq_item *, 36 struct window_pane *, size_t *); 37 38 const struct cmd_entry cmd_capture_pane_entry = { 39 .name = "capture-pane", 40 .alias = "capturep", 41 42 .args = { "ab:CeE:JMNpPqS:Tt:", 0, 0, NULL }, 43 .usage = "[-aCeJMNpPqT] " CMD_BUFFER_USAGE " [-E end-line] " 44 "[-S start-line] " CMD_TARGET_PANE_USAGE, 45 46 .target = { 't', CMD_FIND_PANE, 0 }, 47 48 .flags = CMD_AFTERHOOK, 49 .exec = cmd_capture_pane_exec 50 }; 51 52 const struct cmd_entry cmd_clear_history_entry = { 53 .name = "clear-history", 54 .alias = "clearhist", 55 56 .args = { "Ht:", 0, 0, NULL }, 57 .usage = "[-H] " CMD_TARGET_PANE_USAGE, 58 59 .target = { 't', CMD_FIND_PANE, 0 }, 60 61 .flags = CMD_AFTERHOOK, 62 .exec = cmd_capture_pane_exec 63 }; 64 65 static char * 66 cmd_capture_pane_append(char *buf, size_t *len, char *line, size_t linelen) 67 { 68 buf = xrealloc(buf, *len + linelen + 1); 69 memcpy(buf + *len, line, linelen); 70 *len += linelen; 71 return (buf); 72 } 73 74 static char * 75 cmd_capture_pane_pending(struct args *args, struct window_pane *wp, 76 size_t *len) 77 { 78 struct evbuffer *pending; 79 char *buf, *line, tmp[5]; 80 size_t linelen; 81 u_int i; 82 83 pending = input_pending(wp->ictx); 84 if (pending == NULL) 85 return (xstrdup("")); 86 87 line = (char *)EVBUFFER_DATA(pending); 88 linelen = EVBUFFER_LENGTH(pending); 89 90 buf = xstrdup(""); 91 if (args_has(args, 'C')) { 92 for (i = 0; i < linelen; i++) { 93 if (line[i] >= ' ' && line[i] != '\\') { 94 tmp[0] = line[i]; 95 tmp[1] = '\0'; 96 } else 97 xsnprintf(tmp, sizeof tmp, "\\%03hho", line[i]); 98 buf = cmd_capture_pane_append(buf, len, tmp, 99 strlen(tmp)); 100 } 101 } else 102 buf = cmd_capture_pane_append(buf, len, line, linelen); 103 return (buf); 104 } 105 106 static char * 107 cmd_capture_pane_history(struct args *args, struct cmdq_item *item, 108 struct window_pane *wp, size_t *len) 109 { 110 struct grid *gd; 111 const struct grid_line *gl; 112 struct screen *s; 113 struct grid_cell *gc = NULL; 114 struct window_mode_entry *wme; 115 int n, join_lines, flags = 0; 116 u_int i, sx, top, bottom, tmp; 117 char *cause, *buf, *line; 118 const char *Sflag, *Eflag; 119 size_t linelen; 120 121 sx = screen_size_x(&wp->base); 122 if (args_has(args, 'a')) { 123 gd = wp->base.saved_grid; 124 if (gd == NULL) { 125 if (!args_has(args, 'q')) { 126 cmdq_error(item, "no alternate screen"); 127 return (NULL); 128 } 129 return (xstrdup("")); 130 } 131 s = &wp->base; 132 } else if (args_has(args, 'M')) { 133 wme = TAILQ_FIRST(&wp->modes); 134 if (wme != NULL && wme->mode->get_screen != NULL) { 135 s = wme->mode->get_screen (wme); 136 gd = s->grid; 137 } else { 138 s = &wp->base; 139 gd = wp->base.grid; 140 } 141 } else { 142 s = &wp->base; 143 gd = wp->base.grid; 144 } 145 146 Sflag = args_get(args, 'S'); 147 if (Sflag != NULL && strcmp(Sflag, "-") == 0) 148 top = 0; 149 else { 150 n = args_strtonum_and_expand(args, 'S', INT_MIN, SHRT_MAX, 151 item, &cause); 152 if (cause != NULL) { 153 top = gd->hsize; 154 free(cause); 155 } else if (n < 0 && (u_int) -n > gd->hsize) 156 top = 0; 157 else 158 top = gd->hsize + n; 159 if (top > gd->hsize + gd->sy - 1) 160 top = gd->hsize + gd->sy - 1; 161 } 162 163 Eflag = args_get(args, 'E'); 164 if (Eflag != NULL && strcmp(Eflag, "-") == 0) 165 bottom = gd->hsize + gd->sy - 1; 166 else { 167 n = args_strtonum_and_expand(args, 'E', INT_MIN, SHRT_MAX, 168 item, &cause); 169 if (cause != NULL) { 170 bottom = gd->hsize + gd->sy - 1; 171 free(cause); 172 } else if (n < 0 && (u_int) -n > gd->hsize) 173 bottom = 0; 174 else 175 bottom = gd->hsize + n; 176 if (bottom > gd->hsize + gd->sy - 1) 177 bottom = gd->hsize + gd->sy - 1; 178 } 179 180 if (bottom < top) { 181 tmp = bottom; 182 bottom = top; 183 top = tmp; 184 } 185 186 join_lines = args_has(args, 'J'); 187 if (args_has(args, 'e')) 188 flags |= GRID_STRING_WITH_SEQUENCES; 189 if (args_has(args, 'C')) 190 flags |= GRID_STRING_ESCAPE_SEQUENCES; 191 if (!join_lines && !args_has(args, 'T')) 192 flags |= GRID_STRING_EMPTY_CELLS; 193 if (!join_lines && !args_has(args, 'N')) 194 flags |= GRID_STRING_TRIM_SPACES; 195 196 buf = NULL; 197 for (i = top; i <= bottom; i++) { 198 line = grid_string_cells(gd, 0, i, sx, &gc, flags, s); 199 linelen = strlen(line); 200 201 buf = cmd_capture_pane_append(buf, len, line, linelen); 202 203 gl = grid_peek_line(gd, i); 204 if (!join_lines || !(gl->flags & GRID_LINE_WRAPPED)) 205 buf[(*len)++] = '\n'; 206 207 free(line); 208 } 209 return (buf); 210 } 211 212 static enum cmd_retval 213 cmd_capture_pane_exec(struct cmd *self, struct cmdq_item *item) 214 { 215 struct args *args = cmd_get_args(self); 216 struct client *c = cmdq_get_client(item); 217 struct window_pane *wp = cmdq_get_target(item)->wp; 218 char *buf, *cause; 219 const char *bufname; 220 size_t len; 221 222 if (cmd_get_entry(self) == &cmd_clear_history_entry) { 223 window_pane_reset_mode_all(wp); 224 grid_clear_history(wp->base.grid); 225 if (args_has(args, 'H')) 226 screen_reset_hyperlinks(wp->screen); 227 return (CMD_RETURN_NORMAL); 228 } 229 230 len = 0; 231 if (args_has(args, 'P')) 232 buf = cmd_capture_pane_pending(args, wp, &len); 233 else 234 buf = cmd_capture_pane_history(args, item, wp, &len); 235 if (buf == NULL) 236 return (CMD_RETURN_ERROR); 237 238 if (args_has(args, 'p')) { 239 if (len > 0 && buf[len - 1] == '\n') 240 len--; 241 if (c->flags & CLIENT_CONTROL) 242 control_write(c, "%.*s", (int)len, buf); 243 else { 244 if (!file_can_print(c)) { 245 cmdq_error(item, "can't write to client"); 246 free(buf); 247 return (CMD_RETURN_ERROR); 248 } 249 file_print_buffer(c, buf, len); 250 file_print(c, "\n"); 251 } 252 free(buf); 253 } else { 254 bufname = NULL; 255 if (args_has(args, 'b')) 256 bufname = args_get(args, 'b'); 257 258 if (paste_set(buf, len, bufname, &cause) != 0) { 259 cmdq_error(item, "%s", cause); 260 free(cause); 261 free(buf); 262 return (CMD_RETURN_ERROR); 263 } 264 } 265 266 return (CMD_RETURN_NORMAL); 267 } 268
Indexes created Tue Aug 04 00:24:20 UTC 2026