1 /* $OpenBSD: cmd-show-environment.c,v 1.23 2017/04/22 10:22:39 nicm Exp $ */ 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 * Show environment. 28 */ 29 30 static enum cmd_retval cmd_show_environment_exec(struct cmd *, 31 struct cmdq_item *); 32 33 static char *cmd_show_environment_escape(struct environ_entry *); 34 static void cmd_show_environment_print(struct cmd *, struct cmdq_item *, 35 struct environ_entry *); 36 37 const 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 50 static char * 51 cmd_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 68 static void 69 cmd_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 91 static enum cmd_retval 92 cmd_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