Annotation of src/usr.bin/tmux/cmd-capture-pane.c, Revision 1.30
1.30 ! nicm 1: /* $OpenBSD: cmd-capture-pane.c,v 1.29 2014/10/08 17:35:58 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.6 nicm 43: 0,
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: {
60: char *buf, *line, tmp[5];
61: size_t linelen;
62: u_int i;
63:
64: if (wp->ictx.since_ground == NULL)
65: return (xstrdup(""));
66:
67: line = EVBUFFER_DATA(wp->ictx.since_ground);
68: linelen = EVBUFFER_LENGTH(wp->ictx.since_ground);
69:
1.24 nicm 70: buf = xstrdup("");
1.23 nicm 71: if (args_has(args, 'C')) {
72: for (i = 0; i < linelen; i++) {
73: if (line[i] >= ' ') {
74: tmp[0] = line[i];
75: tmp[1] = '\0';
76: } else
77: xsnprintf(tmp, sizeof tmp, "\\%03o", line[i]);
78: buf = cmd_capture_pane_append(buf, len, tmp,
79: strlen(tmp));
80: }
81: } else
82: buf = cmd_capture_pane_append(buf, len, line, linelen);
83: return (buf);
84: }
85:
86: char *
87: cmd_capture_pane_history(struct args *args, struct cmd_q *cmdq,
88: struct window_pane *wp, size_t *len)
1.1 nicm 89: {
1.7 nicm 90: struct grid *gd;
1.18 nicm 91: const struct grid_line *gl;
1.23 nicm 92: struct grid_cell *gc = NULL;
93: int n, with_codes, escape_c0, join_lines;
94: u_int i, sx, top, bottom, tmp;
95: char *cause, *buf, *line;
1.28 nicm 96: const char *Sflag, *Eflag;
1.23 nicm 97: size_t linelen;
1.14 nicm 98:
1.23 nicm 99: sx = screen_size_x(&wp->base);
1.20 nicm 100: if (args_has(args, 'a')) {
101: gd = wp->saved_grid;
1.23 nicm 102: if (gd == NULL) {
103: if (!args_has(args, 'q')) {
104: cmdq_error(cmdq, "no alternate screen");
105: return (NULL);
106: }
107: return (xstrdup(""));
1.20 nicm 108: }
1.23 nicm 109: } else
110: gd = wp->base.grid;
111:
1.28 nicm 112: Sflag = args_get(args, 'S');
113: if (Sflag != NULL && strcmp(Sflag, "-") == 0)
1.23 nicm 114: top = 0;
1.28 nicm 115: else {
116: n = args_strtonum(args, 'S', INT_MIN, SHRT_MAX, &cause);
117: if (cause != NULL) {
118: top = gd->hsize;
119: free(cause);
120: } else if (n < 0 && (u_int) -n > gd->hsize)
121: top = 0;
122: else
123: top = gd->hsize + n;
124: if (top > gd->hsize + gd->sy - 1)
125: top = gd->hsize + gd->sy - 1;
126: }
1.23 nicm 127:
1.28 nicm 128: Eflag = args_get(args, 'E');
129: if (Eflag != NULL && strcmp(Eflag, "-") == 0)
1.23 nicm 130: bottom = gd->hsize + gd->sy - 1;
1.28 nicm 131: else {
132: n = args_strtonum(args, 'E', INT_MIN, SHRT_MAX, &cause);
133: if (cause != NULL) {
134: bottom = gd->hsize + gd->sy - 1;
135: free(cause);
136: } else if (n < 0 && (u_int) -n > gd->hsize)
137: bottom = 0;
138: else
139: bottom = gd->hsize + n;
140: if (bottom > gd->hsize + gd->sy - 1)
141: bottom = gd->hsize + gd->sy - 1;
142: }
1.23 nicm 143:
144: if (bottom < top) {
145: tmp = bottom;
146: bottom = top;
147: top = tmp;
1.20 nicm 148: }
1.1 nicm 149:
1.23 nicm 150: with_codes = args_has(args, 'e');
151: escape_c0 = args_has(args, 'C');
152: join_lines = args_has(args, 'J');
153:
1.1 nicm 154: buf = NULL;
1.23 nicm 155: for (i = top; i <= bottom; i++) {
156: line = grid_string_cells(gd, 0, i, sx, &gc, with_codes,
157: escape_c0, !join_lines);
158: linelen = strlen(line);
159:
160: buf = cmd_capture_pane_append(buf, len, line, linelen);
161:
162: gl = grid_peek_line(gd, i);
163: if (!join_lines || !(gl->flags & GRID_LINE_WRAPPED))
164: buf[(*len)++] = '\n';
165:
166: free(line);
167: }
168: return (buf);
169: }
1.1 nicm 170:
1.23 nicm 171: enum cmd_retval
172: cmd_capture_pane_exec(struct cmd *self, struct cmd_q *cmdq)
173: {
174: struct args *args = self->args;
175: struct client *c;
176: struct window_pane *wp;
177: char *buf, *cause;
1.27 nicm 178: const char *bufname;
1.23 nicm 179: size_t len;
1.7 nicm 180:
1.23 nicm 181: if (cmd_find_pane(cmdq, args_get(args, 't'), NULL, &wp) == NULL)
182: return (CMD_RETURN_ERROR);
1.1 nicm 183:
1.23 nicm 184: len = 0;
185: if (args_has(args, 'P'))
186: buf = cmd_capture_pane_pending(args, wp, &len);
187: else
188: buf = cmd_capture_pane_history(args, cmdq, wp, &len);
189: if (buf == NULL)
190: return (CMD_RETURN_ERROR);
1.1 nicm 191:
1.14 nicm 192: if (args_has(args, 'p')) {
1.19 nicm 193: c = cmdq->client;
194: if (c == NULL ||
195: (c->session != NULL && !(c->flags & CLIENT_CONTROL))) {
196: cmdq_error(cmdq, "can't write to stdout");
1.14 nicm 197: return (CMD_RETURN_ERROR);
198: }
199: evbuffer_add(c->stdout_data, buf, len);
1.24 nicm 200: if (args_has(args, 'P') && len > 0)
201: evbuffer_add(c->stdout_data, "\n", 1);
1.14 nicm 202: server_push_stdout(c);
203: } else {
204:
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.14 nicm 211: free(buf);
212: free(cause);
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: }