Return to cmd-set-environment.c CVS log

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

tmux 2.4

    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: }