1/* $OpenBSD$ */
2
3/*
4 * Copyright (c) 2009 Nicholas Marriott <[email protected]>
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
30static enum cmd_retval cmd_show_environment_exec(struct cmd *,
31 struct cmdq_item *);
32
33static char *cmd_show_environment_escape(struct environ_entry *);
34static void cmd_show_environment_print(struct cmd *, struct cmdq_item *,
35 struct environ_entry *);
36
37const 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 .target = { 't', CMD_FIND_SESSION, CMD_FIND_CANFAIL },
45
46 .flags = CMD_AFTERHOOK,
47 .exec = cmd_show_environment_exec
48};
49
50static char *
51cmd_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
68static void
69cmd_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
91static enum cmd_retval
92cmd_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->target.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->target.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->target.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}
137