| File: | d/tclenv.c |
| Warning: | line 365, column 12 Potential leak of memory pointed to by 's' |
Press '?' to see keyboard shortcuts
Keyboard shortcuts:
| 1 | /* | |||
| 2 | * The contents of this file are subject to the Mozilla Public License | |||
| 3 | * Version 1.1 (the "License"); you may not use this file except in | |||
| 4 | * compliance with the License. You may obtain a copy of the License at | |||
| 5 | * http://mozilla.org/. | |||
| 6 | * | |||
| 7 | * Software distributed under the License is distributed on an "AS IS" | |||
| 8 | * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See | |||
| 9 | * the License for the specific language governing rights and limitations | |||
| 10 | * under the License. | |||
| 11 | * | |||
| 12 | * The Original Code is AOLserver Code and related documentation | |||
| 13 | * distributed by AOL. | |||
| 14 | * | |||
| 15 | * The Initial Developer of the Original Code is America Online, | |||
| 16 | * Inc. Portions created by AOL are Copyright (C) 1999 America Online, | |||
| 17 | * Inc. All Rights Reserved. | |||
| 18 | * | |||
| 19 | * Alternatively, the contents of this file may be used under the terms | |||
| 20 | * of the GNU General Public License (the "GPL"), in which case the | |||
| 21 | * provisions of GPL are applicable instead of those above. If you wish | |||
| 22 | * to allow use of your version of this file only under the terms of the | |||
| 23 | * GPL and not to allow others to use your version of this file under the | |||
| 24 | * License, indicate your decision by deleting the provisions above and | |||
| 25 | * replace them with the notice and other provisions required by the GPL. | |||
| 26 | * If you do not delete the provisions above, a recipient may use your | |||
| 27 | * version of this file under either the License or the GPL. | |||
| 28 | */ | |||
| 29 | ||||
| 30 | ||||
| 31 | /* | |||
| 32 | * tclenv.c -- | |||
| 33 | * | |||
| 34 | * Implements "ns_env". | |||
| 35 | */ | |||
| 36 | ||||
| 37 | #include "nsd.h" | |||
| 38 | ||||
| 39 | #ifdef HAVE__NSGETENVIRON | |||
| 40 | # include <crt_externs.h> | |||
| 41 | #elif !defined(_WIN32) | |||
| 42 | extern char **environ; | |||
| 43 | #endif | |||
| 44 | ||||
| 45 | /* | |||
| 46 | * Local functions defined in this file. | |||
| 47 | */ | |||
| 48 | ||||
| 49 | static int PutEnv(Tcl_Interp *interp, const char *name, const char *value); | |||
| 50 | ||||
| 51 | /* | |||
| 52 | * Loca variables defined in this file. | |||
| 53 | */ | |||
| 54 | ||||
| 55 | static Ns_Mutex lock = NULL((void*)0); | |||
| 56 | ||||
| 57 | ||||
| 58 | /* | |||
| 59 | *---------------------------------------------------------------------- | |||
| 60 | * | |||
| 61 | * NsInitTclEnv -- | |||
| 62 | * | |||
| 63 | * Global initialization for tasks. | |||
| 64 | * | |||
| 65 | * Results: | |||
| 66 | * None. | |||
| 67 | * | |||
| 68 | * Side effects: | |||
| 69 | * None. | |||
| 70 | * | |||
| 71 | *---------------------------------------------------------------------- | |||
| 72 | */ | |||
| 73 | ||||
| 74 | void | |||
| 75 | NsInitTclEnv(void) | |||
| 76 | { | |||
| 77 | static bool_Bool initialized = NS_FALSE0; | |||
| 78 | ||||
| 79 | if (!initialized) { | |||
| 80 | Ns_MutexInit(&lock); | |||
| 81 | Ns_MutexSetName(&lock, "ns:env"); | |||
| 82 | initialized = NS_TRUE1; | |||
| 83 | } | |||
| 84 | } | |||
| 85 | ||||
| 86 | /* | |||
| 87 | *---------------------------------------------------------------------- | |||
| 88 | * | |||
| 89 | * Ns_GetEnviron -- | |||
| 90 | * | |||
| 91 | * Return the environment vector. | |||
| 92 | * | |||
| 93 | * Results: | |||
| 94 | * Pointer to environment. | |||
| 95 | * | |||
| 96 | * Side effects: | |||
| 97 | * None. | |||
| 98 | * | |||
| 99 | *---------------------------------------------------------------------- | |||
| 100 | */ | |||
| 101 | ||||
| 102 | char ** | |||
| 103 | Ns_GetEnviron(void) | |||
| 104 | { | |||
| 105 | char **envp; | |||
| 106 | ||||
| 107 | #ifdef HAVE__NSGETENVIRON | |||
| 108 | envp = *_NSGetEnviron(); | |||
| 109 | #else | |||
| 110 | envp = environ; | |||
| 111 | #endif | |||
| 112 | return envp; | |||
| 113 | } | |||
| 114 | ||||
| 115 | ||||
| 116 | /* | |||
| 117 | *---------------------------------------------------------------------- | |||
| 118 | * | |||
| 119 | * Ns_CopyEnviron -- | |||
| 120 | * | |||
| 121 | * Copy the environment to the given dstring along with | |||
| 122 | * an argv vector. | |||
| 123 | * | |||
| 124 | * Results: | |||
| 125 | * Pointer to dsPtr->string. | |||
| 126 | * | |||
| 127 | * Side effects: | |||
| 128 | * None. | |||
| 129 | * | |||
| 130 | *---------------------------------------------------------------------- | |||
| 131 | */ | |||
| 132 | ||||
| 133 | char ** | |||
| 134 | Ns_CopyEnviron(Ns_DStringTcl_DString *dsPtr) | |||
| 135 | { | |||
| 136 | char *const*envp; | |||
| 137 | int i; | |||
| 138 | ||||
| 139 | NS_NONNULL_ASSERT(dsPtr != NULL)((void) (0)); | |||
| 140 | ||||
| 141 | Ns_MutexLock(&lock); | |||
| 142 | envp = Ns_GetEnviron(); | |||
| 143 | for (i = 0; envp[i] != NULL((void*)0); ++i) { | |||
| 144 | Ns_DStringAppendArg(dsPtr, envp[i]); | |||
| 145 | } | |||
| 146 | Ns_MutexUnlock(&lock); | |||
| 147 | ||||
| 148 | return Ns_DStringAppendArgv(dsPtr); | |||
| 149 | } | |||
| 150 | ||||
| 151 | ||||
| 152 | /* | |||
| 153 | *---------------------------------------------------------------------- | |||
| 154 | * | |||
| 155 | * NsTclEnvObjCmd -- | |||
| 156 | * | |||
| 157 | * Implements "ns_env". No attempt is made to avoid the | |||
| 158 | * race condition between finding a variable and using it as it is | |||
| 159 | * assumed the environment would only be modified, if ever, at | |||
| 160 | * startup. | |||
| 161 | * | |||
| 162 | * Results: | |||
| 163 | * Tcl result. | |||
| 164 | * | |||
| 165 | * Side effects: | |||
| 166 | * Environment variables may be updated. | |||
| 167 | * | |||
| 168 | *---------------------------------------------------------------------- | |||
| 169 | */ | |||
| 170 | ||||
| 171 | int | |||
| 172 | NsTclEnvObjCmd(ClientData UNUSED(clientData)UNUSED_clientData __attribute__((__unused__)), Tcl_Interp *interp, int objc, Tcl_Obj *const* objv) | |||
| 173 | { | |||
| 174 | int result, opt; | |||
| 175 | static const char *const opts[] = { | |||
| 176 | "exists", "names", "get", "set", "unset", NULL((void*)0) | |||
| 177 | }; | |||
| 178 | enum { | |||
| 179 | IExistsIdx, INamesIdx, IGetIdx, ISetIdx, IUnsetIdx | |||
| 180 | }; | |||
| 181 | ||||
| 182 | if (objc < 2) { | |||
| ||||
| 183 | Tcl_WrongNumArgs(interp, 1, objv, "command ?args ...?"); | |||
| 184 | result = TCL_ERROR1; | |||
| 185 | ||||
| 186 | } else if (Tcl_GetIndexFromObj(interp, objv[1], opts, "command", 0,Tcl_GetIndexFromObjStruct(interp, objv[1], opts, sizeof(char * ), "command", 0, &opt) | |||
| 187 | &opt)Tcl_GetIndexFromObjStruct(interp, objv[1], opts, sizeof(char * ), "command", 0, &opt) != TCL_OK0) { | |||
| 188 | result = TCL_ERROR1; | |||
| 189 | ||||
| 190 | } else { | |||
| 191 | const char *name, *value; | |||
| 192 | char *const *envp; | |||
| 193 | Tcl_Obj *resultObj; | |||
| 194 | int i; | |||
| 195 | ||||
| 196 | result = TCL_OK0; | |||
| 197 | Ns_MutexLock(&lock); | |||
| 198 | ||||
| 199 | switch (opt) { | |||
| 200 | case IExistsIdx: | |||
| 201 | if (objc != 3) { | |||
| 202 | Tcl_WrongNumArgs(interp, 2, objv, "name"); | |||
| 203 | result = TCL_ERROR1; | |||
| 204 | } else { | |||
| 205 | Tcl_SetObjResult(interp, Tcl_NewBooleanObj((getenv(Tcl_GetString(objv[2])) != NULL) ? 1 : 0)Tcl_NewIntObj(((getenv(Tcl_GetString(objv[2])) != ((void*)0)) ? 1 : 0)!=0)); | |||
| 206 | } | |||
| 207 | break; | |||
| 208 | ||||
| 209 | case INamesIdx: | |||
| 210 | envp = Ns_GetEnviron(); | |||
| 211 | resultObj = Tcl_GetObjResult(interp); | |||
| 212 | for (i = 0; envp[i] != NULL((void*)0); ++i) { | |||
| 213 | Tcl_Obj *obj; | |||
| 214 | ||||
| 215 | name = envp[i]; | |||
| 216 | value = strchr(name, INTCHAR('=')((int)((unsigned char)(('='))))); | |||
| 217 | obj = Tcl_NewStringObj(name, (value != NULL((void*)0)) ? (int)(value - name) : -1); | |||
| 218 | if (Tcl_ListObjAppendElement(interp, resultObj, obj) != TCL_OK0) { | |||
| 219 | result = TCL_ERROR1; | |||
| 220 | break; | |||
| 221 | } | |||
| 222 | } | |||
| 223 | break; | |||
| 224 | ||||
| 225 | case ISetIdx: | |||
| 226 | if (objc != 4) { | |||
| 227 | Tcl_WrongNumArgs(interp, 2, objv, "name value"); | |||
| 228 | result = TCL_ERROR1; | |||
| 229 | ||||
| 230 | } else if (PutEnv(interp, Tcl_GetString(objv[2]), Tcl_GetString(objv[3])) != TCL_OK0) { | |||
| 231 | result = TCL_ERROR1; | |||
| 232 | } | |||
| 233 | break; | |||
| 234 | ||||
| 235 | case IGetIdx: | |||
| 236 | case IUnsetIdx: | |||
| 237 | if (objc != 3 && objc != 4) { | |||
| 238 | Tcl_WrongNumArgs(interp, 2, objv, "?-nocomplain? name"); | |||
| 239 | result = TCL_ERROR1; | |||
| 240 | ||||
| 241 | } else if (objc == 4) { | |||
| 242 | const char *arg = Tcl_GetString(objv[2]); | |||
| 243 | ||||
| 244 | if (!STREQ(arg, "-nocomplain")(((*(arg)) == (*("-nocomplain"))) && (strcmp((arg),("-nocomplain" )) == 0))) { | |||
| 245 | Tcl_WrongNumArgs(interp, 2, objv, "?-nocomplain? name"); | |||
| 246 | result = TCL_ERROR1; | |||
| 247 | } | |||
| 248 | } | |||
| 249 | ||||
| 250 | if (result == TCL_OK0) { | |||
| 251 | name = Tcl_GetString(objv[2]); | |||
| 252 | value = getenv(name); | |||
| 253 | if (value == NULL((void*)0) && objc != 4) { | |||
| 254 | Ns_TclPrintfResult(interp, "no such environment variable: %s", name); | |||
| 255 | result = TCL_ERROR1; | |||
| 256 | ||||
| 257 | } else if ((opt == IUnsetIdx) && (PutEnv(interp, name, NULL((void*)0)) != TCL_OK0)) { | |||
| 258 | result = TCL_ERROR1; | |||
| 259 | ||||
| 260 | } else { | |||
| 261 | Tcl_SetObjResult(interp, Tcl_NewStringObj(value, -1)); | |||
| 262 | } | |||
| 263 | } | |||
| 264 | break; | |||
| 265 | ||||
| 266 | default: | |||
| 267 | /* unexpected value */ | |||
| 268 | assert(opt && 0)((void) (0)); | |||
| 269 | break; | |||
| 270 | } | |||
| 271 | ||||
| 272 | Ns_MutexUnlock(&lock); | |||
| 273 | } | |||
| 274 | return result; | |||
| 275 | } | |||
| 276 | ||||
| 277 | ||||
| 278 | /* | |||
| 279 | *---------------------------------------------------------------------- | |||
| 280 | * | |||
| 281 | * PutEnv -- | |||
| 282 | * | |||
| 283 | * NsTclEnvObjCmd helper routine to update an environment variable. | |||
| 284 | * | |||
| 285 | * Results: | |||
| 286 | * TCL_OK or TCL_ERROR. | |||
| 287 | * | |||
| 288 | * Side effects: | |||
| 289 | * Environment variable is set. | |||
| 290 | * | |||
| 291 | *---------------------------------------------------------------------- | |||
| 292 | */ | |||
| 293 | ||||
| 294 | static int | |||
| 295 | PutEnv(Tcl_Interp *interp, const char *name, const char *value) | |||
| 296 | { | |||
| 297 | char *s; | |||
| 298 | size_t len, nameLength, valueLength; | |||
| 299 | int result = TCL_OK0; | |||
| 300 | ||||
| 301 | #ifdef HAVE_UNSETENV1 | |||
| 302 | if (value == NULL((void*)0)) { | |||
| 303 | unsetenv(name); | |||
| 304 | return result; | |||
| 305 | } | |||
| 306 | #endif | |||
| 307 | ||||
| 308 | /* | |||
| 309 | * In case we have no unsetenv(), we have to deal with the value==NULL | |||
| 310 | * case here. If the value is not NULL, valueLength contains the | |||
| 311 | * terminating NUL character. | |||
| 312 | */ | |||
| 313 | len = nameLength = strlen(name); | |||
| 314 | if (value
| |||
| 315 | valueLength = strlen(value) + 1; | |||
| 316 | len += valueLength + 1u; | |||
| 317 | } else { | |||
| 318 | len += 1u; | |||
| 319 | valueLength = 0u; | |||
| 320 | } | |||
| 321 | ||||
| 322 | /* | |||
| 323 | * Use malloc() directly (and not ns_malloc()) | |||
| 324 | * as putenv() expects. | |||
| 325 | */ | |||
| 326 | s = malloc(len + 1u); | |||
| 327 | if (s == NULL((void*)0)) { | |||
| 328 | Ns_TclPrintfResult(interp, "could not allocate memory for new env entry"); | |||
| 329 | result = TCL_ERROR1; | |||
| 330 | } else { | |||
| 331 | ||||
| 332 | /* | |||
| 333 | * This complication for value == NULL below is needed on | |||
| 334 | * some platforms (Solaris) which do not have unsetenv() | |||
| 335 | * and are picky if we try to pass a value to putenv not | |||
| 336 | * conforming to the "name=value" format. | |||
| 337 | * | |||
| 338 | * This trick will of course work only for platforms which | |||
| 339 | * conform to Single Unix Spec and actually uses the storage | |||
| 340 | * passed to putenv() to hold the environ entry. | |||
| 341 | * However, there are some libc implementations (notably | |||
| 342 | * recent BSDs) that do not obey SUS but copy the presented | |||
| 343 | * string. This method fails on such platforms. | |||
| 344 | */ | |||
| 345 | memcpy(s, name, nameLength); | |||
| 346 | *(s + nameLength) = '='; | |||
| 347 | ||||
| 348 | if (valueLength
| |||
| 349 | /* | |||
| 350 | * Copy the value including the terminatig NUL character. | |||
| 351 | */ | |||
| 352 | memcpy(s + nameLength + 1, value, valueLength); | |||
| 353 | } else { | |||
| 354 | *(s + nameLength + 1u) = '\0'; | |||
| 355 | } | |||
| 356 | ||||
| 357 | if (putenv(s) != 0) { | |||
| 358 | Ns_TclPrintfResult(interp, "could not put environment entry \"%s\": %s", | |||
| 359 | s, Tcl_PosixError(interp)); | |||
| 360 | free(s); | |||
| 361 | result = TCL_ERROR1; | |||
| 362 | } | |||
| 363 | } | |||
| 364 | ||||
| 365 | return result; | |||
| ||||
| 366 | } | |||
| 367 | ||||
| 368 | /* | |||
| 369 | * Local Variables: | |||
| 370 | * mode: c | |||
| 371 | * c-basic-offset: 4 | |||
| 372 | * fill-column: 78 | |||
| 373 | * indent-tabs-mode: nil | |||
| 374 | * End: | |||
| 375 | */ |