[BACK]Return to cmd-capture-pane.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / tmux

Annotation of src/usr.bin/tmux/cmd-capture-pane.c, Revision 1.26

1.26    ! nicm        1: /* $OpenBSD: cmd-capture-pane.c,v 1.25 2013/10/10 12:00:18 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,
                     41:        "[-aCeJpPq] [-b buffer-index] [-E end-line] [-S start-line]"
1.13      nicm       42:        CMD_TARGET_PANE_USAGE,
1.6       nicm       43:        0,
                     44:        NULL,
                     45:        cmd_capture_pane_exec
1.1       nicm       46: };
                     47:
1.23      nicm       48: char *
                     49: cmd_capture_pane_append(char *buf, size_t *len, char *line, size_t linelen)
                     50: {
                     51:        buf = xrealloc(buf, 1, *len + linelen + 1);
                     52:        memcpy(buf + *len, line, linelen);
                     53:        *len += linelen;
                     54:        return (buf);
                     55: }
                     56:
                     57: char *
                     58: cmd_capture_pane_pending(struct args *args, struct window_pane *wp,
                     59:     size_t *len)
                     60: {
                     61:        char    *buf, *line, tmp[5];
                     62:        size_t   linelen;
                     63:        u_int    i;
                     64:
                     65:        if (wp->ictx.since_ground == NULL)
                     66:                return (xstrdup(""));
                     67:
                     68:        line = EVBUFFER_DATA(wp->ictx.since_ground);
                     69:        linelen = EVBUFFER_LENGTH(wp->ictx.since_ground);
                     70:
1.24      nicm       71:        buf = xstrdup("");
1.23      nicm       72:        if (args_has(args, 'C')) {
                     73:                for (i = 0; i < linelen; i++) {
                     74:                        if (line[i] >= ' ') {
                     75:                                tmp[0] = line[i];
                     76:                                tmp[1] = '\0';
                     77:                        } else
                     78:                                xsnprintf(tmp, sizeof tmp, "\\%03o", line[i]);
                     79:                        buf = cmd_capture_pane_append(buf, len, tmp,
                     80:                            strlen(tmp));
                     81:                }
                     82:        } else
                     83:                buf = cmd_capture_pane_append(buf, len, line, linelen);
                     84:        return (buf);
                     85: }
                     86:
                     87: char *
                     88: cmd_capture_pane_history(struct args *args, struct cmd_q *cmdq,
                     89:     struct window_pane *wp, size_t *len)
1.1       nicm       90: {
1.7       nicm       91:        struct grid             *gd;
1.18      nicm       92:        const struct grid_line  *gl;
1.23      nicm       93:        struct grid_cell        *gc = NULL;
                     94:        int                      n, with_codes, escape_c0, join_lines;
                     95:        u_int                    i, sx, top, bottom, tmp;
                     96:        char                    *cause, *buf, *line;
                     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:
                    112:        n = args_strtonum(args, 'S', INT_MIN, SHRT_MAX, &cause);
                    113:        if (cause != NULL) {
                    114:                top = gd->hsize;
                    115:                free(cause);
                    116:        } else if (n < 0 && (u_int) -n > gd->hsize)
                    117:                top = 0;
                    118:        else
                    119:                top = gd->hsize + n;
                    120:        if (top > gd->hsize + gd->sy - 1)
                    121:                top = gd->hsize + gd->sy - 1;
                    122:
                    123:        n = args_strtonum(args, 'E', INT_MIN, SHRT_MAX, &cause);
                    124:        if (cause != NULL) {
                    125:                bottom = gd->hsize + gd->sy - 1;
                    126:                free(cause);
                    127:        } else if (n < 0 && (u_int) -n > gd->hsize)
                    128:                bottom = 0;
                    129:        else
                    130:                bottom = gd->hsize + n;
                    131:        if (bottom > gd->hsize + gd->sy - 1)
                    132:                bottom = gd->hsize + gd->sy - 1;
                    133:
                    134:        if (bottom < top) {
                    135:                tmp = bottom;
                    136:                bottom = top;
                    137:                top = tmp;
1.20      nicm      138:        }
1.1       nicm      139:
1.23      nicm      140:        with_codes = args_has(args, 'e');
                    141:        escape_c0 = args_has(args, 'C');
                    142:        join_lines = args_has(args, 'J');
                    143:
1.1       nicm      144:        buf = NULL;
1.23      nicm      145:        for (i = top; i <= bottom; i++) {
                    146:                line = grid_string_cells(gd, 0, i, sx, &gc, with_codes,
                    147:                    escape_c0, !join_lines);
                    148:                linelen = strlen(line);
                    149:
                    150:                buf = cmd_capture_pane_append(buf, len, line, linelen);
                    151:
                    152:                gl = grid_peek_line(gd, i);
                    153:                if (!join_lines || !(gl->flags & GRID_LINE_WRAPPED))
                    154:                        buf[(*len)++] = '\n';
                    155:
                    156:                free(line);
                    157:        }
                    158:        return (buf);
                    159: }
1.1       nicm      160:
1.23      nicm      161: enum cmd_retval
                    162: cmd_capture_pane_exec(struct cmd *self, struct cmd_q *cmdq)
                    163: {
                    164:        struct args             *args = self->args;
                    165:        struct client           *c;
                    166:        struct window_pane      *wp;
                    167:        char                    *buf, *cause;
                    168:        int                      buffer;
                    169:        u_int                    limit;
                    170:        size_t                   len;
1.7       nicm      171:
1.23      nicm      172:        if (cmd_find_pane(cmdq, args_get(args, 't'), NULL, &wp) == NULL)
                    173:                return (CMD_RETURN_ERROR);
1.1       nicm      174:
1.23      nicm      175:        len = 0;
                    176:        if (args_has(args, 'P'))
                    177:                buf = cmd_capture_pane_pending(args, wp, &len);
                    178:        else
                    179:                buf = cmd_capture_pane_history(args, cmdq, wp, &len);
                    180:        if (buf == NULL)
                    181:                return (CMD_RETURN_ERROR);
1.1       nicm      182:
1.14      nicm      183:        if (args_has(args, 'p')) {
1.19      nicm      184:                c = cmdq->client;
                    185:                if (c == NULL ||
                    186:                    (c->session != NULL && !(c->flags & CLIENT_CONTROL))) {
                    187:                        cmdq_error(cmdq, "can't write to stdout");
1.14      nicm      188:                        return (CMD_RETURN_ERROR);
                    189:                }
                    190:                evbuffer_add(c->stdout_data, buf, len);
1.24      nicm      191:                if (args_has(args, 'P') && len > 0)
                    192:                    evbuffer_add(c->stdout_data, "\n", 1);
1.14      nicm      193:                server_push_stdout(c);
                    194:        } else {
                    195:                limit = options_get_number(&global_options, "buffer-limit");
                    196:                if (!args_has(args, 'b')) {
1.26    ! nicm      197:                        paste_add(buf, len, limit);
1.14      nicm      198:                        return (CMD_RETURN_NORMAL);
                    199:                }
                    200:
                    201:                buffer = args_strtonum(args, 'b', 0, INT_MAX, &cause);
                    202:                if (cause != NULL) {
1.19      nicm      203:                        cmdq_error(cmdq, "buffer %s", cause);
1.14      nicm      204:                        free(buf);
                    205:                        free(cause);
                    206:                        return (CMD_RETURN_ERROR);
                    207:                }
                    208:
1.26    ! nicm      209:                if (paste_replace(buffer, buf, len) != 0) {
1.19      nicm      210:                        cmdq_error(cmdq, "no buffer %d", buffer);
1.14      nicm      211:                        free(buf);
                    212:                        return (CMD_RETURN_ERROR);
                    213:                }
1.1       nicm      214:        }
1.6       nicm      215:
1.12      nicm      216:        return (CMD_RETURN_NORMAL);
1.1       nicm      217: }