Annotation of src/usr.bin/tmux/cmd-capture-pane.c, Revision 1.56
1.56 ! nicm 1: /* $OpenBSD: cmd-capture-pane.c,v 1.55 2021/08/21 10:22:38 nicm Exp $ */
1.1 nicm 2:
3: /*
4: * Copyright (c) 2009 Jonathan Alvarado <radobobo@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:
1.4 nicm 21: #include <stdlib.h>
1.1 nicm 22: #include <string.h>
23:
24: #include "tmux.h"
25:
26: /*
1.14 nicm 27: * Write the entire contents of a pane to a buffer or stdout.
1.1 nicm 28: */
29:
1.41 nicm 30: static enum cmd_retval cmd_capture_pane_exec(struct cmd *, struct cmdq_item *);
1.1 nicm 31:
1.41 nicm 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 *);
1.23 nicm 37:
1.1 nicm 38: const struct cmd_entry cmd_capture_pane_entry = {
1.37 nicm 39: .name = "capture-pane",
40: .alias = "capturep",
41:
1.55 nicm 42: .args = { "ab:CeE:JNpPqS:t:", 0, 0, NULL },
1.48 nicm 43: .usage = "[-aCeJNpPq] " CMD_BUFFER_USAGE " [-E end-line] "
1.47 nicm 44: "[-S start-line] " CMD_TARGET_PANE_USAGE,
1.37 nicm 45:
1.44 nicm 46: .target = { 't', CMD_FIND_PANE, 0 },
1.38 nicm 47:
1.40 nicm 48: .flags = CMD_AFTERHOOK,
1.37 nicm 49: .exec = cmd_capture_pane_exec
1.1 nicm 50: };
51:
1.43 nicm 52: const struct cmd_entry cmd_clear_history_entry = {
53: .name = "clear-history",
54: .alias = "clearhist",
55:
1.55 nicm 56: .args = { "t:", 0, 0, NULL },
1.43 nicm 57: .usage = CMD_TARGET_PANE_USAGE,
58:
1.44 nicm 59: .target = { 't', CMD_FIND_PANE, 0 },
1.43 nicm 60:
61: .flags = CMD_AFTERHOOK,
62: .exec = cmd_capture_pane_exec
63: };
64:
1.39 nicm 65: static char *
1.23 nicm 66: cmd_capture_pane_append(char *buf, size_t *len, char *line, size_t linelen)
67: {
1.29 nicm 68: buf = xrealloc(buf, *len + linelen + 1);
1.23 nicm 69: memcpy(buf + *len, line, linelen);
70: *len += linelen;
71: return (buf);
72: }
73:
1.39 nicm 74: static char *
1.23 nicm 75: cmd_capture_pane_pending(struct args *args, struct window_pane *wp,
76: size_t *len)
77: {
1.32 nicm 78: struct evbuffer *pending;
79: char *buf, *line, tmp[5];
80: size_t linelen;
81: u_int i;
1.23 nicm 82:
1.50 nicm 83: pending = input_pending(wp->ictx);
1.32 nicm 84: if (pending == NULL)
1.23 nicm 85: return (xstrdup(""));
86:
1.32 nicm 87: line = EVBUFFER_DATA(pending);
88: linelen = EVBUFFER_LENGTH(pending);
1.23 nicm 89:
1.24 nicm 90: buf = xstrdup("");
1.23 nicm 91: if (args_has(args, 'C')) {
92: for (i = 0; i < linelen; i++) {
1.42 nicm 93: if (line[i] >= ' ' && line[i] != '\\') {
1.23 nicm 94: tmp[0] = line[i];
95: tmp[1] = '\0';
96: } else
1.31 nicm 97: xsnprintf(tmp, sizeof tmp, "\\%03hho", line[i]);
1.23 nicm 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:
1.39 nicm 106: static char *
1.41 nicm 107: cmd_capture_pane_history(struct args *args, struct cmdq_item *item,
1.23 nicm 108: struct window_pane *wp, size_t *len)
1.1 nicm 109: {
1.7 nicm 110: struct grid *gd;
1.18 nicm 111: const struct grid_line *gl;
1.23 nicm 112: struct grid_cell *gc = NULL;
1.48 nicm 113: int n, with_codes, escape_c0, join_lines, no_trim;
1.23 nicm 114: u_int i, sx, top, bottom, tmp;
115: char *cause, *buf, *line;
1.28 nicm 116: const char *Sflag, *Eflag;
1.23 nicm 117: size_t linelen;
1.14 nicm 118:
1.23 nicm 119: sx = screen_size_x(&wp->base);
1.20 nicm 120: if (args_has(args, 'a')) {
1.51 nicm 121: gd = wp->base.saved_grid;
1.23 nicm 122: if (gd == NULL) {
123: if (!args_has(args, 'q')) {
1.41 nicm 124: cmdq_error(item, "no alternate screen");
1.23 nicm 125: return (NULL);
126: }
127: return (xstrdup(""));
1.20 nicm 128: }
1.23 nicm 129: } else
130: gd = wp->base.grid;
131:
1.28 nicm 132: Sflag = args_get(args, 'S');
133: if (Sflag != NULL && strcmp(Sflag, "-") == 0)
1.23 nicm 134: top = 0;
1.28 nicm 135: else {
1.56 ! nicm 136: n = args_strtonum_and_expand(args, 'S', INT_MIN, SHRT_MAX,
! 137: item, &cause);
1.28 nicm 138: if (cause != NULL) {
139: top = gd->hsize;
140: free(cause);
141: } else if (n < 0 && (u_int) -n > gd->hsize)
142: top = 0;
143: else
144: top = gd->hsize + n;
145: if (top > gd->hsize + gd->sy - 1)
146: top = gd->hsize + gd->sy - 1;
147: }
1.23 nicm 148:
1.28 nicm 149: Eflag = args_get(args, 'E');
150: if (Eflag != NULL && strcmp(Eflag, "-") == 0)
1.23 nicm 151: bottom = gd->hsize + gd->sy - 1;
1.28 nicm 152: else {
1.56 ! nicm 153: n = args_strtonum_and_expand(args, 'E', INT_MIN, SHRT_MAX,
! 154: item, &cause);
1.28 nicm 155: if (cause != NULL) {
156: bottom = gd->hsize + gd->sy - 1;
157: free(cause);
158: } else if (n < 0 && (u_int) -n > gd->hsize)
159: bottom = 0;
160: else
161: bottom = gd->hsize + n;
162: if (bottom > gd->hsize + gd->sy - 1)
163: bottom = gd->hsize + gd->sy - 1;
164: }
1.23 nicm 165:
166: if (bottom < top) {
167: tmp = bottom;
168: bottom = top;
169: top = tmp;
1.20 nicm 170: }
1.1 nicm 171:
1.23 nicm 172: with_codes = args_has(args, 'e');
173: escape_c0 = args_has(args, 'C');
174: join_lines = args_has(args, 'J');
1.48 nicm 175: no_trim = args_has(args, 'N');
1.23 nicm 176:
1.1 nicm 177: buf = NULL;
1.23 nicm 178: for (i = top; i <= bottom; i++) {
179: line = grid_string_cells(gd, 0, i, sx, &gc, with_codes,
1.48 nicm 180: escape_c0, !join_lines && !no_trim);
1.23 nicm 181: linelen = strlen(line);
182:
183: buf = cmd_capture_pane_append(buf, len, line, linelen);
184:
185: gl = grid_peek_line(gd, i);
186: if (!join_lines || !(gl->flags & GRID_LINE_WRAPPED))
187: buf[(*len)++] = '\n';
188:
189: free(line);
190: }
191: return (buf);
192: }
1.1 nicm 193:
1.39 nicm 194: static enum cmd_retval
1.41 nicm 195: cmd_capture_pane_exec(struct cmd *self, struct cmdq_item *item)
1.23 nicm 196: {
1.52 nicm 197: struct args *args = cmd_get_args(self);
1.53 nicm 198: struct client *c = cmdq_get_client(item);
199: struct window_pane *wp = cmdq_get_target(item)->wp;
1.23 nicm 200: char *buf, *cause;
1.27 nicm 201: const char *bufname;
1.23 nicm 202: size_t len;
1.43 nicm 203:
1.52 nicm 204: if (cmd_get_entry(self) == &cmd_clear_history_entry) {
1.46 nicm 205: window_pane_reset_mode_all(wp);
1.43 nicm 206: grid_clear_history(wp->base.grid);
207: return (CMD_RETURN_NORMAL);
208: }
1.1 nicm 209:
1.23 nicm 210: len = 0;
211: if (args_has(args, 'P'))
212: buf = cmd_capture_pane_pending(args, wp, &len);
213: else
1.41 nicm 214: buf = cmd_capture_pane_history(args, item, wp, &len);
1.23 nicm 215: if (buf == NULL)
216: return (CMD_RETURN_ERROR);
1.1 nicm 217:
1.14 nicm 218: if (args_has(args, 'p')) {
1.54 nicm 219: if (len > 0 && buf[len - 1] == '\n')
220: len--;
221: if (c->flags & CLIENT_CONTROL)
222: control_write(c, "%.*s", (int)len, buf);
223: else {
224: if (!file_can_print(c)) {
225: cmdq_error(item, "can't write to client");
226: free(buf);
227: return (CMD_RETURN_ERROR);
228: }
229: file_print_buffer(c, buf, len);
230: file_print(c, "\n");
1.33 nicm 231: free(buf);
1.14 nicm 232: }
233: } else {
1.27 nicm 234: bufname = NULL;
235: if (args_has(args, 'b'))
236: bufname = args_get(args, 'b');
237:
238: if (paste_set(buf, len, bufname, &cause) != 0) {
1.41 nicm 239: cmdq_error(item, "%s", cause);
1.33 nicm 240: free(cause);
1.14 nicm 241: free(buf);
242: return (CMD_RETURN_ERROR);
243: }
1.1 nicm 244: }
1.6 nicm 245:
1.12 nicm 246: return (CMD_RETURN_NORMAL);
1.1 nicm 247: }