Annotation of embedaddon/tmux/cmd-show-environment.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: 
        !            21: #include <stdlib.h>
        !            22: #include <string.h>
        !            23: 
        !            24: #include "tmux.h"
        !            25: 
        !            26: /*
        !            27:  * Show environment.
        !            28:  */
        !            29: 
        !            30: static enum cmd_retval cmd_show_environment_exec(struct cmd *,
        !            31:                            struct cmdq_item *);
        !            32: 
        !            33: static char    *cmd_show_environment_escape(struct environ_entry *);
        !            34: static void     cmd_show_environment_print(struct cmd *, struct cmdq_item *,
        !            35:                     struct environ_entry *);
        !            36: 
        !            37: const struct cmd_entry cmd_show_environment_entry = {
        !            38:        .name = "show-environment",
        !            39:        .alias = "showenv",
        !            40: 
        !            41:        .args = { "gst:", 0, 1 },
        !            42:        .usage = "[-gs] " CMD_TARGET_SESSION_USAGE " [name]",
        !            43: 
        !            44:        .tflag = CMD_SESSION_CANFAIL,
        !            45: 
        !            46:        .flags = CMD_AFTERHOOK,
        !            47:        .exec = cmd_show_environment_exec
        !            48: };
        !            49: 
        !            50: static char *
        !            51: cmd_show_environment_escape(struct environ_entry *envent)
        !            52: {
        !            53:        const char      *value = envent->value;
        !            54:        char             c, *out, *ret;
        !            55: 
        !            56:        out = ret = xmalloc(strlen(value) * 2 + 1); /* at most twice the size */
        !            57:        while ((c = *value++) != '\0') {
        !            58:                /* POSIX interprets $ ` " and \ in double quotes. */
        !            59:                if (c == '$' || c == '`' || c == '"' || c == '\\')
        !            60:                        *out++ = '\\';
        !            61:                *out++ = c;
        !            62:        }
        !            63:        *out = '\0';
        !            64: 
        !            65:        return (ret);
        !            66: }
        !            67: 
        !            68: static void
        !            69: cmd_show_environment_print(struct cmd *self, struct cmdq_item *item,
        !            70:     struct environ_entry *envent)
        !            71: {
        !            72:        char    *escaped;
        !            73: 
        !            74:        if (!args_has(self->args, 's')) {
        !            75:                if (envent->value != NULL)
        !            76:                        cmdq_print(item, "%s=%s", envent->name, envent->value);
        !            77:                else
        !            78:                        cmdq_print(item, "-%s", envent->name);
        !            79:                return;
        !            80:        }
        !            81: 
        !            82:        if (envent->value != NULL) {
        !            83:                escaped = cmd_show_environment_escape(envent);
        !            84:                cmdq_print(item, "%s=\"%s\"; export %s;", envent->name, escaped,
        !            85:                    envent->name);
        !            86:                free(escaped);
        !            87:        } else
        !            88:                cmdq_print(item, "unset %s;", envent->name);
        !            89: }
        !            90: 
        !            91: static enum cmd_retval
        !            92: cmd_show_environment_exec(struct cmd *self, struct cmdq_item *item)
        !            93: {
        !            94:        struct args             *args = self->args;
        !            95:        struct environ          *env;
        !            96:        struct environ_entry    *envent;
        !            97:        const char              *target;
        !            98: 
        !            99:        if ((target = args_get(args, 't')) != NULL) {
        !           100:                if (item->state.tflag.s == NULL) {
        !           101:                        cmdq_error(item, "no such session: %s", target);
        !           102:                        return (CMD_RETURN_ERROR);
        !           103:                }
        !           104:        }
        !           105: 
        !           106:        if (args_has(self->args, 'g'))
        !           107:                env = global_environ;
        !           108:        else {
        !           109:                if (item->state.tflag.s == NULL) {
        !           110:                        target = args_get(args, 't');
        !           111:                        if (target != NULL)
        !           112:                                cmdq_error(item, "no such session: %s", target);
        !           113:                        else
        !           114:                                cmdq_error(item, "no current session");
        !           115:                        return (CMD_RETURN_ERROR);
        !           116:                }
        !           117:                env = item->state.tflag.s->environ;
        !           118:        }
        !           119: 
        !           120:        if (args->argc != 0) {
        !           121:                envent = environ_find(env, args->argv[0]);
        !           122:                if (envent == NULL) {
        !           123:                        cmdq_error(item, "unknown variable: %s", args->argv[0]);
        !           124:                        return (CMD_RETURN_ERROR);
        !           125:                }
        !           126:                cmd_show_environment_print(self, item, envent);
        !           127:                return (CMD_RETURN_NORMAL);
        !           128:        }
        !           129: 
        !           130:        envent = environ_first(env);
        !           131:        while (envent != NULL) {
        !           132:                cmd_show_environment_print(self, item, envent);
        !           133:                envent = environ_next(envent);
        !           134:        }
        !           135:        return (CMD_RETURN_NORMAL);
        !           136: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>