Annotation of src/usr.bin/tmux/cmd-capture-pane.c, Revision 1.36
1.36 ! nicm 1: /* $OpenBSD: cmd-capture-pane.c,v 1.35 2015/11/14 09:41:06 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.19 nicm 30: enum cmd_retval cmd_capture_pane_exec(struct cmd *, struct cmd_q *);
1.1 nicm 31:
1.23 nicm 32: char *cmd_capture_pane_append(char *, size_t *, char *, size_t);
33: char *cmd_capture_pane_pending(struct args *, struct window_pane *,
34: size_t *);
35: char *cmd_capture_pane_history(struct args *, struct cmd_q *,
36: struct window_pane *, size_t *);
37:
1.1 nicm 38: const struct cmd_entry cmd_capture_pane_entry = {
39: "capture-pane", "capturep",
1.23 nicm 40: "ab:CeE:JpPqS:t:", 0, 0,
1.27 nicm 41: "[-aCeJpPq] " CMD_BUFFER_USAGE " [-E end-line] [-S start-line]"
1.13 nicm 42: CMD_TARGET_PANE_USAGE,
1.36 ! nicm 43: CMD_PANE_T,
1.6 nicm 44: cmd_capture_pane_exec
1.1 nicm 45: };
46:
1.23 nicm 47: char *
48: cmd_capture_pane_append(char *buf, size_t *len, char *line, size_t linelen)
49: {
1.29 nicm 50: buf = xrealloc(buf, *len + linelen + 1);
1.23 nicm 51: memcpy(buf + *len, line, linelen);
52: *len += linelen;
53: return (buf);
54: }
55:
56: char *
57: cmd_capture_pane_pending(struct args *args, struct window_pane *wp,
58: size_t *len)
59: {
1.32 nicm 60: struct evbuffer *pending;
61: char *buf, *line, tmp[5];
62: size_t linelen;
63: u_int i;
1.23 nicm 64:
1.32 nicm 65: pending = input_pending(wp);
66: if (pending == NULL)
1.23 nicm 67: return (xstrdup(""));
68:
1.32 nicm 69: line = EVBUFFER_DATA(pending);
70: linelen = EVBUFFER_LENGTH(pending);
1.23 nicm 71:
1.24 nicm 72: buf = xstrdup("");
1.23 nicm 73: if (args_has(args, 'C')) {
74: for (i = 0; i < linelen; i++) {
75: if (line[i] >= ' ') {
76: tmp[0] = line[i];
77: tmp[1] = '\0';
78: } else
1.31 nicm 79: xsnprintf(tmp, sizeof tmp, "\\%03hho", line[i]);
1.23 nicm 80: buf = cmd_capture_pane_append(buf, len, tmp,
81: strlen(tmp));
82: }
83: } else
84: buf = cmd_capture_pane_append(buf, len, line, linelen);
85: return (buf);
86: }
87:
88: char *
89: cmd_capture_pane_history(struct args *args, struct cmd_q *cmdq,
90: struct window_pane *wp, size_t *len)
1.1 nicm 91: {
1.7 nicm 92: struct grid *gd;
1.18 nicm 93: const struct grid_line *gl;
1.23 nicm 94: struct grid_cell *gc = NULL;
95: int n, with_codes, escape_c0, join_lines;
96: u_int i, sx, top, bottom, tmp;
97: char *cause, *buf, *line;
1.28 nicm 98: const char *Sflag, *Eflag;
1.23 nicm 99: size_t linelen;
1.14 nicm 100:
1.23 nicm 101: sx = screen_size_x(&wp->base);
1.20 nicm 102: if (args_has(args, 'a')) {
103: gd = wp->saved_grid;
1.23 nicm 104: if (gd == NULL) {
105: if (!args_has(args, 'q')) {
106: cmdq_error(cmdq, "no alternate screen");
107: return (NULL);
108: }
109: return (xstrdup(""));
1.20 nicm 110: }
1.23 nicm 111: } else
112: gd = wp->base.grid;
113:
1.28 nicm 114: Sflag = args_get(args, 'S');
115: if (Sflag != NULL && strcmp(Sflag, "-") == 0)
1.23 nicm 116: top = 0;
1.28 nicm 117: else {
118: n = args_strtonum(args, 'S', INT_MIN, SHRT_MAX, &cause);
119: if (cause != NULL) {
120: top = gd->hsize;
121: free(cause);
122: } else if (n < 0 && (u_int) -n > gd->hsize)
123: top = 0;
124: else
125: top = gd->hsize + n;
126: if (top > gd->hsize + gd->sy - 1)
127: top = gd->hsize + gd->sy - 1;
128: }
1.23 nicm 129:
1.28 nicm 130: Eflag = args_get(args, 'E');
131: if (Eflag != NULL && strcmp(Eflag, "-") == 0)
1.23 nicm 132: bottom = gd->hsize + gd->sy - 1;
1.28 nicm 133: else {
134: n = args_strtonum(args, 'E', INT_MIN, SHRT_MAX, &cause);
135: if (cause != NULL) {
136: bottom = gd->hsize + gd->sy - 1;
137: free(cause);
138: } else if (n < 0 && (u_int) -n > gd->hsize)
139: bottom = 0;
140: else
141: bottom = gd->hsize + n;
142: if (bottom > gd->hsize + gd->sy - 1)
143: bottom = gd->hsize + gd->sy - 1;
144: }
1.23 nicm 145:
146: if (bottom < top) {
147: tmp = bottom;
148: bottom = top;
149: top = tmp;
1.20 nicm 150: }
1.1 nicm 151:
1.23 nicm 152: with_codes = args_has(args, 'e');
153: escape_c0 = args_has(args, 'C');
154: join_lines = args_has(args, 'J');
155:
1.1 nicm 156: buf = NULL;
1.23 nicm 157: for (i = top; i <= bottom; i++) {
158: line = grid_string_cells(gd, 0, i, sx, &gc, with_codes,
159: escape_c0, !join_lines);
160: linelen = strlen(line);
161:
162: buf = cmd_capture_pane_append(buf, len, line, linelen);
163:
164: gl = grid_peek_line(gd, i);
165: if (!join_lines || !(gl->flags & GRID_LINE_WRAPPED))
166: buf[(*len)++] = '\n';
167:
168: free(line);
169: }
170: return (buf);
171: }
1.1 nicm 172:
1.23 nicm 173: enum cmd_retval
174: cmd_capture_pane_exec(struct cmd *self, struct cmd_q *cmdq)
175: {
176: struct args *args = self->args;
177: struct client *c;
1.36 ! nicm 178: struct window_pane *wp = cmdq->state.tflag.wp;
1.23 nicm 179: char *buf, *cause;
1.27 nicm 180: const char *bufname;
1.23 nicm 181: size_t len;
1.1 nicm 182:
1.23 nicm 183: len = 0;
184: if (args_has(args, 'P'))
185: buf = cmd_capture_pane_pending(args, wp, &len);
186: else
187: buf = cmd_capture_pane_history(args, cmdq, wp, &len);
188: if (buf == NULL)
189: return (CMD_RETURN_ERROR);
1.1 nicm 190:
1.14 nicm 191: if (args_has(args, 'p')) {
1.19 nicm 192: c = cmdq->client;
193: if (c == NULL ||
194: (c->session != NULL && !(c->flags & CLIENT_CONTROL))) {
195: cmdq_error(cmdq, "can't write to stdout");
1.33 nicm 196: free(buf);
1.14 nicm 197: return (CMD_RETURN_ERROR);
198: }
199: evbuffer_add(c->stdout_data, buf, len);
1.34 nicm 200: free(buf);
1.24 nicm 201: if (args_has(args, 'P') && len > 0)
202: evbuffer_add(c->stdout_data, "\n", 1);
1.35 nicm 203: server_client_push_stdout(c);
1.14 nicm 204: } else {
1.27 nicm 205: bufname = NULL;
206: if (args_has(args, 'b'))
207: bufname = args_get(args, 'b');
208:
209: if (paste_set(buf, len, bufname, &cause) != 0) {
210: cmdq_error(cmdq, "%s", cause);
1.33 nicm 211: free(cause);
1.14 nicm 212: free(buf);
213: return (CMD_RETURN_ERROR);
214: }
1.1 nicm 215: }
1.6 nicm 216:
1.12 nicm 217: return (CMD_RETURN_NORMAL);
1.1 nicm 218: }