Return to cmd-pipe-pane.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / tmux

Annotation of embedaddon/tmux/cmd-pipe-pane.c, revision 1.1

1.1     ! misho       1: /* $OpenBSD$ */
        !             2: 
        !             3: /*
        !             4:  * Copyright (c) 2009 Nicholas Marriott <nicholas.marriott@gmail.com>
        !             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: #include <sys/socket.h>
        !            21: 
        !            22: #include <errno.h>
        !            23: #include <fcntl.h>
        !            24: #include <stdlib.h>
        !            25: #include <string.h>
        !            26: #include <time.h>
        !            27: #include <unistd.h>
        !            28: 
        !            29: #include "tmux.h"
        !            30: 
        !            31: /*
        !            32:  * Open pipe to redirect pane output. If already open, close first.
        !            33:  */
        !            34: 
        !            35: static enum cmd_retval cmd_pipe_pane_exec(struct cmd *, struct cmdq_item *);
        !            36: 
        !            37: static void cmd_pipe_pane_error_callback(struct bufferevent *, short, void *);
        !            38: 
        !            39: const struct cmd_entry cmd_pipe_pane_entry = {
        !            40:        .name = "pipe-pane",
        !            41:        .alias = "pipep",
        !            42: 
        !            43:        .args = { "ot:", 0, 1 },
        !            44:        .usage = "[-o] " CMD_TARGET_PANE_USAGE " [command]",
        !            45: 
        !            46:        .tflag = CMD_PANE,
        !            47: 
        !            48:        .flags = CMD_AFTERHOOK,
        !            49:        .exec = cmd_pipe_pane_exec
        !            50: };
        !            51: 
        !            52: static enum cmd_retval
        !            53: cmd_pipe_pane_exec(struct cmd *self, struct cmdq_item *item)
        !            54: {
        !            55:        struct args             *args = self->args;
        !            56:        struct client           *c = item->state.c;
        !            57:        struct window_pane      *wp = item->state.tflag.wp;
        !            58:        struct session          *s = item->state.tflag.s;
        !            59:        struct winlink          *wl = item->state.tflag.wl;
        !            60:        char                    *cmd;
        !            61:        int                      old_fd, pipe_fd[2], null_fd;
        !            62:        struct format_tree      *ft;
        !            63: 
        !            64:        /* Destroy the old pipe. */
        !            65:        old_fd = wp->pipe_fd;
        !            66:        if (wp->pipe_fd != -1) {
        !            67:                bufferevent_free(wp->pipe_event);
        !            68:                close(wp->pipe_fd);
        !            69:                wp->pipe_fd = -1;
        !            70:        }
        !            71: 
        !            72:        /* If no pipe command, that is enough. */
        !            73:        if (args->argc == 0 || *args->argv[0] == '\0')
        !            74:                return (CMD_RETURN_NORMAL);
        !            75: 
        !            76:        /*
        !            77:         * With -o, only open the new pipe if there was no previous one. This
        !            78:         * allows a pipe to be toggled with a single key, for example:
        !            79:         *
        !            80:         *      bind ^p pipep -o 'cat >>~/output'
        !            81:         */
        !            82:        if (args_has(self->args, 'o') && old_fd != -1)
        !            83:                return (CMD_RETURN_NORMAL);
        !            84: 
        !            85:        /* Open the new pipe. */
        !            86:        if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, pipe_fd) != 0) {
        !            87:                cmdq_error(item, "socketpair error: %s", strerror(errno));
        !            88:                return (CMD_RETURN_ERROR);
        !            89:        }
        !            90: 
        !            91:        /* Expand the command. */
        !            92:        ft = format_create(item, FORMAT_NONE, 0);
        !            93:        format_defaults(ft, c, s, wl, wp);
        !            94:        cmd = format_expand_time(ft, args->argv[0], time(NULL));
        !            95:        format_free(ft);
        !            96: 
        !            97:        /* Fork the child. */
        !            98:        switch (fork()) {
        !            99:        case -1:
        !           100:                cmdq_error(item, "fork error: %s", strerror(errno));
        !           101: 
        !           102:                free(cmd);
        !           103:                return (CMD_RETURN_ERROR);
        !           104:        case 0:
        !           105:                /* Child process. */
        !           106:                close(pipe_fd[0]);
        !           107:                clear_signals(1);
        !           108: 
        !           109:                if (dup2(pipe_fd[1], STDIN_FILENO) == -1)
        !           110:                        _exit(1);
        !           111:                if (pipe_fd[1] != STDIN_FILENO)
        !           112:                        close(pipe_fd[1]);
        !           113: 
        !           114:                null_fd = open(_PATH_DEVNULL, O_WRONLY, 0);
        !           115:                if (dup2(null_fd, STDOUT_FILENO) == -1)
        !           116:                        _exit(1);
        !           117:                if (dup2(null_fd, STDERR_FILENO) == -1)
        !           118:                        _exit(1);
        !           119:                if (null_fd != STDOUT_FILENO && null_fd != STDERR_FILENO)
        !           120:                        close(null_fd);
        !           121: 
        !           122:                closefrom(STDERR_FILENO + 1);
        !           123: 
        !           124:                execl(_PATH_BSHELL, "sh", "-c", cmd, (char *) NULL);
        !           125:                _exit(1);
        !           126:        default:
        !           127:                /* Parent process. */
        !           128:                close(pipe_fd[1]);
        !           129: 
        !           130:                wp->pipe_fd = pipe_fd[0];
        !           131:                wp->pipe_off = EVBUFFER_LENGTH(wp->event->input);
        !           132: 
        !           133:                wp->pipe_event = bufferevent_new(wp->pipe_fd,
        !           134:                    NULL, NULL, cmd_pipe_pane_error_callback, wp);
        !           135:                bufferevent_enable(wp->pipe_event, EV_WRITE);
        !           136: 
        !           137:                setblocking(wp->pipe_fd, 0);
        !           138: 
        !           139:                free(cmd);
        !           140:                return (CMD_RETURN_NORMAL);
        !           141:        }
        !           142: }
        !           143: 
        !           144: static void
        !           145: cmd_pipe_pane_error_callback(__unused struct bufferevent *bufev,
        !           146:     __unused short what, void *data)
        !           147: {
        !           148:        struct window_pane      *wp = data;
        !           149: 
        !           150:        bufferevent_free(wp->pipe_event);
        !           151:        close(wp->pipe_fd);
        !           152:        wp->pipe_fd = -1;
        !           153: }