Annotation of src/usr.bin/tmux/cmd-set-environment.c, Revision 1.1
1.1 ! nicm 1: /* $OpenBSD$ */
! 2:
! 3: /*
! 4: * Copyright (c) 2009 Nicholas Marriott <nicm@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:
! 21: #include <stdlib.h>
! 22: #include <string.h>
! 23:
! 24: #include "tmux.h"
! 25:
! 26: /*
! 27: * Set an environment variable.
! 28: */
! 29:
! 30: int cmd_set_environment_exec(struct cmd *, struct cmd_ctx *);
! 31:
! 32: const struct cmd_entry cmd_set_environment_entry = {
! 33: "set-environment", "setenv",
! 34: "[-gru] " CMD_OPTION_SESSION_USAGE,
! 35: 0, CMD_CHFLAG('g')|CMD_CHFLAG('r')|CMD_CHFLAG('u'),
! 36: NULL,
! 37: cmd_option_parse,
! 38: cmd_set_environment_exec,
! 39: cmd_option_free,
! 40: cmd_option_print
! 41: };
! 42:
! 43: int
! 44: cmd_set_environment_exec(struct cmd *self, struct cmd_ctx *ctx)
! 45: {
! 46: struct cmd_option_data *data = self->data;
! 47: struct session *s;
! 48: struct environ *env;
! 49:
! 50: if (*data->option == '\0') {
! 51: ctx->error(ctx, "empty variable name");
! 52: return (-1);
! 53: }
! 54: if (strchr(data->option, '=') != NULL) {
! 55: ctx->error(ctx, "variable name contains =");
! 56: return (-1);
! 57: }
! 58:
! 59: if (data->chflags & CMD_CHFLAG('g'))
! 60: env = &global_environ;
! 61: else {
! 62: if ((s = cmd_find_session(ctx, data->target)) == NULL)
! 63: return (-1);
! 64: env = &s->environ;
! 65: }
! 66:
! 67: if (data->chflags & CMD_CHFLAG('u')) {
! 68: if (data->value != NULL) {
! 69: ctx->error(ctx, "can't specify a value with -u");
! 70: return (-1);
! 71: }
! 72: environ_unset(env, data->option);
! 73: } else if (data->chflags & CMD_CHFLAG('r')) {
! 74: if (data->value != NULL) {
! 75: ctx->error(ctx, "can't specify a value with -r");
! 76: return (-1);
! 77: }
! 78: environ_set(env, data->option, NULL);
! 79: } else {
! 80: if (data->value == NULL) {
! 81: ctx->error(ctx, "no value specified");
! 82: return (-1);
! 83: }
! 84: environ_set(env, data->option, data->value);
! 85: }
! 86:
! 87: return (0);
! 88: }