Annotation of embedaddon/tmux/cmd-set-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: * Set an environment variable.
! 28: */
! 29:
! 30: static enum cmd_retval cmd_set_environment_exec(struct cmd *,
! 31: struct cmdq_item *);
! 32:
! 33: const struct cmd_entry cmd_set_environment_entry = {
! 34: .name = "set-environment",
! 35: .alias = "setenv",
! 36:
! 37: .args = { "grt:u", 1, 2 },
! 38: .usage = "[-gru] " CMD_TARGET_SESSION_USAGE " name [value]",
! 39:
! 40: .tflag = CMD_SESSION_CANFAIL,
! 41:
! 42: .flags = CMD_AFTERHOOK,
! 43: .exec = cmd_set_environment_exec
! 44: };
! 45:
! 46: static enum cmd_retval
! 47: cmd_set_environment_exec(struct cmd *self, struct cmdq_item *item)
! 48: {
! 49: struct args *args = self->args;
! 50: struct environ *env;
! 51: const char *name, *value, *target;
! 52:
! 53: name = args->argv[0];
! 54: if (*name == '\0') {
! 55: cmdq_error(item, "empty variable name");
! 56: return (CMD_RETURN_ERROR);
! 57: }
! 58: if (strchr(name, '=') != NULL) {
! 59: cmdq_error(item, "variable name contains =");
! 60: return (CMD_RETURN_ERROR);
! 61: }
! 62:
! 63: if (args->argc < 2)
! 64: value = NULL;
! 65: else
! 66: value = args->argv[1];
! 67:
! 68: if (args_has(self->args, 'g'))
! 69: env = global_environ;
! 70: else {
! 71: if (item->state.tflag.s == NULL) {
! 72: target = args_get(args, 't');
! 73: if (target != NULL)
! 74: cmdq_error(item, "no such session: %s", target);
! 75: else
! 76: cmdq_error(item, "no current session");
! 77: return (CMD_RETURN_ERROR);
! 78: }
! 79: env = item->state.tflag.s->environ;
! 80: }
! 81:
! 82: if (args_has(self->args, 'u')) {
! 83: if (value != NULL) {
! 84: cmdq_error(item, "can't specify a value with -u");
! 85: return (CMD_RETURN_ERROR);
! 86: }
! 87: environ_unset(env, name);
! 88: } else if (args_has(self->args, 'r')) {
! 89: if (value != NULL) {
! 90: cmdq_error(item, "can't specify a value with -r");
! 91: return (CMD_RETURN_ERROR);
! 92: }
! 93: environ_clear(env, name);
! 94: } else {
! 95: if (value == NULL) {
! 96: cmdq_error(item, "no value specified");
! 97: return (CMD_RETURN_ERROR);
! 98: }
! 99: environ_set(env, name, "%s", value);
! 100: }
! 101:
! 102: return (CMD_RETURN_NORMAL);
! 103: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>