Annotation of src/usr.bin/tmux/cmd-set-environment.c, Revision 1.2
1.2 ! nicm 1: /* $OpenBSD: cmd-set-environment.c,v 1.1 2009/08/08 21:52:43 nicm Exp $ */
1.1 nicm 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",
1.2 ! nicm 34: "[-gru] " CMD_TARGET_SESSION_USAGE " name [value]",
! 35: CMD_ARG12, CMD_CHFLAG('g')|CMD_CHFLAG('r')|CMD_CHFLAG('u'),
1.1 nicm 36: NULL,
1.2 ! nicm 37: cmd_target_parse,
1.1 nicm 38: cmd_set_environment_exec,
1.2 ! nicm 39: cmd_target_free,
! 40: cmd_target_print
1.1 nicm 41: };
42:
43: int
44: cmd_set_environment_exec(struct cmd *self, struct cmd_ctx *ctx)
45: {
1.2 ! nicm 46: struct cmd_target_data *data = self->data;
1.1 nicm 47: struct session *s;
48: struct environ *env;
49:
1.2 ! nicm 50: if (*data->arg == '\0') {
1.1 nicm 51: ctx->error(ctx, "empty variable name");
52: return (-1);
53: }
1.2 ! nicm 54: if (strchr(data->arg, '=') != NULL) {
1.1 nicm 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')) {
1.2 ! nicm 68: if (data->arg2 != NULL) {
1.1 nicm 69: ctx->error(ctx, "can't specify a value with -u");
70: return (-1);
71: }
1.2 ! nicm 72: environ_unset(env, data->arg);
1.1 nicm 73: } else if (data->chflags & CMD_CHFLAG('r')) {
1.2 ! nicm 74: if (data->arg2 != NULL) {
1.1 nicm 75: ctx->error(ctx, "can't specify a value with -r");
76: return (-1);
77: }
1.2 ! nicm 78: environ_set(env, data->arg, NULL);
1.1 nicm 79: } else {
1.2 ! nicm 80: if (data->arg2 == NULL) {
1.1 nicm 81: ctx->error(ctx, "no value specified");
82: return (-1);
83: }
1.2 ! nicm 84: environ_set(env, data->arg, data->arg2);
1.1 nicm 85: }
86:
87: return (0);
88: }